@diogonzafe/tokenwatch 0.1.17 → 0.2.1

package/dist/adapters.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/adapters/postgres.ts","../src/adapters/mysql.ts","../src/adapters/mongodb.ts"],"sourcesContent":["import type { IStorage, UsageEntry } from '../types/index.js'\n\n/**\n * IStorage adapter for PostgreSQL using the `pg` driver.\n *\n * Install peer dep:  npm install pg\n * Types (optional):  npm install -D @types/pg\n *\n * @example\n * ```ts\n * import { Pool } from 'pg'\n * import { createTracker } from '@diogonzafe/tokenwatch'\n * import { PostgresStorage } from '@diogonzafe/tokenwatch/adapters'\n *\n * const pool = new Pool({ connectionString: process.env.DATABASE_URL })\n * const storage = new PostgresStorage(pool)\n * await storage.migrate()          // create table if it doesn't exist\n *\n * const tracker = createTracker({ storage })\n * ```\n */\n\n// Minimal structural types so the adapter compiles without `pg` installed\ninterface QueryClient {\n  query(sql: string, values?: unknown[]): Promise<{ rows: unknown[] }>\n}\n\nexport class PostgresStorage implements IStorage {\n  constructor(private readonly client: QueryClient) {}\n\n  /** Creates the `tokenwatch_usage` table if it does not already exist. */\n  async migrate(): Promise<void> {\n    await this.client.query(`\n      CREATE TABLE IF NOT EXISTS tokenwatch_usage (\n        id            BIGSERIAL PRIMARY KEY,\n        model         TEXT      NOT NULL,\n        input_tokens  INTEGER   NOT NULL,\n        output_tokens INTEGER   NOT NULL,\n        cost_usd      NUMERIC   NOT NULL,\n        session_id    TEXT,\n        user_id       TEXT,\n        timestamp     TIMESTAMPTZ NOT NULL\n      )\n    `)\n  }\n\n  record(entry: UsageEntry): void {\n    this.client\n      .query(\n        `INSERT INTO tokenwatch_usage\n         (model, input_tokens, output_tokens, cost_usd, session_id, user_id, timestamp)\n         VALUES ($1, $2, $3, $4, $5, $6, $7)`,\n        [\n          entry.model,\n          entry.inputTokens,\n          entry.outputTokens,\n          entry.costUSD,\n          entry.sessionId ?? null,\n          entry.userId ?? null,\n          entry.timestamp,\n        ],\n      )\n      .catch((err: unknown) => {\n        console.warn('[tokenwatch] PostgresStorage.record failed:', err)\n      })\n  }\n\n  async getAll(): Promise<UsageEntry[]> {\n    const result = await this.client.query(\n      'SELECT * FROM tokenwatch_usage ORDER BY timestamp ASC',\n    )\n    return (result.rows as Array<Record<string, unknown>>).map(rowToEntry)\n  }\n\n  async clearAll(): Promise<void> {\n    await this.client.query('DELETE FROM tokenwatch_usage')\n  }\n\n  async clearSession(sessionId: string): Promise<void> {\n    await this.client.query(\n      'DELETE FROM tokenwatch_usage WHERE session_id = $1',\n      [sessionId],\n    )\n  }\n}\n\nfunction rowToEntry(r: Record<string, unknown>): UsageEntry {\n  return {\n    model: r['model'] as string,\n    inputTokens: r['input_tokens'] as number,\n    outputTokens: r['output_tokens'] as number,\n    costUSD: Number(r['cost_usd']),\n    ...(r['session_id'] != null && { sessionId: r['session_id'] as string }),\n    ...(r['user_id'] != null && { userId: r['user_id'] as string }),\n    timestamp:\n      r['timestamp'] instanceof Date\n        ? (r['timestamp'] as Date).toISOString()\n        : (r['timestamp'] as string),\n  }\n}\n","import type { IStorage, UsageEntry } from '../types/index.js'\n\n/**\n * IStorage adapter for MySQL / MariaDB using the `mysql2` driver.\n *\n * Install peer dep:  npm install mysql2\n *\n * @example\n * ```ts\n * import mysql from 'mysql2/promise'\n * import { createTracker } from '@diogonzafe/tokenwatch'\n * import { MySQLStorage } from '@diogonzafe/tokenwatch/adapters'\n *\n * const pool = mysql.createPool({ uri: process.env.MYSQL_URL })\n * const storage = new MySQLStorage(pool)\n * await storage.migrate()          // create table if it doesn't exist\n *\n * const tracker = createTracker({ storage })\n * ```\n */\n\n// Minimal structural type so the adapter compiles without `mysql2` installed\ninterface QueryClient {\n  execute(sql: string, values?: unknown[]): Promise<[unknown]>\n}\n\nexport class MySQLStorage implements IStorage {\n  constructor(private readonly client: QueryClient) {}\n\n  /** Creates the `tokenwatch_usage` table if it does not already exist. */\n  async migrate(): Promise<void> {\n    await this.client.execute(`\n      CREATE TABLE IF NOT EXISTS tokenwatch_usage (\n        id            BIGINT UNSIGNED AUTO_INCREMENT PRIMARY KEY,\n        model         VARCHAR(255) NOT NULL,\n        input_tokens  INT          NOT NULL,\n        output_tokens INT          NOT NULL,\n        cost_usd      DECIMAL(18,8) NOT NULL,\n        session_id    VARCHAR(255),\n        user_id       VARCHAR(255),\n        timestamp     DATETIME(3)  NOT NULL\n      )\n    `)\n  }\n\n  record(entry: UsageEntry): void {\n    this.client\n      .execute(\n        `INSERT INTO tokenwatch_usage\n         (model, input_tokens, output_tokens, cost_usd, session_id, user_id, timestamp)\n         VALUES (?, ?, ?, ?, ?, ?, ?)`,\n        [\n          entry.model,\n          entry.inputTokens,\n          entry.outputTokens,\n          entry.costUSD,\n          entry.sessionId ?? null,\n          entry.userId ?? null,\n          entry.timestamp,\n        ],\n      )\n      .catch((err: unknown) => {\n        console.warn('[tokenwatch] MySQLStorage.record failed:', err)\n      })\n  }\n\n  async getAll(): Promise<UsageEntry[]> {\n    const [rows] = await this.client.execute(\n      'SELECT * FROM tokenwatch_usage ORDER BY timestamp ASC',\n    )\n    return (rows as Array<Record<string, unknown>>).map(rowToEntry)\n  }\n\n  async clearAll(): Promise<void> {\n    await this.client.execute('DELETE FROM tokenwatch_usage')\n  }\n\n  async clearSession(sessionId: string): Promise<void> {\n    await this.client.execute(\n      'DELETE FROM tokenwatch_usage WHERE session_id = ?',\n      [sessionId],\n    )\n  }\n}\n\nfunction rowToEntry(r: Record<string, unknown>): UsageEntry {\n  return {\n    model: r['model'] as string,\n    inputTokens: r['input_tokens'] as number,\n    outputTokens: r['output_tokens'] as number,\n    costUSD: Number(r['cost_usd']),\n    ...(r['session_id'] != null && { sessionId: r['session_id'] as string }),\n    ...(r['user_id'] != null && { userId: r['user_id'] as string }),\n    timestamp:\n      r['timestamp'] instanceof Date\n        ? (r['timestamp'] as Date).toISOString()\n        : (r['timestamp'] as string),\n  }\n}\n","import type { IStorage, UsageEntry } from '../types/index.js'\n\n/**\n * IStorage adapter for MongoDB using the official `mongodb` driver.\n *\n * Install peer dep:  npm install mongodb\n *\n * @example\n * ```ts\n * import { MongoClient } from 'mongodb'\n * import { createTracker } from '@diogonzafe/tokenwatch'\n * import { MongoStorage } from '@diogonzafe/tokenwatch/adapters'\n *\n * const client = new MongoClient(process.env.MONGO_URL!)\n * await client.connect()\n *\n * const storage = new MongoStorage(client.db('myapp'))\n * const tracker = createTracker({ storage })\n * ```\n *\n * Recommended index (run once at startup):\n * ```ts\n * await storage.createIndexes()\n * ```\n */\n\n// Minimal structural types so the adapter compiles without `mongodb` installed\ninterface MongoDocument {\n  _id?: unknown\n  model: string\n  inputTokens: number\n  outputTokens: number\n  costUSD: number\n  sessionId?: string | null\n  userId?: string | null\n  timestamp: string\n}\n\ninterface MongoCursor {\n  sort(sort: Record<string, unknown>): MongoCursor\n  toArray(): Promise<MongoDocument[]>\n}\n\ninterface Collection {\n  insertOne(doc: MongoDocument): Promise<unknown>\n  find(filter: Record<string, unknown>): MongoCursor\n  deleteMany(filter: Record<string, unknown>): Promise<unknown>\n  createIndex(index: Record<string, unknown>): Promise<unknown>\n}\n\ninterface Database {\n  collection(name: string): Collection\n}\n\nconst COLLECTION = 'tokenwatch_usage'\n\nexport class MongoStorage implements IStorage {\n  private readonly col: Collection\n\n  constructor(db: Database) {\n    this.col = db.collection(COLLECTION)\n  }\n\n  /** Creates recommended indexes for query performance. Call once at startup. */\n  async createIndexes(): Promise<void> {\n    await this.col.createIndex({ timestamp: 1 })\n    await this.col.createIndex({ sessionId: 1 })\n    await this.col.createIndex({ userId: 1 })\n    await this.col.createIndex({ model: 1 })\n  }\n\n  record(entry: UsageEntry): void {\n    this.col\n      .insertOne({\n        model: entry.model,\n        inputTokens: entry.inputTokens,\n        outputTokens: entry.outputTokens,\n        costUSD: entry.costUSD,\n        sessionId: entry.sessionId ?? null,\n        userId: entry.userId ?? null,\n        timestamp: entry.timestamp,\n      })\n      .catch((err: unknown) => {\n        console.warn('[tokenwatch] MongoStorage.record failed:', err)\n      })\n  }\n\n  async getAll(): Promise<UsageEntry[]> {\n    const docs = await this.col.find({}).sort({ timestamp: 1 }).toArray()\n    return docs.map(docToEntry)\n  }\n\n  async clearAll(): Promise<void> {\n    await this.col.deleteMany({})\n  }\n\n  async clearSession(sessionId: string): Promise<void> {\n    await this.col.deleteMany({ sessionId })\n  }\n}\n\nfunction docToEntry(doc: MongoDocument): UsageEntry {\n  return {\n    model: doc.model,\n    inputTokens: doc.inputTokens,\n    outputTokens: doc.outputTokens,\n    costUSD: doc.costUSD,\n    ...(doc.sessionId != null && { sessionId: doc.sessionId }),\n    ...(doc.userId != null && { userId: doc.userId }),\n    timestamp: doc.timestamp,\n  }\n}\n"],"mappings":";AA2BO,IAAM,kBAAN,MAA0C;AAAA,EAC/C,YAA6B,QAAqB;AAArB;AAAA,EAAsB;AAAA,EAAtB;AAAA;AAAA,EAG7B,MAAM,UAAyB;AAC7B,UAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAWvB;AAAA,EACH;AAAA,EAEA,OAAO,OAAyB;AAC9B,SAAK,OACF;AAAA,MACC;AAAA;AAAA;AAAA,MAGA;AAAA,QACE,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM,aAAa;AAAA,QACnB,MAAM,UAAU;AAAA,QAChB,MAAM;AAAA,MACR;AAAA,IACF,EACC,MAAM,CAAC,QAAiB;AACvB,cAAQ,KAAK,+CAA+C,GAAG;AAAA,IACjE,CAAC;AAAA,EACL;AAAA,EAEA,MAAM,SAAgC;AACpC,UAAM,SAAS,MAAM,KAAK,OAAO;AAAA,MAC/B;AAAA,IACF;AACA,WAAQ,OAAO,KAAwC,IAAI,UAAU;AAAA,EACvE;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,KAAK,OAAO,MAAM,8BAA8B;AAAA,EACxD;AAAA,EAEA,MAAM,aAAa,WAAkC;AACnD,UAAM,KAAK,OAAO;AAAA,MAChB;AAAA,MACA,CAAC,SAAS;AAAA,IACZ;AAAA,EACF;AACF;AAEA,SAAS,WAAW,GAAwC;AAC1D,SAAO;AAAA,IACL,OAAO,EAAE,OAAO;AAAA,IAChB,aAAa,EAAE,cAAc;AAAA,IAC7B,cAAc,EAAE,eAAe;AAAA,IAC/B,SAAS,OAAO,EAAE,UAAU,CAAC;AAAA,IAC7B,GAAI,EAAE,YAAY,KAAK,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAY;AAAA,IACtE,GAAI,EAAE,SAAS,KAAK,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAY;AAAA,IAC7D,WACE,EAAE,WAAW,aAAa,OACrB,EAAE,WAAW,EAAW,YAAY,IACpC,EAAE,WAAW;AAAA,EACtB;AACF;;;ACzEO,IAAM,eAAN,MAAuC;AAAA,EAC5C,YAA6B,QAAqB;AAArB;AAAA,EAAsB;AAAA,EAAtB;AAAA;AAAA,EAG7B,MAAM,UAAyB;AAC7B,UAAM,KAAK,OAAO,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAWzB;AAAA,EACH;AAAA,EAEA,OAAO,OAAyB;AAC9B,SAAK,OACF;AAAA,MACC;AAAA;AAAA;AAAA,MAGA;AAAA,QACE,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM,aAAa;AAAA,QACnB,MAAM,UAAU;AAAA,QAChB,MAAM;AAAA,MACR;AAAA,IACF,EACC,MAAM,CAAC,QAAiB;AACvB,cAAQ,KAAK,4CAA4C,GAAG;AAAA,IAC9D,CAAC;AAAA,EACL;AAAA,EAEA,MAAM,SAAgC;AACpC,UAAM,CAAC,IAAI,IAAI,MAAM,KAAK,OAAO;AAAA,MAC/B;AAAA,IACF;AACA,WAAQ,KAAwC,IAAIA,WAAU;AAAA,EAChE;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,KAAK,OAAO,QAAQ,8BAA8B;AAAA,EAC1D;AAAA,EAEA,MAAM,aAAa,WAAkC;AACnD,UAAM,KAAK,OAAO;AAAA,MAChB;AAAA,MACA,CAAC,SAAS;AAAA,IACZ;AAAA,EACF;AACF;AAEA,SAASA,YAAW,GAAwC;AAC1D,SAAO;AAAA,IACL,OAAO,EAAE,OAAO;AAAA,IAChB,aAAa,EAAE,cAAc;AAAA,IAC7B,cAAc,EAAE,eAAe;AAAA,IAC/B,SAAS,OAAO,EAAE,UAAU,CAAC;AAAA,IAC7B,GAAI,EAAE,YAAY,KAAK,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAY;AAAA,IACtE,GAAI,EAAE,SAAS,KAAK,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAY;AAAA,IAC7D,WACE,EAAE,WAAW,aAAa,OACrB,EAAE,WAAW,EAAW,YAAY,IACpC,EAAE,WAAW;AAAA,EACtB;AACF;;;AC5CA,IAAM,aAAa;AAEZ,IAAM,eAAN,MAAuC;AAAA,EAC3B;AAAA,EAEjB,YAAY,IAAc;AACxB,SAAK,MAAM,GAAG,WAAW,UAAU;AAAA,EACrC;AAAA;AAAA,EAGA,MAAM,gBAA+B;AACnC,UAAM,KAAK,IAAI,YAAY,EAAE,WAAW,EAAE,CAAC;AAC3C,UAAM,KAAK,IAAI,YAAY,EAAE,WAAW,EAAE,CAAC;AAC3C,UAAM,KAAK,IAAI,YAAY,EAAE,QAAQ,EAAE,CAAC;AACxC,UAAM,KAAK,IAAI,YAAY,EAAE,OAAO,EAAE,CAAC;AAAA,EACzC;AAAA,EAEA,OAAO,OAAyB;AAC9B,SAAK,IACF,UAAU;AAAA,MACT,OAAO,MAAM;AAAA,MACb,aAAa,MAAM;AAAA,MACnB,cAAc,MAAM;AAAA,MACpB,SAAS,MAAM;AAAA,MACf,WAAW,MAAM,aAAa;AAAA,MAC9B,QAAQ,MAAM,UAAU;AAAA,MACxB,WAAW,MAAM;AAAA,IACnB,CAAC,EACA,MAAM,CAAC,QAAiB;AACvB,cAAQ,KAAK,4CAA4C,GAAG;AAAA,IAC9D,CAAC;AAAA,EACL;AAAA,EAEA,MAAM,SAAgC;AACpC,UAAM,OAAO,MAAM,KAAK,IAAI,KAAK,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,QAAQ;AACpE,WAAO,KAAK,IAAI,UAAU;AAAA,EAC5B;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,KAAK,IAAI,WAAW,CAAC,CAAC;AAAA,EAC9B;AAAA,EAEA,MAAM,aAAa,WAAkC;AACnD,UAAM,KAAK,IAAI,WAAW,EAAE,UAAU,CAAC;AAAA,EACzC;AACF;AAEA,SAAS,WAAW,KAAgC;AAClD,SAAO;AAAA,IACL,OAAO,IAAI;AAAA,IACX,aAAa,IAAI;AAAA,IACjB,cAAc,IAAI;AAAA,IAClB,SAAS,IAAI;AAAA,IACb,GAAI,IAAI,aAAa,QAAQ,EAAE,WAAW,IAAI,UAAU;AAAA,IACxD,GAAI,IAAI,UAAU,QAAQ,EAAE,QAAQ,IAAI,OAAO;AAAA,IAC/C,WAAW,IAAI;AAAA,EACjB;AACF;","names":["rowToEntry"]}
+{"version":3,"sources":["../src/adapters/postgres.ts","../src/adapters/mysql.ts","../src/adapters/mongodb.ts"],"sourcesContent":["import type { IStorage, UsageEntry } from '../types/index.js'\n\n/**\n * IStorage adapter for PostgreSQL using the `pg` driver.\n *\n * Install peer dep:  npm install pg\n * Types (optional):  npm install -D @types/pg\n *\n * @example\n * ```ts\n * import { Pool } from 'pg'\n * import { createTracker } from '@diogonzafe/tokenwatch'\n * import { PostgresStorage } from '@diogonzafe/tokenwatch/adapters'\n *\n * const pool = new Pool({ connectionString: process.env.DATABASE_URL })\n * const storage = new PostgresStorage(pool)\n * await storage.migrate()          // create table if it doesn't exist\n *\n * const tracker = createTracker({ storage })\n * ```\n */\n\n// Minimal structural types so the adapter compiles without `pg` installed\ninterface QueryClient {\n  query(sql: string, values?: unknown[]): Promise<{ rows: unknown[] }>\n}\n\nexport class PostgresStorage implements IStorage {\n  constructor(private readonly client: QueryClient) {}\n\n  /** Creates the `tokenwatch_usage` table if it does not already exist.\n   *  Also adds new columns for databases created before v0.2.0. */\n  async migrate(): Promise<void> {\n    await this.client.query(`\n      CREATE TABLE IF NOT EXISTS tokenwatch_usage (\n        id               BIGSERIAL PRIMARY KEY,\n        model            TEXT        NOT NULL,\n        input_tokens     INTEGER     NOT NULL,\n        output_tokens    INTEGER     NOT NULL,\n        reasoning_tokens INTEGER     NOT NULL DEFAULT 0,\n        cost_usd         NUMERIC     NOT NULL,\n        session_id       TEXT,\n        user_id          TEXT,\n        feature          TEXT,\n        timestamp        TIMESTAMPTZ NOT NULL\n      )\n    `)\n    // Incremental migrations for databases created before v0.2.0\n    await this.client.query(`\n      ALTER TABLE tokenwatch_usage ADD COLUMN IF NOT EXISTS reasoning_tokens INTEGER NOT NULL DEFAULT 0\n    `).catch(() => { /* column already exists in older Postgres versions that don't support IF NOT EXISTS */ })\n    await this.client.query(`\n      ALTER TABLE tokenwatch_usage ADD COLUMN IF NOT EXISTS feature TEXT\n    `).catch(() => { /* same */ })\n  }\n\n  record(entry: UsageEntry): void {\n    this.client\n      .query(\n        `INSERT INTO tokenwatch_usage\n         (model, input_tokens, output_tokens, reasoning_tokens, cost_usd, session_id, user_id, feature, timestamp)\n         VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)`,\n        [\n          entry.model,\n          entry.inputTokens,\n          entry.outputTokens,\n          entry.reasoningTokens ?? 0,\n          entry.costUSD,\n          entry.sessionId ?? null,\n          entry.userId ?? null,\n          entry.feature ?? null,\n          entry.timestamp,\n        ],\n      )\n      .catch((err: unknown) => {\n        console.warn('[tokenwatch] PostgresStorage.record failed:', err)\n      })\n  }\n\n  async getAll(): Promise<UsageEntry[]> {\n    const result = await this.client.query(\n      'SELECT * FROM tokenwatch_usage ORDER BY timestamp ASC',\n    )\n    return (result.rows as Array<Record<string, unknown>>).map(rowToEntry)\n  }\n\n  async clearAll(): Promise<void> {\n    await this.client.query('DELETE FROM tokenwatch_usage')\n  }\n\n  async clearSession(sessionId: string): Promise<void> {\n    await this.client.query(\n      'DELETE FROM tokenwatch_usage WHERE session_id = $1',\n      [sessionId],\n    )\n  }\n}\n\nfunction rowToEntry(r: Record<string, unknown>): UsageEntry {\n  const reasoningTokens = (r['reasoning_tokens'] as number | null) ?? 0\n  return {\n    model: r['model'] as string,\n    inputTokens: r['input_tokens'] as number,\n    outputTokens: r['output_tokens'] as number,\n    ...(reasoningTokens > 0 && { reasoningTokens }),\n    costUSD: Number(r['cost_usd']),\n    ...(r['session_id'] != null && { sessionId: r['session_id'] as string }),\n    ...(r['user_id'] != null && { userId: r['user_id'] as string }),\n    ...(r['feature'] != null && { feature: r['feature'] as string }),\n    timestamp:\n      r['timestamp'] instanceof Date\n        ? (r['timestamp'] as Date).toISOString()\n        : (r['timestamp'] as string),\n  }\n}\n","import type { IStorage, UsageEntry } from '../types/index.js'\n\n/**\n * IStorage adapter for MySQL / MariaDB using the `mysql2` driver.\n *\n * Install peer dep:  npm install mysql2\n *\n * @example\n * ```ts\n * import mysql from 'mysql2/promise'\n * import { createTracker } from '@diogonzafe/tokenwatch'\n * import { MySQLStorage } from '@diogonzafe/tokenwatch/adapters'\n *\n * const pool = mysql.createPool({ uri: process.env.MYSQL_URL })\n * const storage = new MySQLStorage(pool)\n * await storage.migrate()          // create table if it doesn't exist\n *\n * const tracker = createTracker({ storage })\n * ```\n */\n\n// Minimal structural type so the adapter compiles without `mysql2` installed\ninterface QueryClient {\n  execute(sql: string, values?: unknown[]): Promise<[unknown]>\n}\n\nexport class MySQLStorage implements IStorage {\n  constructor(private readonly client: QueryClient) {}\n\n  /** Creates the `tokenwatch_usage` table if it does not already exist.\n   *  Also adds new columns for databases created before v0.2.0. */\n  async migrate(): Promise<void> {\n    await this.client.execute(`\n      CREATE TABLE IF NOT EXISTS tokenwatch_usage (\n        id               BIGINT UNSIGNED AUTO_INCREMENT PRIMARY KEY,\n        model            VARCHAR(255)  NOT NULL,\n        input_tokens     INT           NOT NULL,\n        output_tokens    INT           NOT NULL,\n        reasoning_tokens INT           NOT NULL DEFAULT 0,\n        cost_usd         DECIMAL(18,8) NOT NULL,\n        session_id       VARCHAR(255),\n        user_id          VARCHAR(255),\n        feature          VARCHAR(255),\n        timestamp        DATETIME(3)   NOT NULL\n      )\n    `)\n    // Incremental migrations for databases created before v0.2.0\n    await this.client.execute(`\n      ALTER TABLE tokenwatch_usage\n        ADD COLUMN IF NOT EXISTS reasoning_tokens INT NOT NULL DEFAULT 0,\n        ADD COLUMN IF NOT EXISTS feature VARCHAR(255)\n    `).catch(() => { /* MySQL < 8.0 may not support IF NOT EXISTS — ignore if columns already exist */ })\n  }\n\n  record(entry: UsageEntry): void {\n    this.client\n      .execute(\n        `INSERT INTO tokenwatch_usage\n         (model, input_tokens, output_tokens, reasoning_tokens, cost_usd, session_id, user_id, feature, timestamp)\n         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`,\n        [\n          entry.model,\n          entry.inputTokens,\n          entry.outputTokens,\n          entry.reasoningTokens ?? 0,\n          entry.costUSD,\n          entry.sessionId ?? null,\n          entry.userId ?? null,\n          entry.feature ?? null,\n          entry.timestamp,\n        ],\n      )\n      .catch((err: unknown) => {\n        console.warn('[tokenwatch] MySQLStorage.record failed:', err)\n      })\n  }\n\n  async getAll(): Promise<UsageEntry[]> {\n    const [rows] = await this.client.execute(\n      'SELECT * FROM tokenwatch_usage ORDER BY timestamp ASC',\n    )\n    return (rows as Array<Record<string, unknown>>).map(rowToEntry)\n  }\n\n  async clearAll(): Promise<void> {\n    await this.client.execute('DELETE FROM tokenwatch_usage')\n  }\n\n  async clearSession(sessionId: string): Promise<void> {\n    await this.client.execute(\n      'DELETE FROM tokenwatch_usage WHERE session_id = ?',\n      [sessionId],\n    )\n  }\n}\n\nfunction rowToEntry(r: Record<string, unknown>): UsageEntry {\n  const reasoningTokens = (r['reasoning_tokens'] as number | null) ?? 0\n  return {\n    model: r['model'] as string,\n    inputTokens: r['input_tokens'] as number,\n    outputTokens: r['output_tokens'] as number,\n    ...(reasoningTokens > 0 && { reasoningTokens }),\n    costUSD: Number(r['cost_usd']),\n    ...(r['session_id'] != null && { sessionId: r['session_id'] as string }),\n    ...(r['user_id'] != null && { userId: r['user_id'] as string }),\n    ...(r['feature'] != null && { feature: r['feature'] as string }),\n    timestamp:\n      r['timestamp'] instanceof Date\n        ? (r['timestamp'] as Date).toISOString()\n        : (r['timestamp'] as string),\n  }\n}\n","import type { IStorage, UsageEntry } from '../types/index.js'\n\n/**\n * IStorage adapter for MongoDB using the official `mongodb` driver.\n *\n * Install peer dep:  npm install mongodb\n *\n * @example\n * ```ts\n * import { MongoClient } from 'mongodb'\n * import { createTracker } from '@diogonzafe/tokenwatch'\n * import { MongoStorage } from '@diogonzafe/tokenwatch/adapters'\n *\n * const client = new MongoClient(process.env.MONGO_URL!)\n * await client.connect()\n *\n * const storage = new MongoStorage(client.db('myapp'))\n * const tracker = createTracker({ storage })\n * ```\n *\n * Recommended index (run once at startup):\n * ```ts\n * await storage.createIndexes()\n * ```\n */\n\n// Minimal structural types so the adapter compiles without `mongodb` installed\ninterface MongoDocument {\n  _id?: unknown\n  model: string\n  inputTokens: number\n  outputTokens: number\n  reasoningTokens?: number\n  costUSD: number\n  sessionId?: string | null\n  userId?: string | null\n  feature?: string | null\n  timestamp: string\n}\n\ninterface MongoCursor {\n  sort(sort: Record<string, unknown>): MongoCursor\n  toArray(): Promise<MongoDocument[]>\n}\n\ninterface Collection {\n  insertOne(doc: MongoDocument): Promise<unknown>\n  find(filter: Record<string, unknown>): MongoCursor\n  deleteMany(filter: Record<string, unknown>): Promise<unknown>\n  createIndex(index: Record<string, unknown>): Promise<unknown>\n}\n\ninterface Database {\n  collection(name: string): Collection\n}\n\nconst COLLECTION = 'tokenwatch_usage'\n\nexport class MongoStorage implements IStorage {\n  private readonly col: Collection\n\n  constructor(db: Database) {\n    this.col = db.collection(COLLECTION)\n  }\n\n  /** Creates recommended indexes for query performance. Call once at startup. */\n  async createIndexes(): Promise<void> {\n    await this.col.createIndex({ timestamp: 1 })\n    await this.col.createIndex({ sessionId: 1 })\n    await this.col.createIndex({ userId: 1 })\n    await this.col.createIndex({ model: 1 })\n  }\n\n  record(entry: UsageEntry): void {\n    this.col\n      .insertOne({\n        model: entry.model,\n        inputTokens: entry.inputTokens,\n        outputTokens: entry.outputTokens,\n        ...(entry.reasoningTokens !== undefined && { reasoningTokens: entry.reasoningTokens }),\n        costUSD: entry.costUSD,\n        sessionId: entry.sessionId ?? null,\n        userId: entry.userId ?? null,\n        ...(entry.feature !== undefined && { feature: entry.feature }),\n        timestamp: entry.timestamp,\n      })\n      .catch((err: unknown) => {\n        console.warn('[tokenwatch] MongoStorage.record failed:', err)\n      })\n  }\n\n  async getAll(): Promise<UsageEntry[]> {\n    const docs = await this.col.find({}).sort({ timestamp: 1 }).toArray()\n    return docs.map(docToEntry)\n  }\n\n  async clearAll(): Promise<void> {\n    await this.col.deleteMany({})\n  }\n\n  async clearSession(sessionId: string): Promise<void> {\n    await this.col.deleteMany({ sessionId })\n  }\n}\n\nfunction docToEntry(doc: MongoDocument): UsageEntry {\n  return {\n    model: doc.model,\n    inputTokens: doc.inputTokens,\n    outputTokens: doc.outputTokens,\n    ...(doc.reasoningTokens != null && doc.reasoningTokens > 0 && { reasoningTokens: doc.reasoningTokens }),\n    costUSD: doc.costUSD,\n    ...(doc.sessionId != null && { sessionId: doc.sessionId }),\n    ...(doc.userId != null && { userId: doc.userId }),\n    ...(doc.feature != null && { feature: doc.feature }),\n    timestamp: doc.timestamp,\n  }\n}\n"],"mappings":";AA2BO,IAAM,kBAAN,MAA0C;AAAA,EAC/C,YAA6B,QAAqB;AAArB;AAAA,EAAsB;AAAA,EAAtB;AAAA;AAAA;AAAA,EAI7B,MAAM,UAAyB;AAC7B,UAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAavB;AAED,UAAM,KAAK,OAAO,MAAM;AAAA;AAAA,KAEvB,EAAE,MAAM,MAAM;AAAA,IAA0F,CAAC;AAC1G,UAAM,KAAK,OAAO,MAAM;AAAA;AAAA,KAEvB,EAAE,MAAM,MAAM;AAAA,IAAa,CAAC;AAAA,EAC/B;AAAA,EAEA,OAAO,OAAyB;AAC9B,SAAK,OACF;AAAA,MACC;AAAA;AAAA;AAAA,MAGA;AAAA,QACE,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM,mBAAmB;AAAA,QACzB,MAAM;AAAA,QACN,MAAM,aAAa;AAAA,QACnB,MAAM,UAAU;AAAA,QAChB,MAAM,WAAW;AAAA,QACjB,MAAM;AAAA,MACR;AAAA,IACF,EACC,MAAM,CAAC,QAAiB;AACvB,cAAQ,KAAK,+CAA+C,GAAG;AAAA,IACjE,CAAC;AAAA,EACL;AAAA,EAEA,MAAM,SAAgC;AACpC,UAAM,SAAS,MAAM,KAAK,OAAO;AAAA,MAC/B;AAAA,IACF;AACA,WAAQ,OAAO,KAAwC,IAAI,UAAU;AAAA,EACvE;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,KAAK,OAAO,MAAM,8BAA8B;AAAA,EACxD;AAAA,EAEA,MAAM,aAAa,WAAkC;AACnD,UAAM,KAAK,OAAO;AAAA,MAChB;AAAA,MACA,CAAC,SAAS;AAAA,IACZ;AAAA,EACF;AACF;AAEA,SAAS,WAAW,GAAwC;AAC1D,QAAM,kBAAmB,EAAE,kBAAkB,KAAuB;AACpE,SAAO;AAAA,IACL,OAAO,EAAE,OAAO;AAAA,IAChB,aAAa,EAAE,cAAc;AAAA,IAC7B,cAAc,EAAE,eAAe;AAAA,IAC/B,GAAI,kBAAkB,KAAK,EAAE,gBAAgB;AAAA,IAC7C,SAAS,OAAO,EAAE,UAAU,CAAC;AAAA,IAC7B,GAAI,EAAE,YAAY,KAAK,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAY;AAAA,IACtE,GAAI,EAAE,SAAS,KAAK,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAY;AAAA,IAC7D,GAAI,EAAE,SAAS,KAAK,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAY;AAAA,IAC9D,WACE,EAAE,WAAW,aAAa,OACrB,EAAE,WAAW,EAAW,YAAY,IACpC,EAAE,WAAW;AAAA,EACtB;AACF;;;ACxFO,IAAM,eAAN,MAAuC;AAAA,EAC5C,YAA6B,QAAqB;AAArB;AAAA,EAAsB;AAAA,EAAtB;AAAA;AAAA;AAAA,EAI7B,MAAM,UAAyB;AAC7B,UAAM,KAAK,OAAO,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAazB;AAED,UAAM,KAAK,OAAO,QAAQ;AAAA;AAAA;AAAA;AAAA,KAIzB,EAAE,MAAM,MAAM;AAAA,IAAoF,CAAC;AAAA,EACtG;AAAA,EAEA,OAAO,OAAyB;AAC9B,SAAK,OACF;AAAA,MACC;AAAA;AAAA;AAAA,MAGA;AAAA,QACE,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM,mBAAmB;AAAA,QACzB,MAAM;AAAA,QACN,MAAM,aAAa;AAAA,QACnB,MAAM,UAAU;AAAA,QAChB,MAAM,WAAW;AAAA,QACjB,MAAM;AAAA,MACR;AAAA,IACF,EACC,MAAM,CAAC,QAAiB;AACvB,cAAQ,KAAK,4CAA4C,GAAG;AAAA,IAC9D,CAAC;AAAA,EACL;AAAA,EAEA,MAAM,SAAgC;AACpC,UAAM,CAAC,IAAI,IAAI,MAAM,KAAK,OAAO;AAAA,MAC/B;AAAA,IACF;AACA,WAAQ,KAAwC,IAAIA,WAAU;AAAA,EAChE;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,KAAK,OAAO,QAAQ,8BAA8B;AAAA,EAC1D;AAAA,EAEA,MAAM,aAAa,WAAkC;AACnD,UAAM,KAAK,OAAO;AAAA,MAChB;AAAA,MACA,CAAC,SAAS;AAAA,IACZ;AAAA,EACF;AACF;AAEA,SAASA,YAAW,GAAwC;AAC1D,QAAM,kBAAmB,EAAE,kBAAkB,KAAuB;AACpE,SAAO;AAAA,IACL,OAAO,EAAE,OAAO;AAAA,IAChB,aAAa,EAAE,cAAc;AAAA,IAC7B,cAAc,EAAE,eAAe;AAAA,IAC/B,GAAI,kBAAkB,KAAK,EAAE,gBAAgB;AAAA,IAC7C,SAAS,OAAO,EAAE,UAAU,CAAC;AAAA,IAC7B,GAAI,EAAE,YAAY,KAAK,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAY;AAAA,IACtE,GAAI,EAAE,SAAS,KAAK,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAY;AAAA,IAC7D,GAAI,EAAE,SAAS,KAAK,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAY;AAAA,IAC9D,WACE,EAAE,WAAW,aAAa,OACrB,EAAE,WAAW,EAAW,YAAY,IACpC,EAAE,WAAW;AAAA,EACtB;AACF;;;ACxDA,IAAM,aAAa;AAEZ,IAAM,eAAN,MAAuC;AAAA,EAC3B;AAAA,EAEjB,YAAY,IAAc;AACxB,SAAK,MAAM,GAAG,WAAW,UAAU;AAAA,EACrC;AAAA;AAAA,EAGA,MAAM,gBAA+B;AACnC,UAAM,KAAK,IAAI,YAAY,EAAE,WAAW,EAAE,CAAC;AAC3C,UAAM,KAAK,IAAI,YAAY,EAAE,WAAW,EAAE,CAAC;AAC3C,UAAM,KAAK,IAAI,YAAY,EAAE,QAAQ,EAAE,CAAC;AACxC,UAAM,KAAK,IAAI,YAAY,EAAE,OAAO,EAAE,CAAC;AAAA,EACzC;AAAA,EAEA,OAAO,OAAyB;AAC9B,SAAK,IACF,UAAU;AAAA,MACT,OAAO,MAAM;AAAA,MACb,aAAa,MAAM;AAAA,MACnB,cAAc,MAAM;AAAA,MACpB,GAAI,MAAM,oBAAoB,UAAa,EAAE,iBAAiB,MAAM,gBAAgB;AAAA,MACpF,SAAS,MAAM;AAAA,MACf,WAAW,MAAM,aAAa;AAAA,MAC9B,QAAQ,MAAM,UAAU;AAAA,MACxB,GAAI,MAAM,YAAY,UAAa,EAAE,SAAS,MAAM,QAAQ;AAAA,MAC5D,WAAW,MAAM;AAAA,IACnB,CAAC,EACA,MAAM,CAAC,QAAiB;AACvB,cAAQ,KAAK,4CAA4C,GAAG;AAAA,IAC9D,CAAC;AAAA,EACL;AAAA,EAEA,MAAM,SAAgC;AACpC,UAAM,OAAO,MAAM,KAAK,IAAI,KAAK,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,QAAQ;AACpE,WAAO,KAAK,IAAI,UAAU;AAAA,EAC5B;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,KAAK,IAAI,WAAW,CAAC,CAAC;AAAA,EAC9B;AAAA,EAEA,MAAM,aAAa,WAAkC;AACnD,UAAM,KAAK,IAAI,WAAW,EAAE,UAAU,CAAC;AAAA,EACzC;AACF;AAEA,SAAS,WAAW,KAAgC;AAClD,SAAO;AAAA,IACL,OAAO,IAAI;AAAA,IACX,aAAa,IAAI;AAAA,IACjB,cAAc,IAAI;AAAA,IAClB,GAAI,IAAI,mBAAmB,QAAQ,IAAI,kBAAkB,KAAK,EAAE,iBAAiB,IAAI,gBAAgB;AAAA,IACrG,SAAS,IAAI;AAAA,IACb,GAAI,IAAI,aAAa,QAAQ,EAAE,WAAW,IAAI,UAAU;AAAA,IACxD,GAAI,IAAI,UAAU,QAAQ,EAAE,QAAQ,IAAI,OAAO;AAAA,IAC/C,GAAI,IAAI,WAAW,QAAQ,EAAE,SAAS,IAAI,QAAQ;AAAA,IAClD,WAAW,IAAI;AAAA,EACjB;AACF;","names":["rowToEntry"]}

