@defai.digital/ax-cli 0.0.34

package/dist/hooks/use-input-history.js

@@ -0,0 +1,69 @@
+import { useState, useCallback } from "react";
+const MAX_HISTORY_SIZE = 1000;
+export function useInputHistory() {
+    const [history, setHistory] = useState([]);
+    const [currentIndex, setCurrentIndex] = useState(-1);
+    const [originalInput, setOriginalInput] = useState("");
+    const addToHistory = useCallback((input) => {
+        if (input.trim() && !history.includes(input.trim())) {
+            setHistory(prev => {
+                const newHistory = [...prev, input.trim()];
+                // Keep only the last MAX_HISTORY_SIZE entries to prevent unbounded growth
+                if (newHistory.length > MAX_HISTORY_SIZE) {
+                    return newHistory.slice(-MAX_HISTORY_SIZE);
+                }
+                return newHistory;
+            });
+        }
+        setCurrentIndex(-1);
+        setOriginalInput("");
+    }, [history]);
+    const navigateHistory = useCallback((direction) => {
+        if (history.length === 0)
+            return null;
+        let newIndex;
+        if (direction === "up") {
+            if (currentIndex === -1) {
+                newIndex = history.length - 1;
+            }
+            else {
+                newIndex = Math.max(0, currentIndex - 1);
+            }
+        }
+        else {
+            if (currentIndex === -1) {
+                return null;
+            }
+            else if (currentIndex === history.length - 1) {
+                newIndex = -1;
+                return originalInput;
+            }
+            else {
+                newIndex = Math.min(history.length - 1, currentIndex + 1);
+            }
+        }
+        setCurrentIndex(newIndex);
+        return newIndex === -1 ? originalInput : history[newIndex];
+    }, [history, currentIndex, originalInput]);
+    const getCurrentHistoryIndex = useCallback(() => currentIndex, [currentIndex]);
+    const resetHistory = useCallback(() => {
+        setHistory([]);
+        setCurrentIndex(-1);
+        setOriginalInput("");
+    }, []);
+    const isNavigatingHistory = useCallback(() => currentIndex !== -1, [currentIndex]);
+    const setOriginalInputCallback = useCallback((input) => {
+        if (currentIndex === -1) {
+            setOriginalInput(input);
+        }
+    }, [currentIndex]);
+    return {
+        addToHistory,
+        navigateHistory,
+        getCurrentHistoryIndex,
+        resetHistory,
+        isNavigatingHistory,
+        setOriginalInput: setOriginalInputCallback,
+    };
+}
+//# sourceMappingURL=use-input-history.js.map

package/dist/hooks/use-input-history.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-input-history.js","sourceRoot":"","sources":["../../src/hooks/use-input-history.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AAW9C,MAAM,gBAAgB,GAAG,IAAI,CAAC;AAE9B,MAAM,UAAU,eAAe;IAC7B,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,QAAQ,CAAW,EAAE,CAAC,CAAC;IACrD,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACrD,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEvD,MAAM,YAAY,GAAG,WAAW,CAAC,CAAC,KAAa,EAAE,EAAE;QACjD,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YACpD,UAAU,CAAC,IAAI,CAAC,EAAE;gBAChB,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC3C,0EAA0E;gBAC1E,IAAI,UAAU,CAAC,MAAM,GAAG,gBAAgB,EAAE,CAAC;oBACzC,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,gBAAgB,CAAC,CAAC;gBAC7C,CAAC;gBACD,OAAO,UAAU,CAAC;YACpB,CAAC,CAAC,CAAC;QACL,CAAC;QACD,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;QACpB,gBAAgB,CAAC,EAAE,CAAC,CAAC;IACvB,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAEd,MAAM,eAAe,GAAG,WAAW,CAAC,CAAC,SAAwB,EAAiB,EAAE;QAC9E,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAEtC,IAAI,QAAgB,CAAC;QAErB,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;YACvB,IAAI,YAAY,KAAK,CAAC,CAAC,EAAE,CAAC;gBACxB,QAAQ,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;YAChC,CAAC;iBAAM,CAAC;gBACN,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,YAAY,KAAK,CAAC,CAAC,EAAE,CAAC;gBACxB,OAAO,IAAI,CAAC;YACd,CAAC;iBAAM,IAAI,YAAY,KAAK,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/C,QAAQ,GAAG,CAAC,CAAC,CAAC;gBACd,OAAO,aAAa,CAAC;YACvB,CAAC;iBAAM,CAAC;gBACN,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;QAED,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC1B,OAAO,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC7D,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,EAAE,aAAa,CAAC,CAAC,CAAC;IAE3C,MAAM,sBAAsB,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,YAAY,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;IAE/E,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE;QACpC,UAAU,CAAC,EAAE,CAAC,CAAC;QACf,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;QACpB,gBAAgB,CAAC,EAAE,CAAC,CAAC;IACvB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,mBAAmB,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;IAEnF,MAAM,wBAAwB,GAAG,WAAW,CAAC,CAAC,KAAa,EAAE,EAAE;QAC7D,IAAI,YAAY,KAAK,CAAC,CAAC,EAAE,CAAC;YACxB,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;IAEnB,OAAO;QACL,YAAY;QACZ,eAAe;QACf,sBAAsB;QACtB,YAAY;QACZ,mBAAmB;QACnB,gBAAgB,EAAE,wBAAwB;KAC3C,CAAC;AACJ,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/dist/index.js

