super-dsh 0.3.1 → 0.3.2-e

package/superd-hub/dist/exec-pty-session.js

@@ -1,228 +0,0 @@
-import { Utf16TailRing } from '../../remote/plugin.js';
-// PTY session machinery — PORTED from better-dsh src/remote/pty-session.ts
-// (spec §8.2 / D12 "原样照抄"), with the ONE sanctioned adaptation: the
-// subprocess spawn is replaced by the PtyChannel seam (exec-pty-channel.ts) —
-// every state transition (init nonce handshake, FIFO dispatch, P8 two-stage
-// interrupt \x03 + injected 130-marker, 3s grace, group-kill fallback, idle
-// TTL, OUTPUT_HARD_CAP with surrogate-pair-safe slicing, I1 waiter hygiene)
-// is the ported logic, unchanged.
-import { buildInitCommand, createNonceFrameParser, genNonce, wrapPtyCommand } from './exec-pty-framing.js';
-/** spec §五.2 逐字。 */
-export const RECONNECT_NOTICE = '[remote: session reconnected to fresh shell; cwd reset to default]';
-const DEFAULT_INIT_TIMEOUT_SEC = 30;
-const INTERRUPT_GRACE_MS = 3_000;
-/** pty 收集内存安全帽（Ruling P15，与 oneshot HARD_CAP 对齐）：模型面只见尾窗，保留尾部。 */
-const OUTPUT_HARD_CAP = 4_000_000;
-export class PtySession {
-    opts;
-    state = 'cold';
-    startPromise = null;
-    queue = Promise.resolve();
-    idleTimer = null;
-    onChunk = null;
-    deathWaiters = [];
-    inFlight = 0;
-    lastSettleAt = 0;
-    constructor(opts) {
-        this.opts = opts;
-        this.opts.channel.onDead(() => this.markDead());
-    }
-    get sessionState() { return this.state; }
-    /** status 面快照：state + busy（有命令在跑）+ idleMs（上次 settle 至今）。 */
-    get snapshot() {
-        return { state: this.state, busy: this.inFlight > 0, idleMs: this.lastSettleAt > 0 ? Date.now() - this.lastSettleAt : null };
-    }
-    /** 同一会话命令严格串行（FIFO）；死亡/冷态先重启。 */
-    dispatch(cmd, dOpts = {}) {
-        const attempt = async () => {
-            const started = Date.now();
-            const reconnected = this.state === 'dead';
-            if (this.state !== 'ready')
-                await this.start();
-            this.inFlight++;
-            return await new Promise((resolve) => {
-                const nonce = genNonce();
-                const ring = new Utf16TailRing(OUTPUT_HARD_CAP);
-                let interrupted = false;
-                let settled = false;
-                let interruptTimer;
-                let killTimer;
-                const deathWaiter = () => finish({ output: ring.text, exit: null, cwd: null, timedOut: interrupted });
-                const cleanup = () => {
-                    this.onChunk = null;
-                    // I1（终审）：已结算 dispatch 的死亡等待必须摘除——否则闭包钉住整段 output，
-                    // 长命会话每调用累积一个 waiter（init 路径的 filter 同款形态）
-                    this.deathWaiters = this.deathWaiters.filter((w) => w !== deathWaiter);
-                    if (interruptTimer !== undefined)
-                        clearTimeout(interruptTimer);
-                    if (killTimer !== undefined)
-                        clearTimeout(killTimer);
-                };
-                const finish = (r) => {
-                    if (settled)
-                        return;
-                    settled = true;
-                    cleanup();
-                    this.inFlight--;
-                    this.lastSettleAt = Date.now();
-                    this.resetIdleTimer();
-                    resolve({ ...r, outputStats: { stdout: ring.stats() }, reconnected, durationMs: Date.now() - started });
-                };
-                const parser = createNonceFrameParser(nonce, {
-                    onOutput: (t) => {
-                        ring.push(t);
-                    },
-                    onFrame: ({ exit, cwd }) => finish({ output: ring.text, exit, cwd, timedOut: interrupted }),
-                });
-                this.onChunk = (c) => parser.feed(c);
-                this.write(wrapPtyCommand(cmd, nonce));
-                if (dOpts.stdin !== undefined)
-                    this.write(dOpts.stdin); // Ruling 11: REPL 式紧随
-                interruptTimer = setTimeout(() => {
-                    interrupted = true;
-                    // P8 两段打断：\x03 中断前台作业后，交互 bash 会放弃当前命令行的剩余部分
-                    // （尾部 marker 不再执行）；紧随注入的同 nonce 130-marker 行在 bash 回到
-                    // 读取态后被执行。无论 bash 弃行还是续行，帧都在宽限内到达且 exit=130，
-                    // 会话存活；若命令 trap 掉 SIGINT，原 marker 先帧、注入行帧后字节被丢弃。
-                    this.write('\x03');
-                    this.write(`printf '\\033]133;D;${nonce};130;%s\\007' "$PWD"\n`);
-                    killTimer = setTimeout(() => {
-                        // Ruling 8 的 kill 兜底：宽限无帧 = 会话不可恢复，杀整组后收尸
-                        // （否则 trap-INT 的卡死进程永生，重连后旧组再不可达）
-                        this.opts.channel.kill();
-                        this.markDead();
-                        finish({ output: ring.text, exit: null, cwd: null, timedOut: true });
-                    }, INTERRUPT_GRACE_MS);
-                }, dOpts.timeoutSec === undefined ? 120_000 : dOpts.timeoutSec * 1_000);
-                this.deathWaiters.push(deathWaiter);
-            });
-        };
-        const run = this.queue.then(attempt, attempt);
-        this.queue = run.then(() => undefined, () => undefined);
-        return run;
-    }
-    /** TTL / row teardown：尽力 exit，1s 后杀组，出池。 */
-    async dispose() {
-        if (this.state === 'disposed')
-            return;
-        this.clearIdleTimer();
-        this.state = 'disposed';
-        this.startPromise = null;
-        this.opts.channel.close();
-        this.opts.onDead?.();
-    }
-    write(s) {
-        this.opts.channel.write(s);
-    }
-    async start() {
-        if (this.state === 'disposed')
-            throw new Error(`[E_SESSION_DISPOSED] remote pty session '${this.opts.key}' was disposed — the pool replaces disposed sessions; this reference is stale`);
-        if (this.startPromise === null) {
-            this.state = 'starting';
-            this.startPromise = this.doStart().catch((err) => {
-                this.startPromise = null;
-                this.markDead();
-                throw err;
-            });
-        }
-        return this.startPromise;
-    }
-    async doStart() {
-        const initTimeoutSec = this.opts.initTimeoutSec ?? DEFAULT_INIT_TIMEOUT_SEC;
-        const initNonce = genNonce();
-        await new Promise((resolve, reject) => {
-            const timer = setTimeout(() => {
-                this.opts.channel.kill();
-                reject(new Error(`[E_SESSION_START] remote pty session '${this.opts.key}' did not initialize in ${initTimeoutSec}s` +
-                    (this.diag().length > 0 ? `; transport stderr tail: ${this.diag()}` : '')));
-            }, initTimeoutSec * 1_000);
-            const waiter = () => {
-                clearTimeout(timer);
-                reject(new Error(`[E_SESSION_DIED] remote pty session '${this.opts.key}' died before initializing` +
-                    (this.diag().length > 0 ? `; transport stderr tail: ${this.diag()}` : '')));
-            };
-            this.deathWaiters.push(waiter);
-            const parser = createNonceFrameParser(initNonce, {
-                onFrame: () => {
-                    clearTimeout(timer);
-                    this.deathWaiters = this.deathWaiters.filter((w) => w !== waiter);
-                    if (this.state === 'disposed') {
-                        // dispose 竞态 init：不得复活已终态会话（否则重挂 idle timer + 幽灵 ready）
-                        reject(new Error(`[E_SESSION_DISPOSED] remote pty session '${this.opts.key}' disposed during init`));
-                        return;
-                    }
-                    this.state = 'ready';
-                    this.resetIdleTimer();
-                    resolve();
-                },
-            });
-            this.onChunk = (c) => parser.feed(c);
-            this.opts.channel.open((chunk) => this.onChunk?.(chunk))
-                .catch((err) => {
-                clearTimeout(timer);
-                this.deathWaiters = this.deathWaiters.filter((w) => w !== waiter);
-                reject(err instanceof Error ? err : new Error(String(err)));
-            });
-            this.write(buildInitCommand(initNonce));
-        });
-    }
-    diag() { return this.opts.channel.diag; }
-    markDead() {
-        if (this.state === 'disposed') {
-            // dispose 是终态：不再翻转 state，但在飞 dispatch 的死亡等待仍需立即结算
-            // （row teardown 不是超时——不得等 123s 看门狗，也不得错标 timedOut）
-            const waiters = this.deathWaiters;
-            this.deathWaiters = [];
-            for (const w of waiters)
-                w();
-            return;
-        }
-        this.state = 'dead';
-        this.startPromise = null;
-        this.clearIdleTimer();
-        const waiters = this.deathWaiters;
-        this.deathWaiters = [];
-        for (const w of waiters)
-            w();
-    }
-    resetIdleTimer() {
-        this.clearIdleTimer();
-        this.idleTimer = setTimeout(() => void this.dispose(), this.opts.idleTtlSec * 1_000);
-        this.idleTimer.unref();
-    }
-    clearIdleTimer() {
-        if (this.idleTimer !== null) {
-            clearTimeout(this.idleTimer);
-            this.idleTimer = null;
-        }
-    }
-}
-/** 会话池：键 → 会话。意外死亡的会话留在池内（下次 dispatch 带 reconnect 提示）。 */
-export class PtyPool {
-    defaults;
-    sessions = new Map();
-    constructor(defaults) {
-        this.defaults = defaults;
-    }
-    get size() { return this.sessions.size; }
-    /** status 面：键上的会话快照（无会话 = undefined → "none — dials on first exec"）。 */
-    inspect(key) {
-        return this.sessions.get(key)?.snapshot;
-    }
-    /** roster 面：池内全部会话。 */
-    list() {
-        return [...this.sessions.entries()].map(([key, s]) => ({ key, snapshot: s.snapshot }));
-    }
-    getOrCreate(key, channelOf) {
-        let s = this.sessions.get(key);
-        if (s === undefined || s.sessionState === 'disposed') {
-            s = new PtySession({ key, channel: channelOf(), ...this.defaults, onDead: () => this.sessions.delete(key) });
-            this.sessions.set(key, s);
-        }
-        return s;
-    }
-    async disposeAll() {
-        await Promise.all([...this.sessions.values()].map((s) => s.dispose()));
-        this.sessions.clear();
-    }
-}