echo-cli-agent 0.1.0

package/dist/src/state/state-store.js

@@ -0,0 +1,125 @@
+import { chmod, mkdir, readFile, rename, writeFile } from 'node:fs/promises';
+import { dirname } from 'node:path';
+const MAX_RUNS = 2_000;
+export class StateStore {
+    filePath;
+    state;
+    writeChain = Promise.resolve();
+    constructor(filePath) {
+        this.filePath = filePath;
+    }
+    async initialize() {
+        await mkdir(dirname(this.filePath), { recursive: true, mode: 0o700 });
+        try {
+            const raw = await readFile(this.filePath, 'utf8');
+            const parsed = JSON.parse(raw);
+            if (parsed.version === 1 || parsed.version === 2 || parsed.version === 3) {
+                this.state = {
+                    version: 3,
+                    runs: parsed.runs ?? {},
+                    sessions: parsed.sessions ?? {},
+                    cliWindows: parsed.cliWindows ?? {}
+                };
+                if (parsed.version !== 3)
+                    await this.persist();
+            }
+            else {
+                throw new Error('unsupported state schema');
+            }
+        }
+        catch (error) {
+            const missing = error.code === 'ENOENT';
+            if (!missing && error instanceof SyntaxError) {
+                await rename(this.filePath, `${this.filePath}.invalid-${Date.now()}`);
+            }
+            else if (!missing && !(error instanceof SyntaxError)) {
+                throw error;
+            }
+            this.state = {
+                version: 3,
+                runs: {},
+                sessions: {},
+                cliWindows: {}
+            };
+            await this.persist();
+        }
+    }
+    getRun(runId) {
+        return this.state?.runs[runId];
+    }
+    async markRun(runId, status, sessionId) {
+        const state = this.requireState();
+        const existing = state.runs[runId];
+        if (existing?.status === 'completed' && status !== 'completed')
+            return;
+        state.runs[runId] = {
+            runId,
+            status,
+            ...(sessionId ? { sessionId } : {}),
+            updatedAt: new Date().toISOString()
+        };
+        const entries = Object.values(state.runs);
+        if (entries.length > MAX_RUNS) {
+            entries
+                .sort((left, right) => left.updatedAt.localeCompare(right.updatedAt))
+                .slice(0, entries.length - MAX_RUNS)
+                .forEach((entry) => delete state.runs[entry.runId]);
+        }
+        await this.persist();
+    }
+    getSession(sessionKey) {
+        return this.state?.sessions[sessionKey];
+    }
+    async upsertSession(record) {
+        this.requireState().sessions[record.sessionKey] = record;
+        await this.persist();
+    }
+    async removeSession(sessionKey) {
+        delete this.requireState().sessions[sessionKey];
+        await this.persist();
+    }
+    getCliWindow(taskId, agentType) {
+        return this.state?.cliWindows[this.cliWindowKey(taskId, agentType)];
+    }
+    async upsertCliWindow(record) {
+        this.requireState().cliWindows[this.cliWindowKey(record.taskId, record.agentType)] = record;
+        await this.persist();
+    }
+    listCliWindows(agentType) {
+        const normalized = agentType?.toLowerCase();
+        return Object.values(this.requireState().cliWindows)
+            .filter((item) => !normalized || item.agentType.toLowerCase() === normalized)
+            .sort((left, right) => left.updatedAt.localeCompare(right.updatedAt));
+    }
+    snapshot() {
+        const state = this.requireState();
+        return {
+            runs: Object.keys(state.runs).length,
+            sessions: Object.keys(state.sessions).length
+        };
+    }
+    recentRuns(limit = 20) {
+        return Object.values(this.requireState().runs)
+            .sort((left, right) => right.updatedAt.localeCompare(left.updatedAt))
+            .slice(0, limit);
+    }
+    requireState() {
+        if (!this.state)
+            throw new Error('state store is not initialized');
+        return this.state;
+    }
+    cliWindowKey(taskId, agentType) {
+        return `${taskId}:${agentType.toLowerCase()}`;
+    }
+    persist() {
+        const serialized = JSON.stringify(this.requireState(), null, 2);
+        this.writeChain = this.writeChain.then(async () => {
+            const temporary = `${this.filePath}.${process.pid}.tmp`;
+            await writeFile(temporary, serialized, { mode: 0o600 });
+            await rename(temporary, this.filePath);
+            await chmod(this.filePath, 0o600).catch(() => undefined);
+        });
+        return this.writeChain;
+    }
+}
+//# sourceMappingURL=state-store.js.map