@@ -0,0 +1,339 @@
+#!/usr/bin/env node
+import React from "react";
+import { render } from "ink";
+import { program } from "commander";
+import * as dotenv from "dotenv";
+import { GrokAgent } from "./agent/grok-agent.js";
+import ChatInterface from "./ui/components/chat-interface.js";
+import { getSettingsManager } from "./utils/settings-manager.js";
+import { ConfirmationService } from "./utils/confirmation-service.js";
+import { createMCPCommand } from "./commands/mcp.js";
+// Load environment variables
+dotenv.config();
+// Disable default SIGINT handling to let Ink handle Ctrl+C
+// We'll handle exit through the input system instead
+process.on("SIGTERM", () => {
+    // Restore terminal to normal mode before exit
+    if (process.stdin.isTTY && process.stdin.setRawMode) {
+        try {
+            process.stdin.setRawMode(false);
+        }
+        catch (e) {
+            // Ignore errors when setting raw mode
+        }
+    }
+    console.log("\nGracefully shutting down...");
+    process.exit(0);
+});
+// Handle uncaught exceptions to prevent hanging
+process.on("uncaughtException", (error) => {
+    console.error("Uncaught exception:", error);
+    process.exit(1);
+});
+process.on("unhandledRejection", (reason, promise) => {
+    console.error("Unhandled rejection at:", promise, "reason:", reason);
+    process.exit(1);
+});
+// Ensure user settings are initialized
+function ensureUserSettingsDirectory() {
+    try {
+        const manager = getSettingsManager();
+        // This will create default settings if they don't exist
+        manager.loadUserSettings();
+    }
+    catch (error) {
+        // Silently ignore errors during setup
+    }
+}
+// Load API key from user settings if not in environment
+function loadApiKey() {
+    const manager = getSettingsManager();
+    return manager.getApiKey();
+}
+// Load base URL from user settings if not in environment
+function loadBaseURL() {
+    const manager = getSettingsManager();
+    return manager.getBaseURL();
+}
+// Save command line settings to user settings file
+async function saveCommandLineSettings(apiKey, baseURL) {
+    try {
+        const manager = getSettingsManager();
+        // Update with command line values
+        if (apiKey) {
+            manager.updateUserSetting("apiKey", apiKey);
+            console.log("✅ API key saved to ~/.grok/user-settings.json");
+        }
+        if (baseURL) {
+            manager.updateUserSetting("baseURL", baseURL);
+            console.log("✅ Base URL saved to ~/.grok/user-settings.json");
+        }
+    }
+    catch (error) {
+        console.warn("⚠️ Could not save settings to file:", error instanceof Error ? error.message : "Unknown error");
+    }
+}
+// Load model from user settings if not in environment
+function loadModel() {
+    // First check environment variables
+    let model = process.env.GROK_MODEL;
+    if (!model) {
+        // Use the unified model loading from settings manager
+        try {
+            const manager = getSettingsManager();
+            model = manager.getCurrentModel();
+        }
+        catch (error) {
+            // Ignore errors, model will remain undefined
+        }
+    }
+    return model;
+}
+// Handle commit-and-push command in headless mode
+async function handleCommitAndPushHeadless(apiKey, baseURL, model, maxToolRounds) {
+    try {
+        const agent = new GrokAgent(apiKey, baseURL, model, maxToolRounds);
+        // Configure confirmation service for headless mode (auto-approve all operations)
+        const confirmationService = ConfirmationService.getInstance();
+        confirmationService.setSessionFlag("allOperations", true);
+        console.log("🤖 Processing commit and push...\n");
+        console.log("> /commit-and-push\n");
+        // First check if there are any changes at all
+        const initialStatusResult = await agent.executeBashCommand("git status --porcelain");
+        if (!initialStatusResult.success || !initialStatusResult.output?.trim()) {
+            console.log("❌ No changes to commit. Working directory is clean.");
+            process.exit(1);
+        }
+        console.log("✅ git status: Changes detected");
+        // Add all changes
+        const addResult = await agent.executeBashCommand("git add .");
+        if (!addResult.success) {
+            console.log(`❌ git add: ${addResult.error || "Failed to stage changes"}`);
+            process.exit(1);
+        }
+        console.log("✅ git add: Changes staged");
+        // Get staged changes for commit message generation
+        const diffResult = await agent.executeBashCommand("git diff --cached");
+        // Generate commit message using AI
+        const commitPrompt = `Generate a concise, professional git commit message for these changes:
+
+Git Status:
+${initialStatusResult.output}
+
+Git Diff (staged changes):
+${diffResult.output || "No staged changes shown"}
+
+Follow conventional commit format (feat:, fix:, docs:, etc.) and keep it under 72 characters.
+Respond with ONLY the commit message, no additional text.`;
+        console.log("🤖 Generating commit message...");
+        const commitMessageEntries = await agent.processUserMessage(commitPrompt);
+        let commitMessage = "";
+        // Extract the commit message from the AI response
+        for (const entry of commitMessageEntries) {
+            if (entry.type === "assistant" && entry.content.trim()) {
+                commitMessage = entry.content.trim();
+                break;
+            }
+        }
+        if (!commitMessage) {
+            console.log("❌ Failed to generate commit message");
+            process.exit(1);
+        }
+        // Clean the commit message
+        const cleanCommitMessage = commitMessage.replace(/^["']|["']$/g, "");
+        console.log(`✅ Generated commit message: "${cleanCommitMessage}"`);
+        // Execute the commit
+        const commitCommand = `git commit -m "${cleanCommitMessage}"`;
+        const commitResult = await agent.executeBashCommand(commitCommand);
+        if (commitResult.success) {
+            console.log(`✅ git commit: ${commitResult.output?.split("\n")[0] || "Commit successful"}`);
+            // If commit was successful, push to remote
+            // First try regular push, if it fails try with upstream setup
+            let pushResult = await agent.executeBashCommand("git push");
+            if (!pushResult.success &&
+                pushResult.error?.includes("no upstream branch")) {
+                console.log("🔄 Setting upstream and pushing...");
+                pushResult = await agent.executeBashCommand("git push -u origin HEAD");
+            }
+            if (pushResult.success) {
+                console.log(`✅ git push: ${pushResult.output?.split("\n")[0] || "Push successful"}`);
+            }
+            else {
+                console.log(`❌ git push: ${pushResult.error || "Push failed"}`);
+                process.exit(1);
+            }
+        }
+        else {
+            console.log(`❌ git commit: ${commitResult.error || "Commit failed"}`);
+            process.exit(1);
+        }
+    }
+    catch (error) {
+        console.error("❌ Error during commit and push:", error.message);
+        process.exit(1);
+    }
+}
+// Headless mode processing function
+async function processPromptHeadless(prompt, apiKey, baseURL, model, maxToolRounds) {
+    try {
+        const agent = new GrokAgent(apiKey, baseURL, model, maxToolRounds);
+        // Configure confirmation service for headless mode (auto-approve all operations)
+        const confirmationService = ConfirmationService.getInstance();
+        confirmationService.setSessionFlag("allOperations", true);
+        // Process the user message
+        const chatEntries = await agent.processUserMessage(prompt);
+        // Convert chat entries to OpenAI compatible message objects
+        const messages = [];
+        for (const entry of chatEntries) {
+            switch (entry.type) {
+                case "user":
+                    messages.push({
+                        role: "user",
+                        content: entry.content,
+                    });
+                    break;
+                case "assistant":
+                    const assistantMessage = {
+                        role: "assistant",
+                        content: entry.content,
+                    };
+                    // Add tool calls if present
+                    if (entry.toolCalls && entry.toolCalls.length > 0) {
+                        assistantMessage.tool_calls = entry.toolCalls.map((toolCall) => ({
+                            id: toolCall.id,
+                            type: "function",
+                            function: {
+                                name: toolCall.function.name,
+                                arguments: toolCall.function.arguments,
+                            },
+                        }));
+                    }
+                    messages.push(assistantMessage);
+                    break;
+                case "tool_result":
+                    if (entry.toolCall) {
+                        messages.push({
+                            role: "tool",
+                            tool_call_id: entry.toolCall.id,
+                            content: entry.content,
+                        });
+                    }
+                    break;
+            }
+        }
+        // Output each message as a separate JSON object
+        for (const message of messages) {
+            console.log(JSON.stringify(message));
+        }
+    }
+    catch (error) {
+        // Output error in OpenAI compatible format
+        console.log(JSON.stringify({
+            role: "assistant",
+            content: `Error: ${error.message}`,
+        }));
+        process.exit(1);
+    }
+}
+program
+    .name("grok")
+    .description("A conversational AI CLI tool powered by Grok with text editor capabilities")
+    .version("1.0.1")
+    .argument("[message...]", "Initial message to send to Grok")
+    .option("-d, --directory <dir>", "set working directory", process.cwd())
+    .option("-k, --api-key <key>", "Grok API key (or set GROK_API_KEY env var)")
+    .option("-u, --base-url <url>", "Grok API base URL (or set GROK_BASE_URL env var)")
+    .option("-m, --model <model>", "AI model to use (e.g., grok-code-fast-1, grok-4-latest) (or set GROK_MODEL env var)")
+    .option("-p, --prompt <prompt>", "process a single prompt and exit (headless mode)")
+    .option("--max-tool-rounds <rounds>", "maximum number of tool execution rounds (default: 400)", "400")
+    .action(async (message, options) => {
+    if (options.directory) {
+        try {
+            process.chdir(options.directory);
+        }
+        catch (error) {
+            console.error(`Error changing directory to ${options.directory}:`, error.message);
+            process.exit(1);
+        }
+    }
+    try {
+        // Get API key from options, environment, or user settings
+        const apiKey = options.apiKey || loadApiKey();
+        const baseURL = options.baseUrl || loadBaseURL();
+        const model = options.model || loadModel();
+        const maxToolRounds = parseInt(options.maxToolRounds) || 400;
+        if (!apiKey) {
+            console.error("❌ Error: API key required. Set GROK_API_KEY environment variable, use --api-key flag, or save to ~/.grok/user-settings.json");
+            process.exit(1);
+        }
+        // Save API key and base URL to user settings if provided via command line
+        if (options.apiKey || options.baseUrl) {
+            await saveCommandLineSettings(options.apiKey, options.baseUrl);
+        }
+        // Headless mode: process prompt and exit
+        if (options.prompt) {
+            await processPromptHeadless(options.prompt, apiKey, baseURL, model, maxToolRounds);
+            return;
+        }
+        // Interactive mode: launch UI
+        const agent = new GrokAgent(apiKey, baseURL, model, maxToolRounds);
+        console.log("🤖 Starting Grok CLI Conversational Assistant...\n");
+        ensureUserSettingsDirectory();
+        // Support variadic positional arguments for multi-word initial message
+        const initialMessage = Array.isArray(message)
+            ? message.join(" ")
+            : message;
+        render(React.createElement(ChatInterface, { agent, initialMessage }));
+    }
+    catch (error) {
+        console.error("❌ Error initializing Grok CLI:", error.message);
+        process.exit(1);
+    }
+});
+// Git subcommand
+const gitCommand = program
+    .command("git")
+    .description("Git operations with AI assistance");
+gitCommand
+    .command("commit-and-push")
+    .description("Generate AI commit message and push to remote")
+    .option("-d, --directory <dir>", "set working directory", process.cwd())
+    .option("-k, --api-key <key>", "Grok API key (or set GROK_API_KEY env var)")
+    .option("-u, --base-url <url>", "Grok API base URL (or set GROK_BASE_URL env var)")
+    .option("-m, --model <model>", "AI model to use (e.g., grok-code-fast-1, grok-4-latest) (or set GROK_MODEL env var)")
+    .option("--max-tool-rounds <rounds>", "maximum number of tool execution rounds (default: 400)", "400")
+    .action(async (options) => {
+    if (options.directory) {
+        try {
+            process.chdir(options.directory);
+        }
+        catch (error) {
+            console.error(`Error changing directory to ${options.directory}:`, error.message);
+            process.exit(1);
+        }
+    }
+    try {
+        // Get API key from options, environment, or user settings
+        const apiKey = options.apiKey || loadApiKey();
+        const baseURL = options.baseUrl || loadBaseURL();
+        const model = options.model || loadModel();
+        const maxToolRounds = parseInt(options.maxToolRounds) || 400;
+        if (!apiKey) {
+            console.error("❌ Error: API key required. Set GROK_API_KEY environment variable, use --api-key flag, or save to ~/.grok/user-settings.json");
+            process.exit(1);
+        }
+        // Save API key and base URL to user settings if provided via command line
+        if (options.apiKey || options.baseUrl) {
+            await saveCommandLineSettings(options.apiKey, options.baseUrl);
+        }
+        await handleCommitAndPushHeadless(apiKey, baseURL, model, maxToolRounds);
+    }
+    catch (error) {
+        console.error("❌ Error during git commit-and-push:", error.message);
+        process.exit(1);
+    }
+});
+// MCP command
+program.addCommand(createMCPCommand());
+program.parse();
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,MAAM,EAAE,MAAM,KAAK,CAAC;AAC7B,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAClD,OAAO,aAAa,MAAM,mCAAmC,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AACjE,OAAO,EAAE,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAGrD,6BAA6B;AAC7B,MAAM,CAAC,MAAM,EAAE,CAAC;AAEhB,2DAA2D;AAC3D,qDAAqD;AAErD,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;IACzB,8CAA8C;IAC9C,IAAI,OAAO,CAAC,KAAK,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;QACpD,IAAI,CAAC;YACH,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAClC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,sCAAsC;QACxC,CAAC;IACH,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;IAC7C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,gDAAgD;AAChD,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,KAAK,EAAE,EAAE;IACxC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,KAAK,CAAC,CAAC;IAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnD,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IACrE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,uCAAuC;AACvC,SAAS,2BAA2B;IAClC,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,kBAAkB,EAAE,CAAC;QACrC,wDAAwD;QACxD,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,sCAAsC;IACxC,CAAC;AACH,CAAC;AAED,wDAAwD;AACxD,SAAS,UAAU;IACjB,MAAM,OAAO,GAAG,kBAAkB,EAAE,CAAC;IACrC,OAAO,OAAO,CAAC,SAAS,EAAE,CAAC;AAC7B,CAAC;AAED,yDAAyD;AACzD,SAAS,WAAW;IAClB,MAAM,OAAO,GAAG,kBAAkB,EAAE,CAAC;IACrC,OAAO,OAAO,CAAC,UAAU,EAAE,CAAC;AAC9B,CAAC;AAED,mDAAmD;AACnD,KAAK,UAAU,uBAAuB,CACpC,MAAe,EACf,OAAgB;IAEhB,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,kBAAkB,EAAE,CAAC;QAErC,kCAAkC;QAClC,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,CAAC,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;QAC/D,CAAC;QACD,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,iBAAiB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAC9C,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,IAAI,CACV,qCAAqC,EACrC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CACzD,CAAC;IACJ,CAAC;AACH,CAAC;AAED,sDAAsD;AACtD,SAAS,SAAS;IAChB,oCAAoC;IACpC,IAAI,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC;IAEnC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,sDAAsD;QACtD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,kBAAkB,EAAE,CAAC;YACrC,KAAK,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;QACpC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,6CAA6C;QAC/C,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,kDAAkD;AAClD,KAAK,UAAU,2BAA2B,CACxC,MAAc,EACd,OAAgB,EAChB,KAAc,EACd,aAAsB;IAEtB,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,CAAC,CAAC;QAEnE,iFAAiF;QACjF,MAAM,mBAAmB,GAAG,mBAAmB,CAAC,WAAW,EAAE,CAAC;QAC9D,mBAAmB,CAAC,cAAc,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QAE1D,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;QAClD,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;QAEpC,8CAA8C;QAC9C,MAAM,mBAAmB,GAAG,MAAM,KAAK,CAAC,kBAAkB,CACxD,wBAAwB,CACzB,CAAC;QAEF,IAAI,CAAC,mBAAmB,CAAC,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;YACnE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAE9C,kBAAkB;QAClB,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;QAE9D,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;YACvB,OAAO,CAAC,GAAG,CACT,cAAc,SAAS,CAAC,KAAK,IAAI,yBAAyB,EAAE,CAC7D,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QAEzC,mDAAmD;QACnD,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC;QAEvE,mCAAmC;QACnC,MAAM,YAAY,GAAG;;;EAGvB,mBAAmB,CAAC,MAAM;;;EAG1B,UAAU,CAAC,MAAM,IAAI,yBAAyB;;;0DAGU,CAAC;QAEvD,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QAE/C,MAAM,oBAAoB,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;QAC1E,IAAI,aAAa,GAAG,EAAE,CAAC;QAEvB,kDAAkD;QAClD,KAAK,MAAM,KAAK,IAAI,oBAAoB,EAAE,CAAC;YACzC,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;gBACvD,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;gBACrC,MAAM;YACR,CAAC;QACH,CAAC;QAED,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;YACnD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,2BAA2B;QAC3B,MAAM,kBAAkB,GAAG,aAAa,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;QACrE,OAAO,CAAC,GAAG,CAAC,gCAAgC,kBAAkB,GAAG,CAAC,CAAC;QAEnE,qBAAqB;QACrB,MAAM,aAAa,GAAG,kBAAkB,kBAAkB,GAAG,CAAC;QAC9D,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,aAAa,CAAC,CAAC;QAEnE,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CACT,iBACE,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,mBACzC,EAAE,CACH,CAAC;YAEF,2CAA2C;YAC3C,8DAA8D;YAC9D,IAAI,UAAU,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAE5D,IACE,CAAC,UAAU,CAAC,OAAO;gBACnB,UAAU,CAAC,KAAK,EAAE,QAAQ,CAAC,oBAAoB,CAAC,EAChD,CAAC;gBACD,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;gBAClD,UAAU,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,yBAAyB,CAAC,CAAC;YACzE,CAAC;YAED,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;gBACvB,OAAO,CAAC,GAAG,CACT,eACE,UAAU,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,iBACvC,EAAE,CACH,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,eAAe,UAAU,CAAC,KAAK,IAAI,aAAa,EAAE,CAAC,CAAC;gBAChE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,iBAAiB,YAAY,CAAC,KAAK,IAAI,eAAe,EAAE,CAAC,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAChE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,oCAAoC;AACpC,KAAK,UAAU,qBAAqB,CAClC,MAAc,EACd,MAAc,EACd,OAAgB,EAChB,KAAc,EACd,aAAsB;IAEtB,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,CAAC,CAAC;QAEnE,iFAAiF;QACjF,MAAM,mBAAmB,GAAG,mBAAmB,CAAC,WAAW,EAAE,CAAC;QAC9D,mBAAmB,CAAC,cAAc,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QAE1D,2BAA2B;QAC3B,MAAM,WAAW,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAE3D,4DAA4D;QAC5D,MAAM,QAAQ,GAAiC,EAAE,CAAC;QAElD,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YAChC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,MAAM;oBACT,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,KAAK,CAAC,OAAO;qBACvB,CAAC,CAAC;oBACH,MAAM;gBAER,KAAK,WAAW;oBACd,MAAM,gBAAgB,GAA+B;wBACnD,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,KAAK,CAAC,OAAO;qBACvB,CAAC;oBAEF,4BAA4B;oBAC5B,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAClD,gBAAgB,CAAC,UAAU,GAAG,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;4BAC/D,EAAE,EAAE,QAAQ,CAAC,EAAE;4BACf,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE;gCACR,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;gCAC5B,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS;6BACvC;yBACF,CAAC,CAAC,CAAC;oBACN,CAAC;oBAED,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;oBAChC,MAAM;gBAER,KAAK,aAAa;oBAChB,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;wBACnB,QAAQ,CAAC,IAAI,CAAC;4BACZ,IAAI,EAAE,MAAM;4BACZ,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE;4BAC/B,OAAO,EAAE,KAAK,CAAC,OAAO;yBACvB,CAAC,CAAC;oBACL,CAAC;oBACD,MAAM;YACV,CAAC;QACH,CAAC;QAED,gDAAgD;QAChD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,2CAA2C;QAC3C,OAAO,CAAC,GAAG,CACT,IAAI,CAAC,SAAS,CAAC;YACb,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,UAAU,KAAK,CAAC,OAAO,EAAE;SACnC,CAAC,CACH,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,OAAO;KACJ,IAAI,CAAC,MAAM,CAAC;KACZ,WAAW,CACV,4EAA4E,CAC7E;KACA,OAAO,CAAC,OAAO,CAAC;KAChB,QAAQ,CAAC,cAAc,EAAE,iCAAiC,CAAC;KAC3D,MAAM,CAAC,uBAAuB,EAAE,uBAAuB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC;KACvE,MAAM,CAAC,qBAAqB,EAAE,4CAA4C,CAAC;KAC3E,MAAM,CACL,sBAAsB,EACtB,kDAAkD,CACnD;KACA,MAAM,CACL,qBAAqB,EACrB,qFAAqF,CACtF;KACA,MAAM,CACL,uBAAuB,EACvB,kDAAkD,CACnD;KACA,MAAM,CACL,4BAA4B,EAC5B,wDAAwD,EACxD,KAAK,CACN;KACA,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE;IACjC,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QACtB,IAAI,CAAC;YACH,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CACX,+BAA+B,OAAO,CAAC,SAAS,GAAG,EACnD,KAAK,CAAC,OAAO,CACd,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAED,IAAI,CAAC;QACH,0DAA0D;QAC1D,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,UAAU,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,WAAW,EAAE,CAAC;QACjD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,SAAS,EAAE,CAAC;QAC3C,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC;QAE7D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,CAAC,KAAK,CACX,6HAA6H,CAC9H,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,0EAA0E;QAC1E,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,uBAAuB,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC;QAED,yCAAyC;QACzC,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,MAAM,qBAAqB,CACzB,OAAO,CAAC,MAAM,EACd,MAAM,EACN,OAAO,EACP,KAAK,EACL,aAAa,CACd,CAAC;YACF,OAAO;QACT,CAAC;QAED,8BAA8B;QAC9B,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,CAAC,CAAC;QACnE,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC;QAElE,2BAA2B,EAAE,CAAC;QAE9B,uEAAuE;QACvE,MAAM,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;YAC3C,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC;YACnB,CAAC,CAAC,OAAO,CAAC;QAEZ,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;IACxE,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAC/D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC,CAAC,CAAC;AAEL,iBAAiB;AACjB,MAAM,UAAU,GAAG,OAAO;KACvB,OAAO,CAAC,KAAK,CAAC;KACd,WAAW,CAAC,mCAAmC,CAAC,CAAC;AAEpD,UAAU;KACP,OAAO,CAAC,iBAAiB,CAAC;KAC1B,WAAW,CAAC,+CAA+C,CAAC;KAC5D,MAAM,CAAC,uBAAuB,EAAE,uBAAuB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC;KACvE,MAAM,CAAC,qBAAqB,EAAE,4CAA4C,CAAC;KAC3E,MAAM,CACL,sBAAsB,EACtB,kDAAkD,CACnD;KACA,MAAM,CACL,qBAAqB,EACrB,qFAAqF,CACtF;KACA,MAAM,CACL,4BAA4B,EAC5B,wDAAwD,EACxD,KAAK,CACN;KACA,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;IACxB,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QACtB,IAAI,CAAC;YACH,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CACX,+BAA+B,OAAO,CAAC,SAAS,GAAG,EACnD,KAAK,CAAC,OAAO,CACd,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAED,IAAI,CAAC;QACH,0DAA0D;QAC1D,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,UAAU,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,WAAW,EAAE,CAAC;QACjD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,SAAS,EAAE,CAAC;QAC3C,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC;QAE7D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,CAAC,KAAK,CACX,6HAA6H,CAC9H,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,0EAA0E;QAC1E,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,uBAAuB,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC;QAED,MAAM,2BAA2B,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,CAAC,CAAC;IAC3E,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QACpE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC,CAAC,CAAC;AAEL,cAAc;AACd,OAAO,CAAC,UAAU,CAAC,gBAAgB,EAAE,CAAC,CAAC;AAEvC,OAAO,CAAC,KAAK,EAAE,CAAC"}

package/dist/mcp/client.d.ts

@@ -0,0 +1,31 @@
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { EventEmitter } from "events";
+import { TransportType, TransportConfig } from "./transports.js";
+export interface MCPServerConfig {
+    name: string;
+    transport: TransportConfig;
+    command?: string;
+    args?: string[];
+    env?: Record<string, string>;
+}
+export interface MCPTool {
+    name: string;
+    description: string;
+    inputSchema: any;
+    serverName: string;
+}
+export declare class MCPManager extends EventEmitter {
+    private clients;
+    private transports;
+    private tools;
+    private pendingConnections;
+    addServer(config: MCPServerConfig): Promise<void>;
+    private _addServerInternal;
+    removeServer(serverName: string): Promise<void>;
+    callTool(toolName: string, arguments_: any): Promise<CallToolResult>;
+    getTools(): MCPTool[];
+    getServers(): string[];
+    shutdown(): Promise<void>;
+    getTransportType(serverName: string): TransportType | undefined;
+    ensureServersInitialized(): Promise<void>;
+}

package/dist/mcp/client.js

@@ -0,0 +1,152 @@
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { EventEmitter } from "events";
+import { createTransport } from "./transports.js";
+export class MCPManager extends EventEmitter {
+    clients = new Map();
+    transports = new Map();
+    tools = new Map();
+    pendingConnections = new Map();
+    async addServer(config) {
+        // Check if already connecting to prevent race condition
+        const pending = this.pendingConnections.get(config.name);
+        if (pending) {
+            return pending;
+        }
+        // Create a promise for this connection attempt
+        const connectionPromise = this._addServerInternal(config);
+        this.pendingConnections.set(config.name, connectionPromise);
+        try {
+            await connectionPromise;
+        }
+        finally {
+            this.pendingConnections.delete(config.name);
+        }
+    }
+    async _addServerInternal(config) {
+        try {
+            // Handle legacy stdio-only configuration
+            let transportConfig = config.transport;
+            if (!transportConfig && config.command) {
+                transportConfig = {
+                    type: 'stdio',
+                    command: config.command,
+                    args: config.args,
+                    env: config.env
+                };
+            }
+            if (!transportConfig) {
+                throw new Error('Transport configuration is required');
+            }
+            // Create transport
+            const transport = createTransport(transportConfig);
+            this.transports.set(config.name, transport);
+            // Create client
+            const client = new Client({
+                name: "ax-cli",
+                version: "1.0.0"
+            }, {
+                capabilities: {
+                    tools: {}
+                }
+            });
+            this.clients.set(config.name, client);
+            // Connect
+            const sdkTransport = await transport.connect();
+            await client.connect(sdkTransport);
+            // List available tools
+            const toolsResult = await client.listTools();
+            // Register tools
+            for (const tool of toolsResult.tools) {
+                const mcpTool = {
+                    name: `mcp__${config.name}__${tool.name}`,
+                    description: tool.description || `Tool from ${config.name} server`,
+                    inputSchema: tool.inputSchema,
+                    serverName: config.name
+                };
+                this.tools.set(mcpTool.name, mcpTool);
+            }
+            this.emit('serverAdded', config.name, toolsResult.tools.length);
+        }
+        catch (error) {
+            // Clean up on error
+            this.clients.delete(config.name);
+            const transport = this.transports.get(config.name);
+            if (transport) {
+                await transport.disconnect().catch(() => { });
+                this.transports.delete(config.name);
+            }
+            this.emit('serverError', config.name, error);
+            throw error;
+        }
+    }
+    async removeServer(serverName) {
+        // Remove tools
+        for (const [toolName, tool] of this.tools.entries()) {
+            if (tool.serverName === serverName) {
+                this.tools.delete(toolName);
+            }
+        }
+        // Disconnect client
+        const client = this.clients.get(serverName);
+        if (client) {
+            await client.close();
+            this.clients.delete(serverName);
+        }
+        // Close transport
+        const transport = this.transports.get(serverName);
+        if (transport) {
+            await transport.disconnect();
+            this.transports.delete(serverName);
+        }
+        this.emit('serverRemoved', serverName);
+    }
+    async callTool(toolName, arguments_) {
+        const tool = this.tools.get(toolName);
+        if (!tool) {
+            throw new Error(`Tool ${toolName} not found`);
+        }
+        const client = this.clients.get(tool.serverName);
+        if (!client) {
+            throw new Error(`Server ${tool.serverName} not connected`);
+        }
+        // Extract the original tool name (remove mcp__servername__ prefix)
+        const originalToolName = toolName.replace(`mcp__${tool.serverName}__`, '');
+        const result = await client.callTool({
+            name: originalToolName,
+            arguments: arguments_
+        });
+        return result;
+    }
+    getTools() {
+        return Array.from(this.tools.values());
+    }
+    getServers() {
+        return Array.from(this.clients.keys());
+    }
+    async shutdown() {
+        const serverNames = Array.from(this.clients.keys());
+        await Promise.all(serverNames.map(name => this.removeServer(name)));
+    }
+    getTransportType(serverName) {
+        const transport = this.transports.get(serverName);
+        return transport?.getType();
+    }
+    async ensureServersInitialized() {
+        if (this.clients.size > 0) {
+            return; // Already initialized
+        }
+        const { loadMCPConfig } = await import('../mcp/config');
+        const config = loadMCPConfig();
+        // Initialize servers in parallel to avoid blocking
+        const initPromises = config.servers.map(async (serverConfig) => {
+            try {
+                await this.addServer(serverConfig);
+            }
+            catch (error) {
+                console.warn(`Failed to initialize MCP server ${serverConfig.name}:`, error);
+            }
+        });
+        await Promise.all(initPromises);
+    }
+}
+//# sourceMappingURL=client.js.map

package/dist/mcp/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/mcp/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAEnE,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,eAAe,EAAgD,MAAM,iBAAiB,CAAC;AAkBhG,MAAM,OAAO,UAAW,SAAQ,YAAY;IAClC,OAAO,GAAwB,IAAI,GAAG,EAAE,CAAC;IACzC,UAAU,GAA8B,IAAI,GAAG,EAAE,CAAC;IAClD,KAAK,GAAyB,IAAI,GAAG,EAAE,CAAC;IACxC,kBAAkB,GAA+B,IAAI,GAAG,EAAE,CAAC;IAEnE,KAAK,CAAC,SAAS,CAAC,MAAuB;QACrC,wDAAwD;QACxD,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACzD,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,+CAA+C;QAC/C,MAAM,iBAAiB,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC1D,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;QAE5D,IAAI,CAAC;YACH,MAAM,iBAAiB,CAAC;QAC1B,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,MAAuB;QACtD,IAAI,CAAC;YACH,yCAAyC;YACzC,IAAI,eAAe,GAAG,MAAM,CAAC,SAAS,CAAC;YACvC,IAAI,CAAC,eAAe,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACvC,eAAe,GAAG;oBAChB,IAAI,EAAE,OAAO;oBACb,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,GAAG,EAAE,MAAM,CAAC,GAAG;iBAChB,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;YACzD,CAAC;YAED,mBAAmB;YACnB,MAAM,SAAS,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;YACnD,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YAE5C,gBAAgB;YAChB,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,OAAO;aACjB,EACD;gBACE,YAAY,EAAE;oBACZ,KAAK,EAAE,EAAE;iBACV;aACF,CACF,CAAC;YAEF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAEtC,UAAU;YACV,MAAM,YAAY,GAAG,MAAM,SAAS,CAAC,OAAO,EAAE,CAAC;YAC/C,MAAM,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAEnC,uBAAuB;YACvB,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAE7C,iBAAiB;YACjB,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;gBACrC,MAAM,OAAO,GAAY;oBACvB,IAAI,EAAE,QAAQ,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,EAAE;oBACzC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,aAAa,MAAM,CAAC,IAAI,SAAS;oBAClE,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,UAAU,EAAE,MAAM,CAAC,IAAI;iBACxB,CAAC;gBACF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAClE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,oBAAoB;YACpB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACnD,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,SAAS,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAA6B,CAAC,CAAC,CAAC;gBACxE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACtC,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAC7C,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,UAAkB;QACnC,eAAe;QACf,KAAK,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;YACpD,IAAI,IAAI,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACnC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,oBAAoB;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC;QAED,kBAAkB;QAClB,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAClD,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,SAAS,CAAC,UAAU,EAAE,CAAC;YAC7B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACrC,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,UAAe;QAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,YAAY,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC,UAAU,gBAAgB,CAAC,CAAC;QAC7D,CAAC;QAED,mEAAmE;QACnE,MAAM,gBAAgB,GAAG,QAAQ,CAAC,OAAO,CAAC,QAAQ,IAAI,CAAC,UAAU,IAAI,EAAE,EAAE,CAAC,CAAC;QAE3E,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;YACnC,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,UAAU;SACtB,CAAC,CAAC;QAEH,OAAO,MAAwB,CAAC;IAClC,CAAC;IAED,QAAQ;QACN,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IACzC,CAAC;IAED,UAAU;QACR,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QACpD,MAAM,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,gBAAgB,CAAC,UAAkB;QACjC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAClD,OAAO,SAAS,EAAE,OAAO,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,wBAAwB;QAC5B,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,sBAAsB;QAChC,CAAC;QAED,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,aAAa,EAAE,CAAC;QAE/B,mDAAmD;QACnD,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE;YAC7D,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YACrC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,IAAI,CAAC,mCAAmC,YAAY,CAAC,IAAI,GAAG,EAAE,KAAK,CAAC,CAAC;YAC/E,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAClC,CAAC;CACF"}

package/dist/mcp/config.d.ts

@@ -0,0 +1,13 @@
+import { MCPServerConfig } from "./client.js";
+export interface MCPConfig {
+    servers: MCPServerConfig[];
+}
+/**
+ * Load MCP configuration from project settings
+ */
+export declare function loadMCPConfig(): MCPConfig;
+export declare function saveMCPConfig(config: MCPConfig): void;
+export declare function addMCPServer(config: MCPServerConfig): void;
+export declare function removeMCPServer(serverName: string): void;
+export declare function getMCPServer(serverName: string): MCPServerConfig | undefined;
+export declare const PREDEFINED_SERVERS: Record<string, MCPServerConfig>;