@fitlab-ai/agent-infra 0.7.7 → 0.8.0

package/dist/lib/server/process-control.js

@@ -0,0 +1,200 @@
+import fs from 'node:fs';
+import path from 'node:path';
+import { spawn, execFileSync } from 'node:child_process';
+import { loadServerConfig } from "./config.js";
+import { runDaemon } from "./daemon.js";
+export function buildStopCommand(pid, platform) {
+    if (platform === 'win32') {
+        return { kind: 'exec', command: 'taskkill', args: ['/PID', String(pid), '/T', '/F'] };
+    }
+    return { kind: 'signal', signal: 'SIGTERM' };
+}
+// Liveness check that treats an exited-but-unreaped daemon as dead.
+//
+// The daemon is spawned detached and re-parented to init when `ai server start`
+// exits. After it terminates, `process.kill(pid, 0)` still succeeds until init
+// reaps the zombie. On Linux we therefore also check /proc state so status/stop
+// /start don't treat an already-exited daemon as running. macOS reaps orphans
+// via launchd and Windows has no zombies, so the kill(pid, 0) result is enough.
+export function isProcessAlive(pid) {
+    try {
+        process.kill(pid, 0);
+    }
+    catch (error) {
+        // EPERM means the process exists but we may not signal it → still alive.
+        return error.code === 'EPERM';
+    }
+    if (process.platform === 'linux') {
+        try {
+            const stat = fs.readFileSync(`/proc/${pid}/stat`, 'utf8');
+            // The process state is the first token after the parenthesised comm.
+            const state = stat.slice(stat.lastIndexOf(')') + 1).trim().charAt(0);
+            if (state === 'Z')
+                return false;
+        }
+        catch {
+            return false; // /proc entry vanished → not alive
+        }
+    }
+    return true;
+}
+function readPid(pidFile) {
+    try {
+        const raw = fs.readFileSync(pidFile, 'utf8').trim();
+        const pid = Number.parseInt(raw, 10);
+        return Number.isInteger(pid) && pid > 0 ? pid : null;
+    }
+    catch {
+        return null;
+    }
+}
+function removePidFile(pidFile) {
+    try {
+        fs.unlinkSync(pidFile);
+    }
+    catch {
+        // Already gone.
+    }
+}
+function delay(ms) {
+    return new Promise((resolve) => {
+        setTimeout(resolve, ms);
+    });
+}
+function enabledAdapterNames(config) {
+    return Object.entries(config.adapters)
+        .filter(([, adapter]) => adapter?.enabled === true)
+        .map(([name]) => name);
+}
+export async function start({ foreground = false } = {}) {
+    const config = loadServerConfig();
+    const pidPath = config.pidFile;
+    // Zombie PID cleanup: a stale PID file from a crashed daemon must not block a
+    // fresh start.
+    const existing = readPid(pidPath);
+    if (existing !== null && isProcessAlive(existing)) {
+        process.stdout.write(`server already running (pid ${existing})\n`);
+        return;
+    }
+    if (existing !== null) {
+        removePidFile(pidPath);
+    }
+    if (foreground) {
+        await runDaemon();
+        return;
+    }
+    const cliEntry = process.argv[1];
+    if (!cliEntry) {
+        throw new Error('server: unable to determine CLI entry point for daemon spawn');
+    }
+    // Re-spawn ourselves detached. process.execArgv is forwarded so the dev path
+    // (node --experimental-strip-types ./bin/cli.ts) and the built path
+    // (node dist/bin/cli.js) both work.
+    const child = spawn(process.execPath, [...process.execArgv, cliEntry, 'server', '__daemon'], {
+        detached: true,
+        stdio: 'ignore'
+    });
+    child.unref();
+    if (typeof child.pid !== 'number') {
+        throw new Error('server: failed to spawn daemon process');
+    }
+    fs.mkdirSync(path.dirname(pidPath), { recursive: true });
+    fs.writeFileSync(pidPath, `${child.pid}\n`);
+    process.stdout.write(`server started (pid ${child.pid})\n`);
+}
+export async function stop() {
+    const config = loadServerConfig();
+    const pid = readPid(config.pidFile);
+    if (pid === null) {
+        process.stdout.write('server is not running (no pid file)\n');
+        return;
+    }
+    if (!isProcessAlive(pid)) {
+        removePidFile(config.pidFile);
+        process.stdout.write('server is not running (removed stale pid file)\n');
+        return;
+    }
+    const command = buildStopCommand(pid, process.platform);
+    if (command.kind === 'exec') {
+        execFileSync(command.command, command.args);
+    }
+    else {
+        process.kill(pid, command.signal);
+        const deadline = Date.now() + 5000;
+        while (isProcessAlive(pid) && Date.now() < deadline) {
+            await delay(100);
+        }
+        if (isProcessAlive(pid)) {
+            process.kill(pid, 'SIGKILL');
+        }
+    }
+    removePidFile(config.pidFile);
+    process.stdout.write(`server stopped (pid ${pid})\n`);
+}
+export function status() {
+    const config = loadServerConfig();
+    const pid = readPid(config.pidFile);
+    if (pid === null || !isProcessAlive(pid)) {
+        process.stdout.write('server: stopped\n');
+        if (pid !== null) {
+            process.stdout.write(`  (stale pid file references pid ${pid})\n`);
+        }
+        return;
+    }
+    let startedAt = 'unknown';
+    try {
+        startedAt = fs.statSync(config.pidFile).mtime.toISOString();
+    }
+    catch {
+        // Leave as unknown.
+    }
+    const adapters = enabledAdapterNames(config);
+    process.stdout.write(`server: running\n` +
+        `  pid: ${pid}\n` +
+        `  started: ${startedAt}\n` +
+        `  adapters: ${adapters.length > 0 ? adapters.join(', ') : '(none)'}\n` +
+        `  pid file: ${config.pidFile}\n` +
+        `  log: ${config.log.path}\n`);
+}
+export async function logs({ follow = false } = {}) {
+    const config = loadServerConfig();
+    const logPath = config.log.path;
+    if (!fs.existsSync(logPath)) {
+        process.stdout.write('server: no log file yet\n');
+        return;
+    }
+    const initial = fs.readFileSync(logPath, 'utf8');
+    process.stdout.write(initial);
+    if (!follow)
+        return;
+    let position = Buffer.byteLength(initial);
+    const watcher = fs.watch(logPath, () => {
+        try {
+            const { size } = fs.statSync(logPath);
+            if (size < position)
+                position = 0; // truncated or rotated
+            if (size > position) {
+                const fd = fs.openSync(logPath, 'r');
+                try {
+                    const buffer = Buffer.alloc(size - position);
+                    fs.readSync(fd, buffer, 0, buffer.length, position);
+                    process.stdout.write(buffer.toString('utf8'));
+                }
+                finally {
+                    fs.closeSync(fd);
+                }
+                position = size;
+            }
+        }
+        catch {
+            // Transient read error during rotation; ignore and wait for next event.
+        }
+    });
+    await new Promise((resolve) => {
+        process.on('SIGINT', () => {
+            watcher.close();
+            resolve();
+        });
+    });
+}
+//# sourceMappingURL=process-control.js.map

