@agentshouse/kit 0.2.1-alpha.9 → 0.2.1-c03721a31074a

package/dist/update.js

@@ -0,0 +1,163 @@
+import { spawn } from 'node:child_process';
+import { createHash } from 'node:crypto';
+import { mkdir, readFile, rm, writeFile } from 'node:fs/promises';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import { run } from './clis.js';
+import { kitHome, workspaceRoot } from './home.js';
+import { service, throughSudo } from './placement.js';
+const BOOTSTRAP_BYTES = 1_000_000;
+// A bootstrap is at most a million bytes, so two minutes carry it over ten kilobytes a second and no stalled host holds the update queue.
+const BOOTSTRAP_MS = 120_000;
+function directory() {
+    return join(kitHome(), 'update');
+}
+function pending() {
+    return join(directory(), 'pending.json');
+}
+function outcome() {
+    return join(directory(), 'outcome');
+}
+function unit(callId) {
+    return `house-kit-update-${callId}`;
+}
+function manager() {
+    return service() === 'user' ? ['--user'] : [];
+}
+function elevated() {
+    return throughSudo() ? ['sudo', '-n'] : [];
+}
+function account() {
+    return throughSudo() ? [`--uid=${process.getuid()}`, `--gid=${process.getgid()}`] : [];
+}
+function placement() {
+    return service() === 'user' ? ['--workspace', workspaceRoot()] : ['--linux'];
+}
+function record() {
+    return `outcome=$1; shift; bash "$0" "$@"; printf %s "$?" > "$outcome"; ${[...elevated(), 'systemctl', ...manager(), 'start', 'house-kit.service'].join(' ')}`;
+}
+async function pendingCall() {
+    try {
+        return JSON.parse(await readFile(pending(), 'utf8')).input_id;
+    }
+    catch {
+        return null;
+    }
+}
+async function recorded() {
+    const status = await readFile(outcome(), 'utf8').catch(() => null);
+    if (status === null)
+        return null;
+    return status.trim() === '0' ? {} : { refused: `the Kit bootstrap exited with ${status.trim()}` };
+}
+async function forget(callId) {
+    if ((await pendingCall()) !== callId)
+        return;
+    await rm(pending(), { force: true });
+    await rm(outcome(), { force: true });
+}
+async function finished(callId) {
+    for (;;) {
+        const result = await recorded();
+        if (result !== null)
+            return result;
+        // systemctl answers at once, so ten seconds only bounds a stuck service manager.
+        if ((await run('systemctl', [...manager(), 'is-active', '--quiet', unit(callId)], 10_000)).status !== 0) {
+            return (await recorded()) ?? { refused: 'the Kit bootstrap did not finish' };
+        }
+        // The bootstrap runs for minutes, so checking each second sees its outcome within a second at little cost.
+        await new Promise((resolve) => setTimeout(resolve, 1000));
+    }
+}
+async function bounded(body, most) {
+    const chunks = [];
+    let bytes = 0;
+    for await (const chunk of body ?? []) {
+        bytes += chunk.length;
+        if (bytes > most)
+            return null;
+        chunks.push(chunk);
+    }
+    return Buffer.concat(chunks, bytes);
+}
+function rerun(script, callId, flags) {
+    return new Promise((resolve) => {
+        const [command, ...args] = [
+            ...elevated(),
+            'systemd-run',
+            ...manager(),
+            ...account(),
+            '--wait',
+            '--collect',
+            '--quiet',
+            `--unit=${unit(callId)}`,
+            `--setenv=HOME=${homedir()}`,
+            `--setenv=HOUSE_KIT_HOME=${kitHome()}`,
+            '/bin/sh',
+            '-c',
+            record(),
+            script,
+            outcome(),
+            ...placement(),
+            ...flags,
+        ];
+        const child = spawn(command, args, { stdio: ['ignore', 'ignore', 'inherit'] });
+        child.on('error', () => resolve({ refused: 'no service manager runs this Kit process' }));
+        child.on('close', () => resolve(null));
+    });
+}
+export class KitUpdate {
+    calls;
+    queue = Promise.resolve();
+    constructor(calls) {
+        this.calls = calls;
+    }
+    async resumed() {
+        const callId = await pendingCall();
+        if (callId === null)
+            return;
+        await this.enqueue(callId, () => finished(callId));
+    }
+    start(call) {
+        void this.enqueue(call.input_id, () => this.carry(call));
+    }
+    enqueue(callId, work) {
+        const before = this.queue;
+        const carried = this.calls.carry(callId, async () => {
+            await before;
+            return { result: await work(), returned: () => forget(callId) };
+        });
+        this.queue = before.then(() => carried);
+        return carried;
+    }
+    async carry(call) {
+        if ((await pendingCall()) === call.input_id)
+            return finished(call.input_id);
+        const deadline = AbortSignal.timeout(BOOTSTRAP_MS);
+        let script;
+        try {
+            const answer = await fetch(call.bootstrap, { signal: deadline });
+            if (!answer.ok)
+                return { refused: `the Kit bootstrap could not be downloaded: ${answer.status}` };
+            script = await bounded(answer.body, BOOTSTRAP_BYTES);
+        }
+        catch (error) {
+            if (!deadline.aborted)
+                throw error;
+            return { refused: `the Kit bootstrap did not arrive within ${BOOTSTRAP_MS / 1000} seconds` };
+        }
+        if (script === null)
+            return { refused: `the Kit bootstrap is larger than ${BOOTSTRAP_BYTES} bytes` };
+        if (createHash('sha256').update(script).digest('hex') !== call.sha256) {
+            return { refused: 'the Kit bootstrap does not match its sha256' };
+        }
+        await mkdir(directory(), { recursive: true, mode: 0o700 });
+        const path = join(directory(), 'connect-linux.sh');
+        await writeFile(path, script, { mode: 0o700 });
+        await rm(outcome(), { force: true });
+        await writeFile(pending(), `${JSON.stringify({ input_id: call.input_id })}\n`, { mode: 0o600 });
+        return (await rerun(path, call.input_id, call.update_clis ? ['--update-clis'] : [])) ?? (await recorded()) ?? {
+            refused: 'the Kit bootstrap did not finish',
+        };
+    }
+}

