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

package/dist/src/tools/ast-edit/local-context-analyzer.js

@@ -0,0 +1,267 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Local context analysis functions for AST parsing and code snippet collection.
+ */
+import { ASTConfig } from './ast-config.js';
+import { KEYWORDS, COMMENT_PREFIXES } from './constants.js';
+import { ContextOptimizer } from './context-optimizer.js';
+/**
+ * Parse AST from content using basic line-by-line analysis.
+ * @param content - Source code content
+ * @param language - Programming language
+ * @returns Array of AST nodes
+ */
+export async function parseAST(content, language) {
+    if (language === 'unknown') {
+        return [];
+    }
+    // Use existing validateASTSyntax logic for basic parsing
+    return extractASTNodes(content, language);
+}
+/**
+ * Extract AST nodes from content by analyzing significant lines.
+ * @param content - Source code content
+ * @param language - Programming language
+ * @returns Array of AST nodes
+ */
+export function extractASTNodes(content, language) {
+    // Simplified AST node extraction
+    const nodes = [];
+    const lines = content.split('\n');
+    lines.forEach((line, index) => {
+        if (isSignificantLine(line, language)) {
+            nodes.push({
+                type: inferNodeType(line, language),
+                text: line.trim(),
+                startPosition: { line: index + 1, column: 0 },
+                endPosition: { line: index + 1, column: line.length },
+                children: [],
+            });
+        }
+    });
+    return nodes;
+}
+/**
+ * Collect code snippets from content, filtering out comments and short lines.
+ * @param content - Source code content
+ * @returns Array of code snippets sorted by relevance
+ */
+export function collectSnippets(content) {
+    const snippets = [];
+    const lines = content.split('\n');
+    lines.forEach((line, index) => {
+        const trimmed = line.trim();
+        const isComment = COMMENT_PREFIXES.some((prefix) => trimmed.startsWith(prefix));
+        if (trimmed.length > 10 && !isComment) {
+            snippets.push({
+                text: trimmed,
+                relevance: calculateRelevance(trimmed),
+                line: index + 1,
+                source: 'local',
+                priority: 3,
+                charLength: trimmed.length,
+            });
+        }
+    });
+    return snippets
+        .sort((a, b) => b.relevance - a.relevance)
+        .slice(0, ASTConfig.MAX_SNIPPETS);
+}
+/**
+ * Build language-specific context by extracting functions, classes, and variables.
+ * @param content - Source code content
+ * @param language - Programming language
+ * @returns Language context with extracted elements
+ */
+export function buildLanguageContext(content, language) {
+    return {
+        functions: extractFunctions(content, language),
+        classes: extractClasses(content, language),
+        variables: extractVariables(content, language),
+    };
+}
+/**
+ * Check if a line is significant (non-comment, non-empty).
+ * @param line - Source code line
+ * @param _language - Programming language (unused but kept for signature compatibility)
+ * @returns True if the line is significant
+ */
+export function isSignificantLine(line, _language) {
+    const trimmed = line.trim();
+    const isComment = 
+    // eslint-disable-next-line sonarjs/expression-complexity -- Existing structure is intentionally preserved; refactoring this boundary is outside the lint slice.
+    trimmed.startsWith('//') ||
+        trimmed.startsWith('#') ||
+        trimmed.startsWith('*') ||
+        trimmed.startsWith('/*') ||
+        trimmed.startsWith('*/');
+    return trimmed.length > 0 && !isComment;
+}
+/**
+ * Infer the AST node type from a line of code.
+ * @param line - Source code line
+ * @param _language - Programming language (unused but kept for signature compatibility)
+ * @returns Node type (function, class, control, return, statement)
+ */
+export function inferNodeType(line, _language) {
+    const trimmed = line.trim();
+    if (trimmed.includes(KEYWORDS.FUNCTION) || trimmed.includes(KEYWORDS.DEF))
+        return 'function';
+    if (trimmed.includes(KEYWORDS.CLASS))
+        return 'class';
+    if (trimmed.includes(KEYWORDS.IF) ||
+        trimmed.includes(KEYWORDS.FOR) ||
+        trimmed.includes(KEYWORDS.WHILE))
+        return 'control';
+    if (trimmed.includes(KEYWORDS.RETURN))
+        return 'return';
+    return 'statement';
+}
+/**
+ * Calculate the relevance score of a code line.
+ * @param line - Source code line
+ * @returns Relevance score
+ */
+export function calculateRelevance(line) {
+    let relevance = 1;
+    if (line.includes(KEYWORDS.FUNCTION) || line.includes(KEYWORDS.DEF))
+        relevance += 3;
+    if (line.includes(KEYWORDS.CLASS))
+        relevance += 2;
+    if (line.includes(KEYWORDS.RETURN))
+        relevance += 1;
+    if (line.length > 50)
+        relevance += 1;
+    return relevance;
+}
+/**
+ * Extract function declarations from content using regex.
+ * @param content - Source code content
+ * @param _language - Programming language
+ * @returns Array of function information
+ */
+export function extractFunctions(content, _language) {
+    const functions = [];
+    const lines = content.split('\n');
+    lines.forEach((line, index) => {
+        const trimmed = line.trim();
+        if (_language === 'typescript' || _language === 'javascript') {
+            const match = trimmed.match(
+            // eslint-disable-next-line sonarjs/regular-expr -- Static regex reviewed for lint hardening; behavior preserved.
+            /function\s+(\w+)\s*\(([^)]*)\)(?::\s*(\w+))?/);
+            if (match) {
+                functions.push({
+                    name: match[1],
+                    parameters: match[2]
+                        .split(',')
+                        .map((p) => p.trim())
+                        .filter((p) => p),
+                    returnType: match[3] || 'unknown',
+                    line: index + 1,
+                });
+            }
+        }
+        else if (_language === 'python') {
+            const match = trimmed.match(
+            // eslint-disable-next-line sonarjs/regular-expr -- Static regex reviewed for lint hardening; behavior preserved.
+            /def\s+(\w+)\s*\(([^)]*)\)\s*(?:->\s*(\w+))?/);
+            if (match) {
+                functions.push({
+                    name: match[1],
+                    parameters: match[2]
+                        .split(',')
+                        .map((p) => p.trim())
+                        .filter((p) => p),
+                    returnType: match[3] || 'unknown',
+                    line: index + 1,
+                });
+            }
+        }
+    });
+    return functions;
+}
+/**
+ * Extract class declarations from content using regex.
+ * @param content - Source code content
+ * @param _language - Programming language
+ * @returns Array of class information
+ */
+export function extractClasses(content, _language) {
+    const classes = [];
+    const lines = content.split('\n');
+    lines.forEach((line, index) => {
+        const trimmed = line.trim();
+        if (trimmed.includes(KEYWORDS.CLASS)) {
+            // eslint-disable-next-line sonarjs/regular-expr -- Static regex reviewed for lint hardening; behavior preserved.
+            const match = trimmed.match(/class\s+(\w+)/);
+            if (match) {
+                classes.push({
+                    name: match[1],
+                    methods: [], // Simplified implementation
+                    properties: [], // Simplified implementation
+                    line: index + 1,
+                });
+            }
+        }
+    });
+    return classes;
+}
+/**
+ * Extract variable declarations from content using regex.
+ * @param content - Source code content
+ * @param _language - Programming language
+ * @returns Array of variable information
+ */
+export function extractVariables(content, _language) {
+    const variables = [];
+    const lines = content.split('\n');
+    lines.forEach((line, index) => {
+        const trimmed = line.trim();
+        if (_language === 'typescript' || _language === 'javascript') {
+            // eslint-disable-next-line sonarjs/regular-expr -- Static regex reviewed for lint hardening; behavior preserved.
+            const match = trimmed.match(/(?:const|let|var)\s+(\w+)\s*:\s*(\w+)/);
+            if (match) {
+                variables.push({
+                    name: match[1],
+                    type: match[2],
+                    line: index + 1,
+                });
+            }
+        }
+    });
+    return variables;
+}
+/**
+ * Optimize context collection by gathering declaration and local snippets.
+ * @param declarations - Array of declarations
+ * @param content - Source code content
+ * @param _workspaceRoot - Workspace root path (unused but kept for signature compatibility)
+ * @returns Optimized array of code snippets
+ */
+export function optimizeContextCollection(declarations, content, _workspaceRoot) {
+    const allSnippets = [];
+    // Collect declaration snippets (highest priority)
+    for (const decl of declarations) {
+        allSnippets.push({
+            text: `${decl.type}: ${decl.name}`,
+            relevance: 5,
+            line: decl.line,
+            source: 'declaration',
+            priority: 1,
+            charLength: decl.name.length + decl.type.length + 2,
+        });
+    }
+    // Collect local snippets
+    const localSnippets = collectSnippets(content);
+    allSnippets.push(...localSnippets.map((snippet) => ({
+        ...snippet,
+        source: 'local',
+        priority: 2,
+        charLength: snippet.text.length,
+    })));
+    return ContextOptimizer.optimizeSnippets(allSnippets);
+}
+//# sourceMappingURL=local-context-analyzer.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"local-context-analyzer.js","sourceRoot":"","sources":["../../../../src/tools/ast-edit/local-context-analyzer.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAWH,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE1D;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,OAAe,EACf,QAAgB;IAEhB,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,yDAAyD;IACzD,OAAO,eAAe,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;AAC5C,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,eAAe,CAAC,OAAe,EAAE,QAAgB;IAC/D,iCAAiC;IACjC,MAAM,KAAK,GAAc,EAAE,CAAC;IAC5B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QAC5B,IAAI,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,CAAC;YACtC,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,aAAa,CAAC,IAAI,EAAE,QAAQ,CAAC;gBACnC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE;gBACjB,aAAa,EAAE,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;gBAC7C,WAAW,EAAE,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE;gBACrD,QAAQ,EAAE,EAAE;aACb,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,eAAe,CAAC,OAAe;IAC7C,MAAM,QAAQ,GAAkB,EAAE,CAAC;IACnC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CACjD,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAC3B,CAAC;QACF,IAAI,OAAO,CAAC,MAAM,GAAG,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,OAAO;gBACb,SAAS,EAAE,kBAAkB,CAAC,OAAO,CAAC;gBACtC,IAAI,EAAE,KAAK,GAAG,CAAC;gBACf,MAAM,EAAE,OAAO;gBACf,QAAQ,EAAE,CAAC;gBACX,UAAU,EAAE,OAAO,CAAC,MAAM;aAC3B,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,QAAQ;SACZ,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC;SACzC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC;AACtC,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,oBAAoB,CAClC,OAAe,EACf,QAAgB;IAEhB,OAAO;QACL,SAAS,EAAE,gBAAgB,CAAC,OAAO,EAAE,QAAQ,CAAC;QAC9C,OAAO,EAAE,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC;QAC1C,SAAS,EAAE,gBAAgB,CAAC,OAAO,EAAE,QAAQ,CAAC;KAC/C,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,iBAAiB,CAAC,IAAY,EAAE,SAAiB;IAC/D,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,MAAM,SAAS;IACb,gKAAgK;IAChK,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;QACxB,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;QACvB,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;QACvB,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;QACxB,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAC3B,OAAO,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;AAC1C,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,aAAa,CAAC,IAAY,EAAE,SAAiB;IAC3D,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC;QACvE,OAAO,UAAU,CAAC;IACpB,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IACrD,IACE,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC7B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC;QAC9B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;QAEhC,OAAO,SAAS,CAAC;IACnB,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,QAAQ,CAAC;IACvD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,IAAY;IAC7C,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC;QACjE,SAAS,IAAI,CAAC,CAAC;IACjB,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,SAAS,IAAI,CAAC,CAAC;IAClD,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,SAAS,IAAI,CAAC,CAAC;IACnD,IAAI,IAAI,CAAC,MAAM,GAAG,EAAE;QAAE,SAAS,IAAI,CAAC,CAAC;IACrC,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAC9B,OAAe,EACf,SAAiB;IAEjB,MAAM,SAAS,GAAmB,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,SAAS,KAAK,YAAY,IAAI,SAAS,KAAK,YAAY,EAAE,CAAC;YAC7D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK;YACzB,iHAAiH;YACjH,8CAA8C,CAC/C,CAAC;YACF,IAAI,KAAK,EAAE,CAAC;gBACV,SAAS,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;oBACd,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC;yBACjB,KAAK,CAAC,GAAG,CAAC;yBACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;yBACpB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;oBACnB,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,SAAS;oBACjC,IAAI,EAAE,KAAK,GAAG,CAAC;iBAChB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;aAAM,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;YAClC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK;YACzB,iHAAiH;YACjH,6CAA6C,CAC9C,CAAC;YACF,IAAI,KAAK,EAAE,CAAC;gBACV,SAAS,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;oBACd,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC;yBACjB,KAAK,CAAC,GAAG,CAAC;yBACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;yBACpB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;oBACnB,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,SAAS;oBACjC,IAAI,EAAE,KAAK,GAAG,CAAC;iBAChB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAC5B,OAAe,EACf,SAAiB;IAEjB,MAAM,OAAO,GAAgB,EAAE,CAAC;IAChC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACrC,iHAAiH;YACjH,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;YAC7C,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;oBACd,OAAO,EAAE,EAAE,EAAE,4BAA4B;oBACzC,UAAU,EAAE,EAAE,EAAE,4BAA4B;oBAC5C,IAAI,EAAE,KAAK,GAAG,CAAC;iBAChB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAC9B,OAAe,EACf,SAAiB;IAEjB,MAAM,SAAS,GAAmB,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,SAAS,KAAK,YAAY,IAAI,SAAS,KAAK,YAAY,EAAE,CAAC;YAC7D,iHAAiH;YACjH,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;YACrE,IAAI,KAAK,EAAE,CAAC;gBACV,SAAS,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;oBACd,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;oBACd,IAAI,EAAE,KAAK,GAAG,CAAC;iBAChB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,yBAAyB,CACvC,YAA2B,EAC3B,OAAe,EACf,cAAsB;IAEtB,MAAM,WAAW,GAAkB,EAAE,CAAC;IAEtC,kDAAkD;IAClD,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,WAAW,CAAC,IAAI,CAAC;YACf,IAAI,EAAE,GAAG,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,EAAE;YAClC,SAAS,EAAE,CAAC;YACZ,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,aAAa;YACrB,QAAQ,EAAE,CAAC;YACX,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC;SACpD,CAAC,CAAC;IACL,CAAC;IAED,yBAAyB;IACzB,MAAM,aAAa,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IAC/C,WAAW,CAAC,IAAI,CACd,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QACjC,GAAG,OAAO;QACV,MAAM,EAAE,OAAgB;QACxB,QAAQ,EAAE,CAAC;QACX,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM;KAChC,CAAC,CAAC,CACJ,CAAC;IAEF,OAAO,gBAAgB,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;AACxD,CAAC"}

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

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { RepositoryContext } from './types.js';
+/**
+ * RepositoryContextProvider handles git operations to collect repository context.
+ */
+export declare class RepositoryContextProvider {
+    collectRepositoryContext(rootPath: string): Promise<RepositoryContext | null>;
+    /**
+     * Get the "Working Set" of files:
+     * 1. Unstaged changes (git diff --name-only)
+     * 2. Staged changes (git diff --name-only --cached)
+     * 3. Recent commits (git log -n <limit> --name-only)
+     */
+    getWorkingSetFiles(workspaceRoot: string, limit?: number): Promise<string[]>;
+    private getGitRemoteUrl;
+    private getCurrentCommit;
+    private getCurrentBranch;
+}

