@wireweave/mcp-server 1.6.2-beta.2 → 1.7.0-beta.0

package/dist/index.js

@@ -3,6 +3,7 @@
 // src/index.ts
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
 import {
   CallToolRequestSchema,
   ListToolsRequestSchema,
@@ -11,6 +12,8 @@ import {
   ListResourcesRequestSchema,
   ReadResourceRequestSchema
 } from "@modelcontextprotocol/sdk/types.js";
+import { createServer } from "http";
+import { randomUUID } from "crypto";
 
 // src/tools.ts
 var tools = [
@@ -1046,152 +1049,236 @@ var apiConfig = {
   apiUrl: API_URL,
   apiKey: API_KEY
 };
+var isHttpMode = process.argv.includes("--http") || process.env.WIREWEAVE_TRANSPORT === "http";
+var HTTP_PORT = parseInt(process.env.WIREWEAVE_MCP_PORT || "3305", 10);
+var HTTP_HOST = process.env.WIREWEAVE_MCP_HOST || "127.0.0.1";
 function log(message, level = "info") {
   const prefix = level === "error" ? "\u274C" : level === "warn" ? "\u26A0\uFE0F" : "\u2713";
-  console.error(`[Wireweave] ${prefix} ${message}`);
+  const output = isHttpMode ? console.log : console.error;
+  output(`[Wireweave] ${prefix} ${message}`);
 }
-var server = new Server(
-  {
-    name: "wireweave-mcp",
-    version: "1.0.0"
-  },
-  {
-    capabilities: {
-      tools: {},
-      prompts: {},
-      resources: {}
+function createMcpServer() {
+  const server = new Server(
+    {
+      name: "wireweave-mcp",
+      version: "1.0.0"
+    },
+    {
+      capabilities: {
+        tools: {},
+        prompts: {},
+        resources: {}
+      }
     }
-  }
-);
-server.setRequestHandler(ListToolsRequestSchema, async () => {
-  return { tools: [...tools, ...localTools] };
-});
-server.setRequestHandler(CallToolRequestSchema, async (request) => {
-  const { name, arguments: args } = request.params;
-  if (LOCAL_TOOL_NAMES.has(name)) {
-    return handleLocalTool(name, args, apiConfig);
-  }
-  const endpoint = toolEndpoints[name];
-  if (!endpoint) {
-    const allTools = [...tools, ...localTools];
-    return {
-      content: [
-        {
-          type: "text",
-          text: JSON.stringify({
-            error: `Unknown tool: ${name}`,
-            availableTools: allTools.map((t) => t.name)
-          }, null, 2)
-        }
-      ],
-      isError: true
-    };
-  }
-  try {
-    const result = await callApi(apiConfig, endpoint, args);
-    return {
-      content: [
-        {
-          type: "text",
-          text: JSON.stringify(result, null, 2)
-        }
-      ]
-    };
-  } catch (error) {
-    return {
-      content: [
-        {
-          type: "text",
-          text: JSON.stringify({
-            error: error instanceof Error ? error.message : "Unknown error"
-          }, null, 2)
-        }
-      ],
-      isError: true
-    };
-  }
-});
-server.setRequestHandler(ListPromptsRequestSchema, async () => {
-  return { prompts };
-});
-server.setRequestHandler(GetPromptRequestSchema, async (request) => {
-  const { name, arguments: args } = request.params;
-  const prompt = prompts.find((p) => p.name === name);
-  if (!prompt) {
-    throw new Error(`Prompt not found: ${name}`);
-  }
-  const template = promptTemplates[name];
-  if (!template) {
-    throw new Error(`Prompt template not found: ${name}`);
-  }
-  let messageText = template;
-  if (args) {
-    for (const [key, value] of Object.entries(args)) {
-      messageText = messageText.replace(new RegExp(`\\{\\{${key}\\}\\}`, "g"), String(value));
+  );
+  server.setRequestHandler(ListToolsRequestSchema, async () => {
+    return { tools: [...tools, ...localTools] };
+  });
+  server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    const { name, arguments: args } = request.params;
+    if (LOCAL_TOOL_NAMES.has(name)) {
+      return handleLocalTool(name, args, apiConfig);
     }
-  }
-  messageText = messageText.replace(/\{\{[^}]+\}\}/g, "");
-  return {
-    description: prompt.description,
-    messages: [
-      {
-        role: "user",
-        content: {
-          type: "text",
-          text: messageText
-        }
+    const endpoint = toolEndpoints[name];
+    if (!endpoint) {
+      const allTools = [...tools, ...localTools];
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify({
+              error: `Unknown tool: ${name}`,
+              availableTools: allTools.map((t) => t.name)
+            }, null, 2)
+          }
+        ],
+        isError: true
+      };
+    }
+    try {
+      const result = await callApi(apiConfig, endpoint, args);
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    } catch (error) {
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify({
+              error: error instanceof Error ? error.message : "Unknown error"
+            }, null, 2)
+          }
+        ],
+        isError: true
+      };
+    }
+  });
+  server.setRequestHandler(ListPromptsRequestSchema, async () => {
+    return { prompts };
+  });
+  server.setRequestHandler(GetPromptRequestSchema, async (request) => {
+    const { name, arguments: args } = request.params;
+    const prompt = prompts.find((p) => p.name === name);
+    if (!prompt) {
+      throw new Error(`Prompt not found: ${name}`);
+    }
+    const template = promptTemplates[name];
+    if (!template) {
+      throw new Error(`Prompt template not found: ${name}`);
+    }
+    let messageText = template;
+    if (args) {
+      for (const [key, value] of Object.entries(args)) {
+        messageText = messageText.replace(new RegExp(`\\{\\{${key}\\}\\}`, "g"), String(value));
       }
-    ]
-  };
-});
-server.setRequestHandler(ListResourcesRequestSchema, async () => {
-  return { resources };
-});
-server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
-  const { uri } = request.params;
-  const resource = resources.find((r) => r.uri === uri);
-  if (!resource) {
-    throw new Error(`Resource not found: ${uri}`);
-  }
-  const toolName = resourceToTool[uri];
-  if (!toolName) {
-    throw new Error(`No tool mapping for resource: ${uri}`);
-  }
-  const endpoint = toolEndpoints[toolName];
-  if (!endpoint) {
-    throw new Error(`Tool endpoint not found: ${toolName}`);
-  }
-  try {
-    const result = await callApi(apiConfig, endpoint, {});
-    const content = typeof result === "string" ? result : JSON.stringify(result, null, 2);
+    }
+    messageText = messageText.replace(/\{\{[^}]+\}\}/g, "");
     return {
-      contents: [
+      description: prompt.description,
+      messages: [
         {
-          uri,
-          mimeType: resource.mimeType,
-          text: content
+          role: "user",
+          content: {
+            type: "text",
+            text: messageText
+          }
         }
       ]
     };
-  } catch (error) {
-    throw new Error(`Failed to fetch resource: ${error instanceof Error ? error.message : "Unknown error"}`);
-  }
-});
-server.onerror = (error) => {
-  log("An error occurred", "error");
-};
-process.on("SIGINT", async () => {
-  await server.close();
-  process.exit(0);
-});
-async function main() {
+  });
+  server.setRequestHandler(ListResourcesRequestSchema, async () => {
+    return { resources };
+  });
+  server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
+    const { uri } = request.params;
+    const resource = resources.find((r) => r.uri === uri);
+    if (!resource) {
+      throw new Error(`Resource not found: ${uri}`);
+    }
+    const toolName = resourceToTool[uri];
+    if (!toolName) {
+      throw new Error(`No tool mapping for resource: ${uri}`);
+    }
+    const endpoint = toolEndpoints[toolName];
+    if (!endpoint) {
+      throw new Error(`Tool endpoint not found: ${toolName}`);
+    }
+    try {
+      const result = await callApi(apiConfig, endpoint, {});
+      const content = typeof result === "string" ? result : JSON.stringify(result, null, 2);
+      return {
+        contents: [
+          {
+            uri,
+            mimeType: resource.mimeType,
+            text: content
+          }
+        ]
+      };
+    } catch (error) {
+      throw new Error(`Failed to fetch resource: ${error instanceof Error ? error.message : "Unknown error"}`);
+    }
+  });
+  server.onerror = () => {
+    log("An error occurred", "error");
+  };
+  return server;
+}
+async function startStdio() {
+  const server = createMcpServer();
   const transport = new StdioServerTransport();
   await server.connect(transport);
-  log(`MCP server started with ${tools.length + localTools.length} tools (${localTools.length} local), ${prompts.length} prompts, ${resources.length} resources`);
+  process.on("SIGINT", async () => {
+    await server.close();
+    process.exit(0);
+  });
+  log(`MCP server started (stdio) with ${tools.length + localTools.length} tools (${localTools.length} local), ${prompts.length} prompts, ${resources.length} resources`);
   if (!API_KEY) {
     log("API key not configured. Get one at https://wireweave.org", "warn");
   }
 }