package/dist/web-wizard.js

@@ -0,0 +1,100 @@
+import { loginPath, onLoginPath, withoutProxy } from './clis.js';
+import { killTree } from './scope.js';
+const TERMINAL = 'xterm-256color';
+const SIGNALLED = 128;
+const NO_BASH = "bash is not on the login shell's PATH";
+const LOST = { refused: 'the Kit lost its connection to House' };
+function causeOf(error) {
+    return error instanceof Error ? error.message : String(error);
+}
+function failed(cause) {
+    process.stderr.write(`kit: a Web Wizard could not start: ${cause}\n`);
+    return { refused: `the script could not start: ${cause}` };
+}
+export class WebWizards {
+    calls;
+    send;
+    runs = new Map();
+    constructor(calls, send) {
+        this.calls = calls;
+        this.send = send;
+    }
+    start(call) {
+        void this.calls.carry(call.input_id, () => this.run(call));
+    }
+    notified(notification) {
+        const run = this.runs.get(String(notification.wizard));
+        if (run === undefined)
+            return;
+        if (notification.type === 'wizard_input')
+            this.typed(run, String(notification.data));
+        if (notification.type === 'wizard_size')
+            this.resized(run, { cols: Number(notification.cols), rows: Number(notification.rows) });
+        if (notification.type === 'wizard_end')
+            this.end(run, {});
+    }
+    disconnected() {
+        for (const run of this.runs.values())
+            this.end(run, LOST);
+    }
+    async run(call) {
+        const run = { terminal: null, size: null, typed: [], ended: null, sized: Promise.withResolvers() };
+        this.runs.set(call.input_id, run);
+        try {
+            return { result: await this.ran(call.input_id, call.script, run) };
+        }
+        finally {
+            this.runs.delete(call.input_id);
+        }
+    }
+    async ran(wizard, script, run) {
+        let terminal;
+        try {
+            const { spawn } = await import('@lydell/node-pty');
+            const shell = await onLoginPath('bash');
+            if (shell === null)
+                return failed(NO_BASH);
+            const path = await loginPath();
+            await run.sized.promise;
+            if (run.ended !== null)
+                return run.ended;
+            terminal = spawn(shell, ['-c', script], {
+                name: TERMINAL,
+                ...run.size,
+                cwd: process.cwd(),
+                env: { ...withoutProxy(), PATH: path },
+            });
+        }
+        catch (error) {
+            return failed(causeOf(error));
+        }
+        run.terminal = terminal;
+        terminal.onData((data) => this.send({ type: 'wizard_output', wizard, data }));
+        for (const data of run.typed.splice(0))
+            terminal.write(data);
+        const exited = await new Promise((resolve) => terminal.onExit(resolve));
+        return run.ended ?? { exit_code: exited.signal ? SIGNALLED + exited.signal : exited.exitCode };
+    }
+    typed(run, data) {
+        if (run.terminal === null)
+            run.typed.push(data);
+        else
+            run.terminal.write(data);
+    }
+    resized(run, size) {
+        run.size = size;
+        run.sized.resolve();
+        try {
+            run.terminal?.resize(size.cols, size.rows);
+        }
+        catch { }
+    }
+    end(run, result) {
+        if (run.ended !== null)
+            return;
+        run.ended = result;
+        run.sized.resolve();
+        if (run.terminal !== null)
+            killTree(run.terminal);
+    }
+}

