@tokentop/ttop 0.2.2 → 0.3.0

package/src/pricing/index.ts

@@ -1,6 +1,12 @@
-import type { ModelPricing, ProviderPlugin, ProviderPricing } from '@/plugins/types/provider.ts';
-import { getModelPricing, getProviderModels, normalizeProviderName, setProviderAliases, clearCache as clearModelsDevCache } from './models-dev.ts';
-import { getFallbackPricing, getFallbackProviderPricing } from './fallback.ts';
+import type { ModelPricing, ProviderPlugin, ProviderPricing } from "@/plugins/types/provider.ts";
+import { getFallbackPricing, getFallbackProviderPricing } from "./fallback.ts";
+import {
+  clearCache as clearModelsDevCache,
+  getModelPricing,
+  getProviderModels,
+  normalizeProviderName,
+  setProviderAliases,
+} from "./models-dev.ts";
 
 export function initPricingFromPlugins(
   plugins: readonly ProviderPlugin[],
@@ -12,8 +18,8 @@ export function initPricingFromPlugins(
     const pricing = plugin.pricing;
     if (
       pricing &&
-      'modelsDevProviderId' in pricing &&
-      typeof (pricing as ProviderPricing).modelsDevProviderId === 'string'
+      "modelsDevProviderId" in pricing &&
+      typeof (pricing as ProviderPricing).modelsDevProviderId === "string"
     ) {
       aliases.set(plugin.id, (pricing as ProviderPricing).modelsDevProviderId!);
     }
@@ -30,10 +36,10 @@ export function initPricingFromPlugins(
 
 export async function getPricing(
   providerId: string,
-  modelId: string
+  modelId: string,
 ): Promise<ModelPricing | null> {
   const normalizedProvider = normalizeProviderName(providerId);
-  
+
   const modelsDevPricing = await getModelPricing(normalizedProvider, modelId);
   if (modelsDevPricing) return modelsDevPricing;
 
@@ -44,10 +50,10 @@ export async function getPricing(
 }
 
 export async function getProviderPricing(
-  providerId: string
+  providerId: string,
 ): Promise<Record<string, ModelPricing>> {
   const normalizedProvider = normalizeProviderName(providerId);
-  
+
   const modelsDevPricing = await getProviderModels(normalizedProvider);
   if (modelsDevPricing && Object.keys(modelsDevPricing).length > 0) {
     return modelsDevPricing;
@@ -61,6 +67,6 @@ export function clearPricingCache(): void {
   clearModelsDevCache();
 }
 
-export { estimateCost, estimateSessionCost, formatCost, formatTokenCount } from './estimator.ts';
-export type { TokenUsage } from './estimator.ts';
-export { FALLBACK_PRICING } from './fallback.ts';
+export type { TokenUsage } from "./estimator.ts";
+export { estimateCost, estimateSessionCost, formatCost, formatTokenCount } from "./estimator.ts";
+export { FALLBACK_PRICING } from "./fallback.ts";

package/src/pricing/models-dev.ts

@@ -1,6 +1,6 @@
-import type { ModelPricing } from '@/plugins/types/provider.ts';
+import type { ModelPricing } from "@/plugins/types/provider.ts";
 
-const MODELS_DEV_API = 'https://models.dev/api.json';
+const MODELS_DEV_API = "https://models.dev/api.json";
 const CACHE_TTL_MS = 3600000;
 
 interface ModelsDevCost {
@@ -63,7 +63,7 @@ export function normalizeProviderName(providerId: string): string {
 
 export async function getModelPricing(
   providerId: string,
-  modelId: string
+  modelId: string,
 ): Promise<ModelPricing | null> {
   const data = await fetchModelsDevData();
   if (!data) return null;
@@ -78,7 +78,7 @@ export async function getModelPricing(
   const pricing: ModelPricing = {
     input: model.cost.input,
     output: model.cost.output,
-    source: 'models.dev',
+    source: "models.dev",
   };
 
   if (model.cost.cache_read !== undefined) {
@@ -92,7 +92,7 @@ export async function getModelPricing(
 }
 
 export async function getProviderModels(
-  providerId: string
+  providerId: string,
 ): Promise<Record<string, ModelPricing> | null> {
   const data = await fetchModelsDevData();
   if (!data) return null;
@@ -109,7 +109,7 @@ export async function getProviderModels(
     const pricing: ModelPricing = {
       input: model.cost.input,
       output: model.cost.output,
-      source: 'models.dev',
+      source: "models.dev",
     };
 
     if (model.cost.cache_read !== undefined) {

package/src/storage/database.ts

@@ -1,6 +1,6 @@
-import { Database } from 'bun:sqlite';
-import * as fs from 'fs/promises';
-import { PATHS } from './paths.ts';
+import { Database } from "bun:sqlite";
+import * as fs from "fs/promises";
+import { PATHS } from "./paths.ts";
 
 let db: Database | null = null;
 
@@ -76,8 +76,8 @@ export async function initDatabase(): Promise<Database> {
   await fs.mkdir(PATHS.data.dir, { recursive: true });
 
   db = new Database(PATHS.data.database, { create: true });
-  db.exec('PRAGMA journal_mode = WAL');
-  db.exec('PRAGMA synchronous = NORMAL');
+  db.exec("PRAGMA journal_mode = WAL");
+  db.exec("PRAGMA synchronous = NORMAL");
   db.exec(SCHEMA);
 
   return db;
@@ -85,7 +85,7 @@ export async function initDatabase(): Promise<Database> {
 
 export function getDatabase(): Database {
   if (!db) {
-    throw new Error('Database not initialized. Call initDatabase() first.');
+    throw new Error("Database not initialized. Call initDatabase() first.");
   }
   return db;
 }
@@ -166,7 +166,7 @@ export function insertUsage(record: UsageRecord): void {
     record.cacheReadTokens ?? 0,
     record.cacheWriteTokens ?? 0,
     record.costUsd,
-    record.projectPath ?? null
+    record.projectPath ?? null,
   );
 }
 
@@ -193,7 +193,7 @@ export function insertUsageBatch(records: UsageRecord[]): void {
         record.cacheReadTokens ?? 0,
         record.cacheWriteTokens ?? 0,
         record.costUsd,
-        record.projectPath ?? null
+        record.projectPath ?? null,
       );
     }
   });
@@ -218,7 +218,7 @@ export function insertProviderSnapshot(snapshot: ProviderSnapshot): void {
     snapshot.tokensInput ?? null,
     snapshot.tokensOutput ?? null,
     snapshot.costUsd ?? null,
-    snapshot.rawJson ?? null
+    snapshot.rawJson ?? null,
   );
 }
 
@@ -228,33 +228,33 @@ export function queryUsage(query: UsageQuery): UsageRecord[] {
   const params: (string | number)[] = [];
 
   if (query.startTime !== undefined) {
-    conditions.push('timestamp >= ?');
+    conditions.push("timestamp >= ?");
     params.push(query.startTime);
   }
   if (query.endTime !== undefined) {
-    conditions.push('timestamp <= ?');
+    conditions.push("timestamp <= ?");
     params.push(query.endTime);
   }
   if (query.provider) {
-    conditions.push('provider = ?');
+    conditions.push("provider = ?");
     params.push(query.provider);
   }
   if (query.model) {
-    conditions.push('model = ?');
+    conditions.push("model = ?");
     params.push(query.model);
   }
   if (query.agent) {
-    conditions.push('agent = ?');
+    conditions.push("agent = ?");
     params.push(query.agent);
   }
   if (query.sessionId) {
-    conditions.push('session_id = ?');
+    conditions.push("session_id = ?");
     params.push(query.sessionId);
   }
 
-  const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(' AND ')}` : '';
-  const limitClause = query.limit ? `LIMIT ${query.limit}` : '';
-  const offsetClause = query.offset ? `OFFSET ${query.offset}` : '';
+  const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+  const limitClause = query.limit ? `LIMIT ${query.limit}` : "";
+  const offsetClause = query.offset ? `OFFSET ${query.offset}` : "";
 
   const sql = `
     SELECT * FROM usage
@@ -303,7 +303,8 @@ export interface UsageSummary {
 
 export function getUsageSummary(startTime: number, endTime: number): UsageSummary {
   const db = getDatabase();
-  const row = db.prepare(`
+  const row = db
+    .prepare(`
     SELECT
       COALESCE(SUM(input_tokens), 0) as total_input,
       COALESCE(SUM(output_tokens), 0) as total_output,
@@ -313,7 +314,8 @@ export function getUsageSummary(startTime: number, endTime: number): UsageSummar
       COUNT(*) as request_count
     FROM usage
     WHERE timestamp >= ? AND timestamp <= ?
-  `).get(startTime, endTime) as {
+  `)
+    .get(startTime, endTime) as {
     total_input: number;
     total_output: number;
     total_cache_read: number;
@@ -342,7 +344,8 @@ export interface ProviderSummary {
 
 export function getUsageByProvider(startTime: number, endTime: number): ProviderSummary[] {
   const db = getDatabase();
-  const rows = db.prepare(`
+  const rows = db
+    .prepare(`
     SELECT
       provider,
       COALESCE(SUM(input_tokens), 0) as total_input,
@@ -353,7 +356,8 @@ export function getUsageByProvider(startTime: number, endTime: number): Provider
     WHERE timestamp >= ? AND timestamp <= ?
     GROUP BY provider
     ORDER BY total_cost DESC
-  `).all(startTime, endTime) as Array<{
+  `)
+    .all(startTime, endTime) as Array<{
     provider: string;
     total_input: number;
     total_output: number;
@@ -381,7 +385,8 @@ export interface ModelSummary {
 
 export function getUsageByModel(startTime: number, endTime: number): ModelSummary[] {
   const db = getDatabase();
-  const rows = db.prepare(`
+  const rows = db
+    .prepare(`
     SELECT
       model,
       provider,
@@ -393,7 +398,8 @@ export function getUsageByModel(startTime: number, endTime: number): ModelSummar
     WHERE timestamp >= ? AND timestamp <= ?
     GROUP BY model, provider
     ORDER BY total_cost DESC
-  `).all(startTime, endTime) as Array<{
+  `)
+    .all(startTime, endTime) as Array<{
     model: string;
     provider: string;
     total_input: number;
@@ -422,12 +428,13 @@ export interface TimeSeriesPoint {
 export function getUsageTimeSeries(
   startTime: number,
   endTime: number,
-  bucketMinutes: number = 5
+  bucketMinutes: number = 5,
 ): TimeSeriesPoint[] {
   const db = getDatabase();
   const bucketSeconds = bucketMinutes * 60;
 
-  const rows = db.prepare(`
+  const rows = db
+    .prepare(`
     SELECT
       (timestamp / ? * ?) as bucket_time,
       COALESCE(SUM(input_tokens + output_tokens), 0) as total_tokens,
@@ -437,7 +444,8 @@ export function getUsageTimeSeries(
     WHERE timestamp >= ? AND timestamp <= ?
     GROUP BY bucket_time
     ORDER BY bucket_time ASC
-  `).all(bucketSeconds, bucketSeconds, startTime, endTime) as Array<{
+  `)
+    .all(bucketSeconds, bucketSeconds, startTime, endTime) as Array<{
     bucket_time: number;
     total_tokens: number;
     total_cost: number;
@@ -456,15 +464,17 @@ export function getProviderSnapshots(
   provider: string,
   startTime: number,
   endTime: number,
-  limit: number = 100
+  limit: number = 100,
 ): ProviderSnapshot[] {
   const db = getDatabase();
-  const rows = db.prepare(`
+  const rows = db
+    .prepare(`
     SELECT * FROM provider_snapshots
     WHERE provider = ? AND timestamp >= ? AND timestamp <= ?
     ORDER BY timestamp DESC
     LIMIT ?
-  `).all(provider, startTime, endTime, limit) as Array<{
+  `)
+    .all(provider, startTime, endTime, limit) as Array<{
     timestamp: number;
     provider: string;
     used_percent: number | null;
@@ -514,29 +524,31 @@ export function upsertDailyAggregate(aggregate: DailyAggregate): void {
     aggregate.totalCacheRead,
     aggregate.totalCacheWrite,
     aggregate.totalCostUsd,
-    aggregate.requestCount
+    aggregate.requestCount,
   );
 }
 
 export function getDailyAggregates(
   startDate: string,
   endDate: string,
-  provider?: string
+  provider?: string,
 ): DailyAggregate[] {
   const db = getDatabase();
-  const conditions = ['date >= ?', 'date <= ?'];
+  const conditions = ["date >= ?", "date <= ?"];
   const params: string[] = [startDate, endDate];
 
   if (provider) {
-    conditions.push('provider = ?');
+    conditions.push("provider = ?");
     params.push(provider);
   }
 
-  const rows = db.prepare(`
+  const rows = db
+    .prepare(`
     SELECT * FROM daily_aggregates
-    WHERE ${conditions.join(' AND ')}
+    WHERE ${conditions.join(" AND ")}
     ORDER BY date DESC, total_cost_usd DESC
-  `).all(...params) as Array<{
+  `)
+    .all(...params) as Array<{
     date: string;
     provider: string;
     model: string | null;
@@ -582,9 +594,9 @@ export function getThisMonthSummary(): UsageSummary {
 
 export function pluginStorageGet(pluginId: string, key: string): string | null {
   const db = getDatabase();
-  const row = db.prepare(
-    'SELECT value FROM plugin_storage WHERE plugin_id = ? AND key = ?',
-  ).get(pluginId, key) as { value: string } | null;
+  const row = db
+    .prepare("SELECT value FROM plugin_storage WHERE plugin_id = ? AND key = ?")
+    .get(pluginId, key) as { value: string } | null;
   return row?.value ?? null;
 }
 
@@ -599,15 +611,13 @@ export function pluginStorageSet(pluginId: string, key: string, value: string):
 
 export function pluginStorageDelete(pluginId: string, key: string): void {
   const db = getDatabase();
-  db.prepare(
-    'DELETE FROM plugin_storage WHERE plugin_id = ? AND key = ?',
-  ).run(pluginId, key);
+  db.prepare("DELETE FROM plugin_storage WHERE plugin_id = ? AND key = ?").run(pluginId, key);
 }
 
 export function pluginStorageHas(pluginId: string, key: string): boolean {
   const db = getDatabase();
-  const row = db.prepare(
-    'SELECT 1 FROM plugin_storage WHERE plugin_id = ? AND key = ?',
-  ).get(pluginId, key);
+  const row = db
+    .prepare("SELECT 1 FROM plugin_storage WHERE plugin_id = ? AND key = ?")
+    .get(pluginId, key);
   return row !== null && row !== undefined;
 }

package/src/storage/db.ts

@@ -1,7 +1,7 @@
-import { Database } from 'bun:sqlite';
-import * as fs from 'fs/promises';
-import { PATHS } from './paths.ts';
-import { applyMigrations } from './migrations/index.ts';
+import { Database } from "bun:sqlite";
+import * as fs from "fs/promises";
+import { applyMigrations } from "./migrations/index.ts";
+import { PATHS } from "./paths.ts";
 
 let db: Database | null = null;
 let currentAppRunId: number | null = null;
@@ -13,10 +13,10 @@ export async function initDatabase(): Promise<Database> {
 
   db = new Database(PATHS.data.database, { create: true });
 
-  db.exec('PRAGMA journal_mode = WAL');
-  db.exec('PRAGMA synchronous = NORMAL');
-  db.exec('PRAGMA foreign_keys = ON');
-  db.exec('PRAGMA busy_timeout = 2000');
+  db.exec("PRAGMA journal_mode = WAL");
+  db.exec("PRAGMA synchronous = NORMAL");
+  db.exec("PRAGMA foreign_keys = ON");
+  db.exec("PRAGMA busy_timeout = 2000");
 
   applyMigrations(db);
 
@@ -27,9 +27,9 @@ export async function initDatabase(): Promise<Database> {
 
   const result = stmt.run(
     Date.now(),
-    process.env.npm_package_version ?? '0.0.0',
+    process.env.npm_package_version ?? "0.0.0",
     process.pid,
-    require('os').hostname()
+    require("os").hostname(),
   );
 
   currentAppRunId = Number(result.lastInsertRowid);
@@ -39,7 +39,7 @@ export async function initDatabase(): Promise<Database> {
 
 export function getDatabase(): Database {
   if (!db) {
-    throw new Error('Database not initialized. Call initDatabase() first.');
+    throw new Error("Database not initialized. Call initDatabase() first.");
   }
   return db;
 }
@@ -50,7 +50,7 @@ export function getAppRunId(): number | null {
 
 export function closeDatabase(): void {
   if (db && currentAppRunId) {
-    db.prepare('UPDATE app_runs SET ended_at = ? WHERE id = ?').run(Date.now(), currentAppRunId);
+    db.prepare("UPDATE app_runs SET ended_at = ? WHERE id = ?").run(Date.now(), currentAppRunId);
   }
   if (db) {
     db.close();

package/src/storage/index.ts

@@ -1,66 +1,65 @@
-export { PATHS } from './paths.ts';
-
 export {
-  initDatabase,
-  getDatabase,
   closeDatabase,
-  isDatabaseInitialized,
   getAppRunId,
-} from './db.ts';
-
+  getDatabase,
+  initDatabase,
+  isDatabaseInitialized,
+} from "./db.ts";
+export { PATHS } from "./paths.ts";
+export type { LatestStreamTotals } from "./repos/agentSessions.ts";
 export {
+  getAgentSession,
+  getLatestStreamTotalsForAllSessions,
+  getRecentSessions,
+  getSessionsByProject,
+  insertAgentSessionSnapshot,
+  upsertAgentSession,
+} from "./repos/agentSessions.ts";
+export {
+  getLatestProviderSnapshot,
   insertProviderSnapshot,
   insertProviderSnapshotBatch,
   queryProviderSnapshots,
-  getLatestProviderSnapshot,
-} from './repos/providerSnapshots.ts';
-
+} from "./repos/providerSnapshots.ts";
+export type {
+  ModelDailyCost,
+  ProjectDailyCost,
+  ProviderDailyCost,
+  SessionActivityPoint,
+} from "./repos/usageEvents.ts";
 export {
+  calculateBurnRate,
+  getSessionActivityTimeline,
+  getTotalUsageInWindow,
   insertUsageEvent,
   insertUsageEventBatch,
-  queryUsageTimeSeries,
-  queryProviderDailyCosts,
   queryModelDailyCosts,
   queryProjectDailyCosts,
-  calculateBurnRate,
-  getTotalUsageInWindow,
-  getSessionActivityTimeline,
-} from './repos/usageEvents.ts';
-
-export type { SessionActivityPoint, ProviderDailyCost, ModelDailyCost, ProjectDailyCost } from './repos/usageEvents.ts';
-
-export {
-  upsertAgentSession,
-  insertAgentSessionSnapshot,
-  getAgentSession,
-  getRecentSessions,
-  getSessionsByProject,
-  getLatestStreamTotalsForAllSessions,
-} from './repos/agentSessions.ts';
-
-export type { LatestStreamTotals } from './repos/agentSessions.ts';
+  queryProviderDailyCosts,
+  queryUsageTimeSeries,
+} from "./repos/usageEvents.ts";
 
 export type {
-  UsageEventSource,
-  SessionStatus,
-  CostSource,
-  PricingSource,
-  AppRunRow,
-  ProviderSnapshotRow,
-  ProviderSnapshotInsert,
   AgentSessionDim,
-  AgentSessionUpsert,
-  AgentSessionSnapshotRow,
   AgentSessionSnapshotInsert,
+  AgentSessionSnapshotRow,
   AgentSessionStreamSnapshotRow,
-  UsageEventRow,
-  UsageEventInsert,
-  HourlyAggregateRow,
+  AgentSessionUpsert,
+  AppRunRow,
+  CostSource,
   DailyAggregateRow,
-  TimeSeriesPoint,
+  HourlyAggregateRow,
+  PricingSource,
+  ProviderSnapshotInsert,
+  ProviderSnapshotRow,
+  SessionStatus,
+  StreamTotals,
   TimeSeriesFilters,
+  TimeSeriesPoint,
+  UsageEventInsert,
+  UsageEventRow,
+  UsageEventSource,
   UsageQueryOptions,
-  StreamTotals,
-} from './types.ts';
+} from "./types.ts";
 
-export { computeStreamDelta } from './types.ts';
+export { computeStreamDelta } from "./types.ts";

package/src/storage/migrations/index.ts

@@ -1,4 +1,4 @@
-import type { Database } from 'bun:sqlite';
+import type { Database } from "bun:sqlite";
 
 export interface Migration {
   version: number;
@@ -167,7 +167,7 @@ export const migrations: Migration[] = [
 ];
 
 export function getCurrentVersion(db: Database): number {
-  const result = db.prepare('PRAGMA user_version').get() as { user_version: number } | null;
+  const result = db.prepare("PRAGMA user_version").get() as { user_version: number } | null;
   return result?.user_version ?? 0;
 }
 

package/src/storage/paths.ts

@@ -1,24 +1,24 @@
-import * as path from 'path';
-import * as os from 'os';
+import * as os from "os";
+import * as path from "path";
 
 const HOME = os.homedir();
 
 export const PATHS = {
   config: {
-    dir: path.join(HOME, '.config/tokentop'),
-    file: path.join(HOME, '.config/tokentop/config.json'),
-    fileJsonc: path.join(HOME, '.config/tokentop/config.jsonc'),
-    plugins: path.join(HOME, '.config/tokentop/plugins'),
+    dir: path.join(HOME, ".config/tokentop"),
+    file: path.join(HOME, ".config/tokentop/config.json"),
+    fileJsonc: path.join(HOME, ".config/tokentop/config.jsonc"),
+    plugins: path.join(HOME, ".config/tokentop/plugins"),
   },
   data: {
-    dir: path.join(HOME, '.local/share/tokentop'),
-    database: path.join(HOME, '.local/share/tokentop/usage.db'),
-    sessions: path.join(HOME, '.local/share/tokentop/sessions'),
-    cache: path.join(HOME, '.local/share/tokentop/cache'),
-    logs: path.join(HOME, '.local/share/tokentop/logs'),
+    dir: path.join(HOME, ".local/share/tokentop"),
+    database: path.join(HOME, ".local/share/tokentop/usage.db"),
+    sessions: path.join(HOME, ".local/share/tokentop/sessions"),
+    cache: path.join(HOME, ".local/share/tokentop/cache"),
+    logs: path.join(HOME, ".local/share/tokentop/logs"),
   },
   cache: {
-    dir: path.join(HOME, '.cache/tokentop'),
-    nodeModules: path.join(HOME, '.cache/tokentop/node_modules'),
+    dir: path.join(HOME, ".cache/tokentop"),
+    nodeModules: path.join(HOME, ".cache/tokentop/node_modules"),
   },
 } as const;