lance-context 0.1.0 → 1.1.0

package/dist/symbols/reference-finder.js

@@ -0,0 +1,376 @@
+/**
+ * Reference finder for locating all usages of a symbol.
+ * Uses Tree-sitter to find identifier references across the codebase.
+ */
+import * as fs from 'fs/promises';
+import * as path from 'path';
+import { glob } from 'glob';
+import * as ts from 'typescript';
+import { SymbolKind } from './types.js';
+import { SymbolExtractor } from './symbol-extractor.js';
+import { getSymbolName, parseNamePath, matchNamePath } from './name-path.js';
+let parserModule = null;
+const loadParserModule = async () => {
+    if (!parserModule) {
+        const mod = await import('web-tree-sitter');
+        parserModule = {
+            Parser: mod.Parser,
+            Language: mod.Language,
+        };
+    }
+    return parserModule;
+};
+const LANGUAGE_CONFIGS = {
+    python: {
+        wasmFile: 'tree-sitter-python.wasm',
+        extensions: ['.py', '.pyi'],
+        identifierTypes: ['identifier'],
+    },
+    go: {
+        wasmFile: 'tree-sitter-go.wasm',
+        extensions: ['.go'],
+        identifierTypes: ['identifier', 'type_identifier', 'field_identifier'],
+    },
+    rust: {
+        wasmFile: 'tree-sitter-rust.wasm',
+        extensions: ['.rs'],
+        identifierTypes: ['identifier', 'type_identifier', 'field_identifier'],
+    },
+    java: {
+        wasmFile: 'tree-sitter-java.wasm',
+        extensions: ['.java'],
+        identifierTypes: ['identifier', 'type_identifier'],
+    },
+    ruby: {
+        wasmFile: 'tree-sitter-ruby.wasm',
+        extensions: ['.rb'],
+        identifierTypes: ['identifier', 'constant'],
+    },
+};
+/**
+ * Default patterns for files to exclude when searching
+ */
+const DEFAULT_EXCLUDE_PATTERNS = [
+    'node_modules/**',
+    '.git/**',
+    'dist/**',
+    'build/**',
+    'coverage/**',
+    '__pycache__/**',
+    '.next/**',
+    '.cache/**',
+];
+/**
+ * Find references to a symbol across the codebase.
+ */
+export class ReferenceFinder {
+    static parser = null;
+    static loadedLanguages = new Map();
+    static initPromise = null;
+    static wasmBasePath = null;
+    projectPath;
+    symbolExtractor;
+    constructor(projectPath) {
+        this.projectPath = projectPath;
+        this.symbolExtractor = new SymbolExtractor(projectPath);
+    }
+    /**
+     * Initialize the tree-sitter parser.
+     */
+    static async initialize() {
+        if (this.initPromise) {
+            return this.initPromise;
+        }
+        this.initPromise = (async () => {
+            this.wasmBasePath = await this.findWasmBasePath();
+            const module = await loadParserModule();
+            const webTreeSitterWasm = path.join(process.cwd(), 'node_modules', 'web-tree-sitter', 'web-tree-sitter.wasm');
+            await module.Parser.init({
+                locateFile: () => webTreeSitterWasm,
+            });
+            this.parser = new module.Parser();
+        })();
+        return this.initPromise;
+    }
+    static async findWasmBasePath() {
+        const { fileURLToPath } = await import('url');
+        const possiblePaths = [
+            path.join(process.cwd(), 'node_modules', '@vscode', 'tree-sitter-wasm', 'wasm'),
+            path.join(fileURLToPath(import.meta.url), '..', '..', '..', 'node_modules', '@vscode', 'tree-sitter-wasm', 'wasm'),
+        ];
+        for (const p of possiblePaths) {
+            try {
+                await fs.access(p);
+                return p;
+            }
+            catch {
+                // Try next
+            }
+        }
+        throw new Error('Could not find @vscode/tree-sitter-wasm package.');
+    }
+    static getLanguageConfig(filepath) {
+        const ext = path.extname(filepath).toLowerCase();
+        for (const [, config] of Object.entries(LANGUAGE_CONFIGS)) {
+            if (config.extensions.includes(ext)) {
+                return config;
+            }
+        }
+        return null;
+    }
+    static async loadLanguage(config) {
+        const cached = this.loadedLanguages.get(config.wasmFile);
+        if (cached) {
+            return cached;
+        }
+        if (!this.wasmBasePath) {
+            throw new Error('ReferenceFinder not initialized.');
+        }
+        const module = await loadParserModule();
+        const wasmPath = path.join(this.wasmBasePath, config.wasmFile);
+        const wasmBytes = await fs.readFile(wasmPath);
+        const language = await module.Language.load(wasmBytes);
+        this.loadedLanguages.set(config.wasmFile, language);
+        return language;
+    }
+    /**
+     * Find all references to a symbol.
+     */
+    async findReferences(options) {
+        const { namePath, relativePath, includeInfo = false, includeKinds, excludeKinds } = options;
+        // First, find the target symbol
+        const fullPath = path.join(this.projectPath, relativePath);
+        const symbols = await this.symbolExtractor.extractSymbols(fullPath, false);
+        const pattern = parseNamePath(namePath);
+        const targetSymbol = this.findSymbolByPattern(symbols, pattern);
+        if (!targetSymbol) {
+            throw new Error(`Symbol not found: ${namePath} in ${relativePath}`);
+        }
+        const symbolName = getSymbolName(targetSymbol.namePath);
+        // Find all files to search
+        const files = await this.findFilesToSearch();
+        const references = [];
+        for (const file of files) {
+            const fileRefs = await this.findReferencesInFile(file, symbolName, targetSymbol, includeInfo, includeKinds, excludeKinds);
+            references.push(...fileRefs);
+        }
+        return references;
+    }
+    findSymbolByPattern(symbols, pattern) {
+        for (const symbol of symbols) {
+            if (matchNamePath(symbol.namePath, pattern)) {
+                return symbol;
+            }
+            if (symbol.children) {
+                const found = this.findSymbolByPattern(symbol.children, pattern);
+                if (found) {
+                    return found;
+                }
+            }
+        }
+        return null;
+    }
+    async findFilesToSearch() {
+        const codeExtensions = [
+            '.ts',
+            '.tsx',
+            '.js',
+            '.jsx',
+            '.mts',
+            '.cts',
+            '.mjs',
+            '.cjs',
+            '.py',
+            '.pyi',
+            '.go',
+            '.rs',
+            '.java',
+            '.kt',
+            '.rb',
+        ];
+        const patterns = codeExtensions.map((ext) => `**/*${ext}`);
+        const files = [];
+        for (const pattern of patterns) {
+            const matches = await glob(pattern, {
+                cwd: this.projectPath,
+                ignore: DEFAULT_EXCLUDE_PATTERNS,
+                absolute: true,
+            });
+            files.push(...matches);
+        }
+        return [...new Set(files)];
+    }
+    async findReferencesInFile(filepath, symbolName, targetSymbol, includeInfo, includeKinds, excludeKinds) {
+        const ext = path.extname(filepath).toLowerCase();
+        const relativePath = path.relative(this.projectPath, filepath);
+        const references = [];
+        try {
+            const content = await fs.readFile(filepath, 'utf-8');
+            const lines = content.split('\n');
+            // Skip the definition file for non-exported symbols
+            if (path.relative(this.projectPath, filepath) === targetSymbol.location.filepath) {
+                // Still find references within the same file, but skip the definition itself
+            }
+            // Use TypeScript for TS/JS files
+            if (['.ts', '.tsx', '.js', '.jsx', '.mts', '.cts', '.mjs', '.cjs'].includes(ext)) {
+                const refs = this.findReferencesWithTypeScript(content, filepath, relativePath, lines, symbolName, targetSymbol);
+                references.push(...refs);
+            }
+            else {
+                // Use tree-sitter for other languages
+                const config = ReferenceFinder.getLanguageConfig(filepath);
+                if (config) {
+                    const refs = await this.findReferencesWithTreeSitter(content, filepath, relativePath, lines, symbolName, targetSymbol, config);
+                    references.push(...refs);
+                }
+            }
+            // Filter by kinds
+            return references.filter((ref) => {
+                if (excludeKinds && excludeKinds.includes(ref.referencingSymbol.kind)) {
+                    return false;
+                }
+                if (includeKinds && !includeKinds.includes(ref.referencingSymbol.kind)) {
+                    return false;
+                }
+                return true;
+            });
+        }
+        catch {
+            // Skip files that can't be read
+            return [];
+        }
+    }
+    findReferencesWithTypeScript(content, _filepath, relativePath, lines, symbolName, targetSymbol) {
+        const references = [];
+        const ext = path.extname(relativePath).toLowerCase();
+        let scriptKind = ts.ScriptKind.TS;
+        if (ext === '.tsx' || ext === '.jsx') {
+            scriptKind = ts.ScriptKind.TSX;
+        }
+        else if (['.js', '.mjs', '.cjs'].includes(ext)) {
+            scriptKind = ts.ScriptKind.JS;
+        }
+        const sourceFile = ts.createSourceFile(relativePath, content, ts.ScriptTarget.Latest, true, scriptKind);
+        const findIdentifiers = (node) => {
+            if (ts.isIdentifier(node) && node.text === symbolName) {
+                const line = sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1;
+                const column = sourceFile.getLineAndCharacterOfPosition(node.getStart()).character;
+                // Skip the definition itself
+                if (relativePath === targetSymbol.location.filepath &&
+                    line === targetSymbol.location.startLine) {
+                    return;
+                }
+                // Get context (surrounding lines)
+                const startLine = Math.max(0, line - 2);
+                const endLine = Math.min(lines.length, line + 1);
+                const codeSnippet = lines.slice(startLine, endLine).join('\n');
+                // Create a minimal referencing symbol
+                const referencingSymbol = {
+                    name: symbolName,
+                    namePath: `/${relativePath}:${line}`,
+                    kind: SymbolKind.Variable, // Default - we don't have full context here
+                    location: {
+                        filepath: relativePath,
+                        startLine: line,
+                        endLine: line,
+                        startColumn: column,
+                        endColumn: column + symbolName.length,
+                    },
+                    depth: 0,
+                };
+                references.push({
+                    referencingSymbol,
+                    codeSnippet,
+                    line,
+                    column,
+                });
+            }
+            ts.forEachChild(node, findIdentifiers);
+        };
+        findIdentifiers(sourceFile);
+        return references;
+    }
+    async findReferencesWithTreeSitter(content, _filepath, relativePath, lines, symbolName, targetSymbol, config) {
+        await ReferenceFinder.initialize();
+        const language = await ReferenceFinder.loadLanguage(config);
+        const parser = ReferenceFinder.parser;
+        if (!parser) {
+            return [];
+        }
+        parser.setLanguage(language);
+        const tree = parser.parse(content);
+        if (!tree) {
+            return [];
+        }
+        const references = [];
+        const findIdentifiers = (node) => {
+            if (config.identifierTypes.includes(node.type) && node.text === symbolName) {
+                const line = node.startPosition.row + 1;
+                const column = node.startPosition.column;
+                // Skip the definition itself
+                if (relativePath === targetSymbol.location.filepath &&
+                    line >= targetSymbol.location.startLine &&
+                    line <= targetSymbol.location.endLine) {
+                    return;
+                }
+                // Get context (surrounding lines)
+                const startLine = Math.max(0, line - 2);
+                const endLine = Math.min(lines.length, line + 1);
+                const codeSnippet = lines.slice(startLine, endLine).join('\n');
+                const referencingSymbol = {
+                    name: symbolName,
+                    namePath: `/${relativePath}:${line}`,
+                    kind: SymbolKind.Variable,
+                    location: {
+                        filepath: relativePath,
+                        startLine: line,
+                        endLine: line,
+                        startColumn: column,
+                        endColumn: column + symbolName.length,
+                    },
+                    depth: 0,
+                };
+                references.push({
+                    referencingSymbol,
+                    codeSnippet,
+                    line,
+                    column,
+                });
+            }
+            for (const child of node.children) {
+                findIdentifiers(child);
+            }
+        };
+        findIdentifiers(tree.rootNode);
+        return references;
+    }
+}
+/**
+ * Format reference results for display.
+ */
+export function formatReferencesResult(references) {
+    if (references.length === 0) {
+        return 'No references found.';
+    }
+    const parts = [];
+    parts.push(`Found ${references.length} reference${references.length === 1 ? '' : 's'}:\n`);
+    // Group by file
+    const byFile = new Map();
+    for (const ref of references) {
+        const file = ref.referencingSymbol.location.filepath;
+        const fileRefs = byFile.get(file) || [];
+        fileRefs.push(ref);
+        byFile.set(file, fileRefs);
+    }
+    for (const [filepath, fileRefs] of byFile) {
+        parts.push(`\n## ${filepath}\n`);
+        for (const ref of fileRefs) {
+            parts.push(`Line ${ref.line}:`);
+            parts.push('```');
+            parts.push(ref.codeSnippet);
+            parts.push('```\n');
+        }
+    }
+    return parts.join('\n');
+}
+//# sourceMappingURL=reference-finder.js.map

