@vybestack/llxprt-code-tools 0.10.0-nightly.260613.1adad3b34

package/dist/src/tools/ast-edit/ast-query-extractor.js

@@ -0,0 +1,178 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { parse } from '@ast-grep/napi';
+import { LANGUAGE_MAP, JAVASCRIPT_FAMILY_EXTENSIONS, } from '../../utils/ast-grep-utils.js';
+import { KEYWORDS, COMMENT_PREFIXES } from './constants.js';
+/**
+ * ASTQueryExtractor handles AST parsing with @ast-grep/napi and declaration extraction.
+ */
+export class ASTQueryExtractor {
+    constructor() { }
+    async extractDeclarations(filePath, content) {
+        // eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- intentional falsy coalescing: file extension may be empty string, should fall through to empty
+        const extension = (filePath.split('.').pop() || '').toLowerCase();
+        const lang = LANGUAGE_MAP[extension];
+        if (!lang) {
+            return this.fallbackExtraction(content, 'unknown');
+        }
+        try {
+            const root = parse(lang, content);
+            const declarations = [];
+            const sgRoot = root.root();
+            if (JAVASCRIPT_FAMILY_EXTENSIONS.includes(extension)) {
+                this.extractJsFamilyDeclarations(sgRoot, declarations);
+            }
+            else if (extension === 'py') {
+                this.extractPythonDeclarations(sgRoot, declarations);
+            }
+            else {
+                return this.fallbackExtraction(content, extension);
+            }
+            return declarations;
+        }
+        catch {
+            return this.fallbackExtraction(content, extension);
+        }
+    }
+    extractJsFamilyDeclarations(sgRoot, declarations) {
+        // Functions
+        sgRoot.findAll({ rule: { kind: 'function_declaration' } }).forEach((n) => {
+            const nameNode = n.field('name');
+            const paramsNode = n.field('parameters');
+            const returnTypeNode = n.field('return_type');
+            if (nameNode != null) {
+                const signature = this.buildSignature(paramsNode, returnTypeNode);
+                declarations.push(this.nodeToDeclaration(n, nameNode.text(), 'function', signature));
+            }
+        });
+        // Methods
+        sgRoot.findAll({ rule: { kind: 'method_definition' } }).forEach((n) => {
+            const nameNode = n.field('name');
+            const paramsNode = n.field('parameters');
+            const returnTypeNode = n.field('return_type');
+            if (nameNode != null) {
+                const signature = this.buildSignature(paramsNode, returnTypeNode);
+                declarations.push(this.nodeToDeclaration(n, nameNode.text(), 'function', signature));
+            }
+        });
+        // Classes
+        sgRoot.findAll({ rule: { kind: 'class_declaration' } }).forEach((n) => {
+            const nameNode = n.field('name');
+            if (nameNode != null) {
+                declarations.push(this.nodeToDeclaration(n, nameNode.text(), 'class'));
+            }
+        });
+        // Variables
+        sgRoot.findAll({ rule: { kind: 'variable_declarator' } }).forEach((n) => {
+            const nameNode = n.field('name');
+            if (nameNode != null) {
+                declarations.push(this.nodeToDeclaration(n, nameNode.text(), 'variable'));
+            }
+        });
+        // Imports
+        sgRoot.findAll({ rule: { kind: 'import_statement' } }).forEach((n) => {
+            const sourceNode = n.field('source');
+            declarations.push(this.nodeToDeclaration(n, sourceNode != null ? sourceNode.text() : 'import', 'import'));
+        });
+    }
+    extractPythonDeclarations(sgRoot, declarations) {
+        sgRoot.findAll({ rule: { kind: 'function_definition' } }).forEach((n) => {
+            const nameNode = n.field('name');
+            const paramsNode = n.field('parameters');
+            const returnTypeNode = n.field('return_type');
+            if (nameNode != null) {
+                const signature = this.buildPythonSignature(paramsNode, returnTypeNode);
+                declarations.push(this.nodeToDeclaration(n, nameNode.text(), 'function', signature));
+            }
+        });
+        sgRoot.findAll({ rule: { kind: 'class_definition' } }).forEach((n) => {
+            const nameNode = n.field('name');
+            if (nameNode != null) {
+                declarations.push(this.nodeToDeclaration(n, nameNode.text(), 'class'));
+            }
+        });
+    }
+    buildSignature(paramsNode, returnTypeNode) {
+        let signature = paramsNode != null ? paramsNode.text() : '()';
+        if (returnTypeNode != null) {
+            signature += returnTypeNode.text();
+        }
+        return signature;
+    }
+    buildPythonSignature(paramsNode, returnTypeNode) {
+        let signature = paramsNode != null ? paramsNode.text() : '()';
+        if (returnTypeNode != null) {
+            signature += ` -> ${returnTypeNode.text()}`;
+        }
+        return signature;
+    }
+    nodeToDeclaration(n, name, type, signature) {
+        const range = n.range();
+        return {
+            name,
+            type,
+            line: range.start.line + 1,
+            column: range.start.column,
+            range: {
+                start: { line: range.start.line + 1, column: range.start.column },
+                end: { line: range.end.line + 1, column: range.end.column },
+            },
+            visibility: 'public',
+            signature,
+        };
+    }
+    fallbackExtraction(content, _language) {
+        // Keep the regex-based fallback for robustness
+        const declarations = [];
+        const lines = content.split('\n');
+        lines.forEach((line, index) => {
+            const trimmed = line.trim();
+            const isComment = COMMENT_PREFIXES.some((prefix) => trimmed.startsWith(prefix));
+            if (!trimmed || isComment)
+                return;
+            if (line.includes(KEYWORDS.FUNCTION) ||
+                line.includes(KEYWORDS.DEF) ||
+                line.includes(KEYWORDS.CLASS)) {
+                const name = this.extractNameBasic(trimmed);
+                const column = Math.max(0, line.indexOf(name));
+                declarations.push({
+                    name,
+                    type: trimmed.includes(KEYWORDS.CLASS) ? 'class' : 'function',
+                    line: index + 1,
+                    column,
+                    signature: this.extractSignatureBasic(trimmed),
+                });
+            }
+        });
+        return declarations.map((decl) => ({
+            ...decl,
+            range: {
+                start: { line: decl.line, column: decl.column },
+                end: { line: decl.line, column: decl.column + decl.name.length },
+            },
+            visibility: 'public',
+            signature: decl.signature,
+        }));
+    }
+    extractNameBasic(line) {
+        // eslint-disable-next-line sonarjs/regular-expr -- Static regex reviewed for lint hardening; behavior preserved.
+        const match = line.match(/(?:function|def|class)\s+(\w+)/);
+        return match ? match[1] : 'unknown';
+    }
+    extractSignatureBasic(line) {
+        // Capture parameters: ( ... ). Use index scanning instead of a regex to
+        // avoid polynomial backtracking on lines with many unmatched parentheses.
+        const open = line.indexOf('(');
+        if (open !== -1) {
+            const close = line.indexOf(')', open + 1);
+            if (close !== -1) {
+                return `(${line.slice(open + 1, close)})`;
+            }
+        }
+        return '';
+    }
+}
+//# sourceMappingURL=ast-query-extractor.js.map

