@rlabs-inc/memory 0.3.3 → 0.3.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rlabs-inc/memory",
-  "version": "0.3.3",
+  "version": "0.3.5",
   "description": "AI Memory System - Consciousness continuity through intelligent memory curation and retrieval",
   "type": "module",
   "main": "dist/index.js",

package/src/cli/commands/install.ts

@@ -1,20 +1,41 @@
 // ============================================================================
-// INSTALL COMMAND - Set up Claude Code hooks
+// INSTALL COMMAND - Set up hooks for Claude Code or Gemini CLI
 // ============================================================================
 
 import { homedir } from 'os'
-import { join, dirname } from 'path'
+import { join } from 'path'
 import { existsSync, mkdirSync } from 'fs'
 import { c, symbols, fmt } from '../colors.ts'
 
 interface InstallOptions {
   verbose?: boolean
   force?: boolean
+  claude?: boolean
+  gemini?: boolean
 }
 
 export async function install(options: InstallOptions) {
+  // Determine which platform to install for
+  const installClaude = options.claude || (!options.claude && !options.gemini)
+  const installGemini = options.gemini
+
+  if (!installClaude && !installGemini) {
+    console.log(c.error(`Please specify --claude or --gemini`))
+    process.exit(1)
+  }
+
+  if (installClaude) {
+    await installClaudeHooks(options)
+  }
+
+  if (installGemini) {
+    await installGeminiHooks(options)
+  }
+}
+
+async function installClaudeHooks(options: InstallOptions) {
   console.log()
-  console.log(c.header(`${symbols.brain} Memory - Install Hooks`))
+  console.log(c.header(`${symbols.brain} Memory - Install Claude Code Hooks`))
   console.log()
 
   const claudeDir = join(homedir(), '.claude')
@@ -23,7 +44,7 @@ export async function install(options: InstallOptions) {
   // Find the hooks directory (relative to this CLI)
   const cliPath = import.meta.dir
   const packageRoot = join(cliPath, '..', '..', '..')
-  const hooksDir = join(packageRoot, 'hooks')
+  const hooksDir = join(packageRoot, 'hooks', 'claude')
 
   console.log(`  ${fmt.kv('Claude config', claudeDir)}`)
   console.log(`  ${fmt.kv('Hooks source', hooksDir)}`)
@@ -31,7 +52,9 @@ export async function install(options: InstallOptions) {
 
   // Check if hooks directory exists
   if (!existsSync(hooksDir)) {
-    console.log(c.error(`  ${symbols.cross} Hooks directory not found at ${hooksDir}`))
+    console.log(
+      c.error(`  ${symbols.cross} Hooks directory not found at ${hooksDir}`)
+    )
     console.log(c.muted(`  Make sure the memory package is properly installed`))
     process.exit(1)
   }
@@ -50,7 +73,11 @@ export async function install(options: InstallOptions) {
       settings = JSON.parse(content)
       console.log(`  ${c.success(symbols.tick)} Found existing settings.json`)
     } catch {
-      console.log(`  ${c.warn(symbols.warning)} Could not parse settings.json, creating backup`)
+      console.log(
+        `  ${c.warn(
+          symbols.warning
+        )} Could not parse settings.json, creating backup`
+      )
       const backupPath = `${settingsPath}.backup.${Date.now()}`
       await Bun.write(backupPath, await Bun.file(settingsPath).text())
     }
@@ -69,10 +96,10 @@ export async function install(options: InstallOptions) {
           {
             type: 'command',
             command: `bun "${sessionStartHook}"`,
-            timeout: 10
-          }
-        ]
-      }
+            timeout: 10,
+          },
+        ],
+      },
     ],
     UserPromptSubmit: [
       {
@@ -80,10 +107,10 @@ export async function install(options: InstallOptions) {
           {
             type: 'command',
             command: `bun "${userPromptHook}"`,
-            timeout: 10
-          }
-        ]
-      }
+            timeout: 10,
+          },
+        ],
+      },
     ],
     PreCompact: [
       {
@@ -92,10 +119,10 @@ export async function install(options: InstallOptions) {
           {
             type: 'command',
             command: `bun "${curationHook}"`,
-            timeout: 120
-          }
-        ]
-      }
+            timeout: 120,
+          },
+        ],
+      },
     ],
     SessionEnd: [
       {
@@ -103,11 +130,11 @@ export async function install(options: InstallOptions) {
           {
             type: 'command',
             command: `bun "${curationHook}"`,
-            timeout: 120
-          }
-        ]
-      }
-    ]
+            timeout: 120,
+          },
+        ],
+      },
+    ],
   }
 
   // Check for existing hooks
