@astroanywhere/agent 0.1.0

package/dist/lib/ssh-installer.js

@@ -0,0 +1,230 @@
+/**
+ * Remote SSH installation utilities for local dev.
+ *
+ * Packs the agent-runner locally, SCPs it to the target host,
+ * installs via npm, and pushes pre-provisioned tokens so the
+ * remote host skips its own device auth entirely.
+ */
+import { execFile as execFileCb } from 'node:child_process';
+import { promisify } from 'node:util';
+import { networkInterfaces } from 'node:os';
+import { resolve, dirname } from 'node:path';
+import { fileURLToPath } from 'node:url';
+const execFile = promisify(execFileCb);
+// ============================================================================
+// Network helpers
+// ============================================================================
+/**
+ * Detect the LAN IP of this machine so remote hosts can reach the dev server.
+ * Returns the first non-internal IPv4 address found.
+ */
+export function detectLocalIP() {
+    const nets = networkInterfaces();
+    for (const ifaces of Object.values(nets)) {
+        if (!ifaces)
+            continue;
+        for (const iface of ifaces) {
+            if (iface.family === 'IPv4' && !iface.internal) {
+                return iface.address;
+            }
+        }
+    }
+    return '127.0.0.1';
+}
+// ============================================================================
+// Remote checks
+// ============================================================================
+/**
+ * SSH to a host and check whether Node.js ≥ 18 is available.
+ * Returns { available, version } or throws on SSH failure.
+ */
+export async function checkRemoteNode(host) {
+    try {
+        const { stdout } = await sshExec(host, 'node --version');
+        const ver = stdout.trim(); // e.g. "v20.11.0"
+        const major = parseInt(ver.replace(/^v/, ''), 10);
+        return { available: major >= 18, version: ver };
+    }
+    catch {
+        return { available: false, version: null };
+    }
+}
+/**
+ * Pack the local agent-runner, SCP it to the remote host, install, and configure.
+ *
+ * Steps:
+ *   1. `npm pack` in the agent-runner directory → tarball
+ *   2. `scp` tarball to remote ~/
+ *   3. `ssh npm install -g <tarball>` on remote
+ *   4. `ssh astro-agent setup --non-interactive --skip-auth` on remote
+ *   5. Push tokens via `ssh astro-agent config --set` commands
+ */
+export async function packAndInstall(opts, onProgress) {
+    const { host, apiUrl, relayUrl, accessToken, refreshToken, wsToken, machineId } = opts;
+    const log = onProgress ?? (() => { });
+    // 1. npm pack
+    const agentRunnerDir = resolve(dirname(fileURLToPath(import.meta.url)), '..', '..');
+    log('Packing agent-runner...');
+    const { stdout: packOut } = await execFile('npm', ['pack', '--pack-destination', '/tmp'], {
+        cwd: agentRunnerDir,
+    });
+    const tarball = `/tmp/${packOut.trim().split('\n').pop()}`;
+    // 2. SCP tarball to remote
+    log(`Copying to ${host.name}...`);
+    const scpArgs = buildScpArgs(host, tarball, '~/astro-agent.tgz');
+    await execFile('scp', scpArgs);
+    // 3. Install to user-local prefix to avoid EACCES on system dirs
+    log(`Installing on ${host.name}...`);
+    const npmPrefix = '~/.local';
+    await sshExec(host, `mkdir -p ${npmPrefix} && npm install -g --prefix ${npmPrefix} ~/astro-agent.tgz`);
+    // 4. Ensure ~/.local/bin is on PATH for this session and future logins
+    const binDir = `${npmPrefix}/bin`;
+    const pathExport = `export PATH="${binDir}:$PATH"`;
+    const pathSetup = [
+        `grep -qxF '${pathExport}' ~/.bashrc 2>/dev/null || echo '${pathExport}' >> ~/.bashrc`,
+        `grep -qxF '${pathExport}' ~/.profile 2>/dev/null || echo '${pathExport}' >> ~/.profile`,
+    ].join(' && ');
+    await sshExec(host, pathSetup);
+    // 5. Run setup --non-interactive --skip-auth (with binDir on PATH)
+    log(`Running setup on ${host.name}...`);
+    await sshExec(host, `${pathExport} && astro-agent setup --non-interactive --skip-auth --api ${apiUrl} --relay ${relayUrl}`);
+    // 6. Push tokens via stdin to avoid exposing them in ps output
+    log(`Configuring tokens on ${host.name}...`);
+    const configJson = JSON.stringify({
+        apiUrl,
+        relayUrl,
+        accessToken,
+        refreshToken,
+        wsToken,
+        machineId,
+    });
+    // Pipe config as JSON via stdin to avoid shell argument exposure on shared machines
+    const configCmd = `${pathExport} && cat > /tmp/.astro-config-$$.json && ` +
+        `astro-agent config --import /tmp/.astro-config-$$.json && ` +
+        `rm -f /tmp/.astro-config-$$.json`;
+    const sshArgs = buildSshArgs(host, configCmd);
+    const { execFile: execFileCbInner } = await import('node:child_process');
+    const child = execFileCbInner('ssh', sshArgs);
+    child.stdin?.write(configJson);
+    child.stdin?.end();
+    await new Promise((resolve, reject) => {
+        child.on('close', (code) => code === 0 ? resolve() : reject(new Error(`ssh config push exited with code ${code}`)));
+        child.on('error', reject);
+    });
+    // Clean up remote tarball
+    await sshExec(host, 'rm -f ~/astro-agent.tgz').catch(() => { });
+    log(`Done — ${host.name} is configured`);
+}
+// ============================================================================
+// SSH / SCP helpers
+// ============================================================================
+export function buildSshArgs(host, command) {
+    const args = [];
+    if (host.port && host.port !== 22) {
+        args.push('-p', String(host.port));
+    }
+    if (host.identityFile) {
+        args.push('-i', host.identityFile);
+    }
+    if (host.proxyJump) {
+        args.push('-J', host.proxyJump);
+    }
+    // Batch mode to avoid password prompts hanging
+    args.push('-o', 'BatchMode=yes', '-o', 'ConnectTimeout=10');
+    const target = host.user ? `${host.user}@${host.hostname}` : host.hostname;
+    args.push(target, command);
+    return args;
+}
+function buildScpArgs(host, localPath, remotePath) {
+    const args = [];
+    if (host.port && host.port !== 22) {
+        args.push('-P', String(host.port));
+    }
+    if (host.identityFile) {
+        args.push('-i', host.identityFile);
+    }
+    if (host.proxyJump) {
+        args.push('-o', `ProxyJump=${host.proxyJump}`);
+    }
+    args.push('-o', 'BatchMode=yes', '-o', 'ConnectTimeout=10');
+    const target = host.user ? `${host.user}@${host.hostname}` : host.hostname;
+    args.push(localPath, `${target}:${remotePath}`);
+    return args;
+}
+export async function sshExec(host, command) {
+    const args = buildSshArgs(host, command);
+    return execFile('ssh', args, { timeout: 60_000 });
+}
+/**
+ * Start agent runners on remote hosts.
+ *
+ * For each host:
+ *   1. Check if an agent is already running (pgrep)
+ *   2. Start via nohup + disown
+ *   3. Verify after 2s with pgrep
+ *
+ * One failure does not block others.
+ */
+export async function startRemoteAgents(hosts, options = {}, onProgress) {
+    const results = [];
+    const log = (host, msg) => onProgress?.(host, msg);
+    for (const host of hosts) {
+        log(host.name, 'Checking for running agent...');
+        // 1. Check if already running
+        try {
+            const { stdout } = await sshExec(host, 'pgrep -f "astro-agent start"');
+            if (stdout.trim()) {
+                log(host.name, 'Agent already running');
+                results.push({ host, success: true, message: 'Already running', alreadyRunning: true });
+                continue;
+            }
+        }
+        catch {
+            // pgrep returns exit code 1 when no match — that's fine
+        }
+        // 2. Build start command with forwarded options
+        const binDir = '~/.local/bin';
+        const pathExport = `export PATH="${binDir}:$PATH"`;
+        const flags = ['--foreground'];
+        if (options.maxTasks)
+            flags.push(`--max-tasks ${options.maxTasks}`);
+        if (options.logLevel)
+            flags.push(`--log-level ${options.logLevel}`);
+        if (options.preserveWorktrees)
+            flags.push('--preserve-worktrees');
+        const startCmd = `astro-agent start ${flags.join(' ')}`;
+        log(host.name, 'Starting agent...');
+        try {
+            await sshExec(host, `${pathExport} && mkdir -p ~/.astro/logs && nohup ${startCmd} > ~/.astro/logs/agent-runner.log 2>&1 & disown`);
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            results.push({ host, success: false, message: `Failed to start: ${msg}` });
+            continue;
+        }
+        // 3. Verify after 2s
+        await new Promise((r) => setTimeout(r, 2000));
+        try {
+            const { stdout } = await sshExec(host, 'pgrep -f "astro-agent start"');
+            if (stdout.trim()) {
+                log(host.name, 'Agent started successfully');
+                results.push({ host, success: true, message: 'Started' });
+            }
+            else {
+                // Fallback: check log tail
+                try {
+                    const { stdout: logTail } = await sshExec(host, 'tail -5 ~/.astro/logs/agent-runner.log 2>/dev/null');
+                    results.push({ host, success: false, message: `Process not found after start. Log tail:\n${logTail}` });
+                }
+                catch {
+                    results.push({ host, success: false, message: 'Process not found after start (no logs available)' });
+                }
+            }
+        }
+        catch {
+            results.push({ host, success: false, message: 'Could not verify agent start (pgrep failed)' });
+        }
+    }
+    return results;
+}
+//# sourceMappingURL=ssh-installer.js.map

