gemini-design-mcp 3.7.2 → 3.8.0

package/build/index.js

@@ -1,6 +1,9 @@
 #!/usr/bin/env node
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
+import express from "express";
+import cors from "cors";
 import { createFrontendSchema, createFrontend } from "./tools/create-frontend.js";
 import { modifyFrontendSchema, modifyFrontend } from "./tools/modify-frontend.js";
 import { snippetFrontendSchema, snippetFrontend } from "./tools/snippet-frontend.js";
@@ -8,7 +11,7 @@ import { generateVibesSchema, generateVibes } from "./tools/generate-vibes.js";
 // Create MCP server
 const server = new McpServer({
     name: "gemini-design-mcp",
-    version: "3.2.0",
+    version: "3.8.0",
 });
 // =============================================================================
 // TOOL 1: CREATE_FRONTEND
@@ -279,9 +282,89 @@ Keywords: minimal, whitespace, gallery, clean, sophisticated`, generateVibesSche
 // START SERVER
 // =============================================================================
 async function main() {
-    const transport = new StdioServerTransport();
-    await server.connect(transport);
-    console.error("gemini-design-mcp v3.2.0 running on stdio");
+    const transportType = process.env.TRANSPORT || "stdio";
+
+    if (transportType === "sse") {
+        // SSE/HTTP mode for remote connections
+        const app = express();
+
+        app.use(cors({
+            origin: "*",
+            methods: ["GET", "POST", "OPTIONS"],
+            allowedHeaders: ["Content-Type", "Authorization", "mcp-session-id"],
+            credentials: true,
+        }));
+        app.use(express.json());
+
+        // Store active transports by session
+        const transports = new Map();
+
+        // Health check endpoint
+        app.get("/health", (req, res) => {
+            res.json({
+                status: "ok",
+                service: "gemini-design-mcp",
+                version: "3.8.0",
+                transport: "sse",
+                timestamp: Date.now(),
+            });
+        });
+
+        // SSE endpoint - establishes connection
+        app.get("/sse", async (req, res) => {
+            const authHeader = req.headers.authorization;
+            const apiKey = authHeader?.replace("Bearer ", "");
+
+            if (!apiKey || !apiKey.startsWith("gd_")) {
+                return res.status(401).json({
+                    error: "Missing or invalid API key. Use Authorization: Bearer gd_xxx",
+                });
+            }
+
+            // Set API_KEY env var for this request (tools read from process.env)
+            process.env.API_KEY = apiKey;
+
+            // Create SSE transport
+            const transport = new SSEServerTransport("/messages", res);
+            const sessionId = `sess_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
+            transports.set(sessionId, transport);
+
+            // Send session ID to client
+            res.setHeader("mcp-session-id", sessionId);
+
+            // Clean up on disconnect
+            res.on("close", () => {
+                transports.delete(sessionId);
+            });
+
+            await server.connect(transport);
+        });
+
+        // Messages endpoint - receives JSON-RPC from client
+        app.post("/messages", async (req, res) => {
+            const sessionId = req.query.sessionId || req.headers["mcp-session-id"];
+            const transport = transports.get(sessionId);
+
+            if (!transport) {
+                return res.status(400).json({ error: "Invalid or expired session" });
+            }
+
+            await transport.handlePostMessage(req, res);
+        });
+
+        const port = parseInt(process.env.PORT || "3000", 10);
+        app.listen(port, "0.0.0.0", () => {
+            console.log(`gemini-design-mcp v3.8.0 running on SSE transport`);
+            console.log(`  SSE endpoint: http://0.0.0.0:${port}/sse`);
+            console.log(`  Messages endpoint: http://0.0.0.0:${port}/messages`);
+            console.log(`  Health check: http://0.0.0.0:${port}/health`);
+        });
+    } else {
+        // Stdio mode (default, for local IDE usage)
+        const transport = new StdioServerTransport();
+        await server.connect(transport);
+        console.error("gemini-design-mcp v3.8.0 running on stdio");
+    }
 }
 main().catch((error) => {
     console.error("Fatal error:", error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gemini-design-mcp",
-  "version": "3.7.2",
+  "version": "3.8.0",
   "description": "MCP server that uses Gemini 3 Pro for frontend/design code generation",
   "main": "build/index.js",
   "bin": {
@@ -11,6 +11,7 @@
     "build": "tsc",
     "dev": "tsc --watch",
     "start": "node build/index.js",
+    "start:sse": "TRANSPORT=sse node build/index.js",
     "prepublishOnly": "echo 'Publishing...'"
   },
   "keywords": [
@@ -28,10 +29,14 @@
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.0.0",
     "@google/genai": "^1.33.0",
-    "zod": "^3.23.0"
+    "zod": "^3.23.0",
+    "express": "^4.21.0",
+    "cors": "^2.8.5"
   },
   "devDependencies": {
     "@types/node": "^20.0.0",
+    "@types/express": "^4.17.21",
+    "@types/cors": "^2.8.17",
     "typescript": "^5.0.0"
   },
   "engines": {