package/dist/src/tools/ast-edit/ast-query-extractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast-query-extractor.js","sourceRoot":"","sources":["../../../../src/tools/ast-edit/ast-query-extractor.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AACvC,OAAO,EACL,YAAY,EACZ,4BAA4B,GAC7B,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAE5D;;GAEG;AACH,MAAM,OAAO,iBAAiB;IAC5B,gBAAe,CAAC;IAEhB,KAAK,CAAC,mBAAmB,CACvB,QAAgB,EAChB,OAAe;QAEf,0KAA0K;QAC1K,MAAM,SAAS,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QAClE,MAAM,IAAI,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;QACrC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YAClC,MAAM,YAAY,GAA0B,EAAE,CAAC;YAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YAE3B,IAAI,4BAA4B,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrD,IAAI,CAAC,2BAA2B,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACzD,CAAC;iBAAM,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;gBAC9B,IAAI,CAAC,yBAAyB,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACvD,CAAC;iBAAM,CAAC;gBACN,OAAO,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YACrD,CAAC;YAED,OAAO,YAAY,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAEO,2BAA2B,CACjC,MAAoD,EACpD,YAAmC;QAEnC,YAAY;QACZ,MAAM,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,sBAAsB,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACvE,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACjC,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACzC,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC9C,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;gBACrB,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;gBAClE,YAAY,CAAC,IAAI,CACf,IAAI,CAAC,iBAAiB,CAAC,CAAC,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,SAAS,CAAC,CAClE,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,UAAU;QACV,MAAM,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACpE,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACjC,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACzC,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC9C,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;gBACrB,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;gBAClE,YAAY,CAAC,IAAI,CACf,IAAI,CAAC,iBAAiB,CAAC,CAAC,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,SAAS,CAAC,CAClE,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,UAAU;QACV,MAAM,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACpE,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;gBACrB,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC;YACzE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,YAAY;QACZ,MAAM,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,qBAAqB,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACtE,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;gBACrB,YAAY,CAAC,IAAI,CACf,IAAI,CAAC,iBAAiB,CAAC,CAAC,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,UAAU,CAAC,CACvD,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,UAAU;QACV,MAAM,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACnE,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACrC,YAAY,CAAC,IAAI,CACf,IAAI,CAAC,iBAAiB,CACpB,CAAC,EACD,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,QAAQ,EACjD,QAAQ,CACT,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,yBAAyB,CAC/B,MAAoD,EACpD,YAAmC;QAEnC,MAAM,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,qBAAqB,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACtE,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACjC,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACzC,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC9C,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;gBACrB,MAAM,SAAS,GAAG,IAAI,CAAC,oBAAoB,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;gBACxE,YAAY,CAAC,IAAI,CACf,IAAI,CAAC,iBAAiB,CAAC,CAAC,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,SAAS,CAAC,CAClE,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACnE,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;gBACrB,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC;YACzE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CACpB,UAA+D,EAC/D,cAAmE;QAEnE,IAAI,SAAS,GAAG,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QAC9D,IAAI,cAAc,IAAI,IAAI,EAAE,CAAC;YAC3B,SAAS,IAAI,cAAc,CAAC,IAAI,EAAE,CAAC;QACrC,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,oBAAoB,CAC1B,UAA+D,EAC/D,cAAmE;QAEnE,IAAI,SAAS,GAAG,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QAC9D,IAAI,cAAc,IAAI,IAAI,EAAE,CAAC;YAC3B,SAAS,IAAI,OAAO,cAAc,CAAC,IAAI,EAAE,EAAE,CAAC;QAC9C,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,iBAAiB,CACvB,CAA+C,EAC/C,IAAY,EACZ,IAAyB,EACzB,SAAkB;QAElB,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;QACxB,OAAO;YACL,IAAI;YACJ,IAAI;YACJ,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC;YAC1B,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM;YAC1B,KAAK,EAAE;gBACL,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE;gBACjE,GAAG,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE;aAC5D;YACD,UAAU,EAAE,QAAQ;YACpB,SAAS;SACV,CAAC;IACJ,CAAC;IAEO,kBAAkB,CACxB,OAAe,EACf,SAAiB;QAEjB,+CAA+C;QAC/C,MAAM,YAAY,GAAkB,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAElC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YAC5B,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CACjD,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAC3B,CAAC;YACF,IAAI,CAAC,OAAO,IAAI,SAAS;gBAAE,OAAO;YAElC,IACE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBAChC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC;gBAC3B,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,EAC7B,CAAC;gBACD,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;gBAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC/C,YAAY,CAAC,IAAI,CAAC;oBAChB,IAAI;oBACJ,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU;oBAC7D,IAAI,EAAE,KAAK,GAAG,CAAC;oBACf,MAAM;oBACN,SAAS,EAAE,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC;iBAC/C,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACjC,GAAG,IAAI;YACP,KAAK,EAAE;gBACL,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE;gBAC/C,GAAG,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;aACjE;YACD,UAAU,EAAE,QAAQ;YACpB,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC,CAAC;IACN,CAAC;IAEO,gBAAgB,CAAC,IAAY;QACnC,iHAAiH;QACjH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;QAC3D,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACtC,CAAC;IAEO,qBAAqB,CAAC,IAAY;QACxC,wEAAwE;QACxE,0EAA0E;QAC1E,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC/B,IAAI,IAAI,KAAK,CAAC,CAAC,EAAE,CAAC;YAChB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;YAC1C,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;gBACjB,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC;YAC5C,CAAC;QACH,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;CACF"}

