@veewo/gitnexus 1.3.11 → 1.4.6-rc

package/dist/core/{kuzu/kuzu-adapter.js → lbug/lbug-adapter.js}

@@ -1,14 +1,21 @@
 import fs from 'fs/promises';
+import { createReadStream } from 'fs';
+import { createInterface } from 'readline';
 import path from 'path';
-import kuzu from 'kuzu';
+import lbug from '@ladybugdb/core';
 import { NODE_TABLES, REL_TABLE_NAME, SCHEMA_QUERIES, EMBEDDING_TABLE_NAME, } from './schema.js';
 import { streamAllCSVsToDisk } from './csv-generator.js';
-import { streamRelationshipPairBucketsFromCsv } from './relationship-pair-buckets.js';
 let db = null;
 let conn = null;
 let currentDbPath = null;
 let ftsLoaded = false;
-// Global session lock for operations that touch module-level kuzu globals.
+export const getOpenLbugDatabase = (dbPath) => {
+    if (currentDbPath !== dbPath || !db) {
+        return null;
+    }
+    return db;
+};
+// Global session lock for operations that touch module-level lbug globals.
 // This guarantees no DB switch can happen while an operation is running.
 let sessionLock = Promise.resolve();
 const runWithSessionLock = async (operation) => {
@@ -26,27 +33,27 @@ const runWithSessionLock = async (operation) => {
     }
 };
 const normalizeCopyPath = (filePath) => filePath.replace(/\\/g, '/');
