summaryrefslogtreecommitdiff
path: root/pkg/db/storage.go
diff options
context:
space:
mode:
Diffstat (limited to 'pkg/db/storage.go')
-rw-r--r--pkg/db/storage.go44
1 files changed, 44 insertions, 0 deletions
diff --git a/pkg/db/storage.go b/pkg/db/storage.go
new file mode 100644
index 0000000..1c1c428
--- /dev/null
+++ b/pkg/db/storage.go
@@ -0,0 +1,44 @@
+package db
+
+import (
+ "bytes"
+ "fmt"
+ "io/ioutil"
+ "os"
+ "path/filepath"
+
+ "github.com/xlgmokha/x/pkg/env"
+ "github.com/xlgmokha/x/pkg/serde"
+ "github.com/xlgmokha/x/pkg/x"
+)
+
+type Entity interface {
+ Identifier() string
+}
+
+type Storage[T Entity] struct {
+ dir string
+}
+
+func New[T Entity](dir string) *Storage[T] {
+ fullPath := x.Must(filepath.Abs(dir))
+ x.Check(os.MkdirAll(fullPath, 0700))
+
+ return &Storage[T]{
+ dir: fullPath,
+ }
+}
+
+func (db *Storage[T]) Save(item T) error {
+ w := new(bytes.Buffer)
+ x.Check(serde.To(w, item, serde.YAML))
+ if env.Fetch("DUMP", "") != "" {
+ fmt.Println(w.String())
+ }
+
+ return ioutil.WriteFile(db.filePathFor(item), w.Bytes(), 0700)
+}
+
+func (db *Storage[T]) filePathFor(item T) string {
+ return fmt.Sprintf("%v/%v.yaml", db.dir, item.Identifier())
+}