@bahulam/code 2.6.0

package/src/tools/glob.mjs

@@ -0,0 +1,117 @@
+/**
+ * Glob Tool — matches Claude Code's exact behavior.
+ *
+ * Features:
+ * - Proper glob matching (not shell find)
+ * - Sort by modification time
+ * - path parameter for directory scoping
+ */
+import fs from 'fs';
+import path from 'path';
+
+/**
+ * Minimal glob implementation without external dependencies.
+ * Supports *, **, and ? wildcards.
+ */
+function globMatch(pattern, str) {
+    const regex = globToRegex(pattern);
+    return regex.test(str);
+}
+
+function globToRegex(pattern) {
+    let regex = '';
+    let i = 0;
+    while (i < pattern.length) {
+        const ch = pattern[i];
+        if (ch === '*') {
+            if (pattern[i + 1] === '*') {
+                // ** matches everything including /
+                regex += '.*';
+                i += 2;
+                if (pattern[i] === '/') i++; // skip trailing /
+                continue;
+            }
+            regex += '[^/]*';
+        } else if (ch === '?') {
+            regex += '[^/]';
+        } else if (ch === '.') {
+            regex += '\\.';
+        } else if (ch === '{') {
+            regex += '(';
+        } else if (ch === '}') {
+            regex += ')';
+        } else if (ch === ',') {
+            regex += '|';
+        } else {
+            regex += ch;
+        }
+        i++;
+    }
+    return new RegExp('^' + regex + '$');
+}
+
+function walkDir(dir, maxDepth = 20, depth = 0) {
+    const results = [];
+    if (depth > maxDepth) return results;
+    try {
+        const entries = fs.readdirSync(dir, { withFileTypes: true });
+        for (const entry of entries) {
+            // Skip hidden dirs and node_modules
+            if (entry.name.startsWith('.') && depth > 0) continue;
+            if (entry.name === 'node_modules') continue;
+
+            const full = path.join(dir, entry.name);
+            if (entry.isDirectory()) {
+                results.push(...walkDir(full, maxDepth, depth + 1));
+            } else {
+                results.push(full);
+            }
+        }
+    } catch {
+        // Permission denied or other error
+    }
+    return results;
+}
+
+export const GlobTool = {
+    name: 'Glob',
+    description: 'Find files matching a glob pattern, sorted by modification time.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            pattern: { type: 'string', description: 'Glob pattern (e.g. "**/*.js")' },
+            path: { type: 'string', description: 'Directory to search in' },
+        },
+        required: ['pattern'],
+    },
+    validateInput(input) { return input.pattern ? [] : ['pattern required']; },
+    async call(input) {
+        try {
+            const baseDir = path.resolve(input.path || '.');
+            if (!fs.existsSync(baseDir)) {
+                return `Error: Directory not found: ${baseDir}`;
+            }
+
+            const allFiles = walkDir(baseDir);
+
+            // Match against pattern
+            const matches = allFiles.filter(f => {
+                const rel = path.relative(baseDir, f);
+                return globMatch(input.pattern, rel) || globMatch(input.pattern, path.basename(f));
+            });
+
+            if (matches.length === 0) return 'No matches found.';
+
+            // Sort by modification time (newest first)
+            matches.sort((a, b) => {
+                try {
+                    return fs.statSync(b).mtimeMs - fs.statSync(a).mtimeMs;
+                } catch { return 0; }
+            });
+
+            return matches.join('\n');
+        } catch (e) {
+            return `Error: ${e.message}`;
+        }
+    },
+};

package/src/tools/grep.mjs

