@bahulam/code 2.6.0

package/src/tools/read.mjs

@@ -0,0 +1,138 @@
+/**
+ * Read Tool — matches Claude Code's exact behavior.
+ *
+ * Features:
+ * - pages parameter for PDF files
+ * - Binary file detection
+ * - Default 2000 line limit
+ * - Line number prefix (cat -n format)
+ * - Graceful file not found handling
+ * - Tracks read files for Edit/Write verification
+ */
+import fs from 'fs';
+import path from 'path';
+
+const DEFAULT_LIMIT = 2000;
+
+// Track which files have been read (used by Edit and Write tools)
+const readFiles = new Set();
+
+export function hasBeenRead(filePath) {
+    return readFiles.has(path.resolve(filePath));
+}
+
+export function markRead(filePath) {
+    readFiles.add(path.resolve(filePath));
+}
+
+// Binary detection: check for null bytes in first 8KB
+function isBinary(buffer) {
+    const len = Math.min(buffer.length, 8192);
+    for (let i = 0; i < len; i++) {
+        if (buffer[i] === 0) return true;
+    }
+    return false;
+}
+
+export const ReadTool = {
+    name: 'Read',
+    description: 'Read a file from the local filesystem.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            file_path: { type: 'string', description: 'Absolute path to the file' },
+            offset: { type: 'number', description: 'Line number to start reading from (0-based)' },
+            limit: { type: 'number', description: 'Number of lines to read (default 2000)' },
+            pages: { type: 'string', description: 'Page range for PDF files (e.g. "1-5")' },
+        },
+        required: ['file_path'],
+    },
+    validateInput(input) {
+        const errors = [];
+        if (!input.file_path) errors.push('file_path is required');
+        return errors;
+    },
+    async call(input) {
+        const filePath = path.resolve(input.file_path);
+
+        // Check existence
+        if (!fs.existsSync(filePath)) {
+            return `Error: File not found: ${filePath}`;
+        }
+
+        // Check if directory
+        const stat = fs.statSync(filePath);
+        if (stat.isDirectory()) {
+            return `Error: ${filePath} is a directory, not a file. Use Bash with ls to list directory contents.`;
+        }
+
+        // PDF handling
+        if (filePath.endsWith('.pdf')) {
+            return readPdf(filePath, input.pages);
+        }
+
+        // Binary detection
+        try {
+            const fd = fs.openSync(filePath, 'r');
+            const buf = Buffer.alloc(8192);
+            const bytesRead = fs.readSync(fd, buf, 0, 8192, 0);
+            fs.closeSync(fd);
+            if (isBinary(buf.subarray(0, bytesRead))) {
+                return `Error: ${filePath} appears to be a binary file. Cannot display binary content.`;
+            }
+        } catch (e) {
+            return `Error: ${e.message}`;
+        }
+
+        try {
+            const content = fs.readFileSync(filePath, 'utf-8');
+            const lines = content.split('\n');
+            const start = input.offset || 0;
+            const limit = input.limit || DEFAULT_LIMIT;
+            const end = Math.min(start + limit, lines.length);
+
+            // Track as read
+            readFiles.add(filePath);
+
+            // Handle empty files
+            if (content === '' || (content.length === 0)) {
+                return '[File exists but is empty]';
+            }
+
+            const output = lines
+                .slice(start, end)
+                .map((l, i) => `${start + i + 1}\t${l}`)
+                .join('\n');
+
+            if (end < lines.length) {
+                return output + `\n\n[File has ${lines.length} lines total. Showing lines ${start + 1}-${end}. Use offset/limit for more.]`;
+            }
+
+            return output;
+        } catch (e) {
+            return `Error: ${e.message}`;
+        }
+    },
+};
+
+function readPdf(filePath, pages) {
+    // PDF reading requires external tools; provide a best-effort
+    // text extraction using a simple approach
+    try {
+        const { execSync } = await_import_child_process();
+        const pageArg = pages ? `-f ${pages.split('-')[0]} -l ${pages.split('-').pop()}` : '-f 1 -l 20';
+        const text = execSync(`pdftotext ${pageArg} "${filePath}" - 2>/dev/null`, {
+            encoding: 'utf-8',
+            timeout: 30000,
+            maxBuffer: 1024 * 1024,
+        });
+        return text || `[PDF file at ${filePath} — could not extract text. Use a PDF viewer.]`;
+    } catch {
+        return `[PDF file at ${filePath} — pdftotext not available. Install poppler-utils for PDF support.]`;
+    }
+}
+
+// Lazy import helper
+function await_import_child_process() {
+    return require('child_process');
+}

