c0de-agent 0.0.0 → 1.0.0

package/dist/cli/index.js

@@ -0,0 +1,149 @@
+#!/usr/bin/env node
+// src/cli/index.ts — c0de CLI bin entry.
+import { fileURLToPath } from 'node:url';
+import { parseArgs } from 'node:util';
+import { loadConfig } from '../core/config.js';
+import { createDB, migrateDB } from '../db/index.js';
+import { runAcpCommand } from './commands/acp.js';
+import { runChatCommand } from './commands/chat.js';
+import { runConfigCommand } from './commands/config.js';
+import { runInitCommand } from './commands/init.js';
+import { runPluginCommand } from './commands/plugin.js';
+import { runServeCommand } from './commands/serve.js';
+import { runUpdateCommand } from './commands/update.js';
+import { buildAgentDeps } from './deps.js';
+const COMMANDS = [
+    {
+        name: 'chat',
+        description: 'Ask a one-shot question; print the answer.',
+        options: [
+            { name: 'model', type: 'string' },
+            { name: 'format', type: 'string' },
+            { name: 'yes', type: 'boolean', short: 'y' },
+        ],
+    },
+    {
+        name: 'serve',
+        description: 'Start the HTTP server (default when no command given).',
+        options: [
+            { name: 'port', type: 'string' },
+            { name: 'open', type: 'boolean' },
+            { name: 'restore', type: 'string' },
+            { name: 'handoff-port', type: 'string' },
+        ],
+    },
+    {
+        name: 'init',
+        description: 'Create a .c0de/config.json in the current project.',
+        options: [{ name: 'force', type: 'boolean', short: 'f' }],
+    },
+    {
+        name: 'config',
+        description: 'Get or set configuration values.',
+        options: [{ name: 'global', type: 'boolean', short: 'g' }],
+    },
+    {
+        name: 'plugin',
+        description: 'List or enable plugins.',
+        options: [],
+    },
+    {
+        name: 'acp',
+        description: 'Run in Agent Client Protocol mode (editor integration).',
+        options: [],
+    },
+    {
+        name: 'update',
+        description: 'Check npm registry for a newer version (--apply to hot-update).',
+        options: [{ name: 'apply', type: 'boolean' }],
+    },
+];
+async function dispatch(argv, overrides = {}) {
+    const [command, ...rest] = argv;
+    const name = command ?? 'serve';
+    if (name === 'serve' && overrides.runServe) {
+        await overrides.runServe();
+        return;
+    }
+    const spec = COMMANDS.find((c) => c.name === name);
+    if (!spec)
+        throw new Error(`cli: unknown command "${name}"`);
+    const { values, positionals } = parseArgs({
+        options: Object.fromEntries((spec.options ?? []).map((o) => [
+            o.name,
+            { type: o.type, ...(o.short ? { short: o.short } : {}) },
+        ])),
+        args: rest,
+        allowPositionals: true,
+        strict: true,
+    });
+    const args = { options: values, positionals };
+    const cwd = process.cwd();
+    switch (name) {
+        case 'serve': {
+            await runServeCommand({ args, cwd });
+            return;
+        }
+        case 'chat': {
+            const config = await loadConfig(cwd);
+            const db = await createDB({ driver: 'pglite' });
+            await migrateDB(db);
+            try {
+                const deps = await buildAgentDeps(config, { db, cwd });
+                await runChatCommand({ args, config, deps });
+            }
+            finally {
+                await db.close();
+            }
+            return;
+        }
+        case 'init': {
+            await runInitCommand({ args, cwd });
+            return;
+        }
+        case 'config': {
+            await runConfigCommand({ args, cwd });
+            return;
+        }
+        case 'plugin': {
+            await runPluginCommand({ args, cwd });
+            return;
+        }
+        case 'acp': {
+            const config = await loadConfig(cwd);
+            const db = await createDB({ driver: 'pglite' });
+            await migrateDB(db);
+            try {
+                const deps = await buildAgentDeps(config, { db, cwd });
+                await runAcpCommand({ config, deps });
+            }
+            finally {
+                await db.close();
+            }
+            return;
+        }
+        case 'update': {
+            await runUpdateCommand({ args, cwd });
+            return;
+        }
+        default: {
+            throw new Error(`cli: unknown command "${name}"`);
+        }
+    }
+}
+async function main() {
+    const argv = process.argv.slice(2);
+    try {
+        await dispatch(argv);
+    }
+    catch (err) {
+        process.stderr.write(`${err instanceof Error ? err.message : String(err)}\n`);
+        process.exit(1);
+    }
+}
+export { COMMANDS, dispatch };
+// bin 入口：仅在直接执行时运行 main（非被 import）。
+const isMain = process.argv[1] === fileURLToPath(import.meta.url);
+if (isMain) {
+    void main();
+}