@@ -0,0 +1,129 @@
+/**
+ * Grep Tool — matches Claude Code's exact behavior.
+ *
+ * Features:
+ * - Case insensitive (-i)
+ * - Line numbers (-n, default true for content mode)
+ * - Context lines (-A, -B, -C)
+ * - output_mode: content, files_with_matches, count
+ * - glob filter and type filter
+ * - head_limit (default 250)
+ * - multiline mode
+ */
+import { spawnSync } from 'child_process';
+import path from 'path';
+
+export const GrepTool = {
+    name: 'Grep',
+    description: 'Search file contents with regex (powered by ripgrep or grep).',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            pattern: { type: 'string', description: 'Regex pattern to search for' },
+            path: { type: 'string', description: 'File or directory to search in' },
+            '-i': { type: 'boolean', description: 'Case insensitive' },
+            '-n': { type: 'boolean', description: 'Show line numbers (default true)' },
+            '-A': { type: 'number', description: 'Lines after match' },
+            '-B': { type: 'number', description: 'Lines before match' },
+            '-C': { type: 'number', description: 'Context lines (before and after)' },
+            context: { type: 'number', description: 'Alias for -C' },
+            output_mode: {
+                type: 'string',
+                enum: ['content', 'files_with_matches', 'count'],
+                description: 'Output mode (default: files_with_matches)',
+            },
+            glob: { type: 'string', description: 'Glob pattern to filter files' },
+            type: { type: 'string', description: 'File type filter (e.g. js, py)' },
+            head_limit: { type: 'number', description: 'Max output lines (default 250)' },
+            multiline: { type: 'boolean', description: 'Enable multiline matching' },
+        },
+        required: ['pattern'],
+    },
+    validateInput(input) { return input.pattern ? [] : ['pattern required']; },
+    async call(input) {
+        try {
+            const dir = path.resolve(input.path || '.');
+            const mode = input.output_mode || 'files_with_matches';
+            const limit = input.head_limit ?? 250;
+
+            // Pass regex and paths as process arguments. Joining these into a
+            // shell command would turn regex operators such as | into pipelines.
+            const args = [];
+            const useRg = hasRipgrep();
+            const command = useRg ? 'rg' : 'grep';
+
+            if (useRg) {
+                if (input['-i']) args.push('-i');
+                if (input.multiline) args.push('-U', '--multiline-dotall');
+
+                if (mode === 'files_with_matches') {
+                    args.push('-l');
+                } else if (mode === 'count') {
+                    args.push('-c');
+                } else {
+                    // content mode
+                    const showLineNumbers = input['-n'] !== false;
+                    if (showLineNumbers) args.push('-n');
+                }
+
+                const ctx = input['-C'] || input.context;
+                if (ctx && mode === 'content') args.push('-C', String(ctx));
+                if (input['-A'] && mode === 'content') args.push('-A', String(input['-A']));
+                if (input['-B'] && mode === 'content') args.push('-B', String(input['-B']));
+
+                if (input.glob) args.push('--glob', input.glob);
+                if (input.type) args.push('--type', input.type);
+
+                args.push('--', input.pattern, dir);
+            } else {
+                args.push('-r');
+                if (input['-i']) args.push('-i');
+
+                if (mode === 'files_with_matches') {
+                    args.push('-l');
+                } else if (mode === 'count') {
+                    args.push('-c');
+                } else {
+                    if (input['-n'] !== false) args.push('-n');
+                }
+
+                const ctx = input['-C'] || input.context;
+                if (ctx && mode === 'content') args.push('-C', String(ctx));
+                if (input['-A'] && mode === 'content') args.push('-A', String(input['-A']));
+                if (input['-B'] && mode === 'content') args.push('-B', String(input['-B']));
+
+                if (input.glob) args.push('--include', input.glob);
+
+                args.push('--', input.pattern, dir);
+            }
+
+            const result = spawnSync(command, args, {
+                encoding: 'utf-8',
+                maxBuffer: 10 * 1024 * 1024,
+                timeout: 30000,
+            });
+
+            if (result.error || (result.status !== 0 && result.status !== 1)) {
+                return 'No matches found.';
+            }
+
+            const output = (result.stdout || '').trim();
+            if (!output) return 'No matches found.';
+            if (limit <= 0) return output;
+            return output.split('\n').slice(0, limit).join('\n');
+        } catch {
+            return 'No matches found.';
+        }
+    },
+};
+
+let _hasRg = null;
+function hasRipgrep() {
+    if (_hasRg !== null) return _hasRg;
+    const result = spawnSync('rg', ['--version'], {
+        encoding: 'utf-8',
+        timeout: 5000,
+    });
+    _hasRg = !result.error && result.status === 0;
+    return _hasRg;
+}