package/dist/src/tools/ast-edit/ast-read-file-invocation.d.ts

@@ -0,0 +1,26 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * AST Read File Tool Invocation - Handles execution of file read operations with context
+ */
+import { type ToolInvocation, type ToolLocation, type ToolResult } from '../tools.js';
+import type { IToolHost } from '../../interfaces/index.js';
+import type { AnsiOutput } from '../../utils/terminalSerializer.js';
+import type { ASTReadFileToolParams } from './types.js';
+import type { ASTContextCollector } from './context-collector.js';
+export declare class ASTReadFileToolInvocation implements ToolInvocation<ASTReadFileToolParams, ToolResult> {
+    private readonly host;
+    params: ASTReadFileToolParams;
+    private readonly contextCollector;
+    constructor(host: IToolHost, params: ASTReadFileToolParams, contextCollector: ASTContextCollector);
+    toolLocations(): ToolLocation[];
+    getDescription(): string;
+    shouldConfirmExecute(): Promise<false>;
+    execute(_signal?: AbortSignal, _updateOutput?: (output: string | AnsiOutput) => void, _terminalColumns?: number, _terminalRows?: number, _setPidCallback?: (pid: number) => void): Promise<ToolResult>;
+    private computeLineRange;
+    private buildReadLlmContent;
+    private formatConnectedFilesContext;
+    private handleReadError;
+}

package/dist/src/tools/ast-edit/ast-read-file-invocation.js