package/dist/cli.js

@@ -98,29 +98,40 @@ var SqliteStorage = class {
   migrate() {
     this.db.exec(`
       CREATE TABLE IF NOT EXISTS usage (
-        id            INTEGER PRIMARY KEY AUTOINCREMENT,
-        model         TEXT    NOT NULL,
-        input_tokens  INTEGER NOT NULL,
-        output_tokens INTEGER NOT NULL,
-        cost_usd      REAL    NOT NULL,
-        session_id    TEXT,
-        user_id       TEXT,
-        timestamp     TEXT    NOT NULL
+        id               INTEGER PRIMARY KEY AUTOINCREMENT,
+        model            TEXT    NOT NULL,
+        input_tokens     INTEGER NOT NULL,
+        output_tokens    INTEGER NOT NULL,
+        reasoning_tokens INTEGER NOT NULL DEFAULT 0,
+        cost_usd         REAL    NOT NULL,
+        session_id       TEXT,
+        user_id          TEXT,
+        feature          TEXT,
+        timestamp        TEXT    NOT NULL
       )
     `);
+    const cols = this.db.prepare(`PRAGMA table_info(usage)`).all().map((c) => c.name);
+    if (!cols.includes("reasoning_tokens")) {
+      this.db.exec(`ALTER TABLE usage ADD COLUMN reasoning_tokens INTEGER NOT NULL DEFAULT 0`);
+    }
+    if (!cols.includes("feature")) {
+      this.db.exec(`ALTER TABLE usage ADD COLUMN feature TEXT`);
+    }
   }
   record(entry) {
     this.db.prepare(
       `INSERT INTO usage
-         (model, input_tokens, output_tokens, cost_usd, session_id, user_id, timestamp)
-         VALUES (?, ?, ?, ?, ?, ?, ?)`
+         (model, input_tokens, output_tokens, reasoning_tokens, cost_usd, session_id, user_id, feature, timestamp)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`
     ).run(
       entry.model,
       entry.inputTokens,
       entry.outputTokens,
+      entry.reasoningTokens ?? 0,
       entry.costUSD,
       entry.sessionId ?? null,
       entry.userId ?? null,
+      entry.feature ?? null,
       entry.timestamp
     );
   }
@@ -130,9 +141,11 @@ var SqliteStorage = class {
       model: r.model,
       inputTokens: r.input_tokens,
       outputTokens: r.output_tokens,
+      ...r.reasoning_tokens > 0 && { reasoningTokens: r.reasoning_tokens },
       costUSD: r.cost_usd,
       ...r.session_id != null && { sessionId: r.session_id },
       ...r.user_id != null && { userId: r.user_id },
+      ...r.feature != null && { feature: r.feature },
       timestamp: r.timestamp
     }));
   }
@@ -1423,6 +1436,7 @@ ${issues}`);
     const byModel = {};
     const bySession = {};
     const byUser = {};
+    const byFeature = {};
     let totalInput = 0;
     let totalOutput = 0;
     let totalCost = 0;
@@ -1432,11 +1446,12 @@ ${issues}`);
       totalOutput += e.outputTokens;
       totalCost += e.costUSD;
       if (e.timestamp > lastTimestamp) lastTimestamp = e.timestamp;
-      const m = byModel[e.model] ??= { costUSD: 0, calls: 0, tokens: { input: 0, output: 0 } };
+      const m = byModel[e.model] ??= { costUSD: 0, calls: 0, tokens: { input: 0, output: 0, reasoning: 0 } };
       m.costUSD += e.costUSD;
       m.calls += 1;
       m.tokens.input += e.inputTokens;
       m.tokens.output += e.outputTokens;
+      m.tokens.reasoning += e.reasoningTokens ?? 0;
       if (e.sessionId) {
         const s = bySession[e.sessionId] ??= { costUSD: 0, calls: 0 };
         s.costUSD += e.costUSD;
@@ -1447,6 +1462,11 @@ ${issues}`);
         u.costUSD += e.costUSD;
         u.calls += 1;
       }