package/src/tools/lint.mjs

@@ -0,0 +1,71 @@
+/**
+ * Lint Tool — run linters and type checkers.
+ *
+ * Always safe: read-only, no side effects (unless --fix).
+ * Agent should use this instead of shell("ruff check .").
+ *
+ * Execution: direct on host (no sandbox needed for read-only).
+ * With --fix: routes through sandbox if available.
+ */
+
+import { BashTool } from './bash.mjs';
+
+const LINTERS = {
+    ruff:    { cmd: (p, fix) => `ruff check ${fix ? '--fix ' : ''}${p}`, lang: 'python' },
+    mypy:    { cmd: (p) => `mypy ${p} --no-error-summary`, lang: 'python' },
+    pyright: { cmd: (p) => `pyright ${p}`, lang: 'python' },
+    pylint:  { cmd: (p) => `pylint ${p} --output-format=text`, lang: 'python' },
+    eslint:  { cmd: (p, fix) => `eslint ${fix ? '--fix ' : ''}${p}`, lang: 'javascript' },
+    tsc:     { cmd: () => 'tsc --noEmit', lang: 'typescript' },
+    flake8:  { cmd: (p) => `flake8 ${p}`, lang: 'python' },
+    biome:   { cmd: (p, fix) => `biome check ${fix ? '--fix ' : ''}${p}`, lang: 'javascript' },
+};
+
+export const LintTool = {
+    name: 'Lint',
+    description: 'Run a linter or type checker. Fast, read-only (unless --fix). Use this instead of Bash for code checking.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            tool: {
+                type: 'string',
+                enum: Object.keys(LINTERS),
+                description: 'Which linter to run',
+            },
+            path: {
+                type: 'string',
+                description: 'File or directory to lint (default: ".")',
+                default: '.',
+            },
+            fix: {
+                type: 'boolean',
+                description: 'Apply auto-fixes (default: false)',
+                default: false,
+            },
+        },
+        required: ['tool'],
+    },
+    validateInput(input) {
+        const errors = [];
+        if (!input.tool) errors.push('tool is required');
+        if (input.tool && !LINTERS[input.tool]) errors.push(`Unknown linter: ${input.tool}. Supported: ${Object.keys(LINTERS).join(', ')}`);
+        return errors;
+    },
+    async call(input) {
+        const linter = LINTERS[input.tool];
+        if (!linter) return `Unknown linter: ${input.tool}`;
+
+        const targetPath = input.path || '.';
+        const command = linter.cmd(targetPath, input.fix);
+
+        const result = await BashTool.call({
+            command,
+            timeout: 30000, // linters should be fast
+            description: `Lint: ${input.tool} ${targetPath}`,
+        });
+
+        return result;
+    },
+    // Metadata for execution policy
+    _executionPolicy: 'direct', // no sandbox for read-only; 'contained' if fix=true
+};

package/src/tools/ls.mjs

