feat: sqlite storage

internal/db/db.go

@@ -0,0 +1,92 @@
+package db
+
+import (
+	"database/sql"
+	"log"
+
+	_ "modernc.org/sqlite"
+)
+
+type Volume struct {
+	Name string `json:"name"`
+	Path string `json:"path"`
+}
+
+// DB is the global database connection
+var DB, err = sql.Open("sqlite", "file:ironmount.db")
+
+// Init initializes the database and creates the volumes table if it doesn't exist
+func Init() {
+	if err != nil {
+		panic(err)
+	}
+
+	_, err = DB.Exec(`
+	CREATE TABLE IF NOT EXISTS volumes (
+		name TEXT PRIMARY KEY,
+		path TEXT NOT NULL
+	);
+	`)
+	if err != nil {
+		panic(err)
+	}
+}
+
+func GetVolumeByName(n string) (*Volume, error) {
+	var path string
+	var name string
+
+	err := DB.QueryRow("SELECT name, path FROM volumes WHERE name = ?", n).Scan(&name, &path)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return &Volume{
+		Name: name,
+		Path: path,
+	}, nil
+}
+
+func CreateVolume(name, path string) error {
+	_, err := DB.Exec("INSERT INTO volumes (name, path) VALUES (?, ?)", name, path)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func RemoveVolume(name string) error {
+	_, err := DB.Exec("DELETE FROM volumes WHERE name = ?", name)
+
+	log.Printf("Removing volume: %s", name)
+	log.Printf("Error removing volume: %v", err)
+
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func ListVolumes() ([]Volume, error) {
+	rows, err := DB.Query("SELECT name, path FROM volumes")
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var volumes []Volume
+	for rows.Next() {
+		var vol Volume
+		if err := rows.Scan(&vol.Name, &vol.Path); err != nil {
+			return nil, err
+		}
+		volumes = append(volumes, vol)
+	}
+
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+
+	return volumes, nil
+}