From 45a297a8e526094e8fce6e2c5c0fd89b381d1765 Mon Sep 17 00:00:00 2001 From: ewy Date: Tue, 14 Apr 2026 16:37:17 +0200 Subject: i have to commit at some point! --- main.go | 122 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 122 insertions(+) create mode 100644 main.go (limited to 'main.go') diff --git a/main.go b/main.go new file mode 100644 index 0000000..fbc6ca4 --- /dev/null +++ b/main.go @@ -0,0 +1,122 @@ +package main + +import ( + "github.com/spf13/pflag" + "os" + "pik/cache" + "pik/crawl" + "pik/flags" + "pik/indexers/pikdex" + "pik/menu" + "pik/model" + "pik/run" + "pik/runner/python" + "pik/runner/shell" + "pik/search" + "pik/spool" +) + +var initializers = []model.HasInit{ + python.Python, +} + +var indexers = []model.Indexer{ + pikdex.Indexer, +} + +var runners = []model.Runner{ + shell.Runner, + python.Python, +} + +var hydrators = []model.Hydrator{ + pikdex.Indexer, +} + +var ForceConfirm = false + +func main() { + pflag.Parse() + for _, i := range initializers { + err := i.Init() + if err != nil { + _, _ = spool.Warn("%v\n", err) + } + } + here, err := os.Getwd() + if err != nil { + panic(err) + } + locs := crawl.RichLocations(here) + last := locs[len(locs)-1] + root, err := os.OpenRoot(last) + if root == nil { + panic(err) + } + fs := root.FS() + if err != nil { + panic(err) + } + var st *model.State + var stateError error + if !*flags.All { + st, stateError = model.NewState(fs, locs, indexers, runners) + } else { + c, err := cache.Load() + if err != nil { + panic(err) + } + st, stateError = cache.LoadState(fs, c, indexers, runners) + } + if stateError != nil { + panic(stateError) + } + + err = cache.Save(st) + if err != nil { + _, _ = spool.Warn("%v", err) + } + + args := pflag.Args() + + if len(args) == 0 { + source, target, err := menu.Show(st, hydrators) + if err != nil { + panic(err) + } + err = run.Run(source.Source, target, args...) + if err != nil { + panic(err) + } + + return + } + + target, src, confirm, _, args := search.Search(st, args...) + if !*flags.All && target == nil && len(args) > 0 { + err := pflag.Set("all", "true") + ForceConfirm = true + if err != nil { + panic(err) + } + main() + return + } + + if target == nil { + _, _ = spool.Print("no target found.") + os.Exit(1) + return + } + + if confirm || ForceConfirm { + if !menu.Confirm(os.Stdin, src, target, args...) { + os.Exit(0) + } + } + + err = run.Run(src, target, args...) + if err != nil { + panic(err) + } +} -- cgit v1.3