dual-brain 0.2.6 → 0.2.7

package/bin/dual-brain.mjs

@@ -1040,43 +1040,58 @@ async function installGlobal() {
     return;
   }
 
-  // Load existing settings (merge, never clobber)
-  let existing = {};
-  if (existsSync(globalSettingsPath)) {
-    try { existing = JSON.parse(readFileSync(globalSettingsPath, 'utf8')); } catch {}
-  }
-
-  // Ensure hooks structure exists
-  if (!existing.hooks) existing.hooks = {};
-  if (!existing.hooks.PreToolUse) existing.hooks.PreToolUse = [];
-  if (!existing.hooks.PostToolUse) existing.hooks.PostToolUse = [];
-
-  // Define dual-brain hooks with ownership marker
-  const DB_MARKER = '# dual-brain-managed';
-  const preToolHooks = [
-    { matcher: 'Edit',        hooks: [{ type: 'command', command: `node ${join(hooksDir, 'head-guard.mjs')} ${DB_MARKER}` }] },
-    { matcher: 'Write',       hooks: [{ type: 'command', command: `node ${join(hooksDir, 'head-guard.mjs')} ${DB_MARKER}` }] },
-    { matcher: 'NotebookEdit',hooks: [{ type: 'command', command: `node ${join(hooksDir, 'head-guard.mjs')} ${DB_MARKER}` }] },
-    { matcher: 'Bash',        hooks: [{ type: 'command', command: `node ${join(hooksDir, 'head-guard.mjs')} ${DB_MARKER}` }] },
-    { matcher: 'Agent',       hooks: [{ type: 'command', command: `node ${join(hooksDir, 'enforce-tier.mjs')} ${DB_MARKER}` }] },
-  ];
-  const postToolHooks = [
-    { matcher: '', hooks: [{ type: 'command', command: `node ${join(hooksDir, 'cost-logger.mjs')} ${DB_MARKER}` }] },
-    { matcher: '', hooks: [{ type: 'command', command: `node ${join(hooksDir, 'auto-update-wrapper.mjs')} ${DB_MARKER}` }] },
-  ];
-
-  // Remove any existing dual-brain hooks (idempotent)
-  const isDBHook = (entry) => entry.hooks?.some(h => h.command?.includes(DB_MARKER));
-  existing.hooks.PreToolUse  = existing.hooks.PreToolUse.filter(e => !isDBHook(e));
-  existing.hooks.PostToolUse = existing.hooks.PostToolUse.filter(e => !isDBHook(e));
+  // Check if project-local hooks already exist (avoids double-firing)
+  const projectLocalSettings = join(pkgRoot, '.claude', 'settings.local.json');
+  const hasProjectLocalHooks = (() => {
+    if (!existsSync(projectLocalSettings)) return false;
+    try {
+      const content = readFileSync(projectLocalSettings, 'utf8');
+      return content.includes('dual-brain') || content.includes('head-guard');
+    } catch { return false; }
+  })();
+
+  if (hasProjectLocalHooks) {
+    console.log('  hooks already configured project-locally, skipping global hooks');
+    console.log('  (project .claude/settings.local.json already contains dual-brain hooks)');
+  } else {
+    // Load existing settings (merge, never clobber)
+    let existing = {};
+    if (existsSync(globalSettingsPath)) {
+      try { existing = JSON.parse(readFileSync(globalSettingsPath, 'utf8')); } catch {}
+    }
 
-  // Add dual-brain hooks
-  existing.hooks.PreToolUse.push(...preToolHooks);
-  existing.hooks.PostToolUse.push(...postToolHooks);
+    // Ensure hooks structure exists
+    if (!existing.hooks) existing.hooks = {};
+    if (!existing.hooks.PreToolUse) existing.hooks.PreToolUse = [];
+    if (!existing.hooks.PostToolUse) existing.hooks.PostToolUse = [];
 
-  // Write merged settings
-  mkdirSync(globalClaudeDir, { recursive: true });
-  writeFileSync(globalSettingsPath, JSON.stringify(existing, null, 2) + '\n');
+    // Define dual-brain hooks with ownership marker
+    const DB_MARKER = '# dual-brain-managed';
+    const preToolHooks = [
+      { matcher: 'Edit',        hooks: [{ type: 'command', command: `node ${join(hooksDir, 'head-guard.mjs')} ${DB_MARKER}` }] },
+      { matcher: 'Write',       hooks: [{ type: 'command', command: `node ${join(hooksDir, 'head-guard.mjs')} ${DB_MARKER}` }] },
+      { matcher: 'NotebookEdit',hooks: [{ type: 'command', command: `node ${join(hooksDir, 'head-guard.mjs')} ${DB_MARKER}` }] },
+      { matcher: 'Bash',        hooks: [{ type: 'command', command: `node ${join(hooksDir, 'head-guard.mjs')} ${DB_MARKER}` }] },
+      { matcher: 'Agent',       hooks: [{ type: 'command', command: `node ${join(hooksDir, 'enforce-tier.mjs')} ${DB_MARKER}` }] },
+    ];
+    const postToolHooks = [
+      { matcher: '', hooks: [{ type: 'command', command: `node ${join(hooksDir, 'cost-logger.mjs')} ${DB_MARKER}` }] },
+      { matcher: '', hooks: [{ type: 'command', command: `node ${join(hooksDir, 'auto-update-wrapper.mjs')} ${DB_MARKER}` }] },
+    ];
+
+    // Remove any existing dual-brain hooks (idempotent)
+    const isDBHook = (entry) => entry.hooks?.some(h => h.command?.includes(DB_MARKER));
+    existing.hooks.PreToolUse  = existing.hooks.PreToolUse.filter(e => !isDBHook(e));
+    existing.hooks.PostToolUse = existing.hooks.PostToolUse.filter(e => !isDBHook(e));
+
+    // Add dual-brain hooks
+    existing.hooks.PreToolUse.push(...preToolHooks);
+    existing.hooks.PostToolUse.push(...postToolHooks);
+
+    // Write merged settings
+    mkdirSync(globalClaudeDir, { recursive: true });
+    writeFileSync(globalSettingsPath, JSON.stringify(existing, null, 2) + '\n');
+  }
 
   // Write minimal global CLAUDE.md (only if none exists, or append section)
   const globalClaudeMd = join(globalClaudeDir, 'CLAUDE.md');
