let5see/clawdbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

RPC: extract stdio loop and tests

Browse Source
This commit is contained in:
Peter Steinberger
2025-12-09 02:37:04 +00:00
parent c568284f1b
commit 59a2cbefcb
3 changed files with 288 additions and 155 deletions
Download Patch File Download Diff File Expand all files Collapse all files

161
src/cli/program.ts
View File

@@ -1,17 +1,13 @@
import chalk from "chalk";
import { Command } from "commander";
import { agentCommand } from "../commands/agent.js";
import { getHealthSnapshot, healthCommand, type HealthSummary } from "../commands/health.js";
import { healthCommand } from "../commands/health.js";
import { sendCommand } from "../commands/send.js";
import { sessionsCommand } from "../commands/sessions.js";
import { getStatusSummary, statusCommand, type StatusSummary } from "../commands/status.js";
import { statusCommand } from "../commands/status.js";
import { loadConfig } from "../config/config.js";
import { danger, info, setVerbose } from "../globals.js";
import { startControlChannel } from "../infra/control-channel.js";
import {
getLastHeartbeatEvent,
onHeartbeatEvent,
} from "../infra/heartbeat-events.js";
import { getResolvedLoggerSettings } from "../logging.js";
import {
loginWeb,
@@ -22,7 +18,8 @@ import {
setHeartbeatsEnabled,
type WebMonitorTuning,
} from "../provider-web.js";
import { defaultRuntime, type RuntimeEnv } from "../runtime.js";
import { runRpcLoop } from "../rpc/loop.js";
import { defaultRuntime } from "../runtime.js";
import { VERSION } from "../version.js";
import {
resolveHeartbeatSeconds,
@@ -33,12 +30,6 @@ import {
startWebChatServer,
} from "../webchat/server.js";
import { createDefaultDeps, logWebSelfId } from "./deps.js";
import { onAgentEvent } from "../infra/agent-events.js";
import { enqueueSystemEvent } from "../infra/system-events.js";
import {
listSystemPresence,
updateSystemPresence,
} from "../infra/system-presence.js";
export function buildProgram() {
const program = new Command();
@@ -241,148 +232,8 @@ Examples:
.command("rpc")
.description("Run stdin/stdout JSON RPC loop for agent sends")
.action(async () => {
const { createInterface } = await import("node:readline");
const rl = createInterface({ input: process.stdin, crlfDelay: Infinity });
const respond = (obj: unknown) => {
try {
console.log(JSON.stringify(obj));
} catch (err) {
console.error(JSON.stringify({ type: "error", error: String(err) }));
}
};
const forwardHeartbeat = (payload: unknown) => {
respond({ type: "event", event: "heartbeat", payload });
};
const forwardAgent = (payload: unknown) => {
respond({ type: "event", event: "agent", payload });
};
const latest = getLastHeartbeatEvent();
if (latest) forwardHeartbeat(latest);
const stopBus = onHeartbeatEvent(forwardHeartbeat);
const stopAgentBus = onAgentEvent(forwardAgent);
rl.on("line", async (line: string) => {
if (!line.trim()) return;
try {
const cmd = JSON.parse(line);
if (cmd.type === "status") {
respond({ type: "result", ok: true });
return;
}
if (cmd.type === "set-heartbeats") {
setHeartbeatsEnabled(Boolean(cmd.enabled));
respond({ type: "result", ok: true });
return;
}
if (cmd.type === "control-request" && cmd.id && cmd.method) {
const id = String(cmd.id);
const method = String(cmd.method);
const params = (cmd.params ?? {}) as Record<string, unknown>;
const controlRespond = (ok: boolean, payload?: unknown, error?: string) =>
respond({ type: "control-response", id, ok, payload, error });
try {
if (method === "health") {
const timeoutMs = typeof params.timeoutMs === "number" ? params.timeoutMs : undefined;
const payload = await getHealthSnapshot(timeoutMs);
controlRespond(true, payload satisfies HealthSummary);
return;
}
if (method === "status") {
const payload = await getStatusSummary();
controlRespond(true, payload satisfies StatusSummary);
return;
}
if (method === "last-heartbeat") {
controlRespond(true, getLastHeartbeatEvent());
return;
}
if (method === "set-heartbeats") {
setHeartbeatsEnabled(Boolean(params.enabled));
controlRespond(true, { ok: true });
return;
}
if (method === "system-event") {
const text = String(params.text ?? "").trim();
if (text) {
enqueueSystemEvent(text);
updateSystemPresence(text);
}
controlRespond(true, { ok: true });
return;
}
if (method === "system-presence") {
controlRespond(true, listSystemPresence());
return;
}
controlRespond(false, undefined, `unknown control method: ${method}`);
} catch (err) {
controlRespond(false, undefined, String(err));
}
return;
}
if (cmd.type !== "send" || !cmd.text) {
respond({ type: "error", error: "unsupported command" });
return;
}
const logs: string[] = [];
const runtime: RuntimeEnv = {
log: (msg: string) => logs.push(String(msg)),
error: (msg: string) => logs.push(String(msg)),
exit: (_code: number): never => {
throw new Error("agentCommand requested exit");
},
};
const opts: {
message: string;
to?: string;
sessionId?: string;
thinking?: string;
deliver?: boolean;
json: boolean;
} = {
message: String(cmd.text),
to: cmd.to ? String(cmd.to) : undefined,
sessionId: cmd.session ? String(cmd.session) : undefined,
thinking: cmd.thinking ? String(cmd.thinking) : undefined,
deliver: Boolean(cmd.deliver),
json: true,
};
try {
await agentCommand(opts, runtime, createDefaultDeps());
const payload = extractPayload(logs);
respond({ type: "result", ok: true, payload });
} catch (err) {
respond({ type: "error", error: String(err) });
}
} catch (err) {
respond({ type: "error", error: `parse error: ${String(err)}` });
}
});
const extractPayload = (logs: string[]) => {
for (const entry of logs.slice().reverse()) {
try {
const parsed = JSON.parse(entry);
if (parsed && typeof parsed === "object" && "payloads" in parsed) {
return parsed;
}
} catch {
// non-JSON log, ignore
}
}
return null;
};
await new Promise(() => {});
stopBus();
stopAgentBus();
await runRpcLoop({ input: process.stdin, output: process.stdout });
await new Promise<never>(() => {});
});
program