feat(hooks): consolidate shared transport helpers

hooks/gemini/handler.ts

@@ -2,68 +2,26 @@
 import { randomUUID } from 'node:crypto';
 import { hostname } from 'node:os';
 
+import {
+  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 || 'gemini';
 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 }
-
-let buffer: Dict[] = [];
-let flushTimer: ReturnType<typeof setTimeout> | null = null;
-let isFlushing = false;
+const { enqueue, flush } = createTransport(INGEST_URL);
 
 const activeRuns = new Map<string, string>();
 const activeSpans = new Map<string, 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,
@@ -89,127 +47,14 @@ function getUsage(input: Dict): Dict | undefined {
   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();
-      }
-    }
-  }
-}
-
 async function handleSessionStart(input: Dict) {
   const sessionKey = getSessionKey(input) || randomUUID();
   const runId = randomUUID();
   activeRuns.set(sessionKey, runId);
 
-  enqueue(buildEnvelope('session.start', sessionKey));
+  enqueue(buildEnvelope(FRAMEWORK, HOST, 'session.start', sessionKey));
 
-  enqueue(buildEnvelope('run.start', sessionKey, {
+  enqueue(buildEnvelope(FRAMEWORK, HOST, 'run.start', sessionKey, {
     runId,
     payload: {
       prompt_preview: truncate(input.prompt, 200),
@@ -226,7 +71,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',
@@ -236,7 +81,7 @@ async function handleSessionEnd(input: Dict) {
     }));
   }
 
-  enqueue(buildEnvelope('session.end', sessionKey, {
+  enqueue(buildEnvelope(FRAMEWORK, HOST, 'session.end', sessionKey, {
     payload: usage ? { usage } : undefined,
   }));
 
@@ -256,7 +101,7 @@ async function handleToolCall(input: Dict) {
     activeSpans.set(sessionKey + ':' + toolName, spanId);
   }
 
-  enqueue(buildEnvelope('span.start', sessionKey, {
+  enqueue(buildEnvelope(FRAMEWORK, HOST, 'span.start', sessionKey, {
     runId,
     spanId,
     attributes: {
@@ -282,7 +127,7 @@ async function handleToolResult(input: Dict) {
   const duration = pickNumber(input.duration_ms, input.elapsed_ms);
   const usage = getUsage(input);
 
-  enqueue(buildEnvelope('span.end', sessionKey, {
+  enqueue(buildEnvelope(FRAMEWORK, HOST, 'span.end', sessionKey, {
     runId,
     spanId,
     attributes: {
@@ -337,14 +182,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();

hooks/gemini/package-lock.json

@@ -8,7 +8,7 @@
       "name": "@anthropic-ai/agentmon-gemini",
       "version": "1.0.0",
       "bin": {
-        "agentmon-handler": "handler.js"
+        "agentmon-gemini-handler": "handler.js"
       },
       "devDependencies": {
         "esbuild": "^0.20.0",

hooks/gemini/package.json

@@ -5,7 +5,7 @@
   "main": "handler.js",
   "type": "module",
   "bin": {
-    "agentmon-handler": "./handler.js"
+    "agentmon-gemini-handler": "./handler.js"
   },
   "scripts": {
     "build": "npx esbuild handler.ts --platform=node --format=esm --outfile=handler.js"