@@ -0,0 +1,149 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * AST Read File Tool Invocation - Handles execution of file read operations with context
+ */
+import * as path from 'path';
+import fs from 'fs';
+import { ToolErrorType } from '../../types/tool-error.js';
+import { makeRelative, shortenPath } from '../../utils/paths.js';
+import { isNodeError } from '../../utils/errors.js';
+import { countLines } from '../../utils/fileUtils.js';
+import { ASTConfig } from './ast-config.js';
+export class ASTReadFileToolInvocation {
+    host;
+    params;
+    contextCollector;
+    constructor(host, params, contextCollector) {
+        this.host = host;
+        this.params = params;
+        this.contextCollector = contextCollector;
+    }
+    toolLocations() {
+        return [{ path: this.params.file_path, line: this.params.offset }];
+    }
+    getDescription() {
+        const relativePath = makeRelative(this.params.file_path, this.host.getTargetDir());
+        return shortenPath(relativePath);
+    }
+    async shouldConfirmExecute() {
+        return false; // Read operations don't need confirmation
+    }
+    async execute(_signal, _updateOutput, _terminalColumns, _terminalRows, _setPidCallback) {
+        try {
+            const fileSystemService = this.host.getFileSystemService?.();
+            const content = fileSystemService?.readTextFile
+                ? await fileSystemService.readTextFile(this.params.file_path)
+                : await fs.promises.readFile(this.params.file_path, 'utf-8');
+            const { selectedContent, startLine, endLine, totalLineCount } = this.computeLineRange(content);
+            const workspaceRoot = this.host.getTargetDir();
+            const enhancedContext = await this.contextCollector.collectEnhancedContext(this.params.file_path, content, workspaceRoot);
+            const readLlmContent = this.buildReadLlmContent(enhancedContext, workspaceRoot, startLine, endLine, totalLineCount);
+            return {
+                llmContent: readLlmContent,
+                returnDisplay: {
+                    content: selectedContent,
+                    fileName: path.basename(this.params.file_path),
+                    filePath: this.params.file_path,
+                    metadata: {
+                        language: enhancedContext.language,
+                        declarationsCount: enhancedContext.declarations.length,
+                    },
+                },
+            };
+        }
+        catch (error) {
+            return this.handleReadError(error);
+        }
+    }
+    computeLineRange(content) {
+        const lines = content.split('\n');
+        const totalLineCount = countLines(lines);
+        const startLine = Math.min(typeof this.params.offset === 'number' && this.params.offset > 0
+            ? Math.max(1, this.params.offset) - 1
+            : 0, totalLineCount);
+        const endLine = typeof this.params.limit === 'number' && this.params.limit > 0
+            ? Math.min(totalLineCount, startLine + this.params.limit)
+            : totalLineCount;
+        const selectedContent = lines.slice(startLine, endLine).join('\n');
+        return { selectedContent, startLine, endLine, totalLineCount };
+    }
+    buildReadLlmContent(enhancedContext, workspaceRoot, startLine, endLine, totalLineCount) {
+        return [
+            `LLXPRT READ: ${this.params.file_path}`,
+            `- Language: ${enhancedContext.language}`,
+            `- Lines ${Math.min(startLine + 1, totalLineCount)}-${endLine} of ${totalLineCount}`,
+            `- Declarations: ${enhancedContext.declarations.length}`,
+            '',
+            'CONTEXT ANALYSIS:',
+            ...enhancedContext.declarations.map((decl) => `- ${decl.type}: ${decl.name}${decl.signature ?? ''} (line ${decl.line})`),
+            '',
+            'RELEVANT SNIPPETS:',
+            ...enhancedContext.relevantSnippets
+                .slice(0, ASTConfig.MAX_DISPLAY_RESULTS)
+                .map((snippet) => `- Line ${snippet.line}: ${snippet.text.substring(0, 60)}...`),
+            this.formatConnectedFilesContext(enhancedContext, workspaceRoot),
+        ]
+            .flat()
+            .filter(Boolean)
+            .join('\n');
+    }
+    formatConnectedFilesContext(enhancedContext, workspaceRoot) {
+        if (!enhancedContext.connectedFiles ||
+            enhancedContext.connectedFiles.length === 0) {
+            return [];
+        }
+        return [
+            '',
+            'WORKING SET CONTEXT:',
+            ...enhancedContext.connectedFiles
+                .map((file) => {
+                const relPath = makeRelative(file.filePath, workspaceRoot);
+                if (file.declarations.length === 0)
+                    return `- ${relPath} (No declarations)`;
+                return [
+                    `- ${relPath}:`,
+                    ...file.declarations.map((d) => `  - ${d.type}: ${d.name}${d.signature ?? ''}`),
+                ];
+            })
+                .flat(),
+        ];
+    }
+    handleReadError(error) {
+        const errorMsg = error instanceof Error ? error.message : String(error);
+        let errorType = ToolErrorType.READ_CONTENT_FAILURE;
+        if (isNodeError(error)) {
+            switch (error.code) {
+                case 'ENOENT':
+                    errorType = ToolErrorType.FILE_NOT_FOUND;
+                    break;
+                case 'EACCES':
+                    errorType = ToolErrorType.PERMISSION_DENIED;
+                    break;
+                case 'EISDIR':
+                    errorType = ToolErrorType.TARGET_IS_DIRECTORY;
+                    break;
+                case 'EMFILE':
+                case 'ENFILE':
+                    errorType = ToolErrorType.READ_CONTENT_FAILURE;
+                    break;
+                default:
+                    errorType = ToolErrorType.READ_CONTENT_FAILURE;
+            }
+        }
+        else {
+            errorType = ToolErrorType.UNKNOWN;
+        }
+        return {
+            llmContent: `Error reading file: ${errorMsg}`,
+            returnDisplay: `Error reading file: ${errorMsg}`,
+            error: {
+                message: errorMsg,
+                type: errorType,
+            },
+        };
+    }
+}
+//# sourceMappingURL=ast-read-file-invocation.js.map