package/dist/src/state/state-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"state-store.js","sourceRoot":"","sources":["../../../src/state/state-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAA;AAC5E,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAA;AA6CnC,MAAM,QAAQ,GAAG,KAAK,CAAA;AAEtB,MAAM,OAAO,UAAU;IAIQ;IAHrB,KAAK,CAAa;IAClB,UAAU,GAAG,OAAO,CAAC,OAAO,EAAE,CAAA;IAEtC,YAA6B,QAAgB;QAAhB,aAAQ,GAAR,QAAQ,CAAQ;IAAG,CAAC;IAEjD,KAAK,CAAC,UAAU;QACd,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;QACrE,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;YACjD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAwB,CAAA;YACrD,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC,EAAE,CAAC;gBACzE,IAAI,CAAC,KAAK,GAAG;oBACX,OAAO,EAAE,CAAC;oBACV,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;oBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE;oBAC/B,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,EAAE;iBACpC,CAAA;gBACD,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC;oBAAE,MAAM,IAAI,CAAC,OAAO,EAAE,CAAA;YAChD,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAA;YAC7C,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAI,KAA+B,CAAC,IAAI,KAAK,QAAQ,CAAA;YAClE,IAAI,CAAC,OAAO,IAAI,KAAK,YAAY,WAAW,EAAE,CAAC;gBAC7C,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,QAAQ,YAAY,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;YACvE,CAAC;iBAAM,IAAI,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,YAAY,WAAW,CAAC,EAAE,CAAC;gBACvD,MAAM,KAAK,CAAA;YACb,CAAC;YACD,IAAI,CAAC,KAAK,GAAG;gBACX,OAAO,EAAE,CAAC;gBACV,IAAI,EAAE,EAAE;gBACR,QAAQ,EAAE,EAAE;gBACZ,UAAU,EAAE,EAAE;aACf,CAAA;YACD,MAAM,IAAI,CAAC,OAAO,EAAE,CAAA;QACtB,CAAC;IAEH,CAAC;IAED,MAAM,CAAC,KAAa;QAClB,OAAO,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAA;IAChC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,KAAa,EAAE,MAAiB,EAAE,SAAkB;QAChE,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,EAAE,CAAA;QACjC,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAClC,IAAI,QAAQ,EAAE,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,WAAW;YAAE,OAAM;QACtE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG;YAClB,KAAK;YACL,MAAM;YACN,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAA;QACD,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QACzC,IAAI,OAAO,CAAC,MAAM,GAAG,QAAQ,EAAE,CAAC;YAC9B,OAAO;iBACJ,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;iBACpE,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,GAAG,QAAQ,CAAC;iBACnC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAA;QACvD,CAAC;QACD,MAAM,IAAI,CAAC,OAAO,EAAE,CAAA;IACtB,CAAC;IAED,UAAU,CAAC,UAAkB;QAC3B,OAAO,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAA;IACzC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAqB;QACvC,IAAI,CAAC,YAAY,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,MAAM,CAAA;QACxD,MAAM,IAAI,CAAC,OAAO,EAAE,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,UAAkB;QACpC,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAA;QAC/C,MAAM,IAAI,CAAC,OAAO,EAAE,CAAA;IACtB,CAAC;IAED,YAAY,CAAC,MAAc,EAAE,SAAiB;QAC5C,OAAO,IAAI,CAAC,KAAK,EAAE,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAA;IACrE,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,MAAuB;QAC3C,IAAI,CAAC,YAAY,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,MAAM,CAAA;QAC3F,MAAM,IAAI,CAAC,OAAO,EAAE,CAAA;IACtB,CAAC;IAED,cAAc,CAAC,SAAkB;QAC/B,MAAM,UAAU,GAAG,SAAS,EAAE,WAAW,EAAE,CAAA;QAC3C,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,UAAU,CAAC;aACjD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,KAAK,UAAU,CAAC;aAC5E,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAA;IACzE,CAAC;IAED,QAAQ;QACN,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,EAAE,CAAA;QACjC,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM;YACpC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM;SAC7C,CAAA;IACH,CAAC;IAED,UAAU,CAAC,KAAK,GAAG,EAAE;QACnB,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC;aAC3C,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;aACpE,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAA;IACpB,CAAC;IAEO,YAAY;QAClB,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAA;QAClE,OAAO,IAAI,CAAC,KAAK,CAAA;IACnB,CAAC;IAEO,YAAY,CAAC,MAAc,EAAE,SAAiB;QACpD,OAAO,GAAG,MAAM,IAAI,SAAS,CAAC,WAAW,EAAE,EAAE,CAAA;IAC/C,CAAC;IAEO,OAAO;QACb,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAC/D,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;YAChD,MAAM,SAAS,GAAG,GAAG,IAAI,CAAC,QAAQ,IAAI,OAAO,CAAC,GAAG,MAAM,CAAA;YACvD,MAAM,SAAS,CAAC,SAAS,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;YACvD,MAAM,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAA;YACtC,MAAM,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAA;QAC1D,CAAC,CAAC,CAAA;QACF,OAAO,IAAI,CAAC,UAAU,CAAA;IACxB,CAAC;CACF"}

package/dist/src/transport/taskmcp-client.d.ts

@@ -0,0 +1,29 @@
+import { EventEmitter } from 'node:events';
+import type { AgentConfig } from '../config.js';
+import type { DiagnosticLogger } from '../logging/diagnostic-log.js';
+import type { AgentCliAck, AgentCliControlAck, AgentCliEvent } from '../types/protocol.js';
+type ClientEvents = {
+    message: [unknown];
+    connection: [boolean];
+};
+export declare class TaskMcpClient extends EventEmitter<ClientEvents> {
+    private readonly config;
+    private readonly deviceId;
+    private readonly logger;
+    private socket;
+    private stopped;
+    private retryAttempt;
+    private reconnectTimer;
+    private heartbeat;
+    constructor(config: AgentConfig, deviceId: string, logger: DiagnosticLogger);
+    get connected(): boolean;
+    start(): void;
+    stop(): void;
+    sendEvent(event: AgentCliEvent): Promise<void>;
+    sendAck(ack: AgentCliAck): Promise<void>;
+    sendControlAck(ack: AgentCliControlAck): Promise<void>;
+    private send;
+    private connect;
+    private scheduleReconnect;
+}
+export {};

package/dist/src/transport/taskmcp-client.js

