kickload-watcher-mcp 0.1.0 → 0.1.2

package/kickload-client.js

@@ -1,254 +1,260 @@
-// kickload-client.js
-// Single source of truth for ALL Kickload API calls.
-// Every endpoint verified against app.py source code.
-//
-// ⚠️  NEW BASE URL: https://kickload.neeyatai.com/api
-// All routes are relative to this base, e.g.:
-//   POST https://kickload.neeyatai.com/api/generate-test-plan
-//   POST https://kickload.neeyatai.com/api/run-test/{filename}
-//   GET  https://kickload.neeyatai.com/api/task-status/{id}
-//   POST https://kickload.neeyatai.com/api/analyzeJTL
-//   GET  https://kickload.neeyatai.com/api/download/{filename}
-
-import FormData from "form-data";
-import fetch    from "node-fetch";
-import fs       from "fs";
-import path     from "path";
-
-const POLL_INTERVAL_MS = 5000;    // 5s between polls — matches backend
-const POLL_TIMEOUT_MS  = 600000;  // 10 minutes max
-
-export class KickloadClient {
-  /**
-   * @param {string} baseUrl  — e.g. "https://kickload.neeyatai.com/api"
-   * @param {string} apiToken — X-API-Token value
-   */
-  constructor(baseUrl, apiToken) {
-    // Normalize: strip trailing slash
-    this.baseUrl = baseUrl.replace(/\/$/, "");
-    this.token   = apiToken;
-  }
-
-  // ─── Private authenticated fetch ─────────────────────────────
-  async _fetch(method, endpoint, { body, headers = {}, isJson = false } = {}) {
-    const url = `${this.baseUrl}${endpoint}`;
-
-    const reqHeaders = {
-      "X-API-Token": this.token,
-      ...headers,
-    };
-
-    if (isJson) {
-      reqHeaders["Content-Type"] = "application/json";
-    }
-
-    const response = await fetch(url, {
-      method,
-      headers: reqHeaders,
-      body,
-      signal: AbortSignal.timeout(POLL_TIMEOUT_MS),
-    });
-
-    const text = await response.text();
-
-    let parsed;
-    try {
-      parsed = JSON.parse(text);
-    } catch {
-      parsed = { _raw: text };
-    }
-
-    if (!response.ok) {
-      const err    = new Error(
-        `Kickload ${method} ${endpoint} → ${response.status} ${response.statusText}\n` +
-        `URL: ${url}\n` +
-        `Response: ${text.substring(0, 500)}`
-      );
-      err.status   = response.status;
-      err.body     = text;
-      err.parsed   = parsed;
-      throw err;
-    }
-
-    return parsed;
-  }
-
-  // ══════════════════════════════════════════════════════════════
-  // STEP 1 — Generate Test Plan
-  // POST /generate-test-plan
-  // Source: app.py line 1260
-  //
-  // Send prompt OR JMX file OR both.
-  // Returns: { status: "success", jmx_filename: "test_plan_xxx.jmx" }
-  // ══════════════════════════════════════════════════════════════
-  async generateTestPlan({ prompt, jmxFilePath } = {}) {
-    if (!prompt && !jmxFilePath) {
-      throw new Error("generateTestPlan: provide at least a prompt or jmxFilePath");
-    }
-
-    const form = new FormData();
-
-    if (prompt) {
-      form.append("prompt", prompt);
-    }
-
-    if (jmxFilePath) {
-      const resolved = path.resolve(jmxFilePath);
-      if (!fs.existsSync(resolved)) {
-        throw new Error(`JMX file not found: ${resolved}`);
-      }
-      form.append("file", fs.createReadStream(resolved), {
-        filename:    path.basename(resolved),
-        contentType: "application/xml",
-      });
-    }
-
-    console.log("\n📋 STEP 1: Generating test plan...");
-    if (prompt)      console.log(`   Prompt  : ${prompt.substring(0, 100)}`);
-    if (jmxFilePath) console.log(`   JMX file: ${jmxFilePath}`);
-
-    const result = await this._fetch("POST", "/generate-test-plan", {
-      body:    form,
-      headers: form.getHeaders(),
-    });
-
-    if (result.status !== "success") {
-      throw new Error(
-        `Test plan generation failed: ${result.message || JSON.stringify(result)}`
-      );
-    }
-
-    console.log(`✅ STEP 1 done — JMX: ${result.jmx_filename}`);
-    return result;
-  }
-
-  // ══════════════════════════════════════════════════════════════
-  // STEP 2 — Run Test
-  // POST /run-test/{jmx_filename}
-  // Source: app.py line 1022
-  //
-  // Returns: { status: "started", task_id: "uuid-xxxx" }
-  // ══════════════════════════════════════════════════════════════
-  async runTest(jmxFilename, { numThreads, loopCount, rampTime } = {}) {
-    if (!jmxFilename) {
-      throw new Error("runTest: jmxFilename is required");
-    }
-
-    const body = {};
-    if (numThreads !== undefined) body.num_threads = numThreads;
-    if (loopCount  !== undefined) body.loop_count  = loopCount;
-    if (rampTime   !== undefined) body.ramp_time   = rampTime;
-
-    console.log(`\n▶️  STEP 2: Running test — ${jmxFilename}`);
-    console.log(`   Overrides: ${JSON.stringify(body) || "none (using JMX defaults)"}`);
-
-    const result = await this._fetch(
-      "POST",
-      `/run-test/${encodeURIComponent(jmxFilename)}`,
-      { body: JSON.stringify(body), isJson: true }
-    );
-
-    if (result.status !== "started") {
-      throw new Error(`Run test failed: ${result.message || JSON.stringify(result)}`);
-    }
-
-    console.log(`✅ STEP 2 done — Task ID: ${result.task_id}`);
-    return result;
-  }
-
-  // ══════════════════════════════════════════════════════════════
-  // STEP 3 — Poll Task Status
-  // GET /task-status/{task_id}
-  // Source: app.py line 899
-  //
-  // Polls every 5s until status = "success" | "error"
-  // Returns: { status, result_file, pdf_file, summary_output }
-  // ══════════════════════════════════════════════════════════════
-  async pollUntilDone(taskId) {
-    if (!taskId) throw new Error("pollUntilDone: taskId is required");
-
-    console.log(`\n⏳ STEP 3: Polling task ${taskId}`);
-    const started = Date.now();
-    let   polls   = 0;
-
-    while (Date.now() - started < POLL_TIMEOUT_MS) {
-      await sleep(POLL_INTERVAL_MS);
-      polls++;
-
-      const result  = await this._fetch("GET", `/task-status/${taskId}`);
-      const elapsed = Math.round((Date.now() - started) / 1000);
-
-      if (result.status === "running") {
-        process.stdout.write(`\r   ⏳ Running... ${elapsed}s (poll #${polls})`);
-        continue;
-      }
-
-      if (result.status === "success") {
-        console.log(`\n✅ STEP 3 done in ${elapsed}s`);
-        console.log(`   JTL: ${result.result_file}`);
-        console.log(`   PDF: ${result.pdf_file}`);
-        return result;
-      }
-
-      if (result.status === "error") {
-        throw new Error(`Task ${taskId} failed: ${result.message}`);
-      }
-
-      throw new Error(`Task ${taskId} unknown status: ${result.status}`);
-    }
-
-    throw new Error(`Task ${taskId} timed out after ${POLL_TIMEOUT_MS / 60000} minutes`);
-  }
-
-  // ══════════════════════════════════════════════════════════════
-  // STEP 4 — Analyze JTL
-  // POST /analyzeJTL
-  // Source: app.py line 1127
-  //
-  // Returns: { filename: "analysis_xxx.pdf" }
-  // ══════════════════════════════════════════════════════════════
-  async analyzeJtl(jtlFilename) {
-    if (!jtlFilename) throw new Error("analyzeJtl: jtlFilename is required");
-
-    console.log(`\n🔬 STEP 4: Analyzing JTL — ${jtlFilename}`);
-
-    const result = await this._fetch("POST", "/analyzeJTL", {
-      body:   JSON.stringify({ filename: jtlFilename }),
-      isJson: true,
-    });
-
-    if (!result.filename) {
-      throw new Error(`JTL analysis returned no filename: ${JSON.stringify(result)}`);
-    }
-
-    console.log(`✅ STEP 4 done — Analysis PDF: ${result.filename}`);
-    return result;
-  }
-
-  // ══════════════════════════════════════════════════════════════
-  // STEP 5 — Get Download URL
-  // GET /download/{filename}?mode=inline
-  // Source: app.py line 1472
-  //
-  // Returns: { status: "success", download_url: "https://s3.amazonaws.com/..." }
-  // ══════════════════════════════════════════════════════════════
-  async getDownloadUrl(filename) {
-    if (!filename) return null;
-
-    try {
-      const result = await this._fetch(
-        "GET",
-        `/download/${encodeURIComponent(filename)}?mode=inline`
-      );
-      return result.download_url || null;
-    } catch (err) {
-      console.warn(`⚠️  Download URL failed for ${filename}: ${err.message}`);
-      return null;
-    }
-  }
-}
-
-// ─── Utility ──────────────────────────────────────────────────
-function sleep(ms) {
-  return new Promise(resolve => setTimeout(resolve, ms));
-}
+// kickload-client.js
+// Single source of truth for ALL Kickload API calls.
+// Every endpoint verified against app.py source code.
+//
+// ⚠️  NEW BASE URL: https://kickload.neeyatai.com/api
+// All routes are relative to this base, e.g.:
+//   POST https://kickload.neeyatai.com/api/generate-test-plan
+//   POST https://kickload.neeyatai.com/api/run-test/{filename}
+//   GET  https://kickload.neeyatai.com/api/task-status/{id}
+//   POST https://kickload.neeyatai.com/api/analyzeJTL
+//   GET  https://kickload.neeyatai.com/api/download/{filename}
+
+import FormData from "form-data";
+import fetch    from "node-fetch";
+import fs       from "fs";
+import path     from "path";
+
+const POLL_INTERVAL_MS = 5000;    // 5s between polls — matches backend
+const POLL_TIMEOUT_MS = 1800000; // 30 min
+
+export class KickloadClient {
+  /**
+   * @param {string} baseUrl  — e.g. "https://kickload.neeyatai.com/api"
+   * @param {string} apiToken — X-API-Token value
+   */
+  constructor(baseUrl, apiToken) {
+    // Normalize: strip trailing slash
+    this.baseUrl = baseUrl.replace(/\/$/, "");
+    this.token   = apiToken;
+  }
+
+  // ─── Private authenticated fetch ─────────────────────────────
+  async _fetch(method, endpoint, { body, headers = {}, isJson = false } = {}) {
+    const url = `${this.baseUrl}${endpoint}`;
+
+    const reqHeaders = {
+      "X-API-Token": this.token,
+      ...headers,
+    };
+
+    if (isJson) {
+      reqHeaders["Content-Type"] = "application/json";
+    }
+
+    const response = await fetch(url, {
+      method,
+      headers: reqHeaders,
+      body,
+      signal: AbortSignal.timeout(POLL_TIMEOUT_MS),
+    });
+
+    const text = await response.text();
+
+    let parsed;
+    try {
+      parsed = JSON.parse(text);
+    } catch {
+      parsed = { _raw: text };
+    }
+
+    if (!response.ok) {
+      const err    = new Error(
+        `Kickload ${method} ${endpoint} → ${response.status} ${response.statusText}\n` +
+        `URL: ${url}\n` +
+        `Response: ${text.substring(0, 500)}`
+      );
+      err.status   = response.status;
+      err.body     = text;
+      err.parsed   = parsed;
+      throw err;
+    }
+
+    return parsed;
+  }
+
+  // ══════════════════════════════════════════════════════════════
+  // STEP 1 — Generate Test Plan
+  // POST /generate-test-plan
+  // Source: app.py line 1260
+  //
+  // Send prompt OR JMX file OR both.
+  // Returns: { status: "success", jmx_filename: "test_plan_xxx.jmx" }
+  // ══════════════════════════════════════════════════════════════
+  async generateTestPlan({ prompt, jmxFilePath } = {}) {
+    if (!prompt && !jmxFilePath) {
+      throw new Error("generateTestPlan: provide at least a prompt or jmxFilePath");
+    }
+
+    const form = new FormData();
+
+    if (prompt) {
+      form.append("prompt", prompt);
+    }
+
+    if (jmxFilePath) {
+      const resolved = path.resolve(jmxFilePath);
+      if (!fs.existsSync(resolved)) {
+        throw new Error(`JMX file not found: ${resolved}`);
+      }
+      form.append("file", fs.createReadStream(resolved), {
+        filename:    path.basename(resolved),
+        contentType: "application/xml",
+      });
+    }
+
+    console.log("\n📋 STEP 1: Generating test plan...");
+    if (prompt)      console.log(`   Prompt  : ${prompt.substring(0, 100)}`);
+    if (jmxFilePath) console.log(`   JMX file: ${jmxFilePath}`);
+
+    const result = await this._fetch("POST", "/generate-test-plan", {
+      body:    form,
+      headers: form.getHeaders(),
+    });
+
+    if (result.status !== "success") {
+      throw new Error(
+        `Test plan generation failed: ${result.message || JSON.stringify(result)}`
+      );
+    }
+
+    console.log(`✅ STEP 1 done — JMX: ${result.jmx_filename}`);
+    return result;
+  }
+
+  // ══════════════════════════════════════════════════════════════
+  // STEP 2 — Run Test
+  // POST /run-test/{jmx_filename}
+  // Source: app.py line 1022
+  //
+  // Returns: { status: "started", task_id: "uuid-xxxx" }
+  // ══════════════════════════════════════════════════════════════
+  async runTest(jmxFilename, { numThreads, loopCount, rampTime } = {}) {
+    if (!jmxFilename) {
+      throw new Error("runTest: jmxFilename is required");
+    }
+
+    const body = {};
+    if (numThreads !== undefined) body.num_threads = numThreads;
+    if (loopCount  !== undefined) body.loop_count  = loopCount;
+    if (rampTime   !== undefined) body.ramp_time   = rampTime;
+
+    console.log(`\n▶️  STEP 2: Running test — ${jmxFilename}`);
+    console.log(`   Overrides: ${JSON.stringify(body) || "none (using JMX defaults)"}`);
+
+    const result = await this._fetch(
+      "POST",
+      `/run-test/${encodeURIComponent(jmxFilename)}`,
+      { body: JSON.stringify(body), isJson: true }
+    );
+
+    if (result.status !== "started") {
+      throw new Error(`Run test failed: ${result.message || JSON.stringify(result)}`);
+    }
+
+    console.log(`✅ STEP 2 done — Task ID: ${result.task_id}`);
+    return result;
+  }
+
+  // ══════════════════════════════════════════════════════════════
+  // STEP 3 — Poll Task Status
+  // GET /task-status/{task_id}
+  // Source: app.py line 899
+  //
+  // Polls every 5s until status = "success" | "error"
+  // Returns: { status, result_file, pdf_file, summary_output }
+  // ══════════════════════════════════════════════════════════════
+  async pollUntilDone(taskId) {
+    if (!taskId) throw new Error("pollUntilDone: taskId is required");
+
+    console.log(`\n⏳ STEP 3: Polling task ${taskId}`);
+    const started = Date.now();
+    let   polls   = 0;
+
+    while (Date.now() - started < POLL_TIMEOUT_MS) {
+      await sleep(POLL_INTERVAL_MS);
+      polls++;
+
+      const result  = await this._fetch("GET", `/task-status/${taskId}`);
+      const elapsed = Math.round((Date.now() - started) / 1000);
+
+      if (result.status === "running") {
+        process.stdout.write(`\r   ⏳ Running... ${elapsed}s (poll #${polls})`);
+        continue;
+      }
+
+      if (result.status === "success") {
+        console.log(`\n✅ STEP 3 done in ${elapsed}s`);
+        console.log(`   JTL: ${result.result_file}`);
+        console.log(`   PDF: ${result.pdf_file}`);
+        return result;
+      }
+
+      if (result.status === "error") {
+        throw new Error(`Task ${taskId} failed: ${result.message}`);
+      }
+
+      throw new Error(`Task ${taskId} unknown status: ${result.status}`);
+    }
+
+    throw new Error(
+      `Task ${taskId} timed out after ${POLL_TIMEOUT_MS / 60000} minutes.\n` +
+      `Possible reasons:\n` +
+      `- Large load test\n` +
+      `- Slow backend\n` +
+      `- KickLoad server delay`
+    );
+  }
+
+  // ══════════════════════════════════════════════════════════════
+  // STEP 4 — Analyze JTL
+  // POST /analyzeJTL
+  // Source: app.py line 1127
+  //
+  // Returns: { filename: "analysis_xxx.pdf" }
+  // ══════════════════════════════════════════════════════════════
+  async analyzeJtl(jtlFilename) {
+    if (!jtlFilename) throw new Error("analyzeJtl: jtlFilename is required");
+
+    console.log(`\n🔬 STEP 4: Analyzing JTL — ${jtlFilename}`);
+
+    const result = await this._fetch("POST", "/analyzeJTL", {
+      body:   JSON.stringify({ filename: jtlFilename }),
+      isJson: true,
+    });
+
+    if (!result.filename) {
+      throw new Error(`JTL analysis returned no filename: ${JSON.stringify(result)}`);
+    }
+
+    console.log(`✅ STEP 4 done — Analysis PDF: ${result.filename}`);
+    return result;
+  }
+
+  // ══════════════════════════════════════════════════════════════
+  // STEP 5 — Get Download URL
+  // GET /download/{filename}?mode=inline
+  // Source: app.py line 1472
+  //
+  // Returns: { status: "success", download_url: "https://s3.amazonaws.com/..." }
+  // ══════════════════════════════════════════════════════════════
+  async getDownloadUrl(filename) {
+    if (!filename) return null;
+
+    try {
+      const result = await this._fetch(
+        "GET",
+        `/download/${encodeURIComponent(filename)}?mode=inline`
+      );
+      return result.download_url || null;
+    } catch (err) {
+      console.warn(`⚠️  Download URL failed for ${filename}: ${err.message}`);
+      return null;
+    }
+  }
+}
+
+// ─── Utility ──────────────────────────────────────────────────
+function sleep(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}

