fix(gateway): stream chat events for agent runs

src/commands/agent.ts

@@ -36,7 +36,10 @@ import {
   type SessionEntry,
   saveSessionStore,
 } from "../config/sessions.js";
-import { emitAgentEvent } from "../infra/agent-events.js";
+import {
+  emitAgentEvent,
+  registerAgentRunContext,
+} from "../infra/agent-events.js";
 import { defaultRuntime, type RuntimeEnv } from "../runtime.js";
 import { resolveTelegramToken } from "../telegram/token.js";
 import { normalizeE164 } from "../utils.js";
@@ -204,6 +207,10 @@ export async function agentCommand(
   } = sessionResolution;
   let sessionEntry = resolvedSessionEntry;
 
+  if (sessionKey) {
+    registerAgentRunContext(sessionId, { sessionKey });
+  }
+
   const resolvedThinkLevel =
     thinkOnce ??
     thinkOverride ??
@@ -413,12 +420,15 @@ export async function agentCommand(
   const payloads = result.payloads ?? [];
   const deliver = opts.deliver === true;
   const bestEffortDeliver = opts.bestEffortDeliver === true;
-  const deliveryProvider = (opts.provider ?? "whatsapp").toLowerCase();
+  const deliveryProviderRaw = (opts.provider ?? "whatsapp").toLowerCase();
+  const deliveryProvider =
+    deliveryProviderRaw === "imsg" ? "imessage" : deliveryProviderRaw;
 
   const whatsappTarget = opts.to ? normalizeE164(opts.to) : allowFrom[0];
   const telegramTarget = opts.to?.trim() || undefined;
   const discordTarget = opts.to?.trim() || undefined;
   const signalTarget = opts.to?.trim() || undefined;
+  const imessageTarget = opts.to?.trim() || undefined;
 
   const logDeliveryError = (err: unknown) => {
     const deliveryTarget =
@@ -428,8 +438,10 @@ export async function agentCommand(
           ? whatsappTarget
           : deliveryProvider === "discord"
             ? discordTarget
-            : deliveryProvider === "signal"
-              ? signalTarget
+          : deliveryProvider === "signal"
+            ? signalTarget
+            : deliveryProvider === "imessage"
+              ? imessageTarget
               : undefined;
     const message = `Delivery failed (${deliveryProvider}${deliveryTarget ? ` to ${deliveryTarget}` : ""}): ${String(err)}`;
     runtime.error?.(message);
@@ -463,6 +475,13 @@ export async function agentCommand(
       if (!bestEffortDeliver) throw err;
       logDeliveryError(err);
     }
+    if (deliveryProvider === "imessage" && !imessageTarget) {
+      const err = new Error(
+        "Delivering to iMessage requires --to <handle|chat_id:ID>",
+      );
+      if (!bestEffortDeliver) throw err;
+      logDeliveryError(err);
+    }
     if (deliveryProvider === "webchat") {
       const err = new Error(
         "Delivering to WebChat is not supported via `clawdis agent`; use WhatsApp/Telegram or run with --deliver=false.",
@@ -475,6 +494,7 @@ export async function agentCommand(
       deliveryProvider !== "telegram" &&
       deliveryProvider !== "discord" &&
       deliveryProvider !== "signal" &&
+      deliveryProvider !== "imessage" &&
       deliveryProvider !== "webchat"
     ) {
       const err = new Error(`Unknown provider: ${deliveryProvider}`);
@@ -621,5 +641,38 @@ export async function agentCommand(
         logDeliveryError(err);
       }
     }
+
+    if (deliveryProvider === "imessage" && imessageTarget) {
+      try {
+        if (media.length === 0) {
+          for (const chunk of chunkText(text, 4000)) {
+            await deps.sendMessageIMessage(imessageTarget, chunk, {
+              maxBytes: cfg.imessage?.mediaMaxMb
+                ? cfg.imessage.mediaMaxMb * 1024 * 1024
+                : cfg.agent?.mediaMaxMb
+                  ? cfg.agent.mediaMaxMb * 1024 * 1024
+                  : undefined,
+            });
+          }
+        } else {
+          let first = true;
+          for (const url of media) {
+            const caption = first ? text : "";
+            first = false;
+            await deps.sendMessageIMessage(imessageTarget, caption, {
+              mediaUrl: url,
+              maxBytes: cfg.imessage?.mediaMaxMb
+                ? cfg.imessage.mediaMaxMb * 1024 * 1024
+                : cfg.agent?.mediaMaxMb
+                  ? cfg.agent.mediaMaxMb * 1024 * 1024
+                  : undefined,
+            });
+          }
+        }
+      } catch (err) {
+        if (!bestEffortDeliver) throw err;
+        logDeliveryError(err);
+      }
+    }
   }
 }