package/dist/lib/ssh-installer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ssh-installer.js","sourceRoot":"","sources":["../../src/lib/ssh-installer.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,QAAQ,IAAI,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAGzC,MAAM,QAAQ,GAAG,SAAS,CAAC,UAAU,CAAC,CAAC;AAEvC,+EAA+E;AAC/E,kBAAkB;AAClB,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,UAAU,aAAa;IAC3B,MAAM,IAAI,GAAG,iBAAiB,EAAE,CAAC;IAEjC,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;QACzC,IAAI,CAAC,MAAM;YAAE,SAAS;QACtB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,MAAM,KAAK,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAC/C,OAAO,KAAK,CAAC,OAAO,CAAC;YACvB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,+EAA+E;AAC/E,gBAAgB;AAChB,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,IAAoB;IAEpB,IAAI,CAAC;QACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;QACzD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,kBAAkB;QAC7C,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAClD,OAAO,EAAE,SAAS,EAAE,KAAK,IAAI,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;IAClD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC7C,CAAC;AACH,CAAC;AAgBD;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,IAAoB,EACpB,UAAkC;IAElC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC;IACvF,MAAM,GAAG,GAAG,UAAU,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAErC,cAAc;IACd,MAAM,cAAc,GAAG,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACpF,GAAG,CAAC,yBAAyB,CAAC,CAAC;IAC/B,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC,MAAM,EAAE,oBAAoB,EAAE,MAAM,CAAC,EAAE;QACxF,GAAG,EAAE,cAAc;KACpB,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;IAE3D,2BAA2B;IAC3B,GAAG,CAAC,cAAc,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC;IAClC,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,EAAE,OAAO,EAAE,mBAAmB,CAAC,CAAC;IACjE,MAAM,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAE/B,iEAAiE;IACjE,GAAG,CAAC,iBAAiB,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC;IACrC,MAAM,SAAS,GAAG,UAAU,CAAC;IAC7B,MAAM,OAAO,CAAC,IAAI,EAAE,YAAY,SAAS,+BAA+B,SAAS,oBAAoB,CAAC,CAAC;IAEvG,uEAAuE;IACvE,MAAM,MAAM,GAAG,GAAG,SAAS,MAAM,CAAC;IAClC,MAAM,UAAU,GAAG,gBAAgB,MAAM,SAAS,CAAC;IACnD,MAAM,SAAS,GAAG;QAChB,cAAc,UAAU,oCAAoC,UAAU,gBAAgB;QACtF,cAAc,UAAU,qCAAqC,UAAU,iBAAiB;KACzF,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACf,MAAM,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IAE/B,mEAAmE;IACnE,GAAG,CAAC,oBAAoB,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC;IACxC,MAAM,OAAO,CACX,IAAI,EACJ,GAAG,UAAU,6DAA6D,MAAM,YAAY,QAAQ,EAAE,CACvG,CAAC;IAEF,+DAA+D;IAC/D,GAAG,CAAC,yBAAyB,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC;IAC7C,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,MAAM;QACN,QAAQ;QACR,WAAW;QACX,YAAY;QACZ,OAAO;QACP,SAAS;KACV,CAAC,CAAC;IACH,oFAAoF;IACpF,MAAM,SAAS,GAAG,GAAG,UAAU,0CAA0C;QACvE,4DAA4D;QAC5D,kCAAkC,CAAC;IACrC,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IAC9C,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;IACzE,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAC9C,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;IAC/B,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC;IACnB,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,oCAAoC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACpH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,0BAA0B;IAC1B,MAAM,OAAO,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAE/D,GAAG,CAAC,UAAU,IAAI,CAAC,IAAI,gBAAgB,CAAC,CAAC;AAC3C,CAAC;AAED,+EAA+E;AAC/E,oBAAoB;AACpB,+EAA+E;AAE/E,MAAM,UAAU,YAAY,CAAC,IAAoB,EAAE,OAAe;IAChE,MAAM,IAAI,GAAa,EAAE,CAAC;IAE1B,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,EAAE,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACrC,CAAC;IACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IACrC,CAAC;IACD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAED,+CAA+C;IAC/C,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE,mBAAmB,CAAC,CAAC;IAE5D,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC;IAC3E,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAE3B,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,YAAY,CAAC,IAAoB,EAAE,SAAiB,EAAE,UAAkB;IAC/E,MAAM,IAAI,GAAa,EAAE,CAAC;IAE1B,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,EAAE,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACrC,CAAC;IACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IACrC,CAAC;IACD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE,mBAAmB,CAAC,CAAC;IAE5D,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC;IAC3E,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,MAAM,IAAI,UAAU,EAAE,CAAC,CAAC;IAEhD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,OAAO,CAC3B,IAAoB,EACpB,OAAe;IAEf,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACzC,OAAO,QAAQ,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;AACpD,CAAC;AAmBD;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,KAAuB,EACvB,UAA8B,EAAE,EAChC,UAAgD;IAEhD,MAAM,OAAO,GAAwB,EAAE,CAAC;IACxC,MAAM,GAAG,GAAG,CAAC,IAAY,EAAE,GAAW,EAAE,EAAE,CAAC,UAAU,EAAE,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAEnE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,+BAA+B,CAAC,CAAC;QAEhD,8BAA8B;QAC9B,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,8BAA8B,CAAC,CAAC;YACvE,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,uBAAuB,CAAC,CAAC;gBACxC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;gBACxF,SAAS;YACX,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,wDAAwD;QAC1D,CAAC;QAED,gDAAgD;QAChD,MAAM,MAAM,GAAG,cAAc,CAAC;QAC9B,MAAM,UAAU,GAAG,gBAAgB,MAAM,SAAS,CAAC;QACnD,MAAM,KAAK,GAAa,CAAC,cAAc,CAAC,CAAC;QACzC,IAAI,OAAO,CAAC,QAAQ;YAAE,KAAK,CAAC,IAAI,CAAC,eAAe,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;QACpE,IAAI,OAAO,CAAC,QAAQ;YAAE,KAAK,CAAC,IAAI,CAAC,eAAe,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;QACpE,IAAI,OAAO,CAAC,iBAAiB;YAAE,KAAK,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAClE,MAAM,QAAQ,GAAG,qBAAqB,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QAExD,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;QACpC,IAAI,CAAC;YACH,MAAM,OAAO,CACX,IAAI,EACJ,GAAG,UAAU,uCAAuC,QAAQ,iDAAiD,CAC9G,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,oBAAoB,GAAG,EAAE,EAAE,CAAC,CAAC;YAC3E,SAAS;QACX,CAAC;QAED,qBAAqB;QACrB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;QAC9C,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,8BAA8B,CAAC,CAAC;YACvE,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClB,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,4BAA4B,CAAC,CAAC;gBAC7C,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YAC5D,CAAC;iBAAM,CAAC;gBACN,2BAA2B;gBAC3B,IAAI,CAAC;oBACH,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,oDAAoD,CAAC,CAAC;oBACtG,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,6CAA6C,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC1G,CAAC;gBAAC,MAAM,CAAC;oBACP,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,mDAAmD,EAAE,CAAC,CAAC;gBACvG,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,6CAA6C,EAAE,CAAC,CAAC;QACjG,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/lib/streaming-prompt.d.ts

