vibeshield-mcp 0.1.0

package/CLAUDE_CONFIG.json

@@ -0,0 +1,13 @@
+{
+    "mcpServers": {
+        "vibeshield": {
+            "command": "node",
+            "args": [
+                "/home/glebs/VibeShield/vibeshield-mcp-shim/index.mjs"
+            ],
+            "env": {
+                "VIBESHIELD_TOKEN": "hoNWJ75uxe900YXmBJ1I1Weh9IWVNnkw"
+            }
+        }
+    }
+}

package/index.mjs

@@ -0,0 +1,113 @@
+#!/usr/bin/env node
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+
+// Product-owned (NOT user config)
+const GATEWAY_URL = "https://vibeshield-gateway-651372710922.europe-north1.run.app";
+const TOOL_PATH = "/mcp/tools/vibeshield.analyze_prompt";
+
+// User-owned (only)
+const TOKEN = process.env.VIBESHIELD_TOKEN ?? process.env.VIBESHIELD_API_KEY;
+
+if (!TOKEN) {
+  console.error(
+    "VibeShield: missing VIBESHIELD_TOKEN. Copy your token from the VibeShield dashboard and paste it into your MCP server env."
+  );
+  process.exit(1);
+}
+
+const server = new Server(
+  { name: "vibeshield", version: "0.1.0" },
+  { capabilities: { tools: {} } }
+);
+
+server.setRequestHandler(ListToolsRequestSchema, async () => ({
+  tools: [
+    {
+      name: "analyze_prompt",
+      description:
+        "Analyze a plain-English prompt, select intent/policy packs, and return an injected rewrittenPrompt.",
+      inputSchema: {
+        type: "object",
+        properties: {
+          prompt: { type: "string", minLength: 1 },
+          context: { type: "object" },
+        },
+        required: ["prompt"],
+        additionalProperties: false,
+      },
+    },
+  ],
+}));
+
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  if (request.params.name !== "analyze_prompt") {
+    return {
+      content: [{ type: "text", text: `Unknown tool: ${request.params.name}` }],
+      isError: true,
+    };
+  }
+
+  const args = request.params.arguments ?? {};
+  if (typeof args.prompt !== "string" || args.prompt.trim().length === 0) {
+    return {
+      content: [{ type: "text", text: "Invalid input: 'prompt' must be a non-empty string." }],
+      isError: true,
+    };
+  }
+
+  try {
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), 60000);
+
+    const response = await fetch(`${GATEWAY_URL}${TOOL_PATH}`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "Authorization": `Bearer ${TOKEN}`,
+      },
+      body: JSON.stringify({ prompt: args.prompt, context: args.context }),
+      signal: controller.signal,
+    });
+
+    clearTimeout(timeoutId);
+
+    const raw = await response.text();
+    if (!response.ok) {
+      if (response.status === 401 || response.status === 403) {
+        return {
+          content: [{ type: "text", text: "VibeShield token is invalid or revoked. Rotate it in the dashboard." }],
+          isError: true
+        };
+      }
+      if (response.status === 429) {
+        return {
+          content: [{ type: "text", text: "VibeShield rate limit exceeded. Try again later." }],
+          isError: true
+        };
+      }
+      return {
+        content: [{ type: "text", text: `VibeShield gateway error ${response.status}: ${raw}` }],
+        isError: true,
+      };
+    }
+
+    // Prefer returning rewrittenPrompt; fallback to raw
+    try {
+      const json = JSON.parse(raw);
+      return { content: [{ type: "text", text: json?.rewrittenPrompt ?? raw }] };
+    } catch {
+      return { content: [{ type: "text", text: raw }] };
+    }
+  } catch (error) {
+    if (error.name === 'AbortError') {
+       return { content: [{ type: "text", text: "VibeShield gateway timed out. Please try again." }], isError: true };
+    }
+    const message = error?.message ?? String(error);
+    console.error("VibeShield analyze_prompt failed:", message);
+    return { content: [{ type: "text", text: message }], isError: true };
+  }
+});
+
+await server.connect(new StdioServerTransport());

package/package.json

@@ -0,0 +1,15 @@
+{
+  "name": "vibeshield-mcp",
+  "version": "0.1.0",
+  "description": "VibeShield MCP shim (stdio)",
+  "type": "module",
+  "bin": {
+    "vibeshield-mcp": "./index.mjs"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.0"
+  }
+}