bonecode 1.2.1 → 1.2.2

package/src/stats.ts

@@ -1,10 +1,49 @@
 /**
  * Token usage and cost statistics
  * Ported from opencode's cli/cmd/stats.ts — adapted for BoneCode's Postgres schema.
+ *
+ * Works on both Postgres (full schema with token columns on sessions) and
+ * SQLite fallback (sessions has only id/title/etc; token data lives on messages).
  */
 
 import { Pool } from "pg"
 
+interface CostColumns {
+  cost: boolean
+  tokensInput: boolean
+  tokensOutput: boolean
+  tokensReasoning: boolean
+  tokensCacheRead: boolean
+  tokensCacheWrite: boolean
+}
+
+let _columnsCache: CostColumns | null = null
+
+async function detectSessionCostColumns(pool: Pool): Promise<CostColumns> {
+  if (_columnsCache) return _columnsCache
+
+  const probe = async (col: string): Promise<boolean> => {
+    try {
+      // SELECT col FROM sessions LIMIT 0 — fails if column doesn't exist, succeeds otherwise.
+      // Works identically on Postgres and SQLite.
+      await pool.query(`SELECT ${col} FROM sessions LIMIT 0`)
+      return true
+    } catch {
+      return false
+    }
+  }
+
+  _columnsCache = {
+    cost: await probe("cost_usd"),
+    tokensInput: await probe("tokens_input"),
+    tokensOutput: await probe("tokens_output"),
+    tokensReasoning: await probe("tokens_reasoning"),
+    tokensCacheRead: await probe("tokens_cache_read"),
+    tokensCacheWrite: await probe("tokens_cache_write"),
+  }
+  return _columnsCache
+}
+
 export interface SessionStats {
   totalSessions: number
   totalMessages: number
@@ -50,10 +89,23 @@ export async function aggregateStats(
     return opts.days
   })()
 
+  // Detect available columns by introspection — works on both Postgres and SQLite
+  const hasCol = await detectSessionCostColumns(pool)
+  const sessionSelect = [
+    "id",
+    "project_id",
+    "created_at",
+    "updated_at",
+    hasCol.cost ? "cost_usd" : "0 AS cost_usd",
+    hasCol.tokensInput ? "tokens_input" : "0 AS tokens_input",
+    hasCol.tokensOutput ? "tokens_output" : "0 AS tokens_output",
+    hasCol.tokensReasoning ? "tokens_reasoning" : "0 AS tokens_reasoning",
+    hasCol.tokensCacheRead ? "tokens_cache_read" : "0 AS tokens_cache_read",
+    hasCol.tokensCacheWrite ? "tokens_cache_write" : "0 AS tokens_cache_write",
+  ].join(", ")
+
   // Load sessions
-  let sessionQuery = `SELECT id, project_id, cost_usd, tokens_input, tokens_output, tokens_reasoning,
-    tokens_cache_read, tokens_cache_write, created_at, updated_at
-    FROM sessions WHERE state != 'deleted'`
+  let sessionQuery = `SELECT ${sessionSelect} FROM sessions WHERE state != 'deleted'`
   const params: unknown[] = []
   let idx = 1