@@ -0,0 +1,48 @@
+/**
+ * StreamingPrompt - Async iterator queue for injecting messages into a running SDK session.
+ *
+ * Ported from Cyrus's StreamingPrompt pattern. Enables mid-execution steering
+ * by allowing new user messages to be pushed into a running `query()` call.
+ *
+ * Usage:
+ *   const prompt = new StreamingPrompt('initial task prompt');
+ *   const gen = query({ prompt, options });
+ *   // Later, inject a steer message:
+ *   prompt.addMessage('Please also fix the tests');
+ *   // When done:
+ *   prompt.complete();
+ */
+interface SDKUserMessage {
+    role: 'user';
+    content: string;
+}
+export declare class StreamingPrompt {
+    private queue;
+    private resolve;
+    private done;
+    private sessionId?;
+    constructor(initialPrompt: string);
+    /**
+     * Add a new user message to the running session.
+     * The SDK will pick this up as the next turn's input.
+     */
+    addMessage(content: string): void;
+    /**
+     * Signal that no more messages will be sent.
+     */
+    complete(): void;
+    /**
+     * Update the session ID (called after first SDK init message).
+     */
+    updateSessionId(id: string): void;
+    /**
+     * Get the current session ID.
+     */
+    getSessionId(): string | undefined;
+    /**
+     * Async iterator protocol — used by the SDK's `query()` to consume messages.
+     */
+    [Symbol.asyncIterator](): AsyncIterator<SDKUserMessage>;
+}
+export {};
+//# sourceMappingURL=streaming-prompt.d.ts.map