@@ -0,0 +1,115 @@
+import { EventEmitter } from 'node:events';
+import WebSocket from 'ws';
+export class TaskMcpClient extends EventEmitter {
+    config;
+    deviceId;
+    logger;
+    socket;
+    stopped = false;
+    retryAttempt = 0;
+    reconnectTimer;
+    heartbeat;
+    constructor(config, deviceId, logger) {
+        super();
+        this.config = config;
+        this.deviceId = deviceId;
+        this.logger = logger;
+    }
+    get connected() {
+        return this.socket?.readyState === WebSocket.OPEN;
+    }
+    start() {
+        this.stopped = false;
+        this.connect();
+    }
+    stop() {
+        this.stopped = true;
+        if (this.reconnectTimer)
+            clearTimeout(this.reconnectTimer);
+        if (this.heartbeat)
+            clearInterval(this.heartbeat);
+        this.socket?.close(1000, 'agent shutdown');
+        this.socket = undefined;
+    }
+    sendEvent(event) {
+        return this.send(event);
+    }
+    sendAck(ack) {
+        return this.send(ack);
+    }
+    sendControlAck(ack) {
+        return this.send(ack);
+    }
+    send(message) {
+        return new Promise((resolveSend, rejectSend) => {
+            if (!this.socket || this.socket.readyState !== WebSocket.OPEN) {
+                rejectSend(new Error('TaskMCP WebSocket is not connected'));
+                return;
+            }
+            this.socket.send(JSON.stringify(message), (error) => {
+                if (error)
+                    rejectSend(error);
+                else
+                    resolveSend();
+            });
+        });
+    }
+    connect() {
+        if (this.stopped)
+            return;
+        const url = new URL(this.config.wsUrl);
+        url.searchParams.set('deviceId', this.deviceId);
+        url.searchParams.set('role', 'orchestrator');
+        url.searchParams.set('platform', this.config.platform);
+        url.searchParams.set('name', this.config.deviceName);
+        url.searchParams.set('endpoint', this.config.endpoint);
+        url.searchParams.set('capabilities', this.config.capabilities.join(','));
+        if (this.config.authMode === 'bearer')
+            url.searchParams.set('token', this.config.authToken);
+        const headers = this.config.authMode === 'mcp' ? { 'X-MCP-Token': this.config.authToken } : undefined;
+        const socket = new WebSocket(url, { ...(headers ? { headers } : {}) });
+        this.socket = socket;
+        socket.on('open', () => {
+            this.retryAttempt = 0;
+            this.logger.info('taskmcp-connected', { deviceId: this.deviceId });
+            this.emit('connection', true);
+            this.heartbeat = setInterval(() => {
+                // The server refreshes Redis presence on an application-level ping.
+                if (socket.readyState === WebSocket.OPEN)
+                    socket.send('{"type":"ping"}');
+            }, 25_000);
+        });
+        socket.on('message', (data) => {
+            try {
+                this.emit('message', JSON.parse(data.toString()));
+            }
+            catch {
+                this.logger.warn('taskmcp-invalid-json');
+            }
+        });
+        socket.on('error', (error) => {
+            this.logger.warn('taskmcp-socket-error', { error: error.message });
+        });
+        socket.on('close', (code) => {
+            if (this.heartbeat)
+                clearInterval(this.heartbeat);
+            this.heartbeat = undefined;
+            if (this.socket === socket)
+                this.socket = undefined;
+            this.emit('connection', false);
+            this.logger.warn('taskmcp-disconnected', { code });
+            this.scheduleReconnect();
+        });
+    }
+    scheduleReconnect() {
+        if (this.stopped || this.reconnectTimer)
+            return;
+        const base = Math.min(30_000, 1_000 * 2 ** this.retryAttempt++);
+        const delay = base + Math.floor(Math.random() * 500);
+        this.reconnectTimer = setTimeout(() => {
+            this.reconnectTimer = undefined;
+            this.connect();
+        }, delay);
+    }
+}
+//# sourceMappingURL=taskmcp-client.js.map