package/dist/src/tools/ast-edit/repository-context-provider.js

@@ -0,0 +1,139 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { promises as fsPromises } from 'fs';
+import * as path from 'path';
+import { spawnSync } from 'child_process';
+const GIT_TIMEOUT_MS = 3000;
+const GIT_MAX_BUFFER = 1024 * 1024;
+/**
+ * RepositoryContextProvider handles git operations to collect repository context.
+ */
+export class RepositoryContextProvider {
+    async collectRepositoryContext(rootPath) {
+        try {
+            const gitUrl = await this.getGitRemoteUrl(rootPath);
+            const commitSha = await this.getCurrentCommit(rootPath);
+            const branch = await this.getCurrentBranch(rootPath);
+            if (!gitUrl && !commitSha) {
+                return null; // Not a git repo or failed to get info
+            }
+            return {
+                // eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- intentional falsy coalescing: git command output fallback
+                gitUrl: gitUrl || 'unknown',
+                // eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- intentional falsy coalescing: git command output fallback
+                commitSha: commitSha || 'unknown',
+                // eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- intentional falsy coalescing: git command output fallback
+                branch: branch || 'unknown',
+                rootPath,
+            };
+        }
+        catch {
+            // Git info unavailable; not a git repo.
+            return null;
+        }
+    }
+    /**
+     * Get the "Working Set" of files:
+     * 1. Unstaged changes (git diff --name-only)
+     * 2. Staged changes (git diff --name-only --cached)
+     * 3. Recent commits (git log -n <limit> --name-only)
+     */
+    async getWorkingSetFiles(workspaceRoot, limit = 5) {
+        const files = new Set();
+        try {
+            const execGit = (args) => {
+                // eslint-disable-next-line sonarjs/no-os-command-from-path -- Project intentionally invokes platform tooling at this trusted boundary; arguments remain explicit and behavior is preserved.
+                const result = spawnSync('git', ['-C', workspaceRoot, ...args], {
+                    encoding: 'utf-8',
+                    stdio: 'pipe',
+                    timeout: GIT_TIMEOUT_MS,
+                    maxBuffer: GIT_MAX_BUFFER,
+                });
+                return result.status === 0 ? result.stdout.trim() : '';
+            };
+            // 1. Unstaged changes
+            execGit(['diff', '--name-only', '-z'])
+                .split('\0')
+                .forEach((f) => f && files.add(f));
+            // 2. Staged changes
+            execGit(['diff', '--name-only', '--cached', '-z'])
+                .split('\0')
+                .forEach((f) => f && files.add(f));
+            // 3. Recent commits
+            // Note: -z works with --name-only in log but we need to ensure format doesn't break it.
+            // Safest is to rely on diffs for working set, but strictly following plan:
+            execGit(['log', `-n${limit}`, '--name-only', '--format=', '-z'])
+                .split('\0')
+                .forEach((f) => f && files.add(f));
+        }
+        catch {
+            // Git command failed; return what we have.
+        }
+        // Filter existing files and convert to absolute paths
+        const validFiles = [];
+        for (const file of files) {
+            if (!file.trim())
+                continue;
+            const absPath = path.resolve(workspaceRoot, file);
+            try {
+                await fsPromises.access(absPath);
+                validFiles.push(absPath);
+            }
+            catch {
+                // File might be deleted
+            }
+        }
+        return validFiles;
+    }
+    async getGitRemoteUrl(repoPath) {
+        try {
+            const result = spawnSync(
+            // eslint-disable-next-line sonarjs/no-os-command-from-path -- Project intentionally invokes platform tooling at this trusted boundary; arguments remain explicit and behavior is preserved.
+            'git', ['-C', repoPath, 'remote', 'get-url', 'origin'], {
+                encoding: 'utf-8',
+                stdio: 'pipe',
+                timeout: GIT_TIMEOUT_MS,
+                maxBuffer: GIT_MAX_BUFFER,
+            });
+            return result.status === 0 ? result.stdout.trim() : null;
+        }
+        catch {
+            return null;
+        }
+    }
+    async getCurrentCommit(repoPath) {
+        try {
+            // eslint-disable-next-line sonarjs/no-os-command-from-path -- Project intentionally invokes platform tooling at this trusted boundary; arguments remain explicit and behavior is preserved.
+            const result = spawnSync('git', ['-C', repoPath, 'rev-parse', 'HEAD'], {
+                encoding: 'utf-8',
+                stdio: 'pipe',
+                timeout: GIT_TIMEOUT_MS,
+                maxBuffer: GIT_MAX_BUFFER,
+            });
+            return result.status === 0 ? result.stdout.trim() : null;
+        }
+        catch {
+            return null;
+        }
+    }
+    async getCurrentBranch(repoPath) {
+        try {
+            const result = spawnSync(
+            // eslint-disable-next-line sonarjs/no-os-command-from-path -- Project intentionally invokes platform tooling at this trusted boundary; arguments remain explicit and behavior is preserved.
+            'git', ['-C', repoPath, 'branch', '--show-current'], {
+                encoding: 'utf-8',
+                stdio: 'pipe',
+                timeout: GIT_TIMEOUT_MS,
+                maxBuffer: GIT_MAX_BUFFER,
+            });
+            return result.status === 0 ? result.stdout.trim() : null;
+        }
+        catch {
+            return null;
+        }
+    }
+}
+//# sourceMappingURL=repository-context-provider.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"repository-context-provider.js","sourceRoot":"","sources":["../../../../src/tools/ast-edit/repository-context-provider.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,IAAI,UAAU,EAAE,MAAM,IAAI,CAAC;AAC5C,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAG1C,MAAM,cAAc,GAAG,IAAI,CAAC;AAC5B,MAAM,cAAc,GAAG,IAAI,GAAG,IAAI,CAAC;AAEnC;;GAEG;AACH,MAAM,OAAO,yBAAyB;IACpC,KAAK,CAAC,wBAAwB,CAC5B,QAAgB;QAEhB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YACpD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YACxD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAErD,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC1B,OAAO,IAAI,CAAC,CAAC,uCAAuC;YACtD,CAAC;YAED,OAAO;gBACL,qIAAqI;gBACrI,MAAM,EAAE,MAAM,IAAI,SAAS;gBAC3B,qIAAqI;gBACrI,SAAS,EAAE,SAAS,IAAI,SAAS;gBACjC,qIAAqI;gBACrI,MAAM,EAAE,MAAM,IAAI,SAAS;gBAC3B,QAAQ;aACT,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,wCAAwC;YACxC,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,kBAAkB,CACtB,aAAqB,EACrB,QAAgB,CAAC;QAEjB,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;QAEhC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,CAAC,IAAc,EAAE,EAAE;gBACjC,4LAA4L;gBAC5L,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC,EAAE;oBAC9D,QAAQ,EAAE,OAAO;oBACjB,KAAK,EAAE,MAAM;oBACb,OAAO,EAAE,cAAc;oBACvB,SAAS,EAAE,cAAc;iBAC1B,CAAC,CAAC;gBACH,OAAO,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzD,CAAC,CAAC;YAEF,sBAAsB;YACtB,OAAO,CAAC,CAAC,MAAM,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;iBACnC,KAAK,CAAC,IAAI,CAAC;iBACX,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAErC,oBAAoB;YACpB,OAAO,CAAC,CAAC,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;iBAC/C,KAAK,CAAC,IAAI,CAAC;iBACX,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAErC,oBAAoB;YACpB,wFAAwF;YACxF,2EAA2E;YAC3E,OAAO,CAAC,CAAC,KAAK,EAAE,KAAK,KAAK,EAAE,EAAE,aAAa,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;iBAC7D,KAAK,CAAC,IAAI,CAAC;iBACX,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC;QAAC,MAAM,CAAC;YACP,2CAA2C;QAC7C,CAAC;QAED,sDAAsD;QACtD,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAAE,SAAS;YAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;YAClD,IAAI,CAAC;gBACH,MAAM,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBACjC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;YAAC,MAAM,CAAC;gBACP,wBAAwB;YAC1B,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,QAAgB;QAC5C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS;YACtB,4LAA4L;YAC5L,KAAK,EACL,CAAC,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,CAAC,EAC/C;gBACE,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,MAAM;gBACb,OAAO,EAAE,cAAc;gBACvB,SAAS,EAAE,cAAc;aAC1B,CACF,CAAC;YACF,OAAO,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QAC3D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,QAAgB;QAC7C,IAAI,CAAC;YACH,4LAA4L;YAC5L,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,CAAC,EAAE;gBACrE,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,MAAM;gBACb,OAAO,EAAE,cAAc;gBACvB,SAAS,EAAE,cAAc;aAC1B,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QAC3D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,QAAgB;QAC7C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS;YACtB,4LAA4L;YAC5L,KAAK,EACL,CAAC,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,gBAAgB,CAAC,EAC5C;gBACE,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,MAAM;gBACb,OAAO,EAAE,cAAc;gBACvB,SAAS,EAAE,cAAc;aAC1B,CACF,CAAC;YACF,OAAO,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QAC3D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;CACF"}

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

