will/agentmon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(hooks): consolidate shared transport helpers

Browse Source
This commit is contained in:
William Valentin
2026-03-26 11:22:27 -07:00
parent d49785cb25
commit fdfcb50e80
10 changed files with 374 additions and 948 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hooks/claude-code/handler.ts
+31 -197
View File
@@ -3,15 +3,22 @@ import { randomUUID } from 'node:crypto';
import { hostname, homedir } from 'node:os';
import { readFileSync, writeFileSync, mkdirSync, unlinkSync } from 'node:fs';
import { join } from 'node:path';
import {
type Dict,
isRecord,
pickString,
pickNumber,
truncate,
buildEnvelope,
createTransport,
readStdin,
} from '../shared/lib';
const INGEST_URL = process.env.AGENTMON_INGEST_URL || 'http://localhost:8080';
const FRAMEWORK = process.env.AGENTMON_FRAMEWORK || 'claude-code';
const HOST = process.env.AGENTMON_HOST || hostname();
const BATCH_SIZE = 10;
const FLUSH_MS = 2000;
const FETCH_TIMEOUT_MS = 500;
interface Dict { [key: string]: any }
const { enqueue, flush } = createTransport(INGEST_URL);
// ── Persisted state (survives between hook subprocess invocations) ──────────
interface SessionState {
@@ -49,60 +56,10 @@ function clearState(sessionKey: string) {
}
// ─────────────────────────────────────────────────────────────────────────────
let buffer: Dict[] = [];
let flushTimer: ReturnType<typeof setTimeout> | null = null;
let isFlushing = false;
const activeRuns = new Map<string, string>();
const activeSpans = new Map<string, string>();
const activeSubagents = new Map<string, { name: string; spanId: string }>();
function isRecord(value: unknown): value is Dict {
return value !== null && typeof value === 'object' && !Array.isArray(value);
}
function pickString(...values: unknown[]): string | undefined {
for (const value of values) {
if (typeof value === 'string' && value.trim() !== '') {
return value;
}
}
return undefined;
}
function pickNumber(...values: unknown[]): number | undefined {
for (const value of values) {
if (typeof value === 'number' && Number.isFinite(value)) {
return value;
}
}
return undefined;
}
function truncate(value: unknown, limit: number): string | undefined {
if (value === undefined || value === null) {
return undefined;
}
const text = typeof value === 'string' ? value : safeJSONStringify(value);
if (!text) {
return undefined;
}
if (text.length <= limit) {
return text;
}
return text.slice(0, limit) + '...';
}
function safeJSONStringify(value: unknown): string {
try {
return JSON.stringify(value);
} catch {
return String(value);
}
}
function getSessionKey(input: Dict): string | undefined {
return pickString(
input.sessionId,
@@ -113,11 +70,11 @@ function getSessionKey(input: Dict): string | undefined {
}
function getUsage(input: Dict): Dict | undefined {
const usage = isRecord(input.usage) ? input.usage :
isRecord(input.llm) ? input.llm :
const usage = isRecord(input.usage) ? input.usage :
isRecord(input.llm) ? input.llm :
undefined;
if (!usage) return undefined;
const result: Dict = {};
if (usage.input_tokens !== undefined) result.input_tokens = usage.input_tokens;
if (usage.output_tokens !== undefined) result.output_tokens = usage.output_tokens;
@@ -126,7 +83,7 @@ function getUsage(input: Dict): Dict | undefined {
if (usage.thinking_tokens !== undefined) result.thinking_tokens = usage.thinking_tokens;
if (usage.total_tokens !== undefined) result.total_tokens = usage.total_tokens;
if (usage.total_cost !== undefined) result.total_cost = usage.total_cost;
return Object.keys(result).length > 0 ? result : undefined;
}
@@ -134,130 +91,17 @@ function getContextWindow(input: Dict): Dict | undefined {
const stats = isRecord(input.context_stats) ? input.context_stats :
isRecord(input.stats) ? input.stats : undefined;
if (!stats) return undefined;
const result: Dict = {};
if (stats.input_tokens !== undefined) result.input_tokens = stats.input_tokens;
if (stats.output_tokens !== undefined) result.output_tokens = stats.output_tokens;
if (stats.used_tokens !== undefined) result.used_tokens = stats.used_tokens;
if (stats.max_tokens !== undefined) result.max_tokens = stats.max_tokens;
if (stats.tokens_remaining !== undefined) result.tokens_remaining = stats.tokens_remaining;
return Object.keys(result).length > 0 ? result : undefined;
}
function buildEnvelope(
type: string,
sessionKey?: string,
opts: {
runId?: string;
spanId?: string;
parentSpanId?: string;
attributes?: Dict;
payload?: Dict;
} = {},
): Dict {
const correlation: Dict = {};
if (sessionKey) {
correlation.session_id = sessionKey;
}
if (opts.runId) {
correlation.run_id = opts.runId;
}
if (opts.spanId) {
correlation.span_id = opts.spanId;
}
if (opts.parentSpanId) {
correlation.parent_span_id = opts.parentSpanId;
}
const envelope: Dict = {
schema: { name: 'agentmon.event', version: 1 },
event: {
id: randomUUID(),
type,
ts: new Date().toISOString(),
source: {
framework: FRAMEWORK,
client_id: HOST,
host: HOST,
},
},
};
if (Object.keys(correlation).length > 0) {
envelope.correlation = correlation;
}
if (opts.attributes && Object.keys(opts.attributes).length > 0) {
envelope.attributes = opts.attributes;
}
if (opts.payload && Object.keys(opts.payload).length > 0) {
envelope.payload = opts.payload;
}
return envelope;
}
function scheduleFlush() {
if (!flushTimer) {
flushTimer = setTimeout(() => {
void flush();
}, FLUSH_MS);
}
}
function enqueue(event: Dict) {
buffer.push(event);
if (buffer.length >= BATCH_SIZE) {
void flush();
} else {
scheduleFlush();
}
}
async function postBatch(batch: Dict[]) {
const controller = new AbortController();
const timeout = setTimeout(() => controller.abort(), FETCH_TIMEOUT_MS);
try {
await fetch(`${INGEST_URL}/v1/events`, {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(batch),
signal: controller.signal,
});
} finally {
clearTimeout(timeout);
}
}
async function flush() {
if (flushTimer) {
clearTimeout(flushTimer);
flushTimer = null;
}
if (isFlushing || buffer.length === 0) {
return;
}
isFlushing = true;
const batch = buffer.splice(0, BATCH_SIZE);
try {
await postBatch(batch);
} catch {
console.debug(`[agentmon] failed to flush ${batch.length} events`);
} finally {
isFlushing = false;
if (buffer.length > 0) {
if (buffer.length >= BATCH_SIZE) {
void flush();
} else {
scheduleFlush();
}
}
}
}
function emitError(sessionKey: string | undefined, runId: string | undefined, spanId: string | undefined, errorValue: unknown) {
if (errorValue === undefined || errorValue === null || errorValue === false) {
return;
@@ -267,7 +111,7 @@ function emitError(sessionKey: string | undefined, runId: string | undefined, sp
const message = pickString(errorRecord.message, errorRecord.error, errorValue) || 'unknown';
const errType = pickString(errorRecord.type, errorRecord.code) || FRAMEWORK;
enqueue(buildEnvelope('error', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'error', sessionKey, {
runId,
spanId,
payload: {
@@ -287,11 +131,11 @@ async function handleSessionStart(input: Dict) {
const contextWindow = getContextWindow(input);
enqueue(buildEnvelope('session.start', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'session.start', sessionKey, {
attributes: contextWindow ? { context_window: contextWindow } : undefined,
}));
enqueue(buildEnvelope('run.start', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'run.start', sessionKey, {
runId,
attributes: {
trigger: pickString(input.trigger_type, input.trigger),
@@ -313,7 +157,7 @@ async function handleSessionEnd(input: Dict) {
const duration = pickNumber(input.duration_ms, input.elapsed_ms);
if (runId) {
enqueue(buildEnvelope('run.end', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'run.end', sessionKey, {
runId,
payload: {
status: 'success',
@@ -325,7 +169,7 @@ async function handleSessionEnd(input: Dict) {
}));
}
enqueue(buildEnvelope('session.end', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'session.end', sessionKey, {
payload: {
...(usage && { usage }),
...(contextWindow && { context_window: contextWindow }),
@@ -346,7 +190,7 @@ async function handlePromptSubmit(input: Dict) {
const prompt = pickString(input.prompt, input.text, input.message);
if (runId && prompt) {
enqueue(buildEnvelope('run.end', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'run.end', sessionKey, {
runId,
payload: {
status: 'success',
@@ -361,7 +205,7 @@ async function handlePromptSubmit(input: Dict) {
saveState(sessionKey, { runId: newRunId, spans: {} });
}
enqueue(buildEnvelope('run.start', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'run.start', sessionKey, {
runId: newRunId,
attributes: {
type: 'user_prompt',
@@ -392,7 +236,7 @@ async function handleToolStart(input: Dict) {
if (sessionKey) saveState(sessionKey, state);
}
enqueue(buildEnvelope('span.start', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'span.start', sessionKey, {
runId,
spanId,
attributes: {
@@ -423,7 +267,7 @@ async function handleToolEnd(input: Dict) {
if (sessionKey) saveState(sessionKey, state);
}
enqueue(buildEnvelope('span.end', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'span.end', sessionKey, {
runId,
spanId,
attributes: {
@@ -462,7 +306,7 @@ async function handleSubagentStart(input: Dict) {
saveState(sessionKey, state);
}
enqueue(buildEnvelope('span.start', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'span.start', sessionKey, {
runId,
spanId,
attributes: {
@@ -493,7 +337,7 @@ async function handleSubagentStop(input: Dict) {
if (sessionKey) saveState(sessionKey, state);
}
enqueue(buildEnvelope('span.end', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'span.end', sessionKey, {
runId,
spanId,
attributes: {
@@ -528,7 +372,7 @@ async function handleCompactStart(input: Dict) {
saveState(sessionKey, state);
}
enqueue(buildEnvelope('span.start', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'span.start', sessionKey, {
runId,
spanId,
attributes: {
@@ -554,7 +398,7 @@ async function handleCompactEnd(input: Dict) {
if (sessionKey) saveState(sessionKey, state);
}
enqueue(buildEnvelope('span.end', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'span.end', sessionKey, {
runId,
spanId,
attributes: {
@@ -585,7 +429,7 @@ async function handleNotification(input: Dict) {
const runId = sessionKey ? (activeRuns.get(sessionKey) || state.runId) : undefined;
if (runId) {
enqueue(buildEnvelope('run.end', sessionKey, {
enqueue(buildEnvelope(FRAMEWORK, HOST, 'run.end', sessionKey, {
runId,
payload: {
status: 'success',
@@ -660,14 +504,4 @@ const handler = async () => {
}
};
async function readStdin(): Promise<string> {
return new Promise((resolve) => {
let data = '';
process.stdin.on('data', (chunk) => data += chunk);
process.stdin.on('end', () => resolve(data));
process.stdin.on('error', () => resolve(''));
setTimeout(() => resolve(data), 100);
});
}
handler();