aisessions 1.0.0

package/src/usage/index.js

@@ -0,0 +1,111 @@
+import { spawnSync } from 'node:child_process'
+
+function run(args) {
+  const r = spawnSync('npx', ['--yes', 'ccusage@latest', ...args, '--json'], {
+    timeout: 30_000, maxBuffer: 16 * 1024 * 1024, encoding: 'utf8',
+    env: { ...process.env, NO_COLOR: '1' },
+  })
+  if (r.status !== 0 || !r.stdout) return null
+  try { return JSON.parse(r.stdout) } catch { return null }
+}
+
+function runLocal(args) {
+  try {
+    const r = spawnSync('ccusage', [...args, '--json'], {
+      timeout: 30_000, maxBuffer: 16 * 1024 * 1024, encoding: 'utf8',
+      env: { ...process.env, NO_COLOR: '1' },
+    })
+    if (r.status !== 0 || !r.stdout) return null
+    return JSON.parse(r.stdout)
+  } catch { return null }
+}
+
+// Normalise the two very different per-agent response shapes into one schema.
+function normalizeEntries(entries, agentId) {
+  return (entries || []).map(d => {
+    // Claude-style keys              Codex-style keys
+    const cost      = d.totalCost         ?? d.costUSD          ?? 0
+    const input     = d.inputTokens       ?? 0
+    const output    = d.outputTokens      ?? 0
+    const cacheRead = d.cacheReadTokens   ?? d.cachedInputTokens ?? 0
+    const cacheWrite= d.cacheCreationTokens ?? 0
+    const reasoning = d.reasoningOutputTokens ?? 0
+    const total     = d.totalTokens       ?? (input + output + cacheRead + cacheWrite)
+
+    let models = []
+    if (Array.isArray(d.modelBreakdowns)) {
+      // Claude format
+      models = d.modelBreakdowns.map(m => ({
+        name:       m.modelName || '?',
+        cost:       m.cost              ?? 0,
+        input:      m.inputTokens       ?? 0,
+        output:     m.outputTokens      ?? 0,
+        cacheRead:  m.cacheReadTokens   ?? 0,
+        cacheWrite: m.cacheCreationTokens ?? 0,
+        reasoning:  0,
+      }))
+    } else if (d.models && typeof d.models === 'object') {
+      // Codex format: { "gpt-5.1": { inputTokens, outputTokens, ... } }
+      models = Object.entries(d.models).map(([name, m]) => ({
+        name,
+        cost:       m.cost              ?? 0,
+        input:      m.inputTokens       ?? 0,
+        output:     m.outputTokens      ?? 0,
+        cacheRead:  m.cachedInputTokens ?? 0,
+        cacheWrite: 0,
+        reasoning:  m.reasoningOutputTokens ?? 0,
+      }))
+    } else if (Array.isArray(d.modelsUsed)) {
+      models = d.modelsUsed.map(name => ({ name, cost: 0, input: 0, output: 0, cacheRead: 0, cacheWrite: 0, reasoning: 0 }))
+    }
+
+    return {
+      date: d.date || d.period || '',
+      agent: agentId || 'all',
+      cost, input, output, cacheRead, cacheWrite, reasoning, total,
+      models,
+    }
+  })
+}
+
+function totals(days) {
+  return days.reduce((a, d) => ({
+    totalCost:           a.totalCost           + d.cost,
+    inputTokens:         a.inputTokens         + d.input,
+    outputTokens:        a.outputTokens        + d.output,
+    cacheReadTokens:     a.cacheReadTokens     + d.cacheRead,
+    cacheCreationTokens: a.cacheCreationTokens + d.cacheWrite,
+    reasoningTokens:     a.reasoningTokens     + d.reasoning,
+    totalTokens:         a.totalTokens         + d.total,
+  }), { totalCost: 0, inputTokens: 0, outputTokens: 0, cacheReadTokens: 0, cacheCreationTokens: 0, reasoningTokens: 0, totalTokens: 0 })
+}
+
+export async function getUsageData({ since, until, agent } = {}) {
+  const dateArgs = []
+  if (since) dateArgs.push('--since', since)
+  if (until) dateArgs.push('--until', until)
+
+  let raw
+  if (agent && agent !== 'all') {
+    // Per-agent command: npx ccusage claude daily --json
+    const agentArgs = [agent, 'daily', ...dateArgs]
+    raw = runLocal(agentArgs) ?? run(agentArgs)
+    if (!raw) {
+      // Fallback: run global and filter by metadata.agents
+      const globalArgs = ['daily', ...dateArgs]
+      const g = runLocal(globalArgs) ?? run(globalArgs)
+      if (g && Array.isArray(g.daily)) {
+        raw = { daily: g.daily.filter(d => (d.metadata?.agents || []).includes(agent)) }
+      }
+    }
+  } else {
+    raw = runLocal(['daily', ...dateArgs]) ?? run(['daily', ...dateArgs])
+  }
+
+  if (!raw) return { available: false, error: 'ccusage not found — run: npm i -g ccusage' }
+
+  const entries = normalizeEntries(raw.daily || (Array.isArray(raw) ? raw : []), agent || 'all')
+  entries.sort((a, b) => (a.date < b.date ? -1 : a.date > b.date ? 1 : 0))
+
+  return { available: true, agent: agent || 'all', daily: entries, totals: totals(entries) }
+}