kickload-watcher-mcp 0.1.0 → 0.1.2

package/index.js

@@ -1,270 +1,266 @@
-#!/usr/bin/env node
-
-// ── Global error guards — must be first ──────────────────────────────────────
-process.on("uncaughtException", (err) => {
-  console.error("\n❌ Uncaught error:", err.message);
-  if (process.env.LOG_LEVEL === "debug") console.error(err.stack);
-  process.exit(1);
-});
-
-process.on("unhandledRejection", (reason) => {
-  const msg = reason instanceof Error ? reason.message : String(reason);
-  console.error("\n❌ Unhandled rejection:", msg);
-  if (process.env.LOG_LEVEL === "debug" && reason instanceof Error) console.error(reason.stack);
-  process.exit(1);
-});
-
-// ── Node version check ────────────────────────────────────────────────────────
-const nodeMajor = parseInt(process.versions.node.split(".")[0]);
-if (nodeMajor < 18) {
-  console.error(`\n❌ Node.js 18+ required. You have ${process.versions.node}.`);
-  console.error("   Download: https://nodejs.org\n");
-  process.exit(1);
-}
-
-// ── First-run setup (writes .env if missing, reloads env in-process) ─────────
-// STARTUP FIX: runFirstRunSetup no longer exits — continues execution after writing .env
-import { runFirstRunSetup, ensureGitignore } from "./setup.js";
-await runFirstRunSetup();
-
-// ── Load env ──────────────────────────────────────────────────────────────────
-import dotenv from "dotenv";
-dotenv.config();
-
-// ── All other imports depend on env being loaded ──────────────────────────────
-import { Server }               from "@modelcontextprotocol/sdk/server/index.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
-
-import { config, validateConfig }                                 from "./config.js";
-import { initEmailSender, sendResultEmail, verifyEmailService }  from "./email-sender.js";
-import { watchIdentityMap, getEmailForUser }                      from "./identity-map.js";
-import { startCompliancePoller, onNewSession }                    from "./compliance-poller.js";
-import { startFileWatcher, onNewApiFile }                         from "./file-watcher.js";
-import { runPhase1Pipeline, handleFileEvent,
-         handleComplianceSession, handleGithubEvent }             from "./pipeline.js";
-import { generateTestPrompt }                                     from "./test-generator.js";
-import { startGithubWebhookServer, onGithubPR }                   from "./github-webhook.js";
-import { ensureNgrokInstalled, stopNgrok }                        from "./ngrok-manager.js";
-import { createLogger }                                           from "./logger.js";
-
-const logger = createLogger("main");
-
-// ── Config validation (exits on missing required fields) ─────────────────────
-validateConfig();
-ensureGitignore();
-
-// ── PROBLEM 2 — Email strict mode: init + verify before anything else ─────────
-console.log("\n🔧 Initializing subsystems...");
-
-try {
-  initEmailSender();
-} catch (err) {
-  console.error(`\n❌ Email initialization failed: ${err.message}`);
-  console.error("   Check EMAIL_PROVIDER and related settings in .env\n");
-  process.exit(1);
-}
-
-// Verify SMTP credentials — exits if broken
-await verifyEmailService();
-
-// ── ngrok binary pre-fetch (only if auth token is configured) ─────────────────
-// PROBLEM 1: We no longer check NGROK_ENABLED at startup for localhost backends
-// (pipeline.js handles that automatically). We still pre-install the binary
-// if a token is present, so the first tunnel starts faster.
-if (config.ngrok.authToken) {
-  logger.info("Pre-fetching ngrok binary...");
-  try {
-    await ensureNgrokInstalled();
-    logger.info("✅ ngrok ready.");
-  } catch (err) {
-    // Non-fatal — pipeline will handle the failure gracefully
-    logger.warn(`ngrok pre-install failed: ${err.message}`);
-    logger.warn("Localhost backends will attempt ngrok on first use.");
-  }
-} else {
-  logger.warn("NGROK_AUTHTOKEN not set — ngrok will attempt unauthenticated tunnels for localhost backends.");
-  logger.warn("Add NGROK_AUTHTOKEN to .env to fix: https://dashboard.ngrok.com");
-}
-
-// ── Init remaining subsystems ────────────────────────────────────────────────
-watchIdentityMap();
-
-onNewSession(handleComplianceSession);
-onNewApiFile(handleFileEvent);
-
-if (config.triggerMode === "claudecode" || config.triggerMode === "both") {
-  startCompliancePoller();
-  startFileWatcher();
-}
-
-if (config.triggerMode === "github" || config.triggerMode === "both") {
-  onGithubPR(handleGithubEvent);
-  startGithubWebhookServer();
-}
-
-// ── Graceful shutdown ─────────────────────────────────────────────────────────
-const shutdown = (sig) => {
-  console.log(`\n🛑 ${sig} — shutting down`);
-  stopNgrok();
-  process.exit(0);
-};
-process.on("SIGINT",  () => shutdown("SIGINT"));
-process.on("SIGTERM", () => shutdown("SIGTERM"));
-
-// ── MCP Server ────────────────────────────────────────────────────────────────
-const server = new Server(
-  { name: "kickload-watcher", version: "1.1.0" },
-  { capabilities: { tools: {} } }
-);
-
-server.setRequestHandler(ListToolsRequestSchema, async () => ({
-  tools: [
-    {
-      name: "run_kickload_test",
-      description: "Trigger a KickLoad performance test for an API endpoint.",
-      inputSchema: {
-        type: "object",
-        properties: {
-          endpoint:        { type: "string", description: "API endpoint path, e.g. /api/checkout" },
-          description:     { type: "string", description: "What the endpoint does" },
-          developer_email: { type: "string", description: "Email to send results to" },
-          num_threads:     { type: "number", description: "Concurrent users" },
-          loop_count:      { type: "number", description: "Loops per user" },
-          ramp_time:       { type: "number", description: "Ramp-up seconds" },
-        },
-        required: ["endpoint", "developer_email"],
-      },
-    },
-    {
-      name: "get_watcher_status",
-      description: "Check the current watcher status.",
-      inputSchema: { type: "object", properties: {}, required: [] },
-    },
-    {
-      name: "lookup_developer",
-      description: "Look up the email for a Claude user ID from users.json.",
-      inputSchema: {
-        type: "object",
-        properties: { claude_user_id: { type: "string" } },
-        required: ["claude_user_id"],
-      },
-    },
-    {
-      name: "send_test_email",
-      description: "Send a test email to verify email configuration.",
-      inputSchema: {
-        type: "object",
-        properties: { to_email: { type: "string" } },
-        required: ["to_email"],
-      },
-    },
-  ],
-}));
-
-server.setRequestHandler(CallToolRequestSchema, async (request) => {
-  const { name, arguments: args } = request.params;
-
-  if (name === "run_kickload_test") {
-    const { endpoint, description, developer_email } = args;
-    try {
-      logger.info(`Manual test: ${endpoint}`);
-      let prompt = `Load test the ${endpoint} endpoint for performance and stability.`;
-      if (description) {
-        const p = await generateTestPrompt({
-          fileContent: `// ${endpoint}\n// ${description}`,
-          detectedEndpoints: [endpoint],
-          originalPrompt: description,
-          fileName: "mcp-manual",
-        });
-        prompt = p.testPrompt;
-      }
-      const result = await runPhase1Pipeline({
-        fileName:          "mcp-manual",
-        fileContent:       `// ${endpoint}`,
-        detectedEndpoints: [endpoint],
-        originalPrompt:    prompt,
-        devEmail:          developer_email,
-        userId:            null,
-        source:            "mcp_manual",
-      });
-      return {
-        content: [{ type: "text", text: JSON.stringify({
-          success:         result.success,
-          status:          result.summary?.passed ? "PASS" : "FAIL",
-          averageLatency:  result.summary?.averageLatency,
-          errorPercentage: result.summary?.errorPercentage,
-          throughput:      result.summary?.throughput,
-          downloadUrl:     result.downloadUrl,
-          emailSentTo:     developer_email,
-        }, null, 2) }],
-      };
-    } catch (err) {
-      logger.error(`run_kickload_test failed: ${err.message}`);
-      return {
-        content: [{ type: "text", text: JSON.stringify({ success: false, error: err.message }) }],
-        isError: true,
-      };
-    }
-  }
-
-  if (name === "get_watcher_status") {
-    return {
-      content: [{ type: "text", text: JSON.stringify({
-        status:            "running",
-        version:           "1.1.0",
-        triggerMode:       config.triggerMode,
-        watchPaths:        config.watcher.watchPaths,
-        emailProvider:     config.email.provider,
-        kickloadBaseUrl:   config.kickload.baseUrl,
-        ngrokAutoMode:     "localhost backends auto-tunneled",
-        ngrokTokenSet:     !!config.ngrok.authToken,
-        compliancePolling: !!config.anthropic.complianceApiKey,
-      }, null, 2) }],
-    };
-  }
-
-  if (name === "lookup_developer") {
-    const email = getEmailForUser(args.claude_user_id);
-    return {
-      content: [{ type: "text", text: JSON.stringify({
-        claudeUserId: args.claude_user_id,
-        email:        email || null,
-        found:        !!email,
-      }, null, 2) }],
-    };
-  }
-
-  if (name === "send_test_email") {
-    try {
-      await sendResultEmail({
-        toEmail:    args.to_email,
-        endpoint:   "/api/test",
-        passed:     true,
-        summary:    { averageLatency: 123, errorPercentage: 0.5, throughput: 4500 },
-        downloadUrl: null,
-        pdfFilename: "test.pdf",
-        durationMs:  45000,
-        testPrompt:  "Test email.",
-      });
-      return { content: [{ type: "text", text: JSON.stringify({ success: true }) }] };
-    } catch (err) {
-      return {
-        content: [{ type: "text", text: JSON.stringify({ success: false, error: err.message }) }],
-        isError: true,
-      };
-    }
-  }
-
-  return { content: [{ type: "text", text: `Unknown tool: ${name}` }], isError: true };
-});
-
-const transport = new StdioServerTransport();
-await server.connect(transport);
-
-console.log("\n✅ KickLoad Watcher running");
-console.log(`   Mode    : ${config.triggerMode}`);
-console.log(`   KickLoad: ${config.kickload.baseUrl}`);
-console.log(`   ngrok   : auto-mode (localhost backends tunneled automatically)`);
-console.log(`   Email   : ${config.email.provider} ✓`);
-console.log(`   Watching: ${config.watcher.watchPaths.join(", ")}\n`);
+#!/usr/bin/env node
+
+// ── Global error guards — must be first ──────────────────────────────────────
+process.on("uncaughtException", (err) => {
+  console.error("\n❌ Uncaught error:", err.message);
+  if (process.env.LOG_LEVEL === "debug") console.error(err.stack);
+  process.exit(1);
+});
+
+process.on("unhandledRejection", (reason) => {
+  const msg = reason instanceof Error ? reason.message : String(reason);
+  console.error("\n❌ Unhandled rejection:", msg);
+  if (process.env.LOG_LEVEL === "debug" && reason instanceof Error) console.error(reason.stack);
+  process.exit(1);
+});
+
+// ── Node version check ────────────────────────────────────────────────────────
+const nodeMajor = parseInt(process.versions.node.split(".")[0]);
+if (nodeMajor < 18) {
+  console.error(`\n❌ Node.js 18+ required. You have ${process.versions.node}.`);
+  console.error("   Download: https://nodejs.org\n");
+  process.exit(1);
+}
+
+// ── First-run setup (writes .env if missing, reloads env in-process) ─────────
+// STARTUP FIX: runFirstRunSetup no longer exits — continues execution after writing .env
+import { runFirstRunSetup, ensureGitignore } from "./setup.js";
+await runFirstRunSetup();
+
+const { default: dotenv } = await import("dotenv");
+dotenv.config();
+// ── All other imports depend on env being loaded ──────────────────────────────
+const { Server } = await import("@modelcontextprotocol/sdk/server/index.js");
+const { StdioServerTransport } = await import("@modelcontextprotocol/sdk/server/stdio.js");
+const { CallToolRequestSchema, ListToolsRequestSchema } = await import("@modelcontextprotocol/sdk/types.js");
+const { config, validateConfig } = await import("./config.js");
+const { initEmailSender, verifyEmailService, sendResultEmail } = await import("./email-sender.js");
+const { watchIdentityMap, getEmailForUser } = await import("./identity-map.js");
+const { startCompliancePoller, onNewSession } = await import("./compliance-poller.js");
+const { startFileWatcher, onNewApiFile } = await import("./file-watcher.js");
+const { runPhase1Pipeline, handleFileEvent, handleComplianceSession, handleGithubEvent } = await import("./pipeline.js");
+const { generateTestPrompt } = await import("./test-generator.js");
+const { startGithubWebhookServer, onGithubPR } = await import("./github-webhook.js");
+const { ensureNgrokInstalled, stopNgrok } = await import("./ngrok-manager.js");
+const { createLogger } = await import("./logger.js");
+
+const logger = createLogger("main");
+
+// ── Config validation (exits on missing required fields) ─────────────────────
+validateConfig();
+ensureGitignore();
+
+// ── PROBLEM 2 — Email strict mode: init + verify before anything else ─────────
+console.log("\n🔧 Initializing subsystems...");
+
+try {
+  initEmailSender();
+} catch (err) {
+  console.error(`\n❌ Email initialization failed: ${err.message}`);
+  console.error("   Check EMAIL_PROVIDER and related settings in .env\n");
+  process.exit(1);
+}
+
+// Verify SMTP credentials — exits if broken
+await verifyEmailService();
+
+// ── ngrok binary pre-fetch (only if auth token is configured) ─────────────────
+// PROBLEM 1: We no longer check NGROK_ENABLED at startup for localhost backends
+// (pipeline.js handles that automatically). We still pre-install the binary
+// if a token is present, so the first tunnel starts faster.
+if (config.ngrok.authToken) {
+  logger.info("Pre-fetching ngrok binary...");
+  try {
+    await ensureNgrokInstalled();
+    logger.info("✅ ngrok ready.");
+  } catch (err) {
+    // Non-fatal — pipeline will handle the failure gracefully
+    logger.warn(`ngrok pre-install failed: ${err.message}`);
+    logger.warn("Localhost backends will attempt ngrok on first use.");
+  }
+} else {
+  logger.warn("NGROK_AUTHTOKEN not set — ngrok will attempt unauthenticated tunnels for localhost backends.");
+  logger.warn("Add NGROK_AUTHTOKEN to .env to fix: https://dashboard.ngrok.com");
+}
+
+// ── Init remaining subsystems ────────────────────────────────────────────────
+watchIdentityMap();
+
+onNewSession(handleComplianceSession);
+onNewApiFile(handleFileEvent);
+
+if (config.triggerMode === "claudecode" || config.triggerMode === "both") {
+  startCompliancePoller();
+  startFileWatcher();
+}
+
+if (config.triggerMode === "github" || config.triggerMode === "both") {
+  onGithubPR(handleGithubEvent);
+  startGithubWebhookServer();
+}
+
+// ── Graceful shutdown ─────────────────────────────────────────────────────────
+const shutdown = (sig) => {
+  console.log(`\n🛑 ${sig} — shutting down`);
+  stopNgrok();
+  process.exit(0);
+};
+process.on("SIGINT",  () => shutdown("SIGINT"));
+process.on("SIGTERM", () => shutdown("SIGTERM"));
+
+// ── MCP Server ────────────────────────────────────────────────────────────────
+const server = new Server(
+  { name: "kickload-watcher-mcp", version: "0.1.1" },
+  { capabilities: { tools: {} } }
+);
+
+server.setRequestHandler(ListToolsRequestSchema, async () => ({
+  tools: [
+    {
+      name: "run_kickload_test",
+      description: "Trigger a KickLoad performance test for an API endpoint.",
+      inputSchema: {
+        type: "object",
+        properties: {
+          endpoint:        { type: "string", description: "API endpoint path, e.g. /api/checkout" },
+          description:     { type: "string", description: "What the endpoint does" },
+          developer_email: { type: "string", description: "Email to send results to" },
+          num_threads:     { type: "number", description: "Concurrent users" },
+          loop_count:      { type: "number", description: "Loops per user" },
+          ramp_time:       { type: "number", description: "Ramp-up seconds" },
+        },
+        required: ["endpoint", "developer_email"],
+      },
+    },
+    {
+      name: "get_watcher_status",
+      description: "Check the current watcher status.",
+      inputSchema: { type: "object", properties: {}, required: [] },
+    },
+    {
+      name: "lookup_developer",
+      description: "Look up the email for a Claude user ID from users.json.",
+      inputSchema: {
+        type: "object",
+        properties: { claude_user_id: { type: "string" } },
+        required: ["claude_user_id"],
+      },
+    },
+    {
+      name: "send_test_email",
+      description: "Send a test email to verify email configuration.",
+      inputSchema: {
+        type: "object",
+        properties: { to_email: { type: "string" } },
+        required: ["to_email"],
+      },
+    },
+  ],
+}));
+
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  const { name, arguments: args } = request.params;
+
+  if (name === "run_kickload_test") {
+    const { endpoint, description, developer_email } = args;
+    try {
+      logger.info(`Manual test: ${endpoint}`);
+      let prompt = `Load test the ${endpoint} endpoint for performance and stability.`;
+      if (description) {
+        const p = await generateTestPrompt({
+          fileContent: `// ${endpoint}\n// ${description}`,
+          detectedEndpoints: [endpoint],
+          originalPrompt: description,
+          fileName: "mcp-manual",
+        });
+        prompt = p.testPrompt;
+      }
+      const result = await runPhase1Pipeline({
+        fileName:          "mcp-manual",
+        fileContent:       `// ${endpoint}`,
+        detectedEndpoints: [endpoint],
+        originalPrompt:    prompt,
+        devEmail:          developer_email,
+        userId:            null,
+        source:            "mcp_manual",
+      });
+      return {
+        content: [{ type: "text", text: JSON.stringify({
+          success:         result.success,
+          status:          result.summary?.passed ? "PASS" : "FAIL",
+          averageLatency:  result.summary?.averageLatency,
+          errorPercentage: result.summary?.errorPercentage,
+          throughput:      result.summary?.throughput,
+          downloadUrl:     result.downloadUrl,
+          emailSentTo:     developer_email,
+        }, null, 2) }],
+      };
+    } catch (err) {
+      logger.error(`run_kickload_test failed: ${err.message}`);
+      return {
+        content: [{ type: "text", text: JSON.stringify({ success: false, error: err.message }) }],
+        isError: true,
+      };
+    }
+  }
+
+  if (name === "get_watcher_status") {
+    return {
+      content: [{ type: "text", text: JSON.stringify({
+        status:            "running",
+        version:           "1.1.0",
+        triggerMode:       config.triggerMode,
+        watchPaths:        config.watcher.watchPaths,
+        emailProvider:     config.email.provider,
+        kickloadBaseUrl:   config.kickload.baseUrl,
+        ngrokAutoMode:     "localhost backends auto-tunneled",
+        ngrokTokenSet:     !!config.ngrok.authToken,
+        compliancePolling: !!config.anthropic.complianceApiKey,
+      }, null, 2) }],
+    };
+  }
+
+  if (name === "lookup_developer") {
+    const email = getEmailForUser(args.claude_user_id);
+    return {
+      content: [{ type: "text", text: JSON.stringify({
+        claudeUserId: args.claude_user_id,
+        email:        email || null,
+        found:        !!email,
+      }, null, 2) }],
+    };
+  }
+
+  if (name === "send_test_email") {
+    try {
+      await sendResultEmail({
+        toEmail:    args.to_email,
+        endpoint:   "/api/test",
+        passed:     true,
+        summary:    { averageLatency: 123, errorPercentage: 0.5, throughput: 4500 },
+        downloadUrl: null,
+        pdfFilename: "test.pdf",
+        durationMs:  45000,
+        testPrompt:  "Test email.",
+      });
+      return { content: [{ type: "text", text: JSON.stringify({ success: true }) }] };
+    } catch (err) {
+      return {
+        content: [{ type: "text", text: JSON.stringify({ success: false, error: err.message }) }],
+        isError: true,
+      };
+    }
+  }
+
+  return { content: [{ type: "text", text: `Unknown tool: ${name}` }], isError: true };
+});
+
+const transport = new StdioServerTransport();
+await server.connect(transport);
+
+console.log("\n✅ KickLoad Watcher running");
+console.log(`   Mode    : ${config.triggerMode}`);
+console.log(`   KickLoad: ${config.kickload.baseUrl}`);
+console.log(`   ngrok   : auto-mode (localhost backends tunneled automatically)`);
+console.log(`   Email   : ${config.email.provider} ✓`);
+console.log(`   Watching: ${config.watcher.watchPaths.join(", ")}\n`);