RPC: stream heartbeat events to menu

This commit is contained in:
Peter Steinberger
2025-12-08 20:18:54 +01:00
parent e3089d60ea
commit ab448988ff
5 changed files with 195 additions and 0 deletions

View File

@@ -23,6 +23,10 @@ import {
resolveHeartbeatSeconds,
resolveReconnectPolicy,
} from "../web/reconnect.js";
import {
readLatestHeartbeat,
tailHeartbeatEvents,
} from "../process/heartbeat-events.js";
import {
ensureWebChatServerFromConfig,
startWebChatServer,
@@ -241,6 +245,14 @@ Examples:
}
};
const forwardHeartbeat = (payload: unknown) => {
respond({ type: "event", event: "heartbeat", payload });
};
const latest = readLatestHeartbeat();
if (latest) forwardHeartbeat(latest);
const stopTail = tailHeartbeatEvents(forwardHeartbeat);
rl.on("line", async (line: string) => {
if (!line.trim()) return;
try {
@@ -311,6 +323,8 @@ Examples:
};
await new Promise(() => {});
stopTail();
});
program

View File

@@ -0,0 +1,78 @@
import fs from "node:fs";
import os from "node:os";
import path from "node:path";
import readline from "node:readline";
export type HeartbeatEvent = {
type: "heartbeat";
ts: number; // epoch ms
status:
| "sent"
| "ok-empty"
| "ok-token"
| "skipped"
| "failed";
to?: string;
preview?: string;
durationMs?: number;
hasMedia?: boolean;
reason?: string;
};
const EVENT_FILENAME = "heartbeat-events.jsonl";
const STATE_FILENAME = "heartbeat-state.json";
function baseDir() {
const dir = path.join(os.homedir(), ".clawdis");
if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
return dir;
}
export function heartbeatEventsPath() {
return path.join(baseDir(), EVENT_FILENAME);
}
export function heartbeatStatePath() {
return path.join(baseDir(), STATE_FILENAME);
}
export function writeHeartbeatEvent(evt: HeartbeatEvent) {
const line = JSON.stringify(evt);
fs.appendFileSync(heartbeatEventsPath(), `${line}\n`, { encoding: "utf8" });
fs.writeFileSync(heartbeatStatePath(), line, { encoding: "utf8" });
}
export function readLatestHeartbeat(): HeartbeatEvent | null {
try {
const txt = fs.readFileSync(heartbeatStatePath(), "utf8");
return JSON.parse(txt) as HeartbeatEvent;
} catch {
return null;
}
}
// Tail the events file and invoke the callback for every new parsed event.
export function tailHeartbeatEvents(onEvent: (evt: HeartbeatEvent) => void) {
const file = heartbeatEventsPath();
if (!fs.existsSync(file)) {
fs.writeFileSync(file, "", { encoding: "utf8" });
}
const stream = fs.createReadStream(file, { encoding: "utf8", flags: "a+" });
const rl = readline.createInterface({ input: stream });
rl.on("line", (line) => {
const trimmed = line.trim();
if (!trimmed) return;
try {
const parsed = JSON.parse(trimmed) as HeartbeatEvent;
if (parsed?.type === "heartbeat") onEvent(parsed);
} catch {
// ignore malformed
}
});
return () => {
rl.close();
stream.close();
};
}

View File

@@ -14,6 +14,10 @@ import { danger, info, isVerbose, logVerbose, success } from "../globals.js";
import { logInfo } from "../logger.js";
import { getChildLogger } from "../logging.js";
import { getQueueSize } from "../process/command-queue.js";
import {
type HeartbeatEvent,
writeHeartbeatEvent,
} from "../process/heartbeat-events.js";
import { defaultRuntime, type RuntimeEnv } from "../runtime.js";
import { jidToE164, normalizeE164 } from "../utils.js";
import { monitorWebInbox } from "./inbound.js";
@@ -78,6 +82,10 @@ const formatDuration = (ms: number) =>
const DEFAULT_REPLY_HEARTBEAT_MINUTES = 30;
export const HEARTBEAT_TOKEN = "HEARTBEAT_OK";
function emitHeartbeatEvent(evt: Omit<HeartbeatEvent, "type" | "ts">) {
writeHeartbeatEvent({ type: "heartbeat", ts: Date.now(), ...evt });
}
export const HEARTBEAT_PROMPT = "HEARTBEAT /think:high";
function elide(text?: string, limit = 400) {
@@ -261,6 +269,12 @@ export async function runWebHeartbeatOnce(opts: {
return;
}
const sendResult = await sender(to, overrideBody, { verbose });
emitHeartbeatEvent({
status: "sent",
to,
preview: overrideBody.slice(0, 160),
hasMedia: false,
});
heartbeatLogger.info(
{
to,
@@ -307,6 +321,7 @@ export async function runWebHeartbeatOnce(opts: {
"heartbeat skipped",
);
if (verbose) console.log(success("heartbeat: ok (empty reply)"));
emitHeartbeatEvent({ status: "ok-empty", to });
return;
}
@@ -328,6 +343,7 @@ export async function runWebHeartbeatOnce(opts: {
"heartbeat skipped",
);
console.log(success("heartbeat: ok (HEARTBEAT_OK)"));
emitHeartbeatEvent({ status: "ok-token", to });
return;
}
@@ -351,6 +367,12 @@ export async function runWebHeartbeatOnce(opts: {
}
const sendResult = await sender(to, finalText, { verbose });
emitHeartbeatEvent({
status: "sent",
to,
preview: finalText.slice(0, 160),
hasMedia,
});
heartbeatLogger.info(
{
to,
@@ -364,6 +386,7 @@ export async function runWebHeartbeatOnce(opts: {
} catch (err) {
heartbeatLogger.warn({ to, error: String(err) }, "heartbeat failed");
console.log(danger(`heartbeat: failed - ${String(err)}`));
emitHeartbeatEvent({ status: "failed", to, reason: String(err) });
throw err;
}
}