aisnitch 0.2.2 → 0.2.4

package/dist/cli/index.js

@@ -8,7 +8,7 @@ import { Command, InvalidArgumentError } from "commander";
 
 // src/package-info.ts
 var AISNITCH_PACKAGE_NAME = "aisnitch";
-var AISNITCH_VERSION = "0.2.2";
+var AISNITCH_VERSION = "0.2.3";
 var AISNITCH_DESCRIPTION = "Universal bridge for AI coding tool activity \u2014 capture, normalize, stream.";
 
 // src/core/events/schema.ts
@@ -3220,8 +3220,16 @@ var EventBus = class {
       },
       "Published event"
     );
-    this.emitter.emit("event", parsedEvent.data);
-    this.emitter.emit(`event:${parsedEvent.data.type}`, parsedEvent.data);
+    try {
+      this.emitter.emit("event", parsedEvent.data);
+    } catch (error) {
+      logger.warn({ error, eventType: parsedEvent.data.type }, "\u{1F4D6} Error in EventBus global subscriber");
+    }
+    try {
+      this.emitter.emit(`event:${parsedEvent.data.type}`, parsedEvent.data);
+    } catch (error) {
+      logger.warn({ error, eventType: parsedEvent.data.type }, "\u{1F4D6} Error in EventBus typed subscriber");
+    }
     return true;
   }
   /**
@@ -3487,6 +3495,7 @@ var WSServer = class {
     });
     socket.on("error", (error) => {
       logger.warn({ error }, "WebSocket consumer error");
+      this.consumers.delete(socket);
     });
     const welcomeMessage = {
       type: "welcome",
@@ -3643,10 +3652,17 @@ var HTTPReceiver = class {
   }
   async handleRequest(request, response, options) {
     this.requestCount += 1;
-    const requestUrl = new URL(
-      request.url ?? "/",
-      `http://${this.host}:${this.port ?? options.port}`
-    );
+    let requestUrl;
+    try {
+      requestUrl = new URL(
+        request.url ?? "/",
+        `http://${this.host}:${this.port ?? options.port}`
+      );
+    } catch {
+      this.invalidRequestCount += 1;
+      this.sendJson(response, 400, { error: "malformed request url" });
+      return;
+    }
     if (request.method === "GET" && requestUrl.pathname === "/health") {
       this.sendJson(response, 200, options.getHealthSnapshot());
       return;
@@ -3987,15 +4003,24 @@ var BaseAdapter = class {
         cwd: data.cwd ?? context.cwd
       }
     });
-    const published = await this.publishEventImplementation(event, {
-      cwd: context.cwd,
-      env: context.env,
-      hookPayload: context.hookPayload,
-      pid: context.pid,
-      sessionId,
-      source: context.source,
-      transcriptPath: context.transcriptPath
-    });
+    let published;
+    try {
+      published = await this.publishEventImplementation(event, {
+        cwd: context.cwd,
+        env: context.env,
+        hookPayload: context.hookPayload,
+        pid: context.pid,
+        sessionId,
+        source: context.source,
+        transcriptPath: context.transcriptPath
+      });
+    } catch (error) {
+      logger.error(
+        { error, eventType: type, adapter: this.name, sessionId },
+        "\u{1F4D6} Failed to publish event \u2014 swallowing to prevent daemon crash"
+      );
+      published = false;
+    }
     if (published) {
       this.eventsEmitted += 1;
     }
@@ -9123,22 +9148,40 @@ var AdapterRegistry = class {
   }
   /**
    * Starts every adapter enabled in the current AISnitch config.
+   * 📖 Each adapter is started independently — one failure does not prevent
+   * the others from starting.
    */
   async startAll(config) {
     for (const adapter of this.list()) {
       if (config.adapters[adapter.name]?.enabled !== true) {
         continue;
       }
-      await adapter.start();
+      try {
+        await adapter.start();
+      } catch (error) {
+        logger.error(
+          { error, adapter: adapter.name },
+          `\u{1F4D6} Failed to start adapter "${adapter.name}" \u2014 skipping`
+        );
+      }
     }
   }
   /**
    * Stops every adapter in reverse registration order.
+   * 📖 Each adapter is stopped independently — one failure does not prevent
+   * the others from being stopped.
    */
   async stopAll() {
     const adapters = this.list().reverse();
     for (const adapter of adapters) {
-      await adapter.stop();
+      try {
+        await adapter.stop();
+      } catch (error) {
+        logger.warn(
+          { error, adapter: adapter.name },
+          `\u{1F4D6} Error stopping adapter "${adapter.name}" \u2014 continuing`
+        );
+      }
     }
   }
 };
