Ki& UdZddlmZddlZddlZddlmZejeZ ddZ ddZ e hdZ ddZdd Zdd Zdd Zdd Zdd ZddZddZddZddZeeeeeeeedZded<eej7ZddZy)zSchema migration framework for the code-review-graph SQLite database. Manages incremental schema changes via versioned migration functions. Each migration is idempotent (uses IF NOT EXISTS / column existence checks). ) annotationsN)Callablec |jdj}|ytt|tt fr |dS|dS#t j$rYywxYw)zRead the current schema version from the metadata table. Returns: int: The schema version (0 if metadata table doesn't exist, 1 if not set). z7SELECT value FROM metadata WHERE key = 'schema_version'rvalue)executefetchoneint isinstancetuplelistsqlite3OperationalError)connrows h/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/code_review_graph/migrations.pyget_schema_versionrsl ll E (*  ;ZeT];3q6NNWNN  # #s!A#AAA'&A'c<|jdt|fy)z-Set the schema version in the metadata table.zIINSERT OR REPLACE INTO metadata (key, value) VALUES ('schema_version', ?)N)rstr)rversions r_set_schema_versionr"sLLS W> edgesflowsnodesmetadata nodes_fts risk_index communitiesflow_snapshotsflow_membershipscommunity_summariesc|tvrtd||jd|d}|Dcgc]}t|tr|dn|d}}||vScc}w)z$Check if a column exists in a table.Unknown table: zPRAGMA table_info()rname) _KNOWN_TABLES ValueErrorrr r )rtablecolumncursorrcolumnss r _has_columnr-0sn M!?5'233 \\.ugQ7 8FLRSSC/s1vS[@SGS W Ts!Ac|tvrtd||jd|fj}|ddkDS)zCheck if a table exists.r$zOSELECT count(*) FROM sqlite_master WHERE type IN ('table', 'view') AND name = ?r)r'r(rr )rr)rs r _table_existsr/9sP M!?5'233 ,,   hj  q6A:rclt|dds'|jdtjdyy)z(v2: Add signature column to nodes table.r signaturez+ALTER TABLE nodes ADD COLUMN signature TEXTz/Migration v2: added 'signature' column to nodesNr-rloggerinfors r _migrate_v2r6Js- tWk 2 BC EF 3rc|jd|jd|jd|jd|jdtjdy)z-v3: Create flows and flow_memberships tables.a CREATE TABLE IF NOT EXISTS flows ( id INTEGER PRIMARY KEY AUTOINCREMENT, name TEXT NOT NULL, entry_point_id INTEGER NOT NULL, depth INTEGER NOT NULL, node_count INTEGER NOT NULL, file_count INTEGER NOT NULL, criticality REAL NOT NULL DEFAULT 0.0, path_json TEXT NOT NULL, created_at TEXT NOT NULL DEFAULT (datetime('now')), updated_at TEXT NOT NULL DEFAULT (datetime('now')) ) z CREATE TABLE IF NOT EXISTS flow_memberships ( flow_id INTEGER NOT NULL, node_id INTEGER NOT NULL, position INTEGER NOT NULL, PRIMARY KEY (flow_id, node_id) ) zKCREATE INDEX IF NOT EXISTS idx_flows_criticality ON flows(criticality DESC)zCCREATE INDEX IF NOT EXISTS idx_flows_entry ON flows(entry_point_id)zQCREATE INDEX IF NOT EXISTS idx_flow_memberships_node ON flow_memberships(node_id)z7Migration v3: created flows and flow_memberships tablesNrr3r4r5s r _migrate_v3r9QshLL    LL  LLU LLM LL[ KKIJrc|jdt|dds&|jdtjd|jd|jd|jdtjd y ) z8v4: Create communities table, add community_id to nodes.a CREATE TABLE IF NOT EXISTS communities ( id INTEGER PRIMARY KEY AUTOINCREMENT, name TEXT NOT NULL, level INTEGER NOT NULL DEFAULT 0, parent_id INTEGER, cohesion REAL NOT NULL DEFAULT 0.0, size INTEGER NOT NULL DEFAULT 0, dominant_language TEXT, description TEXT, created_at TEXT NOT NULL DEFAULT (datetime('now')) ) r community_idz1ALTER TABLE nodes ADD COLUMN community_id INTEGERz2Migration v4: added 'community_id' column to nodeszECREATE INDEX IF NOT EXISTS idx_nodes_community ON nodes(community_id)zKCREATE INDEX IF NOT EXISTS idx_communities_parent ON communities(parent_id)zQCREATE INDEX IF NOT EXISTS idx_communities_cohesion ON communities(cohesion DESC)z'Migration v4: created communities tableN)rr-r3r4r5s r _migrate_v4r<us{LL    tWn 5 HI HILLO LLU LL[ KK9:rcjt|ds'|jdtjdyy)z(v5: Create FTS5 virtual table for nodes.rz CREATE VIRTUAL TABLE nodes_fts USING fts5( name, qualified_name, file_path, signature, content='nodes', content_rowid='rowid', tokenize='porter unicode61' ) z2Migration v5: created nodes_fts FTS5 virtual tableN)r/rr3r4r5s r _migrate_v5r>s1 { +     HI ,rc|jd|jd|jd|jdtjdy)z@v6: Add pre-computed summary tables for token-efficient queries.a CREATE TABLE IF NOT EXISTS community_summaries ( community_id INTEGER PRIMARY KEY, name TEXT NOT NULL, purpose TEXT DEFAULT '', key_symbols TEXT DEFAULT '[]', risk TEXT DEFAULT 'unknown', size INTEGER DEFAULT 0, dominant_language TEXT DEFAULT '', FOREIGN KEY (community_id) REFERENCES communities(id) ) a CREATE TABLE IF NOT EXISTS flow_snapshots ( flow_id INTEGER PRIMARY KEY, name TEXT NOT NULL, entry_point TEXT NOT NULL, critical_path TEXT DEFAULT '[]', criticality REAL DEFAULT 0.0, node_count INTEGER DEFAULT 0, file_count INTEGER DEFAULT 0, FOREIGN KEY (flow_id) REFERENCES flows(id) ) a CREATE TABLE IF NOT EXISTS risk_index ( node_id INTEGER PRIMARY KEY, qualified_name TEXT NOT NULL, risk_score REAL DEFAULT 0.0, caller_count INTEGER DEFAULT 0, test_coverage TEXT DEFAULT 'unknown', security_relevant INTEGER DEFAULT 0, last_computed TEXT DEFAULT '', FOREIGN KEY (node_id) REFERENCES nodes(id) ) zNCREATE INDEX IF NOT EXISTS idx_risk_index_score ON risk_index(risk_score DESC)zVMigration v6: created summary tables (community_summaries, flow_snapshots, risk_index)Nr8r5s r _migrate_v6r@saLL    LL    LL    LL ) KKDErcr|jd|jdtjdy)z;v7: Add compound edge indexes for summary and risk queries.zQCREATE INDEX IF NOT EXISTS idx_edges_target_kind ON edges(target_qualified, kind)zQCREATE INDEX IF NOT EXISTS idx_edges_source_kind ON edges(source_qualified, kind)z)Migration v7: added compound edge indexesNr8r5s r _migrate_v7rBs5LL + LL + KK;r) z/dict[int, Callable[[sqlite3.Connection], None]] MIGRATIONSct|}|tk\rytjd|tt t j D]I}||kr tjd| t ||t|||jKtjdty#tj$r*|jtjd|dwxYw)zRun all pending migrations in order. Each migration runs in its own transaction. The schema_version metadata entry is updated after each successful migration. Nz+Schema version %d -> %d: running migrationszRunning migration v%dz"Migration v%d failed, rolling backT)exc_infoz-Migrations complete, now at schema version %d) rLATEST_VERSIONr3r4sortedrQkeysrcommitrErrorrollbackerror)rcurrentrs rrun_migrationsr\s !&G.  KK=wW*//+,  g   +W5  w  % g . KKM  KK?P }}  MMO LL=wQUL V  s .*B55=C2)rsqlite3.Connectionreturnr )rr]rr r^None)rr]r)rr*rr^bool)rr]r)rr^r`)rr]r^r_)__doc__ __future__rloggingrtypingr getLogger__name__r3rr frozensetr'r-r/r6r9r<r>r@rBrDrHrQ__annotations__maxrVrTr\rrrks #   8 $$  "G!KH;< J+E\ => ?$ ? ; Z__&'Qr