homegraph 1.1.2 → 1.2.0

package/dist/db/queries.js

@@ -4,12 +4,48 @@
  *
  * Prepared statements for CRUD operations on the knowledge graph.
  */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.QueryBuilder = void 0;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
 const utils_1 = require("../utils");
 const query_utils_1 = require("../search/query-utils");
 const query_parser_1 = require("../search/query-parser");
 const generated_detection_1 = require("../extraction/generated-detection");
+const arkts_1 = require("../extraction/languages/arkts");
 /**
  * Path-only heuristic for files that should not be candidates for
  * "dominant file" detection: test/spec files and tool-generated files.
@@ -103,11 +139,49 @@ class QueryBuilder {
     // Node cache for frequently accessed nodes (LRU-style, max 1000 entries)
     nodeCache = new Map();
     maxCacheSize = 1000;
+    /** Attached prebuilt OHOS API db (see arkts.ts). */
+    ohosApiDbPath = null;
     // Prepared statements (lazily initialized)
     stmts = {};
     constructor(db) {
         this.db = db;
     }
+    /** Attach a versioned OHOS API db for federated symbol lookup. */
+    attachOhosApiDb(dbPath) {
+        if (dbPath === this.ohosApiDbPath)
+            return;
+        if (this.ohosApiDbPath) {
+            try {
+                this.db.exec('DETACH DATABASE ohos_api');
+            }
+            catch {
+                /* already detached */
+            }
+            this.ohosApiDbPath = null;
+        }
+        if (!dbPath)
+            return;
+        const resolved = path.resolve(dbPath);
+        if (!fs.existsSync(resolved)) {
+            throw new Error(`OHOS API db not found: ${resolved}`);
+        }
+        const escaped = resolved.replace(/'/g, "''");
+        this.db.exec(`ATTACH DATABASE '${escaped}' AS ohos_api`);
+        this.ohosApiDbPath = resolved;
+        this.stmts = {};
+        this.nodeCache.clear();
+    }
+    getOhosApiDbPath() {
+        return this.ohosApiDbPath;
+    }
+    hasOhosApiAttached() {
+        return this.ohosApiDbPath != null;
+    }
+    mapOhosApiNodeRow(row) {
+        const node = rowToNode(row);
+        node.filePath = (0, arkts_1.markOhosApiFilePath)(node.filePath);
+        return node;
+    }
     /** Set the normalized project-name tokens used to down-weight non-discriminative
      * query words in path scoring (#720). Called once when the project opens. */
     setProjectNameTokens(tokens) {
@@ -125,20 +199,20 @@ class QueryBuilder {
      */
     insertNode(node) {
         if (!this.stmts.insertNode) {
-            this.stmts.insertNode = this.db.prepare(`
-        INSERT OR REPLACE INTO nodes (
-          id, kind, name, qualified_name, file_path, language,
-          start_line, end_line, start_column, end_column,
-          docstring, signature, visibility,
-          is_exported, is_async, is_static, is_abstract,
-          decorators, type_parameters, return_type, updated_at
-        ) VALUES (
-          @id, @kind, @name, @qualifiedName, @filePath, @language,
-          @startLine, @endLine, @startColumn, @endColumn,
-          @docstring, @signature, @visibility,
-          @isExported, @isAsync, @isStatic, @isAbstract,
-          @decorators, @typeParameters, @returnType, @updatedAt
-        )
+            this.stmts.insertNode = this.db.prepare(`
+        INSERT OR REPLACE INTO nodes (
+          id, kind, name, qualified_name, file_path, language,
+          start_line, end_line, start_column, end_column,
+          docstring, signature, visibility,
+          is_exported, is_async, is_static, is_abstract,
+          decorators, type_parameters, return_type, updated_at
+        ) VALUES (
+          @id, @kind, @name, @qualifiedName, @filePath, @language,
+          @startLine, @endLine, @startColumn, @endColumn,
+          @docstring, @signature, @visibility,
+          @isExported, @isAsync, @isStatic, @isAbstract,
+          @decorators, @typeParameters, @returnType, @updatedAt
+        )
       `);
         }
         // Validate required fields to prevent SQLite bind errors
@@ -196,29 +270,29 @@ class QueryBuilder {
      */
     updateNode(node) {
         if (!this.stmts.updateNode) {
-            this.stmts.updateNode = this.db.prepare(`
-        UPDATE nodes SET
-          kind = @kind,
-          name = @name,
-          qualified_name = @qualifiedName,
-          file_path = @filePath,
-          language = @language,
-          start_line = @startLine,
-          end_line = @endLine,
-          start_column = @startColumn,
-          end_column = @endColumn,
-          docstring = @docstring,
-          signature = @signature,
-          visibility = @visibility,
-          is_exported = @isExported,
-          is_async = @isAsync,
-          is_static = @isStatic,
-          is_abstract = @isAbstract,
-          decorators = @decorators,
-          type_parameters = @typeParameters,
-          return_type = @returnType,
-          updated_at = @updatedAt
-        WHERE id = @id
+            this.stmts.updateNode = this.db.prepare(`
+        UPDATE nodes SET
+          kind = @kind,
+          name = @name,
+          qualified_name = @qualifiedName,
+          file_path = @filePath,
+          language = @language,
+          start_line = @startLine,
+          end_line = @endLine,
+          start_column = @startColumn,
+          end_column = @endColumn,
+          docstring = @docstring,
+          signature = @signature,
+          visibility = @visibility,
+          is_exported = @isExported,
+          is_async = @isAsync,
+          is_static = @isStatic,
+          is_abstract = @isAbstract,
+          decorators = @decorators,
+          type_parameters = @typeParameters,
+          return_type = @returnType,
+          updated_at = @updatedAt
+        WHERE id = @id
       `);
         }
         // Invalidate cache before update
@@ -426,15 +500,15 @@ class QueryBuilder {
             // protobuf stub) outranks the real `server/etcdserver/server.go`
             // (470 edges) by 4×, and the boost would push the agent toward
             // generated code.
-            this.stmts.getDominantFile = this.db.prepare(`
-        SELECT n.file_path AS file_path, COUNT(*) AS edge_count
-        FROM edges e
-        JOIN nodes n ON e.source = n.id
-        JOIN nodes m ON e.target = m.id
-        WHERE n.file_path = m.file_path
-        GROUP BY n.file_path
-        ORDER BY edge_count DESC
-        LIMIT 20
+            this.stmts.getDominantFile = this.db.prepare(`
+        SELECT n.file_path AS file_path, COUNT(*) AS edge_count
+        FROM edges e
+        JOIN nodes n ON e.source = n.id
+        JOIN nodes m ON e.target = m.id
+        WHERE n.file_path = m.file_path
+        GROUP BY n.file_path
+        ORDER BY edge_count DESC
+        LIMIT 20
       `);
         }
         const rows = this.stmts.getDominantFile.all();
@@ -461,13 +535,13 @@ class QueryBuilder {
      */
     getTopRouteFile() {
         if (!this.stmts.getTopRouteFile) {
-            this.stmts.getTopRouteFile = this.db.prepare(`
-        SELECT file_path, COUNT(*) AS cnt
-        FROM nodes
-        WHERE kind = 'route'
-        GROUP BY file_path
-        ORDER BY cnt DESC
-        LIMIT 20
+            this.stmts.getTopRouteFile = this.db.prepare(`
+        SELECT file_path, COUNT(*) AS cnt
+        FROM nodes
+        WHERE kind = 'route'
+        GROUP BY file_path
+        ORDER BY cnt DESC
+        LIMIT 20
       `);
         }
         const rows = this.stmts.getTopRouteFile.all();
@@ -498,21 +572,21 @@ class QueryBuilder {
             // Edge kind varies across framework resolvers: Spring/Rails/
             // Laravel/Drupal emit `references`, Express emits `calls`. Accept
             // both — the semantic is the same (route → its handler).
-            this.stmts.getRoutingManifest = this.db.prepare(`
-        SELECT
-          r.name AS url,
-          h.name AS handler,
-          h.file_path AS handler_file,
-          h.start_line AS handler_line,
-          h.kind AS handler_kind
-        FROM nodes r
-        JOIN edges e ON e.source = r.id
-        JOIN nodes h ON e.target = h.id
-        WHERE r.kind = 'route'
-          AND e.kind IN ('references', 'calls')
-          AND h.kind IN ('function', 'method', 'class')
-        ORDER BY r.file_path, r.start_line
-        LIMIT ?
+            this.stmts.getRoutingManifest = this.db.prepare(`
+        SELECT
+          r.name AS url,
+          h.name AS handler,
+          h.file_path AS handler_file,
+          h.start_line AS handler_line,
+          h.kind AS handler_kind
+        FROM nodes r
+        JOIN edges e ON e.source = r.id
+        JOIN nodes h ON e.target = h.id
+        WHERE r.kind = 'route'
+          AND e.kind IN ('references', 'calls')
+          AND h.kind IN ('function', 'method', 'class')
+        ORDER BY r.file_path, r.start_line
+        LIMIT ?
       `);
         }
         const rows = this.stmts.getRoutingManifest.all(limit);
@@ -586,7 +660,15 @@ class QueryBuilder {
             this.stmts.getNodesByName = this.db.prepare('SELECT * FROM nodes WHERE name = ?');
         }
         const rows = this.stmts.getNodesByName.all(name);
-        return rows.map(rowToNode);
+        const main = rows.map(rowToNode);
+        if (!this.hasOhosApiAttached())
+            return main;
+        const apiRows = this.db
+            .prepare('SELECT * FROM ohos_api.nodes WHERE name = ?')
+            .all(name);
+        const seen = new Set(main.map((n) => n.id));
+        const api = apiRows.map((r) => this.mapOhosApiNodeRow(r)).filter((n) => !seen.has(n.id));
+        return [...main, ...api];
     }
     /**
      * Get nodes by exact qualified name match (uses idx_nodes_qualified_name index)
@@ -722,8 +804,44 @@ class QueryBuilder {
                 return lowered.some((n) => nm.includes(n));
             });
         }
+        if (this.hasOhosApiAttached() && text.length >= 2) {
+            results = this.mergeOhosApiTextSearch(text, { kinds, languages, limit }, results);
+        }
         return results;
     }
+    mergeOhosApiTextSearch(text, options, existing) {
+        const { kinds, languages, limit = 100 } = options;
+        const seen = new Set(existing.map((r) => r.node.id));
+        const merged = [...existing];
+        const apiLimit = Math.min(20, Math.max(5, Math.floor(limit / 4)));
+        let sql = `
+      SELECT nodes.*, 0.9 as score
+      FROM ohos_api.nodes
+      WHERE (name LIKE ? OR qualified_name LIKE ?)
+    `;
+        const pattern = `%${text}%`;
+        const params = [pattern, pattern];
+        if (kinds && kinds.length > 0) {
+            sql += ` AND kind IN (${kinds.map(() => '?').join(',')})`;
+            params.push(...kinds);
+        }
+        if (languages && languages.length > 0) {
+            sql += ` AND language IN (${languages.map(() => '?').join(',')})`;
+            params.push(...languages);
+        }
+        sql += ' LIMIT ?';
+        params.push(apiLimit);
+        const rows = this.db.prepare(sql).all(...params);
+        for (const row of rows) {
+            const node = this.mapOhosApiNodeRow(row);
+            if (seen.has(node.id))
+                continue;
+            seen.add(node.id);
+            merged.push({ node, score: row.score });
+        }
+        merged.sort((a, b) => b.score - a.score);
+        return merged.slice(0, limit);
+    }
     /**
      * Match-everything path used when the user supplied only field
      * filters (`kind:function lang:typescript`) with no text. Returns
@@ -838,11 +956,11 @@ class QueryBuilder {
         // Fetch 5x requested limit so post-hoc rescoring (kindBonus, pathRelevance,
         // nameMatchBonus) can promote results that BM25 alone undervalues.
         const ftsLimit = Math.max(limit * 5, 100);
-        let sql = `
-      SELECT nodes.*, bm25(nodes_fts, 0, 20, 5, 1, 2) as score
-      FROM nodes_fts
-      JOIN nodes ON nodes_fts.id = nodes.id
-      WHERE nodes_fts MATCH ?
+        let sql = `
+      SELECT nodes.*, bm25(nodes_fts, 0, 20, 5, 1, 2) as score
+      FROM nodes_fts
+      JOIN nodes ON nodes_fts.id = nodes.id
+      WHERE nodes_fts MATCH ?
     `;
         const params = [ftsQuery];
         if (kinds && kinds.length > 0) {
@@ -873,21 +991,21 @@ class QueryBuilder {
      */
     searchNodesLike(query, options) {
         const { kinds, languages, limit = 100, offset = 0 } = options;
-        let sql = `
-      SELECT nodes.*,
-        CASE
-          WHEN name = ? THEN 1.0
-          WHEN name LIKE ? THEN 0.9
-          WHEN name LIKE ? THEN 0.8
-          WHEN qualified_name LIKE ? THEN 0.7
-          ELSE 0.5
-        END as score
-      FROM nodes
-      WHERE (
-        name LIKE ? OR
-        qualified_name LIKE ? OR
-        name LIKE ?
-      )
+        let sql = `
+      SELECT nodes.*,
+        CASE
+          WHEN name = ? THEN 1.0
+          WHEN name LIKE ? THEN 0.9
+          WHEN name LIKE ? THEN 0.8
+          WHEN qualified_name LIKE ? THEN 0.7
+          ELSE 0.5
+        END as score
+      FROM nodes
+      WHERE (
+        name LIKE ? OR
+        qualified_name LIKE ? OR
+        name LIKE ?
+      )
     `;
         // Pattern variants for better matching
         const exactMatch = query;
@@ -961,10 +1079,10 @@ class QueryBuilder {
         const allResults = [];
         const seenIds = new Set();
         for (const name of names) {
-            let sql = `
-        SELECT nodes.*, 1.0 as score
-        FROM nodes
-        WHERE name COLLATE NOCASE = ?
+            let sql = `
+        SELECT nodes.*, 1.0 as score
+        FROM nodes
+        WHERE name COLLATE NOCASE = ?
       `;
             const params = [name];
             if (kinds && kinds.length > 0) {
@@ -994,6 +1112,32 @@ class QueryBuilder {
                 seenIds.add(r.node.id);
                 allResults.push(r);
             }
+            if (this.hasOhosApiAttached()) {
+                let apiSql = `
+          SELECT nodes.*, 0.95 as score
+          FROM ohos_api.nodes
+          WHERE name COLLATE NOCASE = ?
+        `;
+                const apiParams = [name];
+                if (kinds && kinds.length > 0) {
+                    apiSql += ` AND kind IN (${kinds.map(() => '?').join(',')})`;
+                    apiParams.push(...kinds);
+                }
+                if (languages && languages.length > 0) {
+                    apiSql += ` AND language IN (${languages.map(() => '?').join(',')})`;
+                    apiParams.push(...languages);
+                }
+                apiSql += ' LIMIT ?';
+                apiParams.push(perNameLimit);
+                const apiRows = this.db.prepare(apiSql).all(...apiParams);
+                for (const row of apiRows) {
+                    const node = this.mapOhosApiNodeRow(row);
+                    if (seenIds.has(node.id))
+                        continue;
+                    seenIds.add(node.id);
+                    allResults.push({ node, score: row.score });
+                }
+            }
         }
         // Sort all results by score so co-located results bubble up
         allResults.sort((a, b) => b.score - a.score);
@@ -1008,10 +1152,10 @@ class QueryBuilder {
      */
     findNodesByNameSubstring(substring, options = {}) {
         const { kinds, languages, limit = 30, excludePrefix } = options;
-        let sql = `
-      SELECT nodes.*, 1.0 as score
-      FROM nodes
-      WHERE name LIKE ?
+        let sql = `
+      SELECT nodes.*, 1.0 as score
+      FROM nodes
+      WHERE name LIKE ?
     `;
         const params = [`%${substring}%`];
         // Exclude prefix matches (handled by FTS-based prefix search in Step 2b)
@@ -1043,9 +1187,9 @@ class QueryBuilder {
      */
     insertEdge(edge) {
         if (!this.stmts.insertEdge) {
-            this.stmts.insertEdge = this.db.prepare(`
-        INSERT OR IGNORE INTO edges (source, target, kind, metadata, line, col, provenance)
-        VALUES (@source, @target, @kind, @metadata, @line, @col, @provenance)
+            this.stmts.insertEdge = this.db.prepare(`
+        INSERT OR IGNORE INTO edges (source, target, kind, metadata, line, col, provenance)
+        VALUES (@source, @target, @kind, @metadata, @line, @col, @provenance)
       `);
         }
         this.stmts.insertEdge.run({
@@ -1161,12 +1305,12 @@ class QueryBuilder {
      * it. One indexed query (idx_nodes_file_path + idx_edges_target_kind).
      */
     getDependentFilePaths(filePath) {
-        const sql = `SELECT DISTINCT src.file_path AS fp
-      FROM edges e
-      JOIN nodes tgt ON tgt.id = e.target
-      JOIN nodes src ON src.id = e.source
-      WHERE tgt.file_path = ?
-        AND e.kind != 'contains'
+        const sql = `SELECT DISTINCT src.file_path AS fp
+      FROM edges e
+      JOIN nodes tgt ON tgt.id = e.target
+      JOIN nodes src ON src.id = e.source
+      WHERE tgt.file_path = ?
+        AND e.kind != 'contains'
         AND src.file_path != ?`;
         const rows = this.db.prepare(sql).all(filePath, filePath);
         return rows.map((r) => r.fp);
@@ -1178,12 +1322,12 @@ class QueryBuilder {
      * (all kinds except `contains`); same reason imports-only is insufficient.
      */
     getDependencyFilePaths(filePath) {
-        const sql = `SELECT DISTINCT tgt.file_path AS fp
-      FROM edges e
-      JOIN nodes src ON src.id = e.source
-      JOIN nodes tgt ON tgt.id = e.target
-      WHERE src.file_path = ?
-        AND e.kind != 'contains'
+        const sql = `SELECT DISTINCT tgt.file_path AS fp
+      FROM edges e
+      JOIN nodes src ON src.id = e.source
+      JOIN nodes tgt ON tgt.id = e.target
+      WHERE src.file_path = ?
+        AND e.kind != 'contains'
         AND tgt.file_path != ?`;
         const rows = this.db.prepare(sql).all(filePath, filePath);
         return rows.map((r) => r.fp);
@@ -1199,12 +1343,12 @@ class QueryBuilder {
      * {@link getDependentFilePaths}: all kinds except `contains`.
      */
     getCrossFileIncomingEdgesWithTarget(filePath) {
-        const sql = `SELECT e.*, tgt.name AS target_name, tgt.kind AS target_kind
-      FROM edges e
-      JOIN nodes tgt ON tgt.id = e.target
-      JOIN nodes src ON src.id = e.source
-      WHERE tgt.file_path = ?
-        AND e.kind != 'contains'
+        const sql = `SELECT e.*, tgt.name AS target_name, tgt.kind AS target_kind
+      FROM edges e
+      JOIN nodes tgt ON tgt.id = e.target
+      JOIN nodes src ON src.id = e.source
+      WHERE tgt.file_path = ?
+        AND e.kind != 'contains'
         AND src.file_path != ?`;
         const rows = this.db.prepare(sql).all(filePath, filePath);
         return rows.map(row => ({
@@ -1219,10 +1363,10 @@ class QueryBuilder {
      */
     deleteArkTSCrossFileCallEdges() {
         const result = this.db
-            .prepare(`DELETE FROM edges
-         WHERE kind = 'calls'
-           AND provenance = 'heuristic'
-           AND metadata LIKE '%"synthesizedBy":"arkanalyzer"%'
+            .prepare(`DELETE FROM edges
+         WHERE kind = 'calls'
+           AND provenance = 'heuristic'
+           AND metadata LIKE '%"synthesizedBy":"arkanalyzer"%'
            AND metadata LIKE '%"sourceFile"%'`)
             .run();
         return result.changes;
@@ -1235,17 +1379,17 @@ class QueryBuilder {
      */
     upsertFile(file) {
         if (!this.stmts.upsertFile) {
-            this.stmts.upsertFile = this.db.prepare(`
-        INSERT INTO files (path, content_hash, language, size, modified_at, indexed_at, node_count, errors)
-        VALUES (@path, @contentHash, @language, @size, @modifiedAt, @indexedAt, @nodeCount, @errors)
-        ON CONFLICT(path) DO UPDATE SET
-          content_hash = @contentHash,
-          language = @language,
-          size = @size,
-          modified_at = @modifiedAt,
-          indexed_at = @indexedAt,
-          node_count = @nodeCount,
-          errors = @errors
+            this.stmts.upsertFile = this.db.prepare(`
+        INSERT INTO files (path, content_hash, language, size, modified_at, indexed_at, node_count, errors)
+        VALUES (@path, @contentHash, @language, @size, @modifiedAt, @indexedAt, @nodeCount, @errors)
+        ON CONFLICT(path) DO UPDATE SET
+          content_hash = @contentHash,
+          language = @language,
+          size = @size,
+          modified_at = @modifiedAt,
+          indexed_at = @indexedAt,
+          node_count = @nodeCount,
+          errors = @errors
       `);
         }
         this.stmts.upsertFile.run({
@@ -1319,9 +1463,9 @@ class QueryBuilder {
      */
     insertUnresolvedRef(ref) {
         if (!this.stmts.insertUnresolved) {
-            this.stmts.insertUnresolved = this.db.prepare(`
-        INSERT INTO unresolved_refs (from_node_id, reference_name, reference_kind, line, col, candidates, file_path, language)
-        VALUES (@fromNodeId, @referenceName, @referenceKind, @line, @col, @candidates, @filePath, @language)
+            this.stmts.insertUnresolved = this.db.prepare(`
+        INSERT INTO unresolved_refs (from_node_id, reference_name, reference_kind, line, col, candidates, file_path, language)
+        VALUES (@fromNodeId, @referenceName, @referenceKind, @line, @col, @candidates, @filePath, @language)
       `);
         }
         this.stmts.insertUnresolved.run({
@@ -1485,8 +1629,16 @@ class QueryBuilder {
     deleteResolvedReferences(fromNodeIds) {
         if (fromNodeIds.length === 0)
             return;
-        const placeholders = fromNodeIds.map(() => '?').join(',');
-        this.db.prepare(`DELETE FROM unresolved_refs WHERE from_node_id IN (${placeholders})`).run(...fromNodeIds);
+        // Chunk under SQLite's parameter limit, matching every other IN-list in
+        // this file. The internal resolution path uses deleteSpecificResolvedReferences
+        // instead, but QueryBuilder is part of the public API, so a library consumer
+        // passing more ids than SQLITE_MAX_VARIABLE_NUMBER (32766 on the bundled
+        // node:sqlite) would otherwise hit "too many SQL variables". (#540, #1001)
+        for (let i = 0; i < fromNodeIds.length; i += SQLITE_PARAM_CHUNK_SIZE) {
+            const chunk = fromNodeIds.slice(i, i + SQLITE_PARAM_CHUNK_SIZE);
+            const placeholders = chunk.map(() => '?').join(',');
+            this.db.prepare(`DELETE FROM unresolved_refs WHERE from_node_id IN (${placeholders})`).run(...chunk);
+        }
     }
     /**
      * Delete specific resolved references by (fromNodeId, referenceName, referenceKind) tuples.
@@ -1523,11 +1675,11 @@ class QueryBuilder {
      */
     getStats() {
         // Single query for all three aggregate counts
-        const counts = this.db.prepare(`
-      SELECT
-        (SELECT COUNT(*) FROM nodes) AS node_count,
-        (SELECT COUNT(*) FROM edges) AS edge_count,
-        (SELECT COUNT(*) FROM files) AS file_count
+        const counts = this.db.prepare(`
+      SELECT
+        (SELECT COUNT(*) FROM nodes) AS node_count,
+        (SELECT COUNT(*) FROM edges) AS edge_count,
+        (SELECT COUNT(*) FROM files) AS file_count
     `).get();
         const nodesByKind = {};
         const nodeKindRows = this.db
@@ -1588,12 +1740,31 @@ class QueryBuilder {
         }
         return result;
     }
+    // ===========================================================================
+    // MCP Query Cache
+    // ===========================================================================
+    clearMcpQueryCache() {
+        this.db.exec('DELETE FROM mcp_query_cache');
+    }
+    getMcpQueryCache(cacheKey) {
+        const row = this.db
+            .prepare('SELECT tool, response FROM mcp_query_cache WHERE cache_key = ?')
+            .get(cacheKey);
+        return row ?? null;
+    }
+    setMcpQueryCache(cacheKey, tool, response, createdAt) {
+        this.db
+            .prepare('INSERT INTO mcp_query_cache (cache_key, tool, response, created_at) VALUES (?, ?, ?, ?) ' +
+            'ON CONFLICT(cache_key) DO UPDATE SET tool = excluded.tool, response = excluded.response, created_at = excluded.created_at')
+            .run(cacheKey, tool, response, createdAt);
+    }
     /**
      * Clear all data from the database
      */
     clear() {
         this.nodeCache.clear();
         this.db.transaction(() => {
+            this.db.exec('DELETE FROM mcp_query_cache');
             this.db.exec('DELETE FROM unresolved_refs');
             this.db.exec('DELETE FROM edges');
             this.db.exec('DELETE FROM nodes');