@autohq/cli 0.1.563 → 0.1.564

package/dist/agent-bridge.js

@@ -30870,7 +30870,7 @@ Object.assign(lookup, {
 // package.json
 var package_default = {
   name: "@autohq/cli",
-  version: "0.1.563",
+  version: "0.1.564",
   license: "SEE LICENSE IN README.md",
   publishConfig: {
     access: "public"
@@ -30973,7 +30973,8 @@ var AgentBridgeOutputAckTimeoutError = class extends Error {
     this.name = "AgentBridgeOutputAckTimeoutError";
   }
 };
-var AGENT_BRIDGE_TERMINAL_AUTH_DRAIN_TIMEOUT_MS = 6e4;
+var AGENT_BRIDGE_TERMINAL_AUTH_DRAIN_NO_PROGRESS_TIMEOUT_MS = 6e4;
+var AGENT_BRIDGE_TERMINAL_AUTH_DRAIN_ABSOLUTE_TIMEOUT_MS = 30 * 6e4;
 var AGENT_BRIDGE_TERMINAL_AUTH_DRAIN_REDIAL_MS = 1e3;
 async function runAgentBridgeSocket(options) {
   const socket = lookup(`${options.bridgeUrl}${RUNTIME_BRIDGE_NAMESPACE}`, {
@@ -31050,6 +31051,7 @@ async function runAgentBridgeSocket(options) {
         onBootstrap: options.onBootstrap,
         createHandler: (bootstrap) => options.createHandler({
           emitOutput: (output) => emitOutputWithAck(socket, output, options.runtimeLogger),
+          onOutputAckProgress: terminalAuthDrain.handleOutputAckProgress,
           cycleSocket: () => {
             options.writeOutput?.(
               "agent_bridge_socket_cycle reason=output_ack_retries_exhausted"
@@ -31336,7 +31338,12 @@ function createTerminalAuthDrainController(input) {
     if (!current) {
       return;
     }
-    clearTimeout(current.timeout);
+    if (current.absoluteTimeout) {
+      clearTimeout(current.absoluteTimeout);
+    }
+    if (current.noProgressTimeout) {
+      clearTimeout(current.noProgressTimeout);
+    }
     if (current.redialTimer) {
       clearTimeout(current.redialTimer);
     }
@@ -31362,6 +31369,34 @@ function createTerminalAuthDrainController(input) {
       context
     );
   };
+  const timeoutDrain = (current, input2) => {
+    if (state !== current) {
+      return;
+    }
+    logDroppedPendingOutputs(input2.reason);
+    settle(() => current.reject(new Error(input2.error)));
+  };
+  const scheduleNoProgressTimeout = (current) => {
+    if (current.noProgressTimeout) {
+      clearTimeout(current.noProgressTimeout);
+    }
+    current.noProgressTimeout = setTimeout(() => {
+      timeoutDrain(current, {
+        reason: "drain_no_progress_timeout",
+        error: `Timed out draining pending bridge output after terminal auth with no output ack progress for ${AGENT_BRIDGE_TERMINAL_AUTH_DRAIN_NO_PROGRESS_TIMEOUT_MS}ms`
+      });
+    }, AGENT_BRIDGE_TERMINAL_AUTH_DRAIN_NO_PROGRESS_TIMEOUT_MS);
+    current.noProgressTimeout.unref?.();
+  };
+  const scheduleAbsoluteTimeout = (current) => {
+    current.absoluteTimeout = setTimeout(() => {
+      timeoutDrain(current, {
+        reason: "drain_absolute_timeout",
+        error: `Timed out draining pending bridge output after terminal auth at the ${AGENT_BRIDGE_TERMINAL_AUTH_DRAIN_ABSOLUTE_TIMEOUT_MS}ms absolute safety bound`
+      });
+    }, AGENT_BRIDGE_TERMINAL_AUTH_DRAIN_ABSOLUTE_TIMEOUT_MS);
+    current.absoluteTimeout.unref?.();
+  };
   const scheduleRedial = () => {
     const current = state;
     if (!current || current.redialTimer) {
@@ -31401,25 +31436,18 @@ function createTerminalAuthDrainController(input) {
       resolveDrain = resolve4;
       rejectDrain = reject;
     });
-    const timeout = setTimeout(() => {
-      logDroppedPendingOutputs("drain_timeout");
-      settle(() => {
-        rejectDrain(
-          new Error(
-            `Timed out draining pending bridge output after terminal auth in ${AGENT_BRIDGE_TERMINAL_AUTH_DRAIN_TIMEOUT_MS}ms`
-          )
-        );
-      });
-    }, AGENT_BRIDGE_TERMINAL_AUTH_DRAIN_TIMEOUT_MS);
-    timeout.unref?.();
-    state = {
+    const current = {
+      absoluteTimeout: null,
+      noProgressTimeout: null,
       promise: promise2,
       redialTimer: null,
       reject: rejectDrain,
       resolve: resolveDrain,
-      startedAt,
-      timeout
+      startedAt
     };
+    state = current;
+    scheduleNoProgressTimeout(current);
+    scheduleAbsoluteTimeout(current);
     input.writeOutput?.("agent_bridge_terminal_auth_drain_started");
     input.runtimeLogger?.warn("agent_bridge_terminal_auth_drain_started");
     input.socket.auth = { token: input.token, terminalDrain: true };
@@ -31487,11 +31515,20 @@ function createTerminalAuthDrainController(input) {
     }
     return terminalAuthExit;
   };
+  const handleOutputAckProgress = () => {
+    const current = state;
+    if (!current) {
+      return false;
+    }
+    scheduleNoProgressTimeout(current);
+    return true;
+  };
   return {
     begin,
     handleConnected,
     handleConnectError,
     handleDisconnect,
+    handleOutputAckProgress,
     suppressesReconnect: () => terminalAuthExit || state !== null
   };
 }
@@ -72602,6 +72639,7 @@ var AgentBridgeOutputBuffer = class {
     if (isSuccessfulOutputAck(ack)) {
       this.pendingOutputs.delete(output.outputSeq);
       this.discardOnFailureOutputSeqs.delete(output.outputSeq);
+      this.input.onOutputAckProgress?.();
       this.input.runtimeLogger?.info(
         "agent_bridge_output_buffer_emit_ready",
         this.outputLogContext(output, {
@@ -77921,6 +77959,7 @@ function createHarnessCommandHandler(input) {
   const config2 = resolveHarnessConfig(input.bootstrap);
   const base = {
     emitOutput: input.emitOutput,
+    ...input.onOutputAckProgress ? { onOutputAckProgress: input.onOutputAckProgress } : {},
     ...input.cycleSocket ? { cycleSocket: input.cycleSocket } : {},
     ...input.outputSeqStart !== void 0 ? { outputSeqStart: input.outputSeqStart } : {},
     ...input.runtimeLogger ? { runtimeLogger: input.runtimeLogger } : {},