i have to commit at some point!

4 files changed, 253 insertions, 0 deletions

diff --git a/indexers/pikdex/hydrate.go b/indexers/pikdex/hydrate.go
new file mode 100644
index 0000000..078f3a6
--- /dev/null
+++ b/indexers/pikdex/hydrate.go
@@ -0,0 +1,18 @@
+package pikdex
+
+import (
+	"pik/model"
+)
+
+func (u *uwudex) Hydrate(src *model.Source, result *model.HydratedSource) error {
+	mod := u.mods[src.Path]
+	if mod.Path != "" {
+		if mod.Aliases != nil {
+			result.Aliases = append(result.Aliases, mod.Aliases...)
+		}
+		if mod.Icon != "" {
+			result.Icon = mod.Icon
+		}
+	}
+	return nil
+}
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
+}
diff --git a/indexers/pikdex/index_test.go b/indexers/pikdex/index_test.go
new file mode 100644
index 0000000..885b29f
--- /dev/null
+++ b/indexers/pikdex/index_test.go
@@ -0,0 +1,54 @@
+package pikdex
+
+import (
+	"github.com/stretchr/testify/assert"
+	"io/fs"
+	"testing"
+	"testing/fstest"
+)
+
+func TestUwudex_WantsWalk_AnyRoot(t *testing.T) {
+	for _, r := range Roots {
+		data := fstest.MapFS{
+			r: &fstest.MapFile{
+				Data: nil,
+				Mode: fs.ModeDir,
+			},
+		}
+		u := &uwudex{}
+		result, root, err := u.WantsWalk(data)
+		assert.Equal(t, root, r)
+		assert.NoError(t, err)
+		assert.True(t, result)
+	}
+}
+
+func TestUwudex_WantsWalk_TwoRoots(t *testing.T) {
+	data := fstest.MapFS{
+		Roots[0]: &fstest.MapFile{
+			Data: nil,
+			Mode: fs.ModeDir,
+		},
+		Roots[1]: &fstest.MapFile{
+			Data: nil,
+			Mode: fs.ModeDir,
+		},
+	}
+	u := &uwudex{}
+	result, r, err := u.WantsWalk(data)
+	// no guarantee we pick any one lol
+	assert.Contains(t, Roots, r)
+	assert.NoError(t, err)
+	assert.True(t, result)
+}
+
+func TestUwudex_WantsWalk_NoRoots(t *testing.T) {
+	data := fstest.MapFS{
+		"asdf.txt": &fstest.MapFile{},
+	}
+	u := &uwudex{}
+	result, r, err := u.WantsWalk(data)
+	assert.Equal(t, "", r)
+	assert.NoError(t, err)
+	assert.False(t, result)
+}
diff --git a/indexers/pikdex/meta.go b/indexers/pikdex/meta.go
new file mode 100644
index 0000000..af1af93
--- /dev/null
+++ b/indexers/pikdex/meta.go
@@ -0,0 +1,27 @@
+package pikdex
+
+import (
+	"strings"
+)
+
+type MetaSetter func(s *SourceData, content string)
+
+var MetaFiles = map[string]MetaSetter{
+	".alias": func(s *SourceData, content string) {
+		split := strings.Split(content, "\n")
+		s.Aliases = make([]string, 0, len(split))
+		for _, line := range split {
+			stripped := strip(line)
+			if stripped != "" {
+				s.Aliases = append(s.Aliases, stripped)
+			}
+		}
+	},
+	".icon": func(s *SourceData, content string) {
+		s.Icon = string([]rune(strip(content))[0:2])
+	},
+}
+
+func strip(input string) string {
+	return strings.TrimSpace(input)
+}