package/dist/cli/modes/acp.d.ts

@@ -0,0 +1,18 @@
+type ACPRequest = {
+    id: number | string | null;
+    method: string;
+    params: Record<string, unknown>;
+};
+type ACPHandler = (params: Record<string, unknown>) => Promise<Record<string, unknown>>;
+type AcpLoopOptions = {
+    reader: AsyncGenerator<string>;
+    writer: (line: string) => void;
+    handlers: Record<string, ACPHandler>;
+};
+declare function parseACPRequest(line: string): ACPRequest | null;
+declare function formatACPResponse(id: number | string | null, result: Record<string, unknown>): string;
+declare function formatACPError(id: number | string | null, code: number, message: string): string;
+declare function formatACPEvent(method: string, params: Record<string, unknown>): string;
+declare function runAcpLoop(opts: AcpLoopOptions): Promise<void>;
+export type { ACPHandler, ACPRequest, AcpLoopOptions };
+export { formatACPError, formatACPEvent, formatACPResponse, parseACPRequest, runAcpLoop };

package/dist/cli/modes/acp.js

@@ -0,0 +1,57 @@
+function parseACPRequest(line) {
+    let obj;
+    try {
+        obj = JSON.parse(line);
+    }
+    catch {
+        return null;
+    }
+    if (obj === null || typeof obj !== 'object')
+        return null;
+    const o = obj;
+    if (typeof o.method !== 'string')
+        return null;
+    return {
+        id: (o.id ?? null),
+        method: o.method,
+        params: (o.params ?? {}),
+    };
+}
+function formatACPResponse(id, result) {
+    return JSON.stringify({ jsonrpc: '2.0', id, result });
+}
+function formatACPError(id, code, message) {
+    return JSON.stringify({ jsonrpc: '2.0', id, error: { code, message } });
+}
+function formatACPEvent(method, params) {
+    return JSON.stringify({ jsonrpc: '2.0', method, params });
+}
+async function runAcpLoop(opts) {
+    for await (const line of opts.reader) {
+        const trimmed = line.trim();
+        if (!trimmed)
+            continue;
+        const req = parseACPRequest(trimmed);
+        if (!req) {
+            opts.writer(formatACPError(null, -32700, 'Parse error'));
+            continue;
+        }
+        const handler = opts.handlers[req.method];
+        if (!handler) {
+            if (req.id !== null)
+                opts.writer(formatACPError(req.id, -32601, `Method not found: ${req.method}`));
+            continue;
+        }
+        try {
+            const result = await handler(req.params);
+            if (req.id !== null)
+                opts.writer(formatACPResponse(req.id, result));
+        }
+        catch (err) {
+            if (req.id !== null) {
+                opts.writer(formatACPError(req.id, -32603, err instanceof Error ? err.message : String(err)));
+            }
+        }
+    }
+}
+export { formatACPError, formatACPEvent, formatACPResponse, parseACPRequest, runAcpLoop };

package/dist/cli/modes/print.d.ts

@@ -0,0 +1,16 @@
+import type { LoopDeps } from '../../core/loop.js';
+import type { AgentEvent } from '../../shared/types/agent.js';
+import type { Config } from '../../shared/types/config.js';
+type PrintOptions = {
+    model?: string;
+    format?: 'text' | 'json';
+    maxTokens?: number;
+    /** 事件观察回调（CLI 用于把 tool/thinking 写到 stderr）。 */
+    onEvent?: (event: AgentEvent) => void;
+};
+/** 从事件流累积 assistant 文本。纯函数。 */
+declare function collectAssistantText(events: AgentEvent[]): string;
+/** Print 模式：临时会话 → agent → 收集文本 → 返回。 */
+declare function runPrintMode(config: Config, message: string, deps: LoopDeps, opts?: PrintOptions): Promise<string>;
+export type { PrintOptions };
+export { collectAssistantText, runPrintMode };

