Fix terminal issue
Build and Push Spoon Images / quality (push) Successful in 1m40s
Build and Push Spoon Images / build-images (push) Successful in 3m53s

This commit is contained in:
Gabriel Brown
2026-06-25 00:30:11 -04:00
parent 3f1fee4e44
commit b09295570d
2 changed files with 116 additions and 72 deletions
+5
View File
@@ -1,3 +1,4 @@
import { mkdir } from 'node:fs/promises';
import path from 'node:path';
import type { Readable } from 'node:stream';
import { execa } from 'execa';
@@ -350,6 +351,10 @@ export const ensureUserContainer = async (args: {
{ reject: false, stdin: 'ignore' },
);
if (inspect.exitCode === 0 && inspect.stdout.trim() === 'true') return name;
// The box mounts the per-user home, but it's created before the thread's clone
// populates it — ensure it exists first, since podman (unlike docker) refuses to
// bind-mount a missing source directory (statfs: no such file or directory).
await mkdir(args.workdir, { recursive: true });
// Not running: remove any stale container, then start fresh.
await execa(containerRuntime(), ['rm', '-f', name], { reject: false });
await execa(
+106 -67
View File
@@ -1,7 +1,7 @@
import { spawn } from 'node:child_process';
import type { ChildProcessWithoutNullStreams } from 'node:child_process';
import type { Server } from 'node:http';
import type { Duplex } from 'node:stream';
import type { WebSocket } from 'ws';
import Docker from 'dockerode';
import { WebSocketServer } from 'ws';
import { env } from './env';
@@ -9,7 +9,14 @@ import { verifyTerminalToken } from './terminal-token';
import { acquireUserBox, releaseUserBox } from './user-container';
import { getTerminalWorkspace } from './worker';
const docker = new Docker();
const clampDimension = (value: unknown) => {
const n = Math.trunc(Number(value));
if (!Number.isFinite(n)) return undefined;
return Math.min(Math.max(n, 1), 1000);
};
// Single-quote a string for a POSIX shell.
const shellQuote = (value: string) => `'${value.replaceAll("'", `'\\''`)}'`;
const bridge = async (ws: WebSocket, jobId: string) => {
const workspace = getTerminalWorkspace(jobId);
@@ -18,6 +25,58 @@ const bridge = async (ws: WebSocket, jobId: string) => {
return;
}
// bun can't load node-pty (native ABI mismatch) and dockerode can't attach to
// podman, so we drive the runtime CLI (`<runtime> exec -i`) and allocate the PTY
// *inside* the container with `script`, bridging the plain pipes to the socket.
//
// Register the message handler immediately and buffer input/size until the exec
// is ready (acquiring the box can take seconds on first connect), so the initial
// resize and early keystrokes aren't dropped.
const procHolder: { current?: ChildProcessWithoutNullStreams } = {};
const pendingInput: Buffer[] = [];
let cols = 80;
let rows = 24;
ws.on('message', (data: Buffer, isBinary: boolean) => {
if (!isBinary) {
// Text frames are control messages (resize); anything else is raw input.
try {
const message = JSON.parse(data.toString('utf8')) as {
type?: string;
cols?: number;
rows?: number;
};
if (message.type === 'resize') {
const c = clampDimension(message.cols);
const r = clampDimension(message.rows);
if (c && r) {
cols = c;
rows = r;
}
return;
}
} catch {
// fall through: treat as raw input
}
}
if (procHolder.current) procHolder.current.stdin.write(data);
else pendingInput.push(data);
});
let acquired = false;
let released = false;
// Read through a function so TS doesn't narrow `released` to a constant — the
// cleanup handler flips it asynchronously when the socket closes.
const isReleased = () => released;
const cleanup = () => {
if (released) return;
released = true;
procHolder.current?.kill();
if (acquired) releaseUserBox(workspace.username);
};
ws.on('close', cleanup);
ws.on('error', cleanup);
// Hold the per-user box open while this terminal is connected; the agent and
// the terminal share the exact same container (Phase 2).
let boxName: string;
@@ -27,6 +86,7 @@ const bridge = async (ws: WebSocket, jobId: string) => {
workdir: workspace.workdir,
containerHome: workspace.containerHome,
});
acquired = true;
} catch (error) {
ws.close(
1011,
@@ -35,79 +95,58 @@ const bridge = async (ws: WebSocket, jobId: string) => {
return;
}
let stream: Duplex | undefined;
let exec: Docker.Exec | undefined;
try {
exec = await docker.getContainer(boxName).exec({
// Reattach a persistent tmux session across reconnects when tmux is
// available; otherwise fall back to a plain login shell.
Cmd: [
if (isReleased()) return; // client disconnected during startup; cleanup ran
// Reattach a persistent tmux session across reconnects when available, else a
// plain login shell. `stty` sizes the PTY to the client's viewport up front.
const launcher =
`stty rows ${rows} cols ${cols} 2>/dev/null; ` +
// Reattach a persistent tmux session when tmux is present; otherwise fall back
// to an interactive login shell (`-i` so it prints a prompt and line-edits).
// Check with `command -v` rather than `exec tmux || …`: a failed `exec` makes a
// non-interactive shell exit before the `||`, so the fallback never runs.
'if command -v tmux >/dev/null 2>&1; then exec tmux new-session -A -s spoon; ' +
'else exec bash -il; fi';
const envFlags = [
'-e',
'TERM=xterm-256color',
'-e',
`HOME=${workspace.containerHome}`,
...workspace.secrets.flatMap((s) => ['-e', `${s.name}=${s.value}`]),
];
const proc = spawn(
env.containerRuntime,
[
'exec',
'-i',
...envFlags,
'-w',
workspace.containerRepo,
boxName,
'/bin/bash',
'-lc',
'exec tmux new-session -A -s spoon 2>/dev/null || exec bash -l',
`exec script -qfc ${shellQuote(launcher)} /dev/null`,
],
AttachStdin: true,
AttachStdout: true,
AttachStderr: true,
Tty: true,
WorkingDir: workspace.containerRepo,
Env: [
'TERM=xterm-256color',
`HOME=${workspace.containerHome}`,
...workspace.secrets.map((s) => `${s.name}=${s.value}`),
],
});
stream = await exec.start({ hijack: true, stdin: true, Tty: true });
} catch (error) {
ws.close(
1011,
`Failed to start terminal: ${error instanceof Error ? error.message : 'unknown error'}`,
{ stdio: ['pipe', 'pipe', 'pipe'] },
);
releaseUserBox(workspace.username);
return;
}
procHolder.current = proc;
const activeStream = stream;
const activeExec = exec;
// Replay any keystrokes the client sent before the process was ready.
for (const buffered of pendingInput) proc.stdin.write(buffered);
pendingInput.length = 0;
activeStream.on('data', (chunk: Buffer) => {
const forward = (chunk: Buffer) => {
if (ws.readyState === ws.OPEN) ws.send(chunk, { binary: true });
});
activeStream.on('end', () => ws.close());
activeStream.on('error', () => ws.close());
ws.on('message', (data: Buffer, isBinary: boolean) => {
if (isBinary) {
activeStream.write(data);
return;
}
// Text frames are control messages (resize); anything else is treated as
// input for resilience.
try {
const message = JSON.parse(data.toString('utf8')) as {
type?: string;
cols?: number;
rows?: number;
};
if (message.type === 'resize' && message.cols && message.rows) {
void activeExec.resize({ w: message.cols, h: message.rows });
return;
}
} catch {
// fall through: treat as raw input
}
activeStream.write(data);
proc.stdout.on('data', forward);
proc.stderr.on('data', forward);
proc.on('exit', () => {
if (ws.readyState === ws.OPEN) ws.close();
});
proc.on('error', () => {
if (ws.readyState === ws.OPEN) ws.close();
});
let released = false;
const cleanup = () => {
if (released) return;
released = true;
activeStream.end();
releaseUserBox(workspace.username);
};
ws.on('close', cleanup);
ws.on('error', cleanup);
};
/**