@closeup1202/klag 0.3.0 → 0.3.1

package/dist/cli/index.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 
 // src/cli/index.ts
-import chalk5 from "chalk";
+import chalk6 from "chalk";
 import { Command } from "commander";
 
 // src/analyzer/burstDetector.ts
@@ -363,7 +363,7 @@ import chalk from "chalk";
 import Table from "cli-table3";
 
 // src/types/index.ts
-var VERSION = "0.3.0";
+var VERSION = "0.3.1";
 function classifyLag(lag, consumeRate) {
   if (lag === 0n) return "OK";
   if (consumeRate !== void 0) {
@@ -646,6 +646,104 @@ function parseConfig(filePath) {
   return obj;
 }
 
+// src/cli/groupPicker.ts
+import chalk4 from "chalk";
+import prompts from "prompts";
+var GROUP_STATE_ORDER = {
+  Stable: 0,
+  Empty: 1,
+  PreparingRebalance: 2,
+  CompletingRebalance: 3,
+  Dead: 4
+};
+async function pickGroup(options) {
+  const kafka = createKafkaClient("klag-picker", { ...options, groupId: "" });
+  const admin = kafka.admin();
+  process.stdout.write(chalk4.gray("  Fetching consumer groups..."));
+  let groups = [];
+  try {
+    await admin.connect();
+    const result = await admin.listGroups();
+    groups = result.groups;
+  } finally {
+    await admin.disconnect();
+  }
+  process.stdout.write(`\r${" ".repeat(40)}\r`);
+  if (groups.length === 0) {
+    console.error(chalk4.red("\n\u274C No consumer groups found on this broker\n"));
+    process.exit(1);
+  }
+  const kafka2 = createKafkaClient("klag-picker-desc", {
+    ...options,
+    groupId: ""
+  });
+  const admin2 = kafka2.admin();
+  process.stdout.write(chalk4.gray("  Loading group states...  "));
+  const stateMap = /* @__PURE__ */ new Map();
+  try {
+    await admin2.connect();
+    const groupIds = groups.map((g) => g.groupId);
+    const described = await admin2.describeGroups(groupIds);
+    for (const g of described.groups) {
+      stateMap.set(g.groupId, g.state);
+    }
+  } catch {
+  } finally {
+    await admin2.disconnect();
+  }
+  process.stdout.write(`\r${" ".repeat(40)}\r`);
+  const sorted = [...groups].sort((a, b) => {
+    const stateA = GROUP_STATE_ORDER[stateMap.get(a.groupId) ?? ""] ?? 99;
+    const stateB = GROUP_STATE_ORDER[stateMap.get(b.groupId) ?? ""] ?? 99;
+    return stateA !== stateB ? stateA - stateB : a.groupId.localeCompare(b.groupId);
+  });
+  const choices = sorted.map((g) => {
+    const state = stateMap.get(g.groupId);
+    const stateLabel = state ? stateColor(state) : chalk4.gray("unknown");
+    return {
+      title: `${g.groupId}  ${stateLabel}`,
+      value: g.groupId
+    };
+  });
+  console.log("");
+  const response = await prompts(
+    {
+      type: "autocomplete",
+      name: "groupId",
+      message: "Select a consumer group",
+      choices,
+      suggest: (input, choices2) => Promise.resolve(
+        choices2.filter(
+          (c) => c.value.toLowerCase().includes(input.toLowerCase())
+        )
+      )
+    },
+    {
+      onCancel: () => {
+        console.log(chalk4.gray("\n  Cancelled\n"));
+        process.exit(0);
+      }
+    }
+  );
+  console.log("");
+  return response.groupId;
+}
+function stateColor(state) {
+  switch (state) {
+    case "Stable":
+      return chalk4.green(`(${state})`);
+    case "Empty":
+      return chalk4.gray(`(${state})`);
+    case "PreparingRebalance":
+    case "CompletingRebalance":
+      return chalk4.yellow(`(${state})`);
+    case "Dead":
+      return chalk4.red(`(${state})`);
+    default:
+      return chalk4.gray(`(${state})`);
+  }
+}
+
 // src/cli/validators.ts
 import { existsSync as existsSync2 } from "fs";
 import { InvalidArgumentError } from "commander";
@@ -699,7 +797,7 @@ function parseCertPath(value) {
 }
 
 // src/cli/watcher.ts
-import chalk4 from "chalk";
+import chalk5 from "chalk";
 var MAX_RETRIES = 3;
 function clearScreen() {
   process.stdout.write("\x1Bc");
@@ -715,31 +813,31 @@ function printWatchHeader(intervalMs, updatedAt) {
     hour12: false
   });
   console.log(
-    chalk4.bold.cyan("\u26A1 klag") + chalk4.gray(`  v${VERSION}`) + "  \u2502  " + chalk4.yellow("watch mode") + "  \u2502  " + chalk4.gray(`${intervalSec}s refresh`) + "  \u2502  " + chalk4.gray("Ctrl+C to exit")
+    chalk5.bold.cyan("\u26A1 klag") + chalk5.gray(`  v${VERSION}`) + "  \u2502  " + chalk5.yellow("watch mode") + "  \u2502  " + chalk5.gray(`${intervalSec}s refresh`) + "  \u2502  " + chalk5.gray("Ctrl+C to exit")
   );
-  console.log(chalk4.gray(`   Last updated: ${timeStr} (${tz})`));
+  console.log(chalk5.gray(`   Last updated: ${timeStr} (${tz})`));
 }
 function printWatchError(message, retryCount, retryIn) {
   clearScreen();
   console.log(
-    chalk4.bold.cyan("\u26A1 klag") + chalk4.gray(`  v${VERSION}`) + "  \u2502  " + chalk4.yellow("watch mode") + "  \u2502  " + chalk4.gray("Ctrl+C to exit")
+    chalk5.bold.cyan("\u26A1 klag") + chalk5.gray(`  v${VERSION}`) + "  \u2502  " + chalk5.yellow("watch mode") + "  \u2502  " + chalk5.gray("Ctrl+C to exit")
   );
   console.log("");
-  console.error(chalk4.red(`   \u274C Error: ${message}`));
+  console.error(chalk5.red(`   \u274C Error: ${message}`));
   console.log(
-    chalk4.yellow(`   Retrying ${retryCount}/${MAX_RETRIES}... in ${retryIn}s`)
+    chalk5.yellow(`   Retrying ${retryCount}/${MAX_RETRIES}... in ${retryIn}s`)
   );
   console.log("");
 }
 function printWatchFatal(message) {
   clearScreen();
   console.log(
-    chalk4.bold.cyan("\u26A1 klag") + chalk4.gray(`  v${VERSION}`) + "  \u2502  " + chalk4.yellow("watch mode")
+    chalk5.bold.cyan("\u26A1 klag") + chalk5.gray(`  v${VERSION}`) + "  \u2502  " + chalk5.yellow("watch mode")
   );
   console.log("");
-  console.error(chalk4.red(`   \u274C Error: ${message}`));
+  console.error(chalk5.red(`   \u274C Error: ${message}`));
   console.error(
-    chalk4.red(`   All ${MAX_RETRIES} retries failed \u2014 exiting watch mode`)
+    chalk5.red(`   All ${MAX_RETRIES} retries failed \u2014 exiting watch mode`)
   );
   console.log("");
 }
@@ -762,7 +860,7 @@ async function runOnce(options, noRate, previous) {
     const topics = [...new Set(snapshot.partitions.map((p) => p.topic))];
     const waitSec = (options.intervalMs ?? 5e3) / 1e3;
     process.stdout.write(
-      chalk4.gray(`  Sampling rates... (waiting ${waitSec}s)   `)
+      chalk5.gray(`  Sampling rates... (waiting ${waitSec}s)   `)
     );
     rateSnapshot = await collectRate(options, topics);
     process.stdout.write(`\r${" ".repeat(50)}\r`);
@@ -779,7 +877,7 @@ function printCountdown(seconds) {
     let remaining = seconds;
     const tick = () => {
       process.stdout.write(
-        `\r${chalk4.gray(`   [\u25CF] Next refresh in ${remaining}s...`)}   `
+        `\r${chalk5.gray(`   [\u25CF] Next refresh in ${remaining}s...`)}   `
       );
       if (remaining === 0) {
         process.stdout.write(`\r${" ".repeat(40)}\r`);
@@ -804,12 +902,12 @@ function getFriendlyMessage(err, broker) {
 }
 async function startWatch(options, noRate) {
   process.on("SIGINT", () => {
-    console.log(chalk4.gray("\n\n  Watch mode exited\n"));
+    console.log(chalk5.gray("\n\n  Watch mode exited\n"));
     process.exit(0);
   });
   const intervalMs = options.intervalMs ?? 5e3;
   const waitSec = Math.ceil(intervalMs / 1e3);
-  process.stdout.write(chalk4.gray("  Connecting to broker..."));
+  process.stdout.write(chalk5.gray("  Connecting to broker..."));
   let errorCount = 0;
   let previousSnapshot;
   while (true) {
@@ -840,7 +938,10 @@ program.name("klag").description("Kafka consumer lag root cause analyzer").versi
   "Kafka broker address",
   parseBroker,
   "localhost:9092"
-).requiredOption("-g, --group <groupId>", "Consumer group ID").option(
+).option(
+  "-g, --group <groupId>",
+  "Consumer group ID (omit to pick interactively)"
+).option(
   "-i, --interval <ms>",
   "Rate sampling interval in ms",
   parseInterval,
@@ -865,12 +966,11 @@ program.name("klag").description("Kafka consumer lag root cause analyzer").versi
     const rc = loaded?.config ?? {};
     if (loaded) {
       process.stderr.write(
-        chalk5.gray(`  Using config: ${loaded.loadedFrom}
+        chalk6.gray(`  Using config: ${loaded.loadedFrom}
 `)
       );
     }
     const broker = options.broker !== "localhost:9092" ? options.broker : rc.broker ?? options.broker;
-    const groupId = options.group ?? rc.group;
     const intervalMs = options.interval !== 5e3 ? options.interval : rc.interval ?? options.interval;
     const timeoutMs = options.timeout !== 5e3 ? options.timeout : rc.timeout ?? options.timeout;
     const auth = buildAuthOptions({
@@ -882,18 +982,14 @@ program.name("klag").description("Kafka consumer lag root cause analyzer").versi
       saslUsername: options.saslUsername ?? rc.sasl?.username,
       saslPassword: options.saslPassword ?? rc.sasl?.password
     });
-    const kafkaOptions = {
-      broker,
-      groupId,
-      intervalMs,
-      timeoutMs,
-      ...auth
-    };
+    const baseOptions = { broker, intervalMs, timeoutMs, ...auth };
+    const groupId = options.group ?? rc.group ?? await pickGroup(baseOptions);
+    const kafkaOptions = { ...baseOptions, groupId };
     if (options.watch) {
       await startWatch(kafkaOptions, options.rate === false);
       return;
     }
-    process.stdout.write(chalk5.gray("  Connecting to broker..."));
+    process.stdout.write(chalk6.gray("  Connecting to broker..."));
     const snapshot = await collectLag(kafkaOptions);
     process.stdout.write(`\r${" ".repeat(50)}\r`);
     let rateSnapshot;
@@ -901,7 +997,7 @@ program.name("klag").description("Kafka consumer lag root cause analyzer").versi
       const topics = [...new Set(snapshot.partitions.map((p) => p.topic))];
       const waitSec = (kafkaOptions.intervalMs ?? 5e3) / 1e3;
       process.stdout.write(
-        chalk5.gray(`  Sampling rates... (waiting ${waitSec}s)   `)
+        chalk6.gray(`  Sampling rates... (waiting ${waitSec}s)   `)
       );
       rateSnapshot = await collectRate(kafkaOptions, topics);
       process.stdout.write(`\r${" ".repeat(50)}\r`);
@@ -929,14 +1025,14 @@ program.name("klag").description("Kafka consumer lag root cause analyzer").versi
     process.stdout.write(`\r${" ".repeat(50)}\r`);
     const message = err instanceof Error ? err.message : String(err);
     if (message.includes("ECONNREFUSED") || message.includes("ETIMEDOUT") || message.includes("Connection error") || message.includes("connect ECONNREFUSED")) {
-      console.error(chalk5.red(`
+      console.error(chalk6.red(`
 \u274C Cannot connect to broker
 `));
-      console.error(chalk5.yellow("   Check the following:"));
-      console.error(chalk5.gray(`   \u2022 Is Kafka running: docker ps`));
-      console.error(chalk5.gray(`   \u2022 Broker address: ${options.broker}`));
+      console.error(chalk6.yellow("   Check the following:"));
+      console.error(chalk6.gray(`   \u2022 Is Kafka running: docker ps`));
+      console.error(chalk6.gray(`   \u2022 Broker address: ${options.broker}`));
       console.error(
-        chalk5.gray(
+        chalk6.gray(
           `   \u2022 Port accessibility: nc -zv ${options.broker.split(":")[0]} ${options.broker.split(":")[1]}`
         )
       );
@@ -944,18 +1040,18 @@ program.name("klag").description("Kafka consumer lag root cause analyzer").versi
       process.exit(1);
     }
     if (message.includes("SASLAuthenticationFailed") || message.includes("Authentication failed") || message.includes("SASL")) {
-      console.error(chalk5.red(`
+      console.error(chalk6.red(`
 \u274C SASL authentication failed
 `));
-      console.error(chalk5.yellow("   Check the following:"));
+      console.error(chalk6.yellow("   Check the following:"));
       console.error(
-        chalk5.gray(`   \u2022 Mechanism: ${options.saslMechanism ?? "(none)"}`)
+        chalk6.gray(`   \u2022 Mechanism: ${options.saslMechanism ?? "(none)"}`)
       );
       console.error(
-        chalk5.gray(`   \u2022 Username:  ${options.saslUsername ?? "(none)"}`)
+        chalk6.gray(`   \u2022 Username:  ${options.saslUsername ?? "(none)"}`)
       );
       console.error(
-        chalk5.gray(
+        chalk6.gray(
           `   \u2022 Password:  set via KLAG_SASL_PASSWORD or --sasl-password`
         )
       );
@@ -963,21 +1059,21 @@ program.name("klag").description("Kafka consumer lag root cause analyzer").versi
       process.exit(1);
     }
     if (message.includes("not found") || message.includes("Dead state")) {
-      console.error(chalk5.red(`
+      console.error(chalk6.red(`
 \u274C Consumer group not found
 `));
-      console.error(chalk5.yellow("   Check the following:"));
-      console.error(chalk5.gray(`   \u2022 Group ID: ${options.group}`));
-      console.error(chalk5.gray(`   \u2022 List existing groups:`));
+      console.error(chalk6.yellow("   Check the following:"));
+      console.error(chalk6.gray(`   \u2022 Group ID: ${options.group}`));
+      console.error(chalk6.gray(`   \u2022 List existing groups:`));
       console.error(
-        chalk5.gray(
+        chalk6.gray(
           `     kafka-consumer-groups.sh --bootstrap-server ${options.broker} --list`
         )
       );
       console.error("");
       process.exit(1);
     }
-    console.error(chalk5.red(`
+    console.error(chalk6.red(`
 \u274C Error: ${message}
 `));
     process.exit(1);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@closeup1202/klag",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "Kafka consumer lag root cause analyzer",
   "type": "module",
   "bin": {
@@ -39,11 +39,13 @@
     "chalk": "^5.6.2",
     "cli-table3": "^0.6.5",
     "commander": "^14.0.3",
-    "kafkajs": "^2.2.4"
+    "kafkajs": "^2.2.4",
+    "prompts": "^2.4.2"
   },
   "devDependencies": {
     "@biomejs/biome": "^2.4.8",
     "@types/node": "^25.5.0",
+    "@types/prompts": "^2.4.9",
     "tsup": "^8.5.1",
     "tsx": "^4.21.0",
     "typescript": "^6.0.2",