package/src/tools/registry.mjs

@@ -0,0 +1,116 @@
+/**
+ * Tool Registry — validateInput/call interface.
+ * Mirrors Claude Code's tool dispatch system.
+ * Registers all 25+ built-in tools.
+ */
+
+import { BashTool } from './bash.mjs';
+import { ReadTool } from './read.mjs';
+import { EditTool } from './edit.mjs';
+import { WriteTool } from './write.mjs';
+import { GlobTool } from './glob.mjs';
+import { GrepTool } from './grep.mjs';
+import { AgentTool } from './agent.mjs';
+import { WebFetchTool } from './web-fetch.mjs';
+import { WebSearchTool } from './web-search.mjs';
+import { TodoWriteTool } from './todo-write.mjs';
+import { NotebookEditTool } from './notebook-edit.mjs';
+import { MultiEditTool } from './multi-edit.mjs';
+import { LsTool } from './ls.mjs';
+import { ToolSearchTool } from './tool-search.mjs';
+import { AskUserTool } from './ask-user.mjs';
+import { EnterWorktreeTool } from './enter-worktree.mjs';
+import { ExitWorktreeTool } from './exit-worktree.mjs';
+import { SkillTool } from './skill.mjs';
+import { SendMessageTool } from './send-message.mjs';
+import { RemoteTriggerTool } from './remote-trigger.mjs';
+import { CronCreateTool } from './cron-create.mjs';
+import { CronDeleteTool } from './cron-delete.mjs';
+import { CronListTool } from './cron-list.mjs';
+import { LspTool } from './lsp.mjs';
+import { ReadMcpResourceTool } from './read-mcp-resource.mjs';
+
+const BUILTIN_TOOLS = [
+    BashTool,
+    ReadTool,
+    EditTool,
+    WriteTool,
+    GlobTool,
+    GrepTool,
+    AgentTool,
+    WebFetchTool,
+    WebSearchTool,
+    TodoWriteTool,
+    NotebookEditTool,
+    MultiEditTool,
+    LsTool,
+    ToolSearchTool,
+    AskUserTool,
+    EnterWorktreeTool,
+    ExitWorktreeTool,
+    SkillTool,
+    SendMessageTool,
+    RemoteTriggerTool,
+    CronCreateTool,
+    CronDeleteTool,
+    CronListTool,
+    LspTool,
+    ReadMcpResourceTool,
+];
+
+export function createToolRegistry() {
+    const tools = new Map();
+    for (const Tool of BUILTIN_TOOLS) {
+        tools.set(Tool.name, Tool);
+    }
+
+    const registry = {
+        list() {
+            return [...tools.values()].map(t => ({
+                name: t.name,
+                description: t.description,
+                input_schema: t.inputSchema,
+            }));
+        },
+
+        async call(name, input) {
+            const tool = tools.get(name);
+            if (!tool) throw new Error(`Unknown tool: ${name}`);
+            const errors = tool.validateInput?.(input) || [];
+            if (errors.length > 0) return `Validation error: ${errors.join(', ')}`;
+            const result = await tool.call(input);
+
+            return result;
+        },
+
+        register(tool) {
+            tools.set(tool.name, tool);
+        },
+
+        get(name) {
+            return tools.get(name);
+        },
+
+        has(name) {
+            return tools.has(name);
+        },
+
+        registerMcpTools(mcpTools, callFn) {
+            ToolSearchTool._mcpTools = mcpTools;
+
+            for (const mcpTool of mcpTools) {
+                const wrapper = {
+                    name: mcpTool.name,
+                    description: mcpTool.description || '',
+                    inputSchema: mcpTool.inputSchema || { type: 'object', properties: {} },
+                    validateInput() { return []; },
+                    async call(input) { return callFn(mcpTool.name, input); },
+                };
+                tools.set(mcpTool.name, wrapper);
+            }
+        },
+    };
+
+    ToolSearchTool._registry = registry;
+    return registry;
+}

