claude-brain 0.15.2 → 0.17.0

package/src/hooks/installer.ts

@@ -1,194 +1,244 @@
-/**
- * Phase 17: Hook Installer
- * Manages installation/removal of Claude Code hooks in ~/.claude/settings.json.
- */
-
-import { readFileSync, writeFileSync, existsSync, mkdirSync, renameSync } from 'node:fs'
-import { join, dirname } from 'node:path'
-import { fileURLToPath } from 'node:url'
-import { homedir } from 'node:os'
-import { getClaudeBrainHome } from '@/config/home'
-
-const CLAUDE_SETTINGS_PATH = join(homedir(), '.claude', 'settings.json')
-const HOOK_MARKER = 'claude-brain-hook'
-
-/** Get the path where the hook script should be installed */
-export function getHookScriptPath(): string {
-  return join(getClaudeBrainHome(), 'hooks', 'brain-hook.ts')
-}
-
-/** Read Claude Code settings.json, creating if needed */
-function readSettings(): Record<string, any> {
-  if (!existsSync(CLAUDE_SETTINGS_PATH)) return {}
-  try {
-    return JSON.parse(readFileSync(CLAUDE_SETTINGS_PATH, 'utf-8'))
-  } catch {
-    return {}
-  }
-}
-
-/** Write settings.json atomically (write temp → rename) */
-function writeSettings(settings: Record<string, any>): void {
-  const dir = dirname(CLAUDE_SETTINGS_PATH)
-  if (!existsSync(dir)) {
-    mkdirSync(dir, { recursive: true })
-  }
-
-  const tmpPath = CLAUDE_SETTINGS_PATH + '.tmp'
-  writeFileSync(tmpPath, JSON.stringify(settings, null, 2) + '\n', 'utf-8')
-  renameSync(tmpPath, CLAUDE_SETTINGS_PATH)
-}
-
-/** Build the hook command string */
-function buildHookCommand(event: string): string {
-  const scriptPath = getHookScriptPath()
-  return `bun "${scriptPath}" --event ${event} # ${HOOK_MARKER}`
-}
-
-/**
- * Install Claude Brain hooks into Claude Code settings.
- * Adds PostToolUse and Stop hook entries.
- */
-export function installHooks(): { installed: boolean; message: string } {
-  if (isHooksInstalled()) {
-    return { installed: true, message: 'Hooks already installed' }
-  }
-
-  const settings = readSettings()
-
-  // Ensure hooks object exists
-  if (!settings.hooks) settings.hooks = {}
-
-  // PostToolUse hook — runs on every tool completion
-  if (!settings.hooks.PostToolUse) settings.hooks.PostToolUse = []
-  settings.hooks.PostToolUse.push({
-    matcher: '',
-    hooks: [{
-      type: 'command',
-      command: buildHookCommand('PostToolUse'),
-    }],
-  })
-
-  // Stop hook — triggers session-end summary
-  if (!settings.hooks.Stop) settings.hooks.Stop = []
-  settings.hooks.Stop.push({
-    matcher: '',
-    hooks: [{
-      type: 'command',
-      command: buildHookCommand('Stop'),
-    }],
-  })
-
-  writeSettings(settings)
-
-  // Copy hook script to install location
-  copyHookScript()
-
-  return { installed: true, message: 'Hooks installed successfully' }
-}
-
-/**
- * Remove Claude Brain hooks from Claude Code settings.
- */
-export function uninstallHooks(): { uninstalled: boolean; message: string } {
-  if (!isHooksInstalled()) {
-    return { uninstalled: true, message: 'Hooks not installed' }
-  }
-
-  const settings = readSettings()
-
-  if (settings.hooks) {
-    // Remove our entries from PostToolUse
-    if (Array.isArray(settings.hooks.PostToolUse)) {
-      settings.hooks.PostToolUse = settings.hooks.PostToolUse.filter(
-        (entry: any) => !isOurHookEntry(entry)
-      )
-      if (settings.hooks.PostToolUse.length === 0) {
-        delete settings.hooks.PostToolUse
-      }
-    }
-
-    // Remove our entries from Stop
-    if (Array.isArray(settings.hooks.Stop)) {
-      settings.hooks.Stop = settings.hooks.Stop.filter(
-        (entry: any) => !isOurHookEntry(entry)
-      )
-      if (settings.hooks.Stop.length === 0) {
-        delete settings.hooks.Stop
-      }
-    }
-
-    // Clean up empty hooks object
-    if (Object.keys(settings.hooks).length === 0) {
-      delete settings.hooks
-    }
-  }
-
-  writeSettings(settings)
-
-  return { uninstalled: true, message: 'Hooks uninstalled successfully' }
-}
-
-/**
- * Check if hooks are already installed.
- */
-export function isHooksInstalled(): boolean {
-  const settings = readSettings()
-  if (!settings.hooks) return false
-
-  const hasPostToolUse = Array.isArray(settings.hooks.PostToolUse) &&
-    settings.hooks.PostToolUse.some((entry: any) => isOurHookEntry(entry))
-
-  const hasStop = Array.isArray(settings.hooks.Stop) &&
-    settings.hooks.Stop.some((entry: any) => isOurHookEntry(entry))
-
-  return hasPostToolUse || hasStop
-}
-
-/** Check if a hook entry belongs to us (by marker in command) */
-function isOurHookEntry(entry: any): boolean {
-  if (!entry || !Array.isArray(entry.hooks)) return false
-  return entry.hooks.some(
-    (h: any) => typeof h.command === 'string' && h.command.includes(HOOK_MARKER)
-  )
-}
-
-/** All files needed by brain-hook.ts at runtime */
-const HOOK_FILES = [
-  'brain-hook.ts',
-  'capture.ts',
-  'queue.ts',
-  'types.ts',
-  'passive-classifier.ts',
-]
-
-/** Copy the hook script and all its dependencies to the install location */
-function copyHookScript(): void {
-  const destDir = dirname(getHookScriptPath())
-
-  if (!existsSync(destDir)) {
-    mkdirSync(destDir, { recursive: true })
-  }
-
-  // Use Bun's import.meta.dir if available, otherwise fileURLToPath for Windows compat
-  // (new URL(...).pathname returns "/C:/..." on Windows, breaking existsSync)
-  const srcDir = (import.meta as any).dir ?? dirname(fileURLToPath(import.meta.url))
-
-  for (const file of HOOK_FILES) {
-    const srcPath = join(srcDir, file)
-    const destPath = join(destDir, file)
-
-    if (existsSync(srcPath)) {
-      const content = readFileSync(srcPath, 'utf-8')
-      writeFileSync(destPath, content, 'utf-8')
-    } else {
-      // Try .js extension for compiled version
-      const jsSrcPath = srcPath.replace('.ts', '.js')
-      const jsDestPath = destPath.replace('.ts', '.js')
-      if (existsSync(jsSrcPath)) {
-        const content = readFileSync(jsSrcPath, 'utf-8')
-        writeFileSync(jsDestPath, content, 'utf-8')
-      }
-    }
-  }
-}
+/**
+ * Phase 17: Hook Installer
+ * Manages installation/removal of Claude Code hooks in ~/.claude/settings.json.
+ */
+
+import { readFileSync, writeFileSync, existsSync, mkdirSync, renameSync } from 'node:fs'
+import { join, dirname } from 'node:path'
+import { fileURLToPath } from 'node:url'
+import { homedir } from 'node:os'
+import { getClaudeBrainHome } from '@/config/home'
+
+const CLAUDE_SETTINGS_PATH = join(homedir(), '.claude', 'settings.json')
+const HOOK_MARKER = 'claude-brain-hook'
+
+/** Get the path where the brain-hook script should be installed */
+export function getHookScriptPath(): string {
+  return join(getClaudeBrainHome(), 'hooks', 'brain-hook.ts')
+}
+
+/** Get the path where the context-hook script should be installed */
+export function getContextHookScriptPath(): string {
+  return join(getClaudeBrainHome(), 'hooks', 'context-hook.ts')
+}
+
+/** Read Claude Code settings.json, creating if needed */
+function readSettings(): Record<string, any> {
+  if (!existsSync(CLAUDE_SETTINGS_PATH)) return {}
+  try {
+    return JSON.parse(readFileSync(CLAUDE_SETTINGS_PATH, 'utf-8'))
+  } catch {
+    return {}
+  }
+}
+
+/** Write settings.json atomically (write temp → rename) */
+function writeSettings(settings: Record<string, any>): void {
+  const dir = dirname(CLAUDE_SETTINGS_PATH)
+  if (!existsSync(dir)) {
+    mkdirSync(dir, { recursive: true })
+  }
+
+  const tmpPath = CLAUDE_SETTINGS_PATH + '.tmp'
+  writeFileSync(tmpPath, JSON.stringify(settings, null, 2) + '\n', 'utf-8')
+  renameSync(tmpPath, CLAUDE_SETTINGS_PATH)
+}
+
+/** Build the hook command string, embedding the port so hooks work regardless of env */
+function buildHookCommand(event: string, script: 'brain-hook' | 'context-hook' = 'brain-hook'): string {
+  const scriptPath = script === 'context-hook' ? getContextHookScriptPath() : getHookScriptPath()
+  const port = process.env.PORT || process.env.CLAUDE_BRAIN_PORT || '3000'
+  return `CLAUDE_BRAIN_PORT=${port} bun "${scriptPath}" --event ${event} # ${HOOK_MARKER}`
+}
+
+/**
+ * Install Claude Brain hooks into Claude Code settings.
+ * Adds PostToolUse and Stop hook entries.
+ */
+export function installHooks(): { installed: boolean; message: string } {
+  if (isHooksInstalled()) {
+    return { installed: true, message: 'Hooks already installed' }
+  }
+
+  const settings = readSettings()
+
+  // Ensure hooks object exists
+  if (!settings.hooks) settings.hooks = {}
+
+  // PostToolUse hook — runs on every tool completion
+  if (!settings.hooks.PostToolUse) settings.hooks.PostToolUse = []
+  settings.hooks.PostToolUse.push({
+    matcher: '',
+    hooks: [{
+      type: 'command',
+      command: buildHookCommand('PostToolUse'),
+    }],
+  })
+
+  // Stop hook — triggers session-end summary
+  if (!settings.hooks.Stop) settings.hooks.Stop = []
+  settings.hooks.Stop.push({
+    matcher: '',
+    hooks: [{
+      type: 'command',
+      command: buildHookCommand('Stop'),
+    }],
+  })
+
+  // UserPromptSubmit hook — injects relevant memories into every prompt
+  if (!settings.hooks.UserPromptSubmit) settings.hooks.UserPromptSubmit = []
+  settings.hooks.UserPromptSubmit.push({
+    matcher: '',
+    hooks: [{
+      type: 'command',
+      command: buildHookCommand('UserPromptSubmit', 'context-hook'),
+    }],
+  })
+
+  // SessionStart hook — injects project context on session start/resume
+  if (!settings.hooks.SessionStart) settings.hooks.SessionStart = []
+  settings.hooks.SessionStart.push({
+    matcher: 'startup,resume,compact',
+    hooks: [{
+      type: 'command',
+      command: buildHookCommand('SessionStart', 'context-hook'),
+    }],
+  })
+
+  writeSettings(settings)
+
+  // Copy hook script to install location
+  copyHookScript()
+
+  return { installed: true, message: 'Hooks installed successfully' }
+}
+
+/**
+ * Remove Claude Brain hooks from Claude Code settings.
+ */
+export function uninstallHooks(): { uninstalled: boolean; message: string } {
+  if (!isHooksInstalled()) {
+    return { uninstalled: true, message: 'Hooks not installed' }
+  }
+
+  const settings = readSettings()
+
+  if (settings.hooks) {
+    // Remove our entries from PostToolUse
+    if (Array.isArray(settings.hooks.PostToolUse)) {
+      settings.hooks.PostToolUse = settings.hooks.PostToolUse.filter(
+        (entry: any) => !isOurHookEntry(entry)
+      )
+      if (settings.hooks.PostToolUse.length === 0) {
+        delete settings.hooks.PostToolUse
+      }
+    }
+
+    // Remove our entries from Stop
+    if (Array.isArray(settings.hooks.Stop)) {
+      settings.hooks.Stop = settings.hooks.Stop.filter(
+        (entry: any) => !isOurHookEntry(entry)
+      )
+      if (settings.hooks.Stop.length === 0) {
+        delete settings.hooks.Stop
+      }
+    }
+
+    // Remove our entries from UserPromptSubmit
+    if (Array.isArray(settings.hooks.UserPromptSubmit)) {
+      settings.hooks.UserPromptSubmit = settings.hooks.UserPromptSubmit.filter(
+        (entry: any) => !isOurHookEntry(entry)
+      )
+      if (settings.hooks.UserPromptSubmit.length === 0) {
+        delete settings.hooks.UserPromptSubmit
+      }
+    }
+
+    // Remove our entries from SessionStart
+    if (Array.isArray(settings.hooks.SessionStart)) {
+      settings.hooks.SessionStart = settings.hooks.SessionStart.filter(
+        (entry: any) => !isOurHookEntry(entry)
+      )
+      if (settings.hooks.SessionStart.length === 0) {
+        delete settings.hooks.SessionStart
+      }
+    }
+
+    // Clean up empty hooks object
+    if (Object.keys(settings.hooks).length === 0) {
+      delete settings.hooks
+    }
+  }
+
+  writeSettings(settings)
+
+  return { uninstalled: true, message: 'Hooks uninstalled successfully' }
+}
+
+/**
+ * Check if hooks are already installed.
+ */
+export function isHooksInstalled(): boolean {
+  const settings = readSettings()
+  if (!settings.hooks) return false
+
+  const hasPostToolUse = Array.isArray(settings.hooks.PostToolUse) &&
+    settings.hooks.PostToolUse.some((entry: any) => isOurHookEntry(entry))
+
+  const hasStop = Array.isArray(settings.hooks.Stop) &&
+    settings.hooks.Stop.some((entry: any) => isOurHookEntry(entry))
+
+  const hasUserPromptSubmit = Array.isArray(settings.hooks.UserPromptSubmit) &&
+    settings.hooks.UserPromptSubmit.some((entry: any) => isOurHookEntry(entry))
+
+  return hasPostToolUse || hasStop || hasUserPromptSubmit
+}
+
+/** Check if a hook entry belongs to us (by marker in command) */
+function isOurHookEntry(entry: any): boolean {
+  if (!entry || !Array.isArray(entry.hooks)) return false
+  return entry.hooks.some(
+    (h: any) => typeof h.command === 'string' && h.command.includes(HOOK_MARKER)
+  )
+}
+
+/** All files needed by brain-hook.ts and context-hook.ts at runtime */
+const HOOK_FILES = [
+  'brain-hook.ts',
+  'context-hook.ts',
+  'capture.ts',
+  'queue.ts',
+  'types.ts',
+  'passive-classifier.ts',
+]
+
+/** Copy the hook script and all its dependencies to the install location */
+function copyHookScript(): void {
+  const destDir = dirname(getHookScriptPath())
+
+  if (!existsSync(destDir)) {
+    mkdirSync(destDir, { recursive: true })
+  }
+
+  // Use Bun's import.meta.dir if available, otherwise fileURLToPath for Windows compat
+  // (new URL(...).pathname returns "/C:/..." on Windows, breaking existsSync)
+  const srcDir = (import.meta as any).dir ?? dirname(fileURLToPath(import.meta.url))
+
+  for (const file of HOOK_FILES) {
+    const srcPath = join(srcDir, file)
+    const destPath = join(destDir, file)
+
+    if (existsSync(srcPath)) {
+      const content = readFileSync(srcPath, 'utf-8')
+      writeFileSync(destPath, content, 'utf-8')
+    } else {
+      // Try .js extension for compiled version
+      const jsSrcPath = srcPath.replace('.ts', '.js')
+      const jsDestPath = destPath.replace('.ts', '.js')
+      if (existsSync(jsSrcPath)) {
+        const content = readFileSync(jsSrcPath, 'utf-8')
+        writeFileSync(jsDestPath, content, 'utf-8')
+      }
+    }
+  }
+}