package/dist/src/transport/taskmcp-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"taskmcp-client.js","sourceRoot":"","sources":["../../../src/transport/taskmcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAA;AAC1C,OAAO,SAAS,MAAM,IAAI,CAAA;AAU1B,MAAM,OAAO,aAAc,SAAQ,YAA0B;IAQxC;IACA;IACA;IATX,MAAM,CAAuB;IAC7B,OAAO,GAAG,KAAK,CAAA;IACf,YAAY,GAAG,CAAC,CAAA;IAChB,cAAc,CAA4B;IAC1C,SAAS,CAA4B;IAE7C,YACmB,MAAmB,EACnB,QAAgB,EAChB,MAAwB;QAEzC,KAAK,EAAE,CAAA;QAJU,WAAM,GAAN,MAAM,CAAa;QACnB,aAAQ,GAAR,QAAQ,CAAQ;QAChB,WAAM,GAAN,MAAM,CAAkB;IAG3C,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,EAAE,UAAU,KAAK,SAAS,CAAC,IAAI,CAAA;IACnD,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,GAAG,KAAK,CAAA;QACpB,IAAI,CAAC,OAAO,EAAE,CAAA;IAChB,CAAC;IAED,IAAI;QACF,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QACnB,IAAI,IAAI,CAAC,cAAc;YAAE,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;QAC1D,IAAI,IAAI,CAAC,SAAS;YAAE,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QACjD,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAA;QAC1C,IAAI,CAAC,MAAM,GAAG,SAAS,CAAA;IACzB,CAAC;IAED,SAAS,CAAC,KAAoB;QAC5B,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACzB,CAAC;IAED,OAAO,CAAC,GAAgB;QACtB,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACvB,CAAC;IAED,cAAc,CAAC,GAAuB;QACpC,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACvB,CAAC;IAEO,IAAI,CAAC,OAAyD;QACpE,OAAO,IAAI,OAAO,CAAC,CAAC,WAAW,EAAE,UAAU,EAAE,EAAE;YAC7C,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;gBAC9D,UAAU,CAAC,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC,CAAA;gBAC3D,OAAM;YACR,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE;gBAClD,IAAI,KAAK;oBAAE,UAAU,CAAC,KAAK,CAAC,CAAA;;oBACvB,WAAW,EAAE,CAAA;YACpB,CAAC,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC;IAEO,OAAO;QACb,IAAI,IAAI,CAAC,OAAO;YAAE,OAAM;QACxB,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QACtC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC/C,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;QAC5C,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;QACtD,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAA;QACpD,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;QACtD,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAA;QACxE,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,KAAK,QAAQ;YAAE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;QAE3F,MAAM,OAAO,GACX,IAAI,CAAC,MAAM,CAAC,QAAQ,KAAK,KAAK,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,SAAS,CAAA;QACvF,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,GAAG,EAAE,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAA;QACtE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;QAEpB,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACrB,IAAI,CAAC,YAAY,GAAG,CAAC,CAAA;YACrB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAA;YAClE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,CAAA;YAC7B,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE;gBAChC,oEAAoE;gBACpE,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI;oBAAE,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;YAC1E,CAAC,EAAE,MAAM,CAAC,CAAA;QACZ,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;YAC5B,IAAI,CAAC;gBACH,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAA;YACnD,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAA;YAC1C,CAAC;QACH,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QACpE,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YAC1B,IAAI,IAAI,CAAC,SAAS;gBAAE,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;YACjD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;YAC1B,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM;gBAAE,IAAI,CAAC,MAAM,GAAG,SAAS,CAAA;YACnD,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,KAAK,CAAC,CAAA;YAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAA;YAClD,IAAI,CAAC,iBAAiB,EAAE,CAAA;QAC1B,CAAC,CAAC,CAAA;IACJ,CAAC;IAEO,iBAAiB;QACvB,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,cAAc;YAAE,OAAM;QAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,GAAG,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC,CAAA;QAC/D,MAAM,KAAK,GAAG,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAA;QACpD,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAA;YAC/B,IAAI,CAAC,OAAO,EAAE,CAAA;QAChB,CAAC,EAAE,KAAK,CAAC,CAAA;IACX,CAAC;CACF"}

package/dist/src/types/protocol.d.ts

@@ -0,0 +1,98 @@
+import { z } from 'zod';
+export declare const AgentCliRequestSchema: z.ZodObject<{
+    type: z.ZodLiteral<"agent_cli">;
+    run_id: z.ZodString;
+    task_id: z.ZodString;
+    conversation_id: z.ZodOptional<z.ZodString>;
+    user_message_id: z.ZodOptional<z.ZodString>;
+    agent_type: z.ZodDefault<z.ZodString>;
+    action: z.ZodDefault<z.ZodEnum<{
+        prompt: "prompt";
+        session_create: "session_create";
+        session_list: "session_list";
+        model_catalog: "model_catalog";
+        run_cancel: "run_cancel";
+    }>>;
+    prompt: z.ZodOptional<z.ZodString>;
+    window_name: z.ZodOptional<z.ZodString>;
+    workdir: z.ZodNullable<z.ZodOptional<z.ZodString>>;
+    model: z.ZodNullable<z.ZodOptional<z.ZodString>>;
+    reasoning_effort: z.ZodNullable<z.ZodOptional<z.ZodString>>;
+    session_id: z.ZodNullable<z.ZodOptional<z.ZodString>>;
+    request_kind: z.ZodOptional<z.ZodEnum<{
+        new: "new";
+        followup: "followup";
+        resume: "resume";
+    }>>;
+    config: z.ZodDefault<z.ZodObject<{
+        executable: z.ZodOptional<z.ZodString>;
+        arguments: z.ZodOptional<z.ZodUnion<readonly [z.ZodArray<z.ZodString>, z.ZodString]>>;
+        protocol_arguments: z.ZodOptional<z.ZodUnion<readonly [z.ZodArray<z.ZodString>, z.ZodString]>>;
+        timeout_seconds: z.ZodOptional<z.ZodNumber>;
+        environment_file: z.ZodOptional<z.ZodString>;
+        base_url: z.ZodOptional<z.ZodString>;
+    }, z.core.$loose>>;
+}, z.core.$strip>;
+export type AgentCliRequest = z.infer<typeof AgentCliRequestSchema>;
+export type AgentCliStage = 'agent_received' | 'process_started' | 'session_initialized' | 'prompt_delivered' | 'completed' | 'process_exited' | 'failed';
+export type AgentCliEvent = {
+    type: 'agent_cli.event';
+    run_id: string;
+    stage: AgentCliStage;
+    session_id?: string;
+    data: Record<string, unknown>;
+};
+export type AgentCliAck = {
+    type: 'agent_cli.ack';
+    run_id: string;
+    task_id?: string;
+    conversation_id?: string;
+    user_message_id?: string;
+    agent_type?: string;
+    session_id?: string;
+    status: 'done' | 'failed' | 'timeout' | 'cancelled';
+    exit_code?: number;
+    final_text?: string;
+    stderr?: string;
+    duration_ms?: number;
+};
+export type AgentCliControlWindow = {
+    task_id: string;
+    agent_type: string;
+    name: string;
+    session_id?: string;
+};
+export type AgentCliControlAck = {
+    type: 'agent_cli.control_ack';
+    run_id: string;
+    action: 'session_create' | 'session_list' | 'model_catalog' | 'run_cancel';
+    status: 'done' | 'failed';
+    task_id?: string;
+    agent_type?: string;
+    name?: string;
+    session_id?: string;
+    windows?: AgentCliControlWindow[];
+    catalog?: AgentCliModelCatalog;
+    error?: string;
+};
+export type AgentCliModelCatalogEntry = {
+    id: string;
+    name: string;
+    description?: string;
+    default_reasoning_effort?: string;
+    supported_reasoning_efforts: string[];
+};
+export type AgentCliModelCatalog = {
+    agent_type: string;
+    default_model?: string;
+    default_reasoning_effort?: string;
+    fetched_at: string;
+    models: AgentCliModelCatalogEntry[];
+};
+export type DeliveryResult = {
+    runId: string;
+    sessionId: string;
+    sessionKey: string;
+    created: boolean;
+    duplicate: boolean;
+};

