i have to commit at some point!

1 files changed, 154 insertions, 0 deletions

diff --git a/indexers/pikdex/index.go b/indexers/pikdex/index.go
new file mode 100644
index 0000000..e365d6f
--- /dev/null
+++ b/indexers/pikdex/index.go
@@ -0,0 +1,154 @@
+package pikdex
+
+import (
+	"errors"
+	"io/fs"
+	"os"
+	"path"
+	"path/filepath"
+	"pik/model"
+	"pik/spool"
+	"slices"
+	"strings"
+)
+
+var Roots = []string{
+
+	// current name
+	".pik",
+	"_pik",
+
+	// program names from a previous life
+	".godo",
+	"_godo",
+	".pik",
+	"_uwu",
+
+	//utility
+	".bin",
+	"_bin",
+	"tasks",
+	".tasks",
+	"_tasks",
+}
+
+var SkippedFolders = []string{
+	".git",
+	".config",
+	".idea",
+}
+
+func init() {
+	// add own executable name to uwudexable dirs
+	self, err := os.Executable()
+	if strings.HasSuffix(self, ".test") {
+		return
+	}
+	if err != nil {
+		_, _ = spool.Warn("%v\n", err)
+		return
+	}
+	self = strings.TrimSuffix(self, ".exe")
+	Roots = append(Roots, "."+self, "_"+self)
+}
+
+var Indexer = &uwudex{mods: make(map[string]*SourceData)}
+
+type uwudex struct {
+	mods map[string]*SourceData
+}
+
+type SourceData struct {
+	Aliases []string
+	Icon    string
+	Path    string
+}
+
+func (u *uwudex) Index(absPath string, f fs.FS, runners []model.Runner) ([]model.Target, error) {
+	wants, root, err := u.WantsWalk(f)
+	if !wants {
+		return nil, err
+	}
+	var targets []model.Target
+	mod := u.mods[absPath]
+	if mod == nil {
+		u.mods[absPath] = &SourceData{
+			Path: absPath,
+		}
+		mod = u.mods[absPath]
+	}
+	err = fs.WalkDir(f, root, func(p string, d fs.DirEntry, err error) error {
+
+		if !d.IsDir() {
+			for trigger, applier := range MetaFiles {
+
+				// during the crawl, we might find meta files
+				expectedLocation := filepath.Join(absPath, root, trigger)
+				actualLocation := filepath.Join(absPath, p)
+				if expectedLocation != actualLocation {
+					continue
+				}
+
+				content, err := os.ReadFile(expectedLocation)
+				if err != nil {
+					spool.Warn("%v\n", err)
+					continue
+				}
+				applier(mod, string(content))
+
+			}
+		}
+
+		if d.IsDir() {
+			_, dirName := path.Split(p)
+			if slices.Contains(SkippedFolders, dirName) {
+				return fs.SkipDir
+			}
+		}
+
+		for _, r := range runners {
+			wants, err := r.Wants(f, p, d)
+			if err != nil {
+				// TODO: Gracefully keep crawling?
+				return err
+			}
+			if wants {
+				t, err := r.CreateTarget(f, absPath, p, d)
+				if err != nil {
+					// TODO: Gracefully keep crawling?
+					return err
+				}
+				targets = append(targets, t)
+			}
+			if err != nil {
+				// TODO: Gracefully keep crawling?
+				return err
+			}
+		}
+		return nil
+	})
+	u.mods[absPath] = mod
+
+	return targets, err
+}
+
+func (u *uwudex) WantsWalk(f fs.FS) (bool, string, error) {
+	entries, err := fs.ReadDir(f, ".")
+	if err != nil {
+		if errors.Is(err, fs.ErrNotExist) {
+			return false, "", nil
+		} else {
+			return false, "", err
+		}
+	}
+
+	for _, e := range entries {
+		for _, r := range Roots {
+			if e.Name() == r && e.IsDir() {
+				return true, r, nil
+			}
+		}
+	}
+
+	return false, "", nil
+}