@@ -0,0 +1,155 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export interface ASTContext {
+    filePath: string;
+    language: string;
+    fileSize: number;
+    astNodes: ASTNode[];
+    declarations: Declaration[];
+    imports: Import[];
+    relevantSnippets: CodeSnippet[];
+    languageContext: {
+        functions: FunctionInfo[];
+        classes: ClassInfo[];
+        variables: VariableInfo[];
+    };
+}
+export interface ASTNode {
+    type: string;
+    text: string;
+    startPosition: Position;
+    endPosition: Position;
+    children: ASTNode[];
+}
+export interface Declaration {
+    name: string;
+    type: 'function' | 'class' | 'variable' | 'import';
+    line: number;
+    column: number;
+    signature?: string;
+}
+export interface CodeSnippet {
+    text: string;
+    relevance: number;
+    line: number;
+    source: 'declaration' | 'changed_file' | 'recent_file' | 'search' | 'local';
+    priority: number;
+    charLength: number;
+}
+export interface Import {
+    module: string;
+    items: string[];
+    line: number;
+}
+export interface FunctionInfo {
+    name: string;
+    parameters: string[];
+    returnType: string;
+    line: number;
+}
+export interface ClassInfo {
+    name: string;
+    methods: string[];
+    properties: string[];
+    line: number;
+}
+export interface VariableInfo {
+    name: string;
+    type: string;
+    line: number;
+}
+export interface Position {
+    line: number;
+    column: number;
+}
+export interface SgNode {
+    range(): {
+        start: Position;
+        end: Position;
+    };
+}
+export interface RepositoryContext {
+    gitUrl: string;
+    commitSha: string;
+    branch: string;
+    rootPath: string;
+}
+export interface SymbolReference {
+    type: 'definition' | 'reference' | 'import';
+    filePath: string;
+    line: number;
+    column: number;
+    sourceModule?: string;
+}
+export interface FileContext {
+    filePath: string;
+    declarations: EnhancedDeclaration[];
+    summary: string;
+}
+export interface CrossFileContext {
+    files: FileContext[];
+}
+export interface ConnectedFile {
+    filePath: string;
+    declarations: EnhancedDeclaration[];
+}
+export interface EnhancedDeclaration extends Declaration {
+    range: {
+        start: Position;
+        end: Position;
+    };
+    documentation?: string;
+    visibility?: 'public' | 'private' | 'protected';
+    signature?: string;
+}
+export interface EnhancedASTContext extends ASTContext {
+    declarations: EnhancedDeclaration[];
+    repositoryContext?: RepositoryContext;
+    relatedFiles?: string[];
+    relatedSymbols?: SymbolReference[];
+    crossFileContext?: CrossFileContext;
+    connectedFiles?: ConnectedFile[];
+}
+export interface ASTEditToolParams {
+    /**
+     * The absolute path to the file to modify
+     */
+    file_path: string;
+    /**
+     * The text to replace
+     */
+    old_string: string;
+    /**
+     * The text to replace it with
+     */
+    new_string: string;
+    /**
+     * Force execution after preview. Default is false.
+     * IMPORTANT: This tool ALWAYS operates in two steps:
+     * 1. First call: Preview changes (force: false or omitted)
+     * 2. Second call: Apply changes (force: true)
+     */
+    force?: boolean;
+    /**
+     * Timestamp (ms) of the file when last read.
+     * If provided, the tool will verify the file hasn't been modified since this time.
+     */
+    last_modified?: number;
+}
+export interface ASTReadFileToolParams {
+    /**
+     * The absolute path to the file to read
+     */
+    file_path: string;
+    /**
+     * The line number to start reading from (optional)
+     */
+    offset?: number;
+    /**
+     * The number of lines to read (optional)
+     */
+    limit?: number;
+}

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

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};
+//# sourceMappingURL=types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/tools/ast-edit/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG"}

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

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Workspace context provider for enriching context with working-set files.
+ */
+import type { ConnectedFile } from './types.js';
+import type { ASTQueryExtractor } from './ast-query-extractor.js';
+import type { RepositoryContextProvider } from './repository-context-provider.js';
+/**
+ * Enrich context with declarations from working-set files.
+ * Gets the current working set from git (unstaged/staged/recent commits),
+ * reads each file, and extracts declarations for a skeleton view.
+ *
+ * @param targetFilePath - The file currently being edited
+ * @param workspaceRoot - The workspace root directory
+ * @param repoProvider - Repository context provider for git operations
+ * @param astExtractor - AST query extractor for declaration extraction
+ * @returns Array of connected files with their declarations
+ */
+export declare function enrichWithWorkingSetContext(targetFilePath: string, workspaceRoot: string, repoProvider: RepositoryContextProvider, astExtractor: ASTQueryExtractor): Promise<ConnectedFile[]>;