package/dist/src/types/protocol.js

@@ -0,0 +1,38 @@
+import { z } from 'zod';
+const StringArrayOrString = z.union([z.array(z.string()), z.string()]);
+export const AgentCliRequestSchema = z.object({
+    type: z.literal('agent_cli'),
+    run_id: z.string().min(1).max(128),
+    task_id: z.string().min(1).max(128),
+    conversation_id: z.string().optional(),
+    user_message_id: z.string().optional(),
+    agent_type: z.string().min(1).max(40).default('claude'),
+    action: z.enum(['prompt', 'session_create', 'session_list', 'model_catalog', 'run_cancel']).default('prompt'),
+    prompt: z.string().min(1).max(256 * 1024).optional(),
+    window_name: z.string().min(1).max(100).optional(),
+    workdir: z.string().min(1).optional().nullable(),
+    model: z.string().optional().nullable(),
+    reasoning_effort: z.string().min(1).max(40).optional().nullable(),
+    session_id: z.string().min(1).max(512).optional().nullable(),
+    request_kind: z.enum(['new', 'followup', 'resume']).optional(),
+    config: z
+        .object({
+        executable: z.string().optional(),
+        arguments: StringArrayOrString.optional(),
+        protocol_arguments: StringArrayOrString.optional(),
+        timeout_seconds: z.number().int().positive().max(86_400).optional(),
+        environment_file: z.string().optional(),
+        base_url: z.string().url().optional()
+    })
+        .passthrough()
+        .default({})
+}).superRefine((request, context) => {
+    if (request.action === 'prompt' && !request.session_id?.trim()) {
+        context.addIssue({
+            code: 'custom',
+            path: ['session_id'],
+            message: 'prompt requests must provide the provider-native session id'
+        });
+    }
+});
+//# sourceMappingURL=protocol.js.map

package/dist/src/types/protocol.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"protocol.js","sourceRoot":"","sources":["../../../src/types/protocol.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;AAEtE,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5C,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC;IAC5B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC;IAClC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC;IACnC,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACtC,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACtC,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;IACvD,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,gBAAgB,EAAE,cAAc,EAAE,eAAe,EAAE,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;IAC7G,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC,QAAQ,EAAE;IACpD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE;IAClD,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;IAChD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;IACvC,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;IACjE,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;IAC5D,YAAY,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE;IAC9D,MAAM,EAAE,CAAC;SACN,MAAM,CAAC;QACN,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACjC,SAAS,EAAE,mBAAmB,CAAC,QAAQ,EAAE;QACzC,kBAAkB,EAAE,mBAAmB,CAAC,QAAQ,EAAE;QAClD,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;QACnE,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACvC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;KACtC,CAAC;SACD,WAAW,EAAE;SACb,OAAO,CAAC,EAAE,CAAC;CACf,CAAC,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE;IAClC,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,IAAI,EAAE,EAAE,CAAC;QAC/D,OAAO,CAAC,QAAQ,CAAC;YACf,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,CAAC,YAAY,CAAC;YACpB,OAAO,EAAE,6DAA6D;SACvE,CAAC,CAAA;IACJ,CAAC;AACH,CAAC,CAAC,CAAA"}

package/dist/tests/dispatch-integration.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/tests/dispatch-integration.test.js