package/dist/lib/server/protocol.js

@@ -0,0 +1,69 @@
+import { VERSION } from "../version.js";
+import { getSkillRunSpec } from "../run/skills.js";
+const TASK_READ = new Set(['ls', 'list', 'status', 'show', 'log', 'decisions']);
+const SANDBOX_READ = new Set(['ls', 'list', 'show']);
+const SANDBOX_WRITE = new Set(['create', 'start']);
+function splitWords(text) {
+    return text.trim().split(/\s+/).filter(Boolean);
+}
+function taskSubcommand(subcommand) {
+    return subcommand === 'list' ? 'ls' : subcommand;
+}
+export function commandHelp() {
+    return [
+        `agent-infra ${VERSION}`,
+        'Built-ins: /help, /ping, /version',
+        'Read: /sandbox ls|show|vm status, /task decisions|log|ls|show|status',
+        'Write: /sandbox create|start',
+        'Exec: /decide <task-ref> <HD-id> <decision>, /run create-task <description>, /run <skill> <task-ref> ...'
+    ].join('\n');
+}
+export function parseCommand(text) {
+    const trimmed = text.trim();
+    if (!trimmed.startsWith('/'))
+        return { kind: 'ignore' };
+    const words = splitWords(trimmed);
+    const [command = '', subcommand = '', ...rest] = words;
+    if (command === '/help')
+        return { kind: 'builtin', name: 'help', role: 'read', args: rest };
+    if (command === '/ping')
+        return { kind: 'builtin', name: 'ping', role: 'read', args: rest };
+    if (command === '/version')
+        return { kind: 'builtin', name: 'version', role: 'read', args: rest };
+    if (command === '/decide') {
+        return { kind: 'ai', role: 'exec', argv: ['decide', subcommand, ...rest] };
+    }
+    if (command === '/task') {
+        if (!TASK_READ.has(subcommand))
+            return { kind: 'error', message: 'Unknown /task command' };
+        return { kind: 'ai', role: 'read', argv: ['task', taskSubcommand(subcommand), ...rest] };
+    }
+    if (command === '/sandbox') {
+        if (subcommand === 'vm') {
+            if (rest[0] !== 'status')
+                return { kind: 'error', message: 'Only /sandbox vm status is allowed' };
+            return { kind: 'ai', role: 'read', argv: ['sandbox', 'vm', 'status'] };
+        }
+        if (SANDBOX_READ.has(subcommand)) {
+            return { kind: 'ai', role: 'read', argv: ['sandbox', taskSubcommand(subcommand), ...rest] };
+        }
+        if (SANDBOX_WRITE.has(subcommand)) {
+            return { kind: 'ai', role: 'write', argv: ['sandbox', subcommand, ...rest] };
+        }
+        if (subcommand === 'rm') {
+            return {
+                kind: 'error',
+                message: '/sandbox rm is not available from IM because it requires interactive confirmation'
+            };
+        }
+        return { kind: 'error', message: 'Unknown /sandbox command' };
+    }
+    if (command === '/run') {
+        const spec = getSkillRunSpec(subcommand);
+        if (!spec)
+            return { kind: 'error', message: `Unknown skill: ${subcommand}` };
+        return { kind: 'ai', role: spec.role, argv: ['run', subcommand, ...rest] };
+    }
+    return { kind: 'error', message: `Unknown command: ${command}` };
+}
+//# sourceMappingURL=protocol.js.map

