codebuff 1.0.273 → 1.0.275

package/dist/code-map/parse.js

@@ -28,27 +28,36 @@ exports.getFileTokenScores = getFileTokenScores;
 exports.parseTokens = parseTokens;
 const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
+const lodash_1 = require("lodash");
 const languages_1 = require("./languages");
 exports.DEBUG_PARSING = false;
 const IGNORE_TOKENS = ['__init__', '__post_init__', '__call__', 'constructor'];
+const MAX_CALLERS = 25;
 async function getFileTokenScores(projectRoot, filePaths) {
     const startTime = Date.now();
     const tokenScores = {};
     const externalCalls = {};
+    const fileCallsMap = new Map();
+    // First pass: collect all identifiers and calls
     for (const filePath of filePaths) {
         const fullPath = path.join(projectRoot, filePath);
-        if (!!(0, languages_1.getLanguageConfig)(fullPath)) {
-            const { identifiers, calls, numLines } = await parseTokens(fullPath);
+        const languageConfig = await (0, languages_1.getLanguageConfig)(fullPath);
+        if (languageConfig) {
+            const { identifiers, calls, numLines } = await parseTokens(fullPath, languageConfig);
             const tokenScoresForFile = {};
             tokenScores[filePath] = tokenScoresForFile;
             const dirs = path.dirname(fullPath).split(path.sep);
             const depth = dirs.length;
             const tokenBaseScore = 0.8 ** depth * Math.sqrt(numLines / (identifiers.length + 1));
+            // Store defined tokens
             for (const identifier of identifiers) {
                 if (!IGNORE_TOKENS.includes(identifier)) {
                     tokenScoresForFile[identifier] = tokenBaseScore;
                 }
             }
+            // Store calls for this file
+            fileCallsMap.set(filePath, calls);
+            // Track external calls
             for (const call of calls) {
                 if (!tokenScoresForFile[call]) {
                     externalCalls[call] = (externalCalls[call] ?? 0) + 1;
@@ -56,58 +65,99 @@ async function getFileTokenScores(projectRoot, filePaths) {
             }
         }
     }
+    // Build a map of tokens to their defining files for O(1) lookup
+    const tokenDefinitionMap = new Map();
+    const highestScores = new Map();
+    for (const [filePath, scores] of Object.entries(tokenScores)) {
+        for (const [token, score] of Object.entries(scores)) {
+            const currentHighestScore = highestScores.get(token) ?? -Infinity;
+            // Keep the file with the higher score for this token
+            if (score > currentHighestScore) {
+                highestScores.set(token, score);
+                tokenDefinitionMap.set(token, filePath);
+            }
+        }
+    }
+    const tokenCallers = {};
+    // For each file's calls, add it as a caller to the defining file's tokens
+    for (const [callingFile, calls] of fileCallsMap.entries()) {
+        for (const call of calls) {
+            const definingFile = tokenDefinitionMap.get(call);
+            if (!definingFile || callingFile === definingFile) {
+                continue;
+            }
+            // Skip token names in default objects, e.g. toString, hasOwnProperty
+            if (call in {}) {
+                continue;
+            }
+            if (!tokenCallers[definingFile]) {
+                tokenCallers[definingFile] = {};
+            }
+            if (!tokenCallers[definingFile][call]) {
+                tokenCallers[definingFile][call] = [];
+            }
+            const callerFiles = tokenCallers[definingFile][call];
+            if (callerFiles.length < MAX_CALLERS &&
+                !callerFiles.includes(callingFile)) {
+                callerFiles.push(callingFile);
+            }
+        }
+    }
+    // Apply call frequency boost to token scores
     for (const scores of Object.values(tokenScores)) {
         for (const token of Object.keys(scores)) {
             const numCalls = externalCalls[token] ?? 0;
             if (typeof numCalls !== 'number')
                 continue;
             scores[token] *= 1 + Math.log(1 + numCalls);
+            // Round to 3 decimal places
+            scores[token] = Math.round(scores[token] * 1000) / 1000;
         }
     }
     if (exports.DEBUG_PARSING) {
         const endTime = Date.now();
         console.log(`Parsed ${filePaths.length} files in ${endTime - startTime}ms`);
-        console.log('externalCalls', externalCalls);
-        // Save exportedTokens to a file
-        const exportedTokensFilePath = path.join(projectRoot, 'exported-tokens.json');
-        try {
-            fs.writeFileSync(exportedTokensFilePath, JSON.stringify(tokenScores, null, 2));
-            console.log(`Exported tokens saved to ${exportedTokensFilePath}`);
-        }
-        catch (error) {
-            console.error(`Failed to save exported tokens to file: ${error}`);
-        }
+        fs.writeFileSync('../debug/debug-parse.json', JSON.stringify({
+            tokenCallers,
+            tokenScores,
+            fileCallsMap,
+            externalCalls,
+        }));
     }
-    return tokenScores;
+    return { tokenScores, tokenCallers };
 }
-async function parseTokens(filePath) {
-    const languageConfig = await (0, languages_1.getLanguageConfig)(filePath);
-    if (languageConfig) {
-        const { parser, query } = languageConfig;
-        try {
-            const sourceCode = fs.readFileSync(filePath, 'utf8');
-            const numLines = sourceCode.match(/\n/g)?.length ?? 0 + 1;
-            const parseResults = parseFile(parser, query, sourceCode);
-            const identifiers = parseResults.identifier;
-            const calls = parseResults['call.identifier'];
-            return {
-                numLines,
-                identifiers: identifiers ?? [],
-                calls: calls ?? [],
-            };
+async function parseTokens(filePath, languageConfig) {
+    const { parser, query } = languageConfig;
+    try {
+        const sourceCode = fs.readFileSync(filePath, 'utf8');
+        const numLines = sourceCode.match(/\n/g)?.length ?? 0 + 1;
+        const parseResults = parseFile(parser, query, sourceCode);
+        const identifiers = (0, lodash_1.uniq)(parseResults.identifier);
+        const calls = (0, lodash_1.uniq)(parseResults['call.identifier']);
+        if (exports.DEBUG_PARSING) {
+            console.log(`\nParsing ${filePath}:`);
+            console.log('Source:', sourceCode);
+            console.log('Parse results:', parseResults);
+            console.log('Identifiers:', identifiers);
+            console.log('Calls:', calls);
         }
-        catch (e) {
-            if (exports.DEBUG_PARSING) {
-                console.error(`Error parsing query: ${e}`);
-                console.log(filePath);
-            }
+        return {
+            numLines,
+            identifiers: identifiers ?? [],
+            calls: calls ?? [],
+        };
+    }
+    catch (e) {
+        if (exports.DEBUG_PARSING) {
+            console.error(`Error parsing query: ${e}`);
+            console.log(filePath);
         }
+        return {
+            numLines: 0,
+            identifiers: [],
+            calls: [],
+        };
     }
-    return {
-        numLines: 0,
-        identifiers: [],
-        calls: [],
-    };
 }
 function parseFile(parser, query, sourceCode) {
     const tree = parser.parse(sourceCode, undefined, {
@@ -115,6 +165,9 @@ function parseFile(parser, query, sourceCode) {
     });
     const captures = query.captures(tree.rootNode);
     const result = {};
+    if (exports.DEBUG_PARSING) {
+        console.log('\nCaptures:', captures);
+    }
     for (const capture of captures) {
         const { name, node } = capture;
         if (!result[name]) {

package/dist/code-map/parse.js.map

@@ -1 +1 @@
-{"version":3,"file":"parse.js","sourceRoot":"","sources":["../parse.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AASA,gDA8DC;AAED,kCA4BC;AArGD,uCAAwB;AACxB,2CAA4B;AAG5B,2CAA+C;AAElC,QAAA,aAAa,GAAG,KAAK,CAAA;AAClC,MAAM,aAAa,GAAG,CAAC,UAAU,EAAE,eAAe,EAAE,UAAU,EAAE,aAAa,CAAC,CAAA;AAEvE,KAAK,UAAU,kBAAkB,CAAC,WAAmB,EAAE,SAAmB;IAC/E,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;IAC5B,MAAM,WAAW,GAAwD,EAAE,CAAA;IAC3E,MAAM,aAAa,GAAgC,EAAE,CAAA;IAErD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAA;QACjD,IAAI,CAAC,CAAC,IAAA,6BAAiB,EAAC,QAAQ,CAAC,EAAE,CAAC;YAClC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,CAAA;YAEpE,MAAM,kBAAkB,GAAgC,EAAE,CAAA;YAC1D,WAAW,CAAC,QAAQ,CAAC,GAAG,kBAAkB,CAAA;YAE1C,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;YACnD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAA;YACzB,MAAM,cAAc,GAClB,GAAG,IAAI,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAA;YAE/D,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;gBACrC,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;oBACxC,kBAAkB,CAAC,UAAU,CAAC,GAAG,cAAc,CAAA;gBACjD,CAAC;YACH,CAAC;YACD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC9B,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAA;gBACtD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;QAChD,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACxC,MAAM,QAAQ,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YAC1C,IAAI,OAAO,QAAQ,KAAK,QAAQ;gBAAE,SAAQ;YAC1C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAA;QAC7C,CAAC;IACH,CAAC;IAED,IAAI,qBAAa,EAAE,CAAC;QAClB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAC1B,OAAO,CAAC,GAAG,CAAC,UAAU,SAAS,CAAC,MAAM,aAAa,OAAO,GAAG,SAAS,IAAI,CAAC,CAAA;QAE3E,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,aAAa,CAAC,CAAA;QAE3C,gCAAgC;QAChC,MAAM,sBAAsB,GAAG,IAAI,CAAC,IAAI,CACtC,WAAW,EACX,sBAAsB,CACvB,CAAA;QACD,IAAI,CAAC;YACH,EAAE,CAAC,aAAa,CACd,sBAAsB,EACtB,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,CACrC,CAAA;YACD,OAAO,CAAC,GAAG,CAAC,4BAA4B,sBAAsB,EAAE,CAAC,CAAA;QACnE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2CAA2C,KAAK,EAAE,CAAC,CAAA;QACnE,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAA;AACpB,CAAC;AAEM,KAAK,UAAU,WAAW,CAAC,QAAgB;IAChD,MAAM,cAAc,GAAG,MAAM,IAAA,6BAAiB,EAAC,QAAQ,CAAC,CAAA;IACxD,IAAI,cAAc,EAAE,CAAC;QACnB,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,cAAc,CAAA;QAExC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;YACpD,MAAM,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,MAAM,IAAI,CAAC,GAAG,CAAC,CAAA;YACzD,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,EAAE,KAAK,EAAE,UAAU,CAAC,CAAA;YACzD,MAAM,WAAW,GAAG,YAAY,CAAC,UAAU,CAAA;YAC3C,MAAM,KAAK,GAAG,YAAY,CAAC,iBAAiB,CAAC,CAAA;YAC7C,OAAO;gBACL,QAAQ;gBACR,WAAW,EAAE,WAAW,IAAI,EAAE;gBAC9B,KAAK,EAAE,KAAK,IAAI,EAAE;aACnB,CAAA;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,qBAAa,EAAE,CAAC;gBAClB,OAAO,CAAC,KAAK,CAAC,wBAAwB,CAAC,EAAE,CAAC,CAAA;gBAC1C,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;YACvB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO;QACL,QAAQ,EAAE,CAAC;QACX,WAAW,EAAE,EAAc;QAC3B,KAAK,EAAE,EAAc;KACtB,CAAA;AACH,CAAC;AAED,SAAS,SAAS,CAChB,MAAc,EACd,KAAmB,EACnB,UAAkB;IAElB,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE;QAC/C,UAAU,EAAE,IAAI,GAAG,IAAI;KACxB,CAAC,CAAA;IACF,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IAC9C,MAAM,MAAM,GAAgC,EAAE,CAAA;IAE9C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,OAAO,CAAA;QAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAClB,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAA;QACnB,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC9B,CAAC;IAED,OAAO,MAAM,CAAA;AACf,CAAC"}
+{"version":3,"file":"parse.js","sourceRoot":"","sources":["../parse.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAqBA,gDAuHC;AAED,kCAqCC;AAnLD,uCAAwB;AACxB,2CAA4B;AAC5B,mCAA6B;AAE7B,2CAA+D;AAElD,QAAA,aAAa,GAAG,KAAK,CAAA;AAClC,MAAM,aAAa,GAAG,CAAC,UAAU,EAAE,eAAe,EAAE,UAAU,EAAE,aAAa,CAAC,CAAA;AAC9E,MAAM,WAAW,GAAG,EAAE,CAAA;AAaf,KAAK,UAAU,kBAAkB,CACtC,WAAmB,EACnB,SAAmB;IAEnB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;IAC5B,MAAM,WAAW,GAAwD,EAAE,CAAA;IAC3E,MAAM,aAAa,GAAgC,EAAE,CAAA;IACrD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAoB,CAAA;IAEhD,gDAAgD;IAChD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAA;QACjD,MAAM,cAAc,GAAG,MAAM,IAAA,6BAAiB,EAAC,QAAQ,CAAC,CAAA;QACxD,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,WAAW,CACxD,QAAQ,EACR,cAAc,CACf,CAAA;YAED,MAAM,kBAAkB,GAAgC,EAAE,CAAA;YAC1D,WAAW,CAAC,QAAQ,CAAC,GAAG,kBAAkB,CAAA;YAE1C,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;YACnD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAA;YACzB,MAAM,cAAc,GAClB,GAAG,IAAI,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAA;YAE/D,uBAAuB;YACvB,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;gBACrC,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;oBACxC,kBAAkB,CAAC,UAAU,CAAC,GAAG,cAAc,CAAA;gBACjD,CAAC;YACH,CAAC;YAED,4BAA4B;YAC5B,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;YAEjC,uBAAuB;YACvB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC9B,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAA;gBACtD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,gEAAgE;IAChE,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAkB,CAAA;IACpD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAA;IAC/C,KAAK,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7D,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACpD,MAAM,mBAAmB,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAA;YACjE,qDAAqD;YACrD,IAAI,KAAK,GAAG,mBAAmB,EAAE,CAAC;gBAChC,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAA;gBAC/B,kBAAkB,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAA;YACzC,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAAmB,EAAE,CAAA;IAEvC,0EAA0E;IAC1E,KAAK,MAAM,CAAC,WAAW,EAAE,KAAK,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;QAC1D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,YAAY,GAAG,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;YACjD,IAAI,CAAC,YAAY,IAAI,WAAW,KAAK,YAAY,EAAE,CAAC;gBAClD,SAAQ;YACV,CAAC;YAED,qEAAqE;YACrE,IAAI,IAAI,IAAI,EAAE,EAAE,CAAC;gBACf,SAAQ;YACV,CAAC;YAED,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,EAAE,CAAC;gBAChC,YAAY,CAAC,YAAY,CAAC,GAAG,EAAE,CAAA;YACjC,CAAC;YAED,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;gBACtC,YAAY,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAA;YACvC,CAAC;YACD,MAAM,WAAW,GAAG,YAAY,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAA;YACpD,IACE,WAAW,CAAC,MAAM,GAAG,WAAW;gBAChC,CAAC,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,EAClC,CAAC;gBACD,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;YAC/B,CAAC;QACH,CAAC;IACH,CAAC;IAED,6CAA6C;IAC7C,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;QAChD,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACxC,MAAM,QAAQ,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YAC1C,IAAI,OAAO,QAAQ,KAAK,QAAQ;gBAAE,SAAQ;YAC1C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAA;YAC3C,4BAA4B;YAC5B,MAAM,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,CAAA;QACzD,CAAC;IACH,CAAC;IAED,IAAI,qBAAa,EAAE,CAAC;QAClB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAC1B,OAAO,CAAC,GAAG,CAAC,UAAU,SAAS,CAAC,MAAM,aAAa,OAAO,GAAG,SAAS,IAAI,CAAC,CAAA;QAE3E,EAAE,CAAC,aAAa,CACd,2BAA2B,EAC3B,IAAI,CAAC,SAAS,CAAC;YACb,YAAY;YACZ,WAAW;YACX,YAAY;YACZ,aAAa;SACd,CAAC,CACH,CAAA;IACH,CAAC;IAED,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,CAAA;AACtC,CAAC;AAEM,KAAK,UAAU,WAAW,CAC/B,QAAgB,EAChB,cAA8B;IAE9B,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,cAAc,CAAA;IAExC,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;QACpD,MAAM,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,MAAM,IAAI,CAAC,GAAG,CAAC,CAAA;QACzD,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,EAAE,KAAK,EAAE,UAAU,CAAC,CAAA;QACzD,MAAM,WAAW,GAAG,IAAA,aAAI,EAAC,YAAY,CAAC,UAAU,CAAC,CAAA;QACjD,MAAM,KAAK,GAAG,IAAA,aAAI,EAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC,CAAA;QAEnD,IAAI,qBAAa,EAAE,CAAC;YAClB,OAAO,CAAC,GAAG,CAAC,aAAa,QAAQ,GAAG,CAAC,CAAA;YACrC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,CAAA;YAClC,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAA;YAC3C,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,WAAW,CAAC,CAAA;YACxC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;QAC9B,CAAC;QAED,OAAO;YACL,QAAQ;YACR,WAAW,EAAE,WAAW,IAAI,EAAE;YAC9B,KAAK,EAAE,KAAK,IAAI,EAAE;SACnB,CAAA;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,qBAAa,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,wBAAwB,CAAC,EAAE,CAAC,CAAA;YAC1C,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QACvB,CAAC;QACD,OAAO;YACL,QAAQ,EAAE,CAAC;YACX,WAAW,EAAE,EAAc;YAC3B,KAAK,EAAE,EAAc;SACtB,CAAA;IACH,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAChB,MAAc,EACd,KAAmB,EACnB,UAAkB;IAElB,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE;QAC/C,UAAU,EAAE,IAAI,GAAG,IAAI;KACxB,CAAC,CAAA;IACF,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IAC9C,MAAM,MAAM,GAAgC,EAAE,CAAA;IAE9C,IAAI,qBAAa,EAAE,CAAC;QAClB,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAA;IACtC,CAAC;IAED,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,OAAO,CAAA;QAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAClB,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAA;QACnB,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC9B,CAAC;IAED,OAAO,MAAM,CAAA;AACf,CAAC"}

package/dist/code-map/parse.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=parse.test.d.ts.map

package/dist/code-map/parse.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse.test.d.ts","sourceRoot":"","sources":["../parse.test.ts"],"names":[],"mappings":""}

package/dist/code-map/parse.test.js

@@ -0,0 +1,218 @@
+"use strict";
+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 (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const bun_test_1 = require("bun:test");
+const parse_1 = require("./parse");
+// Test data
+const tsFile = `
+interface Greeter {
+    greet(name: string): string;
+}
+
+class Greeting implements Greeter {
+    private prefix: string;
+
+    constructor(prefix: string) {
+        this.prefix = prefix;
+    }
+
+    greet(name: string): string {
+        return \`\${this.prefix}, \${name}!\`;
+    }
+
+    static printGreeting(greeter: Greeter, name: string): void {
+        console.log(greeter.greet(name));
+    }
+}
+
+function createGreeter(prefix: string): Greeter {
+    return new Greeting(prefix);
+}
+
+const greeting = createGreeter('Hello');
+Greeting.printGreeting(greeting, 'World');
+`;
+const pyFile = `
+from abc import ABC, abstractmethod
+
+class Greeter(ABC):
+    @abstractmethod
+    def greet(self, name: str) -> str:
+        pass
+
+class Greeting(Greeter):
+    def __init__(self, prefix: str):
+        self.prefix = prefix
+
+    def greet(self, name: str) -> str:
+        return f'{self.prefix}, {name}!'
+
+def print_greeting(greeter: Greeter, name: str):
+    print(greeter.greet(name))
+
+if __name__ == "__main__":
+    greeting = Greeting("Hello")
+    print_greeting(greeting, "World")
+`;
+const multiDefFile1 = `
+export function utils() {
+    console.log('utils from file 1');
+}
+`;
+const multiDefFile2 = `
+// This file is deeper in the directory structure, so it will have a lower base score
+export function utils() {
+    console.log('utils from file 2');
+}
+`;
+const noDefsOnlyCallsFile = `
+import { utils } from './utils';
+utils();
+console.log('no definitions here');
+`;
+const noCallsOnlyDefsFile = `
+export function unusedFunction() {
+    console.log('never called');
+}
+`;
+const emptyFile = '';
+// Pre-read query files
+const tsQueryFile = fs.readFileSync(path.join(__dirname, 'tree-sitter-queries/tree-sitter-typescript-tags.scm'), 'utf8');
+const pyQueryFile = fs.readFileSync(path.join(__dirname, 'tree-sitter-queries/tree-sitter-python-tags.scm'), 'utf8');
+(0, bun_test_1.describe)('getFileTokenScores', () => {
+    const fsReadFileSyncSpy = (0, bun_test_1.spyOn)(fs, 'readFileSync');
+    (0, bun_test_1.beforeEach)(() => {
+        // Mock file system
+        fsReadFileSyncSpy.mockImplementation(((filePath, options) => {
+            const filePathStr = filePath.toString();
+            // Handle query files first
+            if (filePathStr.includes('tree-sitter-typescript-tags.scm')) {
+                return tsQueryFile;
+            }
+            if (filePathStr.includes('tree-sitter-python-tags.scm')) {
+                return pyQueryFile;
+            }
+            const fileName = path.basename(filePathStr);
+            let content;
+            switch (fileName) {
+                case 'test.ts':
+                    content = tsFile;
+                    break;
+                case 'test.py':
+                    content = pyFile;
+                    break;
+                case 'utils1.ts':
+                    content = multiDefFile1;
+                    break;
+                case 'utils2.ts':
+                    content = multiDefFile2;
+                    break;
+                case 'consumer.ts':
+                    content = noDefsOnlyCallsFile;
+                    break;
+                case 'unused.ts':
+                    content = noCallsOnlyDefsFile;
+                    break;
+                case 'empty.ts':
+                    content = emptyFile;
+                    break;
+                default:
+                    throw new Error(`Unexpected file: ${filePath}`);
+            }
+            // Match fs.readFileSync's behavior:
+            // - Return string if encoding is specified
+            // - Return Buffer if no encoding or encoding is null
+            if (typeof options === 'string') {
+                return content;
+            }
+            if (options && typeof options === 'object' && options.encoding) {
+                return content;
+            }
+            return Buffer.from(content);
+        }));
+    });
+    (0, bun_test_1.afterEach)(() => {
+        fsReadFileSyncSpy.mockReset();
+    });
+    bun_test_1.it.skip('should correctly identify tokens and calls in TypeScript', async () => {
+        const result = await (0, parse_1.getFileTokenScores)('/root', ['test.ts']);
+        // Check token identification
+        (0, bun_test_1.expect)(result.tokenScores['test.ts']).toHaveProperty('Greeter');
+        (0, bun_test_1.expect)(result.tokenScores['test.ts']).toHaveProperty('Greeting');
+        (0, bun_test_1.expect)(result.tokenScores['test.ts']).toHaveProperty('createGreeter');
+        (0, bun_test_1.expect)(result.tokenScores['test.ts']).toHaveProperty('greet');
+        (0, bun_test_1.expect)(result.tokenScores['test.ts']).toHaveProperty('printGreeting');
+        // Check calls
+        (0, bun_test_1.expect)(result.tokenCallers['test.ts']['Greeting']).toContain('test.ts');
+        (0, bun_test_1.expect)(result.tokenCallers['test.ts']['createGreeter']).toContain('test.ts');
+        (0, bun_test_1.expect)(result.tokenCallers['test.ts']['printGreeting']).toContain('test.ts');
+        (0, bun_test_1.expect)(result.tokenCallers['test.ts']['greet']).toContain('test.ts');
+    });
+    bun_test_1.it.skip('should correctly identify tokens and calls in Python', async () => {
+        const result = await (0, parse_1.getFileTokenScores)('/root', ['test.py']);
+        // Check token identification
+        (0, bun_test_1.expect)(result.tokenScores['test.py']).toHaveProperty('Greeter');
+        (0, bun_test_1.expect)(result.tokenScores['test.py']).toHaveProperty('Greeting');
+        (0, bun_test_1.expect)(result.tokenScores['test.py']).toHaveProperty('print_greeting');
+        (0, bun_test_1.expect)(result.tokenScores['test.py']).toHaveProperty('greet');
+        // Check calls
+        (0, bun_test_1.expect)(result.tokenCallers['test.py']['Greeting']).toContain('test.py');
+        (0, bun_test_1.expect)(result.tokenCallers['test.py']['print_greeting']).toContain('test.py');
+        (0, bun_test_1.expect)(result.tokenCallers['test.py']['greet']).toContain('test.py');
+    });
+    bun_test_1.it.skip('should use highest scoring definition when token is defined in multiple files', async () => {
+        const result = await (0, parse_1.getFileTokenScores)('/root', [
+            'utils1.ts',
+            'deep/utils2.ts',
+            'consumer.ts',
+        ]);
+        // utils1.ts has a higher score (shallower path)
+        (0, bun_test_1.expect)(result.tokenCallers['utils1.ts']['utils']).toContain('consumer.ts');
+        // utils2.ts should not be chosen as the defining file
+        (0, bun_test_1.expect)(result.tokenCallers['deep/utils2.ts']['utils']).toBeUndefined();
+    });
+    (0, bun_test_1.it)('should handle files with no definitions', async () => {
+        const result = await (0, parse_1.getFileTokenScores)('/root', ['consumer.ts']);
+        // No definitions, only calls
+        (0, bun_test_1.expect)(Object.keys(result.tokenScores['consumer.ts'])).toHaveLength(0);
+        // External calls are tracked but not mapped to callers since definition is unknown
+        (0, bun_test_1.expect)(Object.keys(result.tokenCallers)).toHaveLength(0);
+    });
+    bun_test_1.it.skip('should handle files with no calls', async () => {
+        const result = await (0, parse_1.getFileTokenScores)('/root', ['unused.ts']);
+        // Has definition but no calls
+        (0, bun_test_1.expect)(result.tokenScores['unused.ts']).toHaveProperty('unusedFunction');
+        (0, bun_test_1.expect)(result.tokenCallers['unused.ts']['unusedFunction']).toEqual([]);
+    });
+    (0, bun_test_1.it)('should handle empty files', async () => {
+        const result = await (0, parse_1.getFileTokenScores)('/root', ['empty.ts']);
+        // No definitions or calls
+        (0, bun_test_1.expect)(Object.keys(result.tokenScores['empty.ts'] || {})).toHaveLength(0);
+        (0, bun_test_1.expect)(Object.keys(result.tokenCallers)).toHaveLength(0);
+    });
+});
+//# sourceMappingURL=parse.test.js.map

package/dist/code-map/parse.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse.test.js","sourceRoot":"","sources":["../parse.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AAAA,uCAAwB;AACxB,2CAA4B;AAC5B,uCAA6E;AAC7E,mCAA4C;AAG5C,YAAY;AACZ,MAAM,MAAM,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2Bd,CAAA;AAED,MAAM,MAAM,GAAG;;;;;;;;;;;;;;;;;;;;;CAqBd,CAAA;AAED,MAAM,aAAa,GAAG;;;;CAIrB,CAAA;AAED,MAAM,aAAa,GAAG;;;;;CAKrB,CAAA;AAED,MAAM,mBAAmB,GAAG;;;;CAI3B,CAAA;AAED,MAAM,mBAAmB,GAAG;;;;CAI3B,CAAA;AAED,MAAM,SAAS,GAAG,EAAE,CAAA;AAEpB,uBAAuB;AACvB,MAAM,WAAW,GAAG,EAAE,CAAC,YAAY,CACjC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,qDAAqD,CAAC,EAC3E,MAAM,CACP,CAAA;AAED,MAAM,WAAW,GAAG,EAAE,CAAC,YAAY,CACjC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,iDAAiD,CAAC,EACvE,MAAM,CACP,CAAA;AAED,IAAA,mBAAQ,EAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,MAAM,iBAAiB,GAAG,IAAA,gBAAK,EAAC,EAAE,EAAE,cAAc,CAAC,CAAA;IAEnD,IAAA,qBAAU,EAAC,GAAG,EAAE;QACd,mBAAmB;QACnB,iBAAiB,CAAC,kBAAkB,CAAC,CAAC,CACpC,QAA8B,EAC9B,OAGQ,EACS,EAAE;YACnB,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,EAAE,CAAA;YAEvC,2BAA2B;YAC3B,IAAI,WAAW,CAAC,QAAQ,CAAC,iCAAiC,CAAC,EAAE,CAAC;gBAC5D,OAAO,WAAW,CAAA;YACpB,CAAC;YACD,IAAI,WAAW,CAAC,QAAQ,CAAC,6BAA6B,CAAC,EAAE,CAAC;gBACxD,OAAO,WAAW,CAAA;YACpB,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAA;YAC3C,IAAI,OAAe,CAAA;YACnB,QAAQ,QAAQ,EAAE,CAAC;gBACjB,KAAK,SAAS;oBACZ,OAAO,GAAG,MAAM,CAAA;oBAChB,MAAK;gBACP,KAAK,SAAS;oBACZ,OAAO,GAAG,MAAM,CAAA;oBAChB,MAAK;gBACP,KAAK,WAAW;oBACd,OAAO,GAAG,aAAa,CAAA;oBACvB,MAAK;gBACP,KAAK,WAAW;oBACd,OAAO,GAAG,aAAa,CAAA;oBACvB,MAAK;gBACP,KAAK,aAAa;oBAChB,OAAO,GAAG,mBAAmB,CAAA;oBAC7B,MAAK;gBACP,KAAK,WAAW;oBACd,OAAO,GAAG,mBAAmB,CAAA;oBAC7B,MAAK;gBACP,KAAK,UAAU;oBACb,OAAO,GAAG,SAAS,CAAA;oBACnB,MAAK;gBACP;oBACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,QAAQ,EAAE,CAAC,CAAA;YACnD,CAAC;YAED,oCAAoC;YACpC,2CAA2C;YAC3C,qDAAqD;YACrD,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAChC,OAAO,OAAO,CAAA;YAChB,CAAC;YACD,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;gBAC/D,OAAO,OAAO,CAAA;YAChB,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC7B,CAAC,CAA2B,CAAC,CAAA;IAC/B,CAAC,CAAC,CAAA;IAEF,IAAA,oBAAS,EAAC,GAAG,EAAE;QACb,iBAAiB,CAAC,SAAS,EAAE,CAAA;IAC/B,CAAC,CAAC,CAAA;IAEF,aAAE,CAAC,IAAI,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAkB,EAAC,OAAO,EAAE,CAAC,SAAS,CAAC,CAAC,CAAA;QAE7D,6BAA6B;QAC7B,IAAA,iBAAM,EAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAA;QAC/D,IAAA,iBAAM,EAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAA;QAChE,IAAA,iBAAM,EAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAA;QACrE,IAAA,iBAAM,EAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAA;QAC7D,IAAA,iBAAM,EAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAA;QAErE,cAAc;QACd,IAAA,iBAAM,EAAC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAA;QACvE,IAAA,iBAAM,EAAC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAA;QAC5E,IAAA,iBAAM,EAAC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAA;QAC5E,IAAA,iBAAM,EAAC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAA;IACtE,CAAC,CAAC,CAAA;IAEF,aAAE,CAAC,IAAI,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAkB,EAAC,OAAO,EAAE,CAAC,SAAS,CAAC,CAAC,CAAA;QAE7D,6BAA6B;QAC7B,IAAA,iBAAM,EAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAA;QAC/D,IAAA,iBAAM,EAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAA;QAChE,IAAA,iBAAM,EAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAA;QACtE,IAAA,iBAAM,EAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAA;QAE7D,cAAc;QACd,IAAA,iBAAM,EAAC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAA;QACvE,IAAA,iBAAM,EAAC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS,CAChE,SAAS,CACV,CAAA;QACD,IAAA,iBAAM,EAAC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAA;IACtE,CAAC,CAAC,CAAA;IAEF,aAAE,CAAC,IAAI,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;QAClG,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAkB,EAAC,OAAO,EAAE;YAC/C,WAAW;YACX,gBAAgB;YAChB,aAAa;SACd,CAAC,CAAA;QAEF,gDAAgD;QAChD,IAAA,iBAAM,EAAC,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAA;QAC1E,sDAAsD;QACtD,IAAA,iBAAM,EAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,EAAE,CAAA;IACxE,CAAC,CAAC,CAAA;IAEF,IAAA,aAAE,EAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAkB,EAAC,OAAO,EAAE,CAAC,aAAa,CAAC,CAAC,CAAA;QAEjE,6BAA6B;QAC7B,IAAA,iBAAM,EAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QACtE,mFAAmF;QACnF,IAAA,iBAAM,EAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;IAC1D,CAAC,CAAC,CAAA;IAEF,aAAE,CAAC,IAAI,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAkB,EAAC,OAAO,EAAE,CAAC,WAAW,CAAC,CAAC,CAAA;QAE/D,8BAA8B;QAC9B,IAAA,iBAAM,EAAC,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAA;QACxE,IAAA,iBAAM,EAAC,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IACxE,CAAC,CAAC,CAAA;IAEF,IAAA,aAAE,EAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAkB,EAAC,OAAO,EAAE,CAAC,UAAU,CAAC,CAAC,CAAA;QAE9D,0BAA0B;QAC1B,IAAA,iBAAM,EAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QACzE,IAAA,iBAAM,EAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;IAC1D,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/code-map/scripts/tree-sitter-test.cjs

@@ -0,0 +1,27 @@
+"use strict";
+// CommonJS version to avoid module system issues
+const Parser = require('tree-sitter');
+const { Query } = require('tree-sitter');
+const treeSitterTypescript = require('tree-sitter-typescript');
+async function main() {
+    const parser = new Parser();
+    const language = treeSitterTypescript.typescript;
+    parser.setLanguage(language);
+    // Simple TypeScript code
+    const sourceCode = `
+interface Greeter {
+  greet(): string;
+}
+`;
+    const tree = parser.parse(sourceCode);
+    // Simple query to find interface name
+    const queryString = '(interface_declaration name: (type_identifier) @interface_name)';
+    const query = new Query(parser.getLanguage(), queryString);
+    const captures = query.captures(tree.rootNode);
+    // Log for debugging
+    console.log('Tree:', tree.rootNode.toString());
+    console.log('Captures:', captures);
+    console.log('First capture text:', captures[0]?.node.text);
+}
+main().catch(console.error);
+//# sourceMappingURL=tree-sitter-test.cjs.map

package/dist/code-map/scripts/tree-sitter-test.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"tree-sitter-test.cjs","sourceRoot":"","sources":["../../scripts/tree-sitter-test.cjs"],"names":[],"mappings":";AAAA,iDAAiD;AACjD,MAAM,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAA;AACrC,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC,CAAA;AACxC,MAAM,oBAAoB,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAAA;AAE9D,KAAK,UAAU,IAAI;IACjB,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAA;IAC3B,MAAM,QAAQ,GAAG,oBAAoB,CAAC,UAAU,CAAA;IAEhD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAA;IAE5B,yBAAyB;IACzB,MAAM,UAAU,GAAG;;;;CAIpB,CAAA;IACC,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAA;IAErC,sCAAsC;IACtC,MAAM,WAAW,GAAG,iEAAiE,CAAA;IACrF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,WAAW,CAAC,CAAA;IAC1D,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IAE9C,oBAAoB;IACpB,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAA;IAC9C,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAA;IAClC,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAA;AAC5D,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA"}

package/dist/code-map/scripts/tree-sitter-test.d.cts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=tree-sitter-test.d.cts.map

package/dist/code-map/scripts/tree-sitter-test.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tree-sitter-test.d.cts","sourceRoot":"","sources":["../../scripts/tree-sitter-test.cjs"],"names":[],"mappings":""}

package/dist/code-map/scripts/tree-sitter-test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=tree-sitter-test.d.ts.map

package/dist/code-map/scripts/tree-sitter-test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tree-sitter-test.d.ts","sourceRoot":"","sources":["../../scripts/tree-sitter-test.ts"],"names":[],"mappings":""}

package/dist/code-map/scripts/tree-sitter-test.js

@@ -0,0 +1,54 @@
+"use strict";
+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 (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const tree_sitter_1 = __importDefault(require("tree-sitter"));
+const tree_sitter_2 = require("tree-sitter");
+async function main() {
+    const parser = new tree_sitter_1.default();
+    const languageModule = await Promise.resolve().then(() => __importStar(require('tree-sitter-typescript')));
+    // Cast to any to avoid type mismatch between tree-sitter and tree-sitter-typescript
+    const language = languageModule.typescript;
+    parser.setLanguage(language);
+    // Simple TypeScript code
+    const sourceCode = `
+interface Greeter {
+  greet(): string;
+}
+`;
+    const tree = parser.parse(sourceCode);
+    // Simple query to find interface name
+    const queryString = '(interface_declaration name: (type_identifier) @interface_name)';
+    const query = new tree_sitter_2.Query(parser.getLanguage(), queryString);
+    const captures = query.captures(tree.rootNode);
+    // Log for debugging
+    console.log('Tree:', tree.rootNode.toString());
+    console.log('Captures:', captures);
+    console.log('First capture text:', captures[0]?.node.text);
+}
+main().catch(console.error);
+//# sourceMappingURL=tree-sitter-test.js.map

package/dist/code-map/scripts/tree-sitter-test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tree-sitter-test.js","sourceRoot":"","sources":["../../scripts/tree-sitter-test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,8DAAgC;AAChC,6CAAmC;AAEnC,KAAK,UAAU,IAAI;IACjB,MAAM,MAAM,GAAG,IAAI,qBAAM,EAAE,CAAA;IAC3B,MAAM,cAAc,GAAG,wDAAa,wBAAwB,GAAC,CAAA;IAC7D,oFAAoF;IACpF,MAAM,QAAQ,GAAG,cAAc,CAAC,UAAiB,CAAA;IAEjD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAA;IAE5B,yBAAyB;IACzB,MAAM,UAAU,GAAG;;;;CAIpB,CAAA;IACC,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAA;IAErC,sCAAsC;IACtC,MAAM,WAAW,GAAG,iEAAiE,CAAA;IACrF,MAAM,KAAK,GAAG,IAAI,mBAAK,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,WAAW,CAAC,CAAA;IAC1D,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IAE9C,oBAAoB;IACpB,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAA;IAC9C,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAA;IAClC,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAA;AAC5D,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA"}

package/dist/code-map/test-parser.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=test-parser.d.ts.map

package/dist/code-map/test-parser.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-parser.d.ts","sourceRoot":"","sources":["../test-parser.js"],"names":[],"mappings":""}

package/dist/code-map/test-parser.js

@@ -0,0 +1,42 @@
+"use strict";
+const Parser = require('tree-sitter');
+const { typescript } = require('tree-sitter-typescript');
+const { Query } = require('tree-sitter');
+async function testParser() {
+    try {
+        const parser = new Parser();
+        console.log('Parser created');
+        parser.setLanguage(typescript);
+        console.log('Language set');
+        const sourceCode = `
+interface Greeter {
+  greet(): string;
+}
+`;
+        console.log('Source code:', sourceCode);
+        const tree = parser.parse(sourceCode);
+        console.log('Parse tree:', tree);
+        if (!tree) {
+            throw new Error('Parser returned undefined tree');
+        }
+        const queryString = '(interface_declaration name: (type_identifier) @interface_name)';
+        const query = new Query(parser.getLanguage(), queryString);
+        const captures = query.captures(tree.rootNode);
+        console.log('Tree:', tree.rootNode.toString());
+        console.log('Captures:', captures);
+        console.log('First capture text:', captures[0]?.node.text);
+        if (captures.length !== 1 || captures[0].node.text !== 'Greeter') {
+            throw new Error('Expected to find "Greeter" interface name');
+        }
+        console.log('Test passed!');
+    }
+    catch (err) {
+        console.error('Error:', err);
+        process.exit(1);
+    }
+}
+testParser().catch(err => {
+    console.error('Unhandled error:', err);
+    process.exit(1);
+});
+//# sourceMappingURL=test-parser.js.map

package/dist/code-map/test-parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-parser.js","sourceRoot":"","sources":["../test-parser.js"],"names":[],"mappings":";AAAA,MAAM,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;AACtC,MAAM,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAAC;AACzD,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;AAEzC,KAAK,UAAU,UAAU;IACvB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAE9B,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QAC/B,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAE5B,MAAM,UAAU,GAAG;;;;CAItB,CAAC;QACE,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QAExC,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACtC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;QAEjC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,WAAW,GAAG,iEAAiE,CAAC;QACtF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,WAAW,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE/C,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC/C,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAE3D,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC/D,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IAC9B,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;QAC7B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,UAAU,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE;IACvB,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,GAAG,CAAC,CAAC;IACvC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/code-map/tree-sitter-queries/tree-sitter-c-tags.scm

@@ -8,4 +8,9 @@
 
 (enum_specifier name: (type_identifier) @identifier)
 
-; TODO add calls
+; Function calls
+(call_expression
+  function: (identifier) @call.identifier)
+
+(call_expression
+  function: (field_expression field: (field_identifier) @call.identifier))

package/dist/code-map/tree-sitter-queries/tree-sitter-c_sharp-tags.scm

@@ -10,4 +10,14 @@
  name: (identifier) @identifier
  )
 
-; TODO add calls
+; Method calls
+(invocation_expression
+  function: (identifier) @call.identifier)
+
+(invocation_expression
+  function: (member_access_expression
+    name: (identifier) @call.identifier))
+
+; Constructor calls
+(object_creation_expression
+  type: (identifier) @call.identifier)