feat: store as dir
Signed-off-by: jolheiser <john.olheiser@gmail.com>main v0.0.2
parent
bc07bf116e
commit
a882e0802c
153
main.go
153
main.go
|
@ -8,6 +8,7 @@ import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"io/fs"
|
"io/fs"
|
||||||
"os"
|
"os"
|
||||||
|
"path/filepath"
|
||||||
"sort"
|
"sort"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
@ -19,13 +20,17 @@ import (
|
||||||
var Version = "develop"
|
var Version = "develop"
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
defaultStore, err := xdg.ConfigFile("kv/store.json")
|
configHome := xdg.ConfigHome
|
||||||
if err != nil {
|
if configHome == "" {
|
||||||
defaultStore = "store.json"
|
home, err := os.UserHomeDir()
|
||||||
|
if err != nil {
|
||||||
|
home = "."
|
||||||
|
}
|
||||||
|
configHome = home
|
||||||
}
|
}
|
||||||
|
defaultStore := filepath.Join(configHome, "kv")
|
||||||
fs := flag.NewFlagSet("kv", flag.ContinueOnError)
|
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")
|
fs.StringVar(storeFlag, "s", *storeFlag, "--store")
|
||||||
|
|
||||||
a := &app{
|
a := &app{
|
||||||
|
@ -66,6 +71,8 @@ type app struct {
|
||||||
|
|
||||||
func (a *app) get() *ffcli.Command {
|
func (a *app) get() *ffcli.Command {
|
||||||
fs := flag.NewFlagSet("get", flag.ContinueOnError)
|
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{
|
return &ffcli.Command{
|
||||||
Name: "get",
|
Name: "get",
|
||||||
FlagSet: fs,
|
FlagSet: fs,
|
||||||
|
@ -80,7 +87,7 @@ func (a *app) get() *ffcli.Command {
|
||||||
}
|
}
|
||||||
|
|
||||||
key := strings.ToLower(strings.Join(args, " "))
|
key := strings.ToLower(strings.Join(args, " "))
|
||||||
val, ok := data[key]
|
val, ok := data.Val(*storeFlag, key)
|
||||||
if !ok {
|
if !ok {
|
||||||
return fmt.Errorf("no value found for %q", key)
|
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 {
|
func (a *app) set() *ffcli.Command {
|
||||||
fs := flag.NewFlagSet("set", flag.ContinueOnError)
|
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{
|
return &ffcli.Command{
|
||||||
Name: "set",
|
Name: "set",
|
||||||
FlagSet: fs,
|
FlagSet: fs,
|
||||||
|
@ -105,9 +114,24 @@ func (a *app) set() *ffcli.Command {
|
||||||
return err
|
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]
|
key := args[0]
|
||||||
data[key] = strings.Join(args[1:], " ")
|
store[key] = strings.Join(args[1:], " ")
|
||||||
if err := a.save(data); err != nil {
|
if err := a.save(*storeFlag, store); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
fmt.Printf("set %q\n", key)
|
fmt.Printf("set %q\n", key)
|
||||||
|
@ -118,6 +142,8 @@ func (a *app) set() *ffcli.Command {
|
||||||
|
|
||||||
func (a *app) del() *ffcli.Command {
|
func (a *app) del() *ffcli.Command {
|
||||||
fs := flag.NewFlagSet("del", flag.ContinueOnError)
|
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{
|
return &ffcli.Command{
|
||||||
Name: "del",
|
Name: "del",
|
||||||
FlagSet: fs,
|
FlagSet: fs,
|
||||||
|
@ -131,12 +157,17 @@ func (a *app) del() *ffcli.Command {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
store, ok := data[*storeFlag]
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("no store found for %q", *storeFlag)
|
||||||
|
}
|
||||||
|
|
||||||
key := strings.ToLower(strings.Join(args, " "))
|
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)
|
return fmt.Errorf("no value found for %q", key)
|
||||||
}
|
}
|
||||||
delete(data, key)
|
delete(store, key)
|
||||||
if err := a.save(data); err != nil {
|
if err := a.save(*storeFlag, store); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -149,6 +180,8 @@ func (a *app) del() *ffcli.Command {
|
||||||
|
|
||||||
func (a *app) list() *ffcli.Command {
|
func (a *app) list() *ffcli.Command {
|
||||||
fs := flag.NewFlagSet("list", flag.ContinueOnError)
|
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{
|
return &ffcli.Command{
|
||||||
Name: "list",
|
Name: "list",
|
||||||
FlagSet: fs,
|
FlagSet: fs,
|
||||||
|
@ -160,7 +193,7 @@ func (a *app) list() *ffcli.Command {
|
||||||
|
|
||||||
var keys []string
|
var keys []string
|
||||||
prefix := strings.ToLower(strings.Join(args, " "))
|
prefix := strings.ToLower(strings.Join(args, " "))
|
||||||
for key := range data {
|
for key := range data.Map() {
|
||||||
if strings.HasPrefix(key, prefix) {
|
if strings.HasPrefix(key, prefix) {
|
||||||
keys = append(keys, key)
|
keys = append(keys, key)
|
||||||
}
|
}
|
||||||
|
@ -176,31 +209,83 @@ func (a *app) list() *ffcli.Command {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *app) load() (map[string]string, error) {
|
type store map[string]map[string]string
|
||||||
var m map[string]string
|
|
||||||
fi, err := os.Open(*a.storeLocation)
|
func (s store) Val(store, key string) (string, bool) {
|
||||||
if err != nil {
|
if store != "" {
|
||||||
if errors.Is(err, fs.ErrNotExist) {
|
val, ok := s[store][key]
|
||||||
fi, err = os.Create(*a.storeLocation)
|
return val, ok
|
||||||
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
|
|
||||||
}
|
}
|
||||||
defer fi.Close()
|
val, ok := s.Map()[key]
|
||||||
return m, json.NewDecoder(fi).Decode(&m)
|
return val, ok
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *app) save(m map[string]string) error {
|
func (s store) Map() map[string]string {
|
||||||
fi, err := os.Create(*a.storeLocation)
|
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 {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue