that initial commit

starpunk/database.py

@@ -0,0 +1,104 @@
+"""
+Database initialization and operations for StarPunk
+SQLite database for metadata, sessions, and tokens
+"""
+
+import sqlite3
+from pathlib import Path
+
+
+# Database schema
+SCHEMA_SQL = """
+-- Notes metadata (content is in files)
+CREATE TABLE IF NOT EXISTS notes (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    slug TEXT UNIQUE NOT NULL,
+    file_path TEXT UNIQUE NOT NULL,
+    published BOOLEAN DEFAULT 0,
+    created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    deleted_at TIMESTAMP,
+    content_hash TEXT
+);
+
+CREATE INDEX IF NOT EXISTS idx_notes_created_at ON notes(created_at DESC);
+CREATE INDEX IF NOT EXISTS idx_notes_published ON notes(published);
+CREATE INDEX IF NOT EXISTS idx_notes_slug ON notes(slug);
+CREATE INDEX IF NOT EXISTS idx_notes_deleted_at ON notes(deleted_at);
+
+-- Authentication sessions (IndieLogin)
+CREATE TABLE IF NOT EXISTS sessions (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    session_token TEXT UNIQUE NOT NULL,
+    me TEXT NOT NULL,
+    created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    expires_at TIMESTAMP NOT NULL,
+    last_used_at TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_sessions_token ON sessions(session_token);
+CREATE INDEX IF NOT EXISTS idx_sessions_expires ON sessions(expires_at);
+
+-- Micropub access tokens
+CREATE TABLE IF NOT EXISTS tokens (
+    token TEXT PRIMARY KEY,
+    me TEXT NOT NULL,
+    client_id TEXT,
+    scope TEXT,
+    created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    expires_at TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_tokens_me ON tokens(me);
+
+-- CSRF state tokens (for IndieAuth flow)
+CREATE TABLE IF NOT EXISTS auth_state (
+    state TEXT PRIMARY KEY,
+    created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    expires_at TIMESTAMP NOT NULL
+);
+
+CREATE INDEX IF NOT EXISTS idx_auth_state_expires ON auth_state(expires_at);
+"""
+
+
+def init_db(app=None):
+    """
+    Initialize database schema
+
+    Args:
+        app: Flask application instance (optional, for config access)
+    """
+    if app:
+        db_path = app.config['DATABASE_PATH']
+    else:
+        # Fallback to default path
+        db_path = Path('./data/starpunk.db')
+
+    # Ensure parent directory exists
+    db_path.parent.mkdir(parents=True, exist_ok=True)
+
+    # Create database and schema
+    conn = sqlite3.connect(db_path)
+    try:
+        conn.executescript(SCHEMA_SQL)
+        conn.commit()
+        print(f"Database initialized: {db_path}")
+    finally:
+        conn.close()
+
+
+def get_db(app):
+    """
+    Get database connection
+
+    Args:
+        app: Flask application instance
+
+    Returns:
+        sqlite3.Connection
+    """
+    db_path = app.config['DATABASE_PATH']
+    conn = sqlite3.connect(db_path)
+    conn.row_factory = sqlite3.Row  # Return rows as dictionaries
+    return conn