@@ -0,0 +1,209 @@
+import { once } from 'node:events';
+import { mkdtemp } from 'node:fs/promises';
+import { tmpdir } from 'node:os';
+import { join } from 'node:path';
+import { WebSocketServer } from 'ws';
+import { afterEach, describe, expect, it } from 'vitest';
+import { Dispatcher } from '../src/dispatch/dispatcher.js';
+import { StateStore } from '../src/state/state-store.js';
+import { TaskMcpClient } from '../src/transport/taskmcp-client.js';
+const logger = {
+    error: () => undefined,
+    warn: () => undefined,
+    info: () => undefined,
+    debug: () => undefined,
+    close: () => undefined
+};
+describe('TaskMCP dispatch integration', () => {
+    const clients = [];
+    const servers = [];
+    afterEach(async () => {
+        clients.forEach((client) => client.stop());
+        await Promise.all(servers.map((server) => new Promise((resolve) => server.close(() => resolve()))));
+    });
+    it('authenticates, completes a delivered prompt, and acknowledges a replay without re-delivery', async () => {
+        const server = new WebSocketServer({ host: '127.0.0.1', port: 0 });
+        servers.push(server);
+        await once(server, 'listening');
+        const address = server.address();
+        if (!address || typeof address === 'string')
+            throw new Error('unexpected WebSocket address');
+        const directory = await mkdtemp(join(tmpdir(), 'api-ocra-integration-'));
+        const state = new StateStore(join(directory, 'state.json'));
+        await state.initialize();
+        let deliveries = 0;
+        const sessions = {
+            async deliver(request, hooks) {
+                deliveries += 1;
+                await hooks.onProcessStarted('session-test', 123);
+                await hooks.onSessionInitialized('session-test', true);
+                await hooks.onProcessExited('session-test', {
+                    exitCode: 0,
+                    durationMs: 12
+                });
+                return {
+                    runId: request.run_id,
+                    sessionId: 'session-test',
+                    sessionKey: 'task-test:codex',
+                    created: true,
+                    duplicate: false
+                };
+            }
+        };
+        const config = {
+            wsUrl: `ws://127.0.0.1:${address.port}/ws/echo`,
+            authMode: 'mcp',
+            authToken: 'device-secret',
+            deviceId: 'device-test',
+            deviceName: 'Integration Device',
+            endpoint: 'cli',
+            capabilities: ['codex'],
+            platform: process.platform,
+            allowedRoots: [directory],
+            allowExecutableOverride: false,
+            dataDir: directory,
+            stateFile: join(directory, 'state.json'),
+            logFile: join(directory, 'agent.log'),
+            healthHost: '127.0.0.1',
+            healthPort: 0,
+            logLevel: 'error'
+        };
+        const client = new TaskMcpClient(config, 'device-test', logger);
+        clients.push(client);
+        const dispatcher = new Dispatcher(sessions, state, client, logger);
+        client.on('message', (message) => dispatcher.accept(message));
+        const received = [];
+        const connection = once(server, 'connection');
+        client.start();
+        const [socket, request] = (await connection);
+        const connectedUrl = new URL(request.url ?? '/', 'ws://127.0.0.1');
+        expect(connectedUrl.searchParams.get('deviceId')).toBe('device-test');
+        expect(connectedUrl.searchParams.get('role')).toBe('orchestrator');
+        expect(request.headers['x-mcp-token']).toBe('device-secret');
+        socket.on('message', (data) => received.push(JSON.parse(data.toString())));
+        const dispatch = {
+            type: 'agent_cli',
+            run_id: 'run-test',
+            task_id: 'task-test',
+            agent_type: 'codex',
+            session_id: '019ff004-803d-7e21-8cc9-8494543d2d7e',
+            prompt: 'do the work',
+            workdir: directory,
+            config: {}
+        };
+        socket.send(JSON.stringify(dispatch));
+        await waitUntil(() => received.some((event) => event.type === 'agent_cli.event' && event.stage === 'prompt_delivered'));
+        socket.send(JSON.stringify(dispatch));
+        await waitUntil(() => received.filter((event) => event.type === 'agent_cli.event' && event.stage === 'prompt_delivered').length === 2);
+        await dispatcher.waitForIdle();
+        expect(deliveries).toBe(1);
+        expect(received.filter((event) => event.type === 'agent_cli.event').map((event) => event.stage)).toEqual([
+            'agent_received',
+            'process_started',
+            'session_initialized',
+            'process_exited',
+            'prompt_delivered',
+            'agent_received',
+            'prompt_delivered',
+            'completed'
+        ]);
+        const ack = received.find((event) => event.type === 'agent_cli.ack');
+        expect(ack).toMatchObject({ status: 'done' });
+        expect(ack).not.toHaveProperty('final_text');
+        const events = received.filter((event) => event.type === 'agent_cli.event');
+        expect(events.at(-2)?.data).toEqual({ duplicate: true });
+        expect(events.at(-1)?.data).toEqual({ delivery: 'prompt_delivered', duplicate: true });
+    });
+    it('forwards a run cancellation to the active local session', async () => {
+        const directory = await mkdtemp(join(tmpdir(), 'api-ocra-cancel-'));
+        const state = new StateStore(join(directory, 'state.json'));
+        await state.initialize();
+        const cancelledRuns = [];
+        const controlAcks = [];
+        const sessions = {
+            async deliver() {
+                throw new Error('not used by a cancel request');
+            },
+            cancelRun(runId) {
+                cancelledRuns.push(runId);
+                return true;
+            }
+        };
+        const sender = {
+            sendEvent: async () => undefined,
+            sendAck: async () => undefined,
+            sendControlAck: async (ack) => { controlAcks.push(ack); }
+        };
+        const dispatcher = new Dispatcher(sessions, state, sender, logger);
+        dispatcher.accept({
+            type: 'agent_cli',
+            run_id: 'run-to-cancel',
+            task_id: 'task-test',
+            agent_type: 'codex',
+            action: 'run_cancel',
+            config: {}
+        });
+        await waitUntil(() => controlAcks.length === 1);
+        expect(cancelledRuns).toEqual(['run-to-cancel']);
+        expect(controlAcks[0]).toMatchObject({
+            run_id: 'run-to-cancel',
+            action: 'run_cancel',
+            status: 'done'
+        });
+    });
+    it('includes the sanitized CLI failure detail in the terminal acknowledgement', async () => {
+        const directory = await mkdtemp(join(tmpdir(), 'api-ocra-failure-'));
+        const state = new StateStore(join(directory, 'state.json'));
+        await state.initialize();
+        const acknowledgements = [];
+        const sessions = {
+            async deliver(request, hooks) {
+                await hooks.onProcessStarted('session-failure', 123);
+                await hooks.onSessionInitialized('session-failure', true);
+                await hooks.onProcessExited('session-failure', {
+                    exitCode: 1,
+                    finalText: '403 Forbidden: provider balance is insufficient',
+                    durationMs: 12
+                });
+                return {
+                    runId: request.run_id,
+                    sessionId: 'session-failure',
+                    sessionKey: 'task-failure:codex',
+                    created: true,
+                    duplicate: false
+                };
+            }
+        };
+        const sender = {
+            sendEvent: async () => undefined,
+            sendAck: async (ack) => { acknowledgements.push(ack); },
+            sendControlAck: async () => undefined
+        };
+        const dispatcher = new Dispatcher(sessions, state, sender, logger);
+        dispatcher.accept({
+            type: 'agent_cli',
+            run_id: 'run-failure',
+            task_id: 'task-failure',
+            agent_type: 'codex',
+            session_id: '019ff004-803d-7e21-8cc9-8494543d2d7e',
+            prompt: 'do the work',
+            workdir: directory,
+            config: {}
+        });
+        await waitUntil(() => acknowledgements.length === 1);
+        expect(acknowledgements[0]).toMatchObject({
+            status: 'failed',
+            exit_code: 1,
+            stderr: '403 Forbidden: provider balance is insufficient'
+        });
+    });
+});
+async function waitUntil(predicate) {
+    const deadline = Date.now() + 2_000;
+    while (!predicate()) {
+        if (Date.now() > deadline)
+            throw new Error('timed out waiting for WebSocket event');
+        await new Promise((resolve) => setTimeout(resolve, 10));
+    }
+}
+//# sourceMappingURL=dispatch-integration.test.js.map

