attocode 0.1.0

package/dist/src/agent-tools/lsp-file-tools.js

@@ -0,0 +1,200 @@
+/**
+ * LSP-Aware File Tools
+ *
+ * Enhanced file editing tools that leverage Language Server Protocol
+ * to provide real-time diagnostic feedback after modifications.
+ */
+import { z } from 'zod';
+import * as fs from 'node:fs/promises';
+import * as path from 'node:path';
+// =============================================================================
+// HELPER FUNCTIONS
+// =============================================================================
+function escapeRegExp(string) {
+    return string.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+}
+function formatDiagnostics(diagnostics, filePath) {
+    if (diagnostics.length === 0) {
+        return '';
+    }
+    const errors = diagnostics.filter(d => d.severity === 'error');
+    const warnings = diagnostics.filter(d => d.severity === 'warning');
+    const parts = [];
+    if (errors.length > 0) {
+        parts.push(`\n\n⚠️ ${errors.length} error(s) detected:`);
+        for (const err of errors.slice(0, 5)) {
+            const loc = `${path.basename(filePath)}:${err.range.start.line + 1}:${err.range.start.character + 1}`;
+            parts.push(`  • [${loc}] ${err.message}`);
+        }
+        if (errors.length > 5) {
+            parts.push(`  ... and ${errors.length - 5} more errors`);
+        }
+    }
+    if (warnings.length > 0) {
+        parts.push(`\n📋 ${warnings.length} warning(s):`);
+        for (const warn of warnings.slice(0, 3)) {
+            const loc = `${path.basename(filePath)}:${warn.range.start.line + 1}`;
+            parts.push(`  • [${loc}] ${warn.message}`);
+        }
+        if (warnings.length > 3) {
+            parts.push(`  ... and ${warnings.length - 3} more warnings`);
+        }
+    }
+    return parts.join('\n');
+}
+async function waitForDiagnostics(lsp, filePath, delay) {
+    // Notify LSP about the file - must open first, then notify change
+    try {
+        const content = await fs.readFile(filePath, 'utf-8');
+        // Open the document (required for LSP to track it)
+        lsp.notifyFileOpened(filePath, content);
+        // Small delay then notify change to trigger diagnostics
+        await new Promise(resolve => setTimeout(resolve, 100));
+        lsp.notifyFileChanged(filePath, content, Date.now());
+    }
+    catch {
+        // File might not exist yet, that's ok
+    }
+    // Wait for diagnostics to be computed
+    await new Promise(resolve => setTimeout(resolve, delay));
+    return lsp.getDiagnostics(filePath);
+}
+// =============================================================================
+// LSP-AWARE EDIT FILE TOOL
+// =============================================================================
+const lspEditFileSchema = z.object({
+    path: z.string().describe('Path to the file to edit'),
+    old_string: z.string().describe('Exact string to find (must be unique in file)'),
+    new_string: z.string().describe('String to replace it with'),
+});
+/**
+ * Create an LSP-aware edit_file tool.
+ */
+export function createLSPEditFileTool(config) {
+    const { lspManager, diagnosticDelay = 500, includeWarnings = true } = config;
+    return {
+        name: 'edit_file',
+        description: 'Make a surgical edit by replacing a unique string in a file (with LSP diagnostics)',
+        parameters: lspEditFileSchema.shape,
+        dangerLevel: 'moderate',
+        async execute(args) {
+            const input = lspEditFileSchema.parse(args);
+            try {
+                // Read file
+                const content = await fs.readFile(input.path, 'utf-8');
+                // Count occurrences
+                const regex = new RegExp(escapeRegExp(input.old_string), 'g');
+                const matches = content.match(regex);
+                const count = matches?.length ?? 0;
+                if (count === 0) {
+                    return `String not found in ${input.path}. Make sure old_string exactly matches the file content, including whitespace.`;
+                }
+                if (count > 1) {
+                    return `String found ${count} times in ${input.path}. The old_string must be unique. Include more surrounding context.`;
+                }
+                // Replace
+                const newContent = content.replace(input.old_string, input.new_string);
+                await fs.writeFile(input.path, newContent, 'utf-8');
+                const linesDiff = input.new_string.split('\n').length - input.old_string.split('\n').length;
+                let output = `Successfully edited ${input.path} (${linesDiff >= 0 ? '+' : ''}${linesDiff} lines)`;
+                // Get LSP diagnostics
+                const diagnostics = await waitForDiagnostics(lspManager, input.path, diagnosticDelay);
+                const diagOutput = formatDiagnostics(includeWarnings ? diagnostics : diagnostics.filter(d => d.severity === 'error'), input.path);
+                if (diagOutput) {
+                    output += diagOutput;
+                }
+                else {
+                    output += '\n✅ No errors detected';
+                }
+                return output;
+            }
+            catch (error) {
+                const err = error;
+                if (err.code === 'ENOENT') {
+                    return `File not found: ${input.path}`;
+                }
+                return `Error editing file: ${err.message}`;
+            }
+        },
+    };
+}
+// =============================================================================
+// LSP-AWARE WRITE FILE TOOL
+// =============================================================================
+const lspWriteFileSchema = z.object({
+    path: z.string().describe('Path to the file to write'),
+    content: z.string().describe('Content to write to the file'),
+});
+/**
+ * Create an LSP-aware write_file tool.
+ */
+export function createLSPWriteFileTool(config) {
+    const { lspManager, diagnosticDelay = 500, includeWarnings = true } = config;
+    return {
+        name: 'write_file',
+        description: 'Write content to a file (creates or overwrites) with LSP diagnostics',
+        parameters: lspWriteFileSchema.shape,
+        dangerLevel: 'moderate',
+        async execute(args) {
+            const input = lspWriteFileSchema.parse(args);
+            try {
+                // Ensure directory exists
+                const dir = path.dirname(input.path);
+                await fs.mkdir(dir, { recursive: true });
+                // Check if file exists
+                let action = 'created';
+                try {
+                    await fs.access(input.path);
+                    action = 'overwrote';
+                }
+                catch {
+                    // File doesn't exist, that's fine
+                }
+                await fs.writeFile(input.path, input.content, 'utf-8');
+                const lines = input.content.split('\n').length;
+                let output = `Successfully ${action} ${input.path} (${lines} lines, ${input.content.length} bytes)`;
+                // Get LSP diagnostics
+                const diagnostics = await waitForDiagnostics(lspManager, input.path, diagnosticDelay);
+                const diagOutput = formatDiagnostics(includeWarnings ? diagnostics : diagnostics.filter(d => d.severity === 'error'), input.path);
+                if (diagOutput) {
+                    output += diagOutput;
+                }
+                else {
+                    output += '\n✅ No errors detected';
+                }
+                return output;
+            }
+            catch (error) {
+                const err = error;
+                return `Error writing file: ${err.message}`;
+            }
+        },
+    };
+}
+// =============================================================================
+// FACTORY
+// =============================================================================
+/**
+ * Create all LSP-aware file tools.
+ */
+export function createLSPFileTools(config) {
+    return [
+        createLSPEditFileTool(config),
+        createLSPWriteFileTool(config),
+    ];
+}
+/**
+ * Check if LSP diagnostics are available for a file type.
+ */
+export function isLSPSupportedFile(filePath) {
+    const ext = path.extname(filePath).toLowerCase();
+    const supported = [
+        '.ts', '.tsx', '.js', '.jsx', '.mjs', '.cjs', // TypeScript/JavaScript
+        '.py', '.pyi', // Python
+        '.rs', // Rust
+        '.go', // Go
+        '.json', '.jsonc', // JSON
+    ];
+    return supported.includes(ext);
+}
+//# sourceMappingURL=lsp-file-tools.js.map

