agentlytics 0.1.15 → 0.1.17

package/editors/copilot.js

@@ -171,6 +171,73 @@ function safeParse(str) {
   try { return JSON.parse(str); } catch { return {}; }
 }
 
+// ============================================================
+// Usage / quota data from GitHub Copilot internal API
+// ============================================================
+
+function getCopilotToken() {
+  // GitHub Copilot stores its OAuth token in ~/.config/github-copilot/apps.json
+  const appsPath = path.join(os.homedir(), '.config', 'github-copilot', 'apps.json');
+  try {
+    if (!fs.existsSync(appsPath)) return null;
+    const data = JSON.parse(fs.readFileSync(appsPath, 'utf-8'));
+    for (const entry of Object.values(data)) {
+      if (entry.oauth_token) return { token: entry.oauth_token, user: entry.user || null };
+    }
+  } catch {}
+  return null;
+}
+
+function fetchCopilotStatus(token) {
+  return new Promise((resolve) => {
+    const https = require('https');
+    const req = https.get('https://api.github.com/copilot_internal/v2/token', {
+      headers: {
+        'Authorization': `token ${token}`,
+        'Accept': 'application/json',
+        'User-Agent': 'agentlytics/1.0',
+      },
+      timeout: 10000,
+    }, (res) => {
+      let data = '';
+      res.on('data', (chunk) => { data += chunk; });
+      res.on('end', () => {
+        try { resolve(JSON.parse(data)); } catch { resolve(null); }
+      });
+    });
+    req.on('error', () => resolve(null));
+    req.on('timeout', () => { req.destroy(); resolve(null); });
+  });
+}
+
+async function getUsage() {
+  const creds = getCopilotToken();
+  if (!creds) return null;
+
+  const status = await fetchCopilotStatus(creds.token);
+  if (!status || status.message) return null;
+
+  return {
+    source: 'copilot-cli',
+    plan: {
+      name: status.sku || null,
+      individual: status.individual || false,
+    },
+    features: {
+      chat: status.chat_enabled || false,
+      codeReview: status.code_review_enabled || false,
+      agentMode: status.agent_mode_auto_approval || false,
+    },
+    limits: {
+      quotas: status.limited_user_quotas || null,
+      resetDate: status.limited_user_reset_date || null,
+    },
+    user: {
+      login: creds.user || null,
+    },
+  };
+}
+
 const labels = { 'copilot-cli': 'Copilot CLI' };
 
-module.exports = { name, labels, getChats, getMessages };
+module.exports = { name, labels, getChats, getMessages, getUsage };

package/editors/cursor.js

@@ -341,6 +341,78 @@ function getMessages(chat) {
   return msgs;
 }
 
+// ============================================================
+// Usage / quota data from Cursor REST API
+// ============================================================
+
+function getCursorAccessToken() {
+  try {
+    const db = new Database(GLOBAL_STORAGE_DB, { readonly: true });
+    const row = db.prepare("SELECT value FROM ItemTable WHERE key = 'cursorAuth/accessToken'").get();
+    db.close();
+    return row ? row.value : null;
+  } catch { return null; }
+}
+
+function cursorApiFetch(endpoint, token) {
+  return new Promise((resolve) => {
+    const https = require('https');
+    const url = `https://api2.cursor.sh/auth/${endpoint}`;
+    const req = https.get(url, {
+      headers: { 'Authorization': `Bearer ${token}`, 'Content-Type': 'application/json' },
+      timeout: 10000,
+    }, (res) => {
+      let data = '';
+      res.on('data', (chunk) => { data += chunk; });
+      res.on('end', () => {
+        try { resolve(JSON.parse(data)); } catch { resolve(null); }
+      });
+    });
+    req.on('error', () => resolve(null));
+    req.on('timeout', () => { req.destroy(); resolve(null); });
+  });
+}
+
+async function getUsage() {
+  const token = getCursorAccessToken();
+  if (!token) return null;
+
+  const [profile, usage] = await Promise.all([
+    cursorApiFetch('full_stripe_profile', token),
+    cursorApiFetch('usage', token),
+  ]);
+
+  if (!profile && !usage) return null;
+
+  const result = {
+    source: 'cursor',
+    plan: {
+      name: profile?.individualMembershipType || profile?.membershipType || null,
+      status: profile?.subscriptionStatus || null,
+      isTeamMember: profile?.isTeamMember || false,
+      isYearlyPlan: profile?.isYearlyPlan || false,
+    },
+    usage: {},
+    startOfMonth: usage?.startOfMonth || null,
+  };
+
+  // Parse per-model usage from the usage endpoint
+  if (usage) {
+    for (const [model, data] of Object.entries(usage)) {
+      if (model === 'startOfMonth') continue;
+      result.usage[model] = {
+        numRequests: data.numRequests || 0,
+        numRequestsTotal: data.numRequestsTotal || 0,
+        numTokens: data.numTokens || 0,
+        maxRequestUsage: data.maxRequestUsage || null,
+        maxTokenUsage: data.maxTokenUsage || null,
+      };
+    }
+  }
+
+  return result;
+}
+
 const labels = { 'cursor': 'Cursor' };
 