package/dist/cli/modes/print.js

@@ -0,0 +1,30 @@
+import { createAgent, runAgent } from '../../core/agent.js';
+import { createSession } from '../../session/session.js';
+/** 从事件流累积 assistant 文本。纯函数。 */
+function collectAssistantText(events) {
+    let text = '';
+    for (const e of events) {
+        if (e._tag === 'text_delta')
+            text += e.text;
+    }
+    return text;
+}
+/** Print 模式：临时会话 → agent → 收集文本 → 返回。 */
+async function runPrintMode(config, message, deps, opts = {}) {
+    const session = await createSession(deps.db, 'cli-print');
+    const agentConfig = {
+        provider: config.defaultProvider,
+        model: opts.model ?? config.defaultModel,
+        tools: config.tools.enabled,
+        plugins: config.plugins.enabled,
+        ...(opts.maxTokens !== undefined ? { maxTokens: opts.maxTokens } : {}),
+    };
+    const state = await createAgent(session, agentConfig, deps);
+    const events = [];
+    for await (const event of runAgent(state, [{ _tag: 'text', text: message }], deps)) {
+        events.push(event);
+        opts.onEvent?.(event);
+    }
+    return collectAssistantText(events);
+}
+export { collectAssistantText, runPrintMode };

package/dist/cli/parser.d.ts

@@ -0,0 +1,19 @@
+type OptionType = 'string' | 'boolean';
+type CommandOption = {
+    name: string;
+    type: OptionType;
+    short?: string;
+    default?: unknown;
+};
+type CommandSpec = {
+    name: string;
+    description: string;
+    options?: CommandOption[];
+};
+type CommandArgs = {
+    options: Record<string, unknown>;
+    positionals: string[];
+};
+declare function parseCommand(spec: CommandSpec, argv: string[]): CommandArgs;
+export type { CommandArgs, CommandOption, CommandSpec, OptionType };
+export { parseCommand };

package/dist/cli/parser.js

@@ -0,0 +1,28 @@
+import { parseArgs } from 'node:util';
+function parseCommand(spec, argv) {
+    const opts = {};
+    const defaults = {};
+    for (const o of spec.options ?? []) {
+        const key = o.name;
+        opts[key] = { type: o.type, ...(o.short ? { short: o.short } : {}) };
+        if (o.default !== undefined)
+            defaults[key] = o.default;
+    }
+    const { values, positionals } = parseArgs({
+        options: opts,
+        args: argv,
+        allowPositionals: true,
+        strict: true,
+    });
+    const merged = { ...defaults };
+    for (const k of Object.keys(opts)) {
+        if (values[k] !== undefined) {
+            merged[k] = values[k];
+        }
+        else if (opts[k]?.type === 'boolean' && defaults[k] === undefined) {
+            merged[k] = false;
+        }
+    }
+    return { options: merged, positionals };
+}
+export { parseCommand };

package/dist/cli/utils/output.d.ts

@@ -0,0 +1,11 @@
+declare function platformOpener(platform: string): string;
+type SpawnLike = (cmd: string, args: string[]) => unknown;
+type OpenBrowserOptions = {
+    platform?: string;
+    /** 测试注入。 */
+    spawnFn?: SpawnLike;
+};
+declare function openBrowser(url: string, opts?: OpenBrowserOptions): Promise<void>;
+declare function printStartupBanner(url: string, write?: (s: string) => void): void;
+export type { OpenBrowserOptions, SpawnLike };
+export { openBrowser, platformOpener, printStartupBanner };

package/dist/cli/utils/output.js

@@ -0,0 +1,23 @@
+import { spawn } from 'node:child_process';
+function platformOpener(platform) {
+    if (platform === 'darwin')
+        return 'open';
+    if (platform === 'win32')
+        return 'start';
+    return 'xdg-open';
+}
+async function openBrowser(url, opts = {}) {
+    const platform = opts.platform ?? process.platform;
+    const cmd = platformOpener(platform);
+    const doSpawn = opts.spawnFn ??
+        ((c, a) => {
+            const child = spawn(c, a, { detached: true, stdio: 'ignore' });
+            child.unref();
+            return child;
+        });
+    doSpawn(cmd, [url]);
+}
+function printStartupBanner(url, write = process.stderr.write.bind(process.stderr)) {
+    write(`\nc0de-agent server running at ${url}\nOpen this URL in your browser.\n\n`);
+}
+export { openBrowser, platformOpener, printStartupBanner };