package/dist/src/tools/ast-edit/workspace-context-provider.js

@@ -0,0 +1,39 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Workspace context provider for enriching context with working-set files.
+ */
+import { promises as fsPromises } from 'fs';
+/**
+ * Enrich context with declarations from working-set files.
+ * Gets the current working set from git (unstaged/staged/recent commits),
+ * reads each file, and extracts declarations for a skeleton view.
+ *
+ * @param targetFilePath - The file currently being edited
+ * @param workspaceRoot - The workspace root directory
+ * @param repoProvider - Repository context provider for git operations
+ * @param astExtractor - AST query extractor for declaration extraction
+ * @returns Array of connected files with their declarations
+ */
+export async function enrichWithWorkingSetContext(targetFilePath, workspaceRoot, repoProvider, astExtractor) {
+    const connectedFiles = [];
+    // Phase 2: Working Set Context (Git-based)
+    // Replace BM25 search with working set file declarations
+    const workingSetFiles = await repoProvider.getWorkingSetFiles(workspaceRoot);
+    // Filter out current file
+    const otherFiles = workingSetFiles.filter((f) => f !== targetFilePath);
+    const settled = await Promise.allSettled(otherFiles.map(async (filePath) => {
+        const fileContent = await fsPromises.readFile(filePath, 'utf-8');
+        const declarations = await astExtractor.extractDeclarations(filePath, fileContent);
+        return { filePath, declarations };
+    }));
+    for (const item of settled) {
+        if (item.status === 'fulfilled') {
+            connectedFiles.push(item.value);
+        }
+    }
+    return connectedFiles;
+}
+//# sourceMappingURL=workspace-context-provider.js.map