package/dist/src/tools/ast-edit/ast-read-file-invocation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast-read-file-invocation.js","sourceRoot":"","sources":["../../../../src/tools/ast-edit/ast-read-file-invocation.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,MAAM,IAAI,CAAC;AAMpB,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAKtD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAG5C,MAAM,OAAO,yBAAyB;IAIjB;IACV;IACU;IAHnB,YACmB,IAAe,EACzB,MAA6B,EACnB,gBAAqC;QAFrC,SAAI,GAAJ,IAAI,CAAW;QACzB,WAAM,GAAN,MAAM,CAAuB;QACnB,qBAAgB,GAAhB,gBAAgB,CAAqB;IACrD,CAAC;IAEJ,aAAa;QACX,OAAO,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IACrE,CAAC;IAED,cAAc;QACZ,MAAM,YAAY,GAAG,YAAY,CAC/B,IAAI,CAAC,MAAM,CAAC,SAAS,EACrB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CACzB,CAAC;QACF,OAAO,WAAW,CAAC,YAAY,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,oBAAoB;QACxB,OAAO,KAAK,CAAC,CAAC,0CAA0C;IAC1D,CAAC;IAED,KAAK,CAAC,OAAO,CACX,OAAqB,EACrB,aAAqD,EACrD,gBAAyB,EACzB,aAAsB,EACtB,eAAuC;QAEvC,IAAI,CAAC;YACH,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAE7C,CAAC;YACd,MAAM,OAAO,GAAG,iBAAiB,EAAE,YAAY;gBAC7C,CAAC,CAAC,MAAM,iBAAiB,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;gBAC7D,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAE/D,MAAM,EAAE,eAAe,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,GAC3D,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAEjC,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YAC/C,MAAM,eAAe,GACnB,MAAM,IAAI,CAAC,gBAAgB,CAAC,sBAAsB,CAChD,IAAI,CAAC,MAAM,CAAC,SAAS,EACrB,OAAO,EACP,aAAa,CACd,CAAC;YAEJ,MAAM,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAC7C,eAAe,EACf,aAAa,EACb,SAAS,EACT,OAAO,EACP,cAAc,CACf,CAAC;YAEF,OAAO;gBACL,UAAU,EAAE,cAAc;gBAC1B,aAAa,EAAE;oBACb,OAAO,EAAE,eAAe;oBACxB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;oBAC9C,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;oBAC/B,QAAQ,EAAE;wBACR,QAAQ,EAAE,eAAe,CAAC,QAAQ;wBAClC,iBAAiB,EAAE,eAAe,CAAC,YAAY,CAAC,MAAM;qBACvD;iBACF;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,OAAe;QAMtC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CACxB,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC;YAC9D,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC;YACrC,CAAC,CAAC,CAAC,EACL,cAAc,CACf,CAAC;QACF,MAAM,OAAO,GACX,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC;YAC5D,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;YACzD,CAAC,CAAC,cAAc,CAAC;QACrB,MAAM,eAAe,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnE,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;IACjE,CAAC;IAEO,mBAAmB,CACzB,eAEC,EACD,aAAqB,EACrB,SAAiB,EACjB,OAAe,EACf,cAAsB;QAEtB,OAAO;YACL,gBAAgB,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;YACvC,eAAe,eAAe,CAAC,QAAQ,EAAE;YACzC,WAAW,IAAI,CAAC,GAAG,CAAC,SAAS,GAAG,CAAC,EAAE,cAAc,CAAC,IAAI,OAAO,OAAO,cAAc,EAAE;YACpF,mBAAmB,eAAe,CAAC,YAAY,CAAC,MAAM,EAAE;YACxD,EAAE;YACF,mBAAmB;YACnB,GAAG,eAAe,CAAC,YAAY,CAAC,GAAG,CACjC,CAAC,IAAI,EAAE,EAAE,CACP,KAAK,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,IAAI,EAAE,UAAU,IAAI,CAAC,IAAI,GAAG,CAC5E;YACD,EAAE;YACF,oBAAoB;YACpB,GAAG,eAAe,CAAC,gBAAgB;iBAChC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,mBAAmB,CAAC;iBACvC,GAAG,CACF,CAAC,OAAO,EAAE,EAAE,CACV,UAAU,OAAO,CAAC,IAAI,KAAK,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAChE;YACH,IAAI,CAAC,2BAA2B,CAAC,eAAe,EAAE,aAAa,CAAC;SACjE;aACE,IAAI,EAAE;aACN,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAEO,2BAA2B,CACjC,eAEC,EACD,aAAqB;QAErB,IACE,CAAC,eAAe,CAAC,cAAc;YAC/B,eAAe,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC,EAC3C,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO;YACL,EAAE;YACF,sBAAsB;YACtB,GAAG,eAAe,CAAC,cAAc;iBAC9B,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBACZ,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;gBAC3D,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC;oBAChC,OAAO,KAAK,OAAO,oBAAoB,CAAC;gBAC1C,OAAO;oBACL,KAAK,OAAO,GAAG;oBACf,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CACtB,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,SAAS,IAAI,EAAE,EAAE,CACtD;iBACF,CAAC;YACJ,CAAC,CAAC;iBACD,IAAI,EAAE;SACV,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,KAAc;QACpC,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAExE,IAAI,SAAS,GAAG,aAAa,CAAC,oBAAoB,CAAC;QACnD,IAAI,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC;YACvB,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,QAAQ;oBACX,SAAS,GAAG,aAAa,CAAC,cAAc,CAAC;oBACzC,MAAM;gBACR,KAAK,QAAQ;oBACX,SAAS,GAAG,aAAa,CAAC,iBAAiB,CAAC;oBAC5C,MAAM;gBACR,KAAK,QAAQ;oBACX,SAAS,GAAG,aAAa,CAAC,mBAAmB,CAAC;oBAC9C,MAAM;gBACR,KAAK,QAAQ,CAAC;gBACd,KAAK,QAAQ;oBACX,SAAS,GAAG,aAAa,CAAC,oBAAoB,CAAC;oBAC/C,MAAM;gBACR;oBACE,SAAS,GAAG,aAAa,CAAC,oBAAoB,CAAC;YACnD,CAAC;QACH,CAAC;aAAM,CAAC;YACN,SAAS,GAAG,aAAa,CAAC,OAAO,CAAC;QACpC,CAAC;QAED,OAAO;YACL,UAAU,EAAE,uBAAuB,QAAQ,EAAE;YAC7C,aAAa,EAAE,uBAAuB,QAAQ,EAAE;YAChD,KAAK,EAAE;gBACL,OAAO,EAAE,QAAQ;gBACjB,IAAI,EAAE,SAAS;aAChB;SACF,CAAC;IACJ,CAAC;CACF"}

package/dist/src/tools/ast-edit/constants.d.ts

