claude-pool 0.0.0-stage → 0.1.1

package/src/setup.js

@@ -0,0 +1,218 @@
+import { spawnSync } from 'child_process';
+import readline from 'readline/promises';
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+import { ConfigManager } from './config.js';
+import { colors, printTable, makePill } from './ui.js';
+
+export function checkClaudeCliInstalled() {
+  try {
+    const isWin = process.platform === 'win32';
+    const cmd = isWin ? 'where.exe' : 'which';
+    const res = spawnSync(cmd, ['claude'], { encoding: 'utf8' });
+    return res.status === 0;
+  } catch {
+    return false;
+  }
+}
+
+export function runAccountLogin(configDir, email = null) {
+  const env = { ...process.env };
+  const defaultClaudePath = path.resolve(path.join(os.homedir(), '.claude'));
+  if (path.resolve(configDir) !== defaultClaudePath) {
+    env.CLAUDE_CONFIG_DIR = configDir;
+  } else {
+    delete env.CLAUDE_CONFIG_DIR;
+  }
+
+  delete env.ANTHROPIC_BASE_URL;
+  delete env.ANTHROPIC_API_KEY;
+  delete env.ANTHROPIC_DEFAULT_SONNET_MODEL;
+
+  const cmdArgs = ['auth', 'login'];
+  if (email) cmdArgs.push('--email', email);
+
+  console.log(`${colors.dim}Opening Claude browser authentication...${colors.reset}`);
+  const res = spawnSync('claude', cmdArgs, { env, stdio: 'inherit' });
+  return res.status === 0;
+}
+
+export function checkAuthStatus(configDir) {
+  const env = { ...process.env };
+  const defaultClaudePath = path.resolve(path.join(os.homedir(), '.claude'));
+  if (path.resolve(configDir) !== defaultClaudePath) {
+    env.CLAUDE_CONFIG_DIR = configDir;
+  } else {
+    delete env.CLAUDE_CONFIG_DIR;
+  }
+
+  try {
+    const res = spawnSync('claude', ['auth', 'status'], { env, encoding: 'utf8', timeout: 10000 });
+    if (res.status === 0) {
+      const data = JSON.parse(res.stdout);
+      return { ok: data.loggedIn === true, email: data.email, plan: data.subscriptionType };
+    }
+  } catch {}
+  return { ok: false, email: null, plan: null };
+}
+
+export function installSlashCommands(configDir) {
+  const cmdDir = path.join(configDir, 'commands');
+  try {
+    fs.mkdirSync(cmdDir, { recursive: true });
+    // /pool
+    fs.writeFileSync(
+      path.join(cmdDir, 'pool.md'),
+      '---\ndescription: View Claude Pool token dashboard and status\n---\nRun `claude-pool dashboard` using Bash and display the formatted output.\n',
+      'utf8'
+    );
+    // /limits
+    fs.writeFileSync(
+      path.join(cmdDir, 'limits.md'),
+      '---\ndescription: View remaining session and weekly quotas across all accounts in the pool\n---\nRun `claude-pool limits` using Bash and display the formatted output.\n',
+      'utf8'
+    );
+    // /switch
+    fs.writeFileSync(
+      path.join(cmdDir, 'switch.md'),
+      '---\ndescription: Switch active Claude account in the pool\n---\nRun `claude-pool switch $ARGUMENTS` using Bash and display the result.\nIf switched successfully, inform the user that their session was synchronized and they can run `/exit` to restart immediately on the new account.\n',
+      'utf8'
+    );
+    // Also include /pool-limits and /pool-switch aliases
+    fs.copyFileSync(path.join(cmdDir, 'limits.md'), path.join(cmdDir, 'pool-limits.md'));
+    fs.copyFileSync(path.join(cmdDir, 'switch.md'), path.join(cmdDir, 'pool-switch.md'));
+  } catch {}
+}
+
+export async function runSetupWizard() {
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+
+  console.log();
+  console.log(colors.indigo + colors.bold + '╭─ claude-pool setup ' + '─'.repeat(45) + '╮' + colors.reset);
+  console.log(colors.border + '│' + colors.reset + `  ${colors.bold}Welcome to Claude Pool Setup Wizard${colors.reset}`);
+  console.log(colors.border + '│' + colors.reset + `  Connect multiple Claude subscriptions into one unified token pool.`);
+  console.log(colors.border + '│' + colors.reset + `  ${colors.terracotta}Automatic failover · Continuous session memory · Live token metrics${colors.reset}`);
+  console.log(colors.indigo + colors.bold + '╰' + '─'.repeat(66) + '╯' + colors.reset);
+  console.log();
+
+  if (!checkClaudeCliInstalled()) {
+    console.log(`${colors.red}✗ Claude Code CLI was not found in your PATH.${colors.reset}`);
+    console.log(`Please install Claude Code CLI first by running:\n`);
+    console.log(`  ${colors.indigo}npm install -g @anthropic-ai/claude-code${colors.reset}\n`);
+    console.log(`Once installed, run ${colors.indigo}claude-pool setup${colors.reset} again.`);
+    rl.close();
+    return;
+  }
+
+  console.log(`${colors.emerald}✓ Claude Code CLI detected.${colors.reset}`);
+
+  const cfg = new ConfigManager();
+  const accounts = [];
+  const home = os.homedir();
+
+  // Primary Account
+  const primaryDir = path.join(home, '.claude');
+  fs.mkdirSync(primaryDir, { recursive: true });
+  console.log(`\n${colors.indigo}${colors.bold}Account #1 (Primary):${colors.reset}`);
+
+  const primaryStatus = checkAuthStatus(primaryDir);
+  if (primaryStatus.ok) {
+    console.log(`  ${colors.emerald}✓ Already authenticated as:${colors.reset} ${colors.bold}${primaryStatus.email || 'Logged In'}${colors.reset}`);
+    accounts.push({
+      name: 'primary',
+      configDir: primaryDir,
+      priority: 1,
+      enabled: true,
+      email: primaryStatus.email,
+      subscriptionType: primaryStatus.plan,
+    });
+  } else {
+    const doLogin = await rl.question('  No active session in ~/.claude. Log in now? [Y/n]: ');
+    if (doLogin.toLowerCase() !== 'n') {
+      const email = await rl.question('  Enter email (optional, press Enter to skip): ');
+      runAccountLogin(primaryDir, email.trim() || null);
+      const st = checkAuthStatus(primaryDir);
+      accounts.push({
+        name: 'primary',
+        configDir: primaryDir,
+        priority: 1,
+        enabled: true,
+        email: st.email,
+        subscriptionType: st.plan,
+      });
+    }
+  }
+
+  // Standby Accounts
+  let count = 2;
+  while (true) {
+    console.log();
+    const addMore = await rl.question(`Would you like to add another Claude account? [y/N]: `);
+    if (addMore.trim().toLowerCase() !== 'y') break;
+
+    const defaultName = `account${count}`;
+    const nameInput = await rl.question(`  Enter account name [default: ${defaultName}]: `);
+    const acctName = nameInput.trim() || defaultName;
+
+    const acctDir = path.join(home, `.claude-${acctName}`);
+    fs.mkdirSync(acctDir, { recursive: true });
+
+    const st = checkAuthStatus(acctDir);
+    let email = st.email;
+    let plan = st.plan;
+
+    if (st.ok) {
+      console.log(`  ${colors.emerald}✓ Already authenticated as:${colors.reset} ${colors.bold}${email}${colors.reset}`);
+    } else {
+      console.log(`  Opening browser to sign in for '${acctName}'...`);
+      const emailInput = await rl.question('  Enter email (optional, press Enter to skip): ');
+      runAccountLogin(acctDir, emailInput.trim() || null);
+      const newSt = checkAuthStatus(acctDir);
+      email = newSt.email;
+      plan = newSt.plan;
+    }
+
+    accounts.push({
+      name: acctName,
+      configDir: acctDir,
+      priority: count,
+      enabled: true,
+      email,
+      subscriptionType: plan,
+    });
+    count++;
+  }
+
+  rl.close();
+
+  if (accounts.length > 0) {
+    for (const a of accounts) {
+      installSlashCommands(a.configDir);
+    }
+    cfg.saveAccounts(accounts);
+    console.log(`\n${colors.emerald}✓ Setup complete! Your Claude Pool is configured:${colors.reset}\n`);
+
+    const headers = [
+      { label: 'Rank', align: 'center' },
+      { label: 'Priority', align: 'center' },
+      { label: 'Account' },
+      { label: 'Status', align: 'center' },
+      { label: 'Email' },
+      { label: 'Plan' },
+      { label: 'Config Directory' },
+    ];
+    const rows = accounts.map((a, i) => [
+      i === 0 ? '#1' : `#${i + 1}`,
+      `P${a.priority}`,
+      a.name,
+      makePill(i === 0 ? 'active' : 'standby'),
+      a.email || '-',
+      a.subscriptionType || '-',
+      a.configDir,
+    ]);
+    printTable(headers, rows);
+
+    console.log(`\nTo start coding with automated pooling, simply run:\n  ${colors.indigo}${colors.bold}claude-pool${colors.reset}\n`);
+  }
+}

