@hasna/economy 0.1.0 → 0.2.0

package/dist/mcp/index.js

@@ -74,19 +74,19 @@ var DEFAULT_PRICING;
 var init_pricing = __esm(() => {
   init_database();
   DEFAULT_PRICING = {
-    "claude-opus-4-6": { inputPer1M: 15, outputPer1M: 75, cacheReadPer1M: 1.5, cacheWritePer1M: 18.75 },
-    "claude-opus-4-5": { inputPer1M: 15, outputPer1M: 75, cacheReadPer1M: 1.5, cacheWritePer1M: 18.75 },
+    "claude-opus-4-6": { inputPer1M: 5, outputPer1M: 25, cacheReadPer1M: 0.5, cacheWritePer1M: 6.25 },
+    "claude-opus-4-5": { inputPer1M: 5, outputPer1M: 25, cacheReadPer1M: 0.5, cacheWritePer1M: 6.25 },
     "claude-sonnet-4-6": { inputPer1M: 3, outputPer1M: 15, cacheReadPer1M: 0.3, cacheWritePer1M: 3.75 },
     "claude-sonnet-4-5": { inputPer1M: 3, outputPer1M: 15, cacheReadPer1M: 0.3, cacheWritePer1M: 3.75 },
-    "claude-haiku-4-5": { inputPer1M: 0.8, outputPer1M: 4, cacheReadPer1M: 0.08, cacheWritePer1M: 1 },
+    "claude-haiku-4-5": { inputPer1M: 1, outputPer1M: 5, cacheReadPer1M: 0.1, cacheWritePer1M: 1.25 },
     "claude-3-5-sonnet": { inputPer1M: 3, outputPer1M: 15, cacheReadPer1M: 0.3, cacheWritePer1M: 3.75 },
-    "claude-3-5-haiku": { inputPer1M: 0.8, outputPer1M: 4, cacheReadPer1M: 0.08, cacheWritePer1M: 1 },
+    "claude-3-5-haiku": { inputPer1M: 1, outputPer1M: 5, cacheReadPer1M: 0.1, cacheWritePer1M: 1.25 },
     "claude-3-opus": { inputPer1M: 15, outputPer1M: 75, cacheReadPer1M: 1.5, cacheWritePer1M: 18.75 },
     "claude-3-sonnet": { inputPer1M: 3, outputPer1M: 15, cacheReadPer1M: 0.3, cacheWritePer1M: 3.75 },
     "claude-3-haiku": { inputPer1M: 0.25, outputPer1M: 1.25, cacheReadPer1M: 0.03, cacheWritePer1M: 0.3 },
-    "gpt-5.3-codex": { inputPer1M: 30, outputPer1M: 120, cacheReadPer1M: 7.5, cacheWritePer1M: 0 },
-    "gpt-5.2-codex": { inputPer1M: 30, outputPer1M: 120, cacheReadPer1M: 7.5, cacheWritePer1M: 0 },
-    "gpt-5-codex": { inputPer1M: 30, outputPer1M: 120, cacheReadPer1M: 7.5, cacheWritePer1M: 0 },
+    "gpt-5.3-codex": { inputPer1M: 1.75, outputPer1M: 14, cacheReadPer1M: 0.44, cacheWritePer1M: 0 },
+    "gpt-5.2-codex": { inputPer1M: 1.75, outputPer1M: 14, cacheReadPer1M: 0.44, cacheWritePer1M: 0 },
+    "gpt-5-codex": { inputPer1M: 1.75, outputPer1M: 14, cacheReadPer1M: 0.44, cacheWritePer1M: 0 },
     "gpt-4o": { inputPer1M: 2.5, outputPer1M: 10, cacheReadPer1M: 1.25, cacheWritePer1M: 0 },
     "gpt-4o-mini": { inputPer1M: 0.15, outputPer1M: 0.6, cacheReadPer1M: 0.075, cacheWritePer1M: 0 },
     o1: { inputPer1M: 15, outputPer1M: 60, cacheReadPer1M: 7.5, cacheWritePer1M: 0 },
@@ -285,8 +285,22 @@ function querySummary(db, period) {
            COALESCE(SUM(input_tokens + output_tokens + cache_read_tokens + cache_create_tokens), 0) as tokens
     FROM requests WHERE ${rWhere}
   `).get();
-  const s = db.prepare(`SELECT COUNT(*) as sessions FROM sessions WHERE ${sWhere}`).get();
-  return { total_usd: r.total_usd, requests: r.requests, tokens: r.tokens, sessions: s.sessions, period };
+  const codexTotals = db.prepare(`
+    SELECT COALESCE(SUM(total_cost_usd), 0) as cost_usd,
+           COALESCE(SUM(total_tokens), 0) as tokens,
+           COUNT(*) as sessions
+    FROM sessions
+    WHERE ${sWhere}
+    AND id NOT IN (SELECT DISTINCT session_id FROM requests)
+  `).get();
+  const sessionCount = db.prepare(`SELECT COUNT(*) as sessions FROM sessions WHERE ${sWhere}`).get();
+  return {
+    total_usd: r.total_usd + codexTotals.cost_usd,
+    requests: r.requests,
+    tokens: r.tokens + codexTotals.tokens,
+    sessions: sessionCount.sessions,
+    period
+  };
 }
 function queryModelBreakdown(db) {
   return db.prepare(`
@@ -301,14 +315,14 @@ function queryModelBreakdown(db) {
 }
 function queryProjectBreakdown(db) {
   return db.prepare(`
-    SELECT s.project_path, s.project_name,
-           COUNT(DISTINCT s.id) as sessions,
-           COALESCE(SUM(r.input_tokens + r.output_tokens + r.cache_read_tokens + r.cache_create_tokens), 0) as total_tokens,
-           COUNT(r.id) as requests,
-           COALESCE(SUM(s.total_cost_usd), 0) as cost_usd,
-           MAX(s.started_at) as last_active
-    FROM sessions s LEFT JOIN requests r ON r.session_id = s.id
-    GROUP BY s.project_path ORDER BY cost_usd DESC
+    SELECT project_path, project_name,
+           COUNT(*) as sessions,
+           COALESCE(SUM(total_tokens), 0) as total_tokens,
+           COALESCE(SUM(request_count), 0) as requests,
+           COALESCE(SUM(total_cost_usd), 0) as cost_usd,
+           MAX(started_at) as last_active
+    FROM sessions
+    GROUP BY project_path ORDER BY cost_usd DESC
   `).all();
 }
 function listBudgets(db) {
@@ -383,118 +397,140 @@ import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprot
 
 // src/ingest/claude.ts
 init_database();
-import { readdirSync, readFileSync, existsSync as existsSync2 } from "fs";
+init_pricing();
+import { readdirSync, readFileSync, existsSync as existsSync2, statSync } from "fs";
 import { homedir as homedir2 } from "os";
 import { join as join2, basename } from "path";
-import { randomUUID } from "crypto";
-var TELEMETRY_DIR = join2(homedir2(), ".claude", "telemetry");
 var PROJECTS_DIR = join2(homedir2(), ".claude", "projects");
-function resolveProjectPath(sessionId) {
-  if (!existsSync2(PROJECTS_DIR))
-    return { projectPath: "", projectName: "unknown" };
-  try {
-    const projectDirs = readdirSync(PROJECTS_DIR, { withFileTypes: true }).filter((d) => d.isDirectory());
-    for (const dir of projectDirs) {
-      const sessionFile = join2(PROJECTS_DIR, dir.name, `${sessionId}.jsonl`);
-      if (existsSync2(sessionFile)) {
-        const projectPath = dir.name.replace(/^-/, "/").replace(/-/g, "/");
-        return { projectPath, projectName: basename(projectPath) };
+function dirNameToPath(dirName) {
+  return dirName.replace(/^-/, "/").replace(/-/g, "/").replace(/\/\//g, "/-");
+}
+function collectJsonlFiles(projectDir) {
+  const files = [];
+  function walk(dir) {
+    try {
+      for (const entry of readdirSync(dir, { withFileTypes: true })) {
+        if (entry.isDirectory())
+          walk(join2(dir, entry.name));
+        else if (entry.name.endsWith(".jsonl"))
+          files.push(join2(dir, entry.name));
       }
-    }
-  } catch {}
-  return { projectPath: "", projectName: "unknown" };
+    } catch {}
+  }
+  walk(projectDir);
+  return files;
 }
-async function ingestClaude(db, verbose = false, telemetryDir = TELEMETRY_DIR) {
-  if (!existsSync2(telemetryDir)) {
+async function ingestClaude(db, verbose = false, _telemetryDir) {
+  if (!existsSync2(PROJECTS_DIR)) {
     if (verbose)
-      console.log("Claude telemetry dir not found:", telemetryDir);
+      console.log("Claude projects dir not found:", PROJECTS_DIR);
     return { files: 0, requests: 0, sessions: 0 };
   }
-  const files = readdirSync(telemetryDir).filter((f) => f.endsWith(".json"));
+  let totalFiles = 0;
   let totalRequests = 0;
-  let processedFiles = 0;
   const touchedSessions = new Set;
-  for (const filename of files) {
-    const stateKey = filename;
-    const processed = getIngestState(db, "claude", stateKey);
-    if (processed === "done")
-      continue;
-    const filePath = join2(telemetryDir, filename);
-    let events;
-    try {
-      const raw = readFileSync(filePath, "utf-8");
-      events = JSON.parse(raw);
-      if (!Array.isArray(events))
-        events = [events];
-    } catch {
-      if (verbose)
-        console.log("Skip unreadable:", filename);
-      continue;
-    }
-    for (const event of events) {
-      const ed = event.event_data;
-      if (!ed || ed.event_name !== "tengu_api_success")
+  const projectDirs = readdirSync(PROJECTS_DIR, { withFileTypes: true }).filter((d) => d.isDirectory());
+  for (const projectDirEntry of projectDirs) {
+    const projectDirPath = join2(PROJECTS_DIR, projectDirEntry.name);
+    const projectPath = dirNameToPath(projectDirEntry.name);
+    const projectName = basename(projectPath);
+    const jsonlFiles = collectJsonlFiles(projectDirPath);
+    for (const filePath of jsonlFiles) {
+      const stateKey = filePath.replace(PROJECTS_DIR, "");
+      let fileMtime = "0";
+      try {
+        fileMtime = statSync(filePath).mtimeMs.toString();
+      } catch {
         continue;
-      const meta = ed.additional_metadata;
-      if (!meta)
+      }
+      const processed = getIngestState(db, "claude", stateKey);
+      if (processed === fileMtime)
         continue;
-      const sessionId = ed.session_id ?? randomUUID();
-      const timestamp = ed.client_timestamp ?? new Date().toISOString();
-      const model = meta.model ?? ed.model ?? "unknown";
-      const costUsd = meta.costUSD ?? 0;
-      const inputTokens = meta.inputTokens ?? (meta.uncachedInputTokens ?? 0);
-      const outputTokens = meta.outputTokens ?? 0;
-      const cacheReadTokens = meta.cachedInputTokens ?? 0;
-      const requestId = meta.requestId ?? randomUUID();
-      upsertRequest(db, {
-        id: `claude-${requestId}`,
-        agent: "claude",
-        session_id: sessionId,
-        model,
-        input_tokens: inputTokens,
-        output_tokens: outputTokens,
-        cache_read_tokens: cacheReadTokens,
-        cache_create_tokens: 0,
-        cost_usd: costUsd,
-        duration_ms: meta.durationMs ?? 0,
-        timestamp,
-        source_request_id: requestId
-      });
-      if (!touchedSessions.has(sessionId)) {
-        const { projectPath, projectName } = resolveProjectPath(sessionId);
-        const existing = db.prepare(`SELECT id FROM sessions WHERE id = ?`).get(sessionId);
-        if (!existing) {
-          const session = {
-            id: sessionId,
-            agent: "claude",
-            project_path: projectPath,
-            project_name: projectName,
-            started_at: timestamp,
-            ended_at: null,
-            total_cost_usd: 0,
-            total_tokens: 0,
-            request_count: 0
-          };
-          upsertSession(db, session);
+      let lines;
+      try {
+        lines = readFileSync(filePath, "utf-8").split(`
+`).filter((l) => l.trim());
+      } catch {
+        continue;
+      }
+      const fileBasename = basename(filePath, ".jsonl");
+      const isUuid = /^[0-9a-f-]{36}$/.test(fileBasename);
+      let sessionId = isUuid ? fileBasename : fileBasename.replace(/^agent-/, "");
+      let sessionCwd = projectPath;
+      for (const line of lines) {
+        let entry;
+        try {
+          entry = JSON.parse(line);
+        } catch {
+          continue;
         }
-        touchedSessions.add(sessionId);
+        if (entry.sessionId)
+          sessionId = entry.sessionId;
+        if (entry.cwd)
+          sessionCwd = entry.cwd;
+        if (entry.message?.role !== "assistant")
+          continue;
+        const usage = entry.message.usage;
+        if (!usage)
+          continue;
+        const model = entry.message.model;
+        if (!model)
+          continue;
+        const inputTokens = usage.input_tokens ?? 0;
+        const outputTokens = usage.output_tokens ?? 0;
+        const cacheWriteTokens = usage.cache_creation_input_tokens ?? 0;
+        const cacheReadTokens = usage.cache_read_input_tokens ?? 0;
+        const timestamp = entry.timestamp ?? new Date().toISOString();
+        if (inputTokens + outputTokens + cacheWriteTokens === 0)
+          continue;
+        const costUsd = computeCostFromDb(db, model, inputTokens, outputTokens, cacheReadTokens, cacheWriteTokens);
+        const reqId = `claude-${sessionId}-${timestamp}`;
+        upsertRequest(db, {
+          id: reqId,
+          agent: "claude",
+          session_id: sessionId,
+          model,
+          input_tokens: inputTokens,
+          output_tokens: outputTokens,
+          cache_read_tokens: cacheReadTokens,
+          cache_create_tokens: cacheWriteTokens,
+          cost_usd: costUsd,
+          duration_ms: 0,
+          timestamp,
+          source_request_id: reqId
+        });
+        if (!touchedSessions.has(sessionId)) {
+          const existing = db.prepare(`SELECT id FROM sessions WHERE id = ?`).get(sessionId);
+          if (!existing) {
+            const session = {
+              id: sessionId,
+              agent: "claude",
+              project_path: sessionCwd || projectPath,
+              project_name: basename(sessionCwd || projectPath),
+              started_at: timestamp,
+              ended_at: null,
+              total_cost_usd: 0,
+              total_tokens: 0,
+              request_count: 0
+            };
+            upsertSession(db, session);
+          }
+          touchedSessions.add(sessionId);
+        }
+        totalRequests++;
       }
-      totalRequests++;
+      setIngestState(db, "claude", stateKey, fileMtime);
+      totalFiles++;
     }
-    setIngestState(db, "claude", stateKey, "done");
-    processedFiles++;
-    if (verbose)
-      console.log(`Processed ${filename}: found ${events.length} events`);
   }
   for (const sessionId of touchedSessions) {
     rollupSession(db, sessionId);
   }
-  return { files: processedFiles, requests: totalRequests, sessions: touchedSessions.size };
+  return { files: totalFiles, requests: totalRequests, sessions: touchedSessions.size };
 }
 
 // src/ingest/codex.ts
 init_database();
-init_pricing();
 import { existsSync as existsSync3, readFileSync as readFileSync2 } from "fs";
 import { homedir as homedir3 } from "os";
 import { join as join3, basename as basename2 } from "path";
@@ -529,9 +565,7 @@ async function ingestCodex(db, verbose = false) {
       const processed = getIngestState(db, "codex", stateKey);
       if (processed === "done")
         continue;
-      const inputTokens = Math.floor(thread.tokens_used * 0.6);
-      const outputTokens = thread.tokens_used - inputTokens;
-      const costUsd = computeCost(model, inputTokens, outputTokens);
+      const costUsd = 0;
       const projectPath = thread.cwd ?? "";
       const projectName = projectPath ? basename2(projectPath) : "unknown";
       const startedAt = thread.created_at ? new Date(thread.created_at * 1000).toISOString() : new Date().toISOString();
@@ -632,7 +666,10 @@ server.setRequestHandler(CallToolRequestSchema, async (req) => {
       case "get_cost_summary": {
         const period = a["period"] ?? "today";
         const summary = querySummary(db, period);
-        return { content: [{ type: "text", text: JSON.stringify(summary, null, 2) }] };
+        const fmtUsd = (n) => "$" + n.toLocaleString("en-US", { minimumFractionDigits: 2, maximumFractionDigits: 2 });
+        const fmtTok = (n) => n >= 1e9 ? `${(n / 1e9).toFixed(1)}B` : n >= 1e6 ? `${(n / 1e6).toFixed(1)}M` : n >= 1000 ? `${(n / 1000).toFixed(1)}k` : String(n);
+        const result = { ...summary, summary: `You've spent ${fmtUsd(summary.total_usd)} ${period === "all" ? "total" : period} across ${summary.sessions} sessions (${summary.requests.toLocaleString()} requests, ${fmtTok(summary.tokens)} tokens)` };
+        return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
       }
       case "get_sessions": {
         const sessions = querySessions(db, {

package/dist/server/index.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=index.d.ts.map

package/dist/server/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/server/index.ts"],"names":[],"mappings":""}