mcp-optimizer 0.0.3-alpha.1 → 0.0.5-alpha.1

package/dist/mcpServer.js

@@ -39,6 +39,7 @@ const mcp_js_1 = require("@modelcontextprotocol/sdk/server/mcp.js");
 const zod_1 = require("zod");
 const lighthouseRunner_1 = require("./runner/lighthouseRunner");
 const sse_js_1 = require("@modelcontextprotocol/sdk/server/sse.js");
+const stdio_js_1 = require("@modelcontextprotocol/sdk/server/stdio.js");
 const http = __importStar(require("http"));
 const fixer_1 = require("./fix/fixer");
 class LighthouseMcpServer {
@@ -240,6 +241,37 @@ async function startMcpServer() {
     const port = Number(portEnv || 5000);
     const mcp = new LighthouseMcpServer();
     let sseTransport = null;
+    let stdioTransport = null;
+    // If process stdin/stdout appear to be non-TTY pipes, assume we're being
+    // launched as a stdio child by an MCP host and use the stdio transport.
+    const shouldUseStdio = (process.stdin && !process.stdin.isTTY) && (process.stdout && !process.stdout.isTTY);
+    if (shouldUseStdio) {
+        // Redirect console output to stderr to avoid corrupting the JSON stdio protocol
+        const _orig = { log: console.log, info: console.info, warn: console.warn };
+        console.log = (...args) => { process.stderr.write(args.map(String).join(' ') + '\n'); };
+        console.info = console.log;
+        console.warn = console.log;
+        try {
+            stdioTransport = new stdio_js_1.StdioServerTransport(process.stdin, process.stdout);
+            await mcp.connect(stdioTransport);
+            console.error('Stdio: connected to parent process over stdio');
+            // When running over stdio we do not start the HTTP server; stay alive
+            // and let the parent coordinate messages. Return a promise that
+            // resolves when the transport closes.
+            return new Promise((resolve, reject) => {
+                stdioTransport.onclose = () => resolve();
+                stdioTransport.onerror = (err) => reject(err);
+            });
+        }
+        catch (err) {
+            console.error('Stdio: failed to start transport, falling back to HTTP:', err);
+            stdioTransport = null;
+            // restore console in case we fall back to HTTP server mode
+            console.log = _orig.log;
+            console.info = _orig.info;
+            console.warn = _orig.warn;
+        }
+    }
     const pendingPosts = [];
     const { Readable, Writable } = await (async () => {
         const mod = await Promise.resolve().then(() => __importStar(require('stream')));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-optimizer",
-  "version": "0.0.3-alpha.1",
+  "version": "0.0.5-alpha.1",
   "description": "An MCP server that runs Lighthouse audits and enables LLMs to automatically optimize your code.",
   "main": "dist/index.js",
   "bin": {

package/src/mcpServer.ts

@@ -3,6 +3,7 @@ import { z } from "zod";
 import { runLighthouseAudit } from "./runner/lighthouseRunner";
 import { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
 import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import * as http from 'http';
 import { IncomingMessage, ServerResponse } from 'http';
 import { autoFixFromReport } from './fix/fixer';
@@ -225,6 +226,36 @@ export async function startMcpServer(): Promise<void> {
   const port = Number(portEnv || 5000);
   const mcp = new LighthouseMcpServer();
   let sseTransport: SSEServerTransport | null = null;
+  let stdioTransport: StdioServerTransport | null = null;
+  // If process stdin/stdout appear to be non-TTY pipes, assume we're being
+  // launched as a stdio child by an MCP host and use the stdio transport.
+  const shouldUseStdio = (process.stdin && !process.stdin.isTTY) && (process.stdout && !process.stdout.isTTY);
+  if (shouldUseStdio) {
+    // Redirect console output to stderr to avoid corrupting the JSON stdio protocol
+    const _orig = { log: console.log, info: console.info, warn: console.warn };
+    console.log = (...args: any[]) => { process.stderr.write(args.map(String).join(' ') + '\n'); };
+    console.info = console.log;
+    console.warn = console.log;
+    try {
+      stdioTransport = new StdioServerTransport(process.stdin, process.stdout);
+      await mcp.connect(stdioTransport as unknown as Transport);
+      console.error('Stdio: connected to parent process over stdio');
+      // When running over stdio we do not start the HTTP server; stay alive
+      // and let the parent coordinate messages. Return a promise that
+      // resolves when the transport closes.
+      return new Promise((resolve, reject) => {
+        stdioTransport!.onclose = () => resolve();
+        stdioTransport!.onerror = (err: any) => reject(err);
+      });
+    } catch (err) {
+      console.error('Stdio: failed to start transport, falling back to HTTP:', err);
+      stdioTransport = null;
+      // restore console in case we fall back to HTTP server mode
+      console.log = _orig.log;
+      console.info = _orig.info;
+      console.warn = _orig.warn;
+    }
+  }
   const pendingPosts: Array<{ body: string; url: string | undefined; headers: any }> = [];
 
   const { Readable, Writable } = await (async () => {