-module.exports = { name, labels, getChats, getMessages };
+module.exports = { name, labels, getChats, getMessages, getUsage };

package/editors/index.js

@@ -1,5 +1,6 @@
 const cursor = require('./cursor');
 const windsurf = require('./windsurf');
+const antigravity = require('./antigravity');
 const claude = require('./claude');
 const vscode = require('./vscode');
 const zed = require('./zed');
@@ -12,7 +13,7 @@ const commandcode = require('./commandcode');
 const goose = require('./goose');
 const kiro = require('./kiro');
 
-const editors = [cursor, windsurf, claude, vscode, zed, opencode, codex, gemini, copilot, cursorAgent, commandcode, goose, kiro];
+const editors = [cursor, windsurf, antigravity, claude, vscode, zed, opencode, codex, gemini, copilot, cursorAgent, commandcode, goose, kiro];
 
 // Build a unified source → display-label map from all editor modules
 const editorLabels = {};
@@ -60,4 +61,23 @@ function resetCaches() {
   }
 }
 
-module.exports = { getAllChats, getMessages, editors, editorLabels, resetCaches };
+/**
+ * Get usage / quota data from all editors that support it.
+ * Returns an array of usage objects, one per editor/variant.
+ */
+async function getAllUsage() {
+  const results = [];
+  for (const editor of editors) {
+    if (typeof editor.getUsage !== 'function') continue;
+    try {
+      const usage = await editor.getUsage();
+      if (!usage) continue;
+      // Windsurf returns an array (one per variant), Cursor returns a single object
+      if (Array.isArray(usage)) results.push(...usage);
+      else results.push(usage);
+    } catch { /* skip broken adapters */ }
+  }
+  return results;
+}
+
+module.exports = { getAllChats, getMessages, editors, editorLabels, resetCaches, getAllUsage };

package/editors/opencode.js

@@ -1,6 +1,7 @@
 const path = require('path');
 const fs = require('fs');
 const os = require('os');
+const Database = require('better-sqlite3');
 
 // OpenCode stores data in different locations depending on the platform
 // - Windows: %LOCALAPPDATA%\opencode\storage (not Roaming)
@@ -22,6 +23,118 @@ const SESSION_DIR = path.join(STORAGE_DIR, 'session');
 const MESSAGE_DIR = path.join(STORAGE_DIR, 'message');
 const PART_DIR = path.join(STORAGE_DIR, 'part');
 
