teleportation-cli 1.1.4 → 1.2.0

package/.claude/hooks/post_tool_use.mjs

@@ -15,6 +15,56 @@
 
 import { stdin, stdout, exit, env } from 'node:process';
 import { appendFileSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join, dirname } from 'node:path';
+import { fileURLToPath } from 'node:url';
+
+// Get __dirname equivalent for ES modules
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+/**
+ * Try to import transcript-ingestion from multiple possible paths
+ * This makes the hook work in both development and installed contexts
+ *
+ * IMPORTANT: Only catches module-not-found errors. Syntax errors, runtime
+ * errors, and other exceptions are re-thrown immediately for debugging.
+ */
+async function tryImportTranscriptIngestion() {
+  const possiblePaths = [
+    // If installed globally to ~/.claude/hooks/, lib is in ~/.teleportation/lib/daemon/
+    () => join(homedir(), '.teleportation', 'lib', 'daemon', 'transcript-ingestion.js'),
+    // If hook is still in project directory
+    () => join(process.cwd(), 'lib', 'daemon', 'transcript-ingestion.js'),
+    // Try relative to hook location (development)
+    () => join(__dirname, '..', '..', 'lib', 'daemon', 'transcript-ingestion.js')
+  ];
+
+  let lastError;
+
+  for (const getPath of possiblePaths) {
+    const modulePath = getPath();
+    try {
+      const module = await import(modulePath);
+      return module;
+    } catch (e) {
+      lastError = e;
+
+      // Only skip module-not-found errors - try next path
+      // Re-throw syntax errors, runtime errors, reference errors immediately
+      if (e.code === 'ERR_MODULE_NOT_FOUND' || e.code === 'ENOENT') {
+        continue; // Module doesn't exist at this path, try next
+      }
+
+      // Syntax error, runtime error, etc. - throw immediately with context
+      console.error(`[post_tool_use] Failed to import transcript-ingestion from ${modulePath}:`, e.message);
+      throw e;
+    }
+  }
+
+  // If we get here, module wasn't found in any location
+  throw new Error(`Could not find transcript-ingestion module in any expected location. Last error: ${lastError?.message || 'unknown'}`);
+}
 
 const readStdin = () => new Promise((resolve, reject) => {
   let data = '';
@@ -207,6 +257,10 @@ function buildErrorMessage(tool_output) {
   const { session_id, tool_name, tool_input, tool_output, tool_use_id } = input || {};
   log(`Session: ${session_id}, Tool: ${tool_name}, tool_use_id: ${tool_use_id || 'none'}`);
 
+  // Detect message source
+  const isAutonomousTask = env.TELEPORTATION_TASK_MODE === 'true' || !!env.TELEPORTATION_PARENT_SESSION_ID;
+  const source = isAutonomousTask ? 'autonomous_task' : 'cli_interactive';
+
   // Validate session_id
   if (!isValidSessionId(session_id)) {
     log(`ERROR: Invalid session_id format: ${session_id}`);
@@ -339,6 +393,7 @@ function buildErrorMessage(tool_output) {
       body: JSON.stringify({
         session_id,
         type: completionEventType,
+        source,
         data: completionEventData
       })
     });
@@ -362,6 +417,7 @@ function buildErrorMessage(tool_output) {
         body: JSON.stringify({
           session_id,
           type: 'tool_error',
+          source,
           data: {
             tool_name,
             tool_input,
@@ -387,6 +443,7 @@ function buildErrorMessage(tool_output) {
       body: JSON.stringify({
         session_id,
         type: 'tool_executed',
+        source,
         data: {
           tool_name,
           tool_input,
@@ -401,6 +458,39 @@ function buildErrorMessage(tool_output) {
     log(`Failed to record legacy tool_executed: ${e.message}`);
   }
 
+  // =========================================================================
+  // Real-time transcript ingestion after each tool execution
+  // Ensures timeline stays in sync with assistant responses
+  // =========================================================================
+
+  try {
+    // Import transcript ingestion from multiple possible locations (works in dev + installed contexts)
+    const transcriptModule = await tryImportTranscriptIngestion();
+    const { ingestTranscriptToTimeline } = transcriptModule;
+
+    const claude_session_id = env.CLAUDE_SESSION_ID || session_id;
+    const cwd = env.TELEPORTATION_CWD || process.cwd();
+
+    // Use real-time mode flag to only process recent events (max 10)
+    await ingestTranscriptToTimeline({
+      claude_session_id,
+      parent_session_id: session_id,
+      task_id: null,
+      cwd,
+      config: {
+        relayApiUrl: RELAY_API_URL,
+        apiKey: RELAY_API_KEY
+      },
+      realTimeMode: true, // Fast mode - only recent events
+      maxEvents: 10 // Limit to 10 most recent events (prevents blocking hook)
+    });
+
+    log(`Real-time transcript ingestion completed`);
+  } catch (e) {
+    log(`Real-time transcript ingestion failed (non-fatal): ${e.message}`);
+    // Don't fail the hook if ingestion fails - this is a best-effort feature
+  }
+
   // PostToolUse hooks don't need to output anything
   return exit(0);
 })();