package/src/tools/remote-trigger.mjs

@@ -0,0 +1,84 @@
+/**
+ * RemoteTrigger Tool — trigger remote execution of a task.
+ *
+ * Sends a task to a remote agent endpoint for execution.
+ * Used for distributed agent workflows and scheduled tasks.
+ */
+
+export const RemoteTriggerTool = {
+    name: 'RemoteTrigger',
+    description: 'Trigger remote execution of a task on a remote agent.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            endpoint: {
+                type: 'string',
+                description: 'Remote agent endpoint URL',
+            },
+            task: {
+                type: 'string',
+                description: 'Task description or prompt to execute remotely',
+            },
+            timeout: {
+                type: 'number',
+                description: 'Timeout in milliseconds (default: 300000)',
+            },
+            async: {
+                type: 'boolean',
+                description: 'Fire-and-forget mode (default: false)',
+            },
+        },
+        required: ['task'],
+    },
+
+    validateInput(input) {
+        return input.task ? [] : ['task is required'];
+    },
+
+    async call(input) {
+        const endpoint = input.endpoint || process.env.REMOTE_AGENT_URL;
+        if (!endpoint) {
+            return 'No remote endpoint configured. Set endpoint parameter or REMOTE_AGENT_URL env var.';
+        }
+
+        try {
+            const controller = new AbortController();
+            const timeout = setTimeout(
+                () => controller.abort(),
+                input.timeout || 300000
+            );
+
+            const res = await fetch(endpoint, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    ...(process.env.REMOTE_AGENT_TOKEN && {
+                        Authorization: `Bearer ${process.env.REMOTE_AGENT_TOKEN}`,
+                    }),
+                },
+                body: JSON.stringify({
+                    task: input.task,
+                    async: input.async || false,
+                }),
+                signal: controller.signal,
+            });
+
+            clearTimeout(timeout);
+
+            if (!res.ok) {
+                return `Remote error: HTTP ${res.status} ${res.statusText}`;
+            }
+
+            if (input.async) {
+                return `Task submitted to ${endpoint} (async mode)`;
+            }
+
+            const data = await res.json();
+            return typeof data.result === 'string'
+                ? data.result
+                : JSON.stringify(data, null, 2);
+        } catch (err) {
+            return `Remote trigger error: ${err.message}`;
+        }
+    },
+};

package/src/tools/send-message.mjs

@@ -0,0 +1,64 @@
+/**
+ * SendMessage Tool — send a message to a teammate agent.
+ *
+ * Used in multi-agent (teams) mode to communicate between agents.
+ * Messages are stored in a shared message queue.
+ */
+
+const messageQueue = new Map(); // agentId -> messages[]
+
+export const SendMessageTool = {
+    name: 'SendMessage',
+    description: 'Send a message to a teammate agent in multi-agent mode.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            to: {
+                type: 'string',
+                description: 'Target agent ID or name',
+            },
+            content: {
+                type: 'string',
+                description: 'Message content to send',
+            },
+            type: {
+                type: 'string',
+                enum: ['request', 'response', 'notification', 'handoff'],
+                description: 'Message type (default: notification)',
+            },
+        },
+        required: ['to', 'content'],
+    },
+
+    validateInput(input) {
+        const errors = [];
+        if (!input.to) errors.push('to is required');
+        if (!input.content) errors.push('content is required');
+        return errors;
+    },
+
+    async call(input) {
+        const msg = {
+            id: `msg_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`,
+            from: process.env.AGENT_ID || 'main',
+            to: input.to,
+            type: input.type || 'notification',
+            content: input.content,
+            timestamp: new Date().toISOString(),
+        };
+
+        if (!messageQueue.has(input.to)) {
+            messageQueue.set(input.to, []);
+        }
+        messageQueue.get(input.to).push(msg);
+
+        return `Message sent to "${input.to}" (id: ${msg.id}, type: ${msg.type})`;
+    },
+
+    // Helper: receive messages for an agent
+    receive(agentId) {
+        const messages = messageQueue.get(agentId) || [];
+        messageQueue.set(agentId, []);
+        return messages;
+    },
+};