@@ -0,0 +1,58 @@
+/**
+ * LS Tool — directory listing with metadata.
+ */
+
+import fs from 'fs';
+import path from 'path';
+
+export const LsTool = {
+    name: 'LS',
+    description: 'List directory contents with file sizes and types.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            path: { type: 'string', description: 'Directory path to list (default: cwd)' },
+            all: { type: 'boolean', description: 'Include hidden files (default: false)' },
+        },
+        required: [],
+    },
+
+    validateInput() { return []; },
+
+    async call(input) {
+        const dirPath = path.resolve(input.path || '.');
+
+        try {
+            const entries = fs.readdirSync(dirPath, { withFileTypes: true });
+            const results = [];
+
+            for (const entry of entries) {
+                if (!input.all && entry.name.startsWith('.')) continue;
+
+                const fullPath = path.join(dirPath, entry.name);
+                let size = '';
+                try {
+                    const stat = fs.statSync(fullPath);
+                    size = entry.isDirectory() ? '' : formatSize(stat.size);
+                } catch {
+                    size = '?';
+                }
+
+                const type = entry.isDirectory() ? 'd' : entry.isSymbolicLink() ? 'l' : '-';
+                results.push(`${type} ${size.padStart(8)} ${entry.name}${entry.isDirectory() ? '/' : ''}`);
+            }
+
+            if (results.length === 0) return 'Empty directory';
+            return `${dirPath}:\n${results.join('\n')}`;
+        } catch (err) {
+            return `Error: ${err.message}`;
+        }
+    },
+};
+
+function formatSize(bytes) {
+    if (bytes < 1024) return `${bytes}B`;
+    if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)}K`;
+    if (bytes < 1024 * 1024 * 1024) return `${(bytes / (1024 * 1024)).toFixed(1)}M`;
+    return `${(bytes / (1024 * 1024 * 1024)).toFixed(1)}G`;
+}

package/src/tools/lsp.mjs

@@ -0,0 +1,115 @@
+/**
+ * LSP Tool — Language Server Protocol integration stub.
+ *
+ * Provides a tool interface for language server features:
+ * - diagnostics (errors/warnings)
+ * - completions
+ * - hover information
+ * - go-to-definition
+ * - references
+ *
+ * This is a stub implementation. Full LSP would spawn an actual language
+ * server process and communicate via JSON-RPC.
+ */
+
+import { execSync } from 'child_process';
+import path from 'path';
+
+export const LspTool = {
+    name: 'LSP',
+    description: 'Query language server for diagnostics, completions, hover, and definitions.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            action: {
+                type: 'string',
+                enum: ['diagnostics', 'completions', 'hover', 'definition', 'references'],
+                description: 'LSP action to perform',
+            },
+            file: {
+                type: 'string',
+                description: 'File path to query',
+            },
+            line: {
+                type: 'number',
+                description: 'Line number (0-based)',
+            },
+            character: {
+                type: 'number',
+                description: 'Character position (0-based)',
+            },
+            language: {
+                type: 'string',
+                description: 'Language ID (e.g., "typescript", "python")',
+            },
+        },
+        required: ['action', 'file'],
+    },
+
+    validateInput(input) {
+        const errors = [];
+        if (!input.action) errors.push('action is required');
+        if (!input.file) errors.push('file is required');
+        return errors;
+    },
+
+    async call(input) {
+        const filePath = path.resolve(input.file);
+        const action = input.action;
+
+        switch (action) {
+            case 'diagnostics':
+                return getDiagnostics(filePath, input.language);
+            case 'completions':
+                return `[LSP stub] Completions at ${filePath}:${input.line}:${input.character} not yet implemented.`;
+            case 'hover':
+                return `[LSP stub] Hover at ${filePath}:${input.line}:${input.character} not yet implemented.`;
+            case 'definition':
+                return `[LSP stub] Go-to-definition at ${filePath}:${input.line}:${input.character} not yet implemented.`;
+            case 'references':
+                return `[LSP stub] Find references at ${filePath}:${input.line}:${input.character} not yet implemented.`;
+            default:
+                return `Unknown LSP action: ${action}`;
+        }
+    },
+};
+
+function getDiagnostics(filePath, language) {
+    const ext = path.extname(filePath);
+    const lang = language || extToLanguage(ext);
+
+    try {
+        switch (lang) {
+            case 'typescript':
+            case 'javascript': {
+                const result = execSync(
+                    `npx tsc --noEmit --pretty false "${filePath}" 2>&1 || true`,
+                    { encoding: 'utf-8', timeout: 15000 }
+                );
+                return result.trim() || 'No diagnostics.';
+            }
+            case 'python': {
+                const result = execSync(
+                    `python3 -m py_compile "${filePath}" 2>&1 || true`,
+                    { encoding: 'utf-8', timeout: 10000 }
+                );
+                return result.trim() || 'No diagnostics.';
+            }
+            default:
+                return `[LSP stub] No diagnostic provider for language: ${lang}`;
+        }
+    } catch (err) {
+        return `Diagnostics error: ${err.message}`;
+    }
+}
+
+function extToLanguage(ext) {
+    const map = {
+        '.ts': 'typescript', '.tsx': 'typescript',
+        '.js': 'javascript', '.jsx': 'javascript', '.mjs': 'javascript',
+        '.py': 'python',
+        '.rs': 'rust', '.go': 'go', '.java': 'java',
+        '.rb': 'ruby', '.php': 'php', '.cs': 'csharp',
+    };
+    return map[ext] || 'unknown';
+}

package/src/tools/multi-edit.mjs

@@ -0,0 +1,94 @@
+/**
+ * MultiEdit Tool — apply multiple edits to one or more files atomically.
+ *
+ * Each edit is an { file_path, old_string, new_string } triple.
+ * All edits are validated before any are applied.
+ */
+
+import fs from 'fs';
+import path from 'path';
+
+export const MultiEditTool = {
+    name: 'MultiEdit',
+    description: 'Apply multiple file edits in a single operation. All edits are validated first.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            edits: {
+                type: 'array',
+                items: {
+                    type: 'object',
+                    properties: {
+                        file_path: { type: 'string' },
+                        old_string: { type: 'string' },
+                        new_string: { type: 'string' },
+                    },
+                    required: ['file_path', 'old_string', 'new_string'],
+                },
+                description: 'Array of edits to apply',
+            },
+        },
+        required: ['edits'],
+    },
+
+    validateInput(input) {
+        const errors = [];
+        if (!input.edits || !Array.isArray(input.edits)) {
+            errors.push('edits must be an array');
+            return errors;
+        }
+        for (let i = 0; i < input.edits.length; i++) {
+            const e = input.edits[i];
+            if (!e.file_path) errors.push(`edit[${i}]: file_path required`);
+            if (!e.old_string) errors.push(`edit[${i}]: old_string required`);
+            if (e.old_string === e.new_string) errors.push(`edit[${i}]: old_string must differ from new_string`);
+        }
+        return errors;
+    },
+
+    async call(input) {
+        // Phase 1: Validate all edits
+        const fileContents = new Map();
+        const errors = [];
+
+        for (let i = 0; i < input.edits.length; i++) {
+            const edit = input.edits[i];
+            const filePath = path.resolve(edit.file_path);
+
+            if (!fileContents.has(filePath)) {
+                try {
+                    fileContents.set(filePath, fs.readFileSync(filePath, 'utf-8'));
+                } catch (err) {
+                    errors.push(`edit[${i}]: cannot read ${filePath}: ${err.message}`);
+                    continue;
+                }
+            }
+
+            let content = fileContents.get(filePath);
+            if (!content.includes(edit.old_string)) {
+                errors.push(`edit[${i}]: old_string not found in ${edit.file_path}`);
+            }
+        }
+
+        if (errors.length > 0) {
+            return `Validation failed:\n${errors.join('\n')}`;
+        }
+
+        // Phase 2: Apply all edits
+        const applied = [];
+        for (const edit of input.edits) {
+            const filePath = path.resolve(edit.file_path);
+            let content = fileContents.get(filePath);
+            content = content.replace(edit.old_string, edit.new_string);
+            fileContents.set(filePath, content);
+        }
+
+        // Phase 3: Write all files
+        for (const [filePath, content] of fileContents) {
+            fs.writeFileSync(filePath, content);
+            applied.push(filePath);
+        }
+
+        return `Applied ${input.edits.length} edits to ${applied.length} file(s):\n${applied.join('\n')}`;
+    },
+};

package/src/tools/notebook-edit.mjs

@@ -0,0 +1,96 @@
+/**
+ * NotebookEdit Tool — edit Jupyter notebook (.ipynb) cells.
+ *
+ * Supports operations:
+ * - insert: add a new cell at a position
+ * - replace: replace content of an existing cell
+ * - delete: remove a cell
+ */
+
+import fs from 'fs';
+import path from 'path';
+
+export const NotebookEditTool = {
+    name: 'NotebookEdit',
+    description: 'Edit Jupyter notebook cells. Supports insert, replace, and delete operations.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            notebook_path: { type: 'string', description: 'Path to the .ipynb file' },
+            operation: {
+                type: 'string',
+                enum: ['insert', 'replace', 'delete'],
+                description: 'Operation to perform',
+            },
+            cell_index: { type: 'number', description: 'Cell index (0-based)' },
+            cell_type: {
+                type: 'string',
+                enum: ['code', 'markdown', 'raw'],
+                description: 'Cell type (for insert/replace)',
+            },
+            source: { type: 'string', description: 'Cell content (for insert/replace)' },
+        },
+        required: ['notebook_path', 'operation', 'cell_index'],
+    },
+
+    validateInput(input) {
+        const errors = [];
+        if (!input.notebook_path) errors.push('notebook_path required');
+        if (!['insert', 'replace', 'delete'].includes(input.operation)) {
+            errors.push('operation must be insert, replace, or delete');
+        }
+        if (typeof input.cell_index !== 'number') errors.push('cell_index must be a number');
+        if (input.operation !== 'delete' && !input.source) errors.push('source required for insert/replace');
+        return errors;
+    },
+
+    async call(input) {
+        const filePath = path.resolve(input.notebook_path);
+
+        try {
+            const raw = fs.readFileSync(filePath, 'utf-8');
+            const notebook = JSON.parse(raw);
+
+            if (!notebook.cells || !Array.isArray(notebook.cells)) {
+                return 'Error: invalid notebook format (no cells array)';
+            }
+
+            const cellType = input.cell_type || 'code';
+            const sourceLines = (input.source || '').split('\n').map(l => l + '\n');
+
+            switch (input.operation) {
+                case 'insert': {
+                    const newCell = {
+                        cell_type: cellType,
+                        metadata: {},
+                        source: sourceLines,
+                        ...(cellType === 'code' ? { outputs: [], execution_count: null } : {}),
+                    };
+                    const idx = Math.min(input.cell_index, notebook.cells.length);
+                    notebook.cells.splice(idx, 0, newCell);
+                    break;
+                }
+                case 'replace': {
+                    if (input.cell_index >= notebook.cells.length) {
+                        return `Error: cell_index ${input.cell_index} out of range (${notebook.cells.length} cells)`;
+                    }
+                    notebook.cells[input.cell_index].source = sourceLines;
+                    if (cellType) notebook.cells[input.cell_index].cell_type = cellType;
+                    break;
+                }
+                case 'delete': {
+                    if (input.cell_index >= notebook.cells.length) {
+                        return `Error: cell_index ${input.cell_index} out of range (${notebook.cells.length} cells)`;
+                    }
+                    notebook.cells.splice(input.cell_index, 1);
+                    break;
+                }
+            }
+
+            fs.writeFileSync(filePath, JSON.stringify(notebook, null, 1));
+            return `Notebook updated: ${input.operation} at cell ${input.cell_index} (${notebook.cells.length} cells total)`;
+        } catch (err) {
+            return `Error: ${err.message}`;
+        }
+    },
+};