+      if (e.feature) {
+        const f = byFeature[e.feature] ??= { costUSD: 0, calls: 0 };
+        f.costUSD += e.costUSD;
+        f.calls += 1;
+      }
     }
     return {
       totalCostUSD: totalCost,
@@ -1454,6 +1474,7 @@ ${issues}`);
       byModel,
       bySession,
       byUser,
+      byFeature,
       period: { from: startedAt, to: lastTimestamp }
     };
   }
@@ -1469,16 +1490,18 @@ ${issues}`);
   }
   async function exportCSV() {
     const entries = await Promise.resolve(storage.getAll());
-    const header = "timestamp,model,inputTokens,outputTokens,costUSD,sessionId,userId";
+    const header = "timestamp,model,inputTokens,outputTokens,reasoningTokens,costUSD,sessionId,userId,feature";
     const rows = entries.map(
       (e) => [
         csvEscape(e.timestamp),
         csvEscape(e.model),
         e.inputTokens,
         e.outputTokens,
+        e.reasoningTokens ?? 0,
         e.costUSD.toFixed(8),
         csvEscape(e.sessionId ?? ""),
-        csvEscape(e.userId ?? "")
+        csvEscape(e.userId ?? ""),
+        csvEscape(e.feature ?? "")
       ].join(",")
     );
     return [header, ...rows].join("\n");
@@ -1579,6 +1602,12 @@ async function cmdReport() {
       console.log(`    ${session.padEnd(30)} $${stats.costUSD.toFixed(6)}  (${stats.calls} calls)`);
     }
   }
+  if (Object.keys(report.byFeature).length > 0) {
+    console.log("\n  By feature:");
+    for (const [feature, stats] of Object.entries(report.byFeature)) {
+      console.log(`    ${feature.padEnd(30)} $${stats.costUSD.toFixed(6)}  (${stats.calls} calls)`);
+    }
+  }
   console.log("\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\n");
 }
 function cmdHelp() {