package/dist/symbols/reference-finder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reference-finder.js","sourceRoot":"","sources":["../../src/symbols/reference-finder.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,KAAK,EAAE,MAAM,YAAY,CAAC;AAEjC,OAAO,EAA2B,UAAU,EAAyB,MAAM,YAAY,CAAC;AACxF,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAQ7E,IAAI,YAAY,GAAwB,IAAI,CAAC;AAC7C,MAAM,gBAAgB,GAAG,KAAK,IAA2B,EAAE;IACzD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;QAC5C,YAAY,GAAG;YACb,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,QAAQ,EAAE,GAAG,CAAC,QAAQ;SACvB,CAAC;IACJ,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC,CAAC;AAWF,MAAM,gBAAgB,GAAmC;IACvD,MAAM,EAAE;QACN,QAAQ,EAAE,yBAAyB;QACnC,UAAU,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC;QAC3B,eAAe,EAAE,CAAC,YAAY,CAAC;KAChC;IACD,EAAE,EAAE;QACF,QAAQ,EAAE,qBAAqB;QAC/B,UAAU,EAAE,CAAC,KAAK,CAAC;QACnB,eAAe,EAAE,CAAC,YAAY,EAAE,iBAAiB,EAAE,kBAAkB,CAAC;KACvE;IACD,IAAI,EAAE;QACJ,QAAQ,EAAE,uBAAuB;QACjC,UAAU,EAAE,CAAC,KAAK,CAAC;QACnB,eAAe,EAAE,CAAC,YAAY,EAAE,iBAAiB,EAAE,kBAAkB,CAAC;KACvE;IACD,IAAI,EAAE;QACJ,QAAQ,EAAE,uBAAuB;QACjC,UAAU,EAAE,CAAC,OAAO,CAAC;QACrB,eAAe,EAAE,CAAC,YAAY,EAAE,iBAAiB,CAAC;KACnD;IACD,IAAI,EAAE;QACJ,QAAQ,EAAE,uBAAuB;QACjC,UAAU,EAAE,CAAC,KAAK,CAAC;QACnB,eAAe,EAAE,CAAC,YAAY,EAAE,UAAU,CAAC;KAC5C;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,wBAAwB,GAAG;IAC/B,iBAAiB;IACjB,SAAS;IACT,SAAS;IACT,UAAU;IACV,aAAa;IACb,gBAAgB;IAChB,UAAU;IACV,WAAW;CACZ,CAAC;AAEF;;GAEG;AACH,MAAM,OAAO,eAAe;IAClB,MAAM,CAAC,MAAM,GAAsB,IAAI,CAAC;IACxC,MAAM,CAAC,eAAe,GAA0B,IAAI,GAAG,EAAE,CAAC;IAC1D,MAAM,CAAC,WAAW,GAAyB,IAAI,CAAC;IAChD,MAAM,CAAC,YAAY,GAAkB,IAAI,CAAC;IAE1C,WAAW,CAAS;IACpB,eAAe,CAAkB;IAEzC,YAAY,WAAmB;QAC7B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC;IAC1D,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,KAAK,CAAC,UAAU;QAC7B,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,CAAC,KAAK,IAAI,EAAE;YAC7B,IAAI,CAAC,YAAY,GAAG,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAClD,MAAM,MAAM,GAAG,MAAM,gBAAgB,EAAE,CAAC;YAExC,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CACjC,OAAO,CAAC,GAAG,EAAE,EACb,cAAc,EACd,iBAAiB,EACjB,sBAAsB,CACvB,CAAC;YACF,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC;gBACvB,UAAU,EAAE,GAAG,EAAE,CAAC,iBAAiB;aACpC,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,CAAC,CAAC,EAAE,CAAC;QAEL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,gBAAgB;QACnC,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,aAAa,GAAG;YACpB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,kBAAkB,EAAE,MAAM,CAAC;YAC/E,IAAI,CAAC,IAAI,CACP,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAC9B,IAAI,EACJ,IAAI,EACJ,IAAI,EACJ,cAAc,EACd,SAAS,EACT,kBAAkB,EAClB,MAAM,CACP;SACF,CAAC;QAEF,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;YAC9B,IAAI,CAAC;gBACH,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBACnB,OAAO,CAAC,CAAC;YACX,CAAC;YAAC,MAAM,CAAC;gBACP,WAAW;YACb,CAAC;QACH,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;IACtE,CAAC;IAEO,MAAM,CAAC,iBAAiB,CAAC,QAAgB;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QACjD,KAAK,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC;YAC1D,IAAI,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACpC,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,MAAsB;QACtD,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,gBAAgB,EAAE,CAAC;QACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC/D,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACpD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,OAA8B;QACjD,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,WAAW,GAAG,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC;QAE5F,gCAAgC;QAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;QAC3D,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAE3E,MAAM,OAAO,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;QACxC,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAEhE,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,OAAO,YAAY,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,UAAU,GAAG,aAAa,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAExD,2BAA2B;QAC3B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE7C,MAAM,UAAU,GAAsB,EAAE,CAAC;QAEzC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAC9C,IAAI,EACJ,UAAU,EACV,YAAY,EACZ,WAAW,EACX,YAAY,EACZ,YAAY,CACb,CAAC;YACF,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,mBAAmB,CACzB,OAAiB,EACjB,OAAyC;QAEzC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,aAAa,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC5C,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpB,MAAM,KAAK,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACjE,IAAI,KAAK,EAAE,CAAC;oBACV,OAAO,KAAK,CAAC;gBACf,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,iBAAiB;QAC7B,MAAM,cAAc,GAAG;YACrB,KAAK;YACL,MAAM;YACN,KAAK;YACL,MAAM;YACN,MAAM;YACN,MAAM;YACN,MAAM;YACN,MAAM;YACN,KAAK;YACL,MAAM;YACN,KAAK;YACL,KAAK;YACL,OAAO;YACP,KAAK;YACL,KAAK;SACN,CAAC;QAEF,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC;QAC3D,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE;gBAClC,GAAG,EAAE,IAAI,CAAC,WAAW;gBACrB,MAAM,EAAE,wBAAwB;gBAChC,QAAQ,EAAE,IAAI;aACf,CAAC,CAAC;YACH,KAAK,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC;QACzB,CAAC;QAED,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IAC7B,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,QAAgB,EAChB,UAAkB,EAClB,YAAoB,EACpB,WAAoB,EACpB,YAA2B,EAC3B,YAA2B;QAE3B,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QACjD,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAC/D,MAAM,UAAU,GAAsB,EAAE,CAAC;QAEzC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAElC,oDAAoD;YACpD,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,KAAK,YAAY,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACjF,6EAA6E;YAC/E,CAAC;YAED,iCAAiC;YACjC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACjF,MAAM,IAAI,GAAG,IAAI,CAAC,4BAA4B,CAC5C,OAAO,EACP,QAAQ,EACR,YAAY,EACZ,KAAK,EACL,UAAU,EACV,YAAY,CACb,CAAC;gBACF,UAAU,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;YAC3B,CAAC;iBAAM,CAAC;gBACN,sCAAsC;gBACtC,MAAM,MAAM,GAAG,eAAe,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;gBAC3D,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,4BAA4B,CAClD,OAAO,EACP,QAAQ,EACR,YAAY,EACZ,KAAK,EACL,UAAU,EACV,YAAY,EACZ,MAAM,CACP,CAAC;oBACF,UAAU,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;YAED,kBAAkB;YAClB,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC/B,IAAI,YAAY,IAAI,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;oBACtE,OAAO,KAAK,CAAC;gBACf,CAAC;gBACD,IAAI,YAAY,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;oBACvE,OAAO,KAAK,CAAC;gBACf,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,gCAAgC;YAChC,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,4BAA4B,CAClC,OAAe,EACf,SAAiB,EACjB,YAAoB,EACpB,KAAe,EACf,UAAkB,EAClB,YAAoB;QAEpB,MAAM,UAAU,GAAsB,EAAE,CAAC;QAEzC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,WAAW,EAAE,CAAC;QACrD,IAAI,UAAU,GAAG,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;QAClC,IAAI,GAAG,KAAK,MAAM,IAAI,GAAG,KAAK,MAAM,EAAE,CAAC;YACrC,UAAU,GAAG,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;QACjC,CAAC;aAAM,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACjD,UAAU,GAAG,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;QAChC,CAAC;QAED,MAAM,UAAU,GAAG,EAAE,CAAC,gBAAgB,CACpC,YAAY,EACZ,OAAO,EACP,EAAE,CAAC,YAAY,CAAC,MAAM,EACtB,IAAI,EACJ,UAAU,CACX,CAAC;QAEF,MAAM,eAAe,GAAG,CAAC,IAAa,EAAE,EAAE;YACxC,IAAI,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACtD,MAAM,IAAI,GAAG,UAAU,CAAC,6BAA6B,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC;gBAChF,MAAM,MAAM,GAAG,UAAU,CAAC,6BAA6B,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,SAAS,CAAC;gBAEnF,6BAA6B;gBAC7B,IACE,YAAY,KAAK,YAAY,CAAC,QAAQ,CAAC,QAAQ;oBAC/C,IAAI,KAAK,YAAY,CAAC,QAAQ,CAAC,SAAS,EACxC,CAAC;oBACD,OAAO;gBACT,CAAC;gBAED,kCAAkC;gBAClC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;gBACxC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;gBACjD,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAE/D,sCAAsC;gBACtC,MAAM,iBAAiB,GAAW;oBAChC,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,IAAI,YAAY,IAAI,IAAI,EAAE;oBACpC,IAAI,EAAE,UAAU,CAAC,QAAQ,EAAE,4CAA4C;oBACvE,QAAQ,EAAE;wBACR,QAAQ,EAAE,YAAY;wBACtB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,IAAI;wBACb,WAAW,EAAE,MAAM;wBACnB,SAAS,EAAE,MAAM,GAAG,UAAU,CAAC,MAAM;qBACtC;oBACD,KAAK,EAAE,CAAC;iBACT,CAAC;gBAEF,UAAU,CAAC,IAAI,CAAC;oBACd,iBAAiB;oBACjB,WAAW;oBACX,IAAI;oBACJ,MAAM;iBACP,CAAC,CAAC;YACL,CAAC;YAED,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;QACzC,CAAC,CAAC;QAEF,eAAe,CAAC,UAAU,CAAC,CAAC;QAE5B,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,4BAA4B,CACxC,OAAe,EACf,SAAiB,EACjB,YAAoB,EACpB,KAAe,EACf,UAAkB,EAClB,YAAoB,EACpB,MAAsB;QAEtB,MAAM,eAAe,CAAC,UAAU,EAAE,CAAC;QAEnC,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAE7B,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,UAAU,GAAsB,EAAE,CAAC;QAEzC,MAAM,eAAe,GAAG,CAAC,IAAgB,EAAE,EAAE;YAC3C,IAAI,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC3E,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,CAAC;gBACxC,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;gBAEzC,6BAA6B;gBAC7B,IACE,YAAY,KAAK,YAAY,CAAC,QAAQ,CAAC,QAAQ;oBAC/C,IAAI,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS;oBACvC,IAAI,IAAI,YAAY,CAAC,QAAQ,CAAC,OAAO,EACrC,CAAC;oBACD,OAAO;gBACT,CAAC;gBAED,kCAAkC;gBAClC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;gBACxC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;gBACjD,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAE/D,MAAM,iBAAiB,GAAW;oBAChC,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,IAAI,YAAY,IAAI,IAAI,EAAE;oBACpC,IAAI,EAAE,UAAU,CAAC,QAAQ;oBACzB,QAAQ,EAAE;wBACR,QAAQ,EAAE,YAAY;wBACtB,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,IAAI;wBACb,WAAW,EAAE,MAAM;wBACnB,SAAS,EAAE,MAAM,GAAG,UAAU,CAAC,MAAM;qBACtC;oBACD,KAAK,EAAE,CAAC;iBACT,CAAC;gBAEF,UAAU,CAAC,IAAI,CAAC;oBACd,iBAAiB;oBACjB,WAAW;oBACX,IAAI;oBACJ,MAAM;iBACP,CAAC,CAAC;YACL,CAAC;YAED,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClC,eAAe,CAAC,KAAK,CAAC,CAAC;YACzB,CAAC;QACH,CAAC,CAAC;QAEF,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE/B,OAAO,UAAU,CAAC;IACpB,CAAC;;AAGH;;GAEG;AACH,MAAM,UAAU,sBAAsB,CAAC,UAA6B;IAClE,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,sBAAsB,CAAC;IAChC,CAAC;IAED,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,CAAC,IAAI,CAAC,SAAS,UAAU,CAAC,MAAM,aAAa,UAAU,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;IAE3F,gBAAgB;IAChB,MAAM,MAAM,GAAG,IAAI,GAAG,EAA6B,CAAC;IACpD,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;QAC7B,MAAM,IAAI,GAAG,GAAG,CAAC,iBAAiB,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACrD,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACxC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,MAAM,EAAE,CAAC;QAC1C,KAAK,CAAC,IAAI,CAAC,QAAQ,QAAQ,IAAI,CAAC,CAAC;QAEjC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,QAAQ,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC;YAChC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClB,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAC5B,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}

