@axplusb/kepler 0.0.1 → 1.0.0

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}`;
+        }
+    },
+};

package/src/tools/read-mcp-resource.mjs

@@ -0,0 +1,57 @@
+/**
+ * ReadMcpResource Tool — read a resource from an MCP server.
+ *
+ * MCP resources are identified by URI and can contain text, JSON,
+ * or binary data. This tool reads text/JSON resources.
+ */
+
+export const ReadMcpResourceTool = {
+    name: 'ReadMcpResource',
+    description: 'Read a resource from an MCP server by URI.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            uri: {
+                type: 'string',
+                description: 'Resource URI (e.g., "file:///path" or "mcp://server/resource")',
+            },
+            server: {
+                type: 'string',
+                description: 'MCP server name (if URI does not specify)',
+            },
+        },
+        required: ['uri'],
+    },
+
+    // Set by the MCP integration layer
+    _mcpClients: null,
+
+    validateInput(input) {
+        return input.uri ? [] : ['uri is required'];
+    },
+
+    async call(input) {
+        if (!this._mcpClients || this._mcpClients.length === 0) {
+            return 'No MCP servers connected. Configure MCP servers in settings.';
+        }
+
+        for (const client of this._mcpClients) {
+            try {
+                const result = await client.readResource(input.uri);
+                if (result) {
+                    if (typeof result === 'string') return result;
+                    if (result.contents && Array.isArray(result.contents)) {
+                        return result.contents
+                            .map(c => c.text || JSON.stringify(c))
+                            .join('\n');
+                    }
+                    return JSON.stringify(result, null, 2);
+                }
+            } catch {
+                // Try next client
+            }
+        }
+
+        return `Resource not found: ${input.uri}`;
+    },
+};

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,132 @@
+/**
+ * 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';
+import { ContextRetriever } from '../context/retriever.mjs';
+
+// Tools that modify files — trigger index update after success
+const WRITE_TOOLS = new Set(['Edit', 'Write', 'MultiEdit']);
+
+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);
+
+            // After successful file writes, update BM25 index incrementally
+            // so search stays fresh within the session (~5-50ms, non-blocking)
+            if (WRITE_TOOLS.has(name) && typeof result === 'string' && !result.startsWith('Error')) {
+                const filePath = input.file_path;
+                if (filePath) {
+                    try {
+                        const retriever = new ContextRetriever();
+                        retriever.updateFile(filePath);
+                    } catch { /* index update is best-effort */ }
+                }
+            }
+
+            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;
+    },
+};