pty-manager 1.2.4 → 1.2.7

package/dist/pty-worker.js

@@ -304,6 +304,7 @@ var PTYSession = class extends import_events.EventEmitter {
   _lastActivityAt = null;
   messageCounter = 0;
   logger;
+  sessionRules = [];
   id;
   config;
   get status() {
@@ -318,6 +319,75 @@ var PTYSession = class extends import_events.EventEmitter {
   get lastActivityAt() {
     return this._lastActivityAt ?? void 0;
   }
+  // ─────────────────────────────────────────────────────────────────────────────
+  // Runtime Auto-Response Rules API
+  // ─────────────────────────────────────────────────────────────────────────────
+  /**
+   * Add an auto-response rule to this session.
+   * Session rules are checked before adapter rules.
+   */
+  addAutoResponseRule(rule) {
+    const existingIndex = this.sessionRules.findIndex(
+      (r) => r.pattern.source === rule.pattern.source && r.pattern.flags === rule.pattern.flags
+    );
+    if (existingIndex >= 0) {
+      this.sessionRules[existingIndex] = rule;
+      this.logger.debug(
+        { sessionId: this.id, pattern: rule.pattern.source, type: rule.type },
+        "Replaced existing auto-response rule"
+      );
+    } else {
+      this.sessionRules.push(rule);
+      this.logger.debug(
+        { sessionId: this.id, pattern: rule.pattern.source, type: rule.type },
+        "Added auto-response rule"
+      );
+    }
+  }
+  /**
+   * Remove an auto-response rule by pattern.
+   * Returns true if a rule was removed.
+   */
+  removeAutoResponseRule(pattern) {
+    const initialLength = this.sessionRules.length;
+    this.sessionRules = this.sessionRules.filter(
+      (r) => !(r.pattern.source === pattern.source && r.pattern.flags === pattern.flags)
+    );
+    const removed = this.sessionRules.length < initialLength;
+    if (removed) {
+      this.logger.debug(
+        { sessionId: this.id, pattern: pattern.source },
+        "Removed auto-response rule"
+      );
+    }
+    return removed;
+  }
+  /**
+   * Set all session auto-response rules, replacing existing ones.
+   */
+  setAutoResponseRules(rules) {
+    this.sessionRules = [...rules];
+    this.logger.debug(
+      { sessionId: this.id, count: rules.length },
+      "Set auto-response rules"
+    );
+  }
+  /**
+   * Get all session auto-response rules.
+   */
+  getAutoResponseRules() {
+    return [...this.sessionRules];
+  }
+  /**
+   * Clear all session auto-response rules.
+   */
+  clearAutoResponseRules() {
+    this.sessionRules = [];
+    this.logger.debug({ sessionId: this.id }, "Cleared auto-response rules");
+  }
+  // ─────────────────────────────────────────────────────────────────────────────
+  // Lifecycle
+  // ─────────────────────────────────────────────────────────────────────────────
   /**
    * Start the PTY session
    */
@@ -459,23 +529,27 @@ var PTYSession = class extends import_events.EventEmitter {
     return false;
   }
   /**
-   * Try to match and apply auto-response rules
+   * Try to match and apply auto-response rules.
+   * Session rules are checked first, then adapter rules.
    */
   tryAutoResponse() {
-    const rules = this.adapter.autoResponseRules;
-    if (!rules || rules.length === 0) {
+    const adapterRules = this.adapter.autoResponseRules || [];
+    const allRules = [...this.sessionRules, ...adapterRules];
+    if (allRules.length === 0) {
       return false;
     }
-    for (const rule of rules) {
+    for (const rule of allRules) {
       if (rule.pattern.test(this.outputBuffer)) {
         const safe = rule.safe !== false;
+        const isSessionRule = this.sessionRules.includes(rule);
         if (safe) {
           this.logger.info(
             {
               sessionId: this.id,
               promptType: rule.type,
               description: rule.description,
-              response: rule.response
+              response: rule.response,
+              source: isSessionRule ? "session" : "adapter"
             },
             "Applying auto-response rule"
           );
@@ -946,6 +1020,61 @@ var PTYManager = class extends import_events2.EventEmitter {
   getSession(sessionId) {
     return this.sessions.get(sessionId);
   }
+  // ─────────────────────────────────────────────────────────────────────────────
+  // Runtime Auto-Response Rules API
+  // ─────────────────────────────────────────────────────────────────────────────
+  /**
+   * Add an auto-response rule to a session.
+   * Session rules are checked before adapter rules.
+   */
+  addAutoResponseRule(sessionId, rule) {
+    const session = this.sessions.get(sessionId);
+    if (!session) {
+      throw new Error(`Session not found: ${sessionId}`);
+    }
+    session.addAutoResponseRule(rule);
+  }
+  /**
+   * Remove an auto-response rule from a session by pattern.
+   * Returns true if a rule was removed.
+   */
+  removeAutoResponseRule(sessionId, pattern) {
+    const session = this.sessions.get(sessionId);
+    if (!session) {
+      throw new Error(`Session not found: ${sessionId}`);
+    }
+    return session.removeAutoResponseRule(pattern);
+  }
+  /**
+   * Set all auto-response rules for a session, replacing existing ones.
+   */
+  setAutoResponseRules(sessionId, rules) {
+    const session = this.sessions.get(sessionId);
+    if (!session) {
+      throw new Error(`Session not found: ${sessionId}`);
+    }
+    session.setAutoResponseRules(rules);
+  }
+  /**
+   * Get all auto-response rules for a session.
+   */
+  getAutoResponseRules(sessionId) {
+    const session = this.sessions.get(sessionId);
+    if (!session) {
+      throw new Error(`Session not found: ${sessionId}`);
+    }
+    return session.getAutoResponseRules();
+  }
+  /**
+   * Clear all auto-response rules for a session.
+   */
+  clearAutoResponseRules(sessionId) {
+    const session = this.sessions.get(sessionId);
+    if (!session) {
+      throw new Error(`Session not found: ${sessionId}`);
+    }
+    session.clearAutoResponseRules();
+  }
 };
 
 // src/adapters/shell-adapter.ts
@@ -1185,6 +1314,93 @@ async function handleShutdown() {
   }
   setTimeout(() => process.exit(0), 100);
 }
+function handleRegisterAdapters(modules) {
+  try {
+    for (const modulePath of modules) {
+      const mod = require(modulePath);
+      if (typeof mod.createAllAdapters === "function") {
+        const adapters = mod.createAllAdapters();
+        for (const adapter of adapters) {
+          manager.registerAdapter(adapter);
+        }
+      } else if (Array.isArray(mod.default)) {
+        for (const adapter of mod.default) {
+          manager.registerAdapter(adapter);
+        }
+      } else if (mod.default && typeof mod.default.adapterType === "string") {
+        manager.registerAdapter(mod.default);
+      } else {
+        throw new Error(`Module ${modulePath} does not export createAllAdapters() or adapters`);
+      }
+    }
+    ack("registerAdapters", void 0, true);
+  } catch (err) {
+    ack("registerAdapters", void 0, false, err instanceof Error ? err.message : String(err));
+  }
+}
+function deserializeRule(serialized) {
+  return {
+    pattern: new RegExp(serialized.pattern, serialized.flags || ""),
+    type: serialized.type,
+    response: serialized.response,
+    description: serialized.description,
+    safe: serialized.safe
+  };
+}
+function serializeRule(rule) {
+  return {
+    pattern: rule.pattern.source,
+    flags: rule.pattern.flags || void 0,
+    type: rule.type,
+    response: rule.response,
+    description: rule.description,
+    safe: rule.safe
+  };
+}
+function handleAddRule(id, serializedRule) {
+  try {
+    const rule = deserializeRule(serializedRule);
+    manager.addAutoResponseRule(id, rule);
+    ack("addRule", id, true);
+  } catch (err) {
+    ack("addRule", id, false, err instanceof Error ? err.message : String(err));
+  }
+}
+function handleRemoveRule(id, pattern, flags) {
+  try {
+    const regex = new RegExp(pattern, flags || "");
+    const removed = manager.removeAutoResponseRule(id, regex);
+    ack("removeRule", id, removed, removed ? void 0 : "Rule not found");
+  } catch (err) {
+    ack("removeRule", id, false, err instanceof Error ? err.message : String(err));
+  }
+}
+function handleSetRules(id, serializedRules) {
+  try {
+    const rules = serializedRules.map(deserializeRule);
+    manager.setAutoResponseRules(id, rules);
+    ack("setRules", id, true);
+  } catch (err) {
+    ack("setRules", id, false, err instanceof Error ? err.message : String(err));
+  }
+}
+function handleGetRules(id) {
+  try {
+    const rules = manager.getAutoResponseRules(id);
+    const serialized = rules.map(serializeRule);
+    emit({ event: "rules", id, rules: serialized });
+  } catch (err) {
+    emit({ event: "error", id, message: err instanceof Error ? err.message : String(err) });
+  }
+}
+function handleClearRules(id) {
+  try {
+    manager.clearAutoResponseRules(id);
+    ack("clearRules", id, true);
+  } catch (err) {
+    ack("clearRules", id, false, err instanceof Error ? err.message : String(err));
+  }
+}
 function processCommand(line) {
   let command;
   try {
@@ -1242,6 +1458,48 @@ function processCommand(line) {
     case "shutdown":
       handleShutdown();
       break;
+    case "registerAdapters":
+      if (!command.modules || !Array.isArray(command.modules)) {
+        ack("registerAdapters", void 0, false, "Missing modules array");
+        return;
+      }
+      handleRegisterAdapters(command.modules);
+      break;
+    case "addRule":
+      if (!command.id || !command.rule) {
+        ack("addRule", command.id, false, "Missing id or rule");
+        return;
+      }
+      handleAddRule(command.id, command.rule);
+      break;
+    case "removeRule":
+      if (!command.id || !command.pattern) {
+        ack("removeRule", command.id, false, "Missing id or pattern");
+        return;
+      }
+      handleRemoveRule(command.id, command.pattern, command.flags);
+      break;
+    case "setRules":
+      if (!command.id || !command.rules) {
+        ack("setRules", command.id, false, "Missing id or rules");
+        return;
+      }
+      handleSetRules(command.id, command.rules);
+      break;
+    case "getRules":
+      if (!command.id) {
+        ack("getRules", command.id, false, "Missing id");
+        return;
+      }
+      handleGetRules(command.id);
+      break;
+    case "clearRules":
+      if (!command.id) {
+        ack("clearRules", command.id, false, "Missing id");
+        return;
+      }
+      handleClearRules(command.id);
+      break;
     default:
       emit({ event: "error", message: `Unknown command: ${command.cmd}` });
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pty-manager",
-  "version": "1.2.4",
+  "version": "1.2.7",
   "description": "PTY session manager with lifecycle management, pluggable adapters, and blocking prompt detection",
   "main": "dist/index.js",
   "module": "dist/index.mjs",