package/dist/lib/streaming-prompt.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming-prompt.d.ts","sourceRoot":"","sources":["../../src/lib/streaming-prompt.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,UAAU,cAAc;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;CACjB;AAMD,qBAAa,eAAe;IAC1B,OAAO,CAAC,KAAK,CAAmB;IAChC,OAAO,CAAC,OAAO,CAAmE;IAClF,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,SAAS,CAAC,CAAS;gBAEf,aAAa,EAAE,MAAM;IAOjC;;;OAGG;IACH,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAgBjC;;OAEG;IACH,QAAQ,IAAI,IAAI;IAShB;;OAEG;IACH,eAAe,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAIjC;;OAEG;IACH,YAAY,IAAI,MAAM,GAAG,SAAS;IAIlC;;OAEG;IACH,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,cAAc,CAAC;CAqBxD"}

package/dist/lib/streaming-prompt.js

@@ -0,0 +1,91 @@
+/**
+ * StreamingPrompt - Async iterator queue for injecting messages into a running SDK session.
+ *
+ * Ported from Cyrus's StreamingPrompt pattern. Enables mid-execution steering
+ * by allowing new user messages to be pushed into a running `query()` call.
+ *
+ * Usage:
+ *   const prompt = new StreamingPrompt('initial task prompt');
+ *   const gen = query({ prompt, options });
+ *   // Later, inject a steer message:
+ *   prompt.addMessage('Please also fix the tests');
+ *   // When done:
+ *   prompt.complete();
+ */
+export class StreamingPrompt {
+    queue = [];
+    resolve = null;
+    done = false;
+    sessionId;
+    constructor(initialPrompt) {
+        // Seed the queue with the initial prompt as the first user message
+        this.queue.push({
+            value: { role: 'user', content: initialPrompt },
+        });
+    }
+    /**
+     * Add a new user message to the running session.
+     * The SDK will pick this up as the next turn's input.
+     */
+    addMessage(content) {
+        if (this.done)
+            return;
+        const message = { role: 'user', content };
+        if (this.resolve) {
+            // Consumer is waiting — deliver immediately
+            const r = this.resolve;
+            this.resolve = null;
+            r({ value: message, done: false });
+        }
+        else {
+            // Consumer hasn't asked yet — buffer
+            this.queue.push({ value: message });
+        }
+    }
+    /**
+     * Signal that no more messages will be sent.
+     */
+    complete() {
+        this.done = true;
+        if (this.resolve) {
+            const r = this.resolve;
+            this.resolve = null;
+            r({ value: undefined, done: true });
+        }
+    }
+    /**
+     * Update the session ID (called after first SDK init message).
+     */
+    updateSessionId(id) {
+        this.sessionId = id;
+    }
+    /**
+     * Get the current session ID.
+     */
+    getSessionId() {
+        return this.sessionId;
+    }
+    /**
+     * Async iterator protocol — used by the SDK's `query()` to consume messages.
+     */
+    [Symbol.asyncIterator]() {
+        return {
+            next: () => {
+                // If there's buffered data, return it immediately
+                if (this.queue.length > 0) {
+                    const item = this.queue.shift();
+                    return Promise.resolve({ value: item.value, done: false });
+                }
+                // If stream is complete, signal done
+                if (this.done) {
+                    return Promise.resolve({ value: undefined, done: true });
+                }
+                // Otherwise, wait for the next message
+                return new Promise((resolve) => {
+                    this.resolve = resolve;
+                });
+            },
+        };
+    }
+}
+//# sourceMappingURL=streaming-prompt.js.map