@@ -9243,26 +9286,32 @@ var Pipeline = class {
         await adapter.handleHook(payload);
       });
     }
+    try {
+      this.wsPort = await this.wsServer.start({
+        port: resolvedWsPort,
+        eventBus: this.eventBus,
+        activeTools
+      });
+      this.httpPort = await this.httpReceiver.start({
+        port: resolvedHttpPort,
+        onHook: async (tool, payload) => {
+          await this.handleHook(tool, payload);
+        },
+        getHealthSnapshot: () => this.getHealthSnapshot()
+      });
+      this.socketPath = await this.udsServer.start({
+        socketPath,
+        onEvent: async (event) => {
+          await this.publishEvent(event);
+        }
+      });
+      await this.adapterRegistry.startAll(config);
+    } catch (error) {
+      logger.error({ error }, "\u{1F4D6} Pipeline start failed \u2014 rolling back already-started components");
+      await this.rollbackPartialStart();
+      throw error;
+    }
     this.startedAt = Date.now();
-    this.wsPort = await this.wsServer.start({
-      port: resolvedWsPort,
-      eventBus: this.eventBus,
-      activeTools
-    });
-    this.httpPort = await this.httpReceiver.start({
-      port: resolvedHttpPort,
-      onHook: async (tool, payload) => {
-        await this.handleHook(tool, payload);
-      },
-      getHealthSnapshot: () => this.getHealthSnapshot()
-    });
-    this.socketPath = await this.udsServer.start({
-      socketPath,
-      onEvent: async (event) => {
-        await this.publishEvent(event);
-      }
-    });
-    await this.adapterRegistry.startAll(config);
     logger.info(this.getStatus(), "Core pipeline started");
     return this.getStatus();
   }
