package repos

import (
	"github.com/genjidb/genji"
	"github.com/genjidb/genji/document"
	"github.com/genjidb/genji/types"
	"go.arsenm.dev/lure/internal/db"
)

// FindPkgs looks for packages matching the inputs inside the database.
// It returns a map that maps the package name input to the packages found for it.
// It also returns a slice that contains the names of all packages that were not found.
func FindPkgs(gdb *genji.DB, pkgs []string) (map[string][]db.Package, []string, error) {
	found := map[string][]db.Package{}
	notFound := []string(nil)

	for _, pkgName := range pkgs {
		result, err := db.GetPkgs(gdb, "name LIKE ?", pkgName)
		if err != nil {
			return nil, nil, err
		}

		added := 0
		err = result.Iterate(func(d types.Document) error {
			var pkg db.Package
			err = document.StructScan(d, &pkg)
			if err != nil {
				return err
			}

			added++
			found[pkgName] = append(found[pkgName], pkg)
			return nil
		})
		result.Close()
		if err != nil {
			return nil, nil, err
		}

		if added == 0 {
			result, err := db.GetPkgs(gdb, "? IN provides", pkgName)
			if err != nil {
				return nil, nil, err
			}

			err = result.Iterate(func(d types.Document) error {
				var pkg db.Package
				err = document.StructScan(d, &pkg)
				if err != nil {
					return err
				}

				added++
				found[pkgName] = append(found[pkgName], pkg)
				return nil
			})
			result.Close()
			if err != nil {
				return nil, nil, err
			}
		}

		if added == 0 {
			notFound = append(notFound, pkgName)
		}
	}

	return found, notFound, nil
}