package/dist/lib/streaming-prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming-prompt.js","sourceRoot":"","sources":["../../src/lib/streaming-prompt.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAWH,MAAM,OAAO,eAAe;IAClB,KAAK,GAAgB,EAAE,CAAC;IACxB,OAAO,GAA8D,IAAI,CAAC;IAC1E,IAAI,GAAG,KAAK,CAAC;IACb,SAAS,CAAU;IAE3B,YAAY,aAAqB;QAC/B,mEAAmE;QACnE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;YACd,KAAK,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,UAAU,CAAC,OAAe;QACxB,IAAI,IAAI,CAAC,IAAI;YAAE,OAAO;QAEtB,MAAM,OAAO,GAAmB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;QAE1D,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,4CAA4C;YAC5C,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;YACvB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,qCAAqC;YACrC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;YACvB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,CAAC,CAAC,EAAE,KAAK,EAAE,SAAsC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,EAAU;QACxB,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,CAAC,MAAM,CAAC,aAAa,CAAC;QACpB,OAAO;YACL,IAAI,EAAE,GAA4C,EAAE;gBAClD,kDAAkD;gBAClD,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAG,CAAC;oBACjC,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC7D,CAAC;gBAED,qCAAqC;gBACrC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBACd,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,SAAsC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;gBACxF,CAAC;gBAED,uCAAuC;gBACvC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;oBAC7B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;gBACzB,CAAC,CAAC,CAAC;YACL,CAAC;SACF,CAAC;IACJ,CAAC;CACF"}

