kickload-watcher-mcp 0.1.0

package/setup.js

@@ -0,0 +1,201 @@
+import readline from "readline";
+import fs       from "fs";
+import path     from "path";
+
+const ENV_PATH = path.resolve(".env");
+const EMAIL_RE = /^[^\s@]+@[^\s@]+\.[^\s@]+$/;
+
+/**
+ * STARTUP FIX: After writing .env, we reload env and CONTINUE — never exit.
+ * The process stays alive and proceeds to start the watcher.
+ */
+export async function runFirstRunSetup() {
+  if (fs.existsSync(ENV_PATH)) {
+    console.log("⚠ .env already exists. Delete it to re-run setup.");
+    ensureGitignore();
+    return;
+  }
+
+  printBanner();
+  console.log("No .env found — running one-time setup before starting.\n");
+  console.log("You will need:");
+  console.log("  • ANTHROPIC_API_KEY   — console.anthropic.com");
+  console.log("  • KICKLOAD_API_TOKEN  — kickload.neeyatai.com");
+  console.log("  • Gmail address + App Password  (for result emails)");
+  console.log("  • NGROK_AUTHTOKEN     — dashboard.ngrok.com  (required for localhost backends)\n");
+  console.log("⚠️  Values are stored in a local .env file. Never commit it.\n");
+
+  const anthropicKey  = await askRequired("ANTHROPIC_API_KEY  : ", true);
+  const kickloadToken = await askRequired("KICKLOAD_API_TOKEN : ", true);
+  const smtpEmail     = await askEmail(   "Your Gmail address : ");
+  const smtpPass = await askRequired(
+  "Gmail App Password : (Use App Password, NOT your normal password)\n  Create here: https://myaccount.google.com/apppasswords\n> ",
+  true
+  );
+  const devEmail      = await askEmail(   "Developer email    : ");
+  const ngrokToken = await askRequired(
+  "NGROK_AUTHTOKEN    : (Required for localhost testing)\n  Get it from: https://dashboard.ngrok.com/get-started/your-authtoken\n> ",
+  true
+  );
+
+  if (!ngrokToken) {
+    console.log("\n⚠️  ngrok skipped.");
+    console.log("   Localhost backends will attempt unauthenticated tunnels (may be rate-limited).");
+    console.log("   For reliable testing, add to .env later:");
+    console.log("     NGROK_AUTHTOKEN=<your_token>  (dashboard.ngrok.com)");
+    console.log("");
+  }
+
+  const lines = [
+    "# KickLoad Watcher — generated by setup",
+    "# ⚠️  Keep this file SECRET — never commit to git",
+    "",
+    `ANTHROPIC_API_KEY=${anthropicKey}`,
+    `KICKLOAD_API_TOKEN=${kickloadToken}`,
+    `KICKLOAD_BASE_URL=https://kickload.neeyatai.com/api`,
+    "",
+    `# ngrok: auto-enabled for localhost backends regardless of this flag`,
+    `NGROK_AUTHTOKEN=${ngrokToken || ""}`,
+    "",
+    `EMAIL_PROVIDER=smtp`,
+    `EMAIL_FROM_NAME=KickLoad Watcher`,
+    `EMAIL_FROM_ADDRESS=${smtpEmail}`,
+    `SMTP_HOST=smtp.gmail.com`,
+    `SMTP_PORT=465`,
+    `SMTP_USER=${smtpEmail}`,
+    `SMTP_PASS=${smtpPass}`,
+    "",
+    `DEFAULT_DEVELOPER_EMAIL=${devEmail}`,
+    "",
+    `WATCH_PATHS=.`,
+    `TRIGGER_MODE=claudecode`,
+    `LOG_LEVEL=info`,
+    "",
+    "# Optional: override auto-detected backend URL",
+    "# TARGET_API_BASE_URL=https://your-api.example.com",
+    "",
+  ];
+
+  fs.writeFileSync(ENV_PATH, lines.join("\n"), "utf-8");
+  ensureGitignore();
+
+  // Re-load env immediately so the rest of the process sees the values.
+  const { default: dotenv } = await import("dotenv");
+  dotenv.config({ override: true });
+
+  // STARTUP FIX: Do NOT exit — continue execution
+  console.log("\n✅ Setup complete — starting KickLoad Watcher now...\n");
+}
+
+export function ensureGitignore() {
+  const gitignorePath = path.resolve(".gitignore");
+  const required = [".env", ".bin/"];
+
+  let content = "";
+  try {
+    if (fs.existsSync(gitignorePath)) content = fs.readFileSync(gitignorePath, "utf-8");
+  } catch { /* non-fatal */ }
+
+  const existing = content.split("\n").map(l => l.trim());
+  const missing  = required.filter(r => !existing.includes(r));
+  if (missing.length === 0) return;
+
+  const prefix = content && !content.endsWith("\n") ? "\n" : "";
+  fs.appendFileSync(gitignorePath, `${prefix}# KickLoad Watcher\n${missing.join("\n")}\n`, "utf-8");
+  console.warn("⚠️  Added .env and .bin/ to .gitignore — never commit your API keys.");
+}
+
+// ── Input helpers ─────────────────────────────────────────────────────────────
+
+async function askRequired(prompt, masked) {
+  while (true) {
+    const val = masked ? await inputMasked(prompt) : await inputPlain(prompt);
+    if (val.length > 0) return val;
+    console.log("  This field is required — please try again.");
+  }
+}
+
+async function askEmail(prompt) {
+  while (true) {
+    const val = await inputPlain(prompt);
+    if (EMAIL_RE.test(val)) return val;
+    console.log("  Invalid email address — please try again.");
+  }
+}
+
+async function askOptional(prompt) {
+  return inputPlain(prompt);
+}
+
+function inputPlain(prompt) {
+  return new Promise((resolve) => {
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+    rl.question(prompt, (a) => { rl.close(); resolve(a.trim()); });
+  });
+}
+
+function inputMasked(prompt) {
+  if (!process.stdin.isTTY) return inputPlain(prompt);
+
+  return new Promise((resolve) => {
+    process.stdout.write(prompt);
+
+    let value = "";
+    let done  = false;
+
+    const cleanup = () => {
+      if (done) return;
+      done = true;
+      try { process.stdin.setRawMode(false); } catch { /* ignore */ }
+      process.stdin.removeListener("data", handler);
+      process.stdin.pause();
+    };
+
+    const handler = (raw) => {
+      const char = raw.toString();
+
+      if (char === "\u0003") {
+        cleanup();
+        process.stdout.write("\n");
+        process.exit(0);
+      }
+
+      if (char === "\r" || char === "\n") {
+        cleanup();
+        process.stdout.write("\n");
+        resolve(value.trim());
+        return;
+      }
+
+      if (char === "\u007F" || char === "\b") {
+        if (value.length > 0) {
+          value = value.slice(0, -1);
+          process.stdout.clearLine(0);
+          process.stdout.cursorTo(0);
+          process.stdout.write(prompt + "*".repeat(value.length));
+        }
+        return;
+      }
+
+      value += char;
+      process.stdout.write("*");
+    };
+
+    process.stdin.setRawMode(true);
+    process.stdin.resume();
+    process.stdin.setEncoding("utf8");
+    process.stdin.on("data", handler);
+  });
+}
+
+function printBanner() {
+  console.log(`
+╔══════════════════════════════════════════════╗
+║      KickLoad Watcher  —  First-Run Setup    ║
+║   Automated API Performance Testing (OneQA)  ║
+╚══════════════════════════════════════════════╝`);
+}
+runFirstRunSetup().catch(err => {
+  console.error("❌ Setup failed:", err.message);
+  process.exit(1);
+});

