scai 0.1.57 → 0.1.58

package/dist/db/functionExtractors/extractFromTs.js

@@ -0,0 +1,72 @@
+import { Project, SyntaxKind } from 'ts-morph';
+import path from 'path';
+import { generateEmbedding } from '../../lib/generateEmbedding.js';
+import { log } from '../../utils/log.js';
+import { getDbForRepo } from '../client.js';
+import { markFileAsSkippedTemplate, markFileAsExtractedTemplate, markFileAsFailedTemplate } from '../sqlTemplates.js';
+export async function extractFromTS(filePath, content, fileId) {
+    const db = getDbForRepo();
+    try {
+        const project = new Project({ useInMemoryFileSystem: true });
+        const sourceFile = project.createSourceFile(filePath, content);
+        const functions = [];
+        const allFuncs = [
+            ...sourceFile.getDescendantsOfKind(SyntaxKind.FunctionDeclaration),
+            ...sourceFile.getDescendantsOfKind(SyntaxKind.FunctionExpression),
+            ...sourceFile.getDescendantsOfKind(SyntaxKind.ArrowFunction),
+        ];
+        for (const fn of allFuncs) {
+            const name = fn.getSymbol()?.getName() ?? `${path.basename(filePath)}:<anon>`;
+            const start = fn.getStartLineNumber();
+            const end = fn.getEndLineNumber();
+            const code = fn.getText();
+            functions.push({ name, start_line: start, end_line: end, content: code });
+        }
+        if (functions.length === 0) {
+            log(`⚠️ No functions found in TS file: ${filePath}`);
+            db.prepare(markFileAsSkippedTemplate).run({ id: fileId });
+            return false;
+        }
+        log(`🔍 Found ${functions.length} TypeScript functions in ${filePath}`);
+        for (const fn of functions) {
+            const embedding = await generateEmbedding(fn.content);
+            const result = db.prepare(`
+        INSERT INTO functions (
+          file_id, name, start_line, end_line, content, embedding, lang
+        ) VALUES (
+          @file_id, @name, @start_line, @end_line, @content, @embedding, @lang
+        )
+      `).run({
+                file_id: fileId,
+                name: fn.name,
+                start_line: fn.start_line,
+                end_line: fn.end_line,
+                content: fn.content,
+                embedding: JSON.stringify(embedding),
+                lang: 'ts'
+            });
+            const callerId = result.lastInsertRowid;
+            // Simplified call detection (no walking for now)
+            const callMatches = fn.content.matchAll(/(\w+)\s*\(/g);
+            for (const match of callMatches) {
+                db.prepare(`
+          INSERT INTO function_calls (caller_id, callee_name)
+          VALUES (@caller_id, @callee_name)
+        `).run({
+                    caller_id: callerId,
+                    callee_name: match[1],
+                });
+            }
+            log(`📌 Indexed TS function: ${fn.name}`);
+        }
+        db.prepare(markFileAsExtractedTemplate).run({ id: fileId });
+        log(`✅ Marked TS functions as extracted for ${filePath}`);
+        return true;
+    }
+    catch (err) {
+        log(`❌ Failed to extract from TS file: ${filePath}`);
+        log(`   ↳ ${String(err.message)}`);
+        db.prepare(markFileAsFailedTemplate).run({ id: fileId });
+        return false;
+    }
+}

package/dist/db/functionExtractors/index.js

@@ -5,6 +5,7 @@ import { extractFromJS } from './extractFromJs.js';
 import { extractFromXML } from './extractFromXML.js';
 import { getDbForRepo } from '../client.js';
 import { markFileAsFailedTemplate, markFileAsSkippedByPath } from '../sqlTemplates.js';
+import { extractFromTS } from './extractFromTs.js';
 /**
  * Detects file type and delegates to the appropriate extractor.
  */
@@ -12,10 +13,14 @@ export async function extractFunctionsFromFile(filePath, content, fileId) {
     const type = detectFileType(filePath).trim().toLowerCase();
     const db = getDbForRepo();
     try {
-        if (type === 'js' || type === 'ts' || type === 'javascript' || type === 'typescript') {
+        if (type === 'js' || type === 'javascript') {
             log(`✅ Attempting to extract JS functions from ${filePath}`);
             return await extractFromJS(filePath, content, fileId);
         }
+        if (type === 'ts' || type === 'typescript') {
+            log(`📘 Extracting TS functions from ${filePath}`);
+            return await extractFromTS(filePath, content, fileId);
+        }
         if (type === 'java') {
             log(`❌ Nothing extracted for ${filePath} due to missing implementation`);
             await extractFromJava(filePath, content, fileId);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "scai",
-  "version": "0.1.57",
+  "version": "0.1.58",
   "type": "module",
   "bin": {
     "scai": "./dist/index.js"
@@ -44,7 +44,8 @@
     "commander": "^11.0.0",
     "fast-glob": "^3.3.3",
     "proper-lockfile": "^4.1.2",
-    "string-similarity-js": "^2.1.4"
+    "string-similarity-js": "^2.1.4",
+    "ts-morph": "^26.0.0"
   },
   "devDependencies": {
     "@types/better-sqlite3": "^7.6.13",