package/dist/lib/server/redact.js

@@ -0,0 +1,12 @@
+const SECRET_PATTERNS = [
+    /\b(token|secret|password|passwd|api[_-]?key)=([^\s]+)/gi,
+    /\b(Bearer\s+)[A-Za-z0-9._~+/=-]+/g
+];
+export function redactSecrets(text) {
+    let out = text;
+    for (const pattern of SECRET_PATTERNS) {
+        out = out.replace(pattern, (_match, prefix) => `${prefix}=<redacted>`);
+    }
+    return out;
+}
+//# sourceMappingURL=redact.js.map

package/dist/lib/server/runner.js

@@ -0,0 +1,75 @@
+import { spawn } from 'node:child_process';
+import fs from 'node:fs';
+import path from 'node:path';
+function resolveCommand(file) {
+    if (process.platform !== 'win32' || path.extname(file)) {
+        return file;
+    }
+    const pathValue = process.env.Path || process.env.PATH || '';
+    const extensions = (process.env.PATHEXT || '.COM;.EXE;.BAT;.CMD').split(';').filter(Boolean);
+    for (const dir of pathValue.split(path.delimiter).filter(Boolean)) {
+        for (const extension of extensions) {
+            const lowerCandidate = path.join(dir, `${file}${extension.toLowerCase()}`);
+            if (fs.existsSync(lowerCandidate))
+                return lowerCandidate;
+            const upperCandidate = path.join(dir, `${file}${extension.toUpperCase()}`);
+            if (fs.existsSync(upperCandidate))
+                return upperCandidate;
+        }
+    }
+    return file;
+}
+function needsShell(file) {
+    return process.platform === 'win32' && /\.(?:bat|cmd)$/i.test(file);
+}
+function spawnCapture(file, args, onChunk) {
+    return new Promise((resolve, reject) => {
+        const resolvedFile = resolveCommand(file);
+        const child = spawn(resolvedFile, args, {
+            stdio: ['ignore', 'pipe', 'pipe'],
+            shell: needsShell(resolvedFile)
+        });
+        let stdout = '';
+        let stderr = '';
+        let settled = false;
+        let pendingChunks = Promise.resolve();
+        const rejectOnce = (error) => {
+            if (settled)
+                return;
+            settled = true;
+            reject(error);
+        };
+        const resolveOnce = (result) => {
+            if (settled)
+                return;
+            settled = true;
+            resolve(result);
+        };
+        const enqueueChunk = (chunk) => {
+            if (!onChunk)
+                return;
+            pendingChunks = pendingChunks.then(() => onChunk(chunk));
+            pendingChunks.catch(rejectOnce);
+        };
+        child.stdout.setEncoding('utf8');
+        child.stderr.setEncoding('utf8');
+        child.stdout.on('data', (chunk) => {
+            stdout += chunk;
+            enqueueChunk(chunk);
+        });
+        child.stderr.on('data', (chunk) => {
+            stderr += chunk;
+            enqueueChunk(chunk);
+        });
+        child.on('error', rejectOnce);
+        child.on('close', (exitCode, signal) => {
+            pendingChunks
+                .then(() => resolveOnce({ exitCode, signal, stdout, stderr }))
+                .catch(rejectOnce);
+        });
+    });
+}
+export async function runAi(args, options = {}) {
+    return (options.spawn ?? spawnCapture)('ai', args, options.onChunk);
+}
+//# sourceMappingURL=runner.js.map