package/dist/symbols/symbol-editor.d.ts

@@ -0,0 +1,81 @@
+/**
+ * Symbol editor for modifying symbol definitions in source files.
+ * Provides atomic operations to replace, insert before, or insert after symbols.
+ */
+/**
+ * Result from a symbol edit operation.
+ */
+export interface EditResult {
+    /** Whether the edit was successful */
+    success: boolean;
+    /** Updated file path */
+    filepath: string;
+    /** Symbol that was edited */
+    symbolName: string;
+    /** New line range of the edited content */
+    newRange?: {
+        startLine: number;
+        endLine: number;
+    };
+    /** Error message if failed */
+    error?: string;
+}
+/**
+ * Options for replace_symbol_body.
+ */
+export interface ReplaceSymbolOptions {
+    /** Name path of the symbol to replace */
+    namePath: string;
+    /** Relative path to the file containing the symbol */
+    relativePath: string;
+    /** The new body content for the symbol */
+    body: string;
+}
+/**
+ * Options for insert_before_symbol or insert_after_symbol.
+ */
+export interface InsertSymbolOptions {
+    /** Name path of the reference symbol */
+    namePath: string;
+    /** Relative path to the file containing the symbol */
+    relativePath: string;
+    /** The content to insert */
+    body: string;
+}
+/**
+ * Symbol editor for modifying source files.
+ */
+export declare class SymbolEditor {
+    private projectPath;
+    private symbolExtractor;
+    constructor(projectPath: string);
+    /**
+     * Replace a symbol's body with new content.
+     */
+    replaceSymbolBody(options: ReplaceSymbolOptions): Promise<EditResult>;
+    /**
+     * Insert content before a symbol.
+     */
+    insertBeforeSymbol(options: InsertSymbolOptions): Promise<EditResult>;
+    /**
+     * Insert content after a symbol.
+     */
+    insertAfterSymbol(options: InsertSymbolOptions): Promise<EditResult>;
+    /**
+     * Find a symbol by name path pattern.
+     */
+    private findSymbolByPattern;
+    /**
+     * Detect the indentation used on a line.
+     */
+    private detectIndentation;
+    /**
+     * Apply indentation to a block of code.
+     */
+    private applyIndentation;
+    /**
+     * Write a file atomically (write to temp file, then rename).
+     */
+    private atomicWrite;
+}
+//# sourceMappingURL=symbol-editor.d.ts.map

