@niroai/mcp-server 1.0.0 → 1.0.1

package/index.js

@@ -44,11 +44,15 @@ let sessionId = null;
  */
 function connectSse() {
   return new Promise((resolve, reject) => {
-    const sseUrl = `${API_URL}/mcp/sse`;
-    process.stderr.write(`[niro-mcp] Connecting to ${sseUrl}\n`);
+    const sseUrl = `${API_URL}/mcp/sse?api_key=${encodeURIComponent(API_KEY)}&project_id=${encodeURIComponent(PROJECT_ID)}`;
+    process.stderr.write(`[niro-mcp] Connecting to ${API_URL}/mcp/sse\n`);
 
     const req = httpModule.get(sseUrl, {
-      headers: { "Accept": "text/event-stream" },
+      headers: {
+        "Accept": "text/event-stream",
+        "X-Niro-Api-Key": API_KEY,
+        "X-Niro-Project-Id": PROJECT_ID,
+      },
     }, (res) => {
       if (res.statusCode !== 200) {
         reject(new Error(`SSE connection failed: HTTP ${res.statusCode}`));
@@ -57,31 +61,41 @@ function connectSse() {
 
       let buffer = "";
 
+      let currentEvent = null;
+      let currentData = "";
+
       res.on("data", (chunk) => {
         buffer += chunk.toString();
-        const lines = buffer.split("\n");
-        buffer = lines.pop(); // Keep incomplete line in buffer
-
-        let eventName = null;
-        for (const line of lines) {
-          if (line.startsWith("event:")) {
-            eventName = line.slice(6).trim();
-          } else if (line.startsWith("data:")) {
-            const data = line.slice(5).trim();
-
-            if (eventName === "endpoint") {
-              // First event: receive the message endpoint URL with sessionId
-              const match = data.match(/sessionId=([a-f0-9-]+)/);
-              if (match) {
-                sessionId = match[1];
-                process.stderr.write(`[niro-mcp] Session established: ${sessionId}\n`);
-                resolve();
-              }
-            } else if (eventName === "message") {
-              // Response from the server — write to stdout
-              process.stdout.write(data + "\n");
+
+        // SSE events are delimited by double newline
+        let doubleNewline;
+        while ((doubleNewline = buffer.indexOf("\n\n")) !== -1) {
+          const block = buffer.slice(0, doubleNewline);
+          buffer = buffer.slice(doubleNewline + 2);
+
+          // Parse the SSE block
+          let eventName = null;
+          let dataLines = [];
+          for (const line of block.split("\n")) {
+            if (line.startsWith("event:")) {
+              eventName = line.slice(6).trim();
+            } else if (line.startsWith("data:")) {
+              dataLines.push(line.slice(5));
+            }
+          }
+
+          const data = dataLines.join("\n").trim();
+          if (!eventName || !data) continue;
+
+          if (eventName === "endpoint") {
+            const match = data.match(/sessionId=([a-f0-9-]+)/);
+            if (match) {
+              sessionId = match[1];
+              process.stderr.write(`[niro-mcp] Session established: ${sessionId}\n`);
+              resolve();
             }
-            eventName = null;
+          } else if (eventName === "message") {
+            process.stdout.write(data + "\n");
           }
         }
       });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@niroai/mcp-server",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "Niro MCP Server — code intelligence for AI coding assistants",
   "bin": {
     "mcp-server": "index.js"