package/dist/lib/task-executor.d.ts

@@ -0,0 +1,114 @@
+/**
+ * Task executor with multiplexing support
+ *
+ * Handles concurrent task execution, routing to providers,
+ * and streaming output back over WebSocket
+ */
+import type { Task } from '../types.js';
+import type { WebSocketClient } from './websocket-client.js';
+export interface TaskExecutorOptions {
+    wsClient: WebSocketClient;
+    maxConcurrentTasks?: number;
+    defaultTimeout?: number;
+    useWorktree?: boolean;
+    worktreeRoot?: string;
+    preserveWorktrees?: boolean;
+    allowNonGit?: boolean;
+    useSandbox?: boolean;
+    maxSandboxSize?: number;
+}
+export declare class TaskExecutor {
+    private wsClient;
+    private runningTasks;
+    private taskQueue;
+    private maxConcurrentTasks;
+    private defaultTimeout;
+    private adapters;
+    private useWorktree;
+    private worktreeRoot?;
+    private preserveWorktrees;
+    private jobMonitor;
+    private allowNonGit;
+    private useSandbox;
+    private maxSandboxSize;
+    private gitAvailable;
+    private tasksByDirectory;
+    private pendingSafetyChecks;
+    constructor(options: TaskExecutorOptions);
+    /**
+     * Submit a task for execution (with safety checks)
+     */
+    submitTask(task: Task): Promise<void>;
+    /**
+     * Handle safety decision from user (via server)
+     */
+    handleSafetyDecision(taskId: string, decision: 'proceed' | 'init-git' | 'sandbox' | 'cancel'): Promise<void>;
+    /**
+     * Cancel a running or queued task
+     */
+    cancelTask(taskId: string): boolean;
+    /**
+     * Get current task counts
+     */
+    getTaskCounts(): {
+        running: number;
+        queued: number;
+    };
+    /**
+     * Cancel all tasks and clear queue
+     */
+    cancelAll(): void;
+    /**
+     * Update max concurrent tasks
+     */
+    setMaxConcurrentTasks(max: number): void;
+    /**
+     * Steer a running task by injecting a message into the agent session.
+     * When interrupt=true, interrupts the current turn before injecting.
+     *
+     * For completed tasks with a preserved session, this triggers a full
+     * resume via the SDK's `resume` option for post-completion follow-up.
+     */
+    steerTask(taskId: string, message: string, interrupt?: boolean): Promise<{
+        accepted: boolean;
+        reason?: string;
+    }>;
+    /**
+     * Find the ClaudeSdkAdapter that has a preserved session for the given task.
+     */
+    private findAdapterWithSession;
+    /**
+     * Resume a completed task session for post-completion steering.
+     * Runs in the background, streaming output back through the WebSocket.
+     */
+    private resumeCompletedTask;
+    /**
+     * Perform safety check on working directory
+     */
+    private performSafetyCheck;
+    /**
+     * Request safety decision from user
+     */
+    private requestSafetyDecision;
+    /**
+     * Initialize git repository in a directory
+     */
+    private initializeGit;
+    /**
+     * Track task by directory for parallel execution safety
+     */
+    private trackTaskDirectory;
+    /**
+     * Untrack task from directory
+     */
+    private untrackTaskDirectory;
+    /**
+     * Get count of active tasks in a directory
+     */
+    private getActiveTasksInDirectory;
+    private executeTask;
+    private prepareTaskWorkspace;
+    private getAdapter;
+    private processQueue;
+}
+//# sourceMappingURL=task-executor.d.ts.map

