scai 0.1.115 → 0.1.117

package/dist/utils/buildContextualPrompt.js

@@ -6,75 +6,89 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
     const log = (...args) => console.log("[buildContextualPrompt]", ...args);
     const promptSections = [];
     const seenPaths = new Set();
-    // --- Utility: Summarize text to a few words ---
     function summarizeForPrompt(summary, maxWords = 30) {
         if (!summary)
             return undefined;
         const words = summary.split(/\s+/);
         return words.length <= maxWords ? summary.trim() : words.slice(0, maxWords).join(" ") + " …";
     }
-    // --- Step 1: Top file summary ---
+    // --- Top file summary ---
     if (topFile.summary) {
         promptSections.push(`**Top file:** ${topFile.path}\n${topFile.summary}`);
         seenPaths.add(topFile.path);
     }
-    // ===========================
-    // SECTION A: Database queries
-    // ===========================
-    // --- Step 2a: KG entities/tags ---
+    // --- Knowledge Graph tags ---
     const topEntitiesRows = db.prepare(`
-    SELECT et.entity_type, et.entity_id, tm.name AS tag
-    FROM entity_tags et
-    JOIN tags_master tm ON et.tag_id = tm.id
-    WHERE et.entity_id = ?
-  `).all(topFile.id);
+    SELECT et.entity_type, et.entity_unique_id, tm.name AS tag
+    FROM graph_entity_tags et
+    JOIN graph_tags_master tm ON et.tag_id = tm.id
+    WHERE et.entity_unique_id = ?
+  `).all(topFile.path);
     if (topEntitiesRows.length > 0) {
         const tags = topEntitiesRows.map(r => `- **${r.entity_type}**: ${r.tag}`);
         promptSections.push(`**Knowledge Graph context for ${topFile.path}:**\n${tags.join("\n")}`);
     }
-    // --- Step 2b: Functions in this file ---
+    // --- Imports / Exports ---
+    const imports = db.prepare(`
+    SELECT e.target_unique_id AS imported
+    FROM graph_edges e
+    WHERE e.source_type = 'file'
+      AND e.relation = 'imports'
+      AND e.source_unique_id = ?
+  `).all(topFile.path);
+    if (imports.length) {
+        promptSections.push(`**Imports from ${topFile.path}:**\n` + imports.map(i => `- ${i.imported}`).join("\n"));
+    }
+    const exports = db.prepare(`
+    SELECT e.target_unique_id AS exported
+    FROM graph_edges e
+    WHERE e.source_type = 'file'
+      AND e.relation = 'exports'
+      AND e.source_unique_id = ?
+  `).all(topFile.path);
+    if (exports.length) {
+        promptSections.push(`**Exports from ${topFile.path}:**\n` + exports.map(e => `- ${e.exported}`).join("\n"));
+    }
+    // --- Functions in file ---
     const functionRows = db
-        .prepare(`SELECT name, start_line, end_line, content FROM functions WHERE file_id = ? ORDER BY start_line`)
+        .prepare(`SELECT unique_id, name, start_line, end_line, content FROM functions WHERE file_id = ? ORDER BY start_line`)
         .all(topFile.id);
-    const FUNCTION_LIMIT = 15;
+    const FUNCTION_LIMIT = 5;
     const hasMoreFunctions = functionRows.length > FUNCTION_LIMIT;
     const functionsSummary = functionRows.slice(0, FUNCTION_LIMIT).map(f => {
         const lines = f.content?.split("\n").map(l => l.trim()).filter(Boolean) || ["[no content]"];
-        const preview = lines.slice(0, 3) // first 3 lines
-            .map(l => l.slice(0, 200) + (l.length > 200 ? "…" : ""))
-            .join(" | ");
+        const preview = lines.slice(0, 3).map(l => l.slice(0, 200) + (l.length > 200 ? "…" : "")).join(" | ");
         return `- ${f.name || "[anonymous]"} (lines ${f.start_line}-${f.end_line}) — ${preview}`;
     });
     if (functionsSummary.length) {
         promptSections.push(`**Functions in ${topFile.path} (showing ${functionsSummary.length}${hasMoreFunctions ? ` of ${functionRows.length}` : ""}):**\n${functionsSummary.join("\n")}`);
     }
     // ===============================
-    // SECTION B: Graph / KG traversal
+    // Graph / KG traversal
     // ===============================
-    const kgRelatedStmt = db.prepare(`
-    SELECT DISTINCT f.id, f.path, f.summary
-    FROM edges e
-    JOIN files f ON e.target_id = f.id
-    WHERE e.source_type = 'file'
-      AND e.target_type = 'file'
-      AND e.source_id = ?
-  `);
-    function getRelatedKGFiles(fileId, visited = new Set()) {
-        if (visited.has(fileId))
+    function getRelatedKGFiles(fileUniqueId, visited = new Set()) {
+        if (visited.has(fileUniqueId))
             return [];
-        visited.add(fileId);
-        const rows = kgRelatedStmt.all(fileId);
+        visited.add(fileUniqueId);
+        const rows = db.prepare(`
+      SELECT DISTINCT f.id, f.path, f.summary
+      FROM graph_edges e
+      JOIN files f ON e.target_unique_id = f.path
+      WHERE e.source_type = 'file'
+        AND e.target_type = 'file'
+        AND e.source_unique_id = ?
+    `).all(fileUniqueId);
         let results = [];
         for (const row of rows) {
             results.push(row);
-            results.push(...getRelatedKGFiles(row.id, visited));
+            results.push(...getRelatedKGFiles(row.path, visited));
         }
         return results;
     }
     function buildFileTree(file, depth, visited = new Set()) {
-        if (visited.has(file.id))
+        if (visited.has(file.path))
             return { id: file.id.toString(), path: file.path };
-        visited.add(file.id);
+        visited.add(file.path);
         const maxWordsByDepth = depth >= 3 ? 30 : depth === 2 ? 15 : 0;
         const node = {
             id: file.id.toString(),
@@ -82,9 +96,7 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
             summary: maxWordsByDepth > 0 ? summarizeForPrompt(file.summary, maxWordsByDepth) : undefined,
         };
         if (depth > 1) {
-            const relatedFiles = getRelatedKGFiles(file.id, visited)
-                .map(f => ({ id: f.id, path: f.path, summary: f.summary }))
-                .slice(0, 5); // cap children
+            const relatedFiles = getRelatedKGFiles(file.path).map(f => ({ id: f.id, path: f.path, summary: f.summary })).slice(0, 5);
             const relatedNodes = relatedFiles.map(f => buildFileTree(f, depth - 1, visited));
             if (relatedNodes.length)
                 node.related = relatedNodes;
@@ -93,7 +105,56 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
     }
     const kgTree = buildFileTree({ id: topFile.id, path: topFile.path, summary: topFile.summary }, kgDepth);
     promptSections.push(`**KG-Related Files (JSON tree, depth ${kgDepth}):**\n\`\`\`json\n${JSON.stringify(kgTree, null, 2)}\n\`\`\``);
