pty-manager 1.2.12 → 1.2.14

package/dist/index.js

@@ -307,12 +307,14 @@ var SPECIAL_KEYS = {
 var BRACKETED_PASTE_START = "\x1B[200~";
 var BRACKETED_PASTE_END = "\x1B[201~";
 var PTYSession = class extends import_events.EventEmitter {
-  constructor(adapter, config, logger) {
+  constructor(adapter, config, logger, stallDetectionEnabled, defaultStallTimeoutMs) {
     super();
     this.adapter = adapter;
     this.id = config.id || generateId();
     this.config = { ...config, id: this.id };
     this.logger = logger || consoleLogger;
+    this._stallDetectionEnabled = stallDetectionEnabled ?? false;
+    this._stallTimeoutMs = config.stallTimeoutMs ?? defaultStallTimeoutMs ?? 8e3;
   }
   ptyProcess = null;
   outputBuffer = "";
@@ -323,6 +325,12 @@ var PTYSession = class extends import_events.EventEmitter {
   logger;
   sessionRules = [];
   _lastBlockingPromptHash = null;
+  // Stall detection
+  _stallTimer = null;
+  _stallTimeoutMs;
+  _stallDetectionEnabled;
+  _lastStallHash = null;
+  _stallStartedAt = null;
   id;
   config;
   get status() {
@@ -404,6 +412,123 @@ var PTYSession = class extends import_events.EventEmitter {
     this.logger.debug({ sessionId: this.id }, "Cleared auto-response rules");
   }
   // ─────────────────────────────────────────────────────────────────────────────
+  // Stall Detection
+  // ─────────────────────────────────────────────────────────────────────────────
+  /**
+   * Start or reset the stall detection timer.
+   * Only active when status is "busy" and stall detection is enabled.
+   */
+  resetStallTimer() {
+    this.clearStallTimer();
+    if (!this._stallDetectionEnabled || this._status !== "busy") {
+      return;
+    }
+    this._stallStartedAt = Date.now();
+    this._lastStallHash = null;
+    this._stallTimer = setTimeout(() => {
+      this.onStallTimerFired();
+    }, this._stallTimeoutMs);
+  }
+  /**
+   * Clear the stall detection timer.
+   */
+  clearStallTimer() {
+    if (this._stallTimer) {
+      clearTimeout(this._stallTimer);
+      this._stallTimer = null;
+    }
+    this._stallStartedAt = null;
+  }
+  /**
+   * Called when the stall timer fires (no output for stallTimeoutMs).
+   */
+  onStallTimerFired() {
+    if (this._status !== "busy") {
+      return;
+    }
+    const tail = this.outputBuffer.slice(-500);
+    const hash = this.simpleHash(tail);
+    if (hash === this._lastStallHash) {
+      this._stallTimer = setTimeout(() => this.onStallTimerFired(), this._stallTimeoutMs);
+      return;
+    }
+    this._lastStallHash = hash;
+    const recentRaw = this.outputBuffer.slice(-2e3);
+    const recentOutput = this.stripAnsiForStall(recentRaw);
+    const stallDurationMs = this._stallStartedAt ? Date.now() - this._stallStartedAt : this._stallTimeoutMs;
+    this.logger.debug(
+      { sessionId: this.id, stallDurationMs, bufferTailLength: tail.length },
+      "Stall detected"
+    );
+    this.emit("stall_detected", recentOutput, stallDurationMs);
+    this._stallTimer = setTimeout(() => this.onStallTimerFired(), this._stallTimeoutMs);
+  }
+  /**
+   * Simple string hash for deduplication.
+   */
+  simpleHash(str) {
+    let hash = 0;
+    for (let i = 0; i < str.length; i++) {
+      const char = str.charCodeAt(i);
+      hash = (hash << 5) - hash + char;
+      hash |= 0;
+    }
+    return hash.toString(36);
+  }
+  /**
+   * Strip ANSI codes for stall detection output.
+   * Replaces cursor-forward sequences with spaces first.
+   */
+  stripAnsiForStall(str) {
+    const withSpaces = str.replace(/\x1b\[\d*C/g, " ");
+    return withSpaces.replace(/\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])/g, "");
+  }
+  /**
+   * Handle external stall classification result.
+   * Called by the manager after onStallClassify resolves.
+   */
+  handleStallClassification(classification) {
+    if (this._status !== "busy") {
+      return;
+    }
+    if (!classification || classification.state === "still_working") {
+      this.resetStallTimer();
+      return;
+    }
+    switch (classification.state) {
+      case "waiting_for_input": {
+        const promptInfo = {
+          type: "stall_classified",
+          prompt: classification.prompt,
+          canAutoRespond: !!classification.suggestedResponse
+        };
+        if (classification.suggestedResponse) {
+          this.logger.info(
+            { sessionId: this.id, response: classification.suggestedResponse },
+            "Auto-responding to stall-classified prompt"
+          );
+          this.writeRaw(classification.suggestedResponse + "\r");
+          this.emit("blocking_prompt", promptInfo, true);
+        } else {
+          this.emit("blocking_prompt", promptInfo, false);
+        }
+        break;
+      }
+      case "task_complete":
+        this._status = "ready";
+        this._lastBlockingPromptHash = null;
+        this.outputBuffer = "";
+        this.clearStallTimer();
+        this.emit("ready");
+        this.logger.info({ sessionId: this.id }, "Stall classified as task_complete, transitioning to ready");
+        break;
+      case "error":
+        this.clearStallTimer();
+        this.emit("error", new Error(classification.prompt || "Stall classified as error"));
+        break;
+    }
+  }
+  // ─────────────────────────────────────────────────────────────────────────────
   // Lifecycle
   // ─────────────────────────────────────────────────────────────────────────────
   /**
@@ -461,11 +586,15 @@ var PTYSession = class extends import_events.EventEmitter {
     this.ptyProcess.onData((data) => {
       this._lastActivityAt = /* @__PURE__ */ new Date();
       this.outputBuffer += data;
+      if (this._status === "busy") {
+        this.resetStallTimer();
+      }
       this.emit("output", data);
       if ((this._status === "starting" || this._status === "authenticating") && this.adapter.detectReady(this.outputBuffer)) {
         this._status = "ready";
         this._lastBlockingPromptHash = null;
         this.outputBuffer = "";
+        this.clearStallTimer();
         this.emit("ready");
         this.logger.info({ sessionId: this.id }, "Session ready");
         return;
@@ -478,6 +607,7 @@ var PTYSession = class extends import_events.EventEmitter {
         const loginDetection = this.adapter.detectLogin(this.outputBuffer);
         if (loginDetection.required && this._status !== "authenticating") {
           this._status = "authenticating";
+          this.clearStallTimer();
           this.emit("login_required", loginDetection.instructions, loginDetection.url);
           this.logger.warn(
             { sessionId: this.id, loginType: loginDetection.type },
@@ -489,6 +619,7 @@ var PTYSession = class extends import_events.EventEmitter {
       const exitDetection = this.adapter.detectExit(this.outputBuffer);
       if (exitDetection.exited) {
         this._status = "stopped";
+        this.clearStallTimer();
         this.emit("exit", exitDetection.code || 0);
       }
       if (this._status !== "starting" && this._status !== "authenticating") {
@@ -497,6 +628,7 @@ var PTYSession = class extends import_events.EventEmitter {
     });
     this.ptyProcess.onExit(({ exitCode, signal }) => {
       this._status = "stopped";
+      this.clearStallTimer();
       this.logger.info(
         { sessionId: this.id, exitCode, signal },
         "PTY session exited"
@@ -572,8 +704,11 @@ var PTYSession = class extends import_events.EventEmitter {
     if (allRules.length === 0) {
       return false;
     }
+    let stripped = this.stripAnsiForStall(this.outputBuffer);
+    stripped = stripped.replace(/[│╭╰╮╯─═╌║╔╗╚╝╠╣╦╩╬┌┐└┘├┤┬┴┼●○❯❮▶◀⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏⏺←→↑↓]/g, " ");
+    stripped = stripped.replace(/ {2,}/g, " ");
     for (const rule of allRules) {
-      if (rule.pattern.test(this.outputBuffer)) {
+      if (rule.pattern.test(stripped)) {
         const safe = rule.safe !== false;
         const isSessionRule = this.sessionRules.includes(rule);
         if (safe) {
@@ -664,6 +799,7 @@ var PTYSession = class extends import_events.EventEmitter {
    */
   send(message) {
     this._status = "busy";
+    this.resetStallTimer();
     const msg = {
       id: `${this.id}-msg-${++this.messageCounter}`,
       sessionId: this.id,
@@ -752,6 +888,7 @@ var PTYSession = class extends import_events.EventEmitter {
   kill(signal) {
     if (this.ptyProcess) {
       this._status = "stopping";
+      this.clearStallTimer();
       this.ptyProcess.kill(signal);
       this.logger.info({ sessionId: this.id, signal }, "Killing PTY session");
     }
@@ -821,11 +958,18 @@ var PTYManager = class extends import_events2.EventEmitter {
   maxLogLines;
   logger;
   adapters;
+  // Stall detection config
+  _stallDetectionEnabled;
+  _stallTimeoutMs;
+  _onStallClassify;
   constructor(config = {}) {
     super();
     this.adapters = new AdapterRegistry();
     this.logger = config.logger || consoleLogger2;
     this.maxLogLines = config.maxLogLines || 1e3;
+    this._stallDetectionEnabled = config.stallDetectionEnabled ?? false;
+    this._stallTimeoutMs = config.stallTimeoutMs ?? 8e3;
+    this._onStallClassify = config.onStallClassify;
   }
   /**
    * Register a CLI adapter
@@ -848,7 +992,13 @@ var PTYManager = class extends import_events2.EventEmitter {
       { type: config.type, name: config.name },
       "Spawning session"
     );
-    const session = new PTYSession(adapter, config, this.logger);
+    const session = new PTYSession(
+      adapter,
+      config,
+      this.logger,
+      this._stallDetectionEnabled,
+      this._stallTimeoutMs
+    );
     this.setupSessionEvents(session);
     this.sessions.set(session.id, session);
     this.outputLogs.set(session.id, []);
@@ -892,6 +1042,21 @@ var PTYManager = class extends import_events2.EventEmitter {
     session.on("error", (error) => {
       this.emit("session_error", session.toHandle(), error.message);
     });
+    session.on("stall_detected", (recentOutput, stallDurationMs) => {
+      const handle = session.toHandle();
+      this.emit("stall_detected", handle, recentOutput, stallDurationMs);
+      if (this._onStallClassify) {
+        this._onStallClassify(session.id, recentOutput, stallDurationMs).then((classification) => {
+          session.handleStallClassification(classification);
+        }).catch((err) => {
+          this.logger.error(
+            { sessionId: session.id, error: err },
+            "Stall classification callback failed"
+          );
+          session.handleStallClassification(null);
+        });
+      }
+    });
   }
   /**
    * Stop a session
@@ -1063,6 +1228,22 @@ var PTYManager = class extends import_events2.EventEmitter {
     return this.sessions.get(sessionId);
   }
   // ─────────────────────────────────────────────────────────────────────────────
+  // Stall Detection Configuration
+  // ─────────────────────────────────────────────────────────────────────────────
+  /**
+   * Configure stall detection at runtime.
+   * Affects newly spawned sessions only — existing sessions keep their config.
+   */
+  configureStallDetection(enabled, timeoutMs, classify) {
+    this._stallDetectionEnabled = enabled;
+    if (timeoutMs !== void 0) {
+      this._stallTimeoutMs = timeoutMs;
+    }
+    if (classify !== void 0) {
+      this._onStallClassify = classify;
+    }
+  }
+  // ─────────────────────────────────────────────────────────────────────────────
   // Runtime Auto-Response Rules API
   // ─────────────────────────────────────────────────────────────────────────────
   /**
@@ -1152,7 +1333,9 @@ var BaseCLIAdapter = class {
    * Subclasses should override for CLI-specific detection.
    */
   detectBlockingPrompt(output) {
-    const stripped = this.stripAnsi(output);
+    let stripped = this.stripAnsi(output);
+    stripped = stripped.replace(/[│╭╰╮╯─═╌║╔╗╚╝╠╣╦╩╬┌┐└┘├┤┬┴┼●○❯❮▶◀⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏⏺←→↑↓]/g, " ");
+    stripped = stripped.replace(/ {2,}/g, " ");
     const loginDetection = this.detectLogin(output);
     if (loginDetection.required) {
       return {
@@ -1332,7 +1515,8 @@ var BaseCLIAdapter = class {
    * Helper to strip ANSI escape codes from output
    */
   stripAnsi(str) {
-    return str.replace(/\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])/g, "");
+    const withSpaces = str.replace(/\x1b\[\d*C/g, " ");
+    return withSpaces.replace(/\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])/g, "");
   }
 };
 
@@ -1535,12 +1719,18 @@ var BunCompatiblePTYManager = class extends import_events3.EventEmitter {
   workerPath;
   env;
   adapterModules;
+  _stallDetectionEnabled;
+  _stallTimeoutMs;
+  _onStallClassify;
   constructor(options = {}) {
     super();
     this.nodePath = options.nodePath || "node";
     this.workerPath = options.workerPath || this.findWorkerPath();
     this.env = options.env || {};
     this.adapterModules = options.adapterModules || [];
+    this._stallDetectionEnabled = options.stallDetectionEnabled ?? false;
+    this._stallTimeoutMs = options.stallTimeoutMs ?? 8e3;
+    this._onStallClassify = options.onStallClassify;
     this.readyPromise = new Promise((resolve) => {
       this.readyResolve = resolve;
     });
@@ -1602,6 +1792,13 @@ var BunCompatiblePTYManager = class extends import_events3.EventEmitter {
         if (this.adapterModules.length > 0) {
           this.sendCommand({ cmd: "registerAdapters", modules: this.adapterModules });
         }
+        if (this._stallDetectionEnabled) {
+          this.sendCommand({
+            cmd: "configureStallDetection",
+            enabled: true,
+            timeoutMs: this._stallTimeoutMs
+          });
+        }
         this.ready = true;
         this.readyResolve();
         this.emit("ready");
@@ -1692,6 +1889,30 @@ var BunCompatiblePTYManager = class extends import_events3.EventEmitter {
         }
         break;
       }
+      case "stall_detected": {
+        const session = this.sessions.get(id);
+        if (session) {
+          const recentOutput = event.recentOutput;
+          const stallDurationMs = event.stallDurationMs;
+          this.emit("stall_detected", session, recentOutput, stallDurationMs);
+          if (this._onStallClassify) {
+            this._onStallClassify(id, recentOutput, stallDurationMs).then((classification) => {
+              this.sendCommand({
+                cmd: "classifyStallResult",
+                id,
+                classification
+              });
+            }).catch(() => {
+              this.sendCommand({
+                cmd: "classifyStallResult",
+                id,
+                classification: null
+              });
+            });
+          }
+        }
+        break;
+      }
       case "list": {
         const sessions = event.sessions.map((s) => ({
           ...s,