@@ -1091,12 +1106,16 @@ async function installGlobal() {
     }
   }
 
-  console.log('  + dual-brain hooks installed globally');
-  console.log('    hooks dir: ' + hooksDir);
-  console.log('    settings:  ' + globalSettingsPath);
-  console.log('');
-  console.log('  All new Claude sessions will load dual-brain hooks.');
-  console.log('  Run "dual-brain uninstall --global" to remove.');
+  if (!hasProjectLocalHooks) {
+    console.log('  + dual-brain hooks installed globally');
+    console.log('    hooks dir: ' + hooksDir);
+    console.log('    settings:  ' + globalSettingsPath);
+    console.log('');
+    console.log('  All new Claude sessions will load dual-brain hooks.');
+    console.log('  Run "dual-brain uninstall --global" to remove.');
+  }
+  console.log('  + global CLAUDE.md updated');
+  console.log('    path: ' + globalClaudeDir);
 }
 
 async function uninstallGlobal() {
@@ -2163,6 +2182,15 @@ async function mainScreen(rl, ask) {
     dashSpinner = fx.spinner('Loading dashboard...').start();
   }
 
+  // ── One-time default shell prompt for returning users (never asked before) ─
+  if (profile.setupComplete && !profile.defaultShellAsked) {
+    if (dashSpinner) { dashSpinner.stop(); dashSpinner = null; }
+    const wantsDefault = await askDefaultShell(cwd, rl, fx);
+    profile.defaultShellAsked = true;
+    profile.isDefaultShell = wantsDefault;
+    saveProfile(profile, { cwd });
+  }
+
   const claudeSub = profile?.providers?.claude;
   const openaiSub = profile?.providers?.openai;
 
@@ -3984,6 +4012,62 @@ function saveWizardCredentials(cwd, detectedProviders) {
  * @param {object} rl  readline interface
  * @returns {object|null}  profile object to save, or null if cancelled/skipped
  */
+function setAsDefaultShell(cwd) {
+  const root = cwd || process.cwd();
+  const replitPath = join(root, '.replit');
+  if (!existsSync(replitPath)) return;
+
+  let content = readFileSync(replitPath, 'utf8');
+  const newOnBoot = 'onBoot = "source /home/runner/workspace/.replit-tools/scripts/setup-claude-code.sh 2>/dev/null || true; ln -sf /home/runner/workspace/.replit-tools/.npm-persistent/.npmrc ~/.npmrc 2>/dev/null || true; dual-brain install --global 2>/dev/null || true"';
+
+  if (content.match(/^onBoot\s*=/m)) {
+    content = content.replace(/^onBoot\s*=.*$/m, newOnBoot);
+  } else {
+    content += '\n' + newOnBoot + '\n';
+  }
+  writeFileSync(replitPath, content);
+}
+
+function removeAsDefaultShell(cwd) {
+  const root = cwd || process.cwd();
+  const replitPath = join(root, '.replit');
+  if (!existsSync(replitPath)) return;
+
+  let content = readFileSync(replitPath, 'utf8');
+  const origOnBoot = 'onBoot = "source /home/runner/workspace/.replit-tools/scripts/setup-claude-code.sh 2>/dev/null || true"';
+  if (content.match(/^onBoot\s*=/m)) {
+    content = content.replace(/^onBoot\s*=.*$/m, origOnBoot);
+    writeFileSync(replitPath, content);
+  }
+}
+
+async function askDefaultShell(cwd, rl, fx) {
+  const cl = fx.colors || {};
+  const DIM = cl.dim || '';
+  const BOLD = cl.bold || '';
+  const GRAY = cl.gray || '';
+  const GREEN = cl.green || '';
+  const RST = cl.reset || '';
+
+  process.stdout.write('\n');
+  process.stdout.write(` ${BOLD}Shell startup${RST}\n\n`);
+  process.stdout.write(` ${DIM}dual-brain can start automatically when your shell opens.${RST}\n`);
+  process.stdout.write(` ${DIM}This modifies .replit onBoot. You can change it anytime in Settings.${RST}\n\n`);
+  process.stdout.write(` ${GRAY}[y]${RST} Yes, set as default  ${GRAY}[n]${RST} No, I'll run it manually\n\n`);
+
+  const answer = await new Promise(res => rl.question('  ', (a) => res(a.trim().toLowerCase())));
+  const yes = !answer || answer.startsWith('y');
+
+  if (yes) {
+    setAsDefaultShell(cwd);
+    process.stdout.write(` ${GREEN}+${RST} ${DIM}dual-brain will start on boot. Change anytime in Settings.${RST}\n`);
+  } else {
+    process.stdout.write(` ${DIM}No problem. Run dual-brain anytime from the command line.${RST}\n`);
+  }
+
+  return yes;
+}
+
 async function runOnboardingWizard(_detection, cwd, rl) {
   const fx  = await getFx();
   const cl  = fx.colors || {};
@@ -4278,6 +4362,23 @@ async function runOnboardingWizard(_detection, cwd, rl) {
   finalProfile.bias      = chosenBias;
   finalProfile.workStyle = chosenBias;
 
+  // Ask about default shell (only on first wizard run)
+  if (!finalProfile.defaultShellAsked) {
+    const wantsDefault = await askDefaultShell(cwd, rl, fx);
+    finalProfile.defaultShellAsked = true;
+    finalProfile.isDefaultShell = wantsDefault;
+    saveProfile(finalProfile, { cwd });
+
+    // Also run global install if they said yes
+    if (wantsDefault) {
+      try {
+        execSync('node ' + join(dirname(fileURLToPath(import.meta.url)), 'dual-brain.mjs') + ' install --global', {
+          cwd, stdio: 'pipe', timeout: 10000,
+        });
+      } catch {}
+    }
+  }
+
   return finalProfile;
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dual-brain",
-  "version": "0.2.6",
+  "version": "0.2.7",
   "description": "AI orchestration across Claude + OpenAI subscriptions — smart routing, budget awareness, and dual-brain collaboration",
   "type": "module",
   "bin": {