-    // --- Step 3: File tree (shallow) ---
+    const functionCallsAll = db.prepare(`
+  SELECT source_unique_id, target_unique_id
+  FROM graph_edges
+  WHERE source_type = 'function' AND relation = 'calls'
+    AND source_unique_id IN (
+      SELECT unique_id FROM functions WHERE file_id = ?
+    )
+`).all(topFile.id);
+    const callsByFunction = {};
+    for (const fn of functionRows) {
+        const rows = functionCallsAll
+            .filter(r => r.source_unique_id === fn.unique_id)
+            .slice(0, FUNCTION_LIMIT);
+        // Truncate function content for preview
+        const lines = fn.content?.split("\n").map(l => l.trim()).filter(Boolean) || ["[no content]"];
+        const preview = lines.slice(0, 3).map(l => l.slice(0, 200) + (l.length > 200 ? "…" : "")).join(" | ");
+        callsByFunction[fn.name || fn.unique_id] = {
+            calls: rows.map(r => ({ unique_id: r.target_unique_id })),
+            preview,
+        };
+    }
+    if (Object.keys(callsByFunction).length > 0) {
+        promptSections.push(`**Function-level calls (limited, JSON):**\n\`\`\`json\n${JSON.stringify(callsByFunction, null, 2)}\n\`\`\``);
+    }
+    // --- Function-level "called by" overview (limited) ---
+    const calledByAll = db.prepare(`
+  SELECT source_unique_id, target_unique_id
+  FROM graph_edges
+  WHERE target_type = 'function' AND relation = 'calls'
+    AND target_unique_id IN (
+      SELECT unique_id FROM functions WHERE file_id = ?
+    )
+`).all(topFile.id);
+    const calledByByFunction = {};
+    for (const fn of functionRows) {
+        const rows = calledByAll
+            .filter(r => r.target_unique_id === fn.unique_id)
+            .slice(0, FUNCTION_LIMIT);
+        // Reuse truncated preview
+        const lines = fn.content?.split("\n").map(l => l.trim()).filter(Boolean) || ["[no content]"];
+        const preview = lines.slice(0, 3).map(l => l.slice(0, 200) + (l.length > 200 ? "…" : "")).join(" | ");
+        calledByByFunction[fn.name || fn.unique_id] = {
+            calledBy: rows.map(r => ({ unique_id: r.source_unique_id })),
+            preview,
+        };
+    }
+    if (Object.keys(calledByByFunction).length > 0) {
+        promptSections.push(`**Function-level called-by (limited, JSON):**\n\`\`\`json\n${JSON.stringify(calledByByFunction, null, 2)}\n\`\`\``);
+    }
+    // --- Focused file tree (shallow) ---
     try {
         const fileTree = generateFocusedFileTree(topFile.path, 2);
         if (fileTree) {
@@ -103,7 +164,7 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
     catch (e) {
         console.warn("⚠️ Could not generate file tree:", e);
     }
-    // --- Step 4: Optional code snippet ---
+    // --- Optional code snippet ---
     const MAX_LINES = 50;
     const queryNeedsCode = /\b(code|implementation|function|snippet)\b/i.test(query);
     if ((!topFile.summary || queryNeedsCode) && topFile.code) {
@@ -113,7 +174,7 @@ export async function buildContextualPrompt({ topFile, query, kgDepth = 3, }) {
             snippet += "\n... [truncated]";
         promptSections.push(`**Code Context (first ${MAX_LINES} lines):**\n\`\`\`\n${snippet}\n\`\`\``);
     }
-    // --- Step 5: User query ---
+    // --- User query ---
     promptSections.push(`**Query:** ${query}`);
     const promptText = promptSections.join("\n\n---\n\n");
     log("✅ Contextual prompt built for:", topFile.path);

package/dist/utils/sharedUtils.js

@@ -0,0 +1,8 @@
+import path from 'path';
+import crypto from 'crypto';
+// put this helper at top-level (or import from shared utils)
+export function getUniqueId(name, filePath, startLine, startColumn, content) {
+    const normalizedPath = path.normalize(filePath).replace(/\\/g, '/');
+    const hash = crypto.createHash('md5').update(content).digest('hex').slice(0, 6);
+    return `${name}@${normalizedPath}:${startLine}:${startColumn}:${hash}`;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "scai",
-  "version": "0.1.115",
+  "version": "0.1.117",
   "type": "module",
   "bin": {
     "scai": "./dist/index.js"

package/dist/commands/MigrateCmd.js

@@ -1,15 +0,0 @@
-// src/commands/MigrateCmd.ts
-import path from 'path';
-import { pathToFileURL } from 'url';
-import { fileURLToPath } from 'url';
-export async function runMigrateCommand() {
-    const scriptPath = path.resolve(path.dirname(fileURLToPath(import.meta.url)), '../../dist/scripts/migrateDb.js' // compiled output
-    );
-    try {
-        await import(pathToFileURL(scriptPath).href);
-    }
-    catch (err) {
-        console.error('❌ Failed to run migration script:', err);
-        process.exit(1);
-    }
-}