added sse http support

src/index.ts

@@ -6,6 +6,10 @@ import { z } from "zod";
 import { searchLibraries, fetchLibraryDocumentation } from "./lib/api.js";
 import { formatSearchResults } from "./lib/utils.js";
 import dotenv from "dotenv";
+import { createServer } from "http";
+import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
+import { parse } from "url";
 
 // Load environment variables from .env file if present
 dotenv.config();
@@ -23,21 +27,25 @@ if (process.env.DEFAULT_MINIMUM_TOKENS) {
   }
 }
 
-// Create server instance
+// Store SSE transports by session ID
-const server = new McpServer({
+const sseTransports: Record<string, SSEServerTransport> = {};
-  name: "Context7",
-  description: "Retrieves up-to-date documentation and code examples for any library.",
-  version: "1.0.6",
-  capabilities: {
-    resources: {},
-    tools: {},
-  },
-});
 
-// Register Context7 tools
+// Function to create a new server instance with all tools registered
-server.tool(
+function createServerInstance() {
-  "resolve-library-id",
+  const server = new McpServer({
-  `Resolves a package/product name to a Context7-compatible library ID and returns a list of matching libraries.
+    name: "Context7",
+    description: "Retrieves up-to-date documentation and code examples for any library.",
+    version: "1.0.7",
+    capabilities: {
+      resources: {},
+      tools: {},
+    },
+  });
+
+  // Register Context7 tools
+  server.tool(
+    "resolve-library-id",
+    `Resolves a package/product name to a Context7-compatible library ID and returns a list of matching libraries.
 
 You MUST call this function before 'get-library-docs' to obtain a valid Context7-compatible library ID UNLESS the user explicitly provides a library ID in the format '/org/project' or '/org/project/version' in their query.
 
@@ -56,43 +64,43 @@ Response Format:
 - If no good matches exist, clearly state this and suggest query refinements
 
 For ambiguous queries, request clarification before proceeding with a best-guess match.`,
-  {
+    {
-    libraryName: z
+      libraryName: z
-      .string()
+        .string()
-      .describe("Library name to search for and retrieve a Context7-compatible library ID."),
+        .describe("Library name to search for and retrieve a Context7-compatible library ID."),
-  },
+    },
-  async ({ libraryName }) => {
+    async ({ libraryName }) => {
-    const searchResponse = await searchLibraries(libraryName);
+      const searchResponse = await searchLibraries(libraryName);
+
+      if (!searchResponse || !searchResponse.results) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: "Failed to retrieve library documentation data from Context7",
+            },
+          ],
+        };
+      }
+
+      if (searchResponse.results.length === 0) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: "No documentation libraries available",
+            },
+          ],
+        };
+      }
+
+      const resultsText = formatSearchResults(searchResponse);
 
-    if (!searchResponse || !searchResponse.results) {
       return {
         content: [
           {
             type: "text",
-            text: "Failed to retrieve library documentation data from Context7",
+            text: `Available Libraries (top matches):
-          },
-        ],
-      };
-    }
-
-    if (searchResponse.results.length === 0) {
-      return {
-        content: [
-          {
-            type: "text",
-            text: "No documentation libraries available",
-          },
-        ],
-      };
-    }
-
-    const resultsText = formatSearchResults(searchResponse);
-
-    return {
-      content: [
-        {
-          type: "text",
-          text: `Available Libraries (top matches):
 
 Each result includes:
 - Library ID: Context7-compatible identifier (format: /org/project)
@@ -107,65 +115,151 @@ For best results, select libraries based on name match, trust score, snippet cov
 ----------
 
 ${resultsText}`,
-        },
-      ],
-    };
-  }
-);
-
-server.tool(
-  "get-library-docs",
-  "Fetches up-to-date documentation for a library. You must call 'resolve-library-id' first to obtain the exact Context7-compatible library ID required to use this tool, UNLESS the user explicitly provides a library ID in the format '/org/project' or '/org/project/version' in their query.",
-  {
-    context7CompatibleLibraryID: z
-      .string()
-      .describe(
-        "Exact Context7-compatible library ID (e.g., '/mongodb/docs', '/vercel/next.js', '/supabase/supabase', '/vercel/next.js/v14.3.0-canary.87') retrieved from 'resolve-library-id' or directly from user query in the format '/org/project' or '/org/project/version'."
-      ),
-    topic: z
-      .string()
-      .optional()
-      .describe("Topic to focus documentation on (e.g., 'hooks', 'routing')."),
-    tokens: z
-      .preprocess((val) => (typeof val === "string" ? Number(val) : val), z.number())
-      .transform((val) => (val < DEFAULT_MINIMUM_TOKENS ? DEFAULT_MINIMUM_TOKENS : val))
-      .optional()
-      .describe(
-        `Maximum number of tokens of documentation to retrieve (default: ${DEFAULT_MINIMUM_TOKENS}). Higher values provide more context but consume more tokens.`
-      ),
-  },
-  async ({ context7CompatibleLibraryID, tokens = DEFAULT_MINIMUM_TOKENS, topic = "" }) => {
-    const documentationText = await fetchLibraryDocumentation(context7CompatibleLibraryID, {
-      tokens,
-      topic,
-    });
-
-    if (!documentationText) {
-      return {
-        content: [
-          {
-            type: "text",
-            text: "Documentation not found or not finalized for this library. This might have happened because you used an invalid Context7-compatible library ID. To get a valid Context7-compatible library ID, use the 'resolve-library-id' with the package name you wish to retrieve documentation for.",
           },
         ],
       };
     }
+  );
 
-    return {
+  server.tool(
-      content: [
+    "get-library-docs",
-        {
+    "Fetches up-to-date documentation for a library. You must call 'resolve-library-id' first to obtain the exact Context7-compatible library ID required to use this tool, UNLESS the user explicitly provides a library ID in the format '/org/project' or '/org/project/version' in their query.",
-          type: "text",
+    {
-          text: documentationText,
+      context7CompatibleLibraryID: z
-        },
+        .string()
-      ],
+        .describe(
-    };
+          "Exact Context7-compatible library ID (e.g., '/mongodb/docs', '/vercel/next.js', '/supabase/supabase', '/vercel/next.js/v14.3.0-canary.87') retrieved from 'resolve-library-id' or directly from user query in the format '/org/project' or '/org/project/version'."
-  }
+        ),
-);
+      topic: z
+        .string()
+        .optional()
+        .describe("Topic to focus documentation on (e.g., 'hooks', 'routing')."),
+      tokens: z
+        .preprocess((val) => (typeof val === "string" ? Number(val) : val), z.number())
+        .transform((val) => (val < DEFAULT_MINIMUM_TOKENS ? DEFAULT_MINIMUM_TOKENS : val))
+        .optional()
+        .describe(
+          `Maximum number of tokens of documentation to retrieve (default: ${DEFAULT_MINIMUM_TOKENS}). Higher values provide more context but consume more tokens.`
+        ),
+    },
+    async ({ context7CompatibleLibraryID, tokens = DEFAULT_MINIMUM_TOKENS, topic = "" }) => {
+      const documentationText = await fetchLibraryDocumentation(context7CompatibleLibraryID, {
+        tokens,
+        topic,
+      });
+
+      if (!documentationText) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: "Documentation not found or not finalized for this library. This might have happened because you used an invalid Context7-compatible library ID. To get a valid Context7-compatible library ID, use the 'resolve-library-id' with the package name you wish to retrieve documentation for.",
+            },
+          ],
+        };
+      }
+
+      return {
+        content: [
+          {
+            type: "text",
+            text: documentationText,
+          },
+        ],
+      };
+    }
+  );
+
+  return server;
+}
 
 async function main() {
-  const transport = new StdioServerTransport();
+  const transportType = process.env.MCP_TRANSPORT || "stdio";
-  await server.connect(transport);
+
-  console.error("Context7 Documentation MCP Server running on stdio");
+  if (transportType === "http" || transportType === "sse") {
+    const port = process.env.PORT ? parseInt(process.env.PORT, 10) : 3000;
+    const httpServer = createServer(async (req, res) => {
+      const url = parse(req.url || "").pathname;
+
+      // Set CORS headers for all responses
+      res.setHeader('Access-Control-Allow-Origin', '*');
+      res.setHeader('Access-Control-Allow-Methods', 'GET,POST,OPTIONS,DELETE');
+      res.setHeader('Access-Control-Allow-Headers', 'Content-Type, MCP-Session-Id, mcp-session-id');
+
+      // Handle preflight OPTIONS requests
+      if (req.method === 'OPTIONS') {
+        res.writeHead(200);
+        res.end();
+        return;
+      }
+
+      try {
+        // Create new server instance for each request
+        const requestServer = createServerInstance();
+
+        if (url === "/mcp") {
+          const transport = new StreamableHTTPServerTransport({
+            sessionIdGenerator: undefined
+          });
+          await requestServer.connect(transport);
+          await transport.handleRequest(req, res);
+        } else if (url === "/sse" && req.method === "GET") {
+          // Create new SSE transport for GET request
+          const sseTransport = new SSEServerTransport("/messages", res);
+          // Store the transport by session ID
+          sseTransports[sseTransport.sessionId] = sseTransport;
+          // Clean up transport when connection closes
+          res.on("close", () => {
+            delete sseTransports[sseTransport.sessionId];
+          });
+          await requestServer.connect(sseTransport);
+        } else if (url === "/messages" && req.method === "POST") {
+          // Get session ID from query parameters
+          const parsedUrl = parse(req.url || "", true);
+          const sessionId = parsedUrl.query.sessionId as string;
+          
+          if (!sessionId) {
+            res.writeHead(400);
+            res.end("Missing sessionId parameter");
+            return;
+          }
+
+          // Get existing transport for this session
+          const sseTransport = sseTransports[sessionId];
+          if (!sseTransport) {
+            res.writeHead(400);
+            res.end(`No transport found for sessionId: ${sessionId}`);
+            return;
+          }
+
+          // Handle the POST message with the existing transport
+          await sseTransport.handlePostMessage(req, res);
+        } else if (url === "/ping") {
+          res.writeHead(200, { 'Content-Type': 'text/plain' });
+          res.end('pong');
+        } else {
+          res.writeHead(404);
+          res.end("Not found");
+        }
+      } catch (error) {
+        console.error("Error handling request:", error);
+        if (!res.headersSent) {
+          res.writeHead(500);
+          res.end("Internal Server Error");
+        }
+      }
+    });
+
+    httpServer.listen(port, () => {
+      console.error(`Context7 Documentation MCP Server running on ${transportType.toUpperCase()} at http://localhost:${port}/mcp and legacy SSE at /sse`);
+    });
+  } else {
+    // Stdio transport - this is already stateless by nature
+    const server = createServerInstance();
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    console.error("Context7 Documentation MCP Server running on stdio");
+  }
 }
 
 main().catch((error) => {