@router-agent/runtime 0.3.31 → 0.3.33

package/dist/runtimes/acp/index.js

@@ -1,633 +0,0 @@
-import { appendFileSync, mkdirSync, readFileSync, renameSync, statSync, writeFileSync } from 'node:fs';
-import { join } from 'node:path';
-import { realRunner } from '@router-agent/infra/process.js';
-import { GitWorktreeProvider } from '../../workspaces/git-worktree.js';
-import { workBranch, worktreeName } from '../../naming.js';
-import { WorkerRuntimeError, } from '../../port.js';
-import { latestActivityAt } from '../../status.js';
-import { AcpConnection, AcpProtocolError } from './protocol.js';
-import { decidePermission } from './permissions.js';
-const CLIENT_NAME = 'router-agent';
-const CLIENT_VERSION = '0.1.0';
-const CHANNEL = 'acp';
-const READ_ONLY_TOOLS = { fileAccess: 'read-only', shell: 'deny', allowNetwork: false };
-/**
- * ACP（Agent Client Protocol）运行时。
- *
- * 客户端侧实现：spawn 一个 ACP Agent 子进程，通过 stdio JSON-RPC 驱动
- * initialize → session/new → session/prompt，并处理 session/update 与权限请求。
- * 工作区仍由 GitWorktreeProvider 负责，与 Claude/Orca 运行时保持一致。
- */
-export class AcpRuntime {
-    id;
-    capabilities = {
-        agent: { liveStatus: true, processExit: true, output: false, activity: true, structuredInput: true, nudge: false },
-        workspace: { cleanup: true, prLink: false, discovery: false },
-    };
-    command;
-    args;
-    env;
-    workersDir;
-    home;
-    initializeTimeoutMs;
-    unavailableReason;
-    installHint;
-    probeArgs;
-    gitProvider;
-    workspaces;
-    sessions = new Map();
-    bySessionId = new Map();
-    constructor(opts) {
-        this.id = opts.id ?? 'acp';
-        this.command = opts.command?.trim() || undefined;
-        this.args = opts.args ?? [];
-        this.env = opts.env ?? {};
-        this.workersDir = opts.workersDir;
-        this.home = opts.home;
-        this.initializeTimeoutMs = opts.initializeTimeoutMs ?? 30_000;
-        this.unavailableReason = opts.unavailableReason;
-        this.installHint = opts.installHint;
-        this.probeArgs = opts.probeArgs ?? ['--version'];
-        this.gitProvider = new GitWorktreeProvider({
-            git: opts.git,
-            worktreesRoot: opts.worktreesRoot ?? join(opts.home, 'worktrees'),
-        });
-        this.workspaces = opts.workspaceProvider ?? this.gitProvider;
-    }
-    async start(ctx) {
-        if (!this.command) {
-            throw new WorkerRuntimeError('runtime:unavailable', this.unavailableReason ?? `ACP Agent ${this.id} 不可用`, {
-                retryable: false,
-                detail: {
-                    runtimeId: this.id,
-                    ...(this.unavailableReason ? { reason: this.unavailableReason } : {}),
-                    ...(this.installHint ? { hint: this.installHint } : {}),
-                },
-            });
-        }
-        const ticket = ctx.ticket;
-        const repo = ticket.repo;
-        const baseBranch = ticket.branch;
-        if (!repo)
-            throw new WorkerRuntimeError('workspace:missing-repo', '工单没有指定仓库，拒绝创建 worktree');
-        if (!baseBranch)
-            throw new WorkerRuntimeError('workspace:missing-branch', '工单没有指定起始分支，拒绝创建 worktree');
-        const branch = workBranch(ticket);
-        const name = worktreeName(ticket);
-        const workdir = ctx.workdir || join(this.home, 'worktrees', name);
-        const dir = join(this.workersDir, ticket.id);
-        mkdirSync(dir, { recursive: true });
-        const briefPath = join(dir, 'brief.md');
-        const workspace = await this.workspaces.create({
-            ticketId: ticket.id,
-            repo,
-            baseBranch,
-            workBranch: branch,
-            name,
-            workdir,
-        });
-        const contract = (ctx.brief?.contract ?? `执行工单 ${ticket.id}：${ticket.title}`)
-            .replace('(由运行时决定)', workspace.workdir);
-        writeFileSync(briefPath, contract, 'utf8');
-        const meta = {
-            schemaVersion: 3,
-            ticketId: ticket.id,
-            ...(ctx.assignment.id ? { assignmentId: ctx.assignment.id } : {}),
-            runtime: this.id,
-            handle: '',
-            workdir: workspace.workdir,
-            repo,
-            branch: baseBranch,
-            workBranch: workspace.branch,
-            status: 'starting',
-            startedAt: new Date().toISOString(),
-            tools: ctx.tools ?? READ_ONLY_TOOLS,
-        };
-        this.writeMeta(ticket.id, meta);
-        let connection;
-        try {
-            connection = this.spawnConnection(workspace.workdir);
-            const init = await this.initialize(connection);
-            const sessionId = await this.newSession(connection, workspace.workdir);
-            const session = this.registerSession({
-                ticketId: ticket.id,
-                sessionId,
-                workdir: workspace.workdir,
-                branch: workspace.branch,
-                tools: ctx.tools ?? READ_ONLY_TOOLS,
-                loadSession: init.agentCapabilities?.loadSession === true,
-                connection,
-            });
-            meta.handle = sessionId;
-            meta.sessionId = sessionId;
-            meta.loadSession = session.loadSession;
-            if (connection.pid !== undefined)
-                meta.pid = connection.pid;
-            meta.status = 'running';
-            this.writeMeta(ticket.id, meta);
-            this.prompt(session, `${contract}\n\n完成后用 ra report 上报。`);
-            return {
-                handle: sessionId,
-                workdir: workspace.workdir,
-                branch: workspace.branch,
-                sessionId,
-                workspaceRef: workspace.workspaceRef,
-            };
-        }
-        catch (error) {
-            await connection?.close().catch(() => { });
-            await this.rollback(workspace.workdir, workspace.branch, repo);
-            throw this.launchError(error, workspace.workdir, workspace.branch);
-        }
-    }
-    async deliver(ref, inputs, opts = {}) {
-        const inputIds = inputs.map(input => input.id);
-        if (inputs.length === 0)
-            return { status: 'accepted', inputIds };
-        let meta = this.readMeta(ref.ticketId);
-        let session = this.sessions.get(ref.ticketId);
-        if (session && !session.connection.alive) {
-            this.dropSession(session);
-            session = undefined;
-        }
-        if (!session) {
-            const revived = await this.revive(meta).catch(() => null);
-            if (!revived) {
-                return { status: 'rejected', inputIds, reason: 'acp:agent-gone', retryable: true, retryAfterMs: 30_000 };
-            }
-            session = revived;
-            meta = this.readMeta(ref.ticketId) ?? meta;
-        }
-        if (session.inFlight) {
-            return { status: 'queued', inputIds, reason: 'acp:busy' };
-        }
-        if (opts.redeliver !== true
-            && meta?.lastHandedOffInputId === inputs.at(-1)?.id
-            && meta?.lastHandedOffAttempt === (opts.attempt ?? 1)) {
-            return { status: 'accepted', inputIds };
-        }
-        const text = inputs.map(input => `【Router 输入 ${input.id}】\n${input.text}`).join('\n\n');
-        try {
-            this.prompt(session, `${text}\n\n继续执行工单，完成后用 ra report 上报。`);
-        }
-        catch (error) {
-            return {
-                status: 'rejected',
-                inputIds,
-                reason: error instanceof WorkerRuntimeError ? error.reason : 'acp:deliver-failed',
-                retryable: true,
-                retryAfterMs: 30_000,
-            };
-        }
-        if (meta) {
-            meta.lastHandedOffInputId = inputs.at(-1)?.id;
-            meta.lastHandedOffAttempt = opts.attempt ?? 1;
-            meta.status = 'running';
-            this.writeMeta(ref.ticketId, meta);
-        }
-        return { status: 'accepted', inputIds };
-    }
-    async stop(ref, _reason) {
-        const session = this.sessions.get(ref.ticketId);
-        if (session) {
-            try {
-                session.connection.notify('session/cancel', { sessionId: session.sessionId });
-            }
-            catch {
-                // 进程已退出时 cancel 无意义，继续清理。
-            }
-            this.dropSession(session);
-            await session.connection.close().catch(() => { });
-        }
-        const meta = this.readMeta(ref.ticketId);
-        if (meta) {
-            meta.status = 'stopped';
-            this.writeMeta(ref.ticketId, meta);
-        }
-    }
-    /** 重启对账用：把磁盘上仍可 load 的会话直接接回来，失败返回 false 交给上层兜底。 */
-    async reattach(ref) {
-        const session = this.sessions.get(ref.ticketId);
-        if (session?.connection.alive)
-            return true;
-        if (session)
-            this.dropSession(session);
-        return (await this.revive(this.readMeta(ref.ticketId))) !== null;
-    }
-    async status(ref) {
-        const session = this.sessions.get(ref.ticketId);
-        const meta = this.readMeta(ref.ticketId);
-        if (!session) {
-            if (!meta)
-                return null;
-            // 重启后内存会话已丢，用活动日志的落盘时间估算最后活动，别拿启动时间冒充。
-            const lastActivityAt = this.lastEventAtFromFile(ref.ticketId) ?? meta.startedAt;
-            return {
-                alive: false,
-                phase: 'exited',
-                acceptingInput: meta.loadSession === true && Boolean(meta.sessionId),
-                ...(lastActivityAt ? { lastActivityAt } : {}),
-                ...(meta.lastError ? { detail: { source: this.id, error: meta.lastError } } : { detail: { source: this.id, status: meta.status } }),
-            };
-        }
-        const lastActivityAt = latestActivityAt(session.lastEventAt, session.promptStartedAt) ?? meta?.startedAt;
-        if (!session.connection.alive) {
-            return {
-                alive: false,
-                phase: 'exited',
-                acceptingInput: session.loadSession,
-                ...(lastActivityAt ? { lastActivityAt } : {}),
-                ...(session.connection.exit.code !== null ? { exitCode: session.connection.exit.code } : {}),
-                detail: { source: this.id, status: 'exited' },
-            };
-        }
-        if (session.inFlight) {
-            return {
-                alive: true,
-                phase: 'running',
-                acceptingInput: false,
-                ...(lastActivityAt ? { lastActivityAt } : {}),
-                detail: { source: this.id, status: 'working' },
-            };
-        }
-        return {
-            alive: true,
-            phase: 'running',
-            acceptingInput: true,
-            ...(lastActivityAt ? { lastActivityAt } : {}),
-            detail: {
-                source: this.id,
-                status: session.lastError ? 'error' : 'idle',
-                ...(session.lastError ? { error: session.lastError } : {}),
-                ...(session.lastStopReason ? { stopReason: session.lastStopReason } : {}),
-            },
-        };
-    }
-    async activity(query) {
-        const meta = this.readMeta(query.ref.ticketId);
-        const session = this.sessions.get(query.ref.ticketId);
-        if (!meta && !session)
-            return null;
-        const events = this.readEvents(query.ref.ticketId);
-        const max = query.opts?.maxEvents ?? 40;
-        const tail = events.slice(-max);
-        const sessionId = query.ref.sessionId ?? session?.sessionId ?? meta?.sessionId;
-        const at = tail.at(-1)?.at ?? session?.lastEventAt ?? meta?.startedAt ?? new Date().toISOString();
-        return { at, events: tail, ...(sessionId ? { sessionId } : {}) };
-    }
-    async inspectWorkspace(ref) {
-        return this.workspaces.inspect(ref.workdir);
-    }
-    async removeWorkspace(ref) {
-        const meta = this.readMeta(ref.ticketId);
-        await this.workspaces.remove(ref.workdir, { repo: ref.repo ?? meta?.repo });
-    }
-    async repoSlug(ref) {
-        if (!ref.workdir)
-            return null;
-        return this.gitProvider.remoteRepo(ref.workdir);
-    }
-    async doctor() {
-        const checks = [];
-        if (!this.command) {
-            checks.push({
-                name: `ACP Agent (${this.id})`,
-                status: 'fail',
-                detail: this.unavailableReason ?? '未找到可执行文件',
-                hint: this.installHint ?? `安装该 Agent 后重启服务，或在 runtime.options.${this.id}.command 指定可执行文件`,
-            });
-        }
-        else {
-            const version = await realRunner(this.command).run(this.probeArgs, {
-                cwd: this.home,
-                env: { ...process.env, ...this.env },
-                timeoutMs: 5_000,
-            });
-            checks.push({
-                name: `ACP Agent (${this.id})`,
-                status: version.code === 0 ? 'ok' : 'fail',
-                detail: version.code === 0
-                    ? version.stdout.trim().split('\n')[0] ?? this.command
-                    : `${version.stderr || version.stdout}`.trim().slice(0, 160),
-                ...(version.code === 0 ? {} : { hint: this.installHint ?? `确认 ${this.command} 可执行；命令与参数由 runtime.options.${this.id}.command/args 配置` }),
-            });
-        }
-        const git = await this.gitProvider.version();
-        checks.push({
-            name: 'git worktree',
-            status: git === null ? 'fail' : 'ok',
-            detail: git ?? 'git --version 不可用',
-            ...(git === null ? { hint: '安装 git，并确认 git worktree 可用' } : {}),
-        });
-        return checks;
-    }
-    spawnConnection(cwd) {
-        if (!this.command) {
-            throw new WorkerRuntimeError('runtime:unavailable', this.unavailableReason ?? `ACP Agent ${this.id} 不可用`);
-        }
-        return new AcpConnection({
-            command: this.command,
-            args: this.args,
-            cwd,
-            env: { ...process.env, ...this.env },
-            onNotification: (method, params) => this.onNotification(method, params),
-            onRequest: (method, params) => this.onRequest(method, params),
-            onExit: (code, signal, stderr) => this.onExit(cwd, code, signal, stderr),
-        });
-    }
-    async initialize(connection) {
-        const init = await connection.request('initialize', {
-            protocolVersion: 1,
-            clientCapabilities: {
-                fs: { readTextFile: false, writeTextFile: false },
-                terminal: false,
-            },
-            clientInfo: { name: CLIENT_NAME, title: 'Router Agent', version: CLIENT_VERSION },
-        }, this.initializeTimeoutMs);
-        if (Number(init?.protocolVersion) !== 1) {
-            throw new WorkerRuntimeError('agent:launch-failed', `ACP Agent 协议版本不兼容: ${init?.protocolVersion ?? 'unknown'}`, {
-                detail: { protocolVersion: init?.protocolVersion },
-            });
-        }
-        return init;
-    }
-    async newSession(connection, workdir) {
-        try {
-            const result = await connection.request('session/new', {
-                cwd: workdir,
-                mcpServers: [],
-            }, this.initializeTimeoutMs);
-            if (!result?.sessionId) {
-                throw new WorkerRuntimeError('agent:launch-failed', 'ACP session/new 没有返回 sessionId');
-            }
-            return result.sessionId;
-        }
-        catch (error) {
-            if (error instanceof AcpProtocolError && error.code === -32000) {
-                throw new WorkerRuntimeError('agent:launch-failed', `ACP Agent 未完成认证: ${error.message}`, {
-                    detail: { code: error.code, data: error.data },
-                });
-            }
-            throw error;
-        }
-    }
-    registerSession(input) {
-        const session = { ...input, inFlight: false, suppressUpdates: false };
-        this.sessions.set(session.ticketId, session);
-        this.bySessionId.set(session.sessionId, session);
-        return session;
-    }
-    prompt(session, text) {
-        if (session.inFlight) {
-            throw new WorkerRuntimeError('agent:not-accepting-input', 'ACP 会话仍在执行上一轮');
-        }
-        session.inFlight = true;
-        session.promptStartedAt = new Date().toISOString();
-        session.lastError = undefined;
-        session.connection
-            .request('session/prompt', {
-            sessionId: session.sessionId,
-            prompt: [{ type: 'text', text }],
-        }, 0)
-            .then(result => {
-            if (this.bySessionId.get(session.sessionId) !== session)
-                return;
-            session.inFlight = false;
-            session.promptStartedAt = undefined;
-            session.lastStopReason = result?.stopReason;
-            session.lastEventAt = new Date().toISOString();
-        })
-            .catch(error => {
-            if (this.bySessionId.get(session.sessionId) !== session)
-                return;
-            session.inFlight = false;
-            session.promptStartedAt = undefined;
-            session.lastError = error instanceof Error ? error.message : String(error);
-            this.writeMetaStatus(session.ticketId, { status: 'error', lastError: session.lastError });
-        });
-    }
-    async revive(meta) {
-        if (!meta?.sessionId || meta.loadSession !== true)
-            return null;
-        const connection = this.spawnConnection(meta.workdir);
-        let session;
-        try {
-            const init = await this.initialize(connection);
-            if (init.agentCapabilities?.loadSession !== true) {
-                throw new WorkerRuntimeError('agent:launch-failed', 'ACP Agent 未声明 loadSession，无法恢复会话');
-            }
-            session = this.registerSession({
-                ticketId: meta.ticketId,
-                sessionId: meta.sessionId,
-                workdir: meta.workdir,
-                branch: meta.workBranch,
-                tools: meta.tools ?? READ_ONLY_TOOLS,
-                loadSession: true,
-                connection,
-            });
-            session.suppressUpdates = true;
-            try {
-                await connection.request('session/load', {
-                    sessionId: meta.sessionId,
-                    cwd: meta.workdir,
-                    mcpServers: [],
-                }, this.initializeTimeoutMs);
-            }
-            finally {
-                session.suppressUpdates = false;
-            }
-            meta.status = 'running';
-            if (connection.pid !== undefined)
-                meta.pid = connection.pid;
-            this.writeMeta(meta.ticketId, meta);
-            // 重连本身算一次活动，避免用重启前的 startedAt 立即判 idle。
-            session.lastEventAt = new Date().toISOString();
-            return session;
-        }
-        catch (error) {
-            if (session)
-                this.dropSession(session);
-            await connection.close().catch(() => { });
-            return null;
-        }
-    }
-    dropSession(session) {
-        this.sessions.delete(session.ticketId);
-        this.bySessionId.delete(session.sessionId);
-    }
-    onNotification(method, params) {
-        if (method !== 'session/update')
-            return;
-        const payload = params;
-        const session = payload?.sessionId ? this.bySessionId.get(payload.sessionId) : undefined;
-        if (!session || session.suppressUpdates)
-            return;
-        const update = payload?.update;
-        if (!update)
-            return;
-        const kind = String(update.sessionUpdate ?? '');
-        const at = new Date().toISOString();
-        if (kind === 'agent_message_chunk') {
-            const text = contentText(update.content);
-            if (text)
-                this.recordEvent(session, { at, kind: 'text', text });
-            return;
-        }
-        if (kind === 'tool_call') {
-            const title = typeof update.title === 'string' ? update.title : undefined;
-            const toolKind = typeof update.kind === 'string' ? update.kind : 'tool';
-            const target = toolTarget(update);
-            this.recordEvent(session, {
-                at,
-                kind: 'tool',
-                text: title ?? toolKind,
-                tool: { name: title ?? toolKind, ...(target ? { target } : {}) },
-            });
-        }
-        // tool_call_update / plan / usage 等只更新状态，不重复写入活动。
-    }
-    async onRequest(method, params) {
-        if (method === 'session/request_permission') {
-            const payload = params;
-            const session = payload?.sessionId ? this.bySessionId.get(payload.sessionId) : undefined;
-            const tools = session?.tools ?? READ_ONLY_TOOLS;
-            const options = Array.isArray(payload?.options) ? payload.options : [];
-            const choice = payload?.toolCall ? decidePermission(tools, payload.toolCall, options) : null;
-            if (!choice)
-                return { outcome: { outcome: 'cancelled' } };
-            return { outcome: { outcome: 'selected', optionId: choice.optionId } };
-        }
-        throw new AcpProtocolError(-32601, `ACP 客户端不支持该方法: ${method}`);
-    }
-    onExit(cwd, code, signal, stderr) {
-        for (const session of this.sessions.values()) {
-            if (session.workdir !== cwd || session.connection.alive)
-                continue;
-            session.inFlight = false;
-            session.promptStartedAt = undefined;
-            session.lastError = `ACP agent 退出（code=${code ?? 'null'}, signal=${signal ?? 'null'}）${stderr ? `: ${stderr.slice(-300)}` : ''}`;
-            this.writeMetaStatus(session.ticketId, { status: 'exited', lastError: session.lastError });
-        }
-    }
-    recordEvent(session, event) {
-        session.lastEventAt = event.at;
-        try {
-            const dir = join(this.workersDir, session.ticketId);
-            mkdirSync(dir, { recursive: true });
-            appendFileSync(join(dir, `${CHANNEL}-events.jsonl`), `${JSON.stringify(event)}\n`, 'utf8');
-        }
-        catch {
-            // 活动落盘失败不应影响主流程；内存里仍可从文件缺失时降级。
-        }
-    }
-    readEvents(ticketId) {
-        try {
-            const text = readFileSync(join(this.workersDir, ticketId, `${CHANNEL}-events.jsonl`), 'utf8');
-            const events = [];
-            for (const line of text.split('\n')) {
-                if (!line.trim())
-                    continue;
-                try {
-                    const parsed = JSON.parse(line);
-                    if (parsed && typeof parsed.at === 'string' && (parsed.kind === 'text' || parsed.kind === 'tool')) {
-                        events.push(parsed);
-                    }
-                }
-                catch {
-                    // 跳过损坏行。
-                }
-            }
-            return events;
-        }
-        catch {
-            return [];
-        }
-    }
-    /** 活动日志的最后落盘时间；文件不存在时返回 undefined。 */
-    lastEventAtFromFile(ticketId) {
-        try {
-            return statSync(join(this.workersDir, ticketId, `${CHANNEL}-events.jsonl`)).mtime.toISOString();
-        }
-        catch {
-            return undefined;
-        }
-    }
-    async rollback(workdir, branch, repo) {
-        await this.workspaces.remove(workdir, { repo }).catch(() => { });
-        if (this.workspaces.deleteBranch) {
-            await this.workspaces.deleteBranch(branch, repo).catch(() => { });
-        }
-    }
-    launchError(error, workdir, branch) {
-        if (error instanceof WorkerRuntimeError) {
-            return new WorkerRuntimeError(error.code, error.reason, {
-                retryable: error.retryable,
-                cause: error.cause,
-                detail: { ...error.detail, workdir, branch, command: this.command },
-            });
-        }
-        const message = error instanceof Error ? error.message : String(error);
-        return new WorkerRuntimeError('agent:launch-failed', `ACP Agent 启动失败: ${message}`, {
-            cause: error instanceof Error ? error.stack : undefined,
-            detail: { workdir, branch, command: this.command },
-        });
-    }
-    metaPath(ticketId) {
-        return join(this.workersDir, ticketId, 'meta.json');
-    }
-    readMeta(ticketId) {
-        try {
-            return JSON.parse(readFileSync(this.metaPath(ticketId), 'utf8'));
-        }
-        catch {
-            return null;
-        }
-    }
-    writeMeta(ticketId, meta) {
-        const path = this.metaPath(ticketId);
-        mkdirSync(join(this.workersDir, ticketId), { recursive: true });
-        const tmp = `${path}.tmp`;
-        writeFileSync(tmp, `${JSON.stringify(meta, null, 2)}\n`, 'utf8');
-        renameSync(tmp, path);
-    }
-    writeMetaStatus(ticketId, patch) {
-        const meta = this.readMeta(ticketId);
-        if (!meta)
-            return;
-        this.writeMeta(ticketId, { ...meta, ...patch });
-    }
-}
-function contentText(content) {
-    if (typeof content === 'string')
-        return content;
-    if (Array.isArray(content))
-        return content.map(contentText).join('');
-    if (content && typeof content === 'object') {
-        const record = content;
-        if (record.type === 'text' && typeof record.text === 'string')
-            return record.text;
-    }
-    return '';
-}
-function toolTarget(update) {
-    const locations = update.locations;
-    if (Array.isArray(locations)) {
-        for (const item of locations) {
-            if (item && typeof item === 'object' && typeof item.path === 'string') {
-                return item.path;
-            }
-        }
-    }
-    const raw = update.rawInput;
-    if (raw && typeof raw === 'object') {
-        const record = raw;
-        if (typeof record.command === 'string')
-            return record.command;
-        if (typeof record.file_path === 'string')
-            return record.file_path;
-        if (typeof record.path === 'string')
-            return record.path;
-    }
-    return undefined;
-}
-//# sourceMappingURL=index.js.map