+// OpenCode also stores data in a SQLite database (older/primary store)
+function getOpenCodeDbPath() {
+  const home = os.homedir();
+  switch (process.platform) {
+    case 'win32':
+      return path.join(home, 'AppData', 'Local', 'opencode', 'opencode.db');
+    case 'darwin':
+    case 'linux':
+    default:
+      return path.join(home, '.local', 'share', 'opencode', 'opencode.db');
+  }
+}
+
+const DB_PATH = getOpenCodeDbPath();
+
+// ============================================================
+// Query SQLite using better-sqlite3
+// ============================================================
+
+function queryDb(sql) {
+  if (!fs.existsSync(DB_PATH)) return [];
+  try {
+    const db = new Database(DB_PATH, { readonly: true });
+    const rows = db.prepare(sql).all();
+    db.close();
+    return rows;
+  } catch {
+    return [];
+  }
+}
+
+function getSqliteSessions() {
+  return queryDb(
+    `SELECT s.id, s.title, s.directory, s.time_created, s.time_updated,
+            p.worktree, p.name as project_name,
+            (SELECT count(*) FROM message m WHERE m.session_id = s.id) as msg_count
+     FROM session s LEFT JOIN project p ON s.project_id = p.id
+     ORDER BY s.time_updated DESC`
+  );
+}
+
+function getSqliteMessages(sessionId) {
+  if (!fs.existsSync(DB_PATH)) return [];
+  try {
+    const db = new Database(DB_PATH, { readonly: true });
+    const messages = db.prepare(
+      `SELECT m.id as msg_id, m.data as msg_data, m.time_created
+       FROM message m WHERE m.session_id = ? ORDER BY m.time_created ASC`
+    ).all(sessionId);
+
+    const result = [];
+    for (const msg of messages) {
+      let msgData;
+      try { msgData = JSON.parse(msg.msg_data); } catch { continue; }
+
+      const role = msgData.role;
+      if (!role) continue;
+
+      const parts = db.prepare(
+        `SELECT data FROM part WHERE message_id = ? ORDER BY time_created ASC`
+      ).all(msg.msg_id);
+
+      const contentParts = [];
+      for (const part of parts) {
+        let partData;
+        try { partData = JSON.parse(part.data); } catch { continue; }
+        const type = partData.type;
+
+        if (type === 'text' && partData.text) {
+          contentParts.push(partData.text);
+        } else if (type === 'thinking' || type === 'reasoning') {
+          if (partData.text) contentParts.push(`[thinking] ${partData.text}`);
+        } else if (type === 'tool-call' || type === 'tool_use' || type === 'tool-use' || type === 'tool') {
+          const toolName = partData.name || partData.toolName || partData.tool || 'tool';
+          let argKeys = '';
+          try {
+            const input = typeof partData.input === 'string' ? JSON.parse(partData.input) : (partData.input || partData.args || partData.arguments || partData.state?.input || {});
+            argKeys = typeof input === 'object' ? Object.keys(input).join(', ') : '';
+          } catch {}
+          contentParts.push(`[tool-call: ${toolName}(${argKeys})]`);
+        } else if (type === 'tool-result' || type === 'tool_result') {
+          const preview = (partData.text || partData.output || partData.state?.output || '').substring(0, 500);
+          contentParts.push(`[tool-result] ${preview}`);
+        }
+      }
+
+      const content = contentParts.join('\n');
+      if (!content) continue;
+
+      let modelValue = null;
+      if (typeof msgData.modelID === 'string') {
+        modelValue = msgData.modelID;
+      } else if (msgData.model && typeof msgData.model === 'object' && msgData.model.modelID) {
+        modelValue = msgData.model.modelID;
+      } else if (typeof msgData.model === 'string') {
+        modelValue = msgData.model;
+      }
+
+      result.push({
+        role: role === 'user' ? 'user' : role === 'assistant' ? 'assistant' : role,
+        content,
+        _model: modelValue,
+      });
+    }
+
+    db.close();
+    return result;
+  } catch {
+    return [];
+  }
+}
+
 // ============================================================
 // Scan JSON files from OpenCode storage
 // ============================================================
