cursor-usage-analyzer 0.2.1 → 0.3.0

package/.idea/copilot.data.migration.agent.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="AgentMigrationStateService">
+    <option name="migrationStatus" value="COMPLETED" />
+  </component>
+</project>

package/analyze.js

@@ -3,10 +3,83 @@
 import fs from 'fs';
 import path from 'path';
 import os from 'os';
-import Database from 'better-sqlite3';
 import { generateHTMLReport } from './html-template.js';
 import { parse } from 'csv-parse/sync';
 
+// Database abstraction - try better-sqlite3 first, fall back to sql.js
+let dbEngine = null;
+
+async function loadDatabaseEngine() {
+  // Try better-sqlite3 first (faster, handles large files)
+  try {
+    const { default: Database } = await import('better-sqlite3');
+    // Test that it actually works (native bindings compiled)
+    const testDb = new Database(':memory:');
+    testDb.close();
+    dbEngine = { type: 'better-sqlite3', Database };
+    return;
+  } catch (e) {
+    console.log('Note: better-sqlite3 unavailable, trying sql.js fallback...');
+  }
+  
+  // Fall back to sql.js (pure JS, works everywhere, but 2GB limit)
+  try {
+    const { default: initSqlJs } = await import('sql.js');
+    const SQL = await initSqlJs();
+    dbEngine = { type: 'sql.js', SQL };
+  } catch (e) {
+    throw new Error('No SQLite library available. Please run: npm install -g cursor-usage-analyzer');
+  }
+}
+
+function openDatabase(dbPath) {
+  if (dbEngine.type === 'better-sqlite3') {
+    return new dbEngine.Database(dbPath, { readonly: true });
+  } else {
+    const fileSize = fs.statSync(dbPath).size;
+    if (fileSize > 2 * 1024 * 1024 * 1024) {
+      throw new Error(
+        `Database file (${(fileSize / 1024 / 1024 / 1024).toFixed(2)} GB) exceeds sql.js 2GB limit.\n` +
+        `To fix this, install globally with native compilation:\n` +
+        `  npm install -g cursor-usage-analyzer\n` +
+        `Or clone and build locally:\n` +
+        `  git clone https://github.com/xaverric/cursor-usage-analyzer.git\n` +
+        `  cd cursor-usage-analyzer && npm install && npm rebuild better-sqlite3`
+      );
+    }
+    const dbBuffer = fs.readFileSync(dbPath);
+    return new dbEngine.SQL.Database(dbBuffer);
+  }
+}
+
+function dbQueryAll(db, sql, params = []) {
+  if (dbEngine.type === 'better-sqlite3') {
+    return db.prepare(sql).all(...params);
+  } else {
+    const stmt = db.prepare(sql);
+    if (params.length > 0) stmt.bind(params);
+    const results = [];
+    while (stmt.step()) {
+      results.push(stmt.getAsObject());
+    }
+    stmt.free();
+    return results;
+  }
+}
+
+function dbQueryOne(db, sql, params = []) {
+  if (dbEngine.type === 'better-sqlite3') {
+    return db.prepare(sql).get(...params);
+  } else {
+    const results = dbQueryAll(db, sql, params);
+    return results.length > 0 ? results[0] : null;
+  }
+}
+
+function dbClose(db) {
+  db.close();
+}
+
 // Get Cursor storage paths based on OS
 function getCursorPaths() {
   const platform = os.platform();
@@ -246,7 +319,7 @@ function findWorkspaceFromPath(filePath) {
 }
 
 // Determine workspace name from multiple sources
-function resolveWorkspace(composerData, headers, db, composerId) {
+function resolveWorkspace(composerData, headers, composerId) {
   // Try workspaceId first
   if (composerData.workspaceId) {
     const name = readWorkspaceJson(composerData.workspaceId);
@@ -268,11 +341,20 @@ function resolveWorkspace(composerData, headers, db, composerId) {
     }
   }
   
+  return 'unknown';
+}
+
+// Determine workspace name with database access for messageRequestContext
+function resolveWorkspaceWithDb(composerData, headers, db, composerId) {
+  // Try basic resolution first
+  const basicResult = resolveWorkspace(composerData, headers, composerId);
+  if (basicResult !== 'unknown') return basicResult;
+  
   // Last resort: messageRequestContext
   if (headers?.length > 0) {
     try {
       const contextKey = `messageRequestContext:${composerId}:${headers[0].bubbleId}`;
-      const row = db.prepare("SELECT value FROM cursorDiskKV WHERE key = ?").get(contextKey);
+      const row = dbQueryOne(db, "SELECT value FROM cursorDiskKV WHERE key = ?", [contextKey]);
       if (row) {
         const context = JSON.parse(row.value);
         if (context.projectLayouts?.length > 0) {
@@ -288,7 +370,7 @@ function resolveWorkspace(composerData, headers, db, composerId) {
 }
 
 // Extract conversations from database
-function extractConversations(startTime, endTime, apiCalls = []) {
+async function extractConversations(startTime, endTime, apiCalls = []) {
   const dbPath = path.join(CURSOR_GLOBAL_STORAGE, 'state.vscdb');
 
   if (!fs.existsSync(dbPath)) {
@@ -297,11 +379,11 @@ function extractConversations(startTime, endTime, apiCalls = []) {
   }
 
   try {
-    const db = new Database(dbPath, { readonly: true });
+    const db = openDatabase(dbPath);
 
     // Load all bubbles
     const bubbleMap = {};
-    const bubbleRows = db.prepare("SELECT key, value FROM cursorDiskKV WHERE key LIKE 'bubbleId:%'").all();
+    const bubbleRows = dbQueryAll(db, "SELECT key, value FROM cursorDiskKV WHERE key LIKE 'bubbleId:%'");
 
     for (const row of bubbleRows) {
       try {
@@ -312,9 +394,9 @@ function extractConversations(startTime, endTime, apiCalls = []) {
     }
 
     // Load composers
-    const composerRows = db.prepare(
+    const composerRows = dbQueryAll(db,
       "SELECT key, value FROM cursorDiskKV WHERE key LIKE 'composerData:%' AND value LIKE '%fullConversationHeadersOnly%'"
-    ).all();
+    );
 
     const conversations = [];
 
@@ -354,7 +436,7 @@ function extractConversations(startTime, endTime, apiCalls = []) {
           timestamp,
           messages,
           messageCount: messages.length,
-          workspace: resolveWorkspace(composer, headers, db, composerId),
+          workspace: resolveWorkspaceWithDb(composer, headers, db, composerId),
           model: composer.modelConfig?.modelName || 'unknown',
           contextTokensUsed: composer.contextTokensUsed || 0,
           contextTokenLimit: composer.contextTokenLimit || 0,
@@ -419,7 +501,7 @@ function extractConversations(startTime, endTime, apiCalls = []) {
       });
     }
 
-    db.close();
+    dbClose(db);
     return conversations;
   } catch (error) {
     console.error('Database error:', error.message);
@@ -612,7 +694,8 @@ async function main() {
 
   console.log('Extracting conversations...');
 
-  const conversations = extractConversations(startOfDay, endOfDay, apiCalls);
+  await loadDatabaseEngine();
+  const conversations = await extractConversations(startOfDay, endOfDay, apiCalls);
 
   console.log(`Found ${conversations.length} conversations\n`);
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cursor-usage-analyzer",
-  "version": "0.2.1",
+  "version": "0.3.0",
   "description": "Analyze and visualize your Cursor AI editor usage with interactive reports including API token tracking and costs",
   "main": "analyze.js",
   "type": "module",
@@ -12,13 +12,15 @@
     "today": "node analyze.js --today",
     "yesterday": "node analyze.js --yesterday",
     "this-month": "node analyze.js --this-month",
-    "last-month": "node analyze.js --last-month"
+    "last-month": "node analyze.js --last-month",
+    "postinstall": "npm rebuild better-sqlite3 2>/dev/null || echo 'Note: better-sqlite3 native build failed, will use sql.js fallback (2GB limit)'"
   },
   "keywords": ["cursor", "ai", "analytics", "usage", "chat", "tokens", "statistics"],
   "author": "Daniel Jílek",
   "license": "MIT",
   "dependencies": {
-    "better-sqlite3": "^9.2.2",
+    "better-sqlite3": "^11.7.0",
+    "sql.js": "^1.11.0",
     "csv-parse": "^5.5.3"
   },
   "repository": {