Add sqlite backend

2025-09-24 10:32:10 -07:00
parent 9eedeceb6a
commit 91f23cddc9
12 changed files with 556 additions and 13 deletions
@@ -0,0 +1,113 @@
+package sqlite
+
+import (
+	"database/sql"
+	"fmt"
+	"sync"
+
+	"fiatjaf.com/nostr/eventstore"
+	_ "github.com/mattn/go-sqlite3"
+)
+
+var _ eventstore.Store = (*SqliteBackend)(nil)
+
+type SqliteBackend struct {
+	sync.RWMutex
+	// Path is where the database will be saved
+	Path string
+
+	db           *sql.DB
+	FTSAvailable bool
+}
+
+func (s *SqliteBackend) Close() {
+	if s.db != nil {
+		s.db.Close()
+	}
+}
+
+func (s *SqliteBackend) Init() error {
+	if s.Path == "" {
+		return fmt.Errorf("missing Path")
+	}
+
+	var err error
+	s.db, err = sql.Open("sqlite3", s.Path+"?_journal_mode=WAL&_sync=NORMAL&_cache_size=1000&_foreign_keys=true")
+	if err != nil {
+		return fmt.Errorf("failed to open database: %w", err)
+	}
+
+	// Create tables and indexes
+	if err := s.createSchema(); err != nil {
+		return fmt.Errorf("failed to create schema: %w", err)
+	}
+
+	return nil
+}
+
+func (s *SqliteBackend) createSchema() error {
+	// Create basic schema first
+	basicSchema := `
+	CREATE TABLE IF NOT EXISTS events (
+		id TEXT PRIMARY KEY,
+		created_at INTEGER NOT NULL,
+		kind INTEGER NOT NULL,
+		pubkey TEXT NOT NULL,
+		content TEXT NOT NULL,
+		tags TEXT NOT NULL,
+		sig TEXT NOT NULL
+	);
+
+	CREATE INDEX IF NOT EXISTS idx_events_created_at ON events(created_at);
+	CREATE INDEX IF NOT EXISTS idx_events_kind ON events(kind);
+	CREATE INDEX IF NOT EXISTS idx_events_pubkey ON events(pubkey);
+	CREATE INDEX IF NOT EXISTS idx_events_kind_pubkey ON events(kind, pubkey);
+	CREATE INDEX IF NOT EXISTS idx_events_kind_pubkey_created_at ON events(kind, pubkey, created_at DESC);
+
+	CREATE TABLE IF NOT EXISTS event_tags (
+		event_id TEXT NOT NULL,
+		key TEXT NOT NULL,
+		value TEXT NOT NULL,
+		FOREIGN KEY (event_id) REFERENCES events(id) ON DELETE CASCADE
+	);
+
+	CREATE INDEX IF NOT EXISTS idx_event_tags_event_id ON event_tags(event_id);
+	CREATE INDEX IF NOT EXISTS idx_event_tags_key ON event_tags(key);
+	CREATE INDEX IF NOT EXISTS idx_event_tags_key_value ON event_tags(key, value);
+	`
+
+	if _, err := s.db.Exec(basicSchema); err != nil {
+		return fmt.Errorf("failed to create schema: %w", err)
+	}
+
+	// Try to create FTS5 schema - if it fails, continue without it
+	ftsSchema := `
+	CREATE VIRTUAL TABLE IF NOT EXISTS events_fts USING fts5(
+		content,
+		content='events',
+		content_rowid='rowid'
+	);
+
+	CREATE TRIGGER IF NOT EXISTS events_ai AFTER INSERT ON events BEGIN
+		INSERT INTO events_fts(rowid, content) VALUES (new.rowid, new.content);
+	END;
+
+	CREATE TRIGGER IF NOT EXISTS events_ad AFTER DELETE ON events BEGIN
+		INSERT INTO events_fts(events_fts, rowid, content) VALUES('delete', old.rowid, old.content);
+	END;
+
+	CREATE TRIGGER IF NOT EXISTS events_au AFTER UPDATE ON events BEGIN
+		INSERT INTO events_fts(events_fts, rowid, content) VALUES('delete', old.rowid, old.content);
+		INSERT INTO events_fts(rowid, content) VALUES (new.rowid, new.content);
+	END;
+	`
+
+	if _, err := s.db.Exec(ftsSchema); err != nil {
+		// FTS5 not available, continue without full-text search
+		s.FTSAvailable = false
+	} else {
+		s.FTSAvailable = true
+	}
+
+	return nil
+}