package/dist/cli/utils/prompt.d.ts

@@ -0,0 +1,12 @@
+type PromptInterface = {
+    question: (q: string) => Promise<string>;
+    close: () => void;
+};
+type ConfirmOptions = {
+    defaultYes?: boolean;
+    /** 测试注入。 */
+    rl?: PromptInterface;
+};
+declare function confirm(question: string, opts?: ConfirmOptions): Promise<boolean>;
+export type { ConfirmOptions, PromptInterface };
+export { confirm };

package/dist/cli/utils/prompt.js

@@ -0,0 +1,18 @@
+import readline from 'node:readline/promises';
+async function confirm(question, opts = {}) {
+    const defaultYes = opts.defaultYes ?? false;
+    const hint = defaultYes ? '[Y/n]' : '[y/N]';
+    const rl = opts.rl ?? readline.createInterface({ input: process.stdin, output: process.stdout });
+    const own = !opts.rl;
+    try {
+        const answer = (await rl.question(`${question} ${hint} `)).trim().toLowerCase();
+        if (answer === '')
+            return defaultYes;
+        return answer === 'y' || answer === 'yes';
+    }
+    finally {
+        if (own)
+            rl.close();
+    }
+}
+export { confirm };

package/dist/core/agent.d.ts

@@ -0,0 +1,12 @@
+import type { AgentConfig, AgentEvent, AgentState, AgentStatus } from '../shared/types/agent.js';
+import type { MessageContent, Session } from '../shared/types/message.js';
+import type { AgentDependencies } from './types.js';
+declare function createAgent(session: Session, config: AgentConfig, deps: AgentDependencies): Promise<AgentState>;
+declare function runAgent(state: AgentState, userInput: MessageContent[], deps: AgentDependencies): AsyncGenerator<AgentEvent>;
+declare function pauseAgent(state: AgentState): void;
+declare function resumeAgent(state: AgentState): void;
+declare function abortAgent(state: AgentState): void;
+declare function getAgentStatus(state: AgentState): AgentStatus;
+/** 判断 agent 是否处于暂停态（spec §19.2）。 */
+declare function isAgentPaused(state: AgentState): boolean;
+export { abortAgent, createAgent, getAgentStatus, isAgentPaused, pauseAgent, resumeAgent, runAgent };

package/dist/core/agent.js

