starkbot-cli 0.1.4 → 0.2.0

package/dist/index.js

@@ -867,6 +867,152 @@ var init_connect = __esm({
   }
 });
 
+// src/lib/status.ts
+import ora2 from "ora";
+import chalk2 from "chalk";
+var subtypeColor, toolColor, subagentColor, dimText, StatusTracker;
+var init_status2 = __esm({
+  "src/lib/status.ts"() {
+    "use strict";
+    subtypeColor = chalk2.hex("#7C3AED").bold;
+    toolColor = chalk2.yellow;
+    subagentColor = chalk2.cyan;
+    dimText = chalk2.dim;
+    StatusTracker = class {
+      spinner;
+      activeSubtype = "";
+      activeTools = [];
+      activeSubagents = /* @__PURE__ */ new Map();
+      // label -> subtype
+      paused = false;
+      constructor() {
+        this.spinner = ora2({ color: "magenta", spinner: "dots" });
+      }
+      handleEvent(event) {
+        switch (event.type) {
+          case "tool_call":
+            this.addTool(event.tool_name ?? "unknown");
+            break;
+          case "tool_result":
+            this.removeTool(event.tool_name ?? "unknown");
+            break;
+          case "subagent_spawned":
+            this.addSubagent(event.label ?? "?", event.agent_subtype);
+            break;
+          case "subagent_completed":
+          case "subagent_failed":
+            this.removeSubagent(event.label ?? "?");
+            break;
+          case "subtype_change":
+            this.setSubtype(event.agent_subtype ?? event.label ?? "");
+            break;
+          case "thinking":
+            this.setThinking(event.content);
+            break;
+          case "task_started":
+            if (event.task_name) this.setTask(event.task_name);
+            break;
+          case "task_completed":
+            this.updateSpinner();
+            break;
+          case "text":
+            this.pauseForText();
+            if (event.content) process.stdout.write(event.content);
+            this.resumeAfterText();
+            break;
+          case "done":
+            this.finish();
+            break;
+        }
+      }
+      prefix() {
+        if (this.activeSubtype) {
+          return `${subtypeColor(`[${this.activeSubtype}]`)} `;
+        }
+        return "";
+      }
+      addTool(name) {
+        if (!this.activeTools.includes(name)) {
+          this.activeTools.push(name);
+        }
+        this.updateSpinner();
+      }
+      removeTool(name) {
+        this.activeTools = this.activeTools.filter((t) => t !== name);
+        this.updateSpinner();
+      }
+      addSubagent(label, subtype) {
+        this.activeSubagents.set(label, subtype ?? "");
+        this.updateSpinner();
+      }
+      removeSubagent(label) {
+        this.activeSubagents.delete(label);
+        this.updateSpinner();
+      }
+      setSubtype(subtype) {
+        this.activeSubtype = subtype;
+        this.updateSpinner();
+      }
+      setThinking(message) {
+        const text = `${this.prefix()}${dimText(message || "thinking...")}`;
+        this.ensureSpinning(text);
+      }
+      setTask(taskName) {
+        const text = `${this.prefix()}${dimText(taskName)}`;
+        this.ensureSpinning(text);
+      }
+      updateSpinner() {
+        const parts = [];
+        for (const [label, subtype] of this.activeSubagents) {
+          const st = subtype ? ` (${subtype})` : "";
+          parts.push(subagentColor(`subagent "${label}"${st} running`));
+        }
+        if (this.activeTools.length > 0) {
+          const names = this.activeTools.map((t) => toolColor(t)).join(", ");
+          parts.push(`calling ${names}`);
+        }
+        if (parts.length === 0) {
+          if (this.spinner.isSpinning && !this.paused) {
+            this.spinner.stop();
+          }
+          return;
+        }
+        const text = `${this.prefix()}${parts.join(" | ")}`;
+        this.ensureSpinning(text);
+      }
+      ensureSpinning(text) {
+        if (this.paused) return;
+        if (this.spinner.isSpinning) {
+          this.spinner.text = text;
+        } else {
+          this.spinner.start(text);
+        }
+      }
+      pauseForText() {
+        if (this.spinner.isSpinning) {
+          this.spinner.stop();
+        }
+        this.paused = true;
+      }
+      resumeAfterText() {
+        this.paused = false;
+        if (this.activeTools.length > 0 || this.activeSubagents.size > 0) {
+          this.updateSpinner();
+        }
+      }
+      finish() {
+        if (this.spinner.isSpinning) {
+          this.spinner.stop();
+        }
+        this.activeTools = [];
+        this.activeSubagents.clear();
+        this.activeSubtype = "";
+        this.paused = false;
+      }
+    };
+  }
+});
+
 // src/commands/chat.ts
 var chat_exports = {};
 __export(chat_exports, {
@@ -886,31 +1032,15 @@ function requireGateway() {
     creds.gateway_token
   );
 }
-function printSseEvent(event) {
-  switch (event.type) {
-    case "tool_call": {
-      const name = event.tool_name ?? "unknown";
-      process.stdout.write(dim(`[${name}...] `));
-      break;
-    }
-    case "tool_result":
-      break;
-    case "text":
-      if (event.content) {
-        process.stdout.write(event.content);
-      }
-      break;
-    case "done":
-      break;
-  }
-}
 async function chatOneShotCommand(message) {
   const gw = requireGateway();
+  const status = new StatusTracker();
   process.stdout.write(`${prefix.agent} `);
   try {
-    await gw.chatStream(message, printSseEvent);
+    await gw.chatStream(message, (event) => status.handleEvent(event));
     console.log();
   } catch (err) {
+    status.finish();
     console.log();
     printError(err.message);
     process.exit(1);
@@ -1011,7 +1141,8 @@ async function chatReplCommand() {
           }
           process.stdout.write(`${prefix.agent} `);
           try {
-            await gw.chatStream(trimmed, printSseEvent);
+            const status = new StatusTracker();
+            await gw.chatStream(trimmed, (event) => status.handleEvent(event));
             console.log();
           } catch (err) {
             console.log();
@@ -1031,6 +1162,7 @@ var init_chat = __esm({
     "use strict";
     init_credentials();
     init_gateway_client();
+    init_status2();
     init_ui();
   }
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "starkbot-cli",
-  "version": "0.1.4",
+  "version": "0.2.0",
   "description": "CLI for Starkbot — login, provision, and chat with your bot from the terminal",
   "type": "module",
   "bin": {