@anusornneal/chat-relay 0.7.0 → 0.7.2

package/agent/local-agent.mjs

@@ -12,7 +12,10 @@ const reconnectMs = Number(process.env.RECONNECT_MS ?? 2000);
 const terminalEnabled = process.env.TERMINAL_ENABLED === "1";
 const desktopEnabled = process.env.DESKTOP_ENABLED === "1";
 const MAX_RESPONSE_BYTES = 60 * 1024;
-const terminals = new TerminalManager();
+const terminals = new TerminalManager({
+  batchConcurrency: process.env.TERMINAL_BATCH_CONCURRENCY,
+  maxQueuedJobs: process.env.TERMINAL_BATCH_MAX_QUEUED,
+});
 const files = new FileManager(process.env.ALLOWED_ROOTS);
 const processes = new ProcessManager();
 const desktop = new DesktopManager({ enabled: desktopEnabled });
@@ -52,6 +55,7 @@ async function handlePayload(payload) {
         agentId,
         agentName,
         terminalEnabled,
+        terminalBatch: terminals.getBatchConfig(),
         desktop: desktop.getConfig(),
         allowedRoots: files.getRoots(),
         reconnectMs,
@@ -62,7 +66,7 @@ async function handlePayload(payload) {
     case "fs.stat":
       return files.stat(payload.path);
     case "fs.list":
-      return files.list(payload.path, payload.depth);
+      return files.list(payload.path, payload.depth, payload.offset, payload.limit, payload.maxBytes);
     case "fs.read":
       return files.read(payload.path, payload.offset, payload.length, payload.maxBytes);
     case "fs.readMany":
@@ -88,21 +92,35 @@ async function handlePayload(payload) {
       return processes.kill(payload.pid);
 
     case "desktop.screenshot":
-      return desktop.screenshot();
+      return desktop.screenshot(payload);
     case "desktop.mouse.click":
       return desktop.mouseClick(payload);
     case "desktop.keyboard.input":
       return desktop.keyboardInput(payload);
+    case "desktop.step":
+      return desktop.step(payload);
 
     case "terminal.exec":
       if (!terminalEnabled) return { ok: false, error: "terminal_disabled" };
       return terminals.exec(payload.command, payload.cwd, payload.timeoutMs);
+    case "terminal.batch.start":
+      if (!terminalEnabled) return { ok: false, error: "terminal_disabled" };
+      return terminals.batchStart(payload.jobs, { cwd: payload.cwd, timeoutMs: payload.timeoutMs, concurrency: payload.concurrency, observability: payload.observability });
+    case "terminal.batch.status":
+      if (!terminalEnabled) return { ok: false, error: "terminal_disabled" };
+      return terminals.batchStatus(payload.batchId);
+    case "terminal.batch.read":
+      if (!terminalEnabled) return { ok: false, error: "terminal_disabled" };
+      return terminals.batchRead(payload.batchId, payload.maxChars);
+    case "terminal.batch.cancel":
+      if (!terminalEnabled) return { ok: false, error: "terminal_disabled" };
+      return terminals.batchCancel(payload.batchId);
     case "terminal.start":
       if (!terminalEnabled) return { ok: false, error: "terminal_disabled" };
-      return terminals.start(payload.command, payload.cwd);
+      return terminals.start(payload.command, payload.cwd, payload.observability);
     case "terminal.shell.start":
       if (!terminalEnabled) return { ok: false, error: "terminal_disabled" };
-      return terminals.startShell(payload.cwd);
+      return terminals.startShell(payload.cwd, payload.observability);
     case "terminal.read":
       if (!terminalEnabled) return { ok: false, error: "terminal_disabled" };
       return terminals.read(payload.sessionId, payload.afterSeq, payload.maxChars);
@@ -115,6 +133,9 @@ async function handlePayload(payload) {
     case "terminal.list":
       if (!terminalEnabled) return { ok: false, error: "terminal_disabled" };
       return terminals.list();
+    case "terminal.observability":
+      if (!terminalEnabled) return { ok: false, error: "terminal_disabled" };
+      return terminals.observability();
 
     default:
       return { ok: false, error: "unknown_action" };
@@ -124,6 +145,7 @@ async function handlePayload(payload) {
 function requireReauthorization() {
   if (reauthorizationRequired) return;
   reauthorizationRequired = true;
+  desktop.close();
   process.exitCode = 2;
   console.error("Agent credential was revoked or rejected.");
   console.error('Recovery: run "chat-relay login --force", then "chat-relay remote".');
@@ -209,4 +231,12 @@ function connect() {
   });
 }
 
+function shutdown() {
+  desktop.close();
+}
+
+process.once("SIGINT", shutdown);
+process.once("SIGTERM", shutdown);
+process.once("exit", shutdown);
+
 connect();