90 lines
2.4 KiB
Python
90 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
import sqlite3
|
|
from pathlib import Path
|
|
|
|
|
|
SCHEMA = """
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
id TEXT PRIMARY KEY,
|
|
user_key TEXT NOT NULL,
|
|
created_at TIMESTAMP NOT NULL,
|
|
updated_at TIMESTAMP NOT NULL
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS user_resources (
|
|
id TEXT PRIMARY KEY,
|
|
user_id TEXT NOT NULL,
|
|
app_id TEXT NOT NULL DEFAULT 'default',
|
|
project_id TEXT NOT NULL DEFAULT 'default',
|
|
session_id TEXT NOT NULL,
|
|
original_filename TEXT,
|
|
mime_type TEXT,
|
|
content_type TEXT NOT NULL,
|
|
uri TEXT NOT NULL,
|
|
uri_public BOOLEAN NOT NULL DEFAULT FALSE,
|
|
sha256 TEXT,
|
|
size_bytes INTEGER,
|
|
title TEXT,
|
|
description TEXT,
|
|
status TEXT NOT NULL DEFAULT 'pending',
|
|
error_message TEXT,
|
|
created_at TIMESTAMP NOT NULL,
|
|
updated_at TIMESTAMP NOT NULL,
|
|
deleted_at TIMESTAMP
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_user_resources_user_scope
|
|
ON user_resources (user_id, app_id, project_id, status, deleted_at);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_user_resources_session_id
|
|
ON user_resources (session_id);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_user_resources_user_id
|
|
ON user_resources (user_id);
|
|
|
|
CREATE TABLE IF NOT EXISTS memory_tombstones (
|
|
id TEXT PRIMARY KEY,
|
|
user_id TEXT NOT NULL,
|
|
memory_id TEXT,
|
|
session_id TEXT,
|
|
reason TEXT,
|
|
created_at TIMESTAMP NOT NULL
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_memory_tombstones_user_memory
|
|
ON memory_tombstones (user_id, memory_id);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_memory_tombstones_user_session
|
|
ON memory_tombstones (user_id, session_id);
|
|
|
|
CREATE TABLE IF NOT EXISTS memory_overrides (
|
|
id TEXT PRIMARY KEY,
|
|
user_id TEXT NOT NULL,
|
|
memory_id TEXT,
|
|
session_id TEXT,
|
|
override_text TEXT NOT NULL,
|
|
is_active BOOLEAN NOT NULL DEFAULT TRUE,
|
|
created_at TIMESTAMP NOT NULL,
|
|
updated_at TIMESTAMP NOT NULL
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_memory_overrides_user_memory_active
|
|
ON memory_overrides (user_id, memory_id, is_active);
|
|
"""
|
|
|
|
|
|
def connect(db_path: Path) -> sqlite3.Connection:
|
|
conn = sqlite3.connect(db_path)
|
|
conn.row_factory = sqlite3.Row
|
|
conn.execute("PRAGMA journal_mode=WAL")
|
|
conn.execute("PRAGMA foreign_keys=ON")
|
|
conn.execute("PRAGMA busy_timeout=5000")
|
|
return conn
|
|
|
|
|
|
def init_db(db_path: Path) -> None:
|
|
db_path.parent.mkdir(parents=True, exist_ok=True)
|
|
with connect(db_path) as conn:
|
|
conn.executescript(SCHEMA)
|