claude-mem/hook-templates/user-prompt-submit.js

#!/usr/bin/env bun

/**
 * User Prompt Submit Hook - Streaming SDK Version
 *
 * Starts a streaming SDK session that will process tool responses in real-time.
 * Saves the SDK session ID for post-tool-use and stop hooks to resume.
 */

import path from 'path';
import fs from 'fs';
import { fileURLToPath } from 'url';
import { query } from '@anthropic-ai/claude-agent-sdk';
import { renderSystemPrompt, HOOK_CONFIG } from './shared/hook-prompt-renderer.js';
import { getProjectName } from './shared/path-resolver.js';
import { initializeDatabase, createStreamingSession, updateStreamingSession } from './shared/hook-helpers.js';

const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);

const HOOKS_LOG = path.join(process.env.HOME || '', '.claude-mem', 'logs', 'hooks.log');

function debugLog(message, data = {}) {
  if (process.env.CLAUDE_MEM_DEBUG === 'true') {
    const timestamp = new Date().toISOString();
    const logLine = `[${timestamp}] HOOK DEBUG: ${message} ${JSON.stringify(data)}\n`;
    try {
      fs.appendFileSync(HOOKS_LOG, logLine);
      process.stderr.write(logLine);
    } catch (error) {
      // Silent fail on log errors
    }
  }
}

// Removed: buildStreamingSystemPrompt function
// Now using centralized config from hook-prompt-renderer.js

// =============================================================================
// GRACEFUL SHUTDOWN HANDLERS
// =============================================================================

let db;

function cleanup() {
  if (db) {
    try {
      db.close();
    } catch (err) {
      // Silent fail on cleanup
    }
  }
}

process.on('SIGTERM', () => {
  debugLog('UserPromptSubmit: Received SIGTERM, cleaning up');
  cleanup();
  process.exit(0);
});

process.on('SIGINT', () => {
  debugLog('UserPromptSubmit: Received SIGINT, cleaning up');
  cleanup();
  process.exit(0);
});

// =============================================================================
// MAIN
// =============================================================================

let input = '';
process.stdin.setEncoding('utf8');
process.stdin.on('data', (chunk) => { input += chunk; });

process.stdin.on('end', async () => {
  let payload;
  try {
    payload = input ? JSON.parse(input) : {};
  } catch (error) {
    debugLog('UserPromptSubmit: JSON parse error', { error: error.message });
    console.log(JSON.stringify({ continue: true, suppressOutput: true }));
    process.exit(0);
  }

  const { prompt, cwd, session_id, timestamp } = payload;
  const project = cwd ? getProjectName(cwd) : 'unknown';
  const date = timestamp ? timestamp.split('T')[0] : new Date().toISOString().split('T')[0];

  debugLog('UserPromptSubmit: Starting streaming session', { project, session_id });

  // Immediately signal activity start for UI indicator
  const activityFlagPath = path.join(process.env.HOME || '', '.claude-mem', 'activity.flag');
  try {
    fs.writeFileSync(activityFlagPath, JSON.stringify({ active: true, project, timestamp: Date.now() }));
  } catch (error) {
    // Silent fail - non-critical
  }

  // Generate title and subtitle non-blocking
  if (prompt && session_id && project) {
    import('child_process').then(({ spawn }) => {
      const titleProcess = spawn('claude-mem', [
        'generate-title',
        '--save',
        '--project', project,
        '--session', session_id,
        prompt
      ], {
        stdio: 'ignore',
        detached: true
      });
      titleProcess.unref();
    }).catch(error => {
      debugLog('UserPromptSubmit: Error spawning title generator', { error: error.message });
    });
  }

  try {
    // Initialize database and create session record FIRST
    db = initializeDatabase();

    // Create session record immediately - this gives us a tracking ID
    const sessionRecord = createStreamingSession(db, {
      claude_session_id: session_id,
      project,
      user_prompt: prompt,
      started_at: timestamp
    });

    debugLog('UserPromptSubmit: Created session record', {
      internalId: sessionRecord.id,
      claudeSessionId: session_id
    });

    // Build system prompt using centralized config
    const systemPrompt = renderSystemPrompt({
      project,
      sessionId: session_id,
      date,
      userPrompt: prompt || ''
    });

    // Start SDK session using centralized config
    const response = query({
      prompt: systemPrompt,
      options: {
        model: HOOK_CONFIG.sdk.model,
        allowedTools: HOOK_CONFIG.sdk.allowedTools,
        maxTokens: HOOK_CONFIG.sdk.maxTokensSystem,
        cwd  // SDK will save transcript in this directory
      }
    });

    // Wait for session ID from init message and consume entire stream
    let sdkSessionId = null;
    for await (const message of response) {
      if (message.type === 'system' && message.subtype === 'init') {
        sdkSessionId = message.session_id;
        debugLog('UserPromptSubmit: Got SDK session ID', { sdkSessionId });
      }
      // Don't break - consume entire stream so transcript gets written
    }

    if (sdkSessionId) {
      // Update session record with SDK session ID
      updateStreamingSession(db, sessionRecord.id, {
        sdk_session_id: sdkSessionId
      });

      debugLog('UserPromptSubmit: SDK session started', {
        internalId: sessionRecord.id,
        sdkSessionId
      });
    }

  } catch (error) {
    debugLog('UserPromptSubmit: Error starting SDK session', { error: error.message, stack: error.stack });
  } finally {
    // Always close database connection
    if (db) {
      try {
        db.close();
      } catch (err) {
        debugLog('UserPromptSubmit: Error closing database', { error: err.message });
      }
    }
  }

  // Return success to Claude Code
  console.log(JSON.stringify({ continue: true, suppressOutput: true }));
  process.exit(0);
});