@axiom-lattice/gateway 3.0.2 → 3.0.3

package/src/controllers/mcp-configs.ts

@@ -1,673 +0,0 @@
-/**
- * MCP Server Config Controller
- * Handles MCP server configuration CRUD operations with automatic tool registration
- */
-
-import { FastifyRequest, FastifyReply } from "fastify";
-import {
-  getStoreLattice,
-  mcpManager,
-  McpLatticeManager,
-} from "@axiom-lattice/core";
-import type {
-  McpServerConfigStore,
-  McpServerConfigEntry,
-  CreateMcpServerConfigRequest,
-  UpdateMcpServerConfigRequest,
-  McpServerConfig,
-  McpTool,
-} from "@axiom-lattice/protocols";
-import { randomUUID } from "crypto";
-
-/**
- * Get tenant ID from authenticated user context
- * Falls back to request headers for backward compatibility
- */
-function getTenantId(request: FastifyRequest): string {
-  // First try to get from authenticated user context
-  const userTenantId = (request as any).user?.tenantId;
-  if (userTenantId) {
-    return userTenantId;
-  }
-
-  // Fallback to request headers for backward compatibility
-  return (request.headers["x-tenant-id"] as string) || "default";
-}
-
-/**
- * MCP server config list response
- */
-interface McpServerConfigListResponse {
-  success: boolean;
-  message: string;
-  data: {
-    records: McpServerConfigEntry[];
-    total: number;
-  };
-}
-
-/**
- * MCP server config response
- */
-interface McpServerConfigResponse {
-  success: boolean;
-  message: string;
-  data?: McpServerConfigEntry;
-}
-
-/**
- * Test connection response
- */
-interface TestConnectionResponse {
-  success: boolean;
-  message: string;
-  data?: {
-    connected: boolean;
-    latency?: number;
-    error?: string;
-  };
-}
-
-/**
- * List tools response
- */
-interface ListToolsResponse {
-  success: boolean;
-  message: string;
-  data?: {
-    tools: McpTool[];
-  };
-}
-
-/**
- * Get all MCP server configs for a tenant
- */
-export async function getMcpServerConfigList(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<McpServerConfigListResponse> {
-  const tenantId = getTenantId(request);
-
-  try {
-    const storeLattice = getStoreLattice("default", "mcp");
-    const store: McpServerConfigStore = storeLattice.store;
-
-    const configs = await store.getAllConfigs(tenantId);
-
-    return {
-      success: true,
-      message: "MCP server configurations retrieved successfully",
-      data: {
-        records: configs,
-        total: configs.length,
-      },
-    };
-  } catch (error) {
-    console.error("Failed to get MCP server configs:", error);
-    return {
-      success: false,
-      message: "Failed to retrieve MCP server configurations",
-      data: {
-        records: [],
-        total: 0,
-      },
-    };
-  }
-}
-
-/**
- * Get MCP server config by key
- */
-export async function getMcpServerConfig(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<McpServerConfigResponse> {
-  const tenantId = getTenantId(request);
-  const { key } = request.params as { key: string };
-
-  try {
-    const storeLattice = getStoreLattice("default", "mcp");
-    const store: McpServerConfigStore = storeLattice.store;
-
-    const config = await store.getConfigByKey(tenantId, key);
-
-    if (!config) {
-      return {
-        success: false,
-        message: "MCP server configuration not found",
-      };
-    }
-
-    return {
-      success: true,
-      message: "MCP server configuration retrieved successfully",
-      data: config,
-    };
-  } catch (error) {
-    console.error("Failed to get MCP server config:", error);
-    return {
-      success: false,
-      message: "Failed to retrieve MCP server configuration",
-    };
-  }
-}
-
-/**
- * Create new MCP server config
- */
-export async function createMcpServerConfig(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<McpServerConfigResponse> {
-  const tenantId = getTenantId(request);
-  const body = request.body as CreateMcpServerConfigRequest & {
-    id?: string;
-  };
-
-  try {
-    const storeLattice = getStoreLattice("default", "mcp");
-    const store: McpServerConfigStore = storeLattice.store;
-
-    // Check if key already exists
-    const existing = await store.getConfigByKey(tenantId, body.key);
-    if (existing) {
-      reply.code(409);
-      return {
-        success: false,
-        message: "MCP server configuration with this key already exists",
-      };
-    }
-
-    const id = body.id || randomUUID();
-    const config = await store.createConfig(tenantId, id, body);
-
-    // Auto-connect and register tools
-    try {
-      await connectAndRegisterTools(config);
-      // Update status to connected
-      await store.updateConfig(tenantId, id, { status: "connected" });
-      config.status = "connected";
-    } catch (error) {
-      console.warn("Failed to auto-connect MCP server:", error);
-      // Update status to error but don't fail the creation
-      await store.updateConfig(tenantId, id, { status: "error" });
-      config.status = "error";
-    }
-
-    reply.code(201);
-    return {
-      success: true,
-      message: "MCP server configuration created successfully",
-      data: config,
-    };
-  } catch (error) {
-    console.error("Failed to create MCP server config:", error);
-    return {
-      success: false,
-      message: "Failed to create MCP server configuration",
-    };
-  }
-}
-
-/**
- * Update MCP server config
- */
-export async function updateMcpServerConfig(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<McpServerConfigResponse> {
-  const tenantId = getTenantId(request);
-  const { key } = request.params as { key: string };
-  const updates = request.body as Partial<UpdateMcpServerConfigRequest>;
-
-  try {
-    const storeLattice = getStoreLattice("default", "mcp");
-    const store: McpServerConfigStore = storeLattice.store;
-
-    const existing = await store.getConfigByKey(tenantId, key);
-    if (!existing) {
-      reply.code(404);
-      return {
-        success: false,
-        message: "MCP server configuration not found",
-      };
-    }
-
-    // If config is being updated, we need to reconnect
-    const shouldReconnect = updates.config !== undefined;
-
-    const updated = await store.updateConfig(tenantId, existing.id, updates);
-
-    if (!updated) {
-      return {
-        success: false,
-        message: "Failed to update MCP server configuration",
-      };
-    }
-
-    // Reconnect and re-register tools if config changed
-    if (shouldReconnect) {
-      try {
-        // Reconnect single server without affecting others
-        await mcpManager.reconnectServer(key);
-
-        // Re-register tools with new config
-        await mcpManager.registerToolsToToolLattice(key, updated.selectedTools);
-
-        await store.updateConfig(tenantId, existing.id, { status: "connected" });
-        updated.status = "connected";
-      } catch (error) {
-        console.warn("Failed to reconnect MCP server:", error);
-        await store.updateConfig(tenantId, existing.id, { status: "error" });
-        updated.status = "error";
-      }
-    }
-
-    return {
-      success: true,
-      message: "MCP server configuration updated successfully",
-      data: updated,
-    };
-  } catch (error) {
-    console.error("Failed to update MCP server config:", error);
-    return {
-      success: false,
-      message: "Failed to update MCP server configuration",
-    };
-  }
-}
-
-/**
- * Delete MCP server config by key or id
- */
-export async function deleteMcpServerConfig(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<McpServerConfigResponse> {
-  const tenantId = getTenantId(request);
-  const { keyOrId } = request.params as { keyOrId: string };
-
-  try {
-    const storeLattice = getStoreLattice("default", "mcp");
-    const store: McpServerConfigStore = storeLattice.store;
-
-    // Try to find by key first, then by id
-    let config = await store.getConfigByKey(tenantId, keyOrId);
-    let configKey = keyOrId;
-
-    if (!config) {
-      // Try to find by id
-      config = await store.getConfigById(tenantId, keyOrId);
-      if (config) {
-        configKey = config.key;
-      }
-    }
-
-    if (!config) {
-      reply.code(404);
-      return {
-        success: false,
-        message: "MCP server configuration not found",
-      };
-    }
-
-    // Disconnect and remove from MCP manager
-    try {
-      if (mcpManager.hasServer(configKey)) {
-        await mcpManager.removeServer(configKey);
-      }
-    } catch (error) {
-      console.warn("Failed to remove from MCP manager:", error);
-    }
-
-    const deleted = await store.deleteConfig(tenantId, config.id);
-
-    if (!deleted) {
-      return {
-        success: false,
-        message: "Failed to delete MCP server configuration",
-      };
-    }
-
-    return {
-      success: true,
-      message: "MCP server configuration deleted successfully",
-    };
-  } catch (error) {
-    console.error("Failed to delete MCP server config:", error);
-    return {
-      success: false,
-      message: "Failed to delete MCP server configuration",
-    };
-  }
-}
-
-/**
- * Test MCP server connection
- */
-export async function testMcpServerConnection(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<TestConnectionResponse> {
-  const tenantId = getTenantId(request);
-  const { key } = request.params as { key: string };
-
-  try {
-    const storeLattice = getStoreLattice("default", "mcp");
-    const store: McpServerConfigStore = storeLattice.store;
-
-    const config = await store.getConfigByKey(tenantId, key);
-    if (!config) {
-      reply.code(404);
-      return {
-        success: false,
-        message: "MCP server configuration not found",
-      };
-    }
-
-    // Test connection by creating a standalone client
-    try {
-      const testKey = `__test_${key}_${Date.now()}`;
-      const connection = convertToConnection(config.config);
-      const { latency } = await McpLatticeManager.testConnection(testKey, connection);
-
-      return {
-        success: true,
-        message: "Connection test successful",
-        data: {
-          connected: true,
-          latency,
-        },
-      };
-    } catch (error) {
-      return {
-        success: true,
-        message: "Connection test failed",
-        data: {
-          connected: false,
-          error: error instanceof Error ? error.message : "Unknown error",
-        },
-      };
-    }
-  } catch (error) {
-    console.error("Failed to test MCP server connection:", error);
-    return {
-      success: false,
-      message: "Failed to test MCP server connection",
-      data: {
-        connected: false,
-        error: error instanceof Error ? error.message : "Unknown error",
-      },
-    };
-  }
-}
-
-/**
- * List available tools from an MCP server
- */
-export async function listMcpServerTools(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<ListToolsResponse> {
-  const tenantId = getTenantId(request);
-  const { key } = request.params as { key: string };
-
-  try {
-    const storeLattice = getStoreLattice("default", "mcp");
-    const store: McpServerConfigStore = storeLattice.store;
-
-    const config = await store.getConfigByKey(tenantId, key);
-    if (!config) {
-      reply.code(404);
-      return {
-        success: false,
-        message: "MCP server configuration not found",
-      };
-    }
-
-    // Ensure server is connected
-    if (!mcpManager.hasServer(key)) {
-      await connectAndRegisterTools(config);
-    }
-
-    const tools = await mcpManager.getServerTools(key);
-
-    return {
-      success: true,
-      message: "Tools retrieved successfully",
-      data: {
-        tools,
-      },
-    };
-  } catch (error) {
-    console.error("Failed to list MCP tools:", error);
-    return {
-      success: false,
-      message: "Failed to retrieve tools",
-    };
-  }
-}
-
-/**
- * Connect to MCP server and register tools
- */
-export async function connectMcpServer(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<McpServerConfigResponse> {
-  const tenantId = getTenantId(request);
-  const { key } = request.params as { key: string };
-
-  try {
-    const storeLattice = getStoreLattice("default", "mcp");
-    const store: McpServerConfigStore = storeLattice.store;
-
-    const config = await store.getConfigByKey(tenantId, key);
-    if (!config) {
-      reply.code(404);
-      return {
-        success: false,
-        message: "MCP server configuration not found",
-      };
-    }
-
-    await connectAndRegisterTools(config);
-
-    // Update status
-    const updated = await store.updateConfig(tenantId, config.id, {
-      status: "connected",
-    });
-
-    return {
-      success: true,
-      message: "MCP server connected successfully",
-      data: updated || config,
-    };
-  } catch (error) {
-    console.error("Failed to connect MCP server:", error);
-
-    // Update status to error
-    const storeLattice = getStoreLattice("default", "mcp");
-    const store: McpServerConfigStore = storeLattice.store;
-    const config = await store.getConfigByKey(tenantId, key);
-    if (config) {
-      await store.updateConfig(tenantId, config.id, { status: "error" });
-    }
-
-    return {
-      success: false,
-      message: `Failed to connect MCP server: ${
-        error instanceof Error ? error.message : "Unknown error"
-      }`,
-    };
-  }
-}
-
-/**
- * Disconnect from MCP server
- */
-export async function disconnectMcpServer(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<McpServerConfigResponse> {
-  const tenantId = getTenantId(request);
-  const { key } = request.params as { key: string };
-
-  try {
-    const storeLattice = getStoreLattice("default", "mcp");
-    const store: McpServerConfigStore = storeLattice.store;
-
-    const config = await store.getConfigByKey(tenantId, key);
-    if (!config) {
-      reply.code(404);
-      return {
-        success: false,
-        message: "MCP server configuration not found",
-      };
-    }
-
-    // Disconnect from MCP manager
-    if (mcpManager.hasServer(key)) {
-      await mcpManager.removeServer(key);
-    }
-
-    // Update status
-    const updated = await store.updateConfig(tenantId, config.id, {
-      status: "disconnected",
-    });
-
-    return {
-      success: true,
-      message: "MCP server disconnected successfully",
-      data: updated || config,
-    };
-  } catch (error) {
-    console.error("Failed to disconnect MCP server:", error);
-    return {
-      success: false,
-      message: "Failed to disconnect MCP server",
-    };
-  }
-}
-
-/**
- * Test MCP server connection and list tools without saving config
- * Used for multi-step configuration flow
- */
-export async function testMcpServerTools(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<ListToolsResponse> {
-  const body = request.body as {
-    config: McpServerConfig;
-  };
-
-  try {
-    if (!body.config) {
-      reply.code(400);
-      return {
-        success: false,
-        message: "config is required",
-      };
-    }
-
-    const testKey = `__test_${Date.now()}`;
-
-    // Create standalone client for testing — does not affect global mcpManager
-    const connection = convertToConnection(body.config);
-    const { tools } = await McpLatticeManager.testConnection(testKey, connection);
-
-    return {
-      success: true,
-      message: "Tools retrieved successfully",
-      data: {
-        tools,
-      },
-    };
-  } catch (error) {
-    console.error("Failed to test MCP server tools:", error);
-    return {
-      success: false,
-      message: `Failed to retrieve tools: ${
-        error instanceof Error ? error.message : String(error)
-      }`,
-    };
-  }
-}
-
-/**
- * Convert McpServerConfig to Connection format compatible with @langchain/mcp-adapters
- */
-export function convertToConnection(config: McpServerConfig): any {
-  const baseConfig = {
-    env: config.env,
-  };
-
-  if (config.transport === "stdio") {
-    return {
-      ...baseConfig,
-      transport: "stdio" as const,
-      command: config.command!,
-      args: config.args || [],
-    };
-  } else {
-    // For HTTP/SSE transports
-    return {
-      ...baseConfig,
-      transport: config.transport === "streamable_http" ? "http" : config.transport,
-      url: config.url!,
-    };
-  }
-}
-
-/**
- * Helper function to connect to MCP server and register selected tools
- */
-async function connectAndRegisterTools(
-  config: McpServerConfigEntry
-): Promise<void> {
-  // Add server to MCP manager
-  const connection = convertToConnection(config.config);
-  mcpManager.addServer(config.key, connection);
-
-  // Connect only unconnected servers (new server gets connected)
-  await mcpManager.connect();
-
-  // Register tools for this server (handles unregister + filter + prefix)
-  await mcpManager.registerToolsToToolLattice(config.key, config.selectedTools);
-}
-
-/**
- * Register MCP server config routes
- */
-export function registerMcpServerConfigRoutes(app: any): void {
-  // Get all configs
-  app.get("/api/mcp-servers", getMcpServerConfigList);
-
-  // Get config by key
-  app.get("/api/mcp-servers/:key", getMcpServerConfig);
-
-  // Create config
-  app.post("/api/mcp-servers", createMcpServerConfig);
-
-  // Update config
-  app.put("/api/mcp-servers/:key", updateMcpServerConfig);
-
-  // Delete config by key or id
-  app.delete("/api/mcp-servers/:keyOrId", deleteMcpServerConfig);
-
-  // Test connection
-  app.post("/api/mcp-servers/:key/test", testMcpServerConnection);
-
-  // List available tools
-  app.get("/api/mcp-servers/:key/tools", listMcpServerTools);
-
-  // Connect to server
-  app.post("/api/mcp-servers/:key/connect", connectMcpServer);
-
-  // Disconnect from server
-  app.post("/api/mcp-servers/:key/disconnect", disconnectMcpServer);
-
-  // Test tools without saving config (for multi-step config flow)
-  app.post("/api/mcp-servers/test-tools", testMcpServerTools);
-}