package main

import (
	"context"
	"sync"

	"github.com/Jguer/aur"
	"github.com/leonelquinteros/gotext"

	"github.com/Jguer/yay/v12/pkg/db"
	"github.com/Jguer/yay/v12/pkg/dep"
	"github.com/Jguer/yay/v12/pkg/runtime"
	"github.com/Jguer/yay/v12/pkg/sync/srcinfo"
	"github.com/Jguer/yay/v12/pkg/sync/workdir"
)

func infoToInstallInfo(info []aur.Pkg) []map[string]*dep.InstallInfo {
	installInfo := make([]map[string]*dep.InstallInfo, 1)
	installInfo[0] = map[string]*dep.InstallInfo{}

	for i := range info {
		pkg := &info[i]
		installInfo[0][pkg.Name] = &dep.InstallInfo{
			AURBase: &pkg.PackageBase,
			Source:  dep.AUR,
		}
	}

	return installInfo
}

// createDevelDB forces yay to create a DB of the existing development packages.
func createDevelDB(ctx context.Context, run *runtime.Runtime, dbExecutor db.Executor) error {
	remoteNames := dbExecutor.InstalledRemotePackageNames()

	run.QueryBuilder.Execute(ctx, dbExecutor, remoteNames)
	info, err := run.AURClient.Get(ctx, &aur.Query{
		Needles:  remoteNames,
		By:       aur.Name,
		Contains: false,
	})
	if err != nil {
		return err
	}

	preper := workdir.NewPreparerWithoutHooks(dbExecutor, run.CmdBuilder, run.Cfg, run.Logger.Child("workdir"), false)

	mapInfo := infoToInstallInfo(info)
	pkgBuildDirsByBase, err := preper.Run(ctx, run, mapInfo)
	if err != nil {
		return err
	}

	srcinfos, err := srcinfo.ParseSrcinfoFilesByBase(run.Logger.Child("srcinfo"), pkgBuildDirsByBase, false)
	if err != nil {
		return err
	}

	var wg sync.WaitGroup
	for i := range srcinfos {
		for iP := range srcinfos[i].Packages {
			wg.Add(1)

			go func(i string, iP int) {
				run.VCSStore.Update(ctx, srcinfos[i].Packages[iP].Pkgname, srcinfos[i].Source)
				wg.Done()
			}(i, iP)
		}
	}

	wg.Wait()
	run.Logger.OperationInfoln(gotext.Get("GenDB finished. No packages were installed"))

	return err
}