pty-manager 1.2.20 → 1.2.22

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;
@@ -277,6 +277,15 @@ var PTYSession = class extends EventEmitter {
     this.logger = logger || consoleLogger;
     this._stallDetectionEnabled = stallDetectionEnabled ?? false;
     this._stallTimeoutMs = config.stallTimeoutMs ?? defaultStallTimeoutMs ?? 8e3;
+    if (config.ruleOverrides) {
+      for (const [key, value] of Object.entries(config.ruleOverrides)) {
+        if (value === null) {
+          this._disabledRulePatterns.add(key);
+        } else {
+          this._ruleOverrides.set(key, value);
+        }
+      }
+    }
   }
   ptyProcess = null;
   outputBuffer = "";
@@ -288,6 +297,8 @@ var PTYSession = class extends EventEmitter {
   sessionRules = [];
   _firedOnceRules = /* @__PURE__ */ new Set();
   _lastBlockingPromptHash = null;
+  _ruleOverrides = /* @__PURE__ */ new Map();
+  _disabledRulePatterns = /* @__PURE__ */ new Set();
   // Stall detection
   _stallTimer = null;
   _stallTimeoutMs;
@@ -295,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() {
@@ -470,10 +484,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;
   }
@@ -510,6 +526,7 @@ var PTYSession = class extends EventEmitter {
             this.writeRaw(resp + "\r");
           }
           this.emit("blocking_prompt", promptInfo, true);
+          this.outputBuffer = "";
         } else {
           this.emit("blocking_prompt", promptInfo, false);
         }
@@ -530,6 +547,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
   // ─────────────────────────────────────────────────────────────────────────────
   /**
@@ -591,10 +641,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;
@@ -604,6 +650,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") {
@@ -679,6 +734,7 @@ var PTYSession = class extends EventEmitter {
             this.writeRaw(resp + "\r");
           }
           this._lastBlockingPromptHash = null;
+          this.outputBuffer = "";
           this.emit("blocking_prompt", promptInfo, true);
           return true;
         }
@@ -706,7 +762,10 @@ var PTYSession = class extends EventEmitter {
    * Session rules are checked first, then adapter rules.
    */
   tryAutoResponse() {
-    const adapterRules = this.adapter.autoResponseRules || [];
+    const adapterRules = (this.adapter.autoResponseRules || []).filter((r) => !this._disabledRulePatterns.has(r.pattern.source)).map((r) => {
+      const override = this._ruleOverrides.get(r.pattern.source);
+      return override ? { ...r, ...override } : r;
+    });
     const allRules = [...this.sessionRules, ...adapterRules];
     if (allRules.length === 0) {
       return false;
@@ -822,6 +881,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}`,
@@ -932,6 +992,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");
     }
@@ -1085,6 +1146,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);
@@ -1938,6 +2005,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) {