initial commit

pkg/db/db.go

@@ -0,0 +1,53 @@
+// Copyright (C) 2023 Marius Schellenberger
+
+package db
+
+import (
+	"io"
+	"path/filepath"
+
+	"git.giftfish.de/ston1th/godrop/v2"
+	"git.giftfish.de/ston1th/keyctl/pkg/core"
+	"git.giftfish.de/ston1th/keyctl/pkg/store"
+	"github.com/go-logr/logr"
+)
+
+const storeFile = "store.db"
+
+type DB struct {
+	log   logr.Logger
+	store store.Store
+}
+
+func NewPlain(log logr.Logger, cfg *core.Config) (db *DB, err error) {
+	db = &DB{log: log}
+	dbFile := filepath.Join(cfg.DataDir, storeFile)
+	err = godrop.Unveil(dbFile, "rwc")
+	if err != nil {
+		return
+	}
+	db.store, err = store.NewBoltStore(dbFile, nil)
+	return
+}
+
+func New(log logr.Logger, cfg *core.Config) (db *DB, err error) {
+	db, err = NewPlain(log, cfg)
+	if err != nil {
+		return
+	}
+
+	err = db.RunMigrations()
+	return
+}
+
+func (db *DB) Dump(w io.Writer) error {
+	return db.store.Dump(w)
+}
+
+func (db *DB) Restore(r io.Reader) error {
+	return db.store.Restore(r)
+}
+
+func (db *DB) Close() error {
+	return db.store.Close()
+}