apex-dev 1.0.0 → 2.0.0

package/src/ui.js

@@ -0,0 +1,269 @@
+'use strict';
+
+const chalk = require('chalk');
+const ora = require('ora');
+const boxen = require('boxen');
+const figures = require('figures');
+const readline = require('readline');
+const { highlight } = require('cli-highlight');
+const path = require('path');
+const { execSync } = require('child_process');
+const { MAX_TOOL_ITERATIONS, PROJECT_ROOT, session, timestamp } = require('./config');
+
+// ===== Theme (Amp-inspired) =====
+const t = {
+  primary:  chalk.hex('#6366f1'),
+  accent:   chalk.hex('#818cf8'),
+  dim:      chalk.dim,
+  muted:    chalk.gray,
+  text:     chalk.white,
+  bold:     chalk.bold,
+  green:    chalk.green,
+  yellow:   chalk.yellow,
+  red:      chalk.red,
+  blue:     chalk.blue,
+  cyan:     chalk.cyan,
+  italic:   chalk.italic,
+};
+
+// ===== Layout =====
+const COLS = Math.min(process.stdout.columns || 80, 100);
+
+function hr() {
+  return t.dim('─'.repeat(COLS));
+}
+
+function stripAnsi(str) {
+  return str.replace(/\x1b\[[0-9;]*m/g, '');
+}
+
+function indent(str, n = 0) {
+  if (n === 0) return str;
+  const prefix = ' '.repeat(n);
+  return str.split('\n').map(l => prefix + l).join('\n');
+}
+
+// ===== Header =====
+function showHeader() {
+  console.clear();
+  let branch = '';
+  try { branch = execSync('git rev-parse --abbrev-ref HEAD 2>/dev/null', { encoding: 'utf-8', cwd: PROJECT_ROOT }).trim(); } catch {}
+
+  const name = t.primary.bold('⚡ Apex');
+  const branchStr = branch ? t.dim('on ') + t.text(branch) : '';
+  const cwd = t.dim(path.basename(PROJECT_ROOT));
+
+  console.log();
+  console.log(`${name}  ${cwd}  ${branchStr}`);
+  console.log(hr());
+}
+
+// ===== Welcome =====
+function showWelcome() {
+  console.log();
+  console.log(t.bold('How can I help?'));
+  console.log(t.dim(`Tools available · Max ${MAX_TOOL_ITERATIONS} iterations per turn`));
+  console.log();
+}
+
+// ===== Messages =====
+function showUserMessage(text) {
+  console.log();
+  console.log(t.blue.bold('You'));
+  console.log(text);
+  console.log();
+}
+
+function showAssistantHeader() {
+  console.log(t.primary.bold('Apex'));
+}
+
+function showAssistantMessage(text) {
+  const lines = text.split('\n');
+  const formatted = [];
+  let inCodeBlock = false;
+  let codeLines = [];
+  let codeLang = '';
+
+  for (const line of lines) {
+    if (line.startsWith('```') && !inCodeBlock) {
+      inCodeBlock = true;
+      codeLang = line.slice(3).trim() || 'code';
+      codeLines = [];
+    } else if (line.startsWith('```') && inCodeBlock) {
+      inCodeBlock = false;
+      formatted.push(renderCodeBlock(codeLines.join('\n'), codeLang));
+    } else if (inCodeBlock) {
+      codeLines.push(line);
+    } else {
+      const processed = line.replace(/`([^`]+)`/g, (_, code) => t.cyan(code));
+      formatted.push(processed);
+    }
+  }
+
+  console.log(formatted.join('\n'));
+  console.log();
+}
+
+// ===== Code Block =====
+function renderCodeBlock(code, lang) {
+  const label = t.dim(`── ${lang} ──`);
+  let highlighted;
+  try {
+    highlighted = highlight(code, { language: lang, ignoreIllegals: true });
+  } catch {
+    highlighted = code;
+  }
+
+  const cLines = highlighted.split('\n').map((line, i) => {
+    return t.dim(String(i + 1).padStart(3) + ' │ ') + line;
+  });
+
+  return `${label}\n${cLines.join('\n')}\n${t.dim('─'.repeat(Math.min(COLS, 60)))}`;
+}
+
+// ===== Tool Display =====
+function toolBadge(name) {
+  return t.dim('[') + t.accent(name) + t.dim(']');
+}
+
+function toolDetail(name, args) {
+  switch (name) {
+    case 'Bash': return args.command || '';
+    case 'Grep': return `"${args.pattern}"${args.path ? ` in ${args.path}` : ''}`;
+    case 'Glob': return args.pattern || '';
+    case 'ListDir': return args.path || '.';
+    case 'Read': {
+      let d = args.path || '';
+      if (args.start_line) d += `:${args.start_line}-${args.end_line || ''}`;
+      return d;
+    }
+    case 'Write': return args.path || '';
+    case 'Edit': return args.path || '';
+    case 'Patch': return `${args.path} (${(args.edits || []).length} edits)`;
+    case 'UndoEdit': return args.path || '';
+    case 'Task': return args.description || '';
+    case 'CodeReview': {
+      const extra = (args.files || []).length;
+      const modCount = session.filesModified.size;
+      return `${modCount} modified${extra ? ` + ${extra} extra` : ''}`;
+    }
+    default: return JSON.stringify(args).slice(0, 60);
+  }
+}
+
+async function showToolCall(name, detail, startTime) {
+  const truncDetail = detail.length > 60 ? detail.slice(0, 57) + '...' : detail;
+  const spinner = ora({
+    text: `${toolBadge(name)} ${t.dim(truncDetail)}`,
+    prefixText: '  ',
+    spinner: 'dots',
+    color: 'white',
+  }).start();
+  return spinner;
+}
+
+function finishToolCall(spinner, name, detail, startTime, success = true) {
+  const elapsed = Date.now() - startTime;
+  const icon = success ? t.green('✓') : t.red('✗');
+  const timeStr = elapsed < 1000 ? `${elapsed}ms` : `${(elapsed / 1000).toFixed(1)}s`;
+  const truncDetail = detail.length > 60 ? detail.slice(0, 57) + '...' : detail;
+
+  spinner.stopAndPersist({
+    symbol: icon,
+    text: `${toolBadge(name)} ${t.dim(truncDetail)}  ${t.dim(timeStr)}`,
+    prefixText: '  ',
+  });
+}
+
+// ===== Diff =====
+function showDiff(filename, diffText) {
+  console.log(indent(t.bold(path.basename(filename)), 4));
+  const lines = diffText.split('\n');
+  for (const line of lines) {
+    if (line.startsWith('+')) {
+      console.log(indent(t.green(line), 4));
+    } else if (line.startsWith('-')) {
+      console.log(indent(t.red(line), 4));
+    }
+  }
+  console.log();
+}
+
+
+// ===== Prompt =====
+function createPrompt() {
+  return readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+    prompt: '',
+  });
+}
+
+function showPrompt() {
+  console.log(t.dim('  Ctrl+C to cancel · /help for commands'));
+  return t.primary('❯ ');
+}
+
+// ===== Session Summary =====
+function showSessionSummary() {
+  const elapsed = ((Date.now() - session.startTime) / 1000 / 60).toFixed(1);
+  const parts = [
+    `${elapsed} min`,
+    `${session.turnCount} turns`,
+    `${session.toolCallCount} tool calls`,
+    `${session.totalTokens.toLocaleString()} tokens`,
+    `$${session.totalCost.toFixed(4)}`,
+  ];
+  if (session.filesModified.size > 0) {
+    parts.push(`${session.filesModified.size} files modified`);
+  }
+  if (session.commandsRun.length > 0) {
+    parts.push(`${session.commandsRun.length} commands`);
+  }
+
+  console.log();
+  console.log(t.dim('  Session: ') + t.text(parts.join(' · ')));
+  console.log();
+}
+
+// ===== Help =====
+function showHelpMenu() {
+  console.log();
+  console.log(t.bold('Commands'));
+  console.log(t.dim('  /help       ') + 'Show this menu');
+  console.log(t.dim('  /files      ') + 'Show project file tree');
+  console.log(t.dim('  /clear      ') + 'Clear conversation');
+  console.log(t.dim('  /cost       ') + 'Show session stats');
+  console.log(t.dim('  /undo       ') + 'Undo last edit');
+  console.log(t.dim('  /diff       ') + 'Show git diff');
+  console.log(t.dim('  /git <cmd>  ') + 'Run a git command');
+  console.log(t.dim('  /quit       ') + 'Exit');
+  console.log();
+  console.log(t.bold('Tools'));
+  console.log(t.dim('  Read, Write, Edit, Patch, Bash, Grep, Glob, ListDir, UndoEdit, Task, CodeReview'));
+  console.log();
+}
+
+module.exports = {
+  t,
+  COLS,
+  hr,
+  stripAnsi,
+  indent,
+  showHeader,
+  showWelcome,
+  showUserMessage,
+  showAssistantHeader,
+  showAssistantMessage,
+  renderCodeBlock,
+  toolBadge,
+  toolDetail,
+  showToolCall,
+  finishToolCall,
+  showDiff,
+  createPrompt,
+  showPrompt,
+  showSessionSummary,
+  showHelpMenu,
+};