+async function startHttp() {
+  const transports = /* @__PURE__ */ new Map();
+  const httpServer = createServer(async (req, res) => {
+    const url = new URL(req.url || "/", `http://${req.headers.host}`);
+    if (url.pathname !== "/mcp") {
+      if (url.pathname === "/health") {
+        res.writeHead(200, { "Content-Type": "application/json" });
+        res.end(JSON.stringify({
+          status: "ok",
+          transport: "streamable-http",
+          activeSessions: transports.size,
+          tools: tools.length + localTools.length
+        }));
+        return;
+      }
+      res.writeHead(404, { "Content-Type": "application/json" });
+      res.end(JSON.stringify({ error: "Not found. Use /mcp for MCP protocol." }));
+      return;
+    }
+    const sessionId = req.headers["mcp-session-id"];
+    if (sessionId && transports.has(sessionId)) {
+      const existing = transports.get(sessionId);
+      await existing.transport.handleRequest(req, res);
+      return;
+    }
+    if (req.method === "DELETE") {
+      res.writeHead(404);
+      res.end();
+      return;
+    }
+    if (req.method !== "POST") {
+      res.writeHead(400, { "Content-Type": "application/json" });
+      res.end(JSON.stringify({ error: "Session required. Send an initialization POST first." }));
+      return;
+    }
+    const MAX_SESSIONS = parseInt(process.env.WIREWEAVE_MCP_MAX_SESSIONS || "10", 10);
+    if (transports.size >= MAX_SESSIONS) {
+      res.writeHead(503, { "Content-Type": "application/json" });
+      res.end(JSON.stringify({ error: "Too many active sessions" }));
+      return;
+    }
+    const server = createMcpServer();
+    const transport = new StreamableHTTPServerTransport({
+      sessionIdGenerator: () => randomUUID(),
+      onsessioninitialized: (sessionId2) => {
+        transports.set(sessionId2, { server, transport });
+        log(`New session: ${sessionId2}`);
+      }
+    });
+    transport.onclose = () => {
+      if (transport.sessionId) {
+        transports.delete(transport.sessionId);
+        log(`Session closed: ${transport.sessionId}`);
+      }
+    };
+    await server.connect(transport);
+    await transport.handleRequest(req, res);
+  });
+  httpServer.listen(HTTP_PORT, HTTP_HOST, () => {
+    log(`MCP server started (streamable-http) at http://${HTTP_HOST}:${HTTP_PORT}/mcp`);
+    log(`Health check: http://${HTTP_HOST}:${HTTP_PORT}/health`);
+    log(`${tools.length + localTools.length} tools (${localTools.length} local), ${prompts.length} prompts, ${resources.length} resources`);
+    if (!API_KEY) {
+      log("API key not configured. Get one at https://wireweave.org", "warn");
+    }
+  });
+  process.on("SIGINT", async () => {
+    log("Shutting down...");
+    for (const [, { server }] of transports) {
+      await server.close();
+    }
+    httpServer.close();
+    process.exit(0);
+  });
+}
+var main = isHttpMode ? startHttp : startStdio;
 main().catch(() => {
   log("Failed to start server", "error");
   process.exit(1);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wireweave/mcp-server",
-  "version": "1.6.2-beta.2",
+  "version": "1.7.0-beta.0",
   "description": "MCP server for Wireweave DSL - Thin client for API Server",
   "type": "module",
   "main": "dist/index.js",
@@ -12,6 +12,7 @@
     "build": "tsup",
     "dev": "tsup --watch",
     "start": "node dist/index.js",
+    "start:http": "node dist/index.js --http",
     "typecheck": "tsc --noEmit",
     "test": "vitest run"
   },