diff --git a/main.go b/main.go index 3c9005f..7ba73d9 100644 --- a/main.go +++ b/main.go @@ -8,6 +8,7 @@ import ( "fmt" "io/fs" "os" + "path/filepath" "sort" "strings" @@ -19,13 +20,17 @@ import ( var Version = "develop" func main() { - defaultStore, err := xdg.ConfigFile("kv/store.json") - if err != nil { - defaultStore = "store.json" + configHome := xdg.ConfigHome + if configHome == "" { + home, err := os.UserHomeDir() + if err != nil { + home = "." + } + configHome = home } - + defaultStore := filepath.Join(configHome, "kv") fs := flag.NewFlagSet("kv", flag.ContinueOnError) - storeFlag := fs.String("store", defaultStore, "Location of the store on disk") + storeFlag := fs.String("store", defaultStore, "Location of the store directory") fs.StringVar(storeFlag, "s", *storeFlag, "--store") a := &app{ @@ -66,6 +71,8 @@ type app struct { func (a *app) get() *ffcli.Command { fs := flag.NewFlagSet("get", flag.ContinueOnError) + storeFlag := fs.String("store", "", "Use a specific store instead of aggregate") + fs.StringVar(storeFlag, "s", *storeFlag, "--store") return &ffcli.Command{ Name: "get", FlagSet: fs, @@ -80,7 +87,7 @@ func (a *app) get() *ffcli.Command { } key := strings.ToLower(strings.Join(args, " ")) - val, ok := data[key] + val, ok := data.Val(*storeFlag, key) if !ok { return fmt.Errorf("no value found for %q", key) } @@ -92,6 +99,8 @@ func (a *app) get() *ffcli.Command { func (a *app) set() *ffcli.Command { fs := flag.NewFlagSet("set", flag.ContinueOnError) + storeFlag := fs.String("store", "kv", "Store for this key/value") + fs.StringVar(storeFlag, "s", *storeFlag, "--store") return &ffcli.Command{ Name: "set", FlagSet: fs, @@ -105,9 +114,24 @@ func (a *app) set() *ffcli.Command { return err } + store, ok := data[*storeFlag] + if !ok { + fi, err := os.Create(filepath.Join(*a.storeLocation, *storeFlag+".json")) + if err != nil { + return err + } + if _, err := fi.WriteString("{}"); err != nil { + return err + } + if err := fi.Close(); err != nil { + return err + } + store = make(map[string]string) + } + key := args[0] - data[key] = strings.Join(args[1:], " ") - if err := a.save(data); err != nil { + store[key] = strings.Join(args[1:], " ") + if err := a.save(*storeFlag, store); err != nil { return err } fmt.Printf("set %q\n", key) @@ -118,6 +142,8 @@ func (a *app) set() *ffcli.Command { func (a *app) del() *ffcli.Command { fs := flag.NewFlagSet("del", flag.ContinueOnError) + storeFlag := fs.String("store", "kv", "Store for this key/value") + fs.StringVar(storeFlag, "s", *storeFlag, "--store") return &ffcli.Command{ Name: "del", FlagSet: fs, @@ -131,12 +157,17 @@ func (a *app) del() *ffcli.Command { return err } + store, ok := data[*storeFlag] + if !ok { + return fmt.Errorf("no store found for %q", *storeFlag) + } + key := strings.ToLower(strings.Join(args, " ")) - if _, ok := data[key]; !ok { + if _, ok := store[key]; !ok { return fmt.Errorf("no value found for %q", key) } - delete(data, key) - if err := a.save(data); err != nil { + delete(store, key) + if err := a.save(*storeFlag, store); err != nil { return err } @@ -149,6 +180,8 @@ func (a *app) del() *ffcli.Command { func (a *app) list() *ffcli.Command { fs := flag.NewFlagSet("list", flag.ContinueOnError) + storeFlag := fs.String("store", "", "Use a specific store instead of aggregate") + fs.StringVar(storeFlag, "s", *storeFlag, "--store") return &ffcli.Command{ Name: "list", FlagSet: fs, @@ -160,7 +193,7 @@ func (a *app) list() *ffcli.Command { var keys []string prefix := strings.ToLower(strings.Join(args, " ")) - for key := range data { + for key := range data.Map() { if strings.HasPrefix(key, prefix) { keys = append(keys, key) } @@ -176,31 +209,83 @@ func (a *app) list() *ffcli.Command { } } -func (a *app) load() (map[string]string, error) { - var m map[string]string - fi, err := os.Open(*a.storeLocation) - if err != nil { - if errors.Is(err, fs.ErrNotExist) { - fi, err = os.Create(*a.storeLocation) - if err != nil { - return m, err - } - if _, err := fi.WriteString("{}"); err != nil { - return m, err - } - if err := fi.Close(); err != nil { - return m, err - } - return a.load() - } - return m, err +type store map[string]map[string]string + +func (s store) Val(store, key string) (string, bool) { + if store != "" { + val, ok := s[store][key] + return val, ok } - defer fi.Close() - return m, json.NewDecoder(fi).Decode(&m) + val, ok := s.Map()[key] + return val, ok } -func (a *app) save(m map[string]string) error { - fi, err := os.Create(*a.storeLocation) +func (s store) Map() map[string]string { + m := s["kv"] + for sname, ss := range s { + if sname == "kv" { + continue + } + for k, v := range ss { + m[k] = v + } + } + return m +} + +func load(paths ...string) store { + s := make(store) + for _, p := range paths { + func() { + var m map[string]string + fi, err := os.Open(p) + if err != nil { + fmt.Printf("could not open %q: %v\n", p, err) + return + } + defer fi.Close() + if err := json.NewDecoder(fi).Decode(&m); err != nil { + fmt.Printf("could not decode %q: %v\n", p, err) + return + } + storeName := strings.TrimSuffix(filepath.Base(p), ".json") + s[storeName] = m + }() + } + return s +} + +func (a *app) load() (store, error) { + defaultConfig := filepath.Join(*a.storeLocation, "kv.json") + if _, err := os.Stat(defaultConfig); err != nil { + if !errors.Is(err, fs.ErrNotExist) { + return nil, err + } + fi, err := os.Create(defaultConfig) + if err != nil { + return nil, err + } + if _, err := fi.WriteString("{}"); err != nil { + return nil, err + } + if err := fi.Close(); err != nil { + return nil, err + } + } + + dirs, err := os.ReadDir(*a.storeLocation) + if err != nil { + return nil, err + } + paths := make([]string, 0, len(dirs)) + for _, dir := range dirs { + paths = append(paths, filepath.Join(*a.storeLocation, dir.Name())) + } + return load(paths...), nil +} + +func (a *app) save(storeName string, m map[string]string) error { + fi, err := os.Create(filepath.Join(*a.storeLocation, storeName+".json")) if err != nil { return err }