midnight-mcp 0.1.0 → 0.1.1

package/README.md

@@ -26,7 +26,7 @@ Add to your `claude_desktop_config.json`:
 ```
 
 <details>
-<summary><strong>📍 Config file locations</strong></summary>
+<summary><strong>Config file locations</strong></summary>
 
 - **macOS**: `~/Library/Application Support/Claude/claude_desktop_config.json`
 - **Windows**: `%APPDATA%\Claude\claude_desktop_config.json`
@@ -36,6 +36,13 @@ Add to your `claude_desktop_config.json`:
 
 ### Cursor
 
+**Click the button to auto install the MCP for Cursor**
+
+[![Install MCP Server](https://cursor.com/deeplink/mcp-install-dark.svg)](https://cursor.com/en-US/install-mcp?name=midnight&config=eyJjb21tYW5kIjoibnB4IC15IG1pZG5pZ2h0LW1jcCJ9)
+
+---
+**Manual setup instructions are below if you need them:**
+
 Add to Cursor's MCP settings (Settings → MCP → Add Server):
 
 ```json
@@ -62,6 +69,8 @@ Or add to `.cursor/mcp.json` in your project:
 }
 ```
 
+
+
 ### Windsurf
 
 Add to `~/.codeium/windsurf/mcp_config.json`:
@@ -87,11 +96,11 @@ Restart your editor after adding the config. All features work out of the box—
 
 By default, the MCP uses a **hosted API** for semantic search:
 
-- ✅ **Zero configuration** — just install and use
-- ✅ **Semantic search** works immediately
-- ✅ **No API keys** needed
+- **Zero configuration** — just install and use
+- **Semantic search** works immediately
+- **No API keys** needed
 
-> **📊 Quality Metrics**: To ensure the MCP stays accurate as Midnight's codebase evolves rapidly, we collect anonymous usage metrics (query counts, relevance scores) to monitor search quality. No query content or personal data is stored. This helps us identify when re-indexing is needed and improve results over time.
+> **Quality Metrics**: To ensure the MCP stays accurate as Midnight's codebase evolves rapidly, we collect anonymous usage metrics (query counts, relevance scores) to monitor search quality. No query content or personal data is stored. This helps us identify when re-indexing is needed and improve results over time.
 
 ### Local Mode (Optional)
 
@@ -143,11 +152,11 @@ Add `"GITHUB_TOKEN": "ghp_..."` for higher GitHub API rate limits (60 → 5000 r
 | `midnight-get-latest-syntax`      | Latest syntax reference                     |
 | `midnight-health-check`           | Check server health status                  |
 | `midnight-get-status`             | Get rate limits and cache stats             |
-| `midnight-generate-contract` ✨   | AI-generate contracts from natural language |
-| `midnight-review-contract` ✨     | AI-powered security review of contracts     |
-| `midnight-document-contract` ✨   | AI-generate documentation for contracts     |
+| `midnight-generate-contract`      | AI-generate contracts from natural language |
+| `midnight-review-contract`        | AI-powered security review of contracts     |
+| `midnight-document-contract`      | AI-generate documentation for contracts     |
 
-> **✨ New AI-Powered Tools** require a client with [MCP Sampling](https://spec.modelcontextprotocol.io/specification/client/sampling/) support (e.g., Claude Desktop). Without sampling, these tools will return a helpful message instead.
+> **Note:** The AI-powered tools require a client with [MCP Sampling](https://spec.modelcontextprotocol.io/specification/client/sampling/) support (e.g., Claude Desktop). Without sampling, these tools will return a helpful message instead.
 
 ### Resource Templates (4)
 

package/dist/server.d.ts

@@ -1,4 +1,8 @@
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+/**
+ * Clear all subscriptions (useful for server restart/testing)
+ */
+export declare function clearSubscriptions(): void;
 /**
  * Create and configure the MCP server
  */

package/dist/server.js

@@ -15,6 +15,13 @@ const SERVER_INFO = {
 };
 // Resource subscriptions tracking
 const resourceSubscriptions = new Set();
+/**
+ * Clear all subscriptions (useful for server restart/testing)
+ */
+export function clearSubscriptions() {
+    resourceSubscriptions.clear();
+    logger.debug("Subscriptions cleared");
+}
 // Resource templates for parameterized resources (RFC 6570 URI Templates)
 const resourceTemplates = [
     {
@@ -295,7 +302,7 @@ function registerSubscriptionHandlers(server) {
         const isValid = validPrefixes.some((prefix) => uri.startsWith(prefix));
         if (!isValid) {
             logger.warn(`Invalid subscription URI: ${uri}`);
-            return {};
+            throw new Error(`Invalid subscription URI: ${uri}. Valid prefixes: ${validPrefixes.join(", ")}`);
         }
         resourceSubscriptions.add(uri);
         logger.debug(`Active subscriptions: ${resourceSubscriptions.size}`);
@@ -355,7 +362,17 @@ function setupSampling(server) {
             }, 
             // Use a schema that matches the expected response
             {
-                parse: (data) => data,
+                parse: (data) => {
+                    const response = data;
+                    // Basic validation of expected response structure
+                    if (!response || typeof response !== "object") {
+                        throw new Error("Invalid sampling response: expected object");
+                    }
+                    if (!response.content || typeof response.content !== "object") {
+                        throw new Error("Invalid sampling response: missing content");
+                    }
+                    return response;
+                },
                 _def: { typeName: "SamplingResponse" },
             });
             return response;

package/dist/services/sampling.js

@@ -191,10 +191,31 @@ Respond in JSON format:
             maxTokens: 4096,
             temperature: 0.2,
         });
-        // Parse JSON response
+        // Parse JSON response with error handling
         const jsonMatch = response.match(/\{[\s\S]*\}/);
         if (jsonMatch) {
-            return JSON.parse(jsonMatch[0]);
+            try {
+                const parsed = JSON.parse(jsonMatch[0]);
+                // Validate expected structure
+                return {
+                    summary: typeof parsed.summary === "string"
+                        ? parsed.summary
+                        : "Review complete",
+                    issues: Array.isArray(parsed.issues) ? parsed.issues : [],
+                    improvedCode: typeof parsed.improvedCode === "string"
+                        ? parsed.improvedCode
+                        : undefined,
+                };
+            }
+            catch (parseError) {
+                logger.warn("Failed to parse JSON from LLM response", {
+                    error: String(parseError),
+                });
+                return {
+                    summary: response,
+                    issues: [],
+                };
+            }
         }
         return {
             summary: response,

package/dist/tools/analyze.js

@@ -230,46 +230,77 @@ function getPrivacyConsiderations(circuit) {
     }
     return considerations;
 }
-// Output schemas for analysis tools
+// Output schemas for analysis tools - aligned with actual function return types
 const analyzeContractOutputSchema = {
     type: "object",
     properties: {
-        name: { type: "string", description: "Contract name" },
-        summary: { type: "string", description: "Brief contract summary" },
-        ledgerState: {
-            type: "array",
-            description: "Ledger state fields",
-            items: {
-                type: "object",
-                properties: {
-                    name: { type: "string" },
-                    type: { type: "string" },
-                    isPrivate: { type: "boolean" },
-                },
+        summary: {
+            type: "object",
+            description: "Summary statistics of the contract",
+            properties: {
+                hasLedger: { type: "boolean" },
+                hasCircuits: { type: "boolean" },
+                hasWitnesses: { type: "boolean" },
+                totalLines: { type: "number" },
+                publicCircuits: { type: "number" },
+                privateCircuits: { type: "number" },
+                publicState: { type: "number" },
+                privateState: { type: "number" },
             },
         },
-        circuits: {
-            type: "array",
-            description: "Circuit definitions",
-            items: {
-                type: "object",
-                properties: {
-                    name: { type: "string" },
-                    isPublic: { type: "boolean" },
-                    parameters: { type: "array", items: { type: "object" } },
-                    purpose: { type: "string" },
+        structure: {
+            type: "object",
+            description: "Contract structure breakdown",
+            properties: {
+                imports: { type: "array", items: { type: "string" } },
+                exports: { type: "array", items: { type: "string" } },
+                ledger: {
+                    type: "array",
+                    description: "Ledger state fields",
+                    items: {
+                        type: "object",
+                        properties: {
+                            name: { type: "string" },
+                            type: { type: "string" },
+                            isPrivate: { type: "boolean" },
+                        },
+                    },
                 },
-            },
-        },
-        witnesses: {
-            type: "array",
-            description: "Witness functions",
-            items: {
-                type: "object",
-                properties: {
-                    name: { type: "string" },
-                    returnType: { type: "string" },
-                    purpose: { type: "string" },
+                circuits: {
+                    type: "array",
+                    description: "Circuit definitions",
+                    items: {
+                        type: "object",
+                        properties: {
+                            name: { type: "string" },
+                            isPublic: { type: "boolean" },
+                            parameters: { type: "array", items: { type: "object" } },
+                            returnType: { type: "string" },
+                        },
+                    },
+                },
+                witnesses: {
+                    type: "array",
+                    description: "Witness functions",
+                    items: {
+                        type: "object",
+                        properties: {
+                            name: { type: "string" },
+                            parameters: { type: "array", items: { type: "object" } },
+                            returnType: { type: "string" },
+                        },
+                    },
+                },
+                types: {
+                    type: "array",
+                    description: "Type definitions",
+                    items: {
+                        type: "object",
+                        properties: {
+                            name: { type: "string" },
+                            definition: { type: "string" },
+                        },
+                    },
                 },
             },
         },
@@ -294,31 +325,14 @@ const analyzeContractOutputSchema = {
             description: "Recommendations for improvement",
         },
     },
-    required: ["summary", "circuits"],
+    required: ["summary", "structure", "securityFindings", "recommendations"],
     description: "Detailed contract analysis with security findings",
 };
 const explainCircuitOutputSchema = {
     type: "object",
     properties: {
-        name: { type: "string", description: "Circuit name" },
-        type: {
-            type: "string",
-            description: "Type: circuit or witness",
-        },
+        circuitName: { type: "string", description: "Circuit name" },
         isPublic: { type: "boolean", description: "Whether it's exported" },
-        plainEnglishExplanation: {
-            type: "string",
-            description: "Plain language explanation",
-        },
-        zkProofImplications: {
-            type: "string",
-            description: "Zero-knowledge proof implications",
-        },
-        privacyConsiderations: {
-            type: "array",
-            items: { type: "string" },
-            description: "Privacy-related considerations",
-        },
         parameters: {
             type: "array",
             items: {
@@ -326,13 +340,37 @@ const explainCircuitOutputSchema = {
                 properties: {
                     name: { type: "string" },
                     type: { type: "string" },
-                    description: { type: "string" },
                 },
             },
+            description: "Circuit parameters",
+        },
+        returnType: { type: "string", description: "Return type" },
+        explanation: {
+            type: "string",
+            description: "Plain language explanation",
+        },
+        operations: {
+            type: "array",
+            items: { type: "string" },
+            description: "Operations performed by the circuit",
+        },
+        zkImplications: {
+            type: "array",
+            items: { type: "string" },
+            description: "Zero-knowledge proof implications",
+        },
+        privacyConsiderations: {
+            type: "array",
+            items: { type: "string" },
+            description: "Privacy-related considerations",
         },
-        returnType: { type: "string" },
     },
-    required: ["name", "plainEnglishExplanation"],
+    required: [
+        "circuitName",
+        "explanation",
+        "zkImplications",
+        "privacyConsiderations",
+    ],
     description: "Detailed circuit explanation with privacy analysis",
 };
 // Tool definitions for MCP

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "midnight-mcp",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Model Context Protocol Server for Midnight Blockchain Development",
   "type": "module",
   "main": "dist/index.js",