package/dist/lib/task-executor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-executor.d.ts","sourceRoot":"","sources":["../../src/lib/task-executor.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,IAAI,EAA4B,MAAM,aAAa,CAAC;AAClE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AA6B7D,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,eAAe,CAAC;IAC1B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAkB;IAClC,OAAO,CAAC,YAAY,CAAuC;IAC3D,OAAO,CAAC,SAAS,CAAc;IAC/B,OAAO,CAAC,kBAAkB,CAAS;IACnC,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAiD;IACjE,OAAO,CAAC,WAAW,CAAU;IAC7B,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,iBAAiB,CAAU;IACnC,OAAO,CAAC,UAAU,CAAkB;IACpC,OAAO,CAAC,WAAW,CAAU;IAC7B,OAAO,CAAC,UAAU,CAAU;IAC5B,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,YAAY,CAAkB;IAGtC,OAAO,CAAC,gBAAgB,CAAuC;IAC/D,OAAO,CAAC,mBAAmB,CAA8C;gBAE7D,OAAO,EAAE,mBAAmB;IAqBxC;;OAEG;IACG,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IA8C3C;;OAEG;IACG,oBAAoB,CACxB,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,SAAS,GAAG,UAAU,GAAG,SAAS,GAAG,QAAQ,GACtD,OAAO,CAAC,IAAI,CAAC;IAWhB;;OAEG;IACH,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAuBnC;;OAEG;IACH,aAAa,IAAI;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE;IAOpD;;OAEG;IACH,SAAS,IAAI,IAAI;IA0BjB;;OAEG;IACH,qBAAqB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAMxC;;;;;;OAMG;IACG,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,UAAQ,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IA6BpH;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAU9B;;;OAGG;YACW,mBAAmB;IA2EjC;;OAEG;YACW,kBAAkB;IAKhC;;OAEG;YACW,qBAAqB;IA8FnC;;OAEG;YACW,aAAa;IAU3B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAM1B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAU5B;;OAEG;IACH,OAAO,CAAC,yBAAyB;YAKnB,WAAW;YA4OX,oBAAoB;YAkEpB,UAAU;IA6BxB,OAAO,CAAC,YAAY;CAUrB"}