@stackmemoryai/stackmemory 0.5.61 → 0.5.62

package/scripts/initialize.ts

@@ -13,6 +13,9 @@ import {
 import { join } from 'path';
 import { execSync } from 'child_process';
 import chalk from 'chalk';
+
+// Parse CLI args for --with-diffmem flag
+const enableDiffMem = process.argv.includes('--with-diffmem');
 // Type-safe environment variable access
 function getEnv(key: string, defaultValue?: string): string {
   const value = process.env[key];
@@ -125,14 +128,93 @@ try {
   console.log(chalk.yellow('⚠') + ' Build failed - run npm run build manually');
 }
 
+// 7. Optional: Initialize DiffMem for long-term user memory
+if (enableDiffMem) {
+  console.log(chalk.blue('\n🧠 Setting up DiffMem integration...\n'));
+
+  const diffmemDir = join(stackDir, 'diffmem');
+  const diffmemStorageDir = join(diffmemDir, 'storage');
+  const diffmemWorktreesDir = join(diffmemDir, 'worktrees');
+
+  // Create DiffMem directories
+  mkdirSync(diffmemStorageDir, { recursive: true });
+  mkdirSync(diffmemWorktreesDir, { recursive: true });
+
+  // Initialize git repo for storage
+  if (!existsSync(join(diffmemStorageDir, '.git'))) {
+    try {
+      execSync('git init', { cwd: diffmemStorageDir, stdio: 'pipe' });
+      execSync('git commit --allow-empty -m "Initialize DiffMem storage"', {
+        cwd: diffmemStorageDir,
+        stdio: 'pipe',
+      });
+      console.log(chalk.green('✓') + ' Initialized DiffMem storage repository');
+    } catch {
+      console.log(
+        chalk.yellow('⚠') + ' Failed to initialize git repo for DiffMem'
+      );
+    }
+  }
+
+  // Create DiffMem config
+  const diffmemConfigPath = join(diffmemDir, 'config.json');
+  const diffmemConfig = {
+    enabled: true,
+    endpoint: 'http://localhost:8000',
+    userId: process.env['USER'] || 'default',
+    storagePath: diffmemStorageDir,
+    worktreePath: diffmemWorktreesDir,
+  };
+  writeFileSync(diffmemConfigPath, JSON.stringify(diffmemConfig, null, 2));
+  console.log(chalk.green('✓') + ' Created DiffMem configuration');
+
+  // Add DiffMem env vars to .env file
+  const envPath = join(projectRoot, '.env');
+  const diffmemEnvVars = `
+# DiffMem Configuration (Long-term User Memory)
+DIFFMEM_ENABLED=true
+DIFFMEM_ENDPOINT=http://localhost:8000
+DIFFMEM_USER_ID=${process.env['USER'] || 'default'}
+DIFFMEM_STORAGE_PATH=${diffmemStorageDir}
+DIFFMEM_WORKTREE_PATH=${diffmemWorktreesDir}
+`;
+
+  if (existsSync(envPath)) {
+    const envContent = readFileSync(envPath, 'utf-8');
+    if (!envContent.includes('DIFFMEM_')) {
+      appendFileSync(envPath, diffmemEnvVars);
+      console.log(chalk.green('✓') + ' Added DiffMem configuration to .env');
+    }
+  } else {
+    writeFileSync(envPath, diffmemEnvVars.trim() + '\n');
+    console.log(chalk.green('✓') + ' Created .env with DiffMem configuration');
+  }
+
+  console.log(chalk.green('✓') + ' DiffMem integration configured');
+  console.log(
+    chalk.gray(
+      '  Note: DiffMem server must be running for user memory features'
+    )
+  );
+}
+
 console.log(chalk.green.bold('\n✅ StackMemory initialized successfully!\n'));
 console.log(chalk.gray('Next steps:'));
 console.log(chalk.gray('1. Add the MCP configuration above to Claude Code'));
 console.log(chalk.gray('2. Restart Claude Code'));
 console.log(chalk.gray('3. Start using context tracking!'));
+if (enableDiffMem) {
+  console.log(chalk.gray('4. Start DiffMem server: npm run diffmem:start'));
+}
 console.log(chalk.gray('\nUseful commands:'));
 console.log(
   chalk.cyan('  npm run mcp:dev') + ' - Start MCP server in dev mode'
 );
 console.log(chalk.cyan('  npm run status') + ' - Check StackMemory status');
-console.log(chalk.cyan('  npm run analyze') + ' - Analyze context usage\n');
+console.log(chalk.cyan('  npm run analyze') + ' - Analyze context usage');
+if (enableDiffMem) {
+  console.log(
+    chalk.cyan('  npm run diffmem:start') + ' - Start DiffMem server'
+  );
+}
+console.log();