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()
+}

pkg/db/key.go

@@ -0,0 +1,48 @@
+// Copyright (C) 2023 Marius Schellenberger
+
+package db
+
+import (
+	"sort"
+
+	"git.giftfish.de/ston1th/goacc/pkg/core"
+	"git.giftfish.de/ston1th/keyctl/pkg/key"
+)
+
+const keyPrefix = "key/"
+
+func (db *DB) GetKeys() (keys core.Keys, err error) {
+	err = db.store.ForEachPrefix(keyPrefix, func(k string, _ []byte) error {
+		k, err := db.GetKey(k)
+		if err == nil {
+			keys = append(keys, k)
+		}
+		return nil
+	})
+	sort.Sort(keys)
+	return
+}
+
+func (db *DB) GetKey(id string) (k core.Key, err error) {
+	k, err = db.GetKeyWithSecret(id)
+	k.Key = ""
+	return
+}
+
+func (db *DB) GetKeyWithSecret(id string) (k core.Key, err error) {
+	err = db.store.Get(keyPrefix+id, &k)
+	return
+}
+
+func (db *DB) CreateKey(name string, size int, enc core.Encoding) (k core.Key, err error) {
+	k, err = key.Generate(name, size, enc)
+	if err != nil {
+		return
+	}
+	err = db.store.Set(keyPrefix+k.ID, &k)
+	return
+}
+
+func (db *DB) DeleteKey(id string) error {
+	return db.store.Delete(keyPrefix + id)
+}

pkg/db/migration.go

@@ -0,0 +1,60 @@
+// Copyright (C) 2022 Marius Schellenberger
+
+package db
+
+import (
+	"errors"
+
+	"git.giftfish.de/ston1th/keyctl/pkg/store"
+)
+
+const (
+	versionKey     = "version/version"
+	currentVersion = 0
+)
+
+type migrator func(*DB, int) error
+
+var migrators = []migrator{
+	func(_ *DB, _ int) error {
+		return nil
+	},
+}
+
+func (db *DB) RunMigrations() (err error) {
+	log := db.log.WithName("migration")
+	var version int
+	err = db.store.Get(versionKey, &version)
+	if err == store.ErrKeyNotFound {
+		err = db.store.Set(versionKey, currentVersion)
+		if err != nil {
+			return
+		}
+		err = db.store.Get(versionKey, &version)
+		if err != nil {
+			return
+		}
+	}
+	log.Info("db versions", "current", version, "new", currentVersion)
+	if currentVersion < version {
+		return errors.New("incompatible versions")
+	}
+	if currentVersion == version {
+		log.Info("nothing to do")
+	}
+	for version < currentVersion {
+		v := version + 1
+		log.Info("running", "version", v)
+		err = migrators[v](db, v)
+		if err != nil {
+			return
+		}
+		err = db.store.Set(versionKey, v)
+		if err != nil {
+			return
+		}
+		version = v
+		log.Info("finished", "version", v)
+	}
+	return
+}