ai-burnrate-tool 1.0.2 → 1.0.3

package/dist/core/parser.js

@@ -1,7 +1,85 @@
 import fs from 'node:fs';
+import { createRequire } from 'node:module';
 import os from 'node:os';
 import path from 'node:path';
 import { calculateTurnCost, loadPricing } from './pricing.js';
+const require = createRequire(import.meta.url);
+function loadAntigravityTitles(geminiDir) {
+    const titles = new Map();
+    let DatabaseSync;
+    try {
+        ({ DatabaseSync } = require('node:sqlite'));
+    }
+    catch {
+        return titles;
+    }
+    const dbCandidates = [
+        path.join(geminiDir, 'antigravity-cli', 'conversation_summaries.db'),
+        path.join(geminiDir, 'antigravity', 'conversation_summaries.db'),
+        path.join(geminiDir, 'conversation_summaries.db'),
+    ];
+    for (const dbPath of dbCandidates) {
+        if (!fs.existsSync(dbPath))
+            continue;
+        let db = null;
+        try {
+            db = new DatabaseSync(dbPath, { readOnly: true });
+            const rows = db
+                .prepare("SELECT conversation_id, title FROM conversation_summaries WHERE title != ''")
+                .all();
+            for (const row of rows) {
+                if (row.conversation_id && row.title && !titles.has(row.conversation_id)) {
+                    titles.set(row.conversation_id, row.title.trim());
+                }
+            }
+        }
+        catch {
+            // Ignore locked or incompatible SQLite DB
+        }
+        finally {
+            try {
+                db?.close?.();
+            }
+            catch {
+                // ignore close error
+            }
+        }
+    }
+    return titles;
+}
+function cleanChatTitle(raw, maxLen = 72) {
+    const userReqMatch = raw.match(/<USER_REQUEST>([\s\S]*?)<\/USER_REQUEST>/i);
+    const base = userReqMatch ? userReqMatch[1] : raw;
+    const cleaned = base
+        .replace(/<local-command-[^>]*>[\s\S]*?<\/local-command-[^>]*>/gi, ' ')
+        .replace(/<command-[^>]*>[\s\S]*?<\/command-[^>]*>/gi, ' ')
+        .replace(/<system-reminder>[\s\S]*?<\/system-reminder>/gi, ' ')
+        .replace(/<[^>]+>/g, ' ')
+        .replace(/^#+\s*/g, '')
+        .replace(/\s+/g, ' ')
+        .trim();
+    if (!cleaned)
+        return '';
+    return cleaned.length > maxLen ? `${cleaned.slice(0, maxLen - 1).trimEnd()}…` : cleaned;
+}
+function extractClaudeUserText(msgContent) {
+    if (typeof msgContent === 'string')
+        return msgContent;
+    if (Array.isArray(msgContent)) {
+        return msgContent
+            .filter((part) => part && part.type === 'text' && typeof part.text === 'string')
+            .map((part) => part.text)
+            .join(' ');
+    }
+    return '';
+}
+function formatSlugTitle(slug) {
+    return slug
+        .split(/[-_]+/)
+        .filter(Boolean)
+        .map((word) => word.charAt(0).toUpperCase() + word.slice(1))
+        .join(' ');
+}
 const fileCache = new Map();
 export function getWatchRoots() {
     const home = os.homedir();
@@ -126,7 +204,7 @@ function extractAntigravitySessionId(filePath) {
         return match[1];
     return path.basename(filePath, '.jsonl');
 }
-function parseAntigravityFile(filePath, stat, pricing) {
+function parseAntigravityFile(filePath, stat, pricing, antigravityTitles) {
     let content;
     try {
         content = fs.readFileSync(filePath, 'utf-8');
@@ -139,6 +217,7 @@ function parseAntigravityFile(filePath, stat, pricing) {
     let currentModel = 'Gemini 4 Argon';
     let workspace = null;
     let lastUserPromptChars = 0;
+    let firstUserTitle = '';
     const turnHistory = [];
     let startedAt = '';
     let lastActiveAt = '';
@@ -164,6 +243,11 @@ function parseAntigravityFile(filePath, stat, pricing) {
             lastActiveAt = timestamp;
         if (obj.type === 'USER_INPUT' && typeof obj.content === 'string') {
             lastUserPromptChars = obj.content.length;
+            if (!firstUserTitle) {
+                const candidateTitle = cleanChatTitle(obj.content);
+                if (candidateTitle)
+                    firstUserTitle = candidateTitle;
+            }
             const modelMatch = obj.content.match(/Model Selection`?\s+from\s+[^\r\n]+?\s+to\s+([^\r\n]+?)\.(?:\s|$)/i);
             if (modelMatch && modelMatch[1]) {
                 currentModel = modelMatch[1].trim();
@@ -245,8 +329,10 @@ function parseAntigravityFile(filePath, stat, pricing) {
     }
     if (turnHistory.length === 0)
         return null;
+    const sessionTitle = antigravityTitles?.get(sessionId) || firstUserTitle || sessionId;
     return summarizeSession({
         sessionId,
+        sessionTitle,
         source: 'antigravity',
         workspace: toRelativeHomePath(workspace || 'Global / CLI'),
         filePath: toRelativeHomePath(filePath),
@@ -269,6 +355,11 @@ function parseClaudeFile(filePath, stat, pricing) {
     let startedAt = '';
     let lastActiveAt = '';
     let lastUserPromptChars = 0;
+    let customTitle = '';
+    let summaryTitle = '';
+    let slugTitle = '';
+    let firstUserTitle = '';
+    let fallbackSlashTitle = '';
     const turnMap = new Map();
     let fallbackCounter = 0;
     for (const rawLine of lines) {
@@ -288,6 +379,15 @@ function parseClaudeFile(filePath, stat, pricing) {
         else if (typeof obj.session_id === 'string' && obj.session_id) {
             sessionId = obj.session_id;
         }
+        if (typeof obj.customTitle === 'string' && obj.customTitle.trim()) {
+            customTitle = cleanChatTitle(obj.customTitle);
+        }
+        if (obj.type === 'summary' && typeof obj.summary === 'string' && obj.summary.trim()) {
+            summaryTitle = cleanChatTitle(obj.summary);
+        }
+        if (!slugTitle && typeof obj.slug === 'string' && obj.slug.trim()) {
+            slugTitle = obj.slug.trim();
+        }
         if (typeof obj.cwd === 'string' && obj.cwd.trim()) {
             workspace = obj.cwd.trim();
         }
@@ -306,6 +406,19 @@ function parseClaudeFile(filePath, stat, pricing) {
             const msgContent = obj.message?.content ?? obj.content;
             const text = typeof msgContent === 'string' ? msgContent : JSON.stringify(msgContent ?? '');
             lastUserPromptChars = text.length;
+            if (!firstUserTitle) {
+                const userText = extractClaudeUserText(msgContent);
+                const candidate = cleanChatTitle(userText);
+                if (candidate && candidate !== '[Request interrupted by user]') {
+                    if (candidate.startsWith('/')) {
+                        if (!fallbackSlashTitle)
+                            fallbackSlashTitle = candidate;
+                    }
+                    else {
+                        firstUserTitle = candidate;
+                    }
+                }
+            }
             continue;
         }
         const isAssistant = obj.type === 'assistant' || obj.message?.role === 'assistant';
@@ -375,8 +488,15 @@ function parseClaudeFile(filePath, stat, pricing) {
     if (turnHistory.length === 0)
         return null;
     const resolvedWs = workspace || decodeClaudeProjectFolder(filePath);
+    const sessionTitle = customTitle ||
+        summaryTitle ||
+        firstUserTitle ||
+        (slugTitle ? formatSlugTitle(slugTitle) : '') ||
+        fallbackSlashTitle ||
+        sessionId;
     return summarizeSession({
         sessionId,
+        sessionTitle,
         source: 'claude',
         workspace: toRelativeHomePath(resolvedWs),
         filePath: toRelativeHomePath(filePath),
@@ -423,6 +543,7 @@ function summarizeSession(params) {
     }
     return {
         sessionId: params.sessionId,
+        sessionTitle: params.sessionTitle,
         source: params.source,
         workspace: params.workspace,
         workspaceName: formatWorkspaceName(params.workspace),
@@ -443,7 +564,7 @@ function summarizeSession(params) {
         turnHistory: params.turnHistory,
     };
 }
-export function parseSingleFile(filePath, source, pricing = loadPricing()) {
+export function parseSingleFile(filePath, source, pricing = loadPricing(), antigravityTitles) {
     if (shouldSkipJsonlFile(filePath, source))
         return null;
     let stat;
@@ -454,12 +575,18 @@ export function parseSingleFile(filePath, source, pricing = loadPricing()) {
         fileCache.delete(filePath);
         return null;
     }
+    const resolvedTitles = source === 'antigravity' ? antigravityTitles || loadAntigravityTitles(getWatchRoots().geminiDir) : undefined;
     const cached = fileCache.get(filePath);
     if (cached && cached.mtimeMs === stat.mtimeMs && cached.size === stat.size) {
+        if (cached.session && resolvedTitles) {
+            const dbTitle = resolvedTitles.get(cached.session.sessionId);
+            if (dbTitle)
+                cached.session.sessionTitle = dbTitle;
+        }
         return cached.session;
     }
     const session = source === 'antigravity'
-        ? parseAntigravityFile(filePath, stat, pricing)
+        ? parseAntigravityFile(filePath, stat, pricing, resolvedTitles)
         : parseClaudeFile(filePath, stat, pricing);
     fileCache.set(filePath, {
         mtimeMs: stat.mtimeMs,
@@ -474,11 +601,12 @@ export function scanAllLogs(forceClearCache = false) {
     }
     const pricing = loadPricing(forceClearCache);
     const { geminiDir, claudeDir } = getWatchRoots();
+    const antigravityTitles = loadAntigravityTitles(geminiDir);
     const geminiFiles = collectJsonlFiles(geminiDir, 'antigravity');
     const claudeFiles = collectJsonlFiles(claudeDir, 'claude');
     const sessions = [];
     for (const file of geminiFiles) {
-        const session = parseSingleFile(file, 'antigravity', pricing);
+        const session = parseSingleFile(file, 'antigravity', pricing, antigravityTitles);
         if (session)
             sessions.push(session);
     }