save-hook.ts

/**
 * Save Hook - PostToolUse
 *
 * Pure HTTP client - sends data to worker, worker handles all database operations
 * including privacy checks. This allows the hook to run under any runtime
 * (Node.js or Bun) since it has no native module dependencies.
 */

import { stdin } from 'process';
import { STANDARD_HOOK_RESPONSE } from './hook-response.js';
import { logger } from '../utils/logger.js';
import { ensureWorkerRunning, getWorkerPort, fetchWithRetry } from '../shared/worker-utils.js';
import { HOOK_TIMEOUTS } from '../shared/hook-constants.js';

export interface PostToolUseInput {
  session_id: string;
  cwd: string;
  tool_name: string;
  tool_input: any;
  tool_response: any;
}

/**
 * Save Hook Main Logic - Fire-and-forget HTTP client
 */
async function saveHook(input?: PostToolUseInput): Promise<void> {
  // Ensure worker is running before any other logic
  await ensureWorkerRunning();

  if (!input) {
    throw new Error('saveHook requires input');
  }

  const { session_id, cwd, tool_name, tool_input, tool_response } = input;

  const port = getWorkerPort();

  const toolStr = logger.formatTool(tool_name, tool_input);

  logger.dataIn('HOOK', `PostToolUse: ${toolStr}`, {
    workerPort: port
  });

  // Validate required fields before sending to worker
  if (!cwd) {
    throw new Error(`Missing cwd in PostToolUse hook input for session ${session_id}, tool ${tool_name}`);
  }

  // Send to worker - worker handles privacy check and database operations
  // Uses fetchWithRetry to handle transient network errors like ECONNRESET
  const response = await fetchWithRetry(
    `http://127.0.0.1:${port}/api/sessions/observations`,
    {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({
        contentSessionId: session_id,
        tool_name,
        tool_input,
        tool_response,
        cwd
      })
    }
  );

  if (!response.ok) {
    throw new Error(`Observation storage failed: ${response.status}`);
  }

  logger.debug('HOOK', 'Observation sent successfully', { toolName: tool_name });

  console.log(STANDARD_HOOK_RESPONSE);
}

// Entry Point
let input = '';
stdin.on('data', (chunk) => input += chunk);
stdin.on('end', async () => {
  let parsed: PostToolUseInput | undefined;
  try {
    parsed = input ? JSON.parse(input) : undefined;
  } catch (error) {
    throw new Error(`Failed to parse hook input: ${error instanceof Error ? error.message : String(error)}`);
  }
  await saveHook(parsed);
});