@@ -0,0 +1,95 @@
+import { appendMessage, getMessages } from '../session/message.js';
+import { getLLMSegments } from '../session/session.js';
+import { generateId } from '../shared/index.js';
+import { listTools } from '../tools/registry.js';
+import { estimateBudget } from './context.js';
+import { agentLoop } from './loop.js';
+import { DEFAULT_SESSION_TITLE, generateSessionTitle } from './title.js';
+async function createAgent(session, config, deps) {
+    const messages = await getMessages(deps.db, session.id);
+    const used = estimateBudget(messages);
+    // 加载会话已有分段：每次 /api/chat 创建新 agent 时，从 DB 恢复完整段历史，
+    // 使本轮新增的 segment/call 追加到既有历史而非覆盖（saveLLMSegments 为全量写回）。
+    const segments = await getLLMSegments(deps.db, session.id);
+    const allTools = listTools(deps.toolRegistry, { config: {}, cwd: deps.cwd });
+    const tools = allTools.filter((t) => config.tools.includes(t.name));
+    return {
+        id: generateId(),
+        session,
+        messages,
+        tools,
+        config,
+        status: { _tag: 'idle' },
+        abortController: new AbortController(),
+        steeringQueue: [],
+        segments,
+        tokenBudget: {
+            total: 128_000,
+            reserved: 25_600,
+            available: 102_400,
+            historyBudget: 76_800,
+            used,
+            keepRecent: 12_800,
+        },
+        calibrationFactor: 1.0,
+    };
+}
+async function* runAgent(state, userInput, deps) {
+    // 幂等重发保护：若 DB 末尾已是相同 user message（服务重启后续传场景），
+    // 跳过 append 避免重复。比较 text parts 内容即可区分不同消息。
+    const last = state.messages[state.messages.length - 1];
+    const isDupUser = last?.role === 'user' &&
+        last.content.length === userInput.length &&
+        last.content.every((p, i) => {
+            const q = userInput[i];
+            return p._tag === 'text' && q?._tag === 'text' ? p.text === q.text : false;
+        });
+    if (!isDupUser) {
+        await appendMessage(deps.db, state.session.id, {
+            role: 'user',
+            content: userInput,
+        });
+    }
+    // 标题生成用纯文本（join text parts），忽略 image/tool 等非文本 part。
+    const titleText = userInput
+        .filter((p) => p._tag === 'text')
+        .map((p) => (p._tag === 'text' ? p.text : ''))
+        .join('');
+    // 第一条用户消息后，后台为会话生成简短标题（fire-and-forget）。
+    // 条件：标题仍是默认占位 + 持久化前无任何消息（即首条消息）。
+    // 失败被吞掉，绝不阻塞主对话流。
+    if (state.session.title === DEFAULT_SESSION_TITLE && state.messages.length === 0) {
+        void generateSessionTitle({
+            db: deps.db,
+            llmRegistry: deps.llmRegistry,
+            config: deps.config,
+            ...(deps.titleChatFn ? { chatFn: deps.titleChatFn } : {}),
+        }, state.session.id, titleText, state.config.provider, state.config.model).catch(() => { });
+    }
+    state.status = { _tag: 'running', turnCount: 0 };
+    yield* agentLoop(state, deps);
+}
+function pauseAgent(state) {
+    if (state.status._tag !== 'running')
+        return;
+    state.status = { _tag: 'paused', pauseReason: 'User requested pause' };
+}
+function resumeAgent(state) {
+    if (state.status._tag !== 'paused')
+        return;
+    state.status = { _tag: 'running', turnCount: 0 };
+}
+function abortAgent(state) {
+    state.abortController.abort();
+    if (state.status._tag === 'running' || state.status._tag === 'paused') {
+        state.status = { _tag: 'stopped', reason: 'aborted' };
+    }
+}
+function getAgentStatus(state) {
+    return state.status;
+}
+/** 判断 agent 是否处于暂停态（spec §19.2）。 */
+function isAgentPaused(state) {
+    return state.status._tag === 'paused';
+}
+export { abortAgent, createAgent, getAgentStatus, isAgentPaused, pauseAgent, resumeAgent, runAgent };

package/dist/core/agents/builtin.d.ts

@@ -0,0 +1,6 @@
+import type { AgentDefinition } from './types.js';
+/** 通用 subagent worker 骨架（叠加角色 prompt）。 */
+declare const WORKER_BASE = "You are a worker agent for delegated tasks.\n\nYou have FULL access to the tools provided. Use them as needed to complete the assigned work.\n\n<directives>\n- You MUST finish only the assigned work and return the minimum useful result. Do not repeat what you wrote to filesystem.\n- You MUST be concise. NEVER include filler, repetition, or tool transcripts. The user cannot see you. Your result is just notes for the main agent.\n- You SHOULD prefer narrow lookups (grep/glob), then read only the needed ranges. Ignore anything beyond your scope.\n- You SHOULD prefer edits to existing files over creating new ones.\n- You NEVER create documentation files (*.md, README) unless explicitly requested.\n- You MUST follow the assignment exactly.\n</directives>\n\nWhen done, call the `yield` tool with your structured result. This is the ONLY way to return a final result.";
+/** 6 个内置 agent（2 primary + 4 subagent）。 */
+declare const BUILTIN_AGENTS: AgentDefinition[];
+export { BUILTIN_AGENTS, WORKER_BASE };

package/dist/core/agents/builtin.js