-export const initKuzu = async (dbPath) => {
-    return runWithSessionLock(() => ensureKuzuInitialized(dbPath));
+export const initLbug = async (dbPath) => {
+    return runWithSessionLock(() => ensureLbugInitialized(dbPath));
 };
 /**
  * Execute multiple queries against one repo DB atomically.
  * While the callback runs, no other request can switch the active DB.
  */
-export const withKuzuDb = async (dbPath, operation) => {
+export const withLbugDb = async (dbPath, operation) => {
     return runWithSessionLock(async () => {
-        await ensureKuzuInitialized(dbPath);
+        await ensureLbugInitialized(dbPath);
         return operation();
     });
 };
-const ensureKuzuInitialized = async (dbPath) => {
+const ensureLbugInitialized = async (dbPath) => {
     if (conn && currentDbPath === dbPath) {
         return { db, conn };
     }
-    await doInitKuzu(dbPath);
+    await doInitLbug(dbPath);
     return { db, conn };
 };
-const doInitKuzu = async (dbPath) => {
+const doInitLbug = async (dbPath) => {
     // Different database requested — close the old one first
     if (conn || db) {
         try {
@@ -64,32 +71,36 @@ const doInitKuzu = async (dbPath) => {
         currentDbPath = null;
         ftsLoaded = false;
     }
-    // kuzu v0.11 stores the database as a single file (not a directory).
-    // If the path already exists, it must be a valid kuzu database file.
+    // LadybugDB stores the database as a single file (not a directory).
+    // If the path already exists, it must be a valid LadybugDB database file.
     // Remove stale empty directories or files from older versions.
     try {
-        const stat = await fs.stat(dbPath);
-        if (stat.isDirectory()) {
-            // Old-style directory database or empty leftover - remove it
-            const files = await fs.readdir(dbPath);
-            if (files.length === 0) {
-                await fs.rmdir(dbPath);
-            }
-            else {
-                // Non-empty directory from older kuzu version - remove entire directory
-                await fs.rm(dbPath, { recursive: true, force: true });
+        const stat = await fs.lstat(dbPath);
+        if (stat.isSymbolicLink()) {
+            // Never follow symlinks — just remove the link itself
+            await fs.unlink(dbPath);
+        }
+        else if (stat.isDirectory()) {
+            // Verify path is within expected storage directory before deleting
+            const realPath = await fs.realpath(dbPath);
+            const parentDir = path.dirname(dbPath);
+            const realParent = await fs.realpath(parentDir);
+            if (!realPath.startsWith(realParent + path.sep) && realPath !== realParent) {
+                throw new Error(`Refusing to delete ${dbPath}: resolved path ${realPath} is outside storage directory`);
             }
+            // Old-style directory database or empty leftover - remove it
+            await fs.rm(dbPath, { recursive: true, force: true });
         }
-        // If it's a file, assume it's an existing kuzu database - kuzu will open it
+        // If it's a file, assume it's an existing LadybugDB database - LadybugDB will open it
     }
     catch {
-        // Path doesn't exist, which is what kuzu wants for a new database
+        // Path doesn't exist, which is what LadybugDB wants for a new database
     }
     // Ensure parent directory exists
     const parentDir = path.dirname(dbPath);
     await fs.mkdir(parentDir, { recursive: true });
-    db = new kuzu.Database(dbPath);
-    conn = new kuzu.Connection(db);
+    db = new lbug.Database(dbPath);
+    conn = new lbug.Connection(db);
     for (const schemaQuery of SCHEMA_QUERIES) {
         try {
             await conn.query(schemaQuery);
@@ -105,9 +116,9 @@ const doInitKuzu = async (dbPath) => {
     currentDbPath = dbPath;
     return { db, conn };
 };
-export const loadGraphToKuzu = async (graph, repoPath, storagePath, onProgress) => {
+export const loadGraphToLbug = async (graph, repoPath, storagePath, onProgress) => {
     if (!conn) {
-        throw new Error('KuzuDB not initialized. Call initKuzu first.');
+        throw new Error('LadybugDB not initialized. Call initLbug first.');
     }
     const log = onProgress || (() => { });
     const csvDir = path.join(storagePath, 'csv');
@@ -121,7 +132,7 @@ export const loadGraphToKuzu = async (graph, repoPath, storagePath, onProgress)
             return 'Process';
         return nodeId.split(':')[0];
     };
-    // Bulk COPY all node CSVs (sequential — KuzuDB allows only one write txn at a time)
+    // Bulk COPY all node CSVs (sequential — LadybugDB allows only one write txn at a time)
     const nodeFiles = [...csvResult.nodeFiles.entries()];
     const totalSteps = nodeFiles.length + 1; // +1 for relationships
     let stepsDone = 0;
@@ -144,33 +155,61 @@ export const loadGraphToKuzu = async (graph, repoPath, storagePath, onProgress)
             }
         }
     }
-    // Bulk COPY relationships — split by FROM→TO label pair (KuzuDB requires it)
-    // Stream relation CSV into per-pair temporary CSV files to avoid retaining
-    // all relationship lines in memory at once.
-    const pairBucketResult = await streamRelationshipPairBucketsFromCsv({
-        relCsvPath: csvResult.relCsvPath,
-        csvDir,
-        validTables,
-        getNodeLabel,
+    // Bulk COPY relationships — split by FROM→TO label pair (LadybugDB requires it)
+    // Stream-read the relation CSV line by line to avoid exceeding V8 max string length
+    let relHeader = '';
+    const relsByPair = new Map();
+    let skippedRels = 0;
+    let totalValidRels = 0;
+    await new Promise((resolve, reject) => {
+        const rl = createInterface({ input: createReadStream(csvResult.relCsvPath, 'utf-8'), crlfDelay: Infinity });
+        let isFirst = true;
+        rl.on('line', (line) => {
+            if (isFirst) {
+                relHeader = line;
+                isFirst = false;
+                return;
+            }
+            if (!line.trim())
+                return;
+            const match = line.match(/"([^"]*)","([^"]*)"/);
+            if (!match) {
+                skippedRels++;
+                return;
+            }
+            const fromLabel = getNodeLabel(match[1]);
+            const toLabel = getNodeLabel(match[2]);
+            if (!validTables.has(fromLabel) || !validTables.has(toLabel)) {
+                skippedRels++;
+                return;
+            }
+            const pairKey = `${fromLabel}|${toLabel}`;
+            let list = relsByPair.get(pairKey);
+            if (!list) {
+                list = [];
+                relsByPair.set(pairKey, list);
+            }
+            list.push(line);
+            totalValidRels++;
+        });
+        rl.on('close', resolve);
+        rl.on('error', reject);
     });
-    const relHeader = pairBucketResult.relHeader;
-    const relsByPair = pairBucketResult.buckets;
-    const skippedRels = pairBucketResult.skippedRels;
-    const totalValidRels = pairBucketResult.totalValidRels;
     const insertedRels = totalValidRels;
     const warnings = [];
-    let fallbackStats = { attempted: 0, succeeded: 0, failed: 0 };
     if (insertedRels > 0) {
         log(`Loading edges: ${insertedRels.toLocaleString()} across ${relsByPair.size} types`);
         let pairIdx = 0;
         let failedPairEdges = 0;
-        const failedPairCsvPaths = [];
-        for (const [pairKey, bucket] of relsByPair) {
+        const failedPairLines = [];
+        for (const [pairKey, lines] of relsByPair) {
             pairIdx++;
             const [fromLabel, toLabel] = pairKey.split('|');
-            const normalizedPath = normalizeCopyPath(bucket.csvPath);
+            const pairCsvPath = path.join(csvDir, `rel_${fromLabel}_${toLabel}.csv`);
+            await fs.writeFile(pairCsvPath, relHeader + '\n' + lines.join('\n'), 'utf-8');
+            const normalizedPath = normalizeCopyPath(pairCsvPath);
             const copyQuery = `COPY ${REL_TABLE_NAME} FROM "${normalizedPath}" (from="${fromLabel}", to="${toLabel}", HEADER=true, ESCAPE='"', DELIM=',', QUOTE='"', PARALLEL=false, auto_detect=false)`;
-            if (pairIdx % 5 === 0 || bucket.rowCount > 1000) {
+            if (pairIdx % 5 === 0 || lines.length > 1000) {
                 log(`Loading edges: ${pairIdx}/${relsByPair.size} types (${fromLabel} -> ${toLabel})`);
             }
             try {
@@ -183,31 +222,20 @@ export const loadGraphToKuzu = async (graph, repoPath, storagePath, onProgress)
                 }
                 catch (retryErr) {
                     const retryMsg = retryErr instanceof Error ? retryErr.message : String(retryErr);
-                    warnings.push(`${fromLabel}->${toLabel} (${bucket.rowCount} edges): ${retryMsg.slice(0, 80)}`);
-                    failedPairEdges += bucket.rowCount;
-                    failedPairCsvPaths.push(bucket.csvPath);
+                    warnings.push(`${fromLabel}->${toLabel} (${lines.length} edges): ${retryMsg.slice(0, 80)}`);
+                    failedPairEdges += lines.length;
+                    failedPairLines.push(...lines);
                 }
             }
-        }
-        if (failedPairCsvPaths.length > 0) {
-            const failedPairLines = [relHeader];
-            for (const failedPairPath of failedPairCsvPaths) {
-                const raw = await fs.readFile(failedPairPath, 'utf-8');
-                const lines = raw
-                    .split('\n')
-                    .slice(1)
-                    .filter(line => line.trim().length > 0);
-                failedPairLines.push(...lines);
-            }
-            log(`Inserting ${failedPairEdges} edges individually (missing schema pairs)`);
-            fallbackStats = await fallbackRelationshipInserts(failedPairLines, validTables, getNodeLabel);
-        }
-        for (const [, bucket] of relsByPair) {
             try {
-                await fs.unlink(bucket.csvPath);
+                await fs.unlink(pairCsvPath);
             }
             catch { }
         }
+        if (failedPairLines.length > 0) {
+            log(`Inserting ${failedPairEdges} edges individually (missing schema pairs)`);
+            await fallbackRelationshipInserts([relHeader, ...failedPairLines], validTables, getNodeLabel);
+        }
     }
     // Cleanup all CSVs
     try {
@@ -234,12 +262,12 @@ export const loadGraphToKuzu = async (graph, repoPath, storagePath, onProgress)
         await fs.rmdir(csvDir);
     }
     catch { }
-    return { success: true, insertedRels, skippedRels, warnings, fallbackStats };
+    return { success: true, insertedRels, skippedRels, warnings };
 };
-// KuzuDB default ESCAPE is '\' (backslash), but our CSV uses RFC 4180 escaping ("" for literal quotes).
+// LadybugDB default ESCAPE is '\' (backslash), but our CSV uses RFC 4180 escaping ("" for literal quotes).
 // Source code content is full of backslashes which confuse the auto-detection.
 // We MUST explicitly set ESCAPE='"' to use RFC 4180 escaping, and disable auto_detect to prevent
-// KuzuDB from overriding our settings based on sample rows.
+// LadybugDB from overriding our settings based on sample rows.
 const COPY_CSV_OPTS = `(HEADER=true, ESCAPE='"', DELIM=',', QUOTE='"', PARALLEL=false, auto_detect=false)`;
 // Multi-language table names that were created with backticks in CODE_ELEMENT_BASE
 // and must always be referenced with backticks in queries
@@ -253,44 +281,35 @@ const escapeTableName = (table) => {
 };
 /** Fallback: insert relationships one-by-one if COPY fails */
 const fallbackRelationshipInserts = async (validRelLines, validTables, getNodeLabel) => {
-    const attempted = Math.max(0, validRelLines.length - 1);
-    if (!conn) {
-        return { attempted, succeeded: 0, failed: attempted };
-    }
+    if (!conn)
+        return;
     const escapeLabel = (label) => {
         return BACKTICK_TABLES.has(label) ? `\`${label}\`` : label;
     };
-    let succeeded = 0;
-    let failed = 0;
     for (let i = 1; i < validRelLines.length; i++) {
         const line = validRelLines[i];
         try {
             const match = line.match(/"([^"]*)","([^"]*)","([^"]*)",([0-9.]+),"([^"]*)",([0-9-]+)/);
-            if (!match) {
-                failed++;
+            if (!match)
                 continue;
-            }
             const [, fromId, toId, relType, confidenceStr, reason, stepStr] = match;
             const fromLabel = getNodeLabel(fromId);
             const toLabel = getNodeLabel(toId);
-            if (!validTables.has(fromLabel) || !validTables.has(toLabel)) {
-                failed++;
+            if (!validTables.has(fromLabel) || !validTables.has(toLabel))
                 continue;
-            }
             const confidence = parseFloat(confidenceStr) || 1.0;
             const step = parseInt(stepStr) || 0;
+            const esc = (s) => s.replace(/'/g, "''").replace(/\\/g, '\\\\').replace(/\n/g, '\\n').replace(/\r/g, '\\r');
             await conn.query(`
-        MATCH (a:${escapeLabel(fromLabel)} {id: '${fromId.replace(/'/g, "''")}' }),
-              (b:${escapeLabel(toLabel)} {id: '${toId.replace(/'/g, "''")}' })
-        CREATE (a)-[:${REL_TABLE_NAME} {type: '${relType}', confidence: ${confidence}, reason: '${reason.replace(/'/g, "''")}', step: ${step}}]->(b)
+        MATCH (a:${escapeLabel(fromLabel)} {id: '${esc(fromId)}' }),
+              (b:${escapeLabel(toLabel)} {id: '${esc(toId)}' })
+        CREATE (a)-[:${REL_TABLE_NAME} {type: '${esc(relType)}', confidence: ${confidence}, reason: '${esc(reason)}', step: ${step}}]->(b)
       `);
-            succeeded++;
         }
         catch {
-            failed++;
+            // skip
         }
     }
-    return { attempted, succeeded, failed };
 };
 /** Tables with isExported column (TypeScript/JS-native types) */
 const TABLES_WITH_EXPORTED = new Set(['Function', 'Class', 'Interface', 'Method', 'CodeElement']);
@@ -308,6 +327,9 @@ const getCopyQuery = (table, filePath) => {
     if (table === 'Process') {
         return `COPY ${t}(id, label, heuristicLabel, processType, stepCount, communities, entryPointId, terminalId) FROM "${filePath}" ${COPY_CSV_OPTS}`;
     }
+    if (table === 'Method') {
+        return `COPY ${t}(id, name, filePath, startLine, endLine, isExported, content, description, parameterCount, returnType) FROM "${filePath}" ${COPY_CSV_OPTS}`;
+    }
     // TypeScript/JS code element tables have isExported; multi-language tables do not
     if (TABLES_WITH_EXPORTED.has(table)) {
         return `COPY ${t}(id, name, filePath, startLine, endLine, isExported, content, description) FROM "${filePath}" ${COPY_CSV_OPTS}`;
@@ -316,16 +338,16 @@ const getCopyQuery = (table, filePath) => {
     return `COPY ${t}(id, name, filePath, startLine, endLine, content, description) FROM "${filePath}" ${COPY_CSV_OPTS}`;
 };
 /**
- * Insert a single node to KuzuDB
+ * Insert a single node to LadybugDB
  * @param label - Node type (File, Function, Class, etc.)
  * @param properties - Node properties
- * @param dbPath - Path to KuzuDB database (optional if already initialized)
+ * @param dbPath - Path to LadybugDB database (optional if already initialized)
  */
-export const insertNodeToKuzu = async (label, properties, dbPath) => {
+export const insertNodeToLbug = async (label, properties, dbPath) => {
     // Use provided dbPath or fall back to module-level db
     const targetDbPath = dbPath || (db ? undefined : null);
     if (!targetDbPath && !db) {
-        throw new Error('KuzuDB not initialized. Provide dbPath or call initKuzu first.');
+        throw new Error('LadybugDB not initialized. Provide dbPath or call initLbug first.');
     }
     try {
         const escapeValue = (v) => {
@@ -334,7 +356,7 @@ export const insertNodeToKuzu = async (label, properties, dbPath) => {
             if (typeof v === 'number')
                 return String(v);
             // Escape backslashes first (for Windows paths), then single quotes
-            return `'${String(v).replace(/\\/g, '\\\\').replace(/'/g, "''")}'`;
+            return `'${String(v).replace(/\\/g, '\\\\').replace(/'/g, "''").replace(/\n/g, '\\n').replace(/\r/g, '\\r')}'`;
         };
         // Build INSERT query based on node type
         const t = escapeTableName(label);
@@ -356,8 +378,8 @@ export const insertNodeToKuzu = async (label, properties, dbPath) => {
         }
         // Use per-query connection if dbPath provided (avoids lock conflicts)
         if (targetDbPath) {
-            const tempDb = new kuzu.Database(targetDbPath);
-            const tempConn = new kuzu.Connection(tempDb);
+            const tempDb = new lbug.Database(targetDbPath);
+            const tempConn = new lbug.Connection(tempDb);
             try {
                 await tempConn.query(query);
                 return true;
@@ -387,12 +409,12 @@ export const insertNodeToKuzu = async (label, properties, dbPath) => {
     }
 };
 /**
- * Batch insert multiple nodes to KuzuDB using a single connection
+ * Batch insert multiple nodes to LadybugDB using a single connection
  * @param nodes - Array of {label, properties} to insert
- * @param dbPath - Path to KuzuDB database
+ * @param dbPath - Path to LadybugDB database
  * @returns Object with success count and error count
  */
-export const batchInsertNodesToKuzu = async (nodes, dbPath) => {
+export const batchInsertNodesToLbug = async (nodes, dbPath) => {
     if (nodes.length === 0)
         return { inserted: 0, failed: 0 };
     const escapeValue = (v) => {
@@ -400,12 +422,12 @@ export const batchInsertNodesToKuzu = async (nodes, dbPath) => {
             return 'NULL';
         if (typeof v === 'number')
             return String(v);
-        // Escape backslashes first (for Windows paths), then single quotes
-        return `'${String(v).replace(/\\/g, '\\\\').replace(/'/g, "''")}'`;
+        // Escape backslashes first (for Windows paths), then single quotes, then newlines
+        return `'${String(v).replace(/\\/g, '\\\\').replace(/'/g, "''").replace(/\n/g, '\\n').replace(/\r/g, '\\r')}'`;
     };
     // Open a single connection for all inserts
-    const tempDb = new kuzu.Database(dbPath);
-    const tempConn = new kuzu.Connection(tempDb);
+    const tempDb = new lbug.Database(dbPath);
+    const tempConn = new lbug.Connection(tempDb);
     let inserted = 0;
     let failed = 0;
     try {
@@ -451,10 +473,10 @@ export const batchInsertNodesToKuzu = async (nodes, dbPath) => {
 };
 export const executeQuery = async (cypher) => {
     if (!conn) {
-        throw new Error('KuzuDB not initialized. Call initKuzu first.');
+        throw new Error('LadybugDB not initialized. Call initLbug first.');
     }
     const queryResult = await conn.query(cypher);
-    // kuzu v0.11 uses getAll() instead of hasNext()/getNext()
+    // LadybugDB uses getAll() instead of hasNext()/getNext()
     // Query returns QueryResult for single queries, QueryResult[] for multi-statement
     const result = Array.isArray(queryResult) ? queryResult[0] : queryResult;
     const rows = await result.getAll();
@@ -462,7 +484,7 @@ export const executeQuery = async (cypher) => {
 };
 export const executeWithReusedStatement = async (cypher, paramsList) => {
     if (!conn) {
-        throw new Error('KuzuDB not initialized. Call initKuzu first.');
+        throw new Error('LadybugDB not initialized. Call initLbug first.');
     }
     if (paramsList.length === 0)
         return;
@@ -483,10 +505,10 @@ export const executeWithReusedStatement = async (cypher, paramsList) => {
             // Log the error and continue with next batch
             console.warn('Batch execution error:', e);
         }
-        // Note: kuzu 0.8.2 PreparedStatement doesn't require explicit close()
+        // Note: LadybugDB PreparedStatement doesn't require explicit close()
     }
 };
-export const getKuzuStats = async () => {
+export const getLbugStats = async () => {
     if (!conn)
         return { nodes: 0, edges: 0 };
     let totalNodes = 0;
@@ -518,7 +540,7 @@ export const getKuzuStats = async () => {
     return { nodes: totalNodes, edges: totalEdges };
 };
 /**
- * Load cached embeddings from KuzuDB before a rebuild.
+ * Load cached embeddings from LadybugDB before a rebuild.
  * Returns all embedding vectors so they can be re-inserted after the graph is reloaded,
  * avoiding expensive re-embedding of unchanged nodes.
  */
@@ -548,7 +570,7 @@ export const loadCachedEmbeddings = async () => {
     catch { /* embedding table may not exist */ }
     return { embeddingNodeIds, embeddings };
 };
-export const closeKuzu = async () => {
+export const closeLbug = async () => {
     if (conn) {
         try {
             await conn.close();
@@ -566,11 +588,11 @@ export const closeKuzu = async () => {
     currentDbPath = null;
     ftsLoaded = false;
 };
-export const isKuzuReady = () => conn !== null && db !== null;
+export const isLbugReady = () => conn !== null && db !== null;
 /**
- * Delete all nodes (and their relationships) for a specific file from KuzuDB
+ * Delete all nodes (and their relationships) for a specific file from LadybugDB
  * @param filePath - The file path to delete nodes for
- * @param dbPath - Optional path to KuzuDB for per-query connection
+ * @param dbPath - Optional path to LadybugDB for per-query connection
  * @returns Object with counts of deleted nodes
  */
 export const deleteNodesForFile = async (filePath, dbPath) => {
@@ -580,12 +602,12 @@ export const deleteNodesForFile = async (filePath, dbPath) => {
     let tempConn = null;
     let targetConn = conn;
     if (usePerQuery) {
-        tempDb = new kuzu.Database(dbPath);
-        tempConn = new kuzu.Connection(tempDb);
+        tempDb = new lbug.Database(dbPath);
+        tempConn = new lbug.Connection(tempDb);
         targetConn = tempConn;
     }
     else if (!conn) {
-        throw new Error('KuzuDB not initialized. Provide dbPath or call initKuzu first.');
+        throw new Error('LadybugDB not initialized. Provide dbPath or call initLbug first.');
     }
     try {
         let deletedNodes = 0;
@@ -650,16 +672,22 @@ export const loadFTSExtension = async () => {
     if (ftsLoaded)
         return;
     if (!conn) {
-        throw new Error('KuzuDB not initialized. Call initKuzu first.');
+        throw new Error('LadybugDB not initialized. Call initLbug first.');
     }
     try {
         await conn.query('INSTALL fts');
         await conn.query('LOAD EXTENSION fts');
+        ftsLoaded = true;
     }
-    catch {
-        // Extension may already be loaded
+    catch (err) {
+        const msg = err?.message || '';
+        if (msg.includes('already loaded') || msg.includes('already installed') || msg.includes('already exists')) {
+            ftsLoaded = true;
+        }
+        else {
+            console.error('GitNexus: FTS extension load failed:', msg);
+        }
     }
-    ftsLoaded = true;
 };
 /**
  * Create a full-text search index on a table
@@ -670,7 +698,7 @@ export const loadFTSExtension = async () => {
  */
 export const createFTSIndex = async (tableName, indexName, properties, stemmer = 'porter') => {
     if (!conn) {
-        throw new Error('KuzuDB not initialized. Call initKuzu first.');
+        throw new Error('LadybugDB not initialized. Call initLbug first.');
     }
     await loadFTSExtension();
     const propList = properties.map(p => `'${p}'`).join(', ');
@@ -695,10 +723,10 @@ export const createFTSIndex = async (tableName, indexName, properties, stemmer =
  */
 export const queryFTS = async (tableName, indexName, query, limit = 20, conjunctive = false) => {
     if (!conn) {
-        throw new Error('KuzuDB not initialized. Call initKuzu first.');
+        throw new Error('LadybugDB not initialized. Call initLbug first.');
     }
-    // Escape single quotes in query
-    const escapedQuery = query.replace(/'/g, "''");
+    // Escape backslashes and single quotes to prevent Cypher injection
+    const escapedQuery = query.replace(/\\/g, '\\\\').replace(/'/g, "''");
     const cypher = `
     CALL QUERY_FTS_INDEX('${tableName}', '${indexName}', '${escapedQuery}', conjunctive := ${conjunctive})
     RETURN node, score
@@ -734,7 +762,7 @@ export const queryFTS = async (tableName, indexName, query, limit = 20, conjunct
  */
 export const dropFTSIndex = async (tableName, indexName) => {
     if (!conn) {
-        throw new Error('KuzuDB not initialized. Call initKuzu first.');
+        throw new Error('LadybugDB not initialized. Call initLbug first.');
     }
     try {
         await conn.query(`CALL DROP_FTS_INDEX('${tableName}', '${indexName}')`);

package/dist/core/{kuzu → lbug}/schema.d.ts

@@ -1,5 +1,5 @@
 /**
- * KuzuDB Schema Definitions
+ * LadybugDB Schema Definitions
  *
  * Hybrid Schema:
  * - Separate node tables for each code element type (File, Function, Class, etc.)
@@ -11,7 +11,7 @@
 export declare const NODE_TABLES: readonly ["File", "Folder", "Function", "Class", "Interface", "Method", "CodeElement", "Community", "Process", "Struct", "Enum", "Macro", "Typedef", "Union", "Namespace", "Trait", "Impl", "TypeAlias", "Const", "Static", "Property", "Record", "Delegate", "Annotation", "Constructor", "Template", "Module"];
 export type NodeTableName = typeof NODE_TABLES[number];
 export declare const REL_TABLE_NAME = "CodeRelation";
-export declare const REL_TYPES: readonly ["CONTAINS", "DEFINES", "IMPORTS", "CALLS", "EXTENDS", "IMPLEMENTS", "MEMBER_OF", "STEP_IN_PROCESS", "UNITY_COMPONENT_IN", "UNITY_COMPONENT_INSTANCE", "UNITY_RESOURCE_SUMMARY", "UNITY_SERIALIZED_TYPE_IN"];
+export declare const REL_TYPES: readonly ["CONTAINS", "DEFINES", "IMPORTS", "CALLS", "EXTENDS", "IMPLEMENTS", "HAS_METHOD", "OVERRIDES", "MEMBER_OF", "STEP_IN_PROCESS"];
 export type RelType = typeof REL_TYPES[number];
 export declare const EMBEDDING_TABLE_NAME = "CodeEmbedding";
 export declare const FILE_SCHEMA = "\nCREATE NODE TABLE File (\n  id STRING,\n  name STRING,\n  filePath STRING,\n  content STRING,\n  PRIMARY KEY (id)\n)";
@@ -19,7 +19,7 @@ export declare const FOLDER_SCHEMA = "\nCREATE NODE TABLE Folder (\n  id STRING,
 export declare const FUNCTION_SCHEMA = "\nCREATE NODE TABLE Function (\n  id STRING,\n  name STRING,\n  filePath STRING,\n  startLine INT64,\n  endLine INT64,\n  isExported BOOLEAN,\n  content STRING,\n  description STRING,\n  PRIMARY KEY (id)\n)";
 export declare const CLASS_SCHEMA = "\nCREATE NODE TABLE Class (\n  id STRING,\n  name STRING,\n  filePath STRING,\n  startLine INT64,\n  endLine INT64,\n  isExported BOOLEAN,\n  content STRING,\n  description STRING,\n  PRIMARY KEY (id)\n)";
 export declare const INTERFACE_SCHEMA = "\nCREATE NODE TABLE Interface (\n  id STRING,\n  name STRING,\n  filePath STRING,\n  startLine INT64,\n  endLine INT64,\n  isExported BOOLEAN,\n  content STRING,\n  description STRING,\n  PRIMARY KEY (id)\n)";
-export declare const METHOD_SCHEMA = "\nCREATE NODE TABLE Method (\n  id STRING,\n  name STRING,\n  filePath STRING,\n  startLine INT64,\n  endLine INT64,\n  isExported BOOLEAN,\n  content STRING,\n  description STRING,\n  PRIMARY KEY (id)\n)";
+export declare const METHOD_SCHEMA = "\nCREATE NODE TABLE Method (\n  id STRING,\n  name STRING,\n  filePath STRING,\n  startLine INT64,\n  endLine INT64,\n  isExported BOOLEAN,\n  content STRING,\n  description STRING,\n  parameterCount INT32,\n  returnType STRING,\n  PRIMARY KEY (id)\n)";
 export declare const CODE_ELEMENT_SCHEMA = "\nCREATE NODE TABLE CodeElement (\n  id STRING,\n  name STRING,\n  filePath STRING,\n  startLine INT64,\n  endLine INT64,\n  isExported BOOLEAN,\n  content STRING,\n  description STRING,\n  PRIMARY KEY (id)\n)";
 export declare const COMMUNITY_SCHEMA = "\nCREATE NODE TABLE Community (\n  id STRING,\n  label STRING,\n  heuristicLabel STRING,\n  keywords STRING[],\n  description STRING,\n  enrichedBy STRING,\n  cohesion DOUBLE,\n  symbolCount INT32,\n  PRIMARY KEY (id)\n)";
 export declare const PROCESS_SCHEMA = "\nCREATE NODE TABLE Process (\n  id STRING,\n  label STRING,\n  heuristicLabel STRING,\n  processType STRING,\n  stepCount INT32,\n  communities STRING[],\n  entryPointId STRING,\n  terminalId STRING,\n  PRIMARY KEY (id)\n)";
@@ -41,7 +41,7 @@ export declare const ANNOTATION_SCHEMA: string;
 export declare const CONSTRUCTOR_SCHEMA: string;
 export declare const TEMPLATE_SCHEMA: string;
 export declare const MODULE_SCHEMA: string;
-export declare const RELATION_SCHEMA = "\nCREATE REL TABLE CodeRelation (\n  FROM File TO File,\n  FROM File TO Folder,\n  FROM File TO Function,\n  FROM File TO Class,\n  FROM File TO Interface,\n  FROM File TO Method,\n  FROM File TO CodeElement,\n  FROM File TO `Struct`,\n  FROM File TO `Enum`,\n  FROM File TO `Macro`,\n  FROM File TO `Typedef`,\n  FROM File TO `Union`,\n  FROM File TO `Namespace`,\n  FROM File TO `Trait`,\n  FROM File TO `Impl`,\n  FROM File TO `TypeAlias`,\n  FROM File TO `Const`,\n  FROM File TO `Static`,\n  FROM File TO `Property`,\n  FROM File TO `Record`,\n  FROM File TO `Delegate`,\n  FROM File TO `Annotation`,\n  FROM File TO `Constructor`,\n  FROM File TO `Template`,\n  FROM File TO `Module`,\n  FROM Folder TO Folder,\n  FROM Folder TO File,\n  FROM Function TO Function,\n  FROM Function TO Method,\n  FROM Function TO Class,\n  FROM Function TO Community,\n  FROM Function TO `Macro`,\n  FROM Function TO `Struct`,\n  FROM Function TO `Template`,\n  FROM Function TO `Enum`,\n  FROM Function TO `Namespace`,\n  FROM Function TO `TypeAlias`,\n  FROM Function TO `Module`,\n  FROM Function TO `Impl`,\n  FROM Function TO Interface,\n  FROM Function TO `Constructor`,\n  FROM Function TO `Property`,\n  FROM Function TO `Const`,\n  FROM Function TO `Typedef`,\n  FROM Function TO `Union`,\n  FROM Class TO Method,\n  FROM Class TO Function,\n  FROM Class TO Class,\n  FROM Class TO File,\n  FROM Class TO CodeElement,\n  FROM Class TO Interface,\n  FROM Class TO Community,\n  FROM Class TO `Template`,\n  FROM Class TO `TypeAlias`,\n  FROM Class TO `Struct`,\n  FROM Class TO `Enum`,\n  FROM Class TO `Annotation`,\n  FROM Class TO `Constructor`,\n  FROM Class TO `Trait`,\n  FROM Class TO `Macro`,\n  FROM Class TO `Impl`,\n  FROM Class TO `Union`,\n  FROM Class TO `Namespace`,\n  FROM Class TO `Typedef`,\n  FROM Class TO `Property`,\n  FROM Class TO `Delegate`,\n  FROM Method TO Function,\n  FROM Method TO Method,\n  FROM Method TO Class,\n  FROM Method TO Community,\n  FROM Method TO `Template`,\n  FROM Method TO `Struct`,\n  FROM Method TO `TypeAlias`,\n  FROM Method TO `Enum`,\n  FROM Method TO `Macro`,\n  FROM Method TO `Namespace`,\n  FROM Method TO `Module`,\n  FROM Method TO `Impl`,\n  FROM Method TO Interface,\n  FROM Method TO `Constructor`,\n  FROM Method TO `Property`,\n  FROM Method TO `Delegate`,\n  FROM `Template` TO `Template`,\n  FROM `Template` TO Function,\n  FROM `Template` TO Method,\n  FROM `Template` TO Class,\n  FROM `Template` TO `Struct`,\n  FROM `Template` TO `TypeAlias`,\n  FROM `Template` TO `Enum`,\n  FROM `Template` TO `Macro`,\n  FROM `Template` TO Interface,\n  FROM `Template` TO `Constructor`,\n  FROM `Module` TO `Module`,\n  FROM CodeElement TO Community,\n  FROM Interface TO Community,\n  FROM Interface TO Function,\n  FROM Interface TO Method,\n  FROM Interface TO Class,\n  FROM Interface TO Interface,\n  FROM Interface TO `TypeAlias`,\n  FROM Interface TO `Struct`,\n  FROM Interface TO `Constructor`,\n  FROM `Struct` TO Community,\n  FROM `Struct` TO `Trait`,\n  FROM `Struct` TO `Struct`,\n  FROM `Struct` TO Class,\n  FROM `Struct` TO `Enum`,\n  FROM `Struct` TO Function,\n  FROM `Struct` TO Method,\n  FROM `Enum` TO Community,\n  FROM `Macro` TO Community,\n  FROM `Macro` TO Function,\n  FROM `Macro` TO Method,\n  FROM `Module` TO Function,\n  FROM `Module` TO Method,\n  FROM `Typedef` TO Community,\n  FROM `Union` TO Community,\n  FROM `Namespace` TO Community,\n  FROM `Namespace` TO `Struct`,\n  FROM `Trait` TO Community,\n  FROM `Impl` TO Community,\n  FROM `Impl` TO `Trait`,\n  FROM `Impl` TO `Struct`,\n  FROM `Impl` TO `Impl`,\n  FROM `TypeAlias` TO Community,\n  FROM `TypeAlias` TO `Trait`,\n  FROM `Const` TO Community,\n  FROM `Static` TO Community,\n  FROM `Property` TO Community,\n  FROM `Property` TO Class,\n  FROM `Property` TO Interface,\n  FROM `Record` TO Community,\n  FROM `Delegate` TO Community,\n  FROM `Annotation` TO Community,\n  FROM `Constructor` TO Community,\n  FROM `Constructor` TO Interface,\n  FROM `Constructor` TO Class,\n  FROM `Constructor` TO Method,\n  FROM `Constructor` TO Function,\n  FROM `Constructor` TO `Constructor`,\n  FROM `Constructor` TO `Struct`,\n  FROM `Constructor` TO `Macro`,\n  FROM `Constructor` TO `Template`,\n  FROM `Constructor` TO `TypeAlias`,\n  FROM `Constructor` TO `Enum`,\n  FROM `Constructor` TO `Annotation`,\n  FROM `Constructor` TO `Impl`,\n  FROM `Constructor` TO `Namespace`,\n  FROM `Constructor` TO `Module`,\n  FROM `Constructor` TO `Property`,\n  FROM `Template` TO Community,\n  FROM `Module` TO Community,\n  FROM Function TO Process,\n  FROM Method TO Process,\n  FROM Class TO Process,\n  FROM Interface TO Process,\n  FROM `Struct` TO Process,\n  FROM `Constructor` TO Process,\n  FROM `Module` TO Process,\n  FROM `Macro` TO Process,\n  FROM `Impl` TO Process,\n  FROM `Typedef` TO Process,\n  FROM `TypeAlias` TO Process,\n  FROM `Enum` TO Process,\n  FROM `Union` TO Process,\n  FROM `Namespace` TO Process,\n  FROM `Trait` TO Process,\n  FROM `Const` TO Process,\n  FROM `Static` TO Process,\n  FROM `Property` TO Process,\n  FROM `Record` TO Process,\n  FROM `Delegate` TO Process,\n  FROM `Annotation` TO Process,\n  FROM `Template` TO Process,\n  FROM CodeElement TO Process,\n  type STRING,\n  confidence DOUBLE,\n  reason STRING,\n  step INT32\n)";
+export declare const RELATION_SCHEMA = "\nCREATE REL TABLE CodeRelation (\n  FROM File TO File,\n  FROM File TO Folder,\n  FROM File TO Function,\n  FROM File TO Class,\n  FROM File TO Interface,\n  FROM File TO Method,\n  FROM File TO CodeElement,\n  FROM File TO `Struct`,\n  FROM File TO `Enum`,\n  FROM File TO `Macro`,\n  FROM File TO `Typedef`,\n  FROM File TO `Union`,\n  FROM File TO `Namespace`,\n  FROM File TO `Trait`,\n  FROM File TO `Impl`,\n  FROM File TO `TypeAlias`,\n  FROM File TO `Const`,\n  FROM File TO `Static`,\n  FROM File TO `Property`,\n  FROM File TO `Record`,\n  FROM File TO `Delegate`,\n  FROM File TO `Annotation`,\n  FROM File TO `Constructor`,\n  FROM File TO `Template`,\n  FROM File TO `Module`,\n  FROM Folder TO Folder,\n  FROM Folder TO File,\n  FROM Function TO Function,\n  FROM Function TO Method,\n  FROM Function TO Class,\n  FROM Function TO Community,\n  FROM Function TO `Macro`,\n  FROM Function TO `Struct`,\n  FROM Function TO `Template`,\n  FROM Function TO `Enum`,\n  FROM Function TO `Namespace`,\n  FROM Function TO `TypeAlias`,\n  FROM Function TO `Module`,\n  FROM Function TO `Impl`,\n  FROM Function TO Interface,\n  FROM Function TO `Constructor`,\n  FROM Function TO `Const`,\n  FROM Function TO `Typedef`,\n  FROM Function TO `Union`,\n  FROM Function TO `Property`,\n  FROM Class TO Method,\n  FROM Class TO Function,\n  FROM Class TO Class,\n  FROM Class TO Interface,\n  FROM Class TO Community,\n  FROM Class TO `Template`,\n  FROM Class TO `TypeAlias`,\n  FROM Class TO `Struct`,\n  FROM Class TO `Enum`,\n  FROM Class TO `Annotation`,\n  FROM Class TO `Constructor`,\n  FROM Class TO `Trait`,\n  FROM Class TO `Macro`,\n  FROM Class TO `Impl`,\n  FROM Class TO `Union`,\n  FROM Class TO `Namespace`,\n  FROM Class TO `Typedef`,\n  FROM Class TO `Property`,\n  FROM Method TO Function,\n  FROM Method TO Method,\n  FROM Method TO Class,\n  FROM Method TO Community,\n  FROM Method TO `Template`,\n  FROM Method TO `Struct`,\n  FROM Method TO `TypeAlias`,\n  FROM Method TO `Enum`,\n  FROM Method TO `Macro`,\n  FROM Method TO `Namespace`,\n  FROM Method TO `Module`,\n  FROM Method TO `Impl`,\n  FROM Method TO Interface,\n  FROM Method TO `Constructor`,\n  FROM Method TO `Property`,\n  FROM `Template` TO `Template`,\n  FROM `Template` TO Function,\n  FROM `Template` TO Method,\n  FROM `Template` TO Class,\n  FROM `Template` TO `Struct`,\n  FROM `Template` TO `TypeAlias`,\n  FROM `Template` TO `Enum`,\n  FROM `Template` TO `Macro`,\n  FROM `Template` TO Interface,\n  FROM `Template` TO `Constructor`,\n  FROM `Module` TO `Module`,\n  FROM CodeElement TO Community,\n  FROM Interface TO Community,\n  FROM Interface TO Function,\n  FROM Interface TO Method,\n  FROM Interface TO Class,\n  FROM Interface TO Interface,\n  FROM Interface TO `TypeAlias`,\n  FROM Interface TO `Struct`,\n  FROM Interface TO `Constructor`,\n  FROM Interface TO `Property`,\n  FROM `Struct` TO Community,\n  FROM `Struct` TO `Trait`,\n  FROM `Struct` TO `Struct`,\n  FROM `Struct` TO Class,\n  FROM `Struct` TO `Enum`,\n  FROM `Struct` TO Function,\n  FROM `Struct` TO Method,\n  FROM `Struct` TO Interface,\n  FROM `Struct` TO `Constructor`,\n  FROM `Struct` TO `Property`,\n  FROM `Enum` TO `Enum`,\n  FROM `Enum` TO Community,\n  FROM `Enum` TO Class,\n  FROM `Enum` TO Interface,\n  FROM `Macro` TO Community,\n  FROM `Macro` TO Function,\n  FROM `Macro` TO Method,\n  FROM `Module` TO Function,\n  FROM `Module` TO Method,\n  FROM `Typedef` TO Community,\n  FROM `Union` TO Community,\n  FROM `Namespace` TO Community,\n  FROM `Namespace` TO `Struct`,\n  FROM `Trait` TO Method,\n  FROM `Trait` TO `Constructor`,\n  FROM `Trait` TO `Property`,\n  FROM `Trait` TO Community,\n  FROM `Impl` TO Method,\n  FROM `Impl` TO `Constructor`,\n  FROM `Impl` TO `Property`,\n  FROM `Impl` TO Community,\n  FROM `Impl` TO `Trait`,\n  FROM `Impl` TO `Struct`,\n  FROM `Impl` TO `Impl`,\n  FROM `TypeAlias` TO Community,\n  FROM `TypeAlias` TO `Trait`,\n  FROM `TypeAlias` TO Class,\n  FROM `Const` TO Community,\n  FROM `Static` TO Community,\n  FROM `Property` TO Community,\n  FROM `Record` TO Method,\n  FROM `Record` TO `Constructor`,\n  FROM `Record` TO `Property`,\n  FROM `Record` TO Community,\n  FROM `Delegate` TO Community,\n  FROM `Annotation` TO Community,\n  FROM `Constructor` TO Community,\n  FROM `Constructor` TO Interface,\n  FROM `Constructor` TO Class,\n  FROM `Constructor` TO Method,\n  FROM `Constructor` TO Function,\n  FROM `Constructor` TO `Constructor`,\n  FROM `Constructor` TO `Struct`,\n  FROM `Constructor` TO `Macro`,\n  FROM `Constructor` TO `Template`,\n  FROM `Constructor` TO `TypeAlias`,\n  FROM `Constructor` TO `Enum`,\n  FROM `Constructor` TO `Annotation`,\n  FROM `Constructor` TO `Impl`,\n  FROM `Constructor` TO `Namespace`,\n  FROM `Constructor` TO `Module`,\n  FROM `Constructor` TO `Property`,\n  FROM `Constructor` TO `Typedef`,\n  FROM `Template` TO Community,\n  FROM `Module` TO Community,\n  FROM Function TO Process,\n  FROM Method TO Process,\n  FROM Class TO Process,\n  FROM Interface TO Process,\n  FROM `Struct` TO Process,\n  FROM `Constructor` TO Process,\n  FROM `Module` TO Process,\n  FROM `Macro` TO Process,\n  FROM `Impl` TO Process,\n  FROM `Typedef` TO Process,\n  FROM `TypeAlias` TO Process,\n  FROM `Enum` TO Process,\n  FROM `Union` TO Process,\n  FROM `Namespace` TO Process,\n  FROM `Trait` TO Process,\n  FROM `Const` TO Process,\n  FROM `Static` TO Process,\n  FROM `Property` TO Process,\n  FROM `Record` TO Process,\n  FROM `Delegate` TO Process,\n  FROM `Annotation` TO Process,\n  FROM `Template` TO Process,\n  FROM CodeElement TO Process,\n  type STRING,\n  confidence DOUBLE,\n  reason STRING,\n  step INT32\n)";
 export declare const EMBEDDING_SCHEMA = "\nCREATE NODE TABLE CodeEmbedding (\n  nodeId STRING,\n  embedding FLOAT[384],\n  PRIMARY KEY (nodeId)\n)";
 /**
  * Create vector index for semantic search