package/dist/src/agent-tools/lsp-file-tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lsp-file-tools.js","sourceRoot":"","sources":["../../../src/agent-tools/lsp-file-tools.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAiBlC,gFAAgF;AAChF,mBAAmB;AACnB,gFAAgF;AAEhF,SAAS,YAAY,CAAC,MAAc;IAClC,OAAO,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;AACvD,CAAC;AAED,SAAS,iBAAiB,CAAC,WAA4B,EAAE,QAAgB;IACvE,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC;IAC/D,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC;IAEnE,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,KAAK,CAAC,IAAI,CAAC,UAAU,MAAM,CAAC,MAAM,qBAAqB,CAAC,CAAC;QACzD,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACrC,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC;YACtG,KAAK,CAAC,IAAI,CAAC,QAAQ,GAAG,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,KAAK,CAAC,IAAI,CAAC,aAAa,MAAM,CAAC,MAAM,GAAG,CAAC,cAAc,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,KAAK,CAAC,IAAI,CAAC,QAAQ,QAAQ,CAAC,MAAM,cAAc,CAAC,CAAC;QAClD,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACxC,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACtE,KAAK,CAAC,IAAI,CAAC,QAAQ,GAAG,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,KAAK,CAAC,IAAI,CAAC,aAAa,QAAQ,CAAC,MAAM,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,KAAK,UAAU,kBAAkB,CAC/B,GAAe,EACf,QAAgB,EAChB,KAAa;IAEb,kEAAkE;IAClE,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACrD,mDAAmD;QACnD,GAAG,CAAC,gBAAgB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACxC,wDAAwD;QACxD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACvD,GAAG,CAAC,iBAAiB,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IACvD,CAAC;IAAC,MAAM,CAAC;QACP,sCAAsC;IACxC,CAAC;IAED,sCAAsC;IACtC,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;IAEzD,OAAO,GAAG,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;AACtC,CAAC;AAED,gFAAgF;AAChF,2BAA2B;AAC3B,gFAAgF;AAEhF,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACjC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;IACrD,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+CAA+C,CAAC;IAChF,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;CAC7D,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,MAA0B;IAC9D,MAAM,EAAE,UAAU,EAAE,eAAe,GAAG,GAAG,EAAE,eAAe,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;IAE7E,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,oFAAoF;QACjG,UAAU,EAAE,iBAAiB,CAAC,KAA2C;QACzE,WAAW,EAAE,UAAU;QACvB,KAAK,CAAC,OAAO,CAAC,IAA6B;YACzC,MAAM,KAAK,GAAG,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAE5C,IAAI,CAAC;gBACH,YAAY;gBACZ,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAEvD,oBAAoB;gBACpB,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,GAAG,CAAC,CAAC;gBAC9D,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACrC,MAAM,KAAK,GAAG,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC;gBAEnC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;oBAChB,OAAO,uBAAuB,KAAK,CAAC,IAAI,gFAAgF,CAAC;gBAC3H,CAAC;gBAED,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;oBACd,OAAO,gBAAgB,KAAK,aAAa,KAAK,CAAC,IAAI,oEAAoE,CAAC;gBAC1H,CAAC;gBAED,UAAU;gBACV,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;gBACvE,MAAM,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;gBAEpD,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;gBAC5F,IAAI,MAAM,GAAG,uBAAuB,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,SAAS,SAAS,CAAC;gBAElG,sBAAsB;gBACtB,MAAM,WAAW,GAAG,MAAM,kBAAkB,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;gBACtF,MAAM,UAAU,GAAG,iBAAiB,CAClC,eAAe,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,OAAO,CAAC,EAC/E,KAAK,CAAC,IAAI,CACX,CAAC;gBAEF,IAAI,UAAU,EAAE,CAAC;oBACf,MAAM,IAAI,UAAU,CAAC;gBACvB,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,wBAAwB,CAAC;gBACrC,CAAC;gBAED,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,GAAG,GAAG,KAA8B,CAAC;gBAC3C,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC1B,OAAO,mBAAmB,KAAK,CAAC,IAAI,EAAE,CAAC;gBACzC,CAAC;gBACD,OAAO,uBAAuB,GAAG,CAAC,OAAO,EAAE,CAAC;YAC9C,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,gFAAgF;AAChF,4BAA4B;AAC5B,gFAAgF;AAEhF,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IAClC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IACtD,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;CAC7D,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAA0B;IAC/D,MAAM,EAAE,UAAU,EAAE,eAAe,GAAG,GAAG,EAAE,eAAe,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;IAE7E,OAAO;QACL,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,sEAAsE;QACnF,UAAU,EAAE,kBAAkB,CAAC,KAA2C;QAC1E,WAAW,EAAE,UAAU;QACvB,KAAK,CAAC,OAAO,CAAC,IAA6B;YACzC,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAE7C,IAAI,CAAC;gBACH,0BAA0B;gBAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACrC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAEzC,uBAAuB;gBACvB,IAAI,MAAM,GAAG,SAAS,CAAC;gBACvB,IAAI,CAAC;oBACH,MAAM,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC5B,MAAM,GAAG,WAAW,CAAC;gBACvB,CAAC;gBAAC,MAAM,CAAC;oBACP,kCAAkC;gBACpC,CAAC;gBAED,MAAM,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBACvD,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;gBAE/C,IAAI,MAAM,GAAG,gBAAgB,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,WAAW,KAAK,CAAC,OAAO,CAAC,MAAM,SAAS,CAAC;gBAEpG,sBAAsB;gBACtB,MAAM,WAAW,GAAG,MAAM,kBAAkB,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;gBACtF,MAAM,UAAU,GAAG,iBAAiB,CAClC,eAAe,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,OAAO,CAAC,EAC/E,KAAK,CAAC,IAAI,CACX,CAAC;gBAEF,IAAI,UAAU,EAAE,CAAC;oBACf,MAAM,IAAI,UAAU,CAAC;gBACvB,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,wBAAwB,CAAC;gBACrC,CAAC;gBAED,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,GAAG,GAAG,KAA8B,CAAC;gBAC3C,OAAO,uBAAuB,GAAG,CAAC,OAAO,EAAE,CAAC;YAC9C,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,gFAAgF;AAChF,UAAU;AACV,gFAAgF;AAEhF;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,MAA0B;IAC3D,OAAO;QACL,qBAAqB,CAAC,MAAM,CAAC;QAC7B,sBAAsB,CAAC,MAAM,CAAC;KAC/B,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,QAAgB;IACjD,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IACjD,MAAM,SAAS,GAAG;QAChB,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAG,wBAAwB;QACvE,KAAK,EAAE,MAAM,EAAoC,SAAS;QAC1D,KAAK,EAA4C,OAAO;QACxD,KAAK,EAA4C,KAAK;QACtD,OAAO,EAAE,QAAQ,EAAgC,OAAO;KACzD,CAAC;IACF,OAAO,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;AACjC,CAAC"}