@episoda/cli 0.2.182 → 0.2.187

package/dist/daemon/daemon-process.js

@@ -3046,7 +3046,7 @@ var require_package = __commonJS({
   "package.json"(exports2, module2) {
     module2.exports = {
       name: "@episoda/cli",
-      version: "0.2.182",
+      version: "0.2.187",
       description: "CLI tool for Episoda local development workflow orchestration",
       main: "dist/index.js",
       types: "dist/index.d.ts",
@@ -12875,7 +12875,7 @@ var pty = __toESM(require("@lydell/node-pty"));
 var INACTIVITY_TIMEOUT_MS3 = 30 * 60 * 1e3;
 var sessions = /* @__PURE__ */ new Map();
 async function handlePtySpawn(payload, client) {
-  const { moduleUid, agent_run_id, command, args, env, cwd, cols = 220, rows = 50 } = payload;
+  const { moduleUid, agent_run_id, command, args, stdin, env, cwd, cols = 220, rows = 50 } = payload;
   if (sessions.has(agent_run_id)) {
     console.warn(`[PTY] Session already exists for agent_run_id ${agent_run_id}, ignoring spawn`);
     return;
@@ -12933,6 +12933,9 @@ async function handlePtySpawn(payload, client) {
       console.error(`[PTY] Failed to send pty_data for ${agent_run_id}:`, err.message);
     });
   });
+  if (typeof stdin === "string" && stdin.trim().length > 0) {
+    proc.write(`${stdin.replace(/\n/g, "\r")}\r`);
+  }
   proc.onExit(({ exitCode }) => {
     const durationMs = Date.now() - session.startedAt;
     if (session.watchdogTimer) clearTimeout(session.watchdogTimer);
@@ -15012,9 +15015,19 @@ var ProjectMessageRouter = class {
     client.on("pty_spawn", async (message) => {
       if (message.type === "pty_spawn") {
         const payload = message.payload;
+        let resolvedPayload = payload;
+        if (!payload.cwd && payload.moduleUid) {
+          const worktreeInfo = await getWorktreeInfoForModule(payload.moduleUid);
+          if (worktreeInfo?.path) {
+            resolvedPayload = {
+              ...payload,
+              cwd: worktreeInfo.path
+            };
+          }
+        }
         console.log(`[Daemon] EP1441: Received pty_spawn for ${projectId}: ${payload.agent_run_id}`);
         client.updateActivity();
-        await handlePtySpawn(payload, client);
+        await handlePtySpawn(resolvedPayload, client);
       }
     });
     client.on("pty_resize", (message) => {
@@ -15311,10 +15324,7 @@ var Daemon = class _Daemon {
     });
   }
   static {
-    this.AGENT_HEARTBEAT_INTERVAL_MS = 15e3;
-  }
-  static {
-    this.AGENT_HEARTBEAT_REQUEST_TIMEOUT_MS = 12e3;
+    this.AGENT_HEARTBEAT_INTERVAL_MS = 6e4;
   }
   static {
     this.HEALTH_PORT = 9999;
@@ -15341,59 +15351,58 @@ var Daemon = class _Daemon {
     loop.stop();
     this.agentHeartbeatLoops.delete(sessionId);
   }
-  async handleServerRequestedAgentStop(sessionId) {
+  async shouldContinueAgentSession(sessionId) {
+    try {
+      const config = await (0, import_core22.loadConfig)();
+      const apiUrl = config?.api_url || "https://episoda.dev";
+      const response = await fetchWithAuth(`${apiUrl}/api/agent-sessions/${encodeURIComponent(sessionId)}/heartbeat`, {
+        method: "POST"
+      });
+      if (!response.ok) {
+        console.warn(`[Daemon] EP1429: Heartbeat API returned ${response.status} for session ${sessionId}; continuing session`);
+        return true;
+      }
+      const payload = await response.json().catch(() => null);
+      return payload?.continue !== false;
+    } catch (error) {
+      console.warn(`[Daemon] EP1429: Heartbeat API check failed for session ${sessionId}; continuing session`, error);
+      return true;
+    }
+  }
+  async stopAgentSessionFromServerRequest(sessionId) {
     try {
       const agentManager = getAgentControlPlane();
+      await agentManager.initialize();
       await agentManager.stopSession(sessionId);
-      console.log(`[Daemon] EP1430: Stopped session ${sessionId} due to server continue=false`);
+      this.agentEventSeq.delete(sessionId);
+      console.log(`[Daemon] EP1429: Stopped agent session ${sessionId} on server heartbeat continue=false`);
     } catch (error) {
-      console.warn(
-        `[Daemon] EP1430: Failed to stop session ${sessionId} after server continue=false:`,
-        error
-      );
+      console.warn(`[Daemon] EP1429: Failed to stop session ${sessionId} after continue=false`, error);
     }
   }
-  startAgentHeartbeatLoop(sessionId) {
+  startAgentHeartbeatLoop(sessionId, client) {
     if (this.agentHeartbeatLoops.has(sessionId)) return;
     let stopped = false;
-    let inFlight = false;
     let timer = null;
     const runHeartbeat = async () => {
-      if (stopped || inFlight) return;
-      inFlight = true;
-      const abortController = new AbortController();
-      const requestTimeout = setTimeout(() => {
-        abortController.abort();
-      }, _Daemon.AGENT_HEARTBEAT_REQUEST_TIMEOUT_MS);
+      if (stopped) return;
       try {
-        const config = await (0, import_core22.loadConfig)();
-        if (!config?.access_token) return;
-        const apiUrl = config.api_url || "https://episoda.dev";
-        const response = await fetchWithAuth(
-          `${apiUrl}/api/agent-sessions/${sessionId}/heartbeat`,
-          {
-            method: "POST",
-            signal: abortController.signal
-          }
-        );
-        if (!response.ok && response.status !== 409) {
-          return;
-        }
-        const payload = await response.json().catch(() => null);
-        const shouldContinue = response.status === 409 ? false : Boolean(payload?.data?.continue ?? payload?.continue ?? true);
+        await client.send({
+          type: "agent_heartbeat",
+          sessionId,
+          pid: process.pid,
+          status: "running"
+        });
+        const shouldContinue = await this.shouldContinueAgentSession(sessionId);
         if (!shouldContinue) {
-          await this.handleServerRequestedAgentStop(sessionId);
           this.stopAgentHeartbeatLoop(sessionId);
+          await this.stopAgentSessionFromServerRequest(sessionId);
         }
       } catch (error) {
-        if (error instanceof Error && error.name === "AbortError") {
-          console.warn(
-            `[Daemon] EP1430: Heartbeat request timed out for session ${sessionId} after ${_Daemon.AGENT_HEARTBEAT_REQUEST_TIMEOUT_MS}ms`
-          );
-        }
-      } finally {
-        clearTimeout(requestTimeout);
-        inFlight = false;
+        console.warn(
+          `[Daemon] EP1429: Failed to send agent_heartbeat for session ${sessionId}:`,
+          error
+        );
       }
     };
     timer = setInterval(() => {
@@ -15848,7 +15857,7 @@ var Daemon = class _Daemon {
           await agentManager.initialize();
           let result;
           if (cmd.action === "start") {
-            this.startAgentHeartbeatLoop(cmd.sessionId);
+            this.startAgentHeartbeatLoop(cmd.sessionId, client);
             const callbacks = createStreamingCallbacks(cmd.sessionId, message.id);
             const agentWorkingDir = await resolveAgentWorkingDirectory({
               command: cmd,
@@ -15889,7 +15898,7 @@ var Daemon = class _Daemon {
               error: startResult.error
             };
           } else if (cmd.action === "message") {
-            this.startAgentHeartbeatLoop(cmd.sessionId);
+            this.startAgentHeartbeatLoop(cmd.sessionId, client);
             const callbacks = createStreamingCallbacks(cmd.sessionId, message.id);
             const sendResult = await agentManager.sendMessage({
               sessionId: cmd.sessionId,