@@ -9270,10 +9319,25 @@ var Pipeline = class {
    * Stops every pipeline component in reverse dependency order.
    */
   async stop() {
-    await this.adapterRegistry?.stopAll();
-    await this.httpReceiver.stop();
-    await this.udsServer.stop();
-    await this.wsServer.stop();
+    const stopSafely = async (label, fn) => {
+      try {
+        await fn();
+      } catch (error) {
+        logger.warn({ error }, `\u{1F4D6} Error while stopping ${label} \u2014 continuing shutdown`);
+      }
+    };
+    await stopSafely("adapter registry", async () => {
+      await this.adapterRegistry?.stopAll();
+    });
+    await stopSafely("HTTP receiver", async () => {
+      await this.httpReceiver.stop();
+    });
+    await stopSafely("UDS server", async () => {
+      await this.udsServer.stop();
+    });
+    await stopSafely("WebSocket server", async () => {
+      await this.wsServer.stop();
+    });
     this.eventBus.unsubscribeAll();
     this.adapterRegistry = null;
     this.enabledTools.clear();
@@ -9290,11 +9354,50 @@ var Pipeline = class {
   registerHookHandler(tool, handler) {
     this.hookHandlers.set(tool, handler);
   }
+  /**
+   * 📖 Rolls back any components that were successfully started before a
+   * failure occurred, preventing orphaned servers or leaking resources.
+   */
+  async rollbackPartialStart() {
+    const stopSafe = async (label, fn) => {
+      try {
+        await fn();
+      } catch (error) {
+        logger.warn({ error }, `\u{1F4D6} Error rolling back ${label}`);
+      }
+    };
+    await stopSafe("adapter registry", async () => {
+      await this.adapterRegistry?.stopAll();
+    });
+    await stopSafe("UDS server", async () => {
+      await this.udsServer.stop();
+    });
+    await stopSafe("HTTP receiver", async () => {
+      await this.httpReceiver.stop();
+    });
+    await stopSafe("WebSocket server", async () => {
+      await this.wsServer.stop();
+    });
+    this.adapterRegistry = null;
+    this.enabledTools.clear();
+    this.hookHandlers.clear();
+  }
   /**
    * Publishes an event after best-effort context enrichment.
+   * 📖 If enrichment fails, the original event is published un-enriched
+   * rather than being dropped entirely.
    */
   async publishEvent(event, context = {}) {
-    const enrichedEvent = await this.contextDetector.enrich(event, context);
+    let enrichedEvent;
+    try {
+      enrichedEvent = await this.contextDetector.enrich(event, context);
+    } catch (error) {
+      logger.warn(
+        { error, eventId: event.id },
+        "\u{1F4D6} Context enrichment failed \u2014 publishing un-enriched event"
+      );
+      enrichedEvent = event;
+    }
     return this.eventBus.publish(enrichedEvent);
   }
   /**
@@ -9330,6 +9433,16 @@ var Pipeline = class {
     };
   }
   async handleHook(tool, payload) {
+    try {
+      await this.handleHookInner(tool, payload);
+    } catch (error) {
+      logger.error(
+        { error, tool },
+        "\u{1F4D6} Unhandled error in hook handler \u2014 swallowing to prevent daemon crash"
+      );
+    }
+  }
+  async handleHookInner(tool, payload) {
     if (!this.enabledTools.has(tool)) {
       logger.debug({ tool }, "Ignoring hook for disabled tool");
       return;
@@ -10505,18 +10618,22 @@ function parseSocketPayload(data) {
   return parsedEvent.success ? parsedEvent.data : null;
 }
 function parseUnknownPayload(data) {
-  if (typeof data === "string") {
-    return JSON.parse(data);
-  }
-  if (Array.isArray(data)) {
-    return JSON.parse(Buffer.concat(data).toString("utf8"));
-  }
-  if (data instanceof ArrayBuffer) {
-    return JSON.parse(
-      Buffer.from(new Uint8Array(data)).toString("utf8")
-    );
+  try {
+    if (typeof data === "string") {
+      return JSON.parse(data);
+    }
+    if (Array.isArray(data)) {
+      return JSON.parse(Buffer.concat(data).toString("utf8"));
+    }
+    if (data instanceof ArrayBuffer) {
+      return JSON.parse(
+        Buffer.from(new Uint8Array(data)).toString("utf8")
+      );
+    }
+    return JSON.parse(Buffer.from(data).toString("utf8"));
+  } catch {
+    return null;
   }
-  return JSON.parse(Buffer.from(data).toString("utf8"));
 }
 
 // src/tui/hooks/useKeyBinds.ts
@@ -11244,16 +11361,20 @@ function ManagedDaemonApp({
 }
 function parseSocketPayload2(data) {
   let parsedPayload;
-  if (typeof data === "string") {
-    parsedPayload = JSON.parse(data);
-  } else if (Array.isArray(data)) {
-    parsedPayload = JSON.parse(Buffer.concat(data).toString("utf8"));
-  } else if (data instanceof ArrayBuffer) {
-    parsedPayload = JSON.parse(
-      Buffer.from(new Uint8Array(data)).toString("utf8")
-    );
-  } else {
-    parsedPayload = JSON.parse(Buffer.from(data).toString("utf8"));
+  try {
+    if (typeof data === "string") {
+      parsedPayload = JSON.parse(data);
+    } else if (Array.isArray(data)) {
+      parsedPayload = JSON.parse(Buffer.concat(data).toString("utf8"));
+    } else if (data instanceof ArrayBuffer) {
+      parsedPayload = JSON.parse(
+        Buffer.from(new Uint8Array(data)).toString("utf8")
+      );
+    } else {
+      parsedPayload = JSON.parse(Buffer.from(data).toString("utf8"));
+    }
+  } catch {
+    return null;
   }
   if (typeof parsedPayload === "object" && parsedPayload !== null && "type" in parsedPayload && parsedPayload.type === "welcome") {
     return null;
@@ -11748,6 +11869,183 @@ function toConfigArgv(options) {
   return options.configPath ? ["--config", options.configPath] : [];
 }
 
+// src/cli/live-logger.ts
+import { once as once5 } from "events";
+import WebSocket5 from "ws";
+function formatLoggerEventBlock(event) {
+  const headerSegments = [
+    colorize(`#${event["aisnitch.seqnum"]}`, EVENT_COLORS[event.type]),
+    colorize(event["aisnitch.tool"], TOOL_COLORS[event["aisnitch.tool"]]),
+    colorize(event.type, EVENT_COLORS[event.type]),
+    colorize(formatSessionLabelFromEvent(event), TUI_THEME.warning),
+    colorize(event.time, TUI_THEME.muted)
+  ];
+  const flattenedLines = flattenEventRecord(event);
+  return [
+    `${colorize("\u256D\u2500", TUI_THEME.frame)} ${headerSegments.join(colorize("  ", TUI_THEME.muted))}`,
+    ...flattenedLines.map((line) => `${colorize("\u2502", TUI_THEME.frame)} ${line}`),
+    colorize("\u2570\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500", TUI_THEME.frame)
+  ].join("\n");
+}
+function formatLoggerWelcomeLine(message) {
+  const tools = message.tools.length > 0 ? message.tools.join(", ") : "none configured";
+  return `${colorize("AISnitch logger attached", TUI_THEME.success)} ${colorize(`v${message.version}`, TUI_THEME.warning)} ${colorize(`tools=${tools}`, TUI_THEME.muted)}`;
+}
+async function attachWebSocketLogger(url, output, filters = {}) {
+  const socket = new WebSocket5(url);
+  socket.on("message", (data) => {
+    const parsedPayload = parseSocketMessage(data);
+    if (isWelcomeMessage(parsedPayload)) {
+      output.stdout(`${formatLoggerWelcomeLine(parsedPayload)}
+`);
+      return;
+    }
+    const parsedEvent = AISnitchEventSchema.safeParse(parsedPayload);
+    if (!parsedEvent.success) {
+      output.stderr(
+        `${colorize("logger:", TUI_THEME.danger)} received an unrecognized event payload.
+`
+      );
+      return;
+    }
+    if (!matchesFilters(parsedEvent.data, filters)) {
+      return;
+    }
+    output.stdout(`${formatLoggerEventBlock(parsedEvent.data)}
+`);
+  });
+  socket.on("error", (error) => {
+    output.stderr(
+      `${colorize("logger:", TUI_THEME.danger)} ${error instanceof Error ? error.message : "unknown socket error"}
+`
+    );
+  });
+  await once5(socket, "open");
+  return async () => {
+    if (socket.readyState === WebSocket5.CLOSING || socket.readyState === WebSocket5.CLOSED) {
+      return;
+    }
+    socket.close();
+    await once5(socket, "close");
+  };
+}
+function matchesFilters(event, filters) {
+  if (filters.tool && event["aisnitch.tool"] !== filters.tool) {
+    return false;
+  }
+  if (filters.type && event.type !== filters.type) {
+    return false;
+  }
+  return true;
+}
+function flattenEventRecord(event) {
+  const lines = [];
+  flattenValue(lines, "", event);
+  return lines;
+}
+function flattenValue(lines, currentPath, value) {
+  if (Array.isArray(value)) {
+    if (value.length === 0) {
+      lines.push(formatLoggerField(currentPath, "[]", "empty"));
+      return;
+    }
+    value.forEach((entry, index) => {
+      flattenValue(lines, `${currentPath}[${index}]`, entry);
+    });
+    return;
+  }
+  if (isRecord10(value)) {
+    const entries = Object.entries(value).sort(
+      ([left], [right]) => left.localeCompare(right)
+    );
+    if (entries.length === 0) {
+      lines.push(formatLoggerField(currentPath, "{}", "empty"));
+      return;
+    }
+    for (const [key, entry] of entries) {
+      const nextPath = currentPath.length === 0 ? key : `${currentPath}.${key}`;
+      flattenValue(lines, nextPath, entry);
+    }
+    return;
+  }
+  lines.push(formatLoggerField(currentPath, value, inferValueKind(value)));
+}
+function formatLoggerField(path, value, kind) {
+  const renderedValue = typeof value === "string" ? JSON.stringify(value) : value === null ? "null" : typeof value === "undefined" ? "undefined" : JSON.stringify(value);
+  return `${colorize(path, TUI_THEME.warning)} ${colorize("=", TUI_THEME.muted)} ${colorize(
+    renderedValue,
+    getValueColor(kind)
+  )}`;
+}
+function inferValueKind(value) {
+  if (value === null) {
+    return "null";
+  }
+  switch (typeof value) {
+    case "string":
+      return "string";
+    case "number":
+      return "number";
+    case "boolean":
+      return "boolean";
+    default:
+      return "other";
+  }
+}
+function getValueColor(kind) {
+  switch (kind) {
+    case "string":
+      return TUI_THEME.panelBody;
+    case "number":
+      return TUI_THEME.success;
+    case "boolean":
+      return TUI_THEME.warning;
+    case "null":
+      return TUI_THEME.danger;
+    case "empty":
+      return TUI_THEME.muted;
+    default:
+      return TUI_THEME.panelBody;
+  }
+}
+function colorize(value, color) {
+  const [red, green, blue] = hexToRgb(color);
+  return `\x1B[38;2;${red};${green};${blue}m${value}\x1B[39m`;
+}
+function hexToRgb(hexColor) {
+  const sanitized = hexColor.slice(1);
+  return [
+    Number.parseInt(sanitized.slice(0, 2), 16),
+    Number.parseInt(sanitized.slice(2, 4), 16),
+    Number.parseInt(sanitized.slice(4, 6), 16)
+  ];
+}
+function parseSocketMessage(data) {
+  try {
+    if (typeof data === "string") {
+      return JSON.parse(data);
+    }
+    if (Array.isArray(data)) {
+      return JSON.parse(Buffer.concat(data).toString("utf8"));
+    }
+    if (data instanceof ArrayBuffer) {
+      return JSON.parse(Buffer.from(new Uint8Array(data)).toString("utf8"));
+    }
+    return JSON.parse(Buffer.from(data).toString("utf8"));
+  } catch {
+    return null;
+  }
+}
+function isWelcomeMessage(payload) {
+  if (!isRecord10(payload)) {
+    return false;
+  }
+  return payload.type === "welcome" && typeof payload.version === "string" && Array.isArray(payload.tools);
+}
+function isRecord10(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+
 // src/cli/runtime.ts
 var execFile11 = promisify11(execFileCallback11);
 var DAEMON_READY_TIMEOUT_MS = 4e3;
@@ -12084,6 +12382,20 @@ function createCliRuntime(dependencies = {}) {
     process.once("SIGINT", () => {
       void shutdown("SIGINT");
     });
+    process.once("uncaughtException", (error) => {
+      output.stderr(
+        `AISnitch crashed: ${error instanceof Error ? error.message : "unknown exception"}
+`
+      );
+      void shutdown("uncaughtException", 1);
+    });
+    process.once("unhandledRejection", (reason) => {
+      output.stderr(
+        `AISnitch rejected a promise: ${reason instanceof Error ? reason.message : "unknown rejection"}
+`
+      );
+      void shutdown("unhandledRejection", 1);
+    });
     await renderForegroundTui({
       configuredAdapters: getEnabledAdapters(config),
       eventBus: pipeline.getEventBus(),
@@ -12225,6 +12537,43 @@ function createCliRuntime(dependencies = {}) {
       }
     });
   }
+  async function logger2(options) {
+    const snapshot = await getStatusSnapshot(options);
+    if (!snapshot.running) {
+      throw new Error(
+        "AISnitch logger requires a running daemon. Start one with `aisnitch start --daemon` or use `aisnitch start` first."
+      );
+    }
+    const closeLogger = await attachWebSocketLogger(
+      `ws://127.0.0.1:${snapshot.wsPort}`,
+      output,
+      {
+        tool: options.tool,
+        type: options.type
+      }
+    );
+    await new Promise((resolve2) => {
+      let closed = false;
+      const shutdown = async () => {
+        if (closed) {
+          return;
+        }
+        closed = true;
+        process.off("SIGINT", handleSigint);
+        process.off("SIGTERM", handleSigterm);
+        await Promise.resolve(closeLogger());
+        resolve2();
+      };
+      const handleSigint = () => {
+        void shutdown();
+      };
+      const handleSigterm = () => {
+        void shutdown();
+      };
+      process.once("SIGINT", handleSigint);
+      process.once("SIGTERM", handleSigterm);
+    });
+  }
   async function mock(selection, options) {
     const pathOptions = toPathOptions2(options);
     const daemonState = await readDaemonState(pathOptions);
@@ -12480,6 +12829,7 @@ function createCliRuntime(dependencies = {}) {
     aiderNotify,
     attach,
     install,
+    logger: logger2,
     mock,
     runDaemonProcess,
     selfUpdateRun,
@@ -12692,6 +13042,8 @@ Examples:
   aisnitch status
   aisnitch attach
   aisnitch attach --view full-data
+  aisnitch logger
+  aisnitch logger --tool claude-code
   aisnitch setup claude-code
   aisnitch setup aider
   aisnitch setup gemini-cli
@@ -12710,6 +13062,7 @@ Examples:
   addAdaptersCommand(program, runtime);
   addSetupCommand(program, runtime);
   addAttachCommand(program, runtime);
+  addLoggerCommand(program, runtime);
   addMockCommand(program, runtime);
   addWrapCommand(program, runtime);
   addInstallCommand(program, runtime);
@@ -12850,6 +13203,21 @@ function addWrapCommand(program, runtime) {
     }
   );
 }
+function addLoggerCommand(program, runtime) {
+  addCommonOptions(
+    program.command("logger").description("Stream exhaustive live event logs without the TUI").option(
+      "--tool <tool>",
+      "Filter the live logger by tool",
+      wrapOptionParser(parseToolFilterOption)
+    ).option(
+      "--type <type>",
+      "Filter the live logger by event type",
+      wrapOptionParser(parseEventTypeFilterOption)
+    )
+  ).action(async (options) => {
+    await runtime.logger(options);
+  });
+}
 function addInstallCommand(program, runtime) {
   addCommonOptions(
     program.command("install").description("Install a macOS LaunchAgent for AISnitch")