@@ -0,0 +1,30 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Code keywords used for pattern matching and analysis.
+ */
+export declare const KEYWORDS: {
+    readonly FUNCTION: "function";
+    readonly DEF: "def";
+    readonly CLASS: "class";
+    readonly IF: "if";
+    readonly FOR: "for";
+    readonly WHILE: "while";
+    readonly RETURN: "return";
+    readonly IMPORT: "import ";
+    readonly FROM: "from ";
+};
+/**
+ * Comment prefixes for various languages.
+ */
+export declare const COMMENT_PREFIXES: string[];
+/**
+ * Regex patterns for code analysis.
+ */
+export declare const REGEX: {
+    readonly IMPORT_MODULE: RegExp;
+};
+export { LANGUAGE_MAP, JAVASCRIPT_FAMILY_EXTENSIONS, } from '../../utils/ast-grep-utils.js';

package/dist/src/tools/ast-edit/constants.js

@@ -0,0 +1,36 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+// ===== Code Keywords =====
+/**
+ * Code keywords used for pattern matching and analysis.
+ */
+export const KEYWORDS = {
+    FUNCTION: 'function',
+    DEF: 'def',
+    CLASS: 'class',
+    IF: 'if',
+    FOR: 'for',
+    WHILE: 'while',
+    RETURN: 'return',
+    IMPORT: 'import ',
+    FROM: 'from ',
+};
+// ===== Comment Patterns =====
+/**
+ * Comment prefixes for various languages.
+ */
+export const COMMENT_PREFIXES = ['//', '#', '*', '/*', '*/'];
+// ===== Regex Patterns =====
+/**
+ * Regex patterns for code analysis.
+ */
+export const REGEX = {
+    // eslint-disable-next-line sonarjs/regular-expr -- Static regex reviewed for lint hardening; behavior preserved.
+    IMPORT_MODULE: /(?:import|from)\s+['"]([^'"]+)['"]/,
+};
+// Re-export shared language mapping from utils
+export { LANGUAGE_MAP, JAVASCRIPT_FAMILY_EXTENSIONS, } from '../../utils/ast-grep-utils.js';
+//# sourceMappingURL=constants.js.map

package/dist/src/tools/ast-edit/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../../src/tools/ast-edit/constants.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,4BAA4B;AAC5B;;GAEG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG;IACtB,QAAQ,EAAE,UAAU;IACpB,GAAG,EAAE,KAAK;IACV,KAAK,EAAE,OAAO;IACd,EAAE,EAAE,IAAI;IACR,GAAG,EAAE,KAAK;IACV,KAAK,EAAE,OAAO;IACd,MAAM,EAAE,QAAQ;IAChB,MAAM,EAAE,SAAS;IACjB,IAAI,EAAE,OAAO;CACL,CAAC;AAEX,+BAA+B;AAC/B;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAE7D,6BAA6B;AAC7B;;GAEG;AACH,MAAM,CAAC,MAAM,KAAK,GAAG;IACnB,iHAAiH;IACjH,aAAa,EAAE,oCAAoC;CAC3C,CAAC;AAEX,+CAA+C;AAC/C,OAAO,EACL,YAAY,EACZ,4BAA4B,GAC7B,MAAM,+BAA+B,CAAC"}

package/dist/src/tools/ast-edit/context-collector.d.ts

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Context collector orchestrates all AST context gathering operations.
+ */
+import type { ASTContext, EnhancedASTContext, EnhancedDeclaration } from './types.js';
+/**
+ * Prioritize important symbols from declarations for lazy cross-file lookups.
+ * [CCR] Reason: Prevents querying low-value symbols (like parameters or local vars) to preserve I/O.
+ * @internal Exported for testing only. Not part of public API.
+ */
+export declare function prioritizeSymbolsFromDeclarations(declarations: EnhancedDeclaration[]): string[];
+/**
+ * Orchestrates all AST context gathering: local analysis, working set, and cross-file relationships.
+ */
+export declare class ASTContextCollector {
+    private astExtractor;
+    private repoProvider;
+    private relationshipAnalyzer;
+    constructor();
+    collectContext(filePath: string, content: string): Promise<ASTContext>;
+    collectEnhancedContext(targetFilePath: string, content: string, workspaceRoot: string): Promise<EnhancedASTContext>;
+}

package/dist/src/tools/ast-edit/context-collector.js