@@ -156,27 +269,64 @@ function getMessagesForSession(sessionId) {
 const name = 'opencode';
 
 function getChats() {
-  const sessions = getAllSessions();
-
-  return sessions.map(s => ({
-    source: 'opencode',
-    composerId: s.id,
-    name: s.title || null,
-    createdAt: s.time?.created || null,
-    lastUpdatedAt: s.time?.updated || null,
-    mode: s.mode || 'opencode',
-    folder: s.directory || null,
-    encrypted: false,
-    bubbleCount: getMessageCount(s.id),
-    _agent: s.agent,
-    _model: s.modelID,
-    _provider: s.providerID,
-    _sessionData: s,
-  })).sort((a, b) => (b.lastUpdatedAt || 0) - (a.lastUpdatedAt || 0));
+  const seen = new Set();
+  const chats = [];
+
+  // 1. JSON file-based sessions (newer storage format)
+  const fileSessions = getAllSessions();
+  for (const s of fileSessions) {
+    seen.add(s.id);
+    chats.push({
+      source: 'opencode',
+      composerId: s.id,
+      name: s.title || null,
+      createdAt: s.time?.created || null,
+      lastUpdatedAt: s.time?.updated || null,
+      mode: s.mode || 'opencode',
+      folder: s.directory || null,
+      encrypted: false,
+      bubbleCount: getMessageCount(s.id),
+      _agent: s.agent,
+      _model: s.modelID,
+      _provider: s.providerID,
+      _sessionData: s,
+      _storageType: 'file',
+    });
+  }
+
+  // 2. SQLite sessions (older/primary store) — add any not already found in files
+  const dbSessions = getSqliteSessions();
+  for (const row of dbSessions) {
+    if (seen.has(row.id)) continue;
+    seen.add(row.id);
+    chats.push({
+      source: 'opencode',
+      composerId: row.id,
+      name: cleanTitle(row.title),
+      createdAt: row.time_created || null,
+      lastUpdatedAt: row.time_updated || null,
+      mode: 'opencode',
+      folder: row.worktree || row.directory || null,
+      encrypted: false,
+      bubbleCount: row.msg_count || 0,
+      _storageType: 'sqlite',
+    });
+  }
+
+  return chats.sort((a, b) => (b.lastUpdatedAt || 0) - (a.lastUpdatedAt || 0));
+}
+
+function cleanTitle(title) {
+  if (!title) return null;
+  if (title.startsWith('New session - ')) return null;
+  return title.substring(0, 120) || null;
 }
 
 function getMessages(chat) {
-  return getMessagesForSession(chat.composerId);
+  // Prefer file-based messages; fall back to SQLite
+  const fileMessages = getMessagesForSession(chat.composerId);
+  if (fileMessages.length > 0) return fileMessages;
+  return getSqliteMessages(chat.composerId);
 }
 
 const labels = { 'opencode': 'OpenCode' };

package/editors/vscode.js

@@ -315,6 +315,75 @@ function getMessages(chat) {
   return messages;
 }
 
+// ============================================================
+// Usage / quota data from GitHub Copilot internal API
+// ============================================================
+
+function getCopilotToken() {
+  const appsPath = path.join(os.homedir(), '.config', 'github-copilot', 'apps.json');
+  try {
+    if (!fs.existsSync(appsPath)) return null;
+    const data = JSON.parse(fs.readFileSync(appsPath, 'utf-8'));
+    // Pick the first available oauth_token
+    for (const entry of Object.values(data)) {
+      if (entry.oauth_token) return { token: entry.oauth_token, user: entry.user || null };
+    }
+  } catch {}
+  return null;
+}
+
+function fetchCopilotStatus(token) {
+  return new Promise((resolve) => {
+    const https = require('https');
+    const req = https.get('https://api.github.com/copilot_internal/v2/token', {
+      headers: {
+        'Authorization': `token ${token}`,
+        'Accept': 'application/json',
+        'User-Agent': 'agentlytics/1.0',
+      },
+      timeout: 10000,
+    }, (res) => {
+      let data = '';
+      res.on('data', (chunk) => { data += chunk; });
+      res.on('end', () => {
+        try { resolve(JSON.parse(data)); } catch { resolve(null); }
+      });
+    });
+    req.on('error', () => resolve(null));
+    req.on('timeout', () => { req.destroy(); resolve(null); });
+  });
+}
+
+async function getUsage() {
+  const creds = getCopilotToken();
+  if (!creds) return null;
+
+  const status = await fetchCopilotStatus(creds.token);
+  if (!status || status.message) return null;
+
+  return {
+    source: 'vscode',
+    plan: {
+      name: status.sku || null,
+      individual: status.individual || false,
+    },
+    features: {
+      chat: status.chat_enabled || false,
+      codeReview: status.code_review_enabled || false,
+      agentMode: status.agent_mode_auto_approval || false,
+      xcode: status.xcode || false,
+      mcp: status.mcp || false,
+    },
+    limits: {
+      quotas: status.limited_user_quotas || null,
+      resetDate: status.limited_user_reset_date || null,
+    },
+    user: {
+      login: creds.user || null,
+    },
+  };
+}
+
 const labels = { 'vscode': 'VS Code', 'vscode-insiders': 'VS Code Insiders' };
 
-module.exports = { name, labels, getChats, getMessages };
+module.exports = { name, labels, getChats, getMessages, getUsage };