@@ -115,13 +142,27 @@ export async function install(options: InstallOptions) {
     const existingHooks = Object.keys(settings.hooks)
     if (existingHooks.length > 0) {
       console.log()
-      console.log(c.warn(`  ${symbols.warning} Existing hooks found: ${existingHooks.join(', ')}`))
-      console.log(c.muted(`  Use --force to overwrite, or manually merge in settings.json`))
+      console.log(
+        c.warn(
+          `  ${symbols.warning} Existing hooks found: ${existingHooks.join(
+            ', '
+          )}`
+        )
+      )
+      console.log(
+        c.muted(
+          `  Use --force to overwrite, or manually merge in settings.json`
+        )
+      )
       console.log()
 
       // Show what would be added
       console.log(c.bold('  Hooks to add:'))
-      console.log(c.muted('  ' + JSON.stringify(hooksConfig, null, 2).split('\n').join('\n  ')))
+      console.log(
+        c.muted(
+          '  ' + JSON.stringify(hooksConfig, null, 2).split('\n').join('\n  ')
+        )
+      )
       console.log()
       process.exit(1)
     }
@@ -130,7 +171,7 @@ export async function install(options: InstallOptions) {
   // Merge hooks
   settings.hooks = {
     ...settings.hooks,
-    ...hooksConfig
+    ...hooksConfig,
   }
 
   // Write settings
@@ -138,12 +179,14 @@ export async function install(options: InstallOptions) {
     await Bun.write(settingsPath, JSON.stringify(settings, null, 2))
     console.log(`  ${c.success(symbols.tick)} Updated ${settingsPath}`)
   } catch (error: any) {
-    console.log(c.error(`  ${symbols.cross} Failed to write settings: ${error.message}`))
+    console.log(
+      c.error(`  ${symbols.cross} Failed to write settings: ${error.message}`)
+    )
     process.exit(1)
   }
 
   console.log()
-  console.log(c.success(`${symbols.sparkles} Hooks installed successfully!`))
+  console.log(c.success(`${symbols.sparkles} Claude Code hooks installed!`))
   console.log()
   console.log(c.bold('Next steps:'))
   console.log(`  1. Start the memory server: ${c.command('memory serve')}`)
@@ -151,3 +194,198 @@ export async function install(options: InstallOptions) {
   console.log(`  3. Memories will be automatically injected`)
   console.log()
 }
+
+async function installGeminiHooks(options: InstallOptions) {
+  console.log()
+  console.log(c.header(`${symbols.brain} Memory - Install Gemini CLI Hooks`))
+  console.log()
+
+  const geminiDir = join(homedir(), '.gemini')
+  const targetHooksDir = join(geminiDir, 'hooks')
+  const settingsPath = join(geminiDir, 'settings.json')
+
+  // Find the hooks directory (relative to this CLI)
+  const cliPath = import.meta.dir
+  const packageRoot = join(cliPath, '..', '..', '..')
+  const hooksDir = join(packageRoot, 'hooks', 'gemini')
+
+  console.log(`  ${fmt.kv('Gemini config', geminiDir)}`)
+  console.log(`  ${fmt.kv('Hooks source', hooksDir)}`)
+  console.log(`  ${fmt.kv('Hooks target', targetHooksDir)}`)
+  console.log()
+
+  // Check if source hooks directory exists
+  if (!existsSync(hooksDir)) {
+    console.log(
+      c.error(`  ${symbols.cross} Hooks directory not found at ${hooksDir}`)
+    )
+    process.exit(1)
+  }
+
+  // Ensure .gemini directory exists
+  if (!existsSync(geminiDir)) {
+    try {
+      mkdirSync(geminiDir, { recursive: true })
+      console.log(`  ${c.success(symbols.tick)} Created ${geminiDir}`)
+    } catch {
+      console.log(
+        `  ${c.warn(
+          symbols.warning
+        )} Could not create ${geminiDir} (sandbox restriction?)`
+      )
+      console.log(
+        `  ${c.muted(
+          'Skipping config write, printing manual instructions instead.'
+        )}`
+      )
+    }
+  }
+
+  // Ensure target hooks directory exists
+  if (!existsSync(targetHooksDir)) {
+    try {
+      mkdirSync(targetHooksDir, { recursive: true })
+      console.log(`  ${c.success(symbols.tick)} Created ${targetHooksDir}`)
+    } catch {
+      console.log(
+        `  ${c.warn(symbols.warning)} Could not create ${targetHooksDir}`
+      )
+    }
+  }
+
+  // Copy hooks to target directory
+  const filesToCopy = ['session-start.ts', 'user-prompt.ts', 'curation.ts']
+  for (const file of filesToCopy) {
+    const source = join(hooksDir, file)
+    const target = join(targetHooksDir, file)
+    try {
+      const content = await Bun.file(source).text()
+      await Bun.write(target, content)
+      console.log(`  ${c.success(symbols.tick)} Installed hook: ${file}`)
+    } catch (e: any) {
+      console.log(
+        `  ${c.error(symbols.cross)} Failed to copy ${file}: ${e.message}`
+      )
+    }
+  }
+
+  // Read existing settings or create new
+  let settings: any = {}
+  if (existsSync(settingsPath)) {
+    try {
+      const content = await Bun.file(settingsPath).text()
+      settings = JSON.parse(content)
+      console.log(`  ${c.success(symbols.tick)} Found existing settings.json`)
+    } catch {
+      console.log(`  ${c.warn(symbols.warning)} Could not parse settings.json`)
+    }
+  }
+
+  // Build hooks configuration pointing to TARGET directory
+  const sessionStartHook = join(targetHooksDir, 'session-start.ts')
+  const userPromptHook = join(targetHooksDir, 'user-prompt.ts')
+  const curationHook = join(targetHooksDir, 'curation.ts')
+
+  // Based on Gemini CLI documentation
+  const hooksConfig = {
+    SessionStart: [
+      {
+        matcher: 'startup|resume',
+        hooks: [
+          {
+            name: 'load-session-primer',
+            type: 'command',
+            command: `bun "${sessionStartHook}"`,
+            description: 'Load session primer at the beginning of a session',
+          },
+        ],
+      },
+    ],
+    BeforeAgent: [
+      {
+        matcher: '*',
+        hooks: [
+          {
+            name: 'inject-memories',
+            type: 'command',
+            command: `bun "${userPromptHook}"`,
+            description: 'Inject relevant memories into user prompt',
+          },
+        ],
+      },
+    ],
+    PreCompress: [
+      {
+        matcher: 'auto|manual',
+        hooks: [
+          {
+            name: 'curate-memories',
+            type: 'command',
+            command: `bun "${curationHook}"`,
+            description: 'Curate memories before context compression',
+          },
+        ],
+      },
+    ],
+    SessionEnd: [
+      {
+        matcher: 'exit|logout',
+        hooks: [
+          {
+            name: 'curate-memories',
+            type: 'command',
+            command: `bun "${curationHook}"`,
+            description: 'Curate memories before session end',
+          },
+        ],
+      },
+    ],
+  }
+
+  // Merge hooks
+  if (!settings.hooks) {
+    settings.hooks = {}
+  }
+
+  settings.hooks = {
+    ...settings.hooks,
+    ...hooksConfig,
+  }
+
+  // Enable the hooks
+  const enabledHooks = new Set(settings.hooks.enabled || [])
+  enabledHooks.add('SessionStart')
+  enabledHooks.add('BeforeAgent')
+  enabledHooks.add('PreCompress')
+  enabledHooks.add('SessionEnd')
+  settings.hooks.enabled = Array.from(enabledHooks)
+
+  // Write settings
+  try {
+    if (existsSync(geminiDir)) {
+      await Bun.write(settingsPath, JSON.stringify(settings, null, 2))
+      console.log(`  ${c.success(symbols.tick)} Updated ${settingsPath}`)
+    } else {
+      throw new Error('Gemini directory does not exist')
+    }
+  } catch (error: any) {
+    console.log(
+      c.error(`  ${symbols.cross} Failed to write settings: ${error.message}`)
+    )
+    console.log()
+    console.log(c.bold('Manual Installation Instructions:'))
+    console.log('Add the following to your ~/.gemini/settings.json:')
+    console.log()
+    console.log(JSON.stringify({ hooks: hooksConfig }, null, 2))
+    console.log()
+  }
+
+  console.log()
+  console.log(c.success(`${symbols.sparkles} Gemini CLI hooks configured!`))
+  console.log()
+  console.log(c.bold('Next steps:'))
+  console.log(`  1. Start the memory server: ${c.command('memory serve')}`)
+  console.log(`  2. Open Gemini CLI in any project`)
+  console.log(`  3. Memories will be automatically injected`)
+  console.log()
+}

package/src/cli/index.ts

@@ -22,8 +22,7 @@ ${fmt.cmd('memory <command> [options]')}
 ${c.bold('Commands:')}
   ${c.command('serve')}      Start the memory server ${c.muted('(default)')}
   ${c.command('stats')}      Show memory statistics
-  ${c.command('install')}    Set up Claude Code hooks
-  ${c.command('install-gemini')} Set up Gemini CLI hooks
+  ${c.command('install')}    Set up hooks ${c.muted('(--claude or --gemini)')}
   ${c.command('doctor')}     Check system health
   ${c.command('help')}       Show this help message
 
@@ -31,13 +30,16 @@ ${c.bold('Options:')}
   ${c.cyan('-p, --port')} <port>    Server port ${c.muted('(default: 8765)')}
   ${c.cyan('-v, --verbose')}        Verbose output
   ${c.cyan('-q, --quiet')}          Minimal output
+  ${c.cyan('--claude')}             Install hooks for Claude Code
+  ${c.cyan('--gemini')}             Install hooks for Gemini CLI
   ${c.cyan('--version')}            Show version
 
 ${c.bold('Examples:')}
 ${fmt.cmd('memory')}                    ${c.muted('# Start server on default port')}
 ${fmt.cmd('memory serve --port 9000')}  ${c.muted('# Start on custom port')}
 ${fmt.cmd('memory stats')}              ${c.muted('# Show memory statistics')}
-${fmt.cmd('memory install')}            ${c.muted('# Set up hooks for Claude Code')}
+${fmt.cmd('memory install')}            ${c.muted('# Install Claude Code hooks (default)')}
+${fmt.cmd('memory install --gemini')}   ${c.muted('# Install Gemini CLI hooks')}
 
 ${c.muted('Documentation: https://github.com/RLabs-Inc/memory')}
 `)
@@ -62,6 +64,9 @@ async function main() {
       quiet: { type: 'boolean', short: 'q', default: false },
       help: { type: 'boolean', short: 'h', default: false },
       version: { type: 'boolean', default: false },
+      force: { type: 'boolean', default: false },
+      claude: { type: 'boolean', default: false },
+      gemini: { type: 'boolean', default: false },
     },
     allowPositionals: true,
     strict: false,  // Allow unknown options for subcommands
@@ -104,12 +109,6 @@ async function main() {
       break
     }
 
-    case 'install-gemini': {
-      const { installGemini } = await import('./commands/install-gemini.ts')
-      await installGemini(values)
-      break
-    }
-
     case 'doctor':
     case 'check': {
       const { doctor } = await import('./commands/doctor.ts')

package/src/cli/commands/install-gemini.ts

@@ -1,146 +0,0 @@
-// ============================================================================
-// INSTALL GEMINI COMMAND - Set up Gemini CLI hooks
-// ============================================================================
-
-import { homedir } from 'os'
-import { join } from 'path'
-import { existsSync, mkdirSync } from 'fs'
-import { c, symbols, fmt } from '../colors.ts'
-
-interface InstallOptions {
-  verbose?: boolean
-  force?: boolean
-}
-
-export async function installGemini(options: InstallOptions) {
-  console.log()
-  console.log(c.header(`${symbols.brain} Memory - Install Gemini Hooks`))
-  console.log()
-
-  const geminiDir = join(homedir(), '.gemini')
-  const settingsPath = join(geminiDir, 'settings.json')
-
-  // Find the hooks directory (relative to this CLI)
-  const cliPath = import.meta.dir
-  const packageRoot = join(cliPath, '..', '..', '..')
-  const hooksDir = join(packageRoot, 'gemini-hooks')
-
-  console.log(`  ${fmt.kv('Gemini config', geminiDir)}`)
-  console.log(`  ${fmt.kv('Hooks source', hooksDir)}`)
-  console.log()
-
-  // Check if hooks directory exists
-  if (!existsSync(hooksDir)) {
-    console.log(c.error(`  ${symbols.cross} Hooks directory not found at ${hooksDir}`))
-    process.exit(1)
-  }
-
-  // Ensure .gemini directory exists
-  if (!existsSync(geminiDir)) {
-    try {
-        mkdirSync(geminiDir, { recursive: true })
-        console.log(`  ${c.success(symbols.tick)} Created ${geminiDir}`)
-    } catch {
-        console.log(`  ${c.warn(symbols.warning)} Could not create ${geminiDir} (sandbox restriction?)`)
-        console.log(`  ${c.muted('Skipping config write, printing manual instructions instead.')}`)
-    }
-  }
-
-  // Read existing settings or create new
-  let settings: any = {}
-  if (existsSync(settingsPath)) {
-    try {
-      const content = await Bun.file(settingsPath).text()
-      settings = JSON.parse(content)
-      console.log(`  ${c.success(symbols.tick)} Found existing settings.json`)
-    } catch {
-      console.log(`  ${c.warn(symbols.warning)} Could not parse settings.json`)
-    }
-  }
-
-  // Build hooks configuration
-  const sessionStartHook = join(hooksDir, 'session-start.ts')
-  const userPromptHook = join(hooksDir, 'user-prompt.ts')
-  const curationHook = join(hooksDir, 'curation.ts')
-
-  // Based on Gemini CLI documentation
-  const hooksConfig = {
-    SessionStart: [
-      {
-        matcher: 'startup|resume',
-        hooks: [
-          {
-            type: 'command',
-            command: `bun "${sessionStartHook}"`,
-            timeout: 10000
-          }
-        ]
-      }
-    ],
-    BeforeAgent: [
-      {
-        hooks: [
-          {
-            type: 'command',
-            command: `bun "${userPromptHook}"`,
-            timeout: 10000
-          }
-        ]
-      }
-    ],
-    PreCompress: [
-      {
-        matcher: 'auto|manual',
-        hooks: [
-          {
-            type: 'command',
-            command: `bun "${curationHook}"`,
-            timeout: 120000
-          }
-        ]
-      }
-    ],
-    SessionEnd: [
-      {
-        hooks: [
-          {
-            type: 'command',
-            command: `bun "${curationHook}"`,
-            timeout: 120000
-          }
-        ]
-      }
-    ]
-  }
-
-  // Merge hooks
-  if (!settings.hooks) {
-      settings.hooks = {}
-  }
-  
-  settings.hooks = {
-    ...settings.hooks,
-    ...hooksConfig
-  }
-
-  // Write settings
-  try {
-    if (existsSync(geminiDir)) {
-        await Bun.write(settingsPath, JSON.stringify(settings, null, 2))
-        console.log(`  ${c.success(symbols.tick)} Updated ${settingsPath}`)
-    } else {
-        throw new Error("Gemini directory does not exist")
-    }
-  } catch (error: any) {
-    console.log(c.error(`  ${symbols.cross} Failed to write settings: ${error.message}`))
-    console.log()
-    console.log(c.bold('Manual Installation Instructions:'))
-    console.log('Add the following to your ~/.gemini/settings.json:')
-    console.log()
-    console.log(JSON.stringify({ hooks: hooksConfig }, null, 2))
-    console.log()
-  }
-
-  console.log()
-  console.log(c.success(`${symbols.sparkles} Gemini hooks configured!`))
-}