package/src/tools/skill.mjs

@@ -0,0 +1,52 @@
+/**
+ * Skill Tool — invoke a loaded skill by name.
+ *
+ * Skills are loaded from .claude/skills/{name}/SKILL.md and provide
+ * specialized prompts injected into the conversation.
+ */
+
+export const SkillTool = {
+    name: 'Skill',
+    description: 'Invoke a skill within the conversation. Skills provide specialized capabilities.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            skill: {
+                type: 'string',
+                description: 'The skill name to invoke (e.g., "commit", "review-pr")',
+            },
+            args: {
+                type: 'string',
+                description: 'Optional arguments for the skill',
+            },
+        },
+        required: ['skill'],
+    },
+
+    // Set by skills loader
+    _skillsLoader: null,
+
+    validateInput(input) {
+        return input.skill ? [] : ['skill name is required'];
+    },
+
+    async call(input) {
+        if (!this._skillsLoader) {
+            return 'Skills system not initialized. No skills available.';
+        }
+
+        try {
+            const skill = this._skillsLoader.get(input.skill);
+            if (!skill) {
+                const available = this._skillsLoader.list();
+                const names = available.map(s => s.name).join(', ');
+                return `Unknown skill: "${input.skill}". Available: ${names || 'none'}`;
+            }
+
+            const result = await this._skillsLoader.run(input.skill, input.args);
+            return result;
+        } catch (err) {
+            return `Skill error: ${err.message}`;
+        }
+    },
+};

package/src/tools/test-runner.mjs

@@ -0,0 +1,49 @@
+/**
+ * Test Runner Tool — run test suites in contained environment.
+ *
+ * Executes in .venv/sandbox when available.
+ * Agent should use this instead of shell("pytest tests/").
+ *
+ * Execution: contained (.venv + OS sandbox if available).
+ */
+
+import { BashTool } from './bash.mjs';
+
+export const TestRunnerTool = {
+    name: 'TestRunner',
+    description: 'Run tests. Executes in project environment. Use this instead of Bash for pytest, jest, vitest, cargo test, go test.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            command: {
+                type: 'string',
+                description: 'Test command (e.g., "pytest tests/test_auth.py -v", "npm test", "cargo test")',
+            },
+            timeout: {
+                type: 'number',
+                description: 'Timeout in ms (default: 120000, max: 300000)',
+                default: 120000,
+            },
+        },
+        required: ['command'],
+    },
+    validateInput(input) {
+        const errors = [];
+        if (!input.command) errors.push('command is required');
+        return errors;
+    },
+    async call(input) {
+        const timeout = Math.min(input.timeout || 120000, 300000);
+
+        // TODO: Wire sandbox.mjs here when sandbox execution is ready
+        // For now, execute directly (same as Bash but with metadata)
+        const result = await BashTool.call({
+            command: input.command,
+            timeout,
+            description: `Test: ${input.command.slice(0, 60)}`,
+        });
+
+        return result;
+    },
+    _executionPolicy: 'contained',
+};

package/src/tools/todo-write.mjs

