(Very) basic setup.

db/db.go

@@ -0,0 +1,37 @@
+package db
+
+import (
+	"errors"
+	badger "github.com/dgraph-io/badger/v3"
+	"log"
+	"os"
+)
+
+var (
+	db *badger.DB
+)
+
+// Init initializes the database.
+func Init() error {
+	dataDir := os.Getenv("MIGO_DATA_DIR")
+
+	if dataDir == "" {
+		return errors.New("MIGO_DATA_DIR not configured")
+	}
+
+	var err error
+	db, err = badger.Open(badger.DefaultOptions(dataDir))
+
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// Close closes the database connection.
+func Close() {
+	if err := db.Close(); err != nil {
+		log.Printf("Error closing database connection: %v", err)
+	}
+}