package/dist/wire.js

@@ -0,0 +1,22 @@
+import { newerThanHouse, updateRequired } from './end.js';
+import { machine } from './placement.js';
+import { refusalOf } from './refusals.js';
+export const KIT_WIRE = 3;
+export const KIT_WIRE_HEADER = 'x-house-kit-wire';
+function newer(text) {
+    let body;
+    try {
+        body = JSON.parse(text);
+    }
+    catch {
+        return null;
+    }
+    return body?.error?.code === 'kit_newer_than_house' ? body.error : null;
+}
+export function endUnserved(text) {
+    if (refusalOf(text)?.code === 'kit_update_required')
+        updateRequired();
+    const refused = newer(text);
+    if (refused !== null)
+        newerThanHouse(refused.release, refused.commands[machine()]);
+}

package/dist/work.js

@@ -0,0 +1,346 @@
+import { randomUUID } from 'node:crypto';
+import { once } from 'node:events';
+import { readFileSync } from 'node:fs';
+import { rm } from 'node:fs/promises';
+import { createServer } from 'node:net';
+import { join } from 'node:path';
+import { RUNNING_JOB } from './events.js';
+import { kitHome, socketDirectory } from './home.js';
+const ENDED_TASK = new Set(['completed', 'failed', 'killed']);
+const SCHEDULING = new Set(['CronCreate', 'CronDelete', 'ScheduleWakeup']);
+const QUEUE_LIST = 'house-kit-queue-';
+const SYNC = 'house-kit-sync-';
+const WAKEUP = /Next wakeup scheduled for (\d\d):(\d\d):(\d\d) \(in (\d+)s\)/;
+const HALF_DAY = 12 * 60 * 60 * 1000;
+export function codexLauncher() {
+    return join(kitHome(), 'launchers', 'codex');
+}
+export class Work {
+    changed;
+    jobs = new Set();
+    constructor(changed) {
+        this.changed = changed;
+    }
+    async open(_cli) {
+        return {};
+    }
+    async settled() { }
+    read(message) {
+        this.take(message);
+        this.changed();
+    }
+    take(message) {
+        const update = message.params?.update;
+        if (message.method !== 'session/update' || update?.asyncTaskId === undefined)
+            return;
+        if (update.sessionUpdate === 'async_task_spawned')
+            this.jobs.add(update.asyncTaskId);
+        if (update.sessionUpdate === 'async_task_state_update' && !RUNNING_JOB.has(update.state))
+            this.jobs.delete(update.asyncTaskId);
+    }
+    busy() {
+        return this.jobs.size > 0;
+    }
+    close() { }
+}
+class ClaudeWork extends Work {
+    cwd;
+    level = new Set();
+    listed = new Set();
+    started = new Set();
+    running = false;
+    hooks = new Set();
+    calls = new Map();
+    crons = new Map();
+    wakeups = [];
+    goal = false;
+    constructor(cwd, changed) {
+        super(changed);
+        this.cwd = cwd;
+    }
+    take(message) {
+        super.take(message);
+        if (message.method === '_claude/sdkMessage')
+            this.sdk(message.params.message);
+        if (message.method === 'session/update')
+            this.update(message.params.update);
+    }
+    busy() {
+        return (super.busy() ||
+            this.level.size > 0 ||
+            this.started.size > 0 ||
+            this.running ||
+            this.hooks.size > 0 ||
+            this.crons.size > 0 ||
+            this.wakeups.length > 0 ||
+            this.goal ||
+            this.durable());
+    }
+    sdk(message) {
+        switch (message.subtype) {
+            case 'background_tasks_changed':
+                this.level = new Set(message.tasks.map((task) => task.task_id));
+                for (const id of this.level) {
+                    this.listed.add(id);
+                    this.started.delete(id);
+                }
+                break;
+            case 'task_started':
+                if (!this.listed.has(message.task_id))
+                    this.started.add(message.task_id);
+                break;
+            case 'task_updated':
+                if (ENDED_TASK.has(message.patch.status))
+                    this.started.delete(message.task_id);
+                break;
+            case 'task_notification':
+                this.started.delete(message.task_id);
+                break;
+            case 'session_state_changed':
+                if (message.state === 'running' && !this.running)
+                    this.wakeups = this.wakeups.filter((due) => due > Date.now());
+                this.running = message.state === 'running';
+                break;
+            case 'hook_started':
+                this.hooks.add(message.hook_id);
+                break;
+            case 'hook_response':
+                this.hooks.delete(message.hook_id);
+                break;
+        }
+    }
+    update(update) {
+        const air = update._meta?.jetbrains?.air;
+        if (update.sessionUpdate === 'session_info_update' && air !== undefined && 'goal' in air) {
+            this.goal = air.goal?.status === 'active';
+        }
+        if (update.sessionUpdate !== 'tool_call' && update.sessionUpdate !== 'tool_call_update')
+            return;
+        const tool = update._meta?.claudeCode?.toolName;
+        if (update.sessionUpdate === 'tool_call' && SCHEDULING.has(tool ?? ''))
+            this.calls.set(update.toolCallId, { tool });
+        const call = this.calls.get(update.toolCallId);
+        if (call === undefined)
+            return;
+        if (update.rawInput !== undefined)
+            call.input = update.rawInput;
+        if (update.content !== undefined)
+            call.content = update.content;
+        if (update.status === 'failed')
+            this.calls.delete(update.toolCallId);
+        if (update.status !== 'completed')
+            return;
+        this.calls.delete(update.toolCallId);
+        if (call.tool === 'CronCreate')
+            this.crons.set(update.toolCallId, JSON.stringify(call.content ?? []));
+        if (call.tool === 'ScheduleWakeup')
+            this.wakeup(call);
+        if (call.tool !== 'CronDelete')
+            return;
+        for (const [id, text] of this.crons)
+            if (text.includes(String(call.input?.id)))
+                this.crons.delete(id);
+    }
+    wakeup(call) {
+        if (call.input?.stop === true) {
+            this.wakeups = [];
+            return;
+        }
+        const time = WAKEUP.exec(JSON.stringify(call.content ?? []));
+        if (time === null) {
+            this.wakeups.push(Infinity);
+            return;
+        }
+        const fires = Date.now() + Number(time[4]) * 1000;
+        const due = new Date(fires);
+        due.setHours(Number(time[1]), Number(time[2]), Number(time[3]), 0);
+        if (due.getTime() - fires > HALF_DAY)
+            due.setDate(due.getDate() - 1);
+        if (fires - due.getTime() > HALF_DAY)
+            due.setDate(due.getDate() + 1);
+        this.wakeups.push(due.getTime());
+    }
+    durable() {
+        let text;
+        try {
+            text = readFileSync(join(this.cwd, '.claude', 'scheduled_tasks.json'), 'utf8');
+        }
+        catch (error) {
+            return error.code !== 'ENOENT';
+        }
+        try {
+            return JSON.parse(text).tasks?.length !== 0;
+        }
+        catch {
+            return true;
+        }
+    }
+}
+class CodexWork extends Work {
+    threads = new Map();
+    queues = new Map();
+    goals = new Set();
+    items = new Set();
+    hooks = new Set();
+    starting = new Set();
+    settling = [];
+    memories = null;
+    lost = false;
+    server = null;
+    path = null;
+    socket = null;
+    asked = 0;
+    heard = 0;
+    proving = -1;
+    proven = -1;
+    async open(cli) {
+        this.path = join(socketDirectory(), `${randomUUID()}.sock`);
+        this.server = createServer((socket) => this.connected(socket));
+        this.server.listen(this.path);
+        await once(this.server, 'listening');
+        return { CODEX_PATH: codexLauncher(), HOUSE_KIT_CODEX: cli, HOUSE_KIT_CODEX_STREAM: this.path };
+    }
+    settled() {
+        return new Promise((resolve) => {
+            this.settling.push(resolve);
+            this.settle();
+        });
+    }
+    take(message) {
+        super.take(message);
+        this.heard++;
+    }
+    busy() {
+        if (super.busy() ||
+            this.lost ||
+            this.memories !== false ||
+            [...this.threads.values()].some(Boolean) ||
+            this.queues.size > 0 ||
+            this.goals.size > 0 ||
+            this.items.size > 0 ||
+            this.hooks.size > 0) {
+            return true;
+        }
+        if (this.proven === this.heard)
+            return false;
+        this.prove();
+        return true;
+    }
+    close() {
+        this.server?.close();
+        if (this.path !== null)
+            void rm(this.path, { force: true });
+    }
+    connected(socket) {
+        this.socket = socket;
+        let buffer = '';
+        socket.on('error', () => this.lose());
+        socket.on('close', () => this.lose());
+        socket.setEncoding('utf8').on('data', (chunk) => {
+            buffer += chunk;
+            for (let index = buffer.indexOf('\n'); index >= 0; index = buffer.indexOf('\n')) {
+                const message = this.parsed(buffer.slice(0, index));
+                buffer = buffer.slice(index + 1);
+                if (message !== null)
+                    this.stream(message, socket);
+            }
+            this.settle();
+            this.changed();
+        });
+    }
+    parsed(line) {
+        try {
+            return JSON.parse(line);
+        }
+        catch {
+            this.lost = true;
+            return null;
+        }
+    }
+    lose() {
+        this.lost = true;
+        this.changed();
+    }
+    prove() {
+        if (this.proving === this.heard)
+            return;
+        this.proving = this.heard;
+        this.socket.write(`${JSON.stringify({ jsonrpc: '2.0', id: `${SYNC}${this.heard}`, method: 'house-kit/sync' })}\n`);
+    }
+    settle() {
+        if (this.starting.size === 0)
+            for (const resolve of this.settling.splice(0))
+                resolve();
+    }
+    stream(message, socket) {
+        const params = message.params;
+        switch (message.method) {
+            case 'thread/status/changed':
+                this.threads.set(params.threadId, params.status.type === 'active' && params.status.activeFlags.length === 0);
+                break;
+            case 'thread/closed':
+                this.threads.delete(params.threadId);
+                break;
+            case 'thread/queue/changed':
+                this.queues.set(params.threadId, ++this.asked);
+                socket.write(`${JSON.stringify({ jsonrpc: '2.0', id: `${QUEUE_LIST}${this.asked}`, method: 'thread/queue/list', params: { threadId: params.threadId } })}\n`);
+                break;
+            case 'thread/goal/updated':
+                if (params.goal.status === 'active')
+                    this.goals.add(params.threadId);
+                else
+                    this.goals.delete(params.threadId);
+                break;
+            case 'thread/goal/cleared':
+                this.goals.delete(params.threadId);
+                break;
+            case 'item/started':
+                this.items.add(`${params.threadId} ${params.item.id}`);
+                break;
+            case 'item/completed':
+                this.items.delete(`${params.threadId} ${params.item.id}`);
+                break;
+            case 'hook/started':
+                this.hooks.add(params.run.id);
+                break;
+            case 'hook/completed':
+                this.hooks.delete(params.run.id);
+                break;
+            case 'mcpServer/startupStatus/updated':
+                if (params.status === 'starting')
+                    this.starting.add(`${params.threadId} ${params.name}`);
+                else
+                    this.starting.delete(`${params.threadId} ${params.name}`);
+                break;
+            case undefined:
+                this.answered(message);
+        }
+    }
+    answered(message) {
+        if (typeof message.id === 'string' && message.id.startsWith(SYNC)) {
+            this.proven = Number(message.id.slice(SYNC.length));
+        }
+        else if (typeof message.id === 'string' && message.id.startsWith(QUEUE_LIST)) {
+            const asked = Number(message.id.slice(QUEUE_LIST.length));
+            for (const [thread, latest] of this.queues) {
+                if (latest === asked && message.result?.data?.length === 0)
+                    this.queues.delete(thread);
+            }
+        }
+        else if (message.result?.config?.features !== undefined) {
+            this.memories = message.result.config.features.memories !== false;
+        }
+    }
+}
+class GrokWork extends Work {
+    busy() {
+        return true;
+    }
+}
+export function workFor(kind, cwd, changed) {
+    if (kind === 'claude-agent-acp')
+        return new ClaudeWork(cwd, changed);
+    if (kind === 'codex-acp')
+        return new CodexWork(changed);
+    return new GrokWork(changed);
+}