@@ -0,0 +1,68 @@
+/**
+ * TodoWrite Tool — in-memory task management.
+ *
+ * Maintains a task list that the agent can use to track work items.
+ * State persists for the duration of the session.
+ */
+
+const todos = [];
+let nextId = 1;
+
+export const TodoWriteTool = {
+    name: 'TodoWrite',
+    description: 'Manage a task list. Supports add, update, complete, and list operations.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            todos: {
+                type: 'array',
+                items: {
+                    type: 'object',
+                    properties: {
+                        id: { type: 'string', description: 'Task ID (auto-assigned if adding)' },
+                        content: { type: 'string', description: 'Task description' },
+                        status: {
+                            type: 'string',
+                            enum: ['pending', 'in_progress', 'completed'],
+                            description: 'Task status',
+                        },
+                        priority: {
+                            type: 'string',
+                            enum: ['high', 'medium', 'low'],
+                            description: 'Task priority',
+                        },
+                    },
+                    required: ['content', 'status'],
+                },
+                description: 'Array of todo items to write (replaces all todos)',
+            },
+        },
+        required: ['todos'],
+    },
+
+    validateInput(input) {
+        if (!input.todos || !Array.isArray(input.todos)) return ['todos must be an array'];
+        return [];
+    },
+
+    async call(input) {
+        // Replace entire todo list (matches Claude Code behavior)
+        todos.length = 0;
+        nextId = 1;
+
+        for (const item of input.todos) {
+            todos.push({
+                id: item.id || String(nextId++),
+                content: item.content,
+                status: item.status || 'pending',
+                priority: item.priority || 'medium',
+            });
+        }
+
+        const summary = todos.map(t =>
+            `[${t.status === 'completed' ? 'x' : t.status === 'in_progress' ? '~' : ' '}] ${t.id}. ${t.content} (${t.priority})`
+        ).join('\n');
+
+        return `Updated ${todos.length} todos:\n${summary}`;
+    },
+};

package/src/tools/tool-search.mjs

@@ -0,0 +1,77 @@
+/**
+ * ToolSearch Tool — deferred tool loading / search.
+ *
+ * Allows the agent to discover and load tools that are not
+ * in the default set. Supports searching MCP servers and
+ * deferred tool definitions.
+ */
+
+export const ToolSearchTool = {
+    name: 'ToolSearch',
+    description: 'Search for and load deferred tools by name or keyword.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            query: {
+                type: 'string',
+                description: 'Tool name or search keywords',
+            },
+            max_results: {
+                type: 'number',
+                description: 'Maximum results to return (default: 5)',
+            },
+        },
+        required: ['query'],
+    },
+
+    // Will be set by the registry when MCP tools are available
+    _mcpTools: [],
+    _registry: null,
+
+    validateInput(input) {
+        return input.query ? [] : ['query is required'];
+    },
+
+    async call(input) {
+        const query = input.query.toLowerCase();
+        const maxResults = input.max_results || 5;
+        const matches = [];
+
+        // Search MCP tools
+        for (const tool of this._mcpTools) {
+            const name = (tool.name || '').toLowerCase();
+            const desc = (tool.description || '').toLowerCase();
+            if (name.includes(query) || desc.includes(query)) {
+                matches.push({
+                    name: tool.name,
+                    description: tool.description,
+                    source: 'mcp',
+                });
+            }
+        }
+
+        // Search registry tools
+        if (this._registry) {
+            for (const tool of this._registry.list()) {
+                const name = (tool.name || '').toLowerCase();
+                const desc = (tool.description || '').toLowerCase();
+                if (name.includes(query) || desc.includes(query)) {
+                    matches.push({
+                        name: tool.name,
+                        description: tool.description,
+                        source: 'builtin',
+                    });
+                }
+            }
+        }
+
+        if (matches.length === 0) {
+            return `No tools found matching "${input.query}"`;
+        }
+
+        const limited = matches.slice(0, maxResults);
+        return limited.map(m =>
+            `[${m.source}] ${m.name}: ${m.description}`
+        ).join('\n');
+    },
+};