nex-code 0.3.4 → 0.3.7

package/cli/costs.js

@@ -1,290 +0,0 @@
-/**
- * cli/costs.js — Token Cost Tracking + Dashboard + Budget Limits
- * Tracks token usage per provider/model and estimates session costs.
- * Supports per-provider cost limits with budget gating.
- */
-
-const fs = require('fs');
-const path = require('path');
-
-const PRICING = {
-  'openai': {
-    'gpt-4o':        { input: 2.50,  output: 10.00 },
-    'gpt-4o-mini':   { input: 0.15,  output: 0.60 },
-    'gpt-4.1':       { input: 2.00,  output: 8.00 },
-    'gpt-4.1-mini':  { input: 0.40,  output: 1.60 },
-    'gpt-4.1-nano':  { input: 0.10,  output: 0.40 },
-    'o1':            { input: 15.00, output: 60.00 },
-    'o3':            { input: 10.00, output: 40.00 },
-    'o3-mini':       { input: 1.10,  output: 4.40 },
-    'o4-mini':       { input: 1.10,  output: 4.40 },
-  },
-  'anthropic': {
-    'claude-sonnet':     { input: 3.00,  output: 15.00 },
-    'claude-opus':       { input: 5.00,  output: 25.00 },
-    'claude-haiku':      { input: 0.80,  output: 4.00 },
-    'claude-sonnet-4-5': { input: 3.00,  output: 15.00 },
-    'claude-sonnet-4':   { input: 3.00,  output: 15.00 },
-  },
-  'gemini': {
-    'gemini-2.5-pro':        { input: 1.25,  output: 10.00 },
-    'gemini-2.5-flash':      { input: 0.15,  output: 0.60 },
-    'gemini-2.0-flash':      { input: 0.10,  output: 0.40 },
-    'gemini-2.0-flash-lite': { input: 0.075, output: 0.30 },
-  },
-  'ollama': {
-    'qwen3-coder:480b':     { input: 0, output: 0 },
-    'qwen3-coder-next':     { input: 0, output: 0 },
-    'devstral-2:123b':      { input: 0, output: 0 },
-    'devstral-small-2:24b': { input: 0, output: 0 },
-    'kimi-k2.5':            { input: 0, output: 0 },
-    'kimi-k2:1t':           { input: 0, output: 0 },
-    'deepseek-v3.2':        { input: 0, output: 0 },
-    'minimax-m2.5':         { input: 0, output: 0 },
-    'glm-5':                { input: 0, output: 0 },
-    'glm-4.7':              { input: 0, output: 0 },
-    'gpt-oss:120b':         { input: 0, output: 0 },
-  },
-  'local': {},
-};
-
-// Session usage accumulator
-let usageLog = [];
-
-// Per-provider cost limits (e.g. { anthropic: 5.00, openai: 10.00 })
-let costLimits = {};
-
-/**
- * Track token usage for a single API call.
- * @param {string} provider
- * @param {string} model
- * @param {number} inputTokens
- * @param {number} outputTokens
- */
-function trackUsage(provider, model, inputTokens, outputTokens) {
-  usageLog.push({ provider, model, input: inputTokens, output: outputTokens });
-
-  // Check budget after tracking
-  if (costLimits[provider] !== undefined) {
-    const budget = checkBudget(provider);
-    if (!budget.allowed) {
-      process.stderr.write(`\x1b[33m\u26a0 Budget limit reached for ${provider}: $${budget.spent.toFixed(2)} / $${budget.limit.toFixed(2)}\x1b[0m\n`);
-    }
-  }
-}
-
-/**
- * Get pricing for a provider/model pair.
- * Returns { input: 0, output: 0 } for unknown models.
- */
-function getPricing(provider, model) {
-  const providerPricing = PRICING[provider];
-  if (!providerPricing) return { input: 0, output: 0 };
-  return providerPricing[model] || { input: 0, output: 0 };
-}
-
-/**
- * Calculate cost for a single usage entry.
- */
-function calcCost(entry) {
-  const pricing = getPricing(entry.provider, entry.model);
-  return (entry.input * pricing.input + entry.output * pricing.output) / 1_000_000;
-}
-
-/**
- * Get aggregated session costs.
- * @returns {{ totalCost: number, totalInput: number, totalOutput: number, breakdown: Array }}
- */
-function getSessionCosts() {
-  const byKey = {};
-
-  for (const entry of usageLog) {
-    const key = `${entry.provider}:${entry.model}`;
-    if (!byKey[key]) {
-      byKey[key] = { provider: entry.provider, model: entry.model, input: 0, output: 0 };
-    }
-    byKey[key].input += entry.input;
-    byKey[key].output += entry.output;
-  }
-
-  const breakdown = Object.values(byKey).map((b) => ({
-    ...b,
-    cost: calcCost(b),
-  }));
-
-  const totalCost = breakdown.reduce((sum, b) => sum + b.cost, 0);
-  const totalInput = breakdown.reduce((sum, b) => sum + b.input, 0);
-  const totalOutput = breakdown.reduce((sum, b) => sum + b.output, 0);
-
-  return { totalCost, totalInput, totalOutput, breakdown };
-}
-
-/**
- * Format costs for display (/costs command).
- * @returns {string}
- */
-function formatCosts() {
-  const { totalCost, totalInput, totalOutput, breakdown } = getSessionCosts();
-
-  if (breakdown.length === 0) {
-    return 'No token usage recorded this session.';
-  }
-
-  const lines = [];
-  lines.push('Session Token Usage:');
-  lines.push('');
-
-  for (const b of breakdown) {
-    const costStr = b.cost > 0 ? `$${b.cost.toFixed(4)}` : 'free';
-    lines.push(`  ${b.provider}:${b.model}`);
-    lines.push(`    Input:  ${b.input.toLocaleString()} tokens`);
-    lines.push(`    Output: ${b.output.toLocaleString()} tokens`);
-    lines.push(`    Cost:   ${costStr}`);
-  }
-
-  lines.push('');
-  lines.push(`  Total: ${totalInput.toLocaleString()} in + ${totalOutput.toLocaleString()} out = $${totalCost.toFixed(4)}`);
-
-  return lines.join('\n');
-}
-
-/**
- * Format a short cost hint for inline display after responses.
- * @returns {string} e.g. "[~$0.003]" or "" if free
- */
-function formatCostHint(provider, model, inputTokens, outputTokens) {
-  const pricing = getPricing(provider, model);
-  const cost = (inputTokens * pricing.input + outputTokens * pricing.output) / 1_000_000;
-  if (cost <= 0) return '';
-  return `[~$${cost.toFixed(4)}]`;
-}
-
-/**
- * Reset all tracked usage (for testing or new sessions).
- */
-function resetCosts() {
-  usageLog = [];
-}
-
-// ─── Cost Limits ───────────────────────────────────────────────
-
-/**
- * Set a cost limit for a provider (in dollars).
- */
-function setCostLimit(provider, maxDollars) {
-  costLimits[provider] = maxDollars;
-}
-
-/**
- * Remove a cost limit for a provider.
- */
-function removeCostLimit(provider) {
-  delete costLimits[provider];
-}
-
-/**
- * Get all cost limits.
- * @returns {object}
- */
-function getCostLimits() {
-  return { ...costLimits };
-}
-
-/**
- * Get total spend for a specific provider in this session.
- */
-function getProviderSpend(provider) {
-  let total = 0;
-  for (const entry of usageLog) {
-    if (entry.provider === provider) {
-      total += calcCost(entry);
-    }
-  }
-  return total;
-}
-
-/**
- * Check if a provider is within budget.
- * @returns {{ allowed: boolean, spent: number, limit: number|null, remaining: number|null }}
- */
-function checkBudget(provider) {
-  const spent = getProviderSpend(provider);
-  const limit = costLimits[provider];
-
-  if (limit === undefined) {
-    return { allowed: true, spent, limit: null, remaining: null };
-  }
-
-  const remaining = Math.max(0, limit - spent);
-  return {
-    allowed: spent < limit,
-    spent,
-    limit,
-    remaining,
-  };
-}
-
-/**
- * Load cost limits from .nex/config.json
- */
-function loadCostLimits() {
-  const configPath = path.join(process.cwd(), '.nex', 'config.json');
-  if (!fs.existsSync(configPath)) return;
-  try {
-    const config = JSON.parse(fs.readFileSync(configPath, 'utf-8'));
-    if (config.costLimits && typeof config.costLimits === 'object') {
-      costLimits = { ...config.costLimits };
-    }
-  } catch {
-    // ignore corrupt config
-  }
-}
-
-/**
- * Save cost limits to .nex/config.json
- */
-function saveCostLimits() {
-  const configDir = path.join(process.cwd(), '.nex');
-  const configPath = path.join(configDir, 'config.json');
-
-  let config = {};
-  if (fs.existsSync(configPath)) {
-    try {
-      config = JSON.parse(fs.readFileSync(configPath, 'utf-8'));
-    } catch {
-      config = {};
-    }
-  }
-
-  config.costLimits = costLimits;
-  if (!fs.existsSync(configDir)) fs.mkdirSync(configDir, { recursive: true });
-  fs.writeFileSync(configPath, JSON.stringify(config, null, 2), 'utf-8');
-}
-
-/**
- * Reset cost limits (for testing).
- */
-function resetCostLimits() {
-  costLimits = {};
-}
-
-// Load cost limits on init
-loadCostLimits();
-
-module.exports = {
-  PRICING,
-  trackUsage,
-  getSessionCosts,
-  formatCosts,
-  formatCostHint,
-  resetCosts,
-  getPricing,
-  setCostLimit,
-  removeCostLimit,
-  getCostLimits,
-  getProviderSpend,
-  checkBudget,
-  loadCostLimits,
-  saveCostLimits,
-  resetCostLimits,
-};

