package router

import (
	"go.jolheiser.com/beaver"
	"go.jolheiser.com/vanity/flags"
	"go.jolheiser.com/vanity/service"
	"strings"
	"sync"

	"go.jolheiser.com/vanity/api"
)

var cache = &packageCache{
	packages: make(map[string]*api.Package),
}

type packageCache struct {
	packages map[string]*api.Package
	sync.Mutex
}

func (c *packageCache) Update(packages map[string]*api.Package) {
	c.Lock()
	c.packages = packages
	c.Unlock()
}

func (c *packageCache) Names() []string {
	names := make([]string, len(c.packages))
	idx := 0
	for name := range c.packages {
		names[idx] = name
		idx++
	}
	return names
}

func updateCache() {
	packages, err := svc.Packages()
	if err != nil {
		beaver.Errorf("could not update packages: %v", err)
		return
	}

	// Filter
	for name, pkg := range packages {
		if err := service.Check(pkg); err != nil {
			beaver.Debug(err)
			delete(packages, name)
			continue
		}
		goMod, err := svc.GoMod(pkg)
		if err != nil {
			beaver.Debugf("No go.mod could be found in the root directory of %s", pkg.Name)
			delete(packages, name)
			continue
		}
		lines := strings.Split(goMod, "\n")
		line := strings.Fields(lines[0])
		if !strings.HasPrefix(line[1], flags.Domain) {
			beaver.Debugf("%s is a Go project, however its module does not include this domain", pkg.Name)
			delete(packages, name)
			continue
		}
		beaver.Debugf("Including %s", pkg.Name)
	}

	// Overrides
	for name, pkg := range packages {
		for key, override := range flags.Override {
			if strings.EqualFold(name, key) {
				beaver.Debugf("Overriding %s -> %s", name, override)
				delete(packages, key)
				pkg.Name = override
				packages[override] = pkg
			}
		}
	}

	// Add packages manually added to config
	for _, pkg := range flags.ConfigPackages {
		packages[pkg.Name] = pkg
	}

	cache.Update(packages)
	canUpdate = false
}