package/dist/lib/server/streamer.js

@@ -0,0 +1,44 @@
+import { redactSecrets } from "./redact.js";
+function chunks(text, size) {
+    if (text.length === 0)
+        return [];
+    const out = [];
+    for (let i = 0; i < text.length; i += size) {
+        out.push(text.slice(i, i + size));
+    }
+    return out;
+}
+export async function streamCommand(options, run, send) {
+    await send(`started ${options.title}`);
+    const size = options.chunkChars ?? 4000;
+    const throttleMs = options.throttleMs ?? 0;
+    let streamed = false;
+    let buffer = '';
+    let lastFlush = 0;
+    const flush = async () => {
+        if (!buffer)
+            return;
+        const text = redactSecrets(buffer);
+        buffer = '';
+        lastFlush = Date.now();
+        for (const chunk of chunks(text, size)) {
+            await send(chunk);
+        }
+    };
+    const emit = async (chunk) => {
+        streamed = true;
+        buffer += chunk;
+        const due = throttleMs === 0 || Date.now() - lastFlush >= throttleMs;
+        if (buffer.length >= size || due) {
+            await flush();
+        }
+    };
+    const result = await run(emit);
+    if (!streamed) {
+        buffer += [result.stdout, result.stderr].filter(Boolean).join('\n');
+    }
+    await flush();
+    await send(`finished ${options.title} exitCode=${result.exitCode} signal=${result.signal ?? 'null'}`);
+    return result;
+}
+//# sourceMappingURL=streamer.js.map

package/dist/lib/task/commands/decisions.js

