mixdog 0.7.18 → 0.8.0

package/src/standalone/usage-dashboard.mjs

@@ -0,0 +1,510 @@
+import { fetchOAuthUsageSnapshot, readCachedOAuthUsageSnapshot } from '../runtime/agent/orchestrator/providers/oauth-usage.mjs';
+import {
+  fetchOpenCodeGoUsageSnapshot,
+  openCodeGoUsageConfigStatus,
+  readCachedOpenCodeGoUsageSnapshot,
+} from '../runtime/agent/orchestrator/providers/opencode-go-usage.mjs';
+import {
+  fetchApiUsageSnapshot,
+  readCachedApiUsageSnapshot,
+} from '../runtime/agent/orchestrator/providers/api-usage.mjs';
+
+function num(value, fallback = null) {
+  if (value === null || value === undefined || value === '') return fallback;
+  const n = Number(value);
+  return Number.isFinite(n) ? n : fallback;
+}
+
+function round(value, digits = 4) {
+  const n = Number(value);
+  if (!Number.isFinite(n)) return null;
+  const scale = 10 ** digits;
+  return Math.round(n * scale) / scale;
+}
+
+function clean(value) {
+  const s = typeof value === 'string' ? value.trim() : '';
+  return s || '';
+}
+
+function money(value) {
+  const n = Number(value);
+  if (!Number.isFinite(n)) return 'n/a';
+  if (n === 0) return '$0';
+  if (n >= 10) return `$${n.toFixed(0)}`;
+  if (n >= 1) return `$${n.toFixed(2)}`;
+  if (n >= 0.01) return `$${n.toFixed(3)}`;
+  return `$${n.toFixed(4)}`;
+}
+
+function compactNumber(value) {
+  const n = Number(value);
+  if (!Number.isFinite(n)) return '';
+  if (Math.abs(n) >= 1_000_000) return `${(n / 1_000_000).toFixed(n >= 10_000_000 ? 0 : 1)}M`;
+  if (Math.abs(n) >= 1_000) return `${(n / 1_000).toFixed(n >= 10_000 ? 0 : 1)}K`;
+  if (Math.abs(n) >= 10) return n.toFixed(0);
+  if (Math.abs(n) >= 1) return n.toFixed(1);
+  return n.toFixed(2);
+}
+
+function normaliseWindows(value, source = 'config') {
+  const entries = Array.isArray(value)
+    ? value.map((entry, index) => [entry?.label || entry?.name || `W${index + 1}`, entry])
+    : value && typeof value === 'object'
+      ? Object.entries(value)
+      : [];
+  return entries.map(([key, entry]) => {
+    if (!entry || typeof entry !== 'object') return null;
+    const limitUsd = num(entry.limitUsd ?? entry.limit_usd ?? entry.budgetUsd ?? entry.budget_usd ?? entry.limit_usd_cents / 100, null);
+    const usedUsd = num(entry.usedUsd ?? entry.used_usd ?? entry.spendUsd ?? entry.spend_usd ?? entry.costUsd ?? entry.cost_usd ?? entry.used_usd_cents / 100, null);
+    const remainingUsd = num(entry.remainingUsd ?? entry.remaining_usd ?? entry.leftUsd ?? entry.left_usd ?? entry.balanceUsd ?? entry.balance_usd ?? entry.remaining_usd_cents / 100, null);
+    const usedPct = num(entry.usedPct ?? entry.used_pct ?? entry.percent ?? entry.pct, null);
+    const remainingCredits = num(entry.remainingCredits ?? entry.remaining_credits ?? entry.creditsRemaining ?? entry.credits_remaining, null);
+    const limitCredits = num(entry.limitCredits ?? entry.limit_credits, null);
+    const usedCredits = num(entry.usedCredits ?? entry.used_credits, null);
+    const resetAt = num(entry.resetAt ?? entry.reset_at ?? entry.resetsAt ?? entry.resets_at, null);
+    const out = {
+      label: clean(entry.label) || String(key || 'USE').toUpperCase(),
+      source: clean(entry.source) || source,
+    };
+    if (remainingUsd !== null) out.remainingUsd = round(remainingUsd, 4);
+    if (limitUsd !== null) out.limitUsd = round(limitUsd, 4);
+    if (usedUsd !== null) out.usedUsd = round(usedUsd, 4);
+    if (usedPct !== null) out.usedPct = round(usedPct, 2);
+    if (remainingCredits !== null) out.remainingCredits = round(remainingCredits, 4);
+    if (limitCredits !== null) out.limitCredits = round(limitCredits, 4);
+    if (usedCredits !== null) out.usedCredits = round(usedCredits, 4);
+    if (resetAt) out.resetAt = resetAt;
+    if (out.usedPct === undefined && out.limitUsd > 0 && out.usedUsd !== undefined) {
+      out.usedPct = round(Math.min(100, out.usedUsd * 100 / out.limitUsd), 2);
+    }
+    return out;
+  }).filter(Boolean);
+}
+
+function localBudget(providerCfg = {}) {
+  const limitUsd = num(
+    providerCfg.limitUsd ?? providerCfg.limit_usd ?? providerCfg.budgetUsd ?? providerCfg.budget_usd
+      ?? providerCfg.monthlyBudgetUsd ?? providerCfg.monthly_budget_usd,
+    null,
+  );
+  const usedUsd = num(providerCfg.usedUsd ?? providerCfg.used_usd ?? providerCfg.spendUsd ?? providerCfg.spend_usd ?? providerCfg.costUsd ?? providerCfg.cost_usd, null);
+  const remainingUsd = num(providerCfg.remainingUsd ?? providerCfg.remaining_usd ?? providerCfg.leftUsd ?? providerCfg.left_usd, null);
+  if (remainingUsd !== null) {
+    return { remainingUsd: round(remainingUsd, 4), usedUsd: usedUsd === null ? null : round(usedUsd, 4), limitUsd: limitUsd === null ? null : round(limitUsd, 4), source: 'local-budget' };
+  }
+  if (limitUsd !== null && usedUsd !== null) {
+    return { remainingUsd: round(Math.max(0, limitUsd - usedUsd), 4), usedUsd: round(usedUsd, 4), limitUsd: round(limitUsd, 4), source: 'local-budget' };
+  }
+  const balance = providerCfg.balance && typeof providerCfg.balance === 'object' ? providerCfg.balance : null;
+  const balanceRemaining = num(balance?.remainingUsd ?? balance?.remaining_usd ?? balance?.balanceUsd ?? balance?.balance_usd, null);
+  if (balanceRemaining !== null) return { remainingUsd: round(balanceRemaining, 4), source: clean(balance?.source) || 'configured-balance' };
+  return null;
+}
+
+function snapshotRemaining(snapshot) {
+  const balanceRemaining = num(snapshot?.balance?.remainingUsd ?? snapshot?.balance?.remaining_usd, null);
+  if (balanceRemaining !== null) return { remainingUsd: round(balanceRemaining, 4), source: clean(snapshot?.balance?.source) || clean(snapshot?.source) || 'provider-api' };
+  const windows = normaliseWindows(snapshot?.quotaWindows, clean(snapshot?.source) || 'provider-api');
+  const usd = windows.filter(w => num(w.remainingUsd, null) !== null);
+  if (usd.length === 1) return { remainingUsd: usd[0].remainingUsd, source: usd[0].source || clean(snapshot?.source) || 'provider-api' };
+  return null;
+}
+
+function snapshotUsage(snapshot) {
+  const usedUsd = num(snapshot?.balance?.usedUsd ?? snapshot?.balance?.used_usd, null);
+  if (usedUsd === null) return null;
+  const limitUsd = num(snapshot?.balance?.limitUsd ?? snapshot?.balance?.limit_usd, null);
+  return {
+    usedUsd: round(usedUsd, 4),
+    limitUsd: limitUsd === null ? null : round(limitUsd, 4),
+    source: clean(snapshot?.balance?.source) || clean(snapshot?.source) || 'provider-api',
+  };
+}
+
+function displayWindow(w) {
+  const label = String(w?.label || 'USE').toUpperCase();
+  if (num(w?.remainingUsd, null) !== null) return `${label} ${money(w.remainingUsd)}`;
+  if (num(w?.usedUsd, null) !== null && num(w?.limitUsd, null) !== null) return `${label} ${money(w.usedUsd)}/${money(w.limitUsd)}`;
+  if (num(w?.remainingCredits, null) !== null && num(w?.limitCredits, null) !== null) return `${label} ${compactNumber(w.remainingCredits)}/${compactNumber(w.limitCredits)}`;
+  if (num(w?.remainingCredits, null) !== null) return `${label} ${compactNumber(w.remainingCredits)}`;
+  if (num(w?.usedCredits, null) !== null && num(w?.limitCredits, null) !== null) return `${label} ${compactNumber(w.usedCredits)}/${compactNumber(w.limitCredits)}`;
+  if (num(w?.usedPct, null) !== null) return `${label} ${Math.round(w.usedPct)}%`;
+  return label;
+}
+
+function providerDescription(id, group) {
+  switch (String(id || '').toLowerCase()) {
+    case 'openai-oauth': return 'ChatGPT/Codex subscription quota';
+    case 'anthropic-oauth': return 'Claude Code subscription quota';
+    case 'grok-oauth': return 'Grok Build subscription quota';
+    case 'opencode-go': return 'OpenCode Go subscription quota';
+    case 'openai': return 'OpenAI API billing';
+    case 'anthropic': return 'Anthropic API billing';
+    case 'deepseek': return 'DeepSeek API billing';
+    case 'gemini': return 'Gemini API billing';
+    case 'xai': return 'xAI API billing';
+    case 'ollama': return 'Local Ollama server';
+    case 'lmstudio': return 'Local LM Studio server';
+    default: return group === 'local' ? 'Local provider' : group === 'oauth' ? 'Subscription quota' : 'API billing';
+  }
+}
+
+function rowTone(row) {
+  if (row.status === 'error') return 'error';
+  if (row.status === 'missing') return 'missing';
+  if (row.status === 'local') return 'local';
+  const remaining = num(row.remainingUsd, null);
+  if (remaining !== null) return remaining <= 1 ? 'danger' : remaining <= 5 ? 'warn' : 'ok';
+  const pct = Math.max(...(row.windows || []).map(w => num(w.usedPct, -1)));
+  if (pct >= 95) return 'danger';
+  if (pct >= 80) return 'warn';
+  if (row.status === 'hidden' || row.status === 'partial' || row.status === 'estimated') return 'warn';
+  return 'ok';
+}
+
+async function oauthSnapshot(providerId, { refresh, getProvider, log }) {
+  const cached = readCachedOAuthUsageSnapshot({ provider: providerId, model: '' });
+  if (!refresh && cached) return cached;
+  if (typeof getProvider === 'function') {
+    try {
+      const providerObj = getProvider(providerId);
+      if (providerObj) return await fetchOAuthUsageSnapshot({ provider: providerId, model: '' }, providerObj, log);
+    } catch {
+      return null;
+    }
+  }
+  return cached;
+}
+
+async function apiSnapshot(providerId, { refresh } = {}) {
+  const cached = readCachedApiUsageSnapshot(providerId);
+  if (!refresh && cached) return cached;
+  try {
+    return await fetchApiUsageSnapshot(providerId, { force: refresh === true });
+  } catch {
+    return cached;
+  }
+}
+
+function baseRow(item, group, providerCfg = {}) {
+  return {
+    id: item.id,
+    label: item.name || item.id,
+    group,
+    type: item.type || group,
+    enabled: item.enabled === true,
+    authenticated: item.authenticated === true,
+    remainingUsd: null,
+    usedUsd: null,
+    limitUsd: null,
+    windows: [],
+    status: 'hidden',
+    source: 'unavailable',
+    sourceLabel: 'unavailable',
+    primary: 'hidden',
+    detail: 'provider does not expose account balance',
+    description: providerDescription(item.id, group),
+    updatedAt: null,
+    includeInTotal: false,
+    totalBucket: null,
+    config: providerCfg,
+  };
+}
+
+function providerRank(row) {
+  const id = String(row?.id || '').toLowerCase();
+  const ranks = {
+    'openai-oauth': 10,
+    'anthropic-oauth': 20,
+    'grok-oauth': 30,
+    'opencode-go': 40,
+    openai: 50,
+    anthropic: 60,
+    deepseek: 70,
+    gemini: 80,
+    xai: 90,
+    lmstudio: 100,
+    ollama: 110,
+  };
+  return ranks[id] ?? 900;
+}
+
+function applyKnownRemaining(row, known, { estimated = false } = {}) {
+  if (!known || num(known.remainingUsd, null) === null) return row;
+  row.remainingUsd = known.remainingUsd;
+  if (known.usedUsd !== null && known.usedUsd !== undefined) row.usedUsd = known.usedUsd;
+  if (known.limitUsd !== null && known.limitUsd !== undefined) row.limitUsd = known.limitUsd;
+  row.status = estimated ? 'estimated' : 'ok';
+  row.source = known.source || (estimated ? 'local-budget' : 'provider-api');
+  row.sourceLabel = estimated ? 'local budget' : 'API';
+  row.primary = `${money(row.remainingUsd)}`;
+  row.detail = row.limitUsd !== null && row.usedUsd !== null
+    ? `${money(row.usedUsd)} / ${money(row.limitUsd)} used`
+    : row.source;
+  row.includeInTotal = true;
+  row.totalBucket = estimated ? 'local' : 'api';
+  return row;
+}
+
+function applyKnownUsage(row, known) {
+  if (!known || num(known.usedUsd, null) === null) return row;
+  row.usedUsd = known.usedUsd;
+  if (known.limitUsd !== null && known.limitUsd !== undefined) row.limitUsd = known.limitUsd;
+  row.status = 'partial';
+  row.source = known.source || 'provider-api';
+  row.sourceLabel = 'usage';
+  row.primary = row.limitUsd !== null && row.limitUsd !== undefined
+    ? `Used ${money(row.usedUsd)}/${money(row.limitUsd)}`
+    : `Used ${money(row.usedUsd)}`;
+  row.detail = 'Spend reported; remaining credit unavailable';
+  row.includeInTotal = false;
+  row.totalBucket = null;
+  return row;
+}
+
+function applyWindowQuota(row, windows, { source = 'quota', detail = 'quota windows' } = {}) {
+  const normalized = normaliseWindows(windows, source);
+  if (!normalized.length) return false;
+  const localEstimate = normalized.every(w => {
+    const s = String(w?.source || '').toLowerCase();
+    return !s || s.includes('local') || s.includes('config');
+  });
+  row.windows = normalized;
+  row.status = localEstimate ? 'estimated' : 'partial';
+  row.source = source;
+  row.sourceLabel = localEstimate ? 'local estimate' : source;
+  row.primary = normalized.map(displayWindow).slice(0, 2).join(' · ');
+  row.detail = localEstimate ? 'Local estimate, not provider-reported' : detail;
+  return true;
+}
+
+function sortedRows(rows) {
+  return rows.slice().sort((a, b) => (
+    providerRank(a) - providerRank(b)
+    || String(a.label).localeCompare(String(b.label))
+  ));
+}
+
+function usageTotal(rows) {
+  const total = rows.reduce((acc, row) => {
+    acc.providerCount += 1;
+    acc[`${row.status}Count`] = (acc[`${row.status}Count`] || 0) + 1;
+    if (row.includeInTotal && num(row.remainingUsd, null) !== null) {
+      acc.knownRemainingUsd += row.remainingUsd;
+      if (row.totalBucket === 'api') acc.apiVerifiedRemainingUsd += row.remainingUsd;
+      else acc.localEstimatedRemainingUsd += row.remainingUsd;
+    }
+    if (row.status === 'missing') acc.notConfiguredCount += 1;
+    return acc;
+  }, {
+    providerCount: 0,
+    knownRemainingUsd: 0,
+    apiVerifiedRemainingUsd: 0,
+    localEstimatedRemainingUsd: 0,
+    hiddenCount: 0,
+    notConfiguredCount: 0,
+    errorCount: 0,
+  });
+
+  total.knownRemainingUsd = round(total.knownRemainingUsd, 4) || 0;
+  total.apiVerifiedRemainingUsd = round(total.apiVerifiedRemainingUsd, 4) || 0;
+  total.localEstimatedRemainingUsd = round(total.localEstimatedRemainingUsd, 4) || 0;
+  return total;
+}
+
+function usageDashboardSnapshot(rows, { checkedAt, refresh = false, checking = false } = {}) {
+  return {
+    title: 'Provider Quotas',
+    subtitle: 'Statusline-style provider quota windows.',
+    checkedAt,
+    refresh: refresh === true,
+    checking: checking === true,
+    total: usageTotal(rows),
+    rows: sortedRows(rows),
+    format: { money },
+  };
+}
+
+function emitUsageDashboard(options, dashboard) {
+  if (typeof options?.onUpdate !== 'function') return;
+  try {
+    options.onUpdate(dashboard);
+  } catch {
+    // UI progress callbacks should never break the usage refresh itself.
+  }
+}
+
+export async function createUsageDashboard(config = {}, options = {}) {
+  const setup = options.setup || { api: [], oauth: [], local: [] };
+  const providers = config.providers || {};
+  const rows = [];
+  const checkedAt = Date.now();
+  const refresh = options.refresh === true;
+  const emit = (checking = true) => emitUsageDashboard(
+    options,
+    usageDashboardSnapshot(rows, { checkedAt, refresh, checking }),
+  );
+
+  const apiTasks = (setup.api || []).map(async (item) => {
+    const providerCfg = providers[item.id] || {};
+    const row = baseRow(item, 'api', providerCfg);
+    row.status = row.authenticated ? 'checking' : 'missing';
+    row.source = row.authenticated ? 'checking' : 'not-configured';
+    row.sourceLabel = row.authenticated ? 'checking' : 'no key';
+    row.primary = row.authenticated ? '' : 'not configured';
+    row.detail = row.authenticated ? 'Checking provider usage' : 'Configure auth';
+    row.tone = rowTone(row);
+    rows.push(row);
+    emit(true);
+
+    if (!row.authenticated) {
+      row.status = 'missing';
+      row.source = 'not-configured';
+      row.sourceLabel = 'no key';
+      row.primary = 'not configured';
+      row.detail = 'Configure auth';
+    } else {
+      let hasQuota = false;
+      if (item.id === 'opencode-go') {
+        const usageStatus = openCodeGoUsageConfigStatus(config);
+        try {
+          const snapshot = refresh
+            ? await fetchOpenCodeGoUsageSnapshot(config, { force: true })
+            : readCachedOpenCodeGoUsageSnapshot() || (usageStatus.ready ? await fetchOpenCodeGoUsageSnapshot(config) : null);
+          if (snapshot) {
+            hasQuota = applyWindowQuota(row, snapshot?.quotaWindows, {
+              source: 'opencode-go-console',
+              detail: 'subscription quota',
+            });
+            row.updatedAt = num(snapshot?.cachedAt, null);
+          } else {
+            row.status = 'missing';
+            row.source = 'usage-auth-missing';
+            row.sourceLabel = 'usage auth';
+            row.primary = '';
+            row.detail = usageStatus.authCookieSet
+              ? 'OpenCode Go usage not found'
+              : 'Set OpenCode web auth cookie for usage';
+          }
+        } catch (err) {
+          if (String(err?.code || '').startsWith('OPENCODE_GO_USAGE_')) {
+            row.status = 'missing';
+            row.source = 'usage-auth-missing';
+            row.sourceLabel = 'usage auth';
+            row.primary = '';
+            row.detail = usageStatus.authCookieSet
+              ? 'OpenCode Go usage not found'
+              : 'Set OpenCode web auth cookie for usage';
+          }
+        }
+      } else {
+        const snapshot = await apiSnapshot(item.id, options);
+        const known = snapshotRemaining(snapshot);
+        const usage = snapshotUsage(snapshot);
+        const windows = normaliseWindows(snapshot?.quotaWindows, clean(snapshot?.source) || 'provider-api');
+        row.updatedAt = num(snapshot?.cachedAt, null);
+        if (known) {
+          applyKnownRemaining(row, known, { estimated: false });
+          row.sourceLabel = 'API';
+          hasQuota = true;
+        } else if (windows.length) {
+          hasQuota = applyWindowQuota(row, windows, {
+            source: clean(snapshot?.source) || 'provider-api',
+            detail: 'provider quota',
+          });
+        } else if (usage) {
+          applyKnownUsage(row, usage);
+          hasQuota = true;
+        }
+      }
+      if (item.id !== 'opencode-go' && !row.includeInTotal && !hasQuota) {
+        row.status = 'hidden';
+        row.source = 'provider-hidden';
+        row.sourceLabel = 'hidden';
+        row.primary = '';
+        row.detail = 'Usage not exposed';
+      }
+    }
+    row.tone = rowTone(row);
+    emit(true);
+    return row;
+  });
+
+  const oauthTasks = (setup.oauth || []).map(async (item) => {
+    const providerCfg = providers[item.id] || {};
+    const row = baseRow(item, 'oauth', providerCfg);
+    row.status = row.authenticated ? 'checking' : 'missing';
+    row.source = row.authenticated ? 'checking' : 'not-configured';
+    row.sourceLabel = row.authenticated ? 'checking' : 'not signed in';
+    row.primary = row.authenticated ? '' : 'not signed in';
+    row.detail = row.authenticated ? 'Checking provider usage' : (item.detail || 'OAuth credentials missing');
+    row.tone = rowTone(row);
+    rows.push(row);
+    emit(true);
+
+    if (!row.authenticated) {
+      row.status = 'missing';
+      row.source = 'not-configured';
+      row.sourceLabel = 'not signed in';
+      row.primary = 'not signed in';
+      row.detail = item.detail || 'OAuth credentials missing';
+    } else {
+      try {
+        const snapshot = await oauthSnapshot(item.id, options);
+        const known = snapshotRemaining(snapshot);
+        const windows = normaliseWindows(snapshot?.quotaWindows, clean(snapshot?.source) || 'provider-api');
+        row.windows = windows;
+        row.updatedAt = num(snapshot?.cachedAt, null);
+        if (known) {
+          applyKnownRemaining(row, known, { estimated: false });
+          row.sourceLabel = 'API';
+          if (windows.length) {
+            row.windows = windows;
+            row.detail = 'subscription quota';
+          }
+        } else if (windows.length) {
+          row.status = 'partial';
+          row.source = clean(snapshot?.source) || 'provider-api';
+          row.sourceLabel = 'API window';
+          row.primary = windows.map(displayWindow).slice(0, 2).join(' · ');
+          row.detail = 'subscription quota';
+        } else {
+          row.status = 'hidden';
+          row.source = 'usage-disabled';
+          row.sourceLabel = 'disabled';
+          row.primary = '';
+          row.detail = 'No current usage query result';
+        }
+      } catch (err) {
+        row.status = 'error';
+        row.source = 'error';
+        row.sourceLabel = 'error';
+        row.primary = 'fetch failed';
+        row.detail = err?.message || String(err);
+      }
+    }
+    row.tone = rowTone(row);
+    emit(true);
+    return row;
+  });
+
+  for (const item of setup.local || []) {
+    if (item.id === 'lmstudio') continue;
+    const row = baseRow(item, 'local', providers[item.id] || {});
+    row.status = item.enabled || item.detected ? 'local' : 'missing';
+    row.source = 'local-provider';
+    row.sourceLabel = item.enabled || item.detected ? 'local' : 'off';
+    row.primary = item.enabled || item.detected ? 'local provider' : 'disabled';
+    row.detail = item.enabled || item.detected ? 'No billing quota' : 'Not running';
+    row.tone = rowTone(row);
+    rows.push(row);
+    emit(true);
+  }
+
+  await Promise.all([...apiTasks, ...oauthTasks]);
+
+  const dashboard = usageDashboardSnapshot(rows, { checkedAt, refresh, checking: false });
+  emitUsageDashboard(options, dashboard);
+  return dashboard;
+}