package/package.json

@@ -1,51 +1,51 @@
-{
-  "name": "kickload-watcher-mcp",
-  "version": "0.1.0",
-  "description": "Automated API performance testing for Claude Code teams via KickLoad",
-  "main": "index.js",
-  "type": "module",
-  "bin": {
-    "kickload-watcher-mcp": "./index.js"
-  },
-  "scripts": {
-    "start": "node index.js",
-    "dev": "node --watch index.js",
-    "setup": "node setup.js"
-  },
-  "dependencies": {
-    "@anthropic-ai/sdk": "^0.39.0",
-    "@modelcontextprotocol/sdk": "^1.0.0",
-    "chokidar": "^3.6.0",
-    "nodemailer": "^6.9.0",
-    "node-fetch": "^3.3.0",
-    "form-data": "^4.0.0",
-    "dotenv": "^16.4.0",
-    "chalk": "^5.3.0",
-    "unzipper": "^0.10.14",
-    "tar": "^6.2.0"
-  },
-  "engines": {
-    "node": ">=18.0.0"
-  },
-  "keywords": [
-    "kickload",
-    "mcp",
-    "claude",
-    "testing",
-    "watcher",
-    "oneqa"
-  ],
-  "author": "NeeyatAI",
-  "license": "MIT",
-  "repository": {
-    "type": "git",
-    "url": "https://github.com/KickLoad/kickload-watcher-mcp.git"
-  },
-  "homepage": "https://github.com/KickLoad/kickload-watcher-mcp",
-  "files": [
-   "**/*.js",
-   "users.json",
-   "README.md"
- ],
-  "preferGlobal": true
-}
+{
+  "name": "kickload-watcher-mcp",
+  "version": "0.1.2",
+  "description": "Automated API performance testing for Claude Code teams via KickLoad",
+  "main": "index.js",
+  "type": "module",
+  "bin": {
+    "kickload-watcher-mcp": "./index.js"
+  },
+  "scripts": {
+    "start": "node index.js",
+    "dev": "node --watch index.js",
+    "setup": "node setup.js"
+  },
+  "dependencies": {
+    "@anthropic-ai/sdk": "^0.39.0",
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "chokidar": "^3.6.0",
+    "nodemailer": "^6.9.0",
+    "node-fetch": "^3.3.0",
+    "form-data": "^4.0.0",
+    "dotenv": "^16.4.0",
+    "chalk": "^5.3.0",
+    "unzipper": "^0.10.14",
+    "tar": "^6.2.0"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "keywords": [
+    "kickload",
+    "mcp",
+    "claude",
+    "testing",
+    "watcher",
+    "oneqa"
+  ],
+  "author": "NeeyatAI",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/KickLoad/kickload-watcher-mcp.git"
+  },
+  "homepage": "https://github.com/KickLoad/kickload-watcher-mcp",
+  "files": [
+    "**/*.js",
+    "users.json",
+    "README.md"
+  ],
+  "preferGlobal": true
+}