// Copyright © 2019 Ettore Di Giacinto <mudler@gentoo.org>
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, see <http://www.gnu.org/licenses/>.

// Recipe is a builder imeplementation.

// It reads a Tree and spit it in human readable form (YAML), called recipe,
// It also loads a tree (recipe) from a YAML (to a db, e.g. BoltDB), allowing to query it
// with the solver, using the package object.
package tree_test

import (
	"io/ioutil"
	"os"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"

	pkg "github.com/mudler/luet/pkg/package"
	"github.com/mudler/luet/pkg/solver"
	. "github.com/mudler/luet/pkg/tree"
)

var _ = Describe("Tree", func() {

	Context("Simple solving with the fixture tree", func() {
		It("writes and reads back the same tree", func() {
			for index := 0; index < 300; index++ { // Just to make sure we don't have false positives
				db := pkg.NewInMemoryDatabase(false)
				generalRecipe := NewCompilerRecipe(db)
				tmpdir, err := ioutil.TempDir("", "package")
				Expect(err).ToNot(HaveOccurred())
				defer os.RemoveAll(tmpdir) // clean up

				err = generalRecipe.Load("../../tests/fixtures/buildableseed")
				Expect(err).ToNot(HaveOccurred())
				Expect(generalRecipe.Tree()).ToNot(BeNil()) // It should be populated back at this point

				Expect(len(generalRecipe.Tree().GetPackageSet().GetPackages())).To(Equal(4))
				err = generalRecipe.Tree().ResolveDeps(1)
				Expect(err).ToNot(HaveOccurred())

				D, err := generalRecipe.Tree().FindPackage(&pkg.DefaultPackage{Name: "d", Category: "test", Version: "1.0"})
				Expect(err).ToNot(HaveOccurred())

				Expect(D.GetRequires()[0].GetName()).To(Equal("c"))
				CfromD := D.GetRequires()[0]
				Expect(len(CfromD.GetRequires()) != 0).To(BeTrue())
				Expect(CfromD.GetRequires()[0].GetName()).To(Equal("b"))

				w, err := generalRecipe.Tree().World()
				Expect(err).ToNot(HaveOccurred())

				s := solver.NewSolver([]pkg.Package{}, w, db)
				pack, err := generalRecipe.Tree().FindPackage(&pkg.DefaultPackage{Name: "d", Category: "test", Version: "1.0"})
				Expect(err).ToNot(HaveOccurred())

				solution, err := s.Install([]pkg.Package{pack})
				Expect(err).ToNot(HaveOccurred())

				solution = solution.Order(pack.GetFingerPrint())

				Expect(solution[0].Package.GetName()).To(Equal("a"))
				Expect(solution[0].Package.Flagged()).To(BeTrue())
				Expect(solution[0].Value).To(BeFalse())

				Expect(solution[1].Package.GetName()).To(Equal("b"))
				Expect(solution[1].Package.Flagged()).To(BeTrue())
				Expect(solution[1].Value).To(BeTrue())

				Expect(solution[2].Package.GetName()).To(Equal("c"))
				Expect(solution[2].Package.Flagged()).To(BeTrue())
				Expect(solution[2].Value).To(BeTrue())

				Expect(solution[3].Package.GetName()).To(Equal("d"))
				Expect(solution[3].Package.Flagged()).To(BeTrue())
				Expect(solution[3].Value).To(BeTrue())
				Expect(len(solution)).To(Equal(4))

				newsolution := solution.Drop(&pkg.DefaultPackage{Name: "d", Category: "test", Version: "1.0"})
				Expect(len(newsolution)).To(Equal(3))

				Expect(newsolution[0].Package.GetName()).To(Equal("a"))
				Expect(newsolution[0].Package.Flagged()).To(BeTrue())
				Expect(newsolution[0].Value).To(BeFalse())

				Expect(newsolution[1].Package.GetName()).To(Equal("b"))
				Expect(newsolution[1].Package.Flagged()).To(BeTrue())
				Expect(newsolution[1].Value).To(BeTrue())

				Expect(newsolution[2].Package.GetName()).To(Equal("c"))
				Expect(newsolution[2].Package.Flagged()).To(BeTrue())
				Expect(newsolution[2].Value).To(BeTrue())

			}
		})
	})

})