package/src/ui.js

@@ -0,0 +1,128 @@
+// Zero-dependency aesthetic UI renderer (Linear / Raycast inspired)
+
+export const colors = {
+  reset: '\x1b[0m',
+  bold: '\x1b[1m',
+  dim: '\x1b[2m',
+  white: '\x1b[37m',
+  indigo: '\x1b[38;2;129;140;248m', // #818cf8
+  terracotta: '\x1b[38;2;249;115;22m', // #f97316
+  emerald: '\x1b[38;2;16;185;129m', // #10b981
+  amber: '\x1b[38;2;245;158;11m', // #f59e0b
+  slate: '\x1b[38;2;100;116;139m', // #64748b
+  border: '\x1b[38;2;51;65;85m', // #334155
+  red: '\x1b[38;2;239;68;68m', // #ef4444
+  bgEmerald: '\x1b[48;2;16;185;129m\x1b[30m\x1b[1m',
+  bgSlate: '\x1b[48;2;100;116;139m\x1b[37m\x1b[1m',
+  bgAmber: '\x1b[48;2;245;158;11m\x1b[30m\x1b[1m',
+  bgRed: '\x1b[48;2;239;68;68m\x1b[37m\x1b[1m',
+};
+
+export function formatTokenCount(n) {
+  if (n >= 1_000_000_000) return `${(n / 1_000_000_000).toFixed(2)}B`;
+  if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`;
+  if (n >= 1_000) return `${(n / 1_000).toFixed(1)}K`;
+  return String(n);
+}
+
+export function makePill(status) {
+  if (status === 'active') return `${colors.bgEmerald} ACTIVE ${colors.reset}`;
+  if (status === 'standby') return `${colors.bgSlate} STANDBY ${colors.reset}`;
+  if (status === 'rate_limited') return `${colors.bgAmber} COOLDOWN ${colors.reset}`;
+  if (status === 'auth_expired') return `${colors.bgRed} EXPIRED ${colors.reset}`;
+  return `${colors.dim} ${status.toUpperCase()} ${colors.reset}`;
+}
+
+export function makeGauge(pctRemaining, width = 10) {
+  if (pctRemaining === null || pctRemaining === undefined) return `${colors.dim}—${colors.reset}`;
+  const pct = Math.max(0, Math.min(100, Math.round(pctRemaining)));
+  const filled = Math.round((pct / 100) * width);
+  const empty = width - filled;
+
+  const barColor = pct > 40 ? colors.emerald : pct > 15 ? colors.amber : colors.red;
+  return `${barColor}${'█'.repeat(filled)}${colors.border}${'░'.repeat(empty)} ${pct}%${colors.reset}`;
+}
+
+// Clean aligned terminal table without external dependencies
+export function printTable(headers, rows) {
+  const colWidths = headers.map((h, i) => {
+    let max = h.label.length;
+    for (const row of rows) {
+      const valStr = String(row[i] || '').replace(/\x1b\[[0-9;]*m/g, ''); // strip ansi
+      if (valStr.length > max) max = valStr.length;
+    }
+    return max + 2;
+  });
+
+  const sepTop = '┌' + colWidths.map(w => '─'.repeat(w)).join('┬') + '┐';
+  const sepMid = '├' + colWidths.map(w => '─'.repeat(w)).join('┼') + '┤';
+  const sepBot = '└' + colWidths.map(w => '─'.repeat(w)).join('┴') + '┘';
+
+  console.log(colors.border + sepTop + colors.reset);
+
+  // Header row
+  const headerStr = headers
+    .map((h, i) => {
+      const pad = colWidths[i] - h.label.length;
+      return h.align === 'right' ? ' '.repeat(pad - 1) + h.label + ' ' : ' ' + h.label + ' '.repeat(pad - 1);
+    })
+    .join(colors.border + '│' + colors.reset);
+  console.log(colors.border + '│' + colors.indigo + colors.bold + headerStr + colors.border + '│' + colors.reset);
+  console.log(colors.border + sepMid + colors.reset);
+
+  // Rows
+  for (const row of rows) {
+    const rowStr = row
+      .map((val, i) => {
+        const raw = String(val || '');
+        const plainLen = raw.replace(/\x1b\[[0-9;]*m/g, '').length;
+        const pad = Math.max(0, colWidths[i] - plainLen);
+        const align = headers[i]?.align || 'left';
+        return align === 'right' ? ' '.repeat(pad - 1) + raw + ' ' : ' ' + raw + ' '.repeat(pad - 1);
+      })
+      .join(colors.border + '│' + colors.reset);
+    console.log(colors.border + '│' + colors.reset + rowStr + colors.border + '│' + colors.reset);
+  }
+
+  console.log(colors.border + sepBot + colors.reset);
+}
+
+export function renderSessionHeader(activeAccount, standbyAccounts, poolStats) {
+  const standbyNames = standbyAccounts.length > 0 ? standbyAccounts.map(a => a.name).join(', ') : 'none';
+  const todayPoolStr = poolStats ? formatTokenCount(poolStats.todayPoolTokens) : '0';
+
+  console.log();
+  console.log(colors.border + '┌─ ' + colors.indigo + colors.bold + 'claude-pool' + colors.border + ' ─'.repeat(60) + '┐' + colors.reset);
+  
+  let remStr = '';
+  if (activeAccount.quota && activeAccount.quota.sessionRemainingPct !== undefined) {
+    const rem = activeAccount.quota.sessionRemainingPct;
+    const color = rem > 30 ? colors.emerald : colors.amber;
+    remStr = ` · ${color}${colors.bold}${rem}% remaining${colors.reset}`;
+    if (activeAccount.quota.sessionReset) {
+      remStr += ` ${colors.slate}(resets ${activeAccount.quota.sessionReset})${colors.reset}`;
+    }
+  } else if (activeAccount.email) {
+    remStr = ` ${colors.slate}· ${activeAccount.email}${colors.reset}`;
+  }
+
+  console.log(colors.border + '│' + colors.reset + `   ${colors.indigo}Active Account${colors.reset}   ${colors.bold}${activeAccount.name}${colors.reset} [Priority ${activeAccount.priority}]${remStr}`);
+  console.log(colors.border + '│' + colors.reset + `   ${colors.indigo}Pool Today${colors.reset}       ${colors.bold}${todayPoolStr} tokens${colors.reset} ${colors.slate}across ${standbyAccounts.length + 1} accounts${colors.reset}`);
+  console.log(colors.border + '│' + colors.reset + `   ${colors.indigo}Standby Queue${colors.reset}    ${standbyNames}`);
+  console.log(colors.border + '└' + '─'.repeat(73) + '┘' + colors.reset);
+  console.log();
+}
+
+export function renderFailoverBanner(oldAccount, newAccount, sessionId) {
+  console.log();
+  console.log(colors.terracotta + '╭─ Auto-Failover Transition ' + '─'.repeat(45) + '╮' + colors.reset);
+  console.log(colors.terracotta + '│' + colors.reset + `  ${colors.amber}${colors.bold}Rate limit reached on ${oldAccount.name}${colors.reset}`);
+  console.log(colors.terracotta + '│' + colors.reset);
+  if (sessionId) {
+    console.log(colors.terracotta + '│' + colors.reset + `  ${colors.emerald}✓ Preserved active session:${colors.reset} ${sessionId.slice(0, 16)}...`);
+  }
+  console.log(colors.terracotta + '│' + colors.reset + `  ${colors.emerald}✓ Synchronized project state to${colors.reset} ${newAccount.name}`);
+  console.log(colors.terracotta + '│' + colors.reset + `  ${colors.indigo}→ Continuing on ${newAccount.name} (Priority ${newAccount.priority})${colors.reset}`);
+  console.log(colors.terracotta + '╰' + '─'.repeat(72) + '╯' + colors.reset);
+  console.log();
+}