@openparachute/agent 0.2.2 → 0.2.3-rc.11
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.parachute/module.json +3 -3
- package/package.json +4 -1
- package/src/agent-defs.ts +9 -0
- package/src/auth.ts +182 -14
- package/src/backends/programmatic.ts +35 -2
- package/src/backends/registry.ts +159 -40
- package/src/backends/types.ts +44 -0
- package/src/daemon.ts +317 -12
- package/src/def-vault-triggers.ts +317 -0
- package/src/preflight.ts +139 -0
- package/src/spawn-agent.ts +16 -0
- package/src/step-up.ts +316 -0
- package/src/terminal-ui.ts +73 -0
- package/src/transports/http-ui.ts +10 -8
- package/src/transports/vault.ts +48 -27
- package/src/ui-kit.ts +6 -3
- package/src/ui-ticket.ts +121 -0
- package/web/ui/dist/assets/index-Dhr5Kl_d.css +1 -0
- package/web/ui/dist/assets/index-Di5MmFZR.js +60 -0
- package/web/ui/dist/index.html +2 -2
- package/src/_parked/interactive-spawn.test.ts +0 -324
- package/src/_parked/interactive-spawn.ts +0 -701
- package/src/agent-defs.test.ts +0 -1504
- package/src/agent-mcp-config.test.ts +0 -115
- package/src/agents.test.ts +0 -360
- package/src/auth.test.ts +0 -46
- package/src/backends/attached-queue.test.ts +0 -376
- package/src/backends/programmatic.test.ts +0 -1715
- package/src/backends/registry.test.ts +0 -1494
- package/src/backends/stream-json.test.ts +0 -570
- package/src/channel-backend-wiring.test.ts +0 -237
- package/src/credentials.test.ts +0 -274
- package/src/cron.test.ts +0 -342
- package/src/daemon-agent-def-api.test.ts +0 -166
- package/src/daemon-agent-defs-api.test.ts +0 -953
- package/src/daemon-agent-env-api.test.ts +0 -338
- package/src/daemon-attached-queue-store.test.ts +0 -65
- package/src/daemon-config-api.test.ts +0 -962
- package/src/daemon-jobs-api.test.ts +0 -271
- package/src/daemon-vault-chat.test.ts +0 -250
- package/src/daemon.test.ts +0 -746
- package/src/def-vaults.test.ts +0 -136
- package/src/delivery-state.test.ts +0 -110
- package/src/effective-env.test.ts +0 -114
- package/src/grants.test.ts +0 -638
- package/src/hub-jwt.test.ts +0 -161
- package/src/jobs.test.ts +0 -245
- package/src/mcp-http.test.ts +0 -265
- package/src/mint-token.test.ts +0 -152
- package/src/module-manifest.test.ts +0 -158
- package/src/programmatic-wiring.test.ts +0 -838
- package/src/registry.test.ts +0 -227
- package/src/resolve-port.test.ts +0 -64
- package/src/routing.test.ts +0 -184
- package/src/runner.test.ts +0 -506
- package/src/sandbox/config.test.ts +0 -150
- package/src/sandbox/egress.test.ts +0 -113
- package/src/sandbox/live-seatbelt.test.ts +0 -277
- package/src/sandbox/mounts.test.ts +0 -154
- package/src/sandbox/sandbox.test.ts +0 -168
- package/src/services-manifest.test.ts +0 -106
- package/src/spa-serve.test.ts +0 -116
- package/src/spawn-agent-cli.test.ts +0 -172
- package/src/spawn-agent.test.ts +0 -1218
- package/src/spawn-deps.test.ts +0 -54
- package/src/terminal-assets.test.ts +0 -50
- package/src/terminal.test.ts +0 -530
- package/src/transports/http-ui.test.ts +0 -455
- package/src/transports/telegram.test.ts +0 -174
- package/src/transports/vault.test.ts +0 -2011
- package/src/ui-kit.test.ts +0 -178
- package/web/ui/dist/assets/index-C-iWdFFV.css +0 -1
- package/web/ui/dist/assets/index-VFETBk0a.js +0 -60
- package/web/ui/tsconfig.json +0 -21
package/src/spawn-deps.test.ts
DELETED
|
@@ -1,54 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Tests for `resolveSpawnDeps` (`src/spawn-deps.ts`) — the real-dep resolver
|
|
3
|
-
* shared by the CLI and the web spawn endpoint.
|
|
4
|
-
*
|
|
5
|
-
* The load-bearing regression guard here is the claude config binding: the
|
|
6
|
-
* sandboxed `claude` MUST get `~/.claude.json` bound read-only, or it runs
|
|
7
|
-
* first-run onboarding whose connectivity check is FATAL under the restricted
|
|
8
|
-
* egress proxy and the tmux session dies instantly ("An unknown error occurred").
|
|
9
|
-
* That bug shipped once; this test ensures the binding stays.
|
|
10
|
-
*/
|
|
11
|
-
|
|
12
|
-
import { describe, test, expect, afterEach } from "bun:test";
|
|
13
|
-
import { mkdtempSync, writeFileSync, rmSync } from "node:fs";
|
|
14
|
-
import { tmpdir, homedir } from "node:os";
|
|
15
|
-
import { join, resolve } from "node:path";
|
|
16
|
-
import { resolveSpawnDeps, SpawnDepsError } from "./spawn-deps.ts";
|
|
17
|
-
|
|
18
|
-
const savedHome = process.env.PARACHUTE_HOME;
|
|
19
|
-
let tmp: string | undefined;
|
|
20
|
-
|
|
21
|
-
afterEach(() => {
|
|
22
|
-
if (savedHome === undefined) delete process.env.PARACHUTE_HOME;
|
|
23
|
-
else process.env.PARACHUTE_HOME = savedHome;
|
|
24
|
-
if (tmp) {
|
|
25
|
-
try { rmSync(tmp, { recursive: true, force: true }); } catch {}
|
|
26
|
-
tmp = undefined;
|
|
27
|
-
}
|
|
28
|
-
});
|
|
29
|
-
|
|
30
|
-
describe("resolveSpawnDeps", () => {
|
|
31
|
-
test("throws SpawnDepsError when there's no operator token", () => {
|
|
32
|
-
tmp = mkdtempSync(join(tmpdir(), "spawn-deps-empty-"));
|
|
33
|
-
process.env.PARACHUTE_HOME = tmp; // no operator.token inside
|
|
34
|
-
expect(() => resolveSpawnDeps()).toThrow(SpawnDepsError);
|
|
35
|
-
});
|
|
36
|
-
|
|
37
|
-
test("binds the claude binary (confined reads) but NOT the operator's ~/.claude", () => {
|
|
38
|
-
tmp = mkdtempSync(join(tmpdir(), "spawn-deps-"));
|
|
39
|
-
process.env.PARACHUTE_HOME = tmp;
|
|
40
|
-
writeFileSync(join(tmp, "operator.token"), "fake-operator-bearer");
|
|
41
|
-
const deps = resolveSpawnDeps();
|
|
42
|
-
// The agent's config/onboarding now lives in its own per-session HOME
|
|
43
|
-
// (seedAgentHome), so we no longer expose the operator's real config.
|
|
44
|
-
expect(deps.runtimeReadOnly).not.toContain(resolve(homedir(), ".claude.json"));
|
|
45
|
-
expect(deps.runtimeReadOnly).not.toContain(resolve(homedir(), ".claude"));
|
|
46
|
-
// The claude BINARY is still bound (needed under confined/scoped reads) when
|
|
47
|
-
// resolvable on PATH — and claudeBin is set to its absolute path.
|
|
48
|
-
const bin = Bun.which("claude");
|
|
49
|
-
if (bin) {
|
|
50
|
-
expect(deps.claudeBin).toBe(bin);
|
|
51
|
-
expect(deps.runtimeReadOnly).toContain(bin);
|
|
52
|
-
}
|
|
53
|
-
});
|
|
54
|
-
});
|
|
@@ -1,50 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Tests for same-origin terminal asset serving (`src/terminal-assets.ts`) — the
|
|
3
|
-
* fix for "xterm.js failed to load (CDN blocked?)". Proves the vendored xterm JS +
|
|
4
|
-
* CSS resolve from the installed packages and serve with the right content-type,
|
|
5
|
-
* and that an unknown asset name is a clean miss (the daemon 404s it).
|
|
6
|
-
*/
|
|
7
|
-
|
|
8
|
-
import { describe, test, expect } from "bun:test";
|
|
9
|
-
import { serveTerminalAsset, TERMINAL_ASSET_NAMES } from "./terminal-assets.ts";
|
|
10
|
-
|
|
11
|
-
describe("serveTerminalAsset", () => {
|
|
12
|
-
test("serves xterm.js as JavaScript with a real body", async () => {
|
|
13
|
-
const res = serveTerminalAsset("xterm.js");
|
|
14
|
-
expect(res).not.toBeNull();
|
|
15
|
-
expect(res!.status).toBe(200);
|
|
16
|
-
expect(res!.headers.get("content-type")).toContain("javascript");
|
|
17
|
-
const body = await res!.text();
|
|
18
|
-
// The UMD bundle defines the Terminal global — a non-trivial body proves the
|
|
19
|
-
// file resolved from the package (not an empty/placeholder response).
|
|
20
|
-
expect(body.length).toBeGreaterThan(1000);
|
|
21
|
-
expect(body).toContain("Terminal");
|
|
22
|
-
});
|
|
23
|
-
|
|
24
|
-
test("serves xterm.css as CSS", async () => {
|
|
25
|
-
const res = serveTerminalAsset("xterm.css");
|
|
26
|
-
expect(res!.status).toBe(200);
|
|
27
|
-
expect(res!.headers.get("content-type")).toContain("css");
|
|
28
|
-
expect((await res!.text()).length).toBeGreaterThan(100);
|
|
29
|
-
});
|
|
30
|
-
|
|
31
|
-
test("serves addon-fit.js", async () => {
|
|
32
|
-
const res = serveTerminalAsset("addon-fit.js");
|
|
33
|
-
expect(res!.status).toBe(200);
|
|
34
|
-
expect(res!.headers.get("content-type")).toContain("javascript");
|
|
35
|
-
});
|
|
36
|
-
|
|
37
|
-
test("sets an immutable cache header (version-pinned)", () => {
|
|
38
|
-
const res = serveTerminalAsset("xterm.js");
|
|
39
|
-
expect(res!.headers.get("cache-control")).toContain("immutable");
|
|
40
|
-
});
|
|
41
|
-
|
|
42
|
-
test("an unknown asset name → null (caller 404s)", () => {
|
|
43
|
-
expect(serveTerminalAsset("evil.js")).toBeNull();
|
|
44
|
-
expect(serveTerminalAsset("../../etc/passwd")).toBeNull();
|
|
45
|
-
});
|
|
46
|
-
|
|
47
|
-
test("the known names are the three xterm assets", () => {
|
|
48
|
-
expect(TERMINAL_ASSET_NAMES.sort()).toEqual(["addon-fit.js", "xterm.css", "xterm.js"]);
|
|
49
|
-
});
|
|
50
|
-
});
|
package/src/terminal.test.ts
DELETED
|
@@ -1,530 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* In-page terminal tests — the WS↔pty relay (`createTerminalWsHandlers`,
|
|
3
|
-
* `parseControlFrame`, src/terminal.ts) + the daemon-side terminal upgrade gate
|
|
4
|
-
* (`authorizeTerminalUpgrade`, src/daemon.ts).
|
|
5
|
-
*
|
|
6
|
-
* The relay tests drive the handler set against a FAKE `ServerWebSocket` and the
|
|
7
|
-
* injectable `spawnTerminal` seam — no real tmux, no real pty, no `Bun.serve`.
|
|
8
|
-
* The fake socket exposes a settable `getBufferedAmount()` so we can simulate
|
|
9
|
-
* the hub's send-buffer filling under a flood and assert the flow-control
|
|
10
|
-
* contract (the load-bearing item): a flood PARKS in the daemon-side coalesce
|
|
11
|
-
* queue, the socket is NOT closed, and output resumes (drains) once the client
|
|
12
|
-
* catches up. The hub's blunt 8 MiB cap therefore never has to fire.
|
|
13
|
-
*
|
|
14
|
-
* The auth tests call the pure `authorizeTerminalUpgrade` directly, using the
|
|
15
|
-
* same sentinel-token `mock.module("./hub-jwt.ts")` harness daemon-config-api.
|
|
16
|
-
* test.ts uses — accept paths run without a live hub/JWKS; the no-token reject
|
|
17
|
-
* still hits the real short-circuit.
|
|
18
|
-
*/
|
|
19
|
-
import { describe, test, expect, mock } from "bun:test";
|
|
20
|
-
import type { ServerWebSocket } from "bun";
|
|
21
|
-
|
|
22
|
-
// Sentinel tokens → fixed scope sets. Mirrors daemon-config-api.test.ts so this
|
|
23
|
-
// process-wide mock stays compatible. ADMIN_TOKEN carries agent:admin (==
|
|
24
|
-
// SCOPE_TERMINAL); READ_TOKEN is under-scoped; LEGACY_ADMIN_TOKEN carries the
|
|
25
|
-
// pre-rename channel:admin scope (dual-accept back-compat); anything else throws.
|
|
26
|
-
const ADMIN_TOKEN = "test-admin-token"; // agent:admin (== SCOPE_TERMINAL)
|
|
27
|
-
const READ_TOKEN = "test-read-token"; // agent:read only (insufficient)
|
|
28
|
-
const LEGACY_ADMIN_TOKEN = "test-legacy-admin-token"; // legacy channel:admin (still accepted)
|
|
29
|
-
import { HubJwtError, looksLikeJwt } from "@openparachute/scope-guard";
|
|
30
|
-
mock.module("./hub-jwt.ts", () => ({
|
|
31
|
-
AGENT_AUDIENCE: "agent",
|
|
32
|
-
CHANNEL_AUDIENCE: "channel", // deprecated alias kept for back-compat
|
|
33
|
-
async validateHubJwt(token: string) {
|
|
34
|
-
// New tokens carry aud:"agent" + agent:* scopes (the daemon mints/validates
|
|
35
|
-
// agent now). A legacy token carries aud:"channel" + channel:* scopes — the
|
|
36
|
-
// resource server dual-accepts both during the rename window.
|
|
37
|
-
const base = { sub: "test", aud: "agent", jti: undefined, clientId: undefined, vaultScope: undefined };
|
|
38
|
-
if (token === ADMIN_TOKEN) return { ...base, scopes: ["agent:read", "agent:send", "agent:admin"] };
|
|
39
|
-
if (token === READ_TOKEN) return { ...base, scopes: ["agent:read"] };
|
|
40
|
-
// Pre-rename token: legacy audience + legacy channel:* scopes. requireScope's
|
|
41
|
-
// dual-accept (hasScope) must still authorize agent:admin from channel:admin.
|
|
42
|
-
if (token === LEGACY_ADMIN_TOKEN)
|
|
43
|
-
return { ...base, aud: "channel", scopes: ["channel:read", "channel:send", "channel:admin"] };
|
|
44
|
-
throw new HubJwtError("issuer", "invalid token");
|
|
45
|
-
},
|
|
46
|
-
HubJwtError,
|
|
47
|
-
looksLikeJwt,
|
|
48
|
-
resetJwksCache() {},
|
|
49
|
-
resetRevocationCache() {},
|
|
50
|
-
}));
|
|
51
|
-
|
|
52
|
-
import {
|
|
53
|
-
createTerminalWsHandlers,
|
|
54
|
-
tmuxAttachEnv,
|
|
55
|
-
parseControlFrame,
|
|
56
|
-
HUB_WS_CAP_BYTES,
|
|
57
|
-
PAUSE_FRAC,
|
|
58
|
-
RESUME_FRAC,
|
|
59
|
-
type TerminalWsData,
|
|
60
|
-
type SpawnTerminalFn,
|
|
61
|
-
type SpawnedTerminal,
|
|
62
|
-
type BunTerminal,
|
|
63
|
-
} from "./terminal.ts";
|
|
64
|
-
import { authorizeTerminalUpgrade } from "./daemon.ts";
|
|
65
|
-
import type { Channel } from "./registry.ts";
|
|
66
|
-
|
|
67
|
-
// ===========================================================================
|
|
68
|
-
// Fakes — a recording pty + a controllable ServerWebSocket.
|
|
69
|
-
// ===========================================================================
|
|
70
|
-
|
|
71
|
-
/** A fake pty that records writes/resizes and lets the test push pty output. */
|
|
72
|
-
class FakePty implements BunTerminal {
|
|
73
|
-
writes: Uint8Array[] = [];
|
|
74
|
-
writeStrings: string[] = [];
|
|
75
|
-
resizes: Array<{ cols: number; rows: number }> = [];
|
|
76
|
-
closed = false;
|
|
77
|
-
/** Captured callbacks from the spawn opts — the test drives pty output via these. */
|
|
78
|
-
onData!: (bytes: Uint8Array) => void;
|
|
79
|
-
onExit!: () => void;
|
|
80
|
-
|
|
81
|
-
write(data: string | ArrayBufferView | ArrayBufferLike): number {
|
|
82
|
-
if (typeof data === "string") {
|
|
83
|
-
this.writeStrings.push(data);
|
|
84
|
-
return data.length;
|
|
85
|
-
}
|
|
86
|
-
const u8 = data instanceof Uint8Array ? data : new Uint8Array(data as ArrayBufferLike);
|
|
87
|
-
this.writes.push(u8);
|
|
88
|
-
return u8.byteLength;
|
|
89
|
-
}
|
|
90
|
-
resize(cols: number, rows: number): void {
|
|
91
|
-
this.resizes.push({ cols, rows });
|
|
92
|
-
}
|
|
93
|
-
close(): void {
|
|
94
|
-
this.closed = true;
|
|
95
|
-
}
|
|
96
|
-
}
|
|
97
|
-
|
|
98
|
-
interface FakeProc {
|
|
99
|
-
readonly exited: Promise<number>;
|
|
100
|
-
killed: boolean;
|
|
101
|
-
kill(signal?: number | string): void;
|
|
102
|
-
}
|
|
103
|
-
|
|
104
|
-
/** A fake ServerWebSocket: records sends/close, settable buffered depth. */
|
|
105
|
-
class FakeWs {
|
|
106
|
-
data: TerminalWsData;
|
|
107
|
-
sent: Uint8Array[] = [];
|
|
108
|
-
closeCalls: Array<{ code?: number; reason?: string }> = [];
|
|
109
|
-
/** Test sets this to simulate the hub send-buffer depth. */
|
|
110
|
-
buffered = 0;
|
|
111
|
-
|
|
112
|
-
constructor(data: TerminalWsData) {
|
|
113
|
-
this.data = data;
|
|
114
|
-
}
|
|
115
|
-
send(bytes: Uint8Array | string): number {
|
|
116
|
-
if (typeof bytes === "string") {
|
|
117
|
-
this.sent.push(new TextEncoder().encode(bytes));
|
|
118
|
-
return bytes.length;
|
|
119
|
-
}
|
|
120
|
-
this.sent.push(bytes);
|
|
121
|
-
return bytes.byteLength;
|
|
122
|
-
}
|
|
123
|
-
getBufferedAmount(): number {
|
|
124
|
-
return this.buffered;
|
|
125
|
-
}
|
|
126
|
-
close(code?: number, reason?: string): void {
|
|
127
|
-
this.closeCalls.push({ code, reason });
|
|
128
|
-
}
|
|
129
|
-
get totalSentBytes(): number {
|
|
130
|
-
return this.sent.reduce((n, b) => n + b.byteLength, 0);
|
|
131
|
-
}
|
|
132
|
-
}
|
|
133
|
-
|
|
134
|
-
/** Build a fake-pty spawn seam; returns the seam + a handle on the created pty. */
|
|
135
|
-
function makeSpawn(): { spawn: SpawnTerminalFn; ptyRef: { current?: FakePty }; procRef: { current?: FakeProc } } {
|
|
136
|
-
const ptyRef: { current?: FakePty } = {};
|
|
137
|
-
const procRef: { current?: FakeProc } = {};
|
|
138
|
-
const spawn: SpawnTerminalFn = (_session, opts) => {
|
|
139
|
-
const pty = new FakePty();
|
|
140
|
-
pty.onData = opts.onData;
|
|
141
|
-
pty.onExit = opts.onExit;
|
|
142
|
-
const proc: FakeProc = { exited: Promise.resolve(0), killed: false, kill() { this.killed = true; } };
|
|
143
|
-
ptyRef.current = pty;
|
|
144
|
-
procRef.current = proc;
|
|
145
|
-
return { terminal: pty, proc } as unknown as SpawnedTerminal;
|
|
146
|
-
};
|
|
147
|
-
return { spawn, ptyRef, procRef };
|
|
148
|
-
}
|
|
149
|
-
|
|
150
|
-
function wsData(over: Partial<TerminalWsData> = {}): TerminalWsData {
|
|
151
|
-
return { session: "eng-agent", channel: "eng", cols: 80, rows: 24, ...over };
|
|
152
|
-
}
|
|
153
|
-
|
|
154
|
-
/** Open a handler set + fake ws + fake pty, returning the lot for a test. */
|
|
155
|
-
function setup(opts: { capBytes?: number } = {}) {
|
|
156
|
-
const { spawn, ptyRef, procRef } = makeSpawn();
|
|
157
|
-
const warnings: string[] = [];
|
|
158
|
-
const handlers = createTerminalWsHandlers({
|
|
159
|
-
spawnTerminal: spawn,
|
|
160
|
-
capBytes: opts.capBytes,
|
|
161
|
-
logger: { warn: (...a: unknown[]) => warnings.push(a.map(String).join(" ")) },
|
|
162
|
-
});
|
|
163
|
-
const ws = new FakeWs(wsData());
|
|
164
|
-
handlers.open(ws as unknown as ServerWebSocket<TerminalWsData>);
|
|
165
|
-
return { handlers, ws, pty: () => ptyRef.current!, proc: () => procRef.current!, warnings };
|
|
166
|
-
}
|
|
167
|
-
|
|
168
|
-
// Cast helper — drive a handler with the fake ws.
|
|
169
|
-
const asWs = (ws: FakeWs) => ws as unknown as ServerWebSocket<TerminalWsData>;
|
|
170
|
-
|
|
171
|
-
// ===========================================================================
|
|
172
|
-
// parseControlFrame — unit cases
|
|
173
|
-
// ===========================================================================
|
|
174
|
-
describe("parseControlFrame", () => {
|
|
175
|
-
test("a valid resize frame parses + clamps", () => {
|
|
176
|
-
expect(parseControlFrame(JSON.stringify({ type: "resize", cols: 120, rows: 40 }))).toEqual({
|
|
177
|
-
type: "resize",
|
|
178
|
-
cols: 120,
|
|
179
|
-
rows: 40,
|
|
180
|
-
});
|
|
181
|
-
});
|
|
182
|
-
|
|
183
|
-
test("out-of-range dims clamp to [1, 9999]", () => {
|
|
184
|
-
expect(parseControlFrame(JSON.stringify({ type: "resize", cols: 0, rows: 999999 }))).toEqual({
|
|
185
|
-
type: "resize",
|
|
186
|
-
cols: 1,
|
|
187
|
-
rows: 9999,
|
|
188
|
-
});
|
|
189
|
-
});
|
|
190
|
-
|
|
191
|
-
test("malformed JSON → null (forwarded as input by the caller)", () => {
|
|
192
|
-
expect(parseControlFrame("{not json")).toBeNull();
|
|
193
|
-
});
|
|
194
|
-
|
|
195
|
-
test("a non-resize control type → null", () => {
|
|
196
|
-
expect(parseControlFrame(JSON.stringify({ type: "evil", cmd: "rm -rf /" }))).toBeNull();
|
|
197
|
-
});
|
|
198
|
-
|
|
199
|
-
test("resize with non-finite dims → null", () => {
|
|
200
|
-
expect(parseControlFrame(JSON.stringify({ type: "resize", cols: "80", rows: NaN }))).toBeNull();
|
|
201
|
-
expect(parseControlFrame(JSON.stringify({ type: "resize" }))).toBeNull();
|
|
202
|
-
});
|
|
203
|
-
|
|
204
|
-
test("a bare JSON scalar / array (not an object) → null", () => {
|
|
205
|
-
expect(parseControlFrame("42")).toBeNull();
|
|
206
|
-
expect(parseControlFrame("null")).toBeNull();
|
|
207
|
-
expect(parseControlFrame("[1,2,3]")).toBeNull();
|
|
208
|
-
});
|
|
209
|
-
});
|
|
210
|
-
|
|
211
|
-
// ===========================================================================
|
|
212
|
-
// Relay — both directions
|
|
213
|
-
// ===========================================================================
|
|
214
|
-
describe("relay — pty ↔ ws", () => {
|
|
215
|
-
test("pty output → ws.send (binary), bytes preserved", () => {
|
|
216
|
-
const { ws, pty } = setup();
|
|
217
|
-
const chunk = new Uint8Array([0x68, 0x69]); // "hi"
|
|
218
|
-
pty().onData(chunk);
|
|
219
|
-
expect(ws.sent).toHaveLength(1);
|
|
220
|
-
expect(Array.from(ws.sent[0]!)).toEqual([0x68, 0x69]);
|
|
221
|
-
});
|
|
222
|
-
|
|
223
|
-
test("ws binary message → terminal.write (keystrokes to the pty)", () => {
|
|
224
|
-
const { handlers, ws, pty } = setup();
|
|
225
|
-
const keys = Buffer.from("ls\n");
|
|
226
|
-
handlers.message(asWs(ws), keys);
|
|
227
|
-
expect(pty().writes).toHaveLength(1);
|
|
228
|
-
expect(Buffer.from(pty().writes[0]!).toString()).toBe("ls\n");
|
|
229
|
-
});
|
|
230
|
-
|
|
231
|
-
test("a text frame that ISN'T a control frame is forwarded to the pty as input (fail-safe)", () => {
|
|
232
|
-
const { handlers, ws, pty } = setup();
|
|
233
|
-
handlers.message(asWs(ws), "plain text");
|
|
234
|
-
expect(pty().writeStrings).toEqual(["plain text"]);
|
|
235
|
-
expect(pty().resizes).toHaveLength(0);
|
|
236
|
-
});
|
|
237
|
-
});
|
|
238
|
-
|
|
239
|
-
// ===========================================================================
|
|
240
|
-
// Control frame — resize
|
|
241
|
-
// ===========================================================================
|
|
242
|
-
describe("control frame — resize", () => {
|
|
243
|
-
test("a {type:resize} text frame → terminal.resize(cols, rows), NOT written as input", () => {
|
|
244
|
-
const { handlers, ws, pty } = setup();
|
|
245
|
-
handlers.message(asWs(ws), JSON.stringify({ type: "resize", cols: 100, rows: 30 }));
|
|
246
|
-
expect(pty().resizes).toEqual([{ cols: 100, rows: 30 }]);
|
|
247
|
-
expect(pty().writes).toHaveLength(0);
|
|
248
|
-
expect(pty().writeStrings).toHaveLength(0);
|
|
249
|
-
});
|
|
250
|
-
});
|
|
251
|
-
|
|
252
|
-
// ===========================================================================
|
|
253
|
-
// THE load-bearing test — backpressure flood stays alive
|
|
254
|
-
// ===========================================================================
|
|
255
|
-
describe("backpressure — a flood parks in the queue, never closes the socket", () => {
|
|
256
|
-
// A tiny cap so the test floods it cheaply. pause @ 50% (500), resume @ 25% (250).
|
|
257
|
-
const CAP = 1000;
|
|
258
|
-
|
|
259
|
-
test("a huge burst while the socket buffer is high → paused + coalesced, socket NOT closed, hub cap never approached", () => {
|
|
260
|
-
const { ws, pty, proc } = setup({ capBytes: CAP });
|
|
261
|
-
|
|
262
|
-
// Simulate the hub's send buffer already full (over the pause mark) and never
|
|
263
|
-
// draining during the burst — exactly the "a build log floods xterm" case.
|
|
264
|
-
ws.buffered = CAP; // 1000 ≥ pauseAt(500)
|
|
265
|
-
|
|
266
|
-
// First chunk goes out, then the post-send buffered check trips pause.
|
|
267
|
-
pty().onData(new Uint8Array(64));
|
|
268
|
-
// Subsequent chunks while paused MUST queue, not send.
|
|
269
|
-
const FLOOD_CHUNK = new Uint8Array(64 * 1024); // 64 KiB each
|
|
270
|
-
for (let i = 0; i < 200; i++) pty().onData(FLOOD_CHUNK); // ~12.8 MiB of pty output
|
|
271
|
-
|
|
272
|
-
// The socket is ALIVE — no close (no 1011, no 1013), pty alive, proc alive.
|
|
273
|
-
expect(ws.closeCalls).toHaveLength(0);
|
|
274
|
-
expect(pty().closed).toBe(false);
|
|
275
|
-
expect(proc().killed).toBe(false);
|
|
276
|
-
|
|
277
|
-
// Our OWN buffered bytes (what the hub sees) never grew past the one chunk we
|
|
278
|
-
// sent before pausing — the flood parked daemon-side, NOT in the hub buffer.
|
|
279
|
-
// i.e. we never let the socket approach the 8 MiB hub cap.
|
|
280
|
-
expect(ws.totalSentBytes).toBeLessThanOrEqual(64);
|
|
281
|
-
expect(ws.totalSentBytes).toBeLessThan(HUB_WS_CAP_BYTES);
|
|
282
|
-
});
|
|
283
|
-
|
|
284
|
-
test("when the client drains, the queued flood is eventually delivered (resume)", () => {
|
|
285
|
-
const { handlers, ws, pty } = setup({ capBytes: CAP });
|
|
286
|
-
|
|
287
|
-
ws.buffered = CAP; // over pause mark
|
|
288
|
-
pty().onData(new Uint8Array(64)); // sent → then pause
|
|
289
|
-
// Queue a bounded flood (well under MAX_QUEUE_BYTES so it parks, not closes).
|
|
290
|
-
const N = 50;
|
|
291
|
-
const CHUNK = 64; // bytes
|
|
292
|
-
for (let i = 0; i < N; i++) pty().onData(new Uint8Array(CHUNK).fill(i));
|
|
293
|
-
expect(ws.closeCalls).toHaveLength(0);
|
|
294
|
-
const sentBeforeDrain = ws.sent.length;
|
|
295
|
-
|
|
296
|
-
// Client catches up: buffer empties. Bun fires drain repeatedly as it clears;
|
|
297
|
-
// drain flushes the queue while buffered stays under the pause mark.
|
|
298
|
-
ws.buffered = 0;
|
|
299
|
-
// A few drain cycles to flush the whole queue (flushQueue drains until empty
|
|
300
|
-
// or the pause mark; with buffered=0 it empties in one pass, but loop to be safe).
|
|
301
|
-
for (let i = 0; i < 5 && ws.sent.length < sentBeforeDrain + N; i++) {
|
|
302
|
-
handlers.drain(asWs(ws));
|
|
303
|
-
}
|
|
304
|
-
|
|
305
|
-
// Everything queued got delivered — the flood was buffered, not dropped.
|
|
306
|
-
expect(ws.sent.length).toBe(sentBeforeDrain + N);
|
|
307
|
-
expect(ws.closeCalls).toHaveLength(0);
|
|
308
|
-
// And the very last queued chunk arrived intact (content preserved through the queue).
|
|
309
|
-
const last = ws.sent[ws.sent.length - 1]!;
|
|
310
|
-
expect(last.byteLength).toBe(CHUNK);
|
|
311
|
-
expect(last[0]).toBe(N - 1);
|
|
312
|
-
});
|
|
313
|
-
});
|
|
314
|
-
|
|
315
|
-
// ===========================================================================
|
|
316
|
-
// Hysteresis — pause @ 50%, resume only under 25%
|
|
317
|
-
// ===========================================================================
|
|
318
|
-
describe("hysteresis — pause @ PAUSE_FRAC, resume only under RESUME_FRAC", () => {
|
|
319
|
-
const CAP = 1000; // pauseAt = 500, resumeAt = 250
|
|
320
|
-
|
|
321
|
-
test("constants are 0.5 / 0.25 (the documented margins)", () => {
|
|
322
|
-
expect(PAUSE_FRAC).toBe(0.5);
|
|
323
|
-
expect(RESUME_FRAC).toBe(0.25);
|
|
324
|
-
});
|
|
325
|
-
|
|
326
|
-
test("does NOT pause while buffered stays under the pause mark", () => {
|
|
327
|
-
const { ws, pty } = setup({ capBytes: CAP });
|
|
328
|
-
ws.buffered = 400; // < pauseAt(500)
|
|
329
|
-
pty().onData(new Uint8Array(10));
|
|
330
|
-
pty().onData(new Uint8Array(10)); // still forwarded, not queued
|
|
331
|
-
expect(ws.sent).toHaveLength(2);
|
|
332
|
-
expect(ws.closeCalls).toHaveLength(0);
|
|
333
|
-
});
|
|
334
|
-
|
|
335
|
-
test("pauses at the 50% mark; a drain to BETWEEN resume(25%) and pause(50%) does NOT resume", () => {
|
|
336
|
-
const { handlers, ws, pty } = setup({ capBytes: CAP });
|
|
337
|
-
ws.buffered = 500; // == pauseAt → pause after the send
|
|
338
|
-
pty().onData(new Uint8Array(10));
|
|
339
|
-
// Now paused: a new pty chunk queues rather than sends.
|
|
340
|
-
pty().onData(new Uint8Array(10));
|
|
341
|
-
expect(ws.sent).toHaveLength(1); // only the first went out
|
|
342
|
-
|
|
343
|
-
// Drain partway — buffer at 300 (still > resumeAt 250). Queue flushes only if
|
|
344
|
-
// under the pause mark (300 < 500 so it flushes), but we stay PAUSED for live
|
|
345
|
-
// forwarding because we're not yet under the resume mark.
|
|
346
|
-
ws.buffered = 300;
|
|
347
|
-
handlers.drain(asWs(ws));
|
|
348
|
-
// A fresh live chunk should STILL queue (paused), proving no premature resume.
|
|
349
|
-
const sentAfterDrain = ws.sent.length;
|
|
350
|
-
pty().onData(new Uint8Array(10));
|
|
351
|
-
expect(ws.sent.length).toBe(sentAfterDrain); // still paused → queued, not sent
|
|
352
|
-
});
|
|
353
|
-
|
|
354
|
-
test("resumes only once buffered falls under the 25% mark (and the queue is empty)", () => {
|
|
355
|
-
const { handlers, ws, pty } = setup({ capBytes: CAP });
|
|
356
|
-
ws.buffered = 600; // pause
|
|
357
|
-
pty().onData(new Uint8Array(10));
|
|
358
|
-
pty().onData(new Uint8Array(10)); // queued
|
|
359
|
-
|
|
360
|
-
// Drain fully under the resume mark → drain flushes the queue + resumes.
|
|
361
|
-
ws.buffered = 100; // < resumeAt(250)
|
|
362
|
-
handlers.drain(asWs(ws));
|
|
363
|
-
// Now resumed: a new live chunk forwards immediately.
|
|
364
|
-
const before = ws.sent.length;
|
|
365
|
-
pty().onData(new Uint8Array(10));
|
|
366
|
-
expect(ws.sent.length).toBe(before + 1);
|
|
367
|
-
});
|
|
368
|
-
});
|
|
369
|
-
|
|
370
|
-
// ===========================================================================
|
|
371
|
-
// 1013 — a hopelessly-stuck client is shed
|
|
372
|
-
// ===========================================================================
|
|
373
|
-
describe("stuck client — closed 1013 when the queue blows past MAX_QUEUE_BYTES", () => {
|
|
374
|
-
// Keep the hub cap tiny so we pause immediately; the queue cap (16 MiB) is the
|
|
375
|
-
// line under test. Flood past 16 MiB while the socket never drains.
|
|
376
|
-
const CAP = 1000;
|
|
377
|
-
|
|
378
|
-
test(">16 MiB queued while the socket stays full → close(1013), pty + viewer torn down", () => {
|
|
379
|
-
const { ws, pty, proc, warnings } = setup({ capBytes: CAP });
|
|
380
|
-
ws.buffered = CAP; // perpetually over the pause mark → everything queues
|
|
381
|
-
pty().onData(new Uint8Array(64)); // first send trips pause
|
|
382
|
-
|
|
383
|
-
// 17 MiB in 1 MiB chunks — crosses the 16 MiB MAX_QUEUE_BYTES line.
|
|
384
|
-
const MiB = new Uint8Array(1024 * 1024);
|
|
385
|
-
for (let i = 0; i < 17; i++) pty().onData(MiB);
|
|
386
|
-
|
|
387
|
-
const closed1013 = ws.closeCalls.find((c) => c.code === 1013);
|
|
388
|
-
expect(closed1013).toBeDefined();
|
|
389
|
-
expect(pty().closed).toBe(true); // pty closed
|
|
390
|
-
expect(proc().killed).toBe(true); // viewer process killed
|
|
391
|
-
expect(warnings.some((w) => w.includes("too far behind") || w.includes("too slow"))).toBe(true);
|
|
392
|
-
});
|
|
393
|
-
});
|
|
394
|
-
|
|
395
|
-
// ===========================================================================
|
|
396
|
-
// Lifecycle — pty exit + client close tear down the viewer (session lives on)
|
|
397
|
-
// ===========================================================================
|
|
398
|
-
describe("lifecycle — teardown", () => {
|
|
399
|
-
test("pty exit closes the socket with a clean 1000 (session ended)", () => {
|
|
400
|
-
const { ws, pty, proc } = setup();
|
|
401
|
-
pty().onExit();
|
|
402
|
-
expect(ws.closeCalls.some((c) => c.code === 1000)).toBe(true);
|
|
403
|
-
expect(pty().closed).toBe(true);
|
|
404
|
-
expect(proc().killed).toBe(true);
|
|
405
|
-
});
|
|
406
|
-
|
|
407
|
-
test("client close releases the pty + kills the viewer proc (idempotent)", () => {
|
|
408
|
-
const { handlers, ws, pty, proc } = setup();
|
|
409
|
-
handlers.close(asWs(ws));
|
|
410
|
-
expect(pty().closed).toBe(true);
|
|
411
|
-
expect(proc().killed).toBe(true);
|
|
412
|
-
// Idempotent — a second close is a no-op (no throw, no double behavior change).
|
|
413
|
-
expect(() => handlers.close(asWs(ws))).not.toThrow();
|
|
414
|
-
});
|
|
415
|
-
});
|
|
416
|
-
|
|
417
|
-
// ===========================================================================
|
|
418
|
-
// tmuxAttachEnv — TERM wiring for the `tmux attach` viewer
|
|
419
|
-
// ===========================================================================
|
|
420
|
-
describe("tmuxAttachEnv — forces a real TERM so `tmux attach` finds terminfo", () => {
|
|
421
|
-
test("sets TERM=xterm-256color (else tmux aborts 'does not support clear')", () => {
|
|
422
|
-
expect(tmuxAttachEnv({ PATH: "/usr/bin" }).TERM).toBe("xterm-256color");
|
|
423
|
-
});
|
|
424
|
-
test("preserves the inherited env + overrides a stale/empty TERM", () => {
|
|
425
|
-
const env = tmuxAttachEnv({ PATH: "/usr/bin", HOME: "/home/op", TERM: "" });
|
|
426
|
-
expect(env.PATH).toBe("/usr/bin");
|
|
427
|
-
expect(env.HOME).toBe("/home/op");
|
|
428
|
-
expect(env.TERM).toBe("xterm-256color"); // empty/missing TERM is the daemon's reality
|
|
429
|
-
});
|
|
430
|
-
});
|
|
431
|
-
|
|
432
|
-
// ===========================================================================
|
|
433
|
-
// Daemon-side terminal auth — authorizeTerminalUpgrade (pure fn)
|
|
434
|
-
// ===========================================================================
|
|
435
|
-
describe("authorizeTerminalUpgrade — operator-gated agent:admin", () => {
|
|
436
|
-
function channels(names: string[] = ["eng"]): Map<string, Channel> {
|
|
437
|
-
const m = new Map<string, Channel>();
|
|
438
|
-
for (const name of names) {
|
|
439
|
-
m.set(name, { name, transport: {} as Channel["transport"], entry: { name, transport: "vault" } });
|
|
440
|
-
}
|
|
441
|
-
return m;
|
|
442
|
-
}
|
|
443
|
-
function req(query = ""): { req: Request; url: URL } {
|
|
444
|
-
const url = new URL(`http://127.0.0.1/terminal/eng${query}`);
|
|
445
|
-
return { req: new Request(url, { headers: { upgrade: "websocket" } }), url };
|
|
446
|
-
}
|
|
447
|
-
|
|
448
|
-
test("missing token → reject (401), no session leaked", async () => {
|
|
449
|
-
const { req: r, url } = req();
|
|
450
|
-
const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
|
|
451
|
-
expect(d.ok).toBe(false);
|
|
452
|
-
if (!d.ok) expect(d.response.status).toBe(401);
|
|
453
|
-
});
|
|
454
|
-
|
|
455
|
-
test("bad/expired token → reject (401)", async () => {
|
|
456
|
-
const { req: r, url } = req("?token=garbage-not-a-real-jwt");
|
|
457
|
-
const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
|
|
458
|
-
expect(d.ok).toBe(false);
|
|
459
|
-
if (!d.ok) expect(d.response.status).toBe(401);
|
|
460
|
-
});
|
|
461
|
-
|
|
462
|
-
test("under-scoped token (agent:read, not agent:admin) → reject (403)", async () => {
|
|
463
|
-
const { req: r, url } = req(`?token=${READ_TOKEN}`);
|
|
464
|
-
const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
|
|
465
|
-
expect(d.ok).toBe(false);
|
|
466
|
-
if (!d.ok) expect(d.response.status).toBe(403);
|
|
467
|
-
});
|
|
468
|
-
|
|
469
|
-
test("valid agent:admin token → ok, with the right tmux session + geometry", async () => {
|
|
470
|
-
const { req: r, url } = req(`?token=${ADMIN_TOKEN}&cols=120&rows=40`);
|
|
471
|
-
const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
|
|
472
|
-
expect(d.ok).toBe(true);
|
|
473
|
-
if (d.ok) {
|
|
474
|
-
expect(d.channel).toBe("eng");
|
|
475
|
-
expect(d.session).toBe("eng-agent"); // <name>-agent (launch-session.sh:38)
|
|
476
|
-
expect(d.cols).toBe(120);
|
|
477
|
-
expect(d.rows).toBe(40);
|
|
478
|
-
}
|
|
479
|
-
});
|
|
480
|
-
|
|
481
|
-
test("legacy channel:admin token still authorizes (dual-accept back-compat)", async () => {
|
|
482
|
-
// A token minted before the channel→agent rename carries channel:admin (and
|
|
483
|
-
// aud:"channel"). requireScope's dual-accept (hasScope) must still authorize
|
|
484
|
-
// the agent:admin-gated terminal until live tokens are re-minted.
|
|
485
|
-
const { req: r, url } = req(`?token=${LEGACY_ADMIN_TOKEN}&cols=100&rows=30`);
|
|
486
|
-
const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
|
|
487
|
-
expect(d.ok).toBe(true);
|
|
488
|
-
if (d.ok) {
|
|
489
|
-
expect(d.session).toBe("eng-agent");
|
|
490
|
-
expect(d.cols).toBe(100);
|
|
491
|
-
expect(d.rows).toBe(30);
|
|
492
|
-
}
|
|
493
|
-
});
|
|
494
|
-
|
|
495
|
-
test("geometry defaults (80×24) when cols/rows absent or out-of-range", async () => {
|
|
496
|
-
const { req: r, url } = req(`?token=${ADMIN_TOKEN}&cols=0&rows=abc`);
|
|
497
|
-
const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
|
|
498
|
-
expect(d.ok).toBe(true);
|
|
499
|
-
if (d.ok) {
|
|
500
|
-
expect(d.cols).toBe(80);
|
|
501
|
-
expect(d.rows).toBe(24);
|
|
502
|
-
}
|
|
503
|
-
});
|
|
504
|
-
|
|
505
|
-
test("an agent name that isn't a configured channel is ACCEPTED → its session (agents have own names)", async () => {
|
|
506
|
-
// The terminal attaches to an AGENT (tmux <name>-agent), not a channel — so a
|
|
507
|
-
// valid-slug name that isn't in the channel map is accepted (a non-existent
|
|
508
|
-
// session just fails to attach downstream with a clean 1000, no 404 here).
|
|
509
|
-
const url = new URL(`http://127.0.0.1/terminal/weaver?token=${ADMIN_TOKEN}`);
|
|
510
|
-
const r = new Request(url, { headers: { upgrade: "websocket" } });
|
|
511
|
-
const d = await authorizeTerminalUpgrade(r, url, channels(["eng"]), "weaver");
|
|
512
|
-
expect(d.ok).toBe(true);
|
|
513
|
-
if (d.ok) expect(d.session).toBe("weaver-agent");
|
|
514
|
-
});
|
|
515
|
-
|
|
516
|
-
test("a path-traversal-shaped name is rejected by the slug guard → 400 (no escape into tmux -t)", async () => {
|
|
517
|
-
const weird = "../../etc";
|
|
518
|
-
const url = new URL(`http://127.0.0.1/terminal/${encodeURIComponent(weird)}?token=${ADMIN_TOKEN}`);
|
|
519
|
-
const r = new Request(url, { headers: { upgrade: "websocket" } });
|
|
520
|
-
const d = await authorizeTerminalUpgrade(r, url, channels(["eng"]), weird);
|
|
521
|
-
expect(d.ok).toBe(false);
|
|
522
|
-
if (!d.ok) expect(d.response.status).toBe(400);
|
|
523
|
-
});
|
|
524
|
-
});
|
|
525
|
-
|
|
526
|
-
// Touch HUB_WS_CAP_BYTES so the import is load-bearing if the relationship to the
|
|
527
|
-
// hub cap ever drifts (the constant mirrors the hub's DEFAULT_MAX_BUFFERED_BYTES).
|
|
528
|
-
test("HUB_WS_CAP_BYTES mirrors the hub's 8 MiB cap", () => {
|
|
529
|
-
expect(HUB_WS_CAP_BYTES).toBe(8 * 1024 * 1024);
|
|
530
|
-
});
|