@@ -0,0 +1,115 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Context collector orchestrates all AST context gathering operations.
+ */
+import { debugLogger } from '../../utils/debugLogger.js';
+import { ASTConfig } from './ast-config.js';
+import { detectLanguage, extractImports } from './language-analysis.js';
+import { ASTQueryExtractor } from './ast-query-extractor.js';
+import { RepositoryContextProvider } from './repository-context-provider.js';
+import { CrossFileRelationshipAnalyzer, getWorkspaceFiles, } from './cross-file-analyzer.js';
+import { parseAST, collectSnippets, buildLanguageContext, optimizeContextCollection, } from './local-context-analyzer.js';
+import { enrichWithWorkingSetContext } from './workspace-context-provider.js';
+const logger = debugLogger;
+/**
+ * Prioritize important symbols from declarations for lazy cross-file lookups.
+ * [CCR] Reason: Prevents querying low-value symbols (like parameters or local vars) to preserve I/O.
+ * @internal Exported for testing only. Not part of public API.
+ */
+export function prioritizeSymbolsFromDeclarations(declarations) {
+    const scores = new Map();
+    // eslint-disable-next-line sonarjs/too-many-break-or-continue-in-loop -- Existing structure is intentionally preserved; refactoring this boundary is outside the lint slice.
+    for (const decl of declarations) {
+        if (decl.name.length < ASTConfig.MIN_SYMBOL_LENGTH)
+            continue;
+        let score = 0;
+        if (decl.type === 'class')
+            score += 10;
+        if (decl.type === 'function')
+            score += 5;
+        if (score === 0)
+            continue;
+        if (decl.visibility === 'public')
+            score += 3;
+        scores.set(decl.name, (scores.get(decl.name) ?? 0) + score);
+    }
+    return Array.from(scores.entries())
+        .sort((a, b) => b[1] - a[1])
+        .map(([name]) => name)
+        .slice(0, ASTConfig.MAX_RELATED_SYMBOLS);
+}
+/**
+ * Orchestrates all AST context gathering: local analysis, working set, and cross-file relationships.
+ */
+export class ASTContextCollector {
+    astExtractor;
+    repoProvider;
+    relationshipAnalyzer;
+    constructor() {
+        this.astExtractor = new ASTQueryExtractor();
+        this.repoProvider = new RepositoryContextProvider();
+        this.relationshipAnalyzer = new CrossFileRelationshipAnalyzer();
+    }
+    async collectContext(filePath, content) {
+        const language = detectLanguage(filePath);
+        return {
+            filePath,
+            language,
+            fileSize: content.length,
+            astNodes: await parseAST(content, language),
+            declarations: await this.astExtractor.extractDeclarations(filePath, content),
+            imports: extractImports(content, language),
+            relevantSnippets: collectSnippets(content),
+            languageContext: buildLanguageContext(content, language),
+        };
+    }
+    async collectEnhancedContext(targetFilePath, content, workspaceRoot) {
+        const startTime = Date.now();
+        const startMemory = process.memoryUsage().heapUsed;
+        // Base context
+        const baseContext = await this.collectContext(targetFilePath, content);
+        const enhancedContext = {
+            ...baseContext,
+            declarations: baseContext.declarations,
+            connectedFiles: [],
+        };
+        // Context optimization
+        enhancedContext.relevantSnippets = optimizeContextCollection(enhancedContext.declarations, content, workspaceRoot);
+        // Phase 2: Working Set Context (Git-based)
+        const connectedFiles = await enrichWithWorkingSetContext(targetFilePath, workspaceRoot, this.repoProvider, this.astExtractor);
+        enhancedContext.connectedFiles = connectedFiles;
+        // Phase 3: Repository context and Cross-file Relationships
+        const repoContext = await this.repoProvider.collectRepositoryContext(workspaceRoot);
+        enhancedContext.repositoryContext = repoContext ?? undefined;
+        // [CCR] Relation: Cross-file relationship analysis segment.
+        // Reason: Optimized to use on-demand findInFiles instead of eager indexing.
+        if (repoContext) {
+            if (ASTConfig.ENABLE_SYMBOL_INDEXING) {
+                const workspaceFiles = await getWorkspaceFiles(workspaceRoot);
+                await this.relationshipAnalyzer.buildSymbolIndex(workspaceFiles);
+                const relatedFiles = await this.relationshipAnalyzer.findRelatedFiles(targetFilePath);
+                enhancedContext.relatedFiles = relatedFiles;
+            }
+            // Prioritize symbols for Lazy search
+            const topSymbols = prioritizeSymbolsFromDeclarations(enhancedContext.declarations);
+            // Execute atomic queries with strict limits and Survivability (Promise.allSettled)
+            const relatedSymbolsTasks = topSymbols.map((symbol) => this.relationshipAnalyzer.findRelatedSymbols(symbol, workspaceRoot));
+            const relatedSymbolsResults = await Promise.allSettled(relatedSymbolsTasks);
+            enhancedContext.relatedSymbols = relatedSymbolsResults
+                .filter((r) => r.status === 'fulfilled')
+                .map((r) => r.value)
+                .flat();
+        }
+        const duration = Date.now() - startTime;
+        const memoryDelta = (process.memoryUsage().heapUsed - startMemory) / 1024 / 1024;
+        const symbolCount = enhancedContext.relatedSymbols !== undefined
+            ? enhancedContext.relatedSymbols.length
+            : 0;
+        logger.debug(`collectEnhancedContext Metrics: ${duration}ms, Delta: ${memoryDelta.toFixed(2)}MB, Symbols: ${symbolCount}`);
+        return enhancedContext;
+    }
+}
+//# sourceMappingURL=context-collector.js.map