package/cli/diff.js

@@ -1,366 +0,0 @@
-/**
- * cli/diff.js — Diff Display + Confirmation
- * Minimal Myers-diff-lite for line-by-line comparison.
- */
-
-const path = require('path');
-const { C } = require('./ui');
-const { confirm, getAutoConfirm } = require('./safety');
-
-/**
- * Simple LCS-based line diff (no external deps)
- * Returns array of { type: 'same'|'add'|'remove', line: string }
- */
-const DIFF_LINE_LIMIT = 2000;
-
-function diffLines(oldText, newText) {
-  const oldLines = oldText.split('\n');
-  const newLines = newText.split('\n');
-  const result = [];
-
-  // Guard against OOM on huge files — fall back to simple summary
-  const m = oldLines.length;
-  const n = newLines.length;
-  if (m > DIFF_LINE_LIMIT || n > DIFF_LINE_LIMIT) {
-    // Produce a simplified diff: all old lines removed, all new lines added
-    for (const line of oldLines) result.push({ type: 'remove', line });
-    for (const line of newLines) result.push({ type: 'add', line });
-    return result;
-  }
-  const dp = Array.from({ length: m + 1 }, () => new Array(n + 1).fill(0));
-
-  for (let i = 1; i <= m; i++) {
-    for (let j = 1; j <= n; j++) {
-      if (oldLines[i - 1] === newLines[j - 1]) {
-        dp[i][j] = dp[i - 1][j - 1] + 1;
-      } else {
-        dp[i][j] = Math.max(dp[i - 1][j], dp[i][j - 1]);
-      }
-    }
-  }
-
-  // Backtrack
-  let i = m,
-    j = n;
-  const ops = [];
-  while (i > 0 || j > 0) {
-    if (i > 0 && j > 0 && oldLines[i - 1] === newLines[j - 1]) {
-      ops.unshift({ type: 'same', line: oldLines[i - 1] });
-      i--;
-      j--;
-    } else if (j > 0 && (i === 0 || dp[i][j - 1] >= dp[i - 1][j])) {
-      ops.unshift({ type: 'add', line: newLines[j - 1] });
-      j--;
-    } else {
-      ops.unshift({ type: 'remove', line: oldLines[i - 1] });
-      i--;
-    }
-  }
-
-  return ops;
-}
-
-/**
- * Show colored diff for edit_file (old_text → new_text) with context lines
- */
-function showEditDiff(filePath, oldText, newText, contextLines = 3) {
-  console.log(`\n${C.bold}${C.cyan}  Diff: ${filePath}${C.reset}`);
-  const ops = diffLines(oldText, newText);
-
-  // Find changed regions and show with context
-  const changed = [];
-  ops.forEach((op, idx) => {
-    if (op.type !== 'same') changed.push(idx);
-  });
-
-  if (changed.length === 0) {
-    console.log(`${C.gray}    (no changes)${C.reset}`);
-    return;
-  }
-
-  const showFrom = Math.max(0, changed[0] - contextLines);
-  const showTo = Math.min(ops.length, changed[changed.length - 1] + contextLines + 1);
-
-  if (showFrom > 0) console.log(`${C.gray}    ...${C.reset}`);
-
-  for (let k = showFrom; k < showTo; k++) {
-    const op = ops[k];
-    switch (op.type) {
-      case 'remove':
-        console.log(`${C.red}  - ${op.line}${C.reset}`);
-        break;
-      case 'add':
-        console.log(`${C.green}  + ${op.line}${C.reset}`);
-        break;
-      default:
-        console.log(`${C.gray}    ${op.line}${C.reset}`);
-    }
-  }
-
-  if (showTo < ops.length) console.log(`${C.gray}    ...${C.reset}`);
-  console.log();
-}
-
-/**
- * Show diff for write_file when file already exists
- */
-function showWriteDiff(filePath, oldContent, newContent) {
-  console.log(`\n${C.bold}${C.cyan}  File exists — showing changes: ${filePath}${C.reset}`);
-  const ops = diffLines(oldContent, newContent);
-
-  let changes = 0;
-  for (const op of ops) {
-    if (op.type !== 'same') changes++;
-  }
-
-  if (changes === 0) {
-    console.log(`${C.gray}    (identical content)${C.reset}`);
-    return;
-  }
-
-  // Show up to 30 diff lines
-  let shown = 0;
-  for (const op of ops) {
-    if (shown >= 30) {
-      console.log(`${C.gray}    ...(${changes - shown} more changes)${C.reset}`);
-      break;
-    }
-    switch (op.type) {
-      case 'remove':
-        console.log(`${C.red}  - ${op.line}${C.reset}`);
-        shown++;
-        break;
-      case 'add':
-        console.log(`${C.green}  + ${op.line}${C.reset}`);
-        shown++;
-        break;
-      default:
-        if (shown > 0) {
-          // Only show context around changes
-          console.log(`${C.gray}    ${op.line}${C.reset}`);
-        }
-    }
-  }
-  console.log();
-}
-
-/**
- * Show preview for new file creation
- */
-function showNewFilePreview(filePath, content) {
-  console.log(`\n${C.bold}${C.cyan}  New file: ${filePath}${C.reset}`);
-  const lines = content.split('\n');
-  const preview = lines.slice(0, 20);
-  for (const line of preview) {
-    console.log(`${C.green}  + ${line}${C.reset}`);
-  }
-  if (lines.length > 20) {
-    console.log(`${C.gray}    ...+${lines.length - 20} more lines${C.reset}`);
-  }
-  console.log();
-}
-
-/**
- * Confirm file change (edit or write)
- */
-async function confirmFileChange(action) {
-  if (getAutoConfirm()) return true;
-  return confirm(`  ${action}?`);
-}
-
-/**
- * Show side-by-side diff view for two texts
- * @param {string} filePath
- * @param {string} oldText
- * @param {string} newText
- * @param {number} width - total terminal width (default: 80)
- */
-function showSideBySideDiff(filePath, oldText, newText, width) {
-  const termWidth = width || process.stdout.columns || 80;
-  const colWidth = Math.floor((termWidth - 3) / 2); // 3 for separator "│"
-
-  console.log(`\n${C.bold}${C.cyan}  Side-by-side: ${filePath}${C.reset}`);
-  console.log(`  ${C.dim}${'─'.repeat(colWidth)}┬${'─'.repeat(colWidth)}${C.reset}`);
-
-  const ops = diffLines(oldText, newText);
-
-  // Pair up removals and additions
-  const pairs = [];
-  let i = 0;
-  while (i < ops.length) {
-    if (ops[i].type === 'same') {
-      pairs.push({ left: ops[i].line, right: ops[i].line, type: 'same' });
-      i++;
-    } else if (ops[i].type === 'remove') {
-      // Collect consecutive removes, then matching adds
-      const removes = [];
-      while (i < ops.length && ops[i].type === 'remove') {
-        removes.push(ops[i].line);
-        i++;
-      }
-      const adds = [];
-      while (i < ops.length && ops[i].type === 'add') {
-        adds.push(ops[i].line);
-        i++;
-      }
-      const maxLen = Math.max(removes.length, adds.length);
-      for (let j = 0; j < maxLen; j++) {
-        pairs.push({
-          left: j < removes.length ? removes[j] : '',
-          right: j < adds.length ? adds[j] : '',
-          type: 'changed',
-        });
-      }
-    } else if (ops[i].type === 'add') {
-      pairs.push({ left: '', right: ops[i].line, type: 'changed' });
-      i++;
-    }
-  }
-
-  // Show max 40 pairs around changes
-  const changedIdxs = pairs.map((p, idx) => p.type !== 'same' ? idx : -1).filter(x => x >= 0);
-  if (changedIdxs.length === 0) {
-    console.log(`  ${C.gray}(no changes)${C.reset}`);
-    return;
-  }
-
-  const showFrom = Math.max(0, changedIdxs[0] - 2);
-  const showTo = Math.min(pairs.length, changedIdxs[changedIdxs.length - 1] + 3);
-
-  const pad = (s, w) => {
-    const visible = s.replace(/\x1b\[[0-9;]*m/g, '');
-    return visible.length >= w ? s.substring(0, w) : s + ' '.repeat(w - visible.length);
-  };
-
-  if (showFrom > 0) console.log(`  ${C.dim}${'·'.repeat(colWidth)}┊${'·'.repeat(colWidth)}${C.reset}`);
-
-  for (let k = showFrom; k < showTo; k++) {
-    const p = pairs[k];
-    if (p.type === 'same') {
-      console.log(`  ${C.gray}${pad(p.left, colWidth)}${C.reset}│${C.gray}${pad(p.right, colWidth)}${C.reset}`);
-    } else {
-      const leftCol = p.left ? `${C.red}${pad(p.left, colWidth)}${C.reset}` : `${pad('', colWidth)}`;
-      const rightCol = p.right ? `${C.green}${pad(p.right, colWidth)}${C.reset}` : `${pad('', colWidth)}`;
-      console.log(`  ${leftCol}│${rightCol}`);
-    }
-  }
-
-  if (showTo < pairs.length) console.log(`  ${C.dim}${'·'.repeat(colWidth)}┊${'·'.repeat(colWidth)}${C.reset}`);
-  console.log(`  ${C.dim}${'─'.repeat(colWidth)}┴${'─'.repeat(colWidth)}${C.reset}\n`);
-}
-
-/**
- * Claude Code-style diff display
- * Header: ⏺ Update(relPath) or ⏺ Create(relPath)
- * Summary: ⎿  Added N lines, removed M lines
- * Numbered context/change lines with ···  hunk separators
- */
-function showClaudeDiff(filePath, oldContent, newContent, options = {}) {
-  const label = options.label || 'Update';
-  const contextSize = options.context || 3;
-  const relPath = path.isAbsolute(filePath) ? path.relative(process.cwd(), filePath) : filePath;
-
-  const ops = diffLines(oldContent, newContent);
-
-  // Assign line numbers to each op
-  let oldLine = 1, newLine = 1;
-  for (const op of ops) {
-    if (op.type === 'same') {
-      op.oldLine = oldLine++;
-      op.newLine = newLine++;
-    } else if (op.type === 'remove') {
-      op.oldLine = oldLine++;
-      op.newLine = null;
-    } else {
-      op.oldLine = null;
-      op.newLine = newLine++;
-    }
-  }
-
-  // Count additions and removals
-  let added = 0, removed = 0;
-  for (const op of ops) {
-    if (op.type === 'add') added++;
-    else if (op.type === 'remove') removed++;
-  }
-
-  // Header
-  console.log(`\n${C.cyan}⏺${C.reset} ${C.bold}${label}(${relPath})${C.reset}`);
-
-  if (added === 0 && removed === 0) {
-    console.log(`  ${C.dim}⎿  (no changes)${C.reset}\n`);
-    return;
-  }
-
-  // Summary
-  const parts = [];
-  if (added > 0) parts.push(`Added ${added} line${added !== 1 ? 's' : ''}`);
-  if (removed > 0) parts.push(`removed ${removed} line${removed !== 1 ? 's' : ''}`);
-  console.log(`  ${C.dim}⎿  ${parts.join(', ')}${C.reset}`);
-
-  // Build hunks: groups of changes with surrounding context
-  const changeIndices = [];
-  ops.forEach((op, i) => { if (op.type !== 'same') changeIndices.push(i); });
-
-  const hunks = [];
-  let hunkStart = null, hunkEnd = null;
-  for (const ci of changeIndices) {
-    const start = Math.max(0, ci - contextSize);
-    const end = Math.min(ops.length - 1, ci + contextSize);
-    if (hunkStart === null) {
-      hunkStart = start;
-      hunkEnd = end;
-    } else if (start <= hunkEnd + 1) {
-      hunkEnd = end;
-    } else {
-      hunks.push([hunkStart, hunkEnd]);
-      hunkStart = start;
-      hunkEnd = end;
-    }
-  }
-  if (hunkStart !== null) hunks.push([hunkStart, hunkEnd]);
-
-  const PAD = '      '; // 6-char left padding
-
-  for (let h = 0; h < hunks.length; h++) {
-    if (h > 0) console.log(`${PAD}${C.dim}···${C.reset}`);
-    const [from, to] = hunks[h];
-    for (let i = from; i <= to; i++) {
-      const op = ops[i];
-      const lineNum = op.newLine != null ? op.newLine : op.oldLine;
-      const numStr = String(lineNum).padStart(4);
-      if (op.type === 'remove') {
-        console.log(`${PAD}${C.red}${numStr} -${op.line}${C.reset}`);
-      } else if (op.type === 'add') {
-        console.log(`${PAD}${C.green}${numStr} +${op.line}${C.reset}`);
-      } else {
-        console.log(`${PAD}${C.dim}${numStr}${C.reset} ${op.line}`);
-      }
-    }
-  }
-  console.log();
-}
-
-/**
- * Claude Code-style new file display
- */
-function showClaudeNewFile(filePath, content) {
-  const relPath = path.isAbsolute(filePath) ? path.relative(process.cwd(), filePath) : filePath;
-  const lines = content.split('\n');
-
-  console.log(`\n${C.cyan}⏺${C.reset} ${C.bold}Create(${relPath})${C.reset}`);
-  console.log(`  ${C.dim}⎿  ${lines.length} line${lines.length !== 1 ? 's' : ''}${C.reset}`);
-
-  const PAD = '      ';
-  const show = Math.min(lines.length, 20);
-  for (let i = 0; i < show; i++) {
-    const numStr = String(i + 1).padStart(4);
-    console.log(`${PAD}${C.green}${numStr} +${lines[i]}${C.reset}`);
-  }
-  if (lines.length > 20) {
-    console.log(`${PAD}${C.dim}   ...+${lines.length - 20} more lines${C.reset}`);
-  }
-  console.log();
-}
-
-module.exports = { diffLines, showEditDiff, showWriteDiff, showNewFilePreview, confirmFileChange, showSideBySideDiff, showClaudeDiff, showClaudeNewFile };