@viraatdas/rudder 3.1.7 → 3.1.8

package/dist/cloud.js

@@ -1,5 +1,6 @@
 import { spawn } from "node:child_process";
 import { createHash, randomUUID } from "node:crypto";
+import { EchoPredictor } from "./predict.js";
 import fsp from "node:fs/promises";
 import os from "node:os";
 import path from "node:path";
@@ -2828,12 +2829,19 @@ async function runAttach(target, options) {
         let result = "failed";
         const splashAllowed = isInteractive && !options.json && !options.quietBanner;
         const splash = splashAllowed ? new AttachSplash(stdout, target.label) : null;
+        // Predictive local echo (src/predict.ts): typed characters appear at the
+        // caret at once and the cloud's frame confirms them. RUDDER_PREDICT=0
+        // turns it off; it only engages once the round trip is worth hiding.
+        const predictor = isInteractive && process.env.RUDDER_PREDICT !== "0"
+            ? new EchoPredictor(stdout.columns ?? 120, stdout.rows ?? 32, (text) => stdout.write(text))
+            : null;
         const sendResize = () => {
             if (socket.readyState !== WebSocket.OPEN) {
                 return;
             }
             const cols = stdout.columns ?? 120;
             const rows = stdout.rows ?? 32;
+            predictor?.resize(cols, rows);
             socket.send(JSON.stringify({ type: "resize", cols, rows }));
         };
         let lastCtrlC = 0;
@@ -2911,6 +2919,7 @@ async function runAttach(target, options) {
                 lastCtrlC = now;
             }
             socket.send(buffer, { binary: true });
+            predictor?.onInput(buffer);
         };
         const onResize = () => {
             if (hasControl) {
@@ -2925,7 +2934,14 @@ async function runAttach(target, options) {
         // as a dropped connection, which attachWithReconnect reopens.
         let lastHeard = Date.now();
         let lastTick = Date.now();
-        socket.on("pong", () => { lastHeard = Date.now(); });
+        let pingSentAt = 0;
+        socket.on("pong", () => {
+            lastHeard = Date.now();
+            if (pingSentAt) {
+                predictor?.noteRtt(Date.now() - pingSentAt);
+                pingSentAt = 0;
+            }
+        });
         socket.on("message", () => { lastHeard = Date.now(); });
         const liveness = setInterval(() => {
             const now = Date.now();
@@ -2940,6 +2956,7 @@ async function runAttach(target, options) {
             }
             try {
                 socket.ping();
+                pingSentAt = Date.now();
             }
             catch {
                 socket.terminate();
@@ -2952,6 +2969,7 @@ async function runAttach(target, options) {
             }
             cleaned = true;
             clearInterval(liveness);
+            predictor?.dispose();
             stdin.off("data", onStdin);
             stdout.off("resize", onResize);
             process.off("SIGINT", onSigint);
@@ -3105,6 +3123,20 @@ async function runAttach(target, options) {
         };
         socket.on("open", () => {
             opened = true;
+            // Measure the round trip at once: predictive echo waits on it.
+            try {
+                socket.ping();
+                pingSentAt = Date.now();
+            }
+            catch {
+                // the liveness timer retries
+            }
+            if (probeMode) {
+                // An idle dashboard sends nothing; a resize makes it repaint, which is
+                // the first frame the probe waits for.
+                socket.send(JSON.stringify({ type: "resize", cols: 119, rows: 32 }));
+                setTimeout(() => socket.send(JSON.stringify({ type: "resize", cols: 120, rows: 32 })), 200);
+            }
             // Disable Nagle on the underlying TCP socket so single keystrokes don't
             // sit in the send buffer waiting for piggyback ACKs (~40ms savings per
             // keypress on the WAN).
@@ -3211,6 +3243,7 @@ async function runAttach(target, options) {
                     setTimeout(sendResize, 150);
                 }
                 stdout.write(data);
+                predictor?.onServerData(data);
                 return;
             }
             const text = Buffer.isBuffer(data)
@@ -3673,6 +3706,9 @@ const LAPTOP_CONFIG_FILES = [
     "~/.codex/config.toml",
     "~/.codex/AGENTS.md",
     "~/.codex/hooks.json",
+    // Service logins agents use from scripts (Kalshi signs requests with this key).
+    "~/.kalshi/config.json",
+    "~/.kalshi/private_key.pem",
 ];
 async function laptopConfigFiles() {
     const names = [...LAPTOP_CONFIG_FILES, "~/.aws/config", "~/.aws/credentials"];