package/dist/src/tools/ast-edit/context-collector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-collector.js","sourceRoot":"","sources":["../../../../src/tools/ast-edit/context-collector.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAOzD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAC7E,OAAO,EACL,6BAA6B,EAC7B,iBAAiB,GAClB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,QAAQ,EACR,eAAe,EACf,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,2BAA2B,EAAE,MAAM,iCAAiC,CAAC;AAE9E,MAAM,MAAM,GAAG,WAAW,CAAC;AAE3B;;;;GAIG;AACH,MAAM,UAAU,iCAAiC,CAC/C,YAAmC;IAEnC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEzC,6KAA6K;IAC7K,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,iBAAiB;YAAE,SAAS;QAE7D,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO;YAAE,KAAK,IAAI,EAAE,CAAC;QACvC,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU;YAAE,KAAK,IAAI,CAAC,CAAC;QACzC,IAAI,KAAK,KAAK,CAAC;YAAE,SAAS;QAC1B,IAAI,IAAI,CAAC,UAAU,KAAK,QAAQ;YAAE,KAAK,IAAI,CAAC,CAAC;QAE7C,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;IAC9D,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;SAChC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;SAC3B,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC;SACrB,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,mBAAmB,CAAC,CAAC;AAC7C,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,mBAAmB;IACtB,YAAY,CAAoB;IAChC,YAAY,CAA4B;IACxC,oBAAoB,CAAgC;IAE5D;QACE,IAAI,CAAC,YAAY,GAAG,IAAI,iBAAiB,EAAE,CAAC;QAC5C,IAAI,CAAC,YAAY,GAAG,IAAI,yBAAyB,EAAE,CAAC;QACpD,IAAI,CAAC,oBAAoB,GAAG,IAAI,6BAA6B,EAAE,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAgB,EAAE,OAAe;QACpD,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;QAE1C,OAAO;YACL,QAAQ;YACR,QAAQ;YACR,QAAQ,EAAE,OAAO,CAAC,MAAM;YACxB,QAAQ,EAAE,MAAM,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC;YAC3C,YAAY,EAAE,MAAM,IAAI,CAAC,YAAY,CAAC,mBAAmB,CACvD,QAAQ,EACR,OAAO,CACR;YACD,OAAO,EAAE,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC;YAC1C,gBAAgB,EAAE,eAAe,CAAC,OAAO,CAAC;YAC1C,eAAe,EAAE,oBAAoB,CAAC,OAAO,EAAE,QAAQ,CAAC;SACzD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,cAAsB,EACtB,OAAe,EACf,aAAqB;QAErB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC;QAEnD,eAAe;QACf,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAEvE,MAAM,eAAe,GAAuB;YAC1C,GAAG,WAAW;YACd,YAAY,EAAE,WAAW,CAAC,YAAqC;YAC/D,cAAc,EAAE,EAAE;SACnB,CAAC;QAEF,uBAAuB;QACvB,eAAe,CAAC,gBAAgB,GAAG,yBAAyB,CAC1D,eAAe,CAAC,YAAY,EAC5B,OAAO,EACP,aAAa,CACd,CAAC;QAEF,2CAA2C;QAC3C,MAAM,cAAc,GAAG,MAAM,2BAA2B,CACtD,cAAc,EACd,aAAa,EACb,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,YAAY,CAClB,CAAC;QACF,eAAe,CAAC,cAAc,GAAG,cAAc,CAAC;QAEhD,2DAA2D;QAC3D,MAAM,WAAW,GACf,MAAM,IAAI,CAAC,YAAY,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC;QAClE,eAAe,CAAC,iBAAiB,GAAG,WAAW,IAAI,SAAS,CAAC;QAE7D,4DAA4D;QAC5D,4EAA4E;QAC5E,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,SAAS,CAAC,sBAAsB,EAAE,CAAC;gBACrC,MAAM,cAAc,GAAG,MAAM,iBAAiB,CAAC,aAAa,CAAC,CAAC;gBAC9D,MAAM,IAAI,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;gBAEjE,MAAM,YAAY,GAChB,MAAM,IAAI,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;gBACnE,eAAe,CAAC,YAAY,GAAG,YAAY,CAAC;YAC9C,CAAC;YAED,qCAAqC;YACrC,MAAM,UAAU,GAAG,iCAAiC,CAClD,eAAe,CAAC,YAAY,CAC7B,CAAC;YAEF,mFAAmF;YACnF,MAAM,mBAAmB,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACpD,IAAI,CAAC,oBAAoB,CAAC,kBAAkB,CAAC,MAAM,EAAE,aAAa,CAAC,CACpE,CAAC;YAEF,MAAM,qBAAqB,GACzB,MAAM,OAAO,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;YAChD,eAAe,CAAC,cAAc,GAAG,qBAAqB;iBACnD,MAAM,CACL,CAAC,CAAC,EAAkD,EAAE,CACpD,CAAC,CAAC,MAAM,KAAK,WAAW,CAC3B;iBACA,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;iBACnB,IAAI,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QACxC,MAAM,WAAW,GACf,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,GAAG,WAAW,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;QAC/D,MAAM,WAAW,GACf,eAAe,CAAC,cAAc,KAAK,SAAS;YAC1C,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,MAAM;YACvC,CAAC,CAAC,CAAC,CAAC;QACR,MAAM,CAAC,KAAK,CACV,mCAAmC,QAAQ,cAAc,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,gBAAgB,WAAW,EAAE,CAC7G,CAAC;QAEF,OAAO,eAAe,CAAC;IACzB,CAAC;CACF"}

package/dist/src/tools/ast-edit/context-optimizer.d.ts

@@ -0,0 +1,29 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Context optimizer for snippet deduplication and context size management.
+ */
+import type { CodeSnippet } from './types.js';
+/**
+ * Handles optimization of code context including snippet management and prompt clipping.
+ */
+export declare class ContextOptimizer {
+    /**
+     * Clip prompt to fit max length limit (corresponds to AST clip_prompt)
+     */
+    static clipPrompt(prompt: string, maxLength: number): string;
+    /**
+     * Check UTF-16 character boundary
+     */
+    private static isCharBoundary;
+    /**
+     * Manage snippets by priority and budget (corresponds to AST snippet collection)
+     */
+    static optimizeSnippets(snippets: CodeSnippet[], maxChars?: number): CodeSnippet[];
+    /**
+     * Truncate overly long snippets
+     */
+    static truncateSnippet(snippet: CodeSnippet): CodeSnippet;
+}