@threadbase-sh/streamer 1.15.4 → 1.16.0

package/dist/index.d.cts

@@ -596,6 +596,7 @@ type ApiDeps = {
     handleGetOutput: (sessionId: string, res: ServerResponse) => void;
     handleSendInput: (sessionId: string, req: IncomingMessage, res: ServerResponse) => Promise<void>;
     handleCancel: (sessionId: string, res: ServerResponse) => void;
+    handleStopSession: (sessionId: string, res: ServerResponse) => Promise<void>;
     handleSetSessionName: (sessionId: string, req: IncomingMessage, res: ServerResponse) => Promise<void>;
     handleUploadFile: (sessionId: string, req: IncomingMessage, res: ServerResponse) => Promise<void>;
     handleAdopt: (sessionId: string, res: ServerResponse) => Promise<void>;
@@ -725,6 +726,7 @@ declare class StreamerServer {
     private log;
     private agentConfig;
     private agentClient;
+    private sessionStatusBus;
     constructor(config: ServerConfig & {
         apiKey: string;
     });
@@ -771,6 +773,7 @@ declare class StreamerServer {
     private handleUploadFile;
     private handleGetOutput;
     private handleCancel;
+    private handleStopSession;
     private handleAdopt;
     private handleStartSession;
     private linkSessionToProject;

package/dist/index.d.ts

@@ -596,6 +596,7 @@ type ApiDeps = {
     handleGetOutput: (sessionId: string, res: ServerResponse) => void;
     handleSendInput: (sessionId: string, req: IncomingMessage, res: ServerResponse) => Promise<void>;
     handleCancel: (sessionId: string, res: ServerResponse) => void;
+    handleStopSession: (sessionId: string, res: ServerResponse) => Promise<void>;
     handleSetSessionName: (sessionId: string, req: IncomingMessage, res: ServerResponse) => Promise<void>;
     handleUploadFile: (sessionId: string, req: IncomingMessage, res: ServerResponse) => Promise<void>;
     handleAdopt: (sessionId: string, res: ServerResponse) => Promise<void>;
@@ -725,6 +726,7 @@ declare class StreamerServer {
     private log;
     private agentConfig;
     private agentClient;
+    private sessionStatusBus;
     constructor(config: ServerConfig & {
         apiKey: string;
     });
@@ -771,6 +773,7 @@ declare class StreamerServer {
     private handleUploadFile;
     private handleGetOutput;
     private handleCancel;
+    private handleStopSession;
     private handleAdopt;
     private handleStartSession;
     private linkSessionToProject;

package/dist/index.js

@@ -1189,6 +1189,7 @@ import {
   ConversationScanner,
   search
 } from "@threadbase-sh/scanner";
+import { EventEmitter } from "events";
 import {
   createReadStream,
   existsSync as existsSync6,
@@ -1845,6 +1846,10 @@ var createSessionRoutes = (deps) => {
     await deps.handleAdopt(c.req.param("id"), c.env.outgoing);
     return alreadyHandled6();
   });
+  app.post("/:id/stop", async (c) => {
+    await deps.handleStopSession(c.req.param("id"), c.env.outgoing);
+    return alreadyHandled6();
+  });
   app.get("/:id", (c) => {
     deps.handleGetSession(c.req.param("id"), c.env.outgoing);
     return alreadyHandled6();
@@ -3927,7 +3932,9 @@ var StreamerServer = class {
   log = getLogger("server");
   agentConfig;
   agentClient = null;
+  sessionStatusBus = new EventEmitter();
   constructor(config) {
+    this.sessionStatusBus.setMaxListeners(0);
     this.apiKey = config.apiKey;
     this.localNoAuth = config.localNoAuth ?? false;
     this.verbose = config.verbose ?? false;
@@ -4044,6 +4051,7 @@ var StreamerServer = class {
         if (resp) {
           this.wsHub.broadcast({ type: "session_update", session: resp });
         }
+        this.sessionStatusBus.emit(`status:${session.id}`, session.status);
       }
     });
     this.agentConfig = readAgentConfig();
@@ -4089,6 +4097,7 @@ var StreamerServer = class {
       handleGetOutput: (id, res) => this.handleGetOutput(id, res),
       handleSendInput: (id, req, res) => this.handleSendInput(id, req, res),
       handleCancel: (id, res) => this.handleCancel(id, res),
+      handleStopSession: (id, res) => this.handleStopSession(id, res),
       handleSetSessionName: (id, req, res) => this.handleSetSessionName(id, req, res),
       handleUploadFile: (id, req, res) => this.handleUploadFile(id, req, res),
       handleAdopt: (id, res) => this.handleAdopt(id, res),
@@ -5267,6 +5276,54 @@ var StreamerServer = class {
       json(res, 400, { error: message });
     }
   }
+  async handleStopSession(sessionId, res) {
+    const STOP_TIMEOUT_MS = 5e3;
+    const session = this.ptyManager.getSession(sessionId);
+    if (!session) {
+      res.writeHead(404, { "Content-Type": "application/json" });
+      res.end(JSON.stringify({ error: "Session not found" }));
+      return;
+    }
+    if (session.status === "idle") {
+      res.writeHead(200, { "Content-Type": "application/json" });
+      res.end(JSON.stringify({ status: "already_idle", sessionId }));
+      return;
+    }
+    res.writeHead(200, {
+      "Content-Type": "application/x-ndjson",
+      "Transfer-Encoding": "chunked",
+      "Cache-Control": "no-cache",
+      "X-Accel-Buffering": "no"
+    });
+    res.write(`${JSON.stringify({ event: "stopping", sessionId })}
+`);
+    const idlePromise = new Promise((resolve2) => {
+      const handler = (status) => {
+        if (status === "idle") {
+          this.sessionStatusBus.off(`status:${sessionId}`, handler);
+          resolve2("idle");
+        }
+      };
+      this.sessionStatusBus.on(`status:${sessionId}`, handler);
+    });
+    const timeoutPromise = new Promise(
+      (resolve2) => setTimeout(() => resolve2("timeout"), STOP_TIMEOUT_MS)
+    );
+    this.ptyManager.putOnHold(sessionId);
+    this.discoveryCache = null;
+    const outcome = await Promise.race([idlePromise, timeoutPromise]);
+    if (outcome === "idle") {
+      res.write(`${JSON.stringify({ event: "stopped", sessionId })}
+`);
+    } else {
+      res.write(`${JSON.stringify({ event: "timeout", sessionId })}
+`);
+      this.log.warn(
+        `[stop] session ${sessionId.slice(0, 8)} did not idle within ${STOP_TIMEOUT_MS}ms`
+      );
+    }
+    res.end();
+  }
   async handleAdopt(sessionId, res) {
     const discovered = await discoverClaudeProcesses();
     this.sessionStore.setDiscovered(discovered);