@@ -0,0 +1,76 @@
+/** 通用 subagent worker 骨架（叠加角色 prompt）。 */
+const WORKER_BASE = `You are a worker agent for delegated tasks.
+
+You have FULL access to the tools provided. Use them as needed to complete the assigned work.
+
+<directives>
+- You MUST finish only the assigned work and return the minimum useful result. Do not repeat what you wrote to filesystem.
+- You MUST be concise. NEVER include filler, repetition, or tool transcripts. The user cannot see you. Your result is just notes for the main agent.
+- You SHOULD prefer narrow lookups (grep/glob), then read only the needed ranges. Ignore anything beyond your scope.
+- You SHOULD prefer edits to existing files over creating new ones.
+- You NEVER create documentation files (*.md, README) unless explicitly requested.
+- You MUST follow the assignment exactly.
+</directives>
+
+When done, call the \`yield\` tool with your structured result. This is the ONLY way to return a final result.`;
+/** Plan 模式 primary agent 的 role prompt（覆盖 role section）。 */
+const PLAN_ROLE = `You are c0de-agent in **Plan Mode**. Your job is to investigate the codebase and produce a clear, actionable plan — NOT to make changes directly.
+
+- Use read-only tools (grep/glob/read) to understand the structure and relevant code.
+- Ask clarifying questions when requirements are ambiguous.
+- When ready, present a concrete implementation plan (files to touch, approach, risks).
+- Do NOT use edit/write tools to modify code. You may run bash for investigation only.`;
+/** 6 个内置 agent（2 primary + 4 subagent）。 */
+const BUILTIN_AGENTS = [
+    {
+        name: 'default',
+        description: '通用助手（默认）。全工具，动态系统提示。',
+        systemPrompt: '',
+        mode: 'primary',
+        source: 'builtin',
+    },
+    {
+        name: 'plan',
+        description: '计划模式（只读）。专注调研与方案设计，不直接改代码。',
+        tools: ['read', 'grep', 'glob', 'bash'],
+        systemPrompt: PLAN_ROLE,
+        mode: 'primary',
+        source: 'builtin',
+    },
+    {
+        name: 'general',
+        description: '通用助手，全工具，可递归派生子任务。默认子 agent。',
+        systemPrompt: `${WORKER_BASE}\n\nYou are a general-purpose agent. Tackle any delegated task with the tools available.`,
+        mode: 'subagent',
+        maxRecursion: 1,
+        source: 'builtin',
+    },
+    {
+        name: 'coder',
+        description: '实现专家，专注写代码实现。可读写文件、执行命令。',
+        systemPrompt: `${WORKER_BASE}\n\nYou specialize as: an implementation engineer. Bring exactly that expertise — write clean, correct, well-tested code. Prefer surgical edits. Verify your work (run relevant tests/commands) before yielding.`,
+        tools: ['read', 'write', 'edit', 'bash', 'grep', 'glob'],
+        mode: 'subagent',
+        maxRecursion: 0,
+        source: 'builtin',
+    },
+    {
+        name: 'researcher',
+        description: '只读代码调研专家，用 grep/glob/read 摸清结构后返回压缩上下文。不改任何文件。',
+        systemPrompt: `${WORKER_BASE}\n\nYou specialize as: a read-only codebase scout. Map the relevant code, return compressed context (key files, structures, signatures). NEVER modify files. NEVER run write/edit/bash.`,
+        tools: ['grep', 'glob', 'read'],
+        mode: 'subagent',
+        maxRecursion: 0,
+        source: 'builtin',
+    },
+    {
+        name: 'reviewer',
+        description: '代码审查专家，返回结构化发现（问题、建议、严重性）。',
+        systemPrompt: `${WORKER_BASE}\n\nYou specialize as: a code reviewer. Read the code under review, assess correctness/quality/risks. Return findings via yield as { findings: [{ severity, file, line, issue, suggestion }], summary: string }. severity ∈ 'critical' | 'warning' | 'info'.`,
+        tools: ['grep', 'glob', 'read'],
+        mode: 'subagent',
+        maxRecursion: 0,
+        source: 'builtin',
+    },
+];
+export { BUILTIN_AGENTS, WORKER_BASE };

package/dist/core/agents/discovery.d.ts

@@ -0,0 +1,6 @@
+import type { AgentDefinition, AgentSource } from './types.js';
+/** 解析单个 markdown 文件为 AgentDefinition。无 frontmatter 返回 null。 */
+declare function loadAgentFile(filePath: string, source: AgentSource): Promise<AgentDefinition | null>;
+/** 加载项目 agents 目录下所有 .md 为 AgentDefinition（源: project）。 */
+declare function loadAgents(projectDir: string): Promise<AgentDefinition[]>;
+export { loadAgentFile, loadAgents };