P1.1: Registry package with full CRUD and tests

SQLite schema (services, components, ports, volumes, cmd, events), migrations, and complete CRUD operations. 7 tests covering: idempotent migration, service CRUD, duplicate name rejection, component CRUD with ports/volumes/cmd, composite PK enforcement, cascade delete, and event insert/query/count/prune. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-03-26 11:18:35 -07:00
parent 3f23b14ef4
commit 6122123064
8 changed files with 963 additions and 4 deletions
--- a/internal/registry/db.go
+++ b/internal/registry/db.go
@@ -0,0 +1,130 @@
+package registry
+
+import (
+	"database/sql"
+	"fmt"
+
+	_ "modernc.org/sqlite"
+)
+
+// Open opens the registry database at the given path and runs migrations.
+func Open(path string) (*sql.DB, error) {
+	db, err := sql.Open("sqlite", path)
+	if err != nil {
+		return nil, fmt.Errorf("open database: %w", err)
+	}
+
+	for _, pragma := range []string{
+		"PRAGMA journal_mode = WAL",
+		"PRAGMA foreign_keys = ON",
+		"PRAGMA busy_timeout = 5000",
+	} {
+		if _, err := db.Exec(pragma); err != nil {
+			_ = db.Close()
+			return nil, fmt.Errorf("exec %q: %w", pragma, err)
+		}
+	}
+
+	if err := migrate(db); err != nil {
+		_ = db.Close()
+		return nil, fmt.Errorf("migrate: %w", err)
+	}
+
+	return db, nil
+}
+
+func migrate(db *sql.DB) error {
+	_, err := db.Exec(`
+		CREATE TABLE IF NOT EXISTS schema_migrations (
+			version INTEGER PRIMARY KEY,
+			applied_at TEXT NOT NULL DEFAULT (datetime('now'))
+		);
+	`)
+	if err != nil {
+		return fmt.Errorf("create migrations table: %w", err)
+	}
+
+	for i, m := range migrations {
+		version := i + 1
+		var count int
+		if err := db.QueryRow("SELECT COUNT(*) FROM schema_migrations WHERE version = ?", version).Scan(&count); err != nil {
+			return fmt.Errorf("check migration %d: %w", version, err)
+		}
+		if count > 0 {
+			continue
+		}
+
+		if _, err := db.Exec(m); err != nil {
+			return fmt.Errorf("run migration %d: %w", version, err)
+		}
+		if _, err := db.Exec("INSERT INTO schema_migrations (version) VALUES (?)", version); err != nil {
+			return fmt.Errorf("record migration %d: %w", version, err)
+		}
+	}
+
+	return nil
+}
+
+var migrations = []string{
+	// Migration 1: initial schema
+	`
+	CREATE TABLE IF NOT EXISTS services (
+		name       TEXT PRIMARY KEY,
+		active     INTEGER NOT NULL DEFAULT 1,
+		created_at TEXT NOT NULL DEFAULT (datetime('now')),
+		updated_at TEXT NOT NULL DEFAULT (datetime('now'))
+	);
+
+	CREATE TABLE IF NOT EXISTS components (
+		name           TEXT NOT NULL,
+		service        TEXT NOT NULL REFERENCES services(name) ON DELETE CASCADE,
+		image          TEXT NOT NULL,
+		network        TEXT NOT NULL DEFAULT 'bridge',
+		user_spec      TEXT NOT NULL DEFAULT '',
+		restart        TEXT NOT NULL DEFAULT 'unless-stopped',
+		desired_state  TEXT NOT NULL DEFAULT 'running',
+		observed_state TEXT NOT NULL DEFAULT 'unknown',
+		version        TEXT NOT NULL DEFAULT '',
+		created_at     TEXT NOT NULL DEFAULT (datetime('now')),
+		updated_at     TEXT NOT NULL DEFAULT (datetime('now')),
+		PRIMARY KEY (service, name)
+	);
+
+	CREATE TABLE IF NOT EXISTS component_ports (
+		service   TEXT NOT NULL,
+		component TEXT NOT NULL,
+		mapping   TEXT NOT NULL,
+		PRIMARY KEY (service, component, mapping),
+		FOREIGN KEY (service, component) REFERENCES components(service, name) ON DELETE CASCADE
+	);
+
+	CREATE TABLE IF NOT EXISTS component_volumes (
+		service   TEXT NOT NULL,
+		component TEXT NOT NULL,
+		mapping   TEXT NOT NULL,
+		PRIMARY KEY (service, component, mapping),
+		FOREIGN KEY (service, component) REFERENCES components(service, name) ON DELETE CASCADE
+	);
+
+	CREATE TABLE IF NOT EXISTS component_cmd (
+		service   TEXT NOT NULL,
+		component TEXT NOT NULL,
+		position  INTEGER NOT NULL,
+		arg       TEXT NOT NULL,
+		PRIMARY KEY (service, component, position),
+		FOREIGN KEY (service, component) REFERENCES components(service, name) ON DELETE CASCADE
+	);
+
+	CREATE TABLE IF NOT EXISTS events (
+		id         INTEGER PRIMARY KEY AUTOINCREMENT,
+		service    TEXT NOT NULL,
+		component  TEXT NOT NULL,
+		prev_state TEXT NOT NULL,
+		new_state  TEXT NOT NULL,
+		timestamp  TEXT NOT NULL DEFAULT (datetime('now'))
+	);
+
+	CREATE INDEX IF NOT EXISTS idx_events_component_time
+		ON events(service, component, timestamp);
+	`,
+}