@@ -0,0 +1,258 @@
+import fs from 'node:fs';
+import path from 'node:path';
+import { formatTable } from "../../table.js";
+import { resolveTaskRef } from "../resolve-ref.js";
+import { parseLedger, HUMAN_DECISION_STATUSES } from "../ledger.js";
+import { extractSubSection } from "../sections.js";
+const USAGE = `Usage: ai task decisions <N | #N | TASK-id> [selector] [options]
+
+Lists the human-decision (HD-) items recorded in a task's review disagreement
+ledger, or prints the full detail block for a single item. Read-only.
+
+  <ref>          Bare numeric / '#N' short id, or a full TASK-YYYYMMDD-HHMMSS id.
+  [selector]     Ordinal (1-based) or HD id (e.g. 'HD-3') to show one item's detail.
+
+Options:
+  --all              Include already-decided (human-decided) items, not just pending.
+  --stage <s>        Filter to one stage: analysis | plan | code.
+  --format <fmt>     Output format: text (default) | markdown.
+  -h, --help         Show this help.
+
+Aliased as 'ai task d'.
+`;
+const STAGES = new Set(['analysis', 'plan', 'code']);
+const FORMATS = new Set(['text', 'markdown']);
+const HD_ID_RE = /^HD-\d+$/;
+function fail(message) {
+    process.stderr.write(`ai task decisions: ${message}\n`);
+    process.exitCode = 1;
+}
+// Returns null and sets the exit code when an option is malformed.
+function parseArgs(args) {
+    const out = { positionals: [], all: false, format: 'text' };
+    for (let i = 0; i < args.length; i += 1) {
+        const a = args[i];
+        if (a === '--all') {
+            out.all = true;
+        }
+        else if (a === '--stage') {
+            const v = args[i + 1];
+            if (v === undefined) {
+                fail('--stage requires a value (analysis|plan|code)');
+                return null;
+            }
+            out.stage = v;
+            i += 1;
+        }
+        else if (a.startsWith('--stage=')) {
+            out.stage = a.slice('--stage='.length);
+        }
+        else if (a === '--format') {
+            const v = args[i + 1];
+            if (v === undefined) {
+                fail('--format requires a value (text|markdown)');
+                return null;
+            }
+            out.format = v;
+            i += 1;
+        }
+        else if (a.startsWith('--format=')) {
+            out.format = a.slice('--format='.length);
+        }
+        else if (a.startsWith('-')) {
+            fail(`unknown option '${a}'`);
+            return null;
+        }
+        else {
+            out.positionals.push(a);
+        }
+    }
+    return out;
+}
+// Parse `<file>.md#anchor` evidence into its filename, when present.
+function evidenceFile(evidence) {
+    const m = /([\w.-]+\.md)#/.exec(evidence);
+    return m ? m[1] : null;
+}
+function roundOf(file) {
+    const m = /-r(\d+)\.md$/.exec(file);
+    return m ? Number.parseInt(m[1], 10) : 1;
+}
+// Locate the `### HD-N` detail block for a row. Prefer the artifact named by the
+// row's evidence anchor; otherwise scan analysis/plan/code artifacts and return
+// the block from the highest-round file that contains it. Returns '' when none
+// is found (caller degrades gracefully — plan B3).
+function findDetailBlock(row, taskDir) {
+    const hinted = evidenceFile(row.evidence);
+    if (hinted) {
+        const p = path.join(taskDir, hinted);
+        if (fs.existsSync(p)) {
+            const block = extractSubSection(fs.readFileSync(p, 'utf8'), row.id);
+            if (block)
+                return block;
+        }
+    }
+    let best = '';
+    let bestRound = -1;
+    let entries;
+    try {
+        entries = fs.readdirSync(taskDir);
+    }
+    catch {
+        return '';
+    }
+    for (const file of entries) {
+        if (!/^(analysis|plan|code)(-r\d+)?\.md$/.test(file))
+            continue;
+        const block = extractSubSection(fs.readFileSync(path.join(taskDir, file), 'utf8'), row.id);
+        if (block && roundOf(file) > bestRound) {
+            best = block;
+            bestRound = roundOf(file);
+        }
+    }
+    return best;
+}
+// Pull the `## 人工裁决` record lines that mention this HD id, so a decided item
+// shows the human's recorded ruling alongside its detail block.
+function findDecisionRecord(id, content) {
+    const lines = content.split('\n');
+    let i = 0;
+    while (i < lines.length && !/^##\s+(人工裁决|Human Decisions?)\s*$/.test(lines[i]))
+        i += 1;
+    if (i >= lines.length)
+        return [];
+    const idRe = new RegExp(`(^|[^\\w-])${id}(?![\\w-])`);
+    const out = [];
+    for (let j = i + 1; j < lines.length; j += 1) {
+        if (/^##\s/.test(lines[j]))
+            break;
+        if (lines[j].trim().startsWith('-') && idRe.test(lines[j]))
+            out.push(lines[j]);
+    }
+    return out;
+}
+function titleOf(row, taskDir) {
+    const block = findDetailBlock(row, taskDir);
+    if (block) {
+        return block
+            .split('\n')[0]
+            .replace(/^###\s+/, '')
+            .replace(/\s*\[needs-human-decision\]\s*$/, '')
+            .trim();
+    }
+    return row.evidence || '(无详情)';
+}
+function renderList(rows, format, taskDir) {
+    if (rows.length === 0) {
+        process.stdout.write('无待裁决项。\n');
+        return;
+    }
+    const headers = ['#', 'ID', 'STAGE', 'SEVERITY', 'STATUS', 'EVIDENCE', 'TITLE'];
+    const data = rows.map((r, i) => [
+        String(i + 1),
+        r.id,
+        r.stage,
+        r.severity,
+        r.status,
+        r.evidence,
+        titleOf(r, taskDir)
+    ]);
+    if (format === 'markdown') {
+        const sep = headers.map(() => '---');
+        const md = [
+            `| ${headers.join(' | ')} |`,
+            `| ${sep.join(' | ')} |`,
+            ...data.map((row) => `| ${row.join(' | ')} |`)
+        ];
+        process.stdout.write(`${md.join('\n')}\n`);
+        return;
+    }
+    process.stdout.write(`${formatTable(headers, data).join('\n')}\n`);
+}
+function renderDetail(rows, selector, format, taskDir, content) {
+    let row;
+    if (/^\d+$/.test(selector)) {
+        const idx = Number.parseInt(selector, 10) - 1;
+        if (idx < 0 || idx >= rows.length) {
+            fail(`ordinal '${selector}' out of range (1..${rows.length})`);
+            return;
+        }
+        row = rows[idx];
+    }
+    else {
+        const want = selector.toUpperCase();
+        const matches = rows.filter((r) => r.id.toUpperCase() === want);
+        if (matches.length === 0) {
+            fail(`no decision item matches '${selector}'`);
+            return;
+        }
+        if (matches.length > 1) {
+            fail(`duplicate id '${selector}' in ledger; select by ordinal instead`);
+            return;
+        }
+        row = matches[0];
+    }
+    const r = row;
+    const block = findDetailBlock(r, taskDir);
+    const lines = [];
+    if (format === 'markdown') {
+        lines.push(`**${r.id}** (${r.stage}/${r.severity}) · status=\`${r.status}\` · evidence: \`${r.evidence}\``, '');
+    }
+    else {
+        lines.push(`${r.id} (${r.stage}/${r.severity}) status=${r.status}`, `evidence: ${r.evidence}`, '');
+    }
+    if (block) {
+        lines.push(block);
+    }
+    else {
+        lines.push(`（详情块未找到：未在任务产物中定位到 \`### ${r.id}\` 锚点，可能为历史产物或尚未写入；evidence 指向 ${r.evidence}）`);
+    }
+    if (r.status === 'human-decided') {
+        const record = findDecisionRecord(r.id, content);
+        if (record.length) {
+            lines.push('', '人工裁定：', ...record);
+        }
+    }
+    process.stdout.write(`${lines.join('\n')}\n`);
+}
+function decisions(args = []) {
+    if (args[0] === '--help' || args[0] === '-h') {
+        process.stdout.write(USAGE);
+        return;
+    }
+    const parsed = parseArgs(args);
+    if (!parsed)
+        return;
+    if (parsed.positionals.length === 0) {
+        process.stdout.write(USAGE);
+        process.exitCode = 1;
+        return;
+    }
+    if (parsed.stage !== undefined && !STAGES.has(parsed.stage)) {
+        fail(`invalid --stage '${parsed.stage}' (expected analysis|plan|code)`);
+        return;
+    }
+    if (!FORMATS.has(parsed.format)) {
+        fail(`invalid --format '${parsed.format}' (expected text|markdown)`);
+        return;
+    }
+    const resolved = resolveTaskRef(parsed.positionals[0]);
+    if (!resolved.ok) {
+        fail(resolved.message);
+        return;
+    }
+    const content = fs.readFileSync(resolved.taskMdPath, 'utf8');
+    let rows = parseLedger(content).filter((r) => HD_ID_RE.test(r.id));
+    rows = rows.filter((r) => parsed.all ? HUMAN_DECISION_STATUSES.has(r.status) : r.status === 'needs-human-decision');
+    if (parsed.stage !== undefined)
+        rows = rows.filter((r) => r.stage === parsed.stage);
+    const selector = parsed.positionals[1];
+    if (selector === undefined) {
+        renderList(rows, parsed.format, resolved.taskDir);
+    }
+    else {
+        renderDetail(rows, selector, parsed.format, resolved.taskDir, content);
+    }
+}
+export { decisions };
+//# sourceMappingURL=decisions.js.map