package/dist/symbols/symbol-editor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"symbol-editor.d.ts","sourceRoot":"","sources":["../../src/symbols/symbol-editor.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAQH;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,sCAAsC;IACtC,OAAO,EAAE,OAAO,CAAC;IACjB,wBAAwB;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,6BAA6B;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,2CAA2C;IAC3C,QAAQ,CAAC,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;IAClD,8BAA8B;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,yCAAyC;IACzC,QAAQ,EAAE,MAAM,CAAC;IACjB,sDAAsD;IACtD,YAAY,EAAE,MAAM,CAAC;IACrB,0CAA0C;IAC1C,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,wCAAwC;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,sDAAsD;IACtD,YAAY,EAAE,MAAM,CAAC;IACrB,4BAA4B;IAC5B,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,qBAAa,YAAY;IACvB,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,eAAe,CAAkB;gBAE7B,WAAW,EAAE,MAAM;IAK/B;;OAEG;IACG,iBAAiB,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,UAAU,CAAC;IA2D3E;;OAEG;IACG,kBAAkB,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,UAAU,CAAC;IAiE3E;;OAEG;IACG,iBAAiB,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,UAAU,CAAC;IAiE1E;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAkB3B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAKzB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAyBxB;;OAEG;YACW,WAAW;CAe1B"}