feat: emit job-state events from rpc
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
import chalk from "chalk";
|
||||
import { randomUUID } from "node:crypto";
|
||||
import { Command } from "commander";
|
||||
import { agentCommand } from "../commands/agent.js";
|
||||
import { healthCommand } from "../commands/health.js";
|
||||
@@ -272,6 +273,14 @@ Examples:
|
||||
return;
|
||||
}
|
||||
|
||||
const jobId = cmd.jobId ? String(cmd.jobId) : randomUUID();
|
||||
const startedAt = Date.now();
|
||||
respond({
|
||||
type: "event",
|
||||
event: "job-state",
|
||||
payload: { id: jobId, state: "started", startedAt },
|
||||
});
|
||||
|
||||
const logs: string[] = [];
|
||||
const runtime: RuntimeEnv = {
|
||||
log: (msg: string) => logs.push(String(msg)),
|
||||
@@ -299,9 +308,31 @@ Examples:
|
||||
|
||||
try {
|
||||
await agentCommand(opts, runtime, createDefaultDeps());
|
||||
const endedAt = Date.now();
|
||||
respond({
|
||||
type: "event",
|
||||
event: "job-state",
|
||||
payload: {
|
||||
id: jobId,
|
||||
state: "done",
|
||||
durationMs: endedAt - startedAt,
|
||||
endedAt,
|
||||
},
|
||||
});
|
||||
const payload = extractPayload(logs);
|
||||
respond({ type: "result", ok: true, payload });
|
||||
} catch (err) {
|
||||
const endedAt = Date.now();
|
||||
respond({
|
||||
type: "event",
|
||||
event: "job-state",
|
||||
payload: {
|
||||
id: jobId,
|
||||
state: "error",
|
||||
durationMs: endedAt - startedAt,
|
||||
endedAt,
|
||||
},
|
||||
});
|
||||
respond({ type: "error", error: String(err) });
|
||||
}
|
||||
} catch (err) {
|
||||
|
||||
@@ -50,11 +50,18 @@ describe("control channel", () => {
|
||||
const frame = { type: "request", id, method, params };
|
||||
client.write(`${JSON.stringify(frame)}\n`);
|
||||
const onData = (chunk: Buffer) => {
|
||||
const line = chunk.toString("utf8").trim();
|
||||
const parsed = JSON.parse(line) as { id?: string };
|
||||
if (parsed.id === id) {
|
||||
client.off("data", onData);
|
||||
resolve(parsed as Record<string, unknown>);
|
||||
const lines = chunk.toString("utf8").trim().split(/\n/);
|
||||
for (const line of lines) {
|
||||
try {
|
||||
const parsed = JSON.parse(line) as { id?: string };
|
||||
if (parsed.id === id) {
|
||||
client.off("data", onData);
|
||||
resolve(parsed as Record<string, unknown>);
|
||||
return;
|
||||
}
|
||||
} catch {
|
||||
/* ignore non-JSON noise */
|
||||
}
|
||||
}
|
||||
};
|
||||
client.on("data", onData);
|
||||
|
||||
Reference in New Issue
Block a user