package/test-generator.js

@@ -0,0 +1,66 @@
+import Anthropic from "@anthropic-ai/sdk";
+import { config }       from "./config.js";
+import { createLogger } from "./logger.js";
+
+const logger = createLogger("test-generator");
+const client = new Anthropic({ apiKey: config.anthropic.apiKey });
+
+export async function generateTestPrompt({ fileContent, detectedEndpoints, originalPrompt, fileName, backendBaseUrl }) {
+  if (!detectedEndpoints || detectedEndpoints.length === 0) {
+    throw new Error("generateTestPrompt: no endpoints provided");
+  }
+
+  const resolvedBase = (backendBaseUrl || process.env.TARGET_API_BASE_URL || "").replace(/\/$/, "");
+
+  if (!resolvedBase) {
+    throw new Error("No backend URL — set TARGET_API_BASE_URL or ensure backend is running");
+  }
+
+  if (resolvedBase.includes("localhost") || resolvedBase.includes("127.0.0.1")) {
+    throw new Error(
+      `Localhost URL cannot be sent to KickLoad: ${resolvedBase}\n` +
+      `Set NGROK_ENABLED=true in .env or set TARGET_API_BASE_URL to a public URL.`
+    );
+  }
+
+  logger.info(`Generating prompt for ${detectedEndpoints.length} endpoint(s) at ${resolvedBase}`);
+
+  const fullUrls = detectedEndpoints.map(ep => `${resolvedBase}${ep}`);
+
+  let numThreads = 5, loopCount = 2, rampTime = 2;
+
+  try {
+    const response = await client.messages.create({
+      model:      config.anthropic.model,
+      max_tokens: 200,
+      system:
+        `You are a JMeter load test engineer. Return ONLY valid JSON — no markdown, no explanation.
+{"num_threads":<5-50>,"loop_count":<2-10>,"ramp_time":<2-10>,"reasoning":"<one sentence>"}
+Be conservative for new APIs.`,
+      messages: [{
+        role:    "user",
+        content: `File: ${fileName}\nPrompt: ${originalPrompt || "N/A"}\nEndpoints: ${detectedEndpoints.join(", ")}\n\nCode:\n${fileContent.substring(0, 2000)}`,
+      }],
+    });
+
+    const parsed = JSON.parse((response.content[0]?.text || "").replace(/```json|```/gi, "").trim());
+    numThreads = clamp(parsed.num_threads, 5, 50);
+    loopCount  = clamp(parsed.loop_count,  2, 10);
+    rampTime   = clamp(parsed.ramp_time,   2, 10);
+    logger.info(`Parameters: threads=${numThreads} loops=${loopCount} ramp=${rampTime}s`);
+  } catch (err) {
+    logger.warn(`Claude unavailable — using defaults: ${err.message}`);
+  }
+
+  const testPrompt =
+    `Generate a JMeter load test. Send HTTP GET requests to the following endpoints: ${fullUrls.join(" ")}. ` +
+    `Use ${numThreads} threads, ${loopCount} loops, ${rampTime} seconds ramp-up. ` +
+    `Validate status 200. Include summary report.`;
+
+  return { testPrompt, numThreads, loopCount, rampTime, endpoints: detectedEndpoints, fullUrls, primaryEndpoint: detectedEndpoints[0] };
+}
+
+function clamp(value, min, max) {
+  const n = Number(value);
+  return isNaN(n) ? min : Math.min(Math.max(n, min), max);
+}

package/users.json

@@ -0,0 +1,5 @@
+{
+  "usr_abc123def456": "john.doe@yourcompany.com",
+  "usr_xyz789ghi012": "jane.smith@yourcompany.com",
+  "usr_mno345pqr678": "dev.team@yourcompany.com"
+}