package/dist/tests/dispatch-integration.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dispatch-integration.test.js","sourceRoot":"","sources":["../../tests/dispatch-integration.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,aAAa,CAAA;AAClC,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAA;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,eAAe,EAAkB,MAAM,IAAI,CAAA;AACpD,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAExD,OAAO,EAAE,UAAU,EAA0C,MAAM,+BAA+B,CAAA;AAElG,OAAO,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAA;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,oCAAoC,CAAA;AAGlE,MAAM,MAAM,GAAqB;IAC/B,KAAK,EAAE,GAAG,EAAE,CAAC,SAAS;IACtB,IAAI,EAAE,GAAG,EAAE,CAAC,SAAS;IACrB,IAAI,EAAE,GAAG,EAAE,CAAC,SAAS;IACrB,KAAK,EAAE,GAAG,EAAE,CAAC,SAAS;IACtB,KAAK,EAAE,GAAG,EAAE,CAAC,SAAS;CACvB,CAAA;AAED,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC5C,MAAM,OAAO,GAAoB,EAAE,CAAA;IACnC,MAAM,OAAO,GAAsB,EAAE,CAAA;IAErC,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAA;QAC1C,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;IAC3G,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4FAA4F,EAAE,KAAK,IAAI,EAAE;QAC1G,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAA;QAClE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACpB,MAAM,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QAC/B,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAA;QAChC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAA;QAE5F,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QACxE,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAA;QAC3D,MAAM,KAAK,CAAC,UAAU,EAAE,CAAA;QACxB,IAAI,UAAU,GAAG,CAAC,CAAA;QAClB,MAAM,QAAQ,GAAoB;YAChC,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK;gBAC1B,UAAU,IAAI,CAAC,CAAA;gBACf,MAAM,KAAK,CAAC,gBAAgB,CAAC,cAAc,EAAE,GAAG,CAAC,CAAA;gBACjD,MAAM,KAAK,CAAC,oBAAoB,CAAC,cAAc,EAAE,IAAI,CAAC,CAAA;gBACtD,MAAM,KAAK,CAAC,eAAe,CAAC,cAAc,EAAE;oBAC1C,QAAQ,EAAE,CAAC;oBACX,UAAU,EAAE,EAAE;iBACf,CAAC,CAAA;gBACF,OAAO;oBACL,KAAK,EAAE,OAAO,CAAC,MAAM;oBACrB,SAAS,EAAE,cAAc;oBACzB,UAAU,EAAE,iBAAiB;oBAC7B,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,KAAK;iBACjB,CAAA;YACH,CAAC;SACF,CAAA;QAED,MAAM,MAAM,GAAgB;YAC1B,KAAK,EAAE,kBAAkB,OAAO,CAAC,IAAI,UAAU;YAC/C,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,eAAe;YAC1B,QAAQ,EAAE,aAAa;YACvB,UAAU,EAAE,oBAAoB;YAChC,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,CAAC,OAAO,CAAC;YACvB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,YAAY,EAAE,CAAC,SAAS,CAAC;YACzB,uBAAuB,EAAE,KAAK;YAC9B,OAAO,EAAE,SAAS;YAClB,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC;YACxC,OAAO,EAAE,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC;YACrC,UAAU,EAAE,WAAW;YACvB,UAAU,EAAE,CAAC;YACb,QAAQ,EAAE,OAAO;SAClB,CAAA;QACD,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,MAAM,EAAE,aAAa,EAAE,MAAM,CAAC,CAAA;QAC/D,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACpB,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;QAClE,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;QAE7D,MAAM,QAAQ,GAAuC,EAAE,CAAA;QACvD,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAA;QAC7C,MAAM,CAAC,KAAK,EAAE,CAAA;QACd,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,UAAU,CAAqD,CAAA;QAChG,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,IAAI,GAAG,EAAE,gBAAgB,CAAC,CAAA;QAClE,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;QACrE,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;QAClE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QAC5D,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAgC,CAAC,CAAC,CAAA;QAEzG,MAAM,QAAQ,GAAG;YACf,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,UAAU;YAClB,OAAO,EAAE,WAAW;YACpB,UAAU,EAAE,OAAO;YACnB,UAAU,EAAE,sCAAsC;YAClD,MAAM,EAAE,aAAa;YACrB,OAAO,EAAE,SAAS;YAClB,MAAM,EAAE,EAAE;SACX,CAAA;QACD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;QACrC,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB,IAAI,KAAK,CAAC,KAAK,KAAK,kBAAkB,CAAC,CAAC,CAAA;QACvH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;QACrC,MAAM,SAAS,CACb,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB,IAAI,KAAK,CAAC,KAAK,KAAK,kBAAkB,CAAC,CAAC,MAAM,KAAK,CAAC,CACtH,CAAA;QACD,MAAM,UAAU,CAAC,WAAW,EAAE,CAAA;QAE9B,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC1B,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,EAA0B,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/H,gBAAgB;YAChB,iBAAiB;YACjB,qBAAqB;YACrB,gBAAgB;YAChB,kBAAkB;YAClB,gBAAgB;YAChB,kBAAkB;YAClB,WAAW;SACZ,CAAC,CAAA;QACF,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,EAAwB,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,eAAe,CAAC,CAAA;QAC1F,MAAM,CAAC,GAAG,CAAC,CAAC,aAAa,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAA;QAC7C,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,CAAC,CAAA;QAC5C,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,EAA0B,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAA;QACnG,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;QACxD,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,kBAAkB,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;IACxF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,kBAAkB,CAAC,CAAC,CAAA;QACnE,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAA;QAC3D,MAAM,KAAK,CAAC,UAAU,EAAE,CAAA;QACxB,MAAM,aAAa,GAAa,EAAE,CAAA;QAClC,MAAM,WAAW,GAAyB,EAAE,CAAA;QAC5C,MAAM,QAAQ,GAAoB;YAChC,KAAK,CAAC,OAAO;gBACX,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAA;YACjD,CAAC;YACD,SAAS,CAAC,KAAK;gBACb,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;gBACzB,OAAO,IAAI,CAAA;YACb,CAAC;SACF,CAAA;QACD,MAAM,MAAM,GAAgB;YAC1B,SAAS,EAAE,KAAK,IAAI,EAAE,CAAC,SAAS;YAChC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,SAAS;YAC9B,cAAc,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,GAAG,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA,CAAC,CAAC;SACzD,CAAA;QACD,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;QAElE,UAAU,CAAC,MAAM,CAAC;YAChB,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,eAAe;YACvB,OAAO,EAAE,WAAW;YACpB,UAAU,EAAE,OAAO;YACnB,MAAM,EAAE,YAAY;YACpB,MAAM,EAAE,EAAE;SACX,CAAC,CAAA;QAEF,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,CAAC,CAAA;QAC/C,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,CAAC,CAAA;QAChD,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YACnC,MAAM,EAAE,eAAe;YACvB,MAAM,EAAE,YAAY;YACpB,MAAM,EAAE,MAAM;SACf,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAA;QACpE,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAA;QAC3D,MAAM,KAAK,CAAC,UAAU,EAAE,CAAA;QACxB,MAAM,gBAAgB,GAAkB,EAAE,CAAA;QAC1C,MAAM,QAAQ,GAAoB;YAChC,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK;gBAC1B,MAAM,KAAK,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAA;gBACpD,MAAM,KAAK,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAA;gBACzD,MAAM,KAAK,CAAC,eAAe,CAAC,iBAAiB,EAAE;oBAC7C,QAAQ,EAAE,CAAC;oBACX,SAAS,EAAE,iDAAiD;oBAC5D,UAAU,EAAE,EAAE;iBACf,CAAC,CAAA;gBACF,OAAO;oBACL,KAAK,EAAE,OAAO,CAAC,MAAM;oBACrB,SAAS,EAAE,iBAAiB;oBAC5B,UAAU,EAAE,oBAAoB;oBAChC,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,KAAK;iBACjB,CAAA;YACH,CAAC;SACF,CAAA;QACD,MAAM,MAAM,GAAgB;YAC1B,SAAS,EAAE,KAAK,IAAI,EAAE,CAAC,SAAS;YAChC,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,GAAG,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA,CAAC,CAAC;YACtD,cAAc,EAAE,KAAK,IAAI,EAAE,CAAC,SAAS;SACtC,CAAA;QACD,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;QAElE,UAAU,CAAC,MAAM,CAAC;YAChB,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,aAAa;YACrB,OAAO,EAAE,cAAc;YACvB,UAAU,EAAE,OAAO;YACnB,UAAU,EAAE,sCAAsC;YAClD,MAAM,EAAE,aAAa;YACrB,OAAO,EAAE,SAAS;YAClB,MAAM,EAAE,EAAE;SACX,CAAC,CAAA;QAEF,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC,CAAC,CAAA;QACpD,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YACxC,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,CAAC;YACZ,MAAM,EAAE,iDAAiD;SAC1D,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,KAAK,UAAU,SAAS,CAAC,SAAwB;IAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAA;IACnC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;QACpB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAA;QACnF,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAA;IACzD,CAAC;AACH,CAAC"}