pty-manager 1.2.21 → 1.3.0

package/dist/index.mjs

@@ -268,7 +268,7 @@ var SPECIAL_KEYS = {
 };
 var BRACKETED_PASTE_START = "\x1B[200~";
 var BRACKETED_PASTE_END = "\x1B[201~";
-var PTYSession = class extends EventEmitter {
+var PTYSession = class _PTYSession extends EventEmitter {
   constructor(adapter, config, logger, stallDetectionEnabled, defaultStallTimeoutMs) {
     super();
     this.adapter = adapter;
@@ -306,6 +306,9 @@ var PTYSession = class extends EventEmitter {
   _lastStallHash = null;
   _stallStartedAt = null;
   _lastContentHash = null;
+  // Task completion detection (idle detection when busy)
+  _taskCompleteTimer = null;
+  static TASK_COMPLETE_DEBOUNCE_MS = 1500;
   id;
   config;
   get status() {
@@ -445,6 +448,19 @@ var PTYSession = class extends EventEmitter {
       return;
     }
     this._lastStallHash = hash;
+    if (this._status === "busy" && this.adapter.detectTaskComplete?.(this.outputBuffer)) {
+      this._status = "ready";
+      this._lastBlockingPromptHash = null;
+      this.outputBuffer = "";
+      this.clearStallTimer();
+      this.emit("status_changed", "ready");
+      this.emit("task_complete");
+      this.logger.info(
+        { sessionId: this.id },
+        "Task complete (adapter fast-path) \u2014 agent returned to idle prompt"
+      );
+      return;
+    }
     const recentRaw = this.outputBuffer.slice(-2e3);
     const recentOutput = this.stripAnsiForStall(recentRaw);
     const stallDurationMs = this._stallStartedAt ? Date.now() - this._stallStartedAt : this._stallTimeoutMs;
@@ -481,10 +497,12 @@ var PTYSession = class extends EventEmitter {
    * word boundaries — e.g. "Do\x1b[5Cyou" becomes "Do you", not "Doyou".
    */
   stripAnsiForStall(str) {
-    let result = str.replace(/\x1b\[\d*[CDABG]/g, " ");
+    let result = str.replace(/\x1b\[\d*[CDABGdEF]/g, " ");
     result = result.replace(/\x1b\[\d*(?:;\d+)?[Hf]/g, " ");
+    result = result.replace(/\x1b\[\d*[JK]/g, " ");
     result = result.replace(/\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])/g, "");
-    result = result.replace(/[│╭╰╮╯─═╌║╔╗╚╝╠╣╦╩╬┌┐└┘├┤┬┴┼●○❯❮▶◀⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏⣾⣽⣻⢿⡿⣟⣯⣷✻✶✳✢⏺←→↑↓]/g, " ");
+    result = result.replace(/[\x00-\x08\x0b\x0c\x0e-\x1f\x7f]/g, "");
+    result = result.replace(/[│╭╰╮╯─═╌║╔╗╚╝╠╣╦╩╬┌┐└┘├┤┬┴┼●○❯❮▶◀⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏⣾⣽⣻⢿⡿⣟⣯⣷✻✶✳✢⏺←→↑↓⬆⬇◆◇▪▫■□▲△▼▽◈⟨⟩⌘⏎⏏⌫⌦⇧⇪⌥]/g, " ");
     result = result.replace(/ {2,}/g, " ");
     return result;
   }
@@ -542,6 +560,39 @@ var PTYSession = class extends EventEmitter {
     }
   }
   // ─────────────────────────────────────────────────────────────────────────────
+  // Task Completion Detection
+  // ─────────────────────────────────────────────────────────────────────────────
+  /**
+   * Schedule a task_complete transition after a debounce period.
+   * If new non-whitespace output arrives before the timer fires,
+   * the timer is cancelled (by cancelTaskComplete in onData).
+   */
+  scheduleTaskComplete() {
+    if (this._taskCompleteTimer) return;
+    this._taskCompleteTimer = setTimeout(() => {
+      this._taskCompleteTimer = null;
+      if (this._status !== "busy") return;
+      if (!this.adapter.detectReady(this.outputBuffer)) return;
+      this._status = "ready";
+      this._lastBlockingPromptHash = null;
+      this.outputBuffer = "";
+      this.clearStallTimer();
+      this.emit("status_changed", "ready");
+      this.emit("task_complete");
+      this.logger.info({ sessionId: this.id }, "Task complete \u2014 agent returned to idle prompt");
+    }, _PTYSession.TASK_COMPLETE_DEBOUNCE_MS);
+  }
+  /**
+   * Cancel a pending task_complete timer (new output arrived that
+   * doesn't match the idle prompt, so the agent is still working).
+   */
+  cancelTaskComplete() {
+    if (this._taskCompleteTimer) {
+      clearTimeout(this._taskCompleteTimer);
+      this._taskCompleteTimer = null;
+    }
+  }
+  // ─────────────────────────────────────────────────────────────────────────────
   // Lifecycle
   // ─────────────────────────────────────────────────────────────────────────────
   /**
@@ -603,10 +654,6 @@ var PTYSession = class extends EventEmitter {
         this.resetStallTimer();
       }
       this.emit("output", data);
-      const blockingPrompt = this.detectAndHandleBlockingPrompt();
-      if (blockingPrompt) {
-        return;
-      }
       if ((this._status === "starting" || this._status === "authenticating") && this.adapter.detectReady(this.outputBuffer)) {
         this._status = "ready";
         this._lastBlockingPromptHash = null;
@@ -616,6 +663,15 @@ var PTYSession = class extends EventEmitter {
         this.logger.info({ sessionId: this.id }, "Session ready");
         return;
       }
+      if (this._status === "busy" && this.adapter.detectReady(this.outputBuffer)) {
+        this.scheduleTaskComplete();
+      } else {
+        this.cancelTaskComplete();
+      }
+      const blockingPrompt = this.detectAndHandleBlockingPrompt();
+      if (blockingPrompt) {
+        return;
+      }
       if (this._status !== "ready" && this._status !== "busy") {
         const loginDetection = this.adapter.detectLogin(this.outputBuffer);
         if (loginDetection.required && this._status !== "authenticating") {
@@ -838,6 +894,7 @@ var PTYSession = class extends EventEmitter {
    */
   send(message) {
     this._status = "busy";
+    this.emit("status_changed", "busy");
     this.resetStallTimer();
     const msg = {
       id: `${this.id}-msg-${++this.messageCounter}`,
@@ -948,6 +1005,7 @@ var PTYSession = class extends EventEmitter {
     if (this.ptyProcess) {
       this._status = "stopping";
       this.clearStallTimer();
+      this.cancelTaskComplete();
       this.ptyProcess.kill(signal);
       this.logger.info({ sessionId: this.id, signal }, "Killing PTY session");
     }
@@ -1101,6 +1159,12 @@ var PTYManager = class extends EventEmitter2 {
     session.on("error", (error) => {
       this.emit("session_error", session.toHandle(), error.message);
     });
+    session.on("status_changed", () => {
+      this.emit("session_status_changed", session.toHandle());
+    });
+    session.on("task_complete", () => {
+      this.emit("task_complete", session.toHandle());
+    });
     session.on("stall_detected", (recentOutput, stallDurationMs) => {
       const handle = session.toHandle();
       this.emit("stall_detected", handle, recentOutput, stallDurationMs);
@@ -1505,6 +1569,14 @@ var BaseCLIAdapter = class {
     }
     return { detected: false };
   }
+  /**
+   * Default task completion detection — delegates to detectReady().
+   * Subclasses should override to match high-confidence completion patterns
+   * (e.g. duration summaries) that short-circuit the LLM stall classifier.
+   */
+  detectTaskComplete(output) {
+    return this.detectReady(output);
+  }
   /**
    * Default input formatting - just return as-is
    */
@@ -1954,6 +2026,24 @@ var BunCompatiblePTYManager = class extends EventEmitter3 {
         }
         break;
       }
+      case "status_changed": {
+        const session = this.sessions.get(id);
+        if (session) {
+          session.status = event.status;
+          session.lastActivityAt = /* @__PURE__ */ new Date();
+          this.emit("session_status_changed", session);
+        }
+        break;
+      }
+      case "task_complete": {
+        const session = this.sessions.get(id);
+        if (session) {
+          session.status = "ready";
+          session.lastActivityAt = /* @__PURE__ */ new Date();
+          this.emit("task_complete", session);
+        }
+        break;
+      }
       case "stall_detected": {
         const session = this.sessions.get(id);
         if (session) {