codebuddy-stats 1.3.1 → 1.3.3

package/dist/index.js

@@ -3,6 +3,7 @@ import fs from 'node:fs';
 import path from 'node:path';
 import { fileURLToPath } from 'node:url';
 import blessed from 'blessed';
+import { performance } from 'node:perf_hooks';
 import { loadUsageData } from './lib/data-loader.js';
 import { resolveProjectName } from './lib/workspace-resolver.js';
 import { compareByCostThenTokens, formatCost, formatNumber, formatPercent, formatTokens, truncate } from './lib/utils.js';
@@ -72,7 +73,7 @@ function printTextReport(data) {
         console.log(`\nTop model:         ${topModel.id} (${formatCost(topModel.cost)})`);
     }
     if (topProject) {
-        const shortName = resolveProjectName(topProject.name, data.workspaceMappings);
+        const shortName = data.projectDisplayNames?.[topProject.name] ?? resolveProjectName(topProject.name, data.workspaceMappings);
         console.log(`Top project:       ${shortName}`);
         console.log(`                   (${formatCost(topProject.cost)})`);
     }
@@ -86,7 +87,7 @@ function printTextReport(data) {
     for (const [project, stats] of Object.entries(projectTotals)
         .sort((a, b) => b[1].cost - a[1].cost)
         .slice(0, 10)) {
-        const shortName = resolveProjectName(project, data.workspaceMappings);
+        const shortName = data.projectDisplayNames?.[project] ?? resolveProjectName(project, data.workspaceMappings);
         console.log(`  ${truncate(shortName, 40)}: ${formatCost(stats.cost)}`); // eslint-disable-line no-console
     }
     console.log('\n' + '='.repeat(50) + '\n');
@@ -185,8 +186,14 @@ async function main() {
         content += ' {gray-fg}(Tab view, s source){/gray-fg}';
         tabBar.setContent(content);
     }
+    // profiling（默认关闭；建议配合 --no-tui 使用以免打乱终端 UI）
+    const prof = process.env.CBS_PROFILE === '1';
+    const profTui = process.env.CBS_PROFILE_TUI === '1';
+    let lastRenderMs = 0;
+    let lastRenderTab = '';
     // 更新内容
     function updateContent() {
+        const t0 = prof ? performance.now() : 0;
         const width = Number(screen.width) || 80;
         const note = state.currentSource === 'code'
             ? `针对 CodeBuddy Code < 2.20.0 版本产生的数据，由于没有请求级别的 model ID，用量是基于当前 CodeBuddy Code 设置的 model ID（${state.data.defaultModelId}）计算价格的`
@@ -225,7 +232,7 @@ async function main() {
         const dailyMaxOffset = Math.max(0, Object.keys(state.data.dailySummary).length - state.dailyPageSize);
         state.dailyScrollOffset = Math.min(state.dailyScrollOffset, dailyMaxOffset);
         state.dailySelectedIndex = Math.min(state.dailySelectedIndex, Math.max(0, Object.keys(state.data.dailySummary).length - 1));
-        const sortedMonths = getSortedMonthsFromDailyData(state.data.dailyData);
+        const sortedMonths = state.data.sortedMonths ?? getSortedMonthsFromDailyData(state.data.dailyData);
         // 如果当前在 Monthly detail 但该月份已不存在，回到列表
         if (state.monthlyDetailMonth && !sortedMonths.includes(state.monthlyDetailMonth)) {
             state.monthlyDetailMonth = null;
@@ -243,27 +250,39 @@ async function main() {
         }
         // Monthly detail 的滚动上限
         if (state.monthlyDetailMonth) {
-            const projects = new Set();
-            for (const [date, dayData] of Object.entries(state.data.dailyData)) {
-                if (!date.startsWith(state.monthlyDetailMonth))
-                    continue;
-                for (const projectName of Object.keys(dayData ?? {}))
-                    projects.add(projectName);
+            const cachedMonthProjects = state.data.monthlyByProject?.[state.monthlyDetailMonth];
+            let projectCount = 0;
+            if (cachedMonthProjects) {
+                projectCount = Object.keys(cachedMonthProjects).length;
             }
-            const maxOffset = Math.max(0, projects.size - state.monthlyDetailPageSize);
+            else {
+                const projects = new Set();
+                for (const [date, dayData] of Object.entries(state.data.dailyData)) {
+                    if (!date.startsWith(state.monthlyDetailMonth))
+                        continue;
+                    for (const projectName of Object.keys(dayData ?? {}))
+                        projects.add(projectName);
+                }
+                projectCount = projects.size;
+            }
+            const maxOffset = Math.max(0, projectCount - state.monthlyDetailPageSize);
             state.monthlyDetailScrollOffset = Math.min(state.monthlyDetailScrollOffset, maxOffset);
         }
         switch (state.currentTab) {
             case 0:
+                lastRenderTab = 'overview';
                 renderOverview(contentBox, state.data, width, innerHeight, note);
                 break;
             case 1:
+                lastRenderTab = 'by-model';
                 renderByModel(contentBox, state.data, state.modelScrollOffset, width, note, state.modelPageSize);
                 break;
             case 2:
+                lastRenderTab = 'by-project';
                 renderByProject(contentBox, state.data, state.projectScrollOffset, width, note, state.projectPageSize);
                 break;
             case 3:
+                lastRenderTab = state.dailyDetailDate ? 'daily-detail' : 'daily';
                 if (state.dailyDetailDate) {
                     renderDailyDetail(contentBox, state.data, state.dailyDetailDate, state.dailyDetailScrollOffset, width, state.dailyDetailPageSize);
                 }
@@ -272,6 +291,7 @@ async function main() {
                 }
                 break;
             case 4:
+                lastRenderTab = state.monthlyDetailMonth ? 'monthly-detail' : 'monthly';
                 if (state.monthlyDetailMonth) {
                     renderMonthlyDetail(contentBox, state.data, state.monthlyDetailMonth, state.monthlyDetailScrollOffset, width, state.monthlyDetailPageSize);
                 }
@@ -280,6 +300,12 @@ async function main() {
                 }
                 break;
         }
+        if (prof) {
+            lastRenderMs = performance.now() - t0;
+            if (profTui) {
+                console.error(`[cbs] render tab=${lastRenderTab} ${lastRenderMs.toFixed(1)}ms`); // eslint-disable-line no-console
+            }
+        }
     }
     // 更新状态栏
     function updateStatusBar() {
@@ -291,9 +317,10 @@ async function main() {
         const reservedForRight = rightContent.length + 2; // 版本号 + 两侧空格
         const availableForLeft = width - reservedForRight;
         let leftContent;
-        const fullContent = ` ${daysInfo} | Source: ${sourceInfo} | Total: ${formatCost(state.data.grandTotal.cost)} | q quit, Tab view, s source, r refresh`;
-        const mediumContent = ` ${daysInfo} | ${sourceInfo} | ${formatCost(state.data.grandTotal.cost)} | q/Tab/s/r`;
-        const shortContent = ` ${sourceInfo} | ${formatCost(state.data.grandTotal.cost)} | q/Tab/s/r`;
+        const perfInfo = prof ? ` | render:${lastRenderTab} ${Math.round(lastRenderMs)}ms` : '';
+        const fullContent = ` ${daysInfo} | Source: ${sourceInfo} | Total: ${formatCost(state.data.grandTotal.cost)}${perfInfo} | q quit, Tab view, s source, r refresh`;
+        const mediumContent = ` ${daysInfo} | ${sourceInfo} | ${formatCost(state.data.grandTotal.cost)}${perfInfo} | q/Tab/s/r`;
+        const shortContent = ` ${sourceInfo} | ${formatCost(state.data.grandTotal.cost)}${perfInfo} | q/Tab/s/r`;
         const minContent = ` ${formatCost(state.data.grandTotal.cost)}`;
         if (fullContent.length <= availableForLeft) {
             leftContent = fullContent;

package/dist/lib/code-usage.worker.js

@@ -0,0 +1,132 @@
+import fs from 'node:fs/promises';
+import fsSync from 'node:fs';
+import path from 'node:path';
+import { createInterface } from 'node:readline';
+import { parentPort, workerData } from 'node:worker_threads';
+import { DEFAULT_MODEL_ID, getPricingForModel, tokensToCost } from './pricing.js';
+function getProjectName(filePath) {
+    const parts = filePath.split(path.sep);
+    const projectsIndex = parts.lastIndexOf('projects');
+    if (projectsIndex !== -1 && projectsIndex < parts.length - 1) {
+        return parts[projectsIndex + 1] ?? 'unknown-project';
+    }
+    return 'unknown-project';
+}
+function extractUsageStats(usage) {
+    const promptTokens = usage.prompt_tokens ?? 0;
+    const completionTokens = usage.completion_tokens ?? 0;
+    const totalTokens = usage.total_tokens ?? promptTokens + completionTokens;
+    const cacheHitTokens = usage.prompt_cache_hit_tokens ?? 0;
+    const cacheMissTokens = usage.prompt_cache_miss_tokens ?? (cacheHitTokens > 0 ? Math.max(promptTokens - cacheHitTokens, 0) : 0);
+    const cacheWriteTokens = usage.cache_creation_input_tokens ?? 0;
+    return {
+        promptTokens,
+        completionTokens,
+        totalTokens,
+        cacheHitTokens,
+        cacheMissTokens,
+        cacheWriteTokens,
+    };
+}
+function computeUsageCost(usage, modelId) {
+    const stats = extractUsageStats(usage);
+    const pricing = getPricingForModel(modelId);
+    let inputCost = 0;
+    if (stats.cacheHitTokens || stats.cacheMissTokens || stats.cacheWriteTokens) {
+        inputCost += tokensToCost(stats.cacheHitTokens, pricing.cacheRead);
+        inputCost += tokensToCost(stats.cacheMissTokens, pricing.prompt);
+        inputCost += tokensToCost(stats.cacheWriteTokens, pricing.cacheWrite);
+    }
+    else {
+        inputCost += tokensToCost(stats.promptTokens, pricing.prompt);
+    }
+    const outputCost = tokensToCost(stats.completionTokens, pricing.completion);
+    return { cost: inputCost + outputCost, stats };
+}
+function ensureDailyModelStats(dailyData, date, project, modelId) {
+    dailyData[date] ??= {};
+    dailyData[date][project] ??= {};
+    dailyData[date][project][modelId] ??= {
+        cost: 0,
+        promptTokens: 0,
+        completionTokens: 0,
+        totalTokens: 0,
+        cacheHitTokens: 0,
+        cacheMissTokens: 0,
+        cacheWriteTokens: 0,
+        requests: 0,
+    };
+    return dailyData[date][project][modelId];
+}
+async function main() {
+    const input = workerData;
+    const files = Array.isArray(input?.files) ? input.files : [];
+    const defaultModelId = typeof input?.defaultModelId === 'string' && input.defaultModelId ? input.defaultModelId : DEFAULT_MODEL_ID;
+    const minDate = typeof input?.minDate === 'string' ? input.minDate : null;
+    const dailyData = {};
+    const modelTotals = {};
+    const projectTotals = {};
+    const grandTotal = { cost: 0, tokens: 0, requests: 0, cacheHitTokens: 0, cacheMissTokens: 0 };
+    for (const filePath of files) {
+        const projectName = getProjectName(filePath);
+        let fileStream;
+        try {
+            fileStream = fsSync.createReadStream(filePath);
+        }
+        catch {
+            continue;
+        }
+        const rl = createInterface({ input: fileStream, crlfDelay: Number.POSITIVE_INFINITY });
+        for await (const line of rl) {
+            try {
+                const record = JSON.parse(line);
+                const usage = record?.providerData?.rawUsage;
+                const timestamp = record?.timestamp;
+                if (!usage || timestamp == null)
+                    continue;
+                const dateObj = new Date(timestamp);
+                if (Number.isNaN(dateObj.getTime()))
+                    continue;
+                const date = dateObj.toISOString().split('T')[0];
+                if (!date)
+                    continue;
+                if (minDate && date < minDate)
+                    continue;
+                const recordModelId = record?.providerData?.model;
+                const modelFromRecord = typeof recordModelId === 'string' ? recordModelId : null;
+                const usedModelId = modelFromRecord || defaultModelId;
+                const { cost, stats: usageStats } = computeUsageCost(usage, usedModelId);
+                const dayStats = ensureDailyModelStats(dailyData, date, projectName, usedModelId);
+                dayStats.cost += cost;
+                dayStats.promptTokens += usageStats.promptTokens;
+                dayStats.completionTokens += usageStats.completionTokens;
+                dayStats.totalTokens += usageStats.totalTokens;
+                dayStats.cacheHitTokens += usageStats.cacheHitTokens;
+                dayStats.cacheMissTokens += usageStats.cacheMissTokens;
+                dayStats.cacheWriteTokens += usageStats.cacheWriteTokens;
+                dayStats.requests += 1;
+                modelTotals[usedModelId] ??= { cost: 0, tokens: 0, requests: 0 };
+                modelTotals[usedModelId].cost += cost;
+                modelTotals[usedModelId].tokens += usageStats.totalTokens;
+                modelTotals[usedModelId].requests += 1;
+                projectTotals[projectName] ??= { cost: 0, tokens: 0, requests: 0 };
+                projectTotals[projectName].cost += cost;
+                projectTotals[projectName].tokens += usageStats.totalTokens;
+                projectTotals[projectName].requests += 1;
+                grandTotal.cost += cost;
+                grandTotal.tokens += usageStats.totalTokens;
+                grandTotal.requests += 1;
+                grandTotal.cacheHitTokens += usageStats.cacheHitTokens;
+                grandTotal.cacheMissTokens += usageStats.cacheMissTokens;
+            }
+            catch {
+                // ignore
+            }
+        }
+    }
+    parentPort?.postMessage({ dailyData, modelTotals, projectTotals, grandTotal });
+}
+main().catch(err => {
+    // 让主线程感知 worker 失败
+    throw err;
+});

package/dist/lib/data-loader.js

@@ -1,12 +1,33 @@
 import fs from 'node:fs/promises';
 import fsSync from 'node:fs';
 import path from 'node:path';
+import os from 'node:os';
+import { createHash } from 'node:crypto';
+import { performance } from 'node:perf_hooks';
 import { createInterface } from 'node:readline';
+import { Worker } from 'node:worker_threads';
 import { getIdeDataDir, getProjectsDir, getSettingsPath } from './paths.js';
 import { DEFAULT_MODEL_ID, getPricingForModel, tokensToCost } from './pricing.js';
 import { compareByCostThenTokens } from './utils.js';
-import { loadWorkspaceMappings } from './workspace-resolver.js';
+import { loadWorkspaceMappings, resolveProjectName } from './workspace-resolver.js';
 export const BASE_DIR = getProjectsDir();
+const memoryCache = new Map();
+function getCacheKey(source, days) {
+    return `${source}:${days ?? 'all'}`;
+}
+function cacheGet(source, days) {
+    return memoryCache.get(getCacheKey(source, days));
+}
+function cacheSet(source, days, entry) {
+    const key = getCacheKey(source, days);
+    memoryCache.set(key, entry);
+    // 简单上限：只保留最近 4 个 key
+    if (memoryCache.size > 4) {
+        const firstKey = memoryCache.keys().next().value;
+        if (firstKey)
+            memoryCache.delete(firstKey);
+    }
+}
 async function loadModelFromSettings() {
     try {
         const settingsPath = getSettingsPath();
@@ -116,24 +137,159 @@ function ensureDailyModelStats(dailyData, date, project, modelId) {
     };
     return dailyData[date][project][modelId];
 }
+function mergeSummaryTotals(target, src) {
+    for (const [k, v] of Object.entries(src)) {
+        target[k] ??= { cost: 0, tokens: 0, requests: 0 };
+        target[k].cost += v.cost;
+        target[k].tokens += v.tokens;
+        target[k].requests += v.requests;
+    }
+}
+function mergeGrandTotal(target, src) {
+    target.cost += src.cost;
+    target.tokens += src.tokens;
+    target.requests += src.requests;
+    target.cacheHitTokens += src.cacheHitTokens;
+    target.cacheMissTokens += src.cacheMissTokens;
+}
+function mergeDailyData(target, src) {
+    for (const [date, byProject] of Object.entries(src)) {
+        target[date] ??= {};
+        for (const [project, byModel] of Object.entries(byProject ?? {})) {
+            target[date][project] ??= {};
+            for (const [modelId, s] of Object.entries(byModel ?? {})) {
+                const t = ensureDailyModelStats(target, date, project, modelId);
+                t.cost += s.cost;
+                t.promptTokens += s.promptTokens;
+                t.completionTokens += s.completionTokens;
+                t.totalTokens += s.totalTokens;
+                t.cacheHitTokens += s.cacheHitTokens;
+                t.cacheMissTokens += s.cacheMissTokens;
+                t.cacheWriteTokens += s.cacheWriteTokens;
+                t.requests += s.requests;
+            }
+        }
+    }
+}
+function chunkArray(arr, chunks) {
+    const n = Math.max(1, Math.floor(chunks));
+    const out = Array.from({ length: n }, () => []);
+    for (let i = 0; i < arr.length; i++)
+        out[i % n].push(arr[i]);
+    return out.filter(a => a.length > 0);
+}
+async function parseCodeUsageWithWorkers(files, defaultModelId, minDate, workerCount) {
+    const chunks = chunkArray(files, Math.min(workerCount, files.length));
+    const workerUrl = new URL('./code-usage.worker.js', import.meta.url);
+    const results = await Promise.all(chunks.map(chunk => new Promise((resolve, reject) => {
+        const worker = new Worker(workerUrl, {
+            workerData: {
+                files: chunk,
+                defaultModelId,
+                minDate,
+            },
+        });
+        worker.once('message', msg => resolve(msg));
+        worker.once('error', err => reject(err));
+        worker.once('exit', code => {
+            if (code !== 0)
+                reject(new Error(`worker exited with code ${code}`));
+        });
+    })));
+    const merged = {
+        dailyData: {},
+        modelTotals: {},
+        projectTotals: {},
+        grandTotal: { cost: 0, tokens: 0, requests: 0, cacheHitTokens: 0, cacheMissTokens: 0 },
+    };
+    for (const r of results) {
+        mergeDailyData(merged.dailyData, r.dailyData);
+        mergeSummaryTotals(merged.modelTotals, r.modelTotals);
+        mergeSummaryTotals(merged.projectTotals, r.projectTotals);
+        mergeGrandTotal(merged.grandTotal, r.grandTotal);
+    }
+    return merged;
+}
 function finalizeAnalysis(defaultModelId, dailyData, modelTotals, projectTotals, grandTotal, workspaceMappings) {
+    const prof = process.env.CBS_PROFILE === '1';
+    const profLog = prof && (process.env.CBS_PROFILE_TUI === '1' || process.argv.includes('--no-tui'));
+    const t0 = prof ? performance.now() : 0;
     const dailySummary = {};
-    for (const date of Object.keys(dailyData)) {
-        dailySummary[date] = { cost: 0, tokens: 0, requests: 0 };
-        for (const project of Object.values(dailyData[date] ?? {})) {
-            for (const model of Object.values(project ?? {})) {
-                dailySummary[date].cost += model.cost;
-                dailySummary[date].tokens += model.totalTokens;
-                dailySummary[date].requests += model.requests;
+    const dailyTops = {};
+    const monthlyAgg = {};
+    const monthlyByProject = {};
+    // 一次性遍历 dailyData：生成 dailySummary + dailyTops + monthly 聚合
+    for (const [date, dayData] of Object.entries(dailyData)) {
+        const daySummary = { cost: 0, tokens: 0, requests: 0 };
+        let topModelId = '-';
+        let topModelCost = 0;
+        let topProjectName = '-';
+        let topProjectCost = 0;
+        const month = date.slice(0, 7);
+        if (month) {
+            monthlyAgg[month] ??= {
+                cost: 0,
+                tokens: 0,
+                requests: 0,
+                cacheHitTokens: 0,
+                cacheMissTokens: 0,
+                modelCost: {},
+                projectCost: {},
+            };
+            monthlyByProject[month] ??= {};
+        }
+        for (const [projectName, models] of Object.entries(dayData ?? {})) {
+            let projectCost = 0;
+            if (month) {
+                monthlyByProject[month][projectName] ??= { cost: 0, tokens: 0, requests: 0, modelCost: {} };
+            }
+            for (const [modelId, stats] of Object.entries(models ?? {})) {
+                const cost = stats.cost;
+                const tokens = stats.totalTokens;
+                const requests = stats.requests;
+                daySummary.cost += cost;
+                daySummary.tokens += tokens;
+                daySummary.requests += requests;
+                projectCost += cost;
+                if (cost > topModelCost) {
+                    topModelCost = cost;
+                    topModelId = modelId;
+                }
+                if (month) {
+                    const mAgg = monthlyAgg[month];
+                    mAgg.cost += cost;
+                    mAgg.tokens += tokens;
+                    mAgg.requests += requests;
+                    mAgg.cacheHitTokens += stats.cacheHitTokens;
+                    mAgg.cacheMissTokens += stats.cacheMissTokens;
+                    mAgg.modelCost[modelId] = (mAgg.modelCost[modelId] ?? 0) + cost;
+                    mAgg.projectCost[projectName] = (mAgg.projectCost[projectName] ?? 0) + cost;
+                    const pAgg = monthlyByProject[month][projectName];
+                    pAgg.cost += cost;
+                    pAgg.tokens += tokens;
+                    pAgg.requests += requests;
+                    pAgg.modelCost[modelId] = (pAgg.modelCost[modelId] ?? 0) + cost;
+                }
+            }
+            if (projectCost > topProjectCost) {
+                topProjectCost = projectCost;
+                topProjectName = projectName;
             }
         }
+        dailySummary[date] = daySummary;
+        dailyTops[date] = { topModelId, topModelCost, topProjectName, topProjectCost };
     }
+    const projectDisplayNames = {};
+    for (const projectName of Object.keys(projectTotals)) {
+        projectDisplayNames[projectName] = resolveProjectName(projectName, workspaceMappings);
+    }
+    const sortedMonths = Object.keys(monthlyAgg).sort().reverse();
     const topModelEntry = Object.entries(modelTotals).sort(compareByCostThenTokens)[0];
     const topProjectEntry = Object.entries(projectTotals).sort((a, b) => b[1].cost - a[1].cost)[0];
     const cacheHitRate = grandTotal.cacheHitTokens + grandTotal.cacheMissTokens > 0
         ? grandTotal.cacheHitTokens / (grandTotal.cacheHitTokens + grandTotal.cacheMissTokens)
         : 0;
-    return {
+    const out = {
         defaultModelId,
         dailyData,
         dailySummary,
@@ -145,22 +301,94 @@ function finalizeAnalysis(defaultModelId, dailyData, modelTotals, projectTotals,
         cacheHitRate,
         activeDays: Object.keys(dailyData).length,
         workspaceMappings,
+        projectDisplayNames,
+        dailyTops,
+        monthlyAgg,
+        monthlyByProject,
+        sortedMonths,
     };
+    if (profLog) {
+        const ms = performance.now() - t0;
+        console.error(`[cbs] finalizeAnalysis ${ms.toFixed(1)}ms (days=${Object.keys(dailyData).length}, projects=${Object.keys(projectTotals).length}, models=${Object.keys(modelTotals).length})`); // eslint-disable-line no-console
+    }
+    return out;
 }
 /**
  * 加载所有用量数据
  */
 export async function loadUsageData(options = {}) {
     const source = options.source ?? 'code';
-    if (source === 'ide') {
-        return loadIdeUsageData(options);
+    const prof = process.env.CBS_PROFILE === '1';
+    const profLog = prof && (process.env.CBS_PROFILE_TUI === '1' || process.argv.includes('--no-tui'));
+    const t0 = prof ? performance.now() : 0;
+    const data = source === 'ide' ? await loadIdeUsageData(options) : await loadCodeUsageData(options);
+    if (profLog) {
+        const ms = performance.now() - t0;
+        const daysInfo = options.days ? `days=${options.days}` : 'days=all';
+        // 用 stderr 避免影响 TUI 输出（建议配合 --no-tui 使用）
+        console.error(`[cbs] loadUsageData source=${source} ${daysInfo} ${ms.toFixed(1)}ms`); // eslint-disable-line no-console
     }
-    return loadCodeUsageData(options);
+    return data;
 }
 async function loadCodeUsageData(options = {}) {
+    const prof = process.env.CBS_PROFILE === '1';
+    const profLog = prof && (process.env.CBS_PROFILE_TUI === '1' || process.argv.includes('--no-tui'));
+    const t0 = prof ? performance.now() : 0;
+    const workerCountEnv = Number.parseInt(process.env.CBS_WORKERS ?? '', 10);
+    const workerCount = Number.isFinite(workerCountEnv)
+        ? Math.max(1, workerCountEnv)
+        : Math.min(4, Math.max(1, os.cpus().length - 1));
+    const useWorkersDefault = workerCount > 1;
+    const useWorkers = useWorkersDefault &&
+        // tsx/dev 场景下没有编译后的 worker 文件，自动回退
+        !import.meta.url.endsWith('.ts');
     const defaultModelId = await loadModelFromSettings();
     const jsonlFiles = await findJsonlFiles(BASE_DIR);
     const minDate = computeMinDate(options.days);
+    // 同进程内存缓存（不落盘）：refresh 时如果源文件未变，直接复用结果
+    const sortedFiles = [...jsonlFiles].sort();
+    const hash = createHash('sha1');
+    hash.update(`code|defaultModelId=${defaultModelId}|minDate=${minDate ?? ''}|files=${sortedFiles.length}\n`);
+    const nonEmptyFiles = [];
+    for (const filePath of sortedFiles) {
+        try {
+            const st = await fs.stat(filePath);
+            hash.update(`${filePath}|${st.size}|${st.mtimeMs}\n`);
+            if (st.size > 0)
+                nonEmptyFiles.push(filePath);
+        }
+        catch {
+            // 忽略暂时不可访问的文件
+        }
+    }
+    const fingerprint = hash.digest('hex');
+    const cached = cacheGet('code', options.days);
+    if (cached?.fingerprint === fingerprint) {
+        if (profLog) {
+            const ms = performance.now() - t0;
+            console.error(`[cbs] loadCodeUsageData cacheHit ${ms.toFixed(1)}ms (files=${sortedFiles.length})`); // eslint-disable-line no-console
+        }
+        return cached.data;
+    }
+    // 并行解析（不落盘）：多核并行 JSON.parse
+    if (useWorkers && nonEmptyFiles.length > 1) {
+        try {
+            const parsed = await parseCodeUsageWithWorkers(nonEmptyFiles, defaultModelId, minDate, workerCount);
+            const data = finalizeAnalysis(defaultModelId, parsed.dailyData, parsed.modelTotals, parsed.projectTotals, parsed.grandTotal);
+            cacheSet('code', options.days, { fingerprint, data });
+            if (profLog) {
+                const ms = performance.now() - t0;
+                console.error(`[cbs] loadCodeUsageData workers=${Math.min(workerCount, nonEmptyFiles.length)} ${ms.toFixed(1)}ms (files=${sortedFiles.length})`); // eslint-disable-line no-console
+            }
+            return data;
+        }
+        catch (err) {
+            if (profLog) {
+                console.error(`[cbs] loadCodeUsageData workersFailed; fallback to single-thread: ${err?.message ?? String(err)}`); // eslint-disable-line no-console
+            }
+            // fallback: 单线程解析
+        }
+    }
     const dailyData = {};
     const modelTotals = {};
     const projectTotals = {};
@@ -171,10 +399,7 @@ async function loadCodeUsageData(options = {}) {
         cacheHitTokens: 0,
         cacheMissTokens: 0,
     };
-    for (const filePath of jsonlFiles) {
-        const fileStat = await fs.stat(filePath);
-        if (fileStat.size === 0)
-            continue;
+    for (const filePath of nonEmptyFiles) {
         const fileStream = fsSync.createReadStream(filePath);
         const rl = createInterface({
             input: fileStream,
@@ -228,7 +453,13 @@ async function loadCodeUsageData(options = {}) {
             }
         }
     }
-    return finalizeAnalysis(defaultModelId, dailyData, modelTotals, projectTotals, grandTotal);
+    const data = finalizeAnalysis(defaultModelId, dailyData, modelTotals, projectTotals, grandTotal);
+    cacheSet('code', options.days, { fingerprint, data });
+    if (profLog) {
+        const ms = performance.now() - t0;
+        console.error(`[cbs] loadCodeUsageData ${ms.toFixed(1)}ms (files=${sortedFiles.length})`); // eslint-disable-line no-console
+    }
+    return data;
 }
 async function findIdeHistoryDirs() {
     const root = getIdeDataDir();
@@ -329,6 +560,9 @@ async function inferIdeModelIdForRequest(conversationDir, request, messageModelC
     return null;
 }
 async function loadIdeUsageData(options = {}) {
+    const prof = process.env.CBS_PROFILE === '1';
+    const profLog = prof && (process.env.CBS_PROFILE_TUI === '1' || process.argv.includes('--no-tui'));
+    const t0 = prof ? performance.now() : 0;
     const defaultModelId = await loadModelFromSettings();
     const minDate = computeMinDate(options.days);
     // 加载工作区映射
@@ -344,6 +578,48 @@ async function loadIdeUsageData(options = {}) {
         cacheMissTokens: 0,
     };
     const historyDirs = await findIdeHistoryDirs();
+    // 同进程内存缓存（不落盘）：refresh 时如果 IDE 索引未变，直接复用结果
+    const sortedHistoryDirs = [...historyDirs].sort();
+    const ideHash = createHash('sha1');
+    ideHash.update(`ide|defaultModelId=${defaultModelId}|minDate=${minDate ?? ''}|historyDirs=${sortedHistoryDirs.length}\n`);
+    const mappingKeys = [...workspaceMappings.keys()].sort();
+    ideHash.update(`mappings=${mappingKeys.length}\n`);
+    for (const k of mappingKeys) {
+        const v = workspaceMappings.get(k);
+        ideHash.update(`${k}|${v?.displayPath ?? ''}\n`);
+    }
+    for (const historyDir of sortedHistoryDirs) {
+        ideHash.update(`dir=${historyDir}\n`);
+        let workspaces = [];
+        try {
+            workspaces = await fs.readdir(historyDir, { withFileTypes: true });
+        }
+        catch {
+            continue;
+        }
+        for (const ws of workspaces) {
+            if (!ws.isDirectory())
+                continue;
+            const workspaceDir = path.join(historyDir, ws.name);
+            const workspaceIndexPath = path.join(workspaceDir, 'index.json');
+            try {
+                const st = await fs.stat(workspaceIndexPath);
+                ideHash.update(`${ws.name}|${st.size}|${st.mtimeMs}\n`);
+            }
+            catch {
+                // ignore
+            }
+        }
+    }
+    const fingerprint = ideHash.digest('hex');
+    const cached = cacheGet('ide', options.days);
+    if (cached?.fingerprint === fingerprint) {
+        if (profLog) {
+            const ms = performance.now() - t0;
+            console.error(`[cbs] loadIdeUsageData cacheHit ${ms.toFixed(1)}ms (historyDirs=${sortedHistoryDirs.length})`); // eslint-disable-line no-console
+        }
+        return cached.data;
+    }
     const messageModelCache = new Map();
     for (const historyDir of historyDirs) {
         let workspaces = [];
@@ -434,5 +710,11 @@ async function loadIdeUsageData(options = {}) {
             }
         }
     }
-    return finalizeAnalysis(defaultModelId, dailyData, modelTotals, projectTotals, grandTotal, workspaceMappings);
+    const data = finalizeAnalysis(defaultModelId, dailyData, modelTotals, projectTotals, grandTotal, workspaceMappings);
+    cacheSet('ide', options.days, { fingerprint, data });
+    if (profLog) {
+        const ms = performance.now() - t0;
+        console.error(`[cbs] loadIdeUsageData ${ms.toFixed(1)}ms (historyDirs=${historyDirs.length})`); // eslint-disable-line no-console
+    }
+    return data;
 }

package/dist/lib/pricing.js

@@ -31,6 +31,24 @@ export const MODEL_PRICING = {
     "gpt-5.1-codex-mini": createPricing(0.25, 0.025, 2.0),
     "gpt-5-codex": createPricing(1.25, 0.125, 10.0),
     // Claude 系列
+    "claude-opus-4.6": {
+        prompt: [
+            { limit: 200_000, pricePerMTok: 5.0 },
+            { limit: Number.POSITIVE_INFINITY, pricePerMTok: 10.0 },
+        ],
+        completion: [
+            { limit: 200_000, pricePerMTok: 25.0 },
+            { limit: Number.POSITIVE_INFINITY, pricePerMTok: 37.5 },
+        ],
+        cacheRead: [
+            { limit: 200_000, pricePerMTok: 0.5 },
+            { limit: Number.POSITIVE_INFINITY, pricePerMTok: 1.0 },
+        ],
+        cacheWrite: [
+            { limit: 200_000, pricePerMTok: 6.25 },
+            { limit: Number.POSITIVE_INFINITY, pricePerMTok: 12.5 },
+        ],
+    },
     "claude-opus-4.5": createPricing(5.0, 0.5, 25.0, 10.0),
     "claude-haiku-4.5": createPricing(1.0, 0.1, 5.0, 1.25),
     "claude-4.5": {

package/dist/views/by-project.js

@@ -20,7 +20,7 @@ export function renderByProject(box, data, scrollOffset = 0, width, note, pageSi
     const visibleProjects = sorted.slice(scrollOffset, scrollOffset + safePageSize);
     for (const [projectName, stats] of visibleProjects) {
         // 简化项目名
-        const shortName = resolveProjectName(projectName, data.workspaceMappings);
+        const shortName = data.projectDisplayNames?.[projectName] ?? resolveProjectName(projectName, data.workspaceMappings);
         content +=
             truncate(shortName, projectCol - 1).padEnd(projectCol) +
                 formatCost(stats.cost).padStart(12) +

package/dist/views/daily-detail.js

@@ -10,7 +10,7 @@ export function renderDailyDetail(box, data, date, scrollOffset = 0, width, page
     }
     const projectDetails = [];
     for (const [projectName, models] of Object.entries(dayData)) {
-        const shortName = resolveProjectName(projectName, data.workspaceMappings);
+        const shortName = data.projectDisplayNames?.[projectName] ?? resolveProjectName(projectName, data.workspaceMappings);
         const modelList = [];
         let totalCost = 0;
         let totalTokens = 0;

package/dist/views/daily.js

@@ -31,23 +31,31 @@ export function renderDaily(box, data, scrollOffset = 0, selectedIndex = 0, widt
         const dayData = dailyData[date];
         if (!daySummary || !dayData)
             continue;
-        // 找出当天 top model 和 project
+        // 找出当天 top model 和 project（优先使用预计算缓存）
         let topModel = { id: '-', cost: 0 };
         let topProject = { name: '-', cost: 0 };
-        for (const [project, models] of Object.entries(dayData)) {
-            let projectCost = 0;
-            for (const [model, stats] of Object.entries(models)) {
-                const modelStats = stats;
-                projectCost += Number(modelStats.cost ?? 0);
-                if (Number(modelStats.cost ?? 0) > topModel.cost) {
-                    topModel = { id: model, cost: Number(modelStats.cost ?? 0) };
+        const cachedTop = data.dailyTops?.[date];
+        if (cachedTop) {
+            topModel = { id: cachedTop.topModelId || '-', cost: cachedTop.topModelCost || 0 };
+            topProject = { name: cachedTop.topProjectName || '-', cost: cachedTop.topProjectCost || 0 };
+        }
+        else {
+            for (const [project, models] of Object.entries(dayData)) {
+                let projectCost = 0;
+                for (const [model, stats] of Object.entries(models)) {
+                    const modelStats = stats;
+                    const c = Number(modelStats.cost ?? 0);
+                    projectCost += c;
+                    if (c > topModel.cost) {
+                        topModel = { id: model, cost: c };
+                    }
+                }
+                if (projectCost > topProject.cost) {
+                    topProject = { name: project, cost: projectCost };
                 }
-            }
-            if (projectCost > topProject.cost) {
-                topProject = { name: project, cost: projectCost };
             }
         }
-        const shortProject = resolveProjectName(topProject.name, data.workspaceMappings);
+        const shortProject = data.projectDisplayNames?.[topProject.name] ?? resolveProjectName(topProject.name, data.workspaceMappings);
         const isSelected = scrollOffset + i === selectedIndex;
         const rowContent = date.padEnd(dateCol) +
             formatCost(daySummary.cost).padStart(costCol) +

package/dist/views/monthly-detail.js

@@ -1,62 +1,99 @@
 import { resolveProjectName } from '../lib/workspace-resolver.js';
 import { formatCost, formatNumber, formatPercent, formatTokens, truncate } from '../lib/utils.js';
 export function renderMonthlyDetail(box, data, month, scrollOffset = 0, width, pageSize) {
-    const aggByProject = {};
+    let projects = [];
     let totalCost = 0;
     let totalTokens = 0;
     let totalRequests = 0;
-    let cacheHitTokens = 0;
-    let cacheMissTokens = 0;
-    for (const [date, dayData] of Object.entries(data.dailyData)) {
-        if (!date.startsWith(month))
-            continue;
-        for (const [projectName, models] of Object.entries(dayData ?? {})) {
-            aggByProject[projectName] ??= {
+    let cacheHitRate = 0;
+    const cachedMonthAgg = data.monthlyAgg?.[month];
+    const cachedByProject = data.monthlyByProject?.[month];
+    if (cachedMonthAgg && cachedByProject) {
+        totalCost = cachedMonthAgg.cost;
+        totalTokens = cachedMonthAgg.tokens;
+        totalRequests = cachedMonthAgg.requests;
+        cacheHitRate =
+            cachedMonthAgg.cacheHitTokens + cachedMonthAgg.cacheMissTokens > 0
+                ? cachedMonthAgg.cacheHitTokens / (cachedMonthAgg.cacheHitTokens + cachedMonthAgg.cacheMissTokens)
+                : 0;
+        projects = Object.entries(cachedByProject)
+            .map(([projectName, agg]) => {
+            let topModelId = '-';
+            let topModelCost = 0;
+            for (const [modelId, c] of Object.entries(agg.modelCost)) {
+                if (c > topModelCost) {
+                    topModelCost = c;
+                    topModelId = modelId;
+                }
+            }
+            const shortName = data.projectDisplayNames?.[projectName] ?? resolveProjectName(projectName, data.workspaceMappings);
+            return {
                 name: projectName,
-                shortName: resolveProjectName(projectName, data.workspaceMappings),
-                cost: 0,
-                tokens: 0,
-                requests: 0,
-                modelCost: {},
+                shortName,
+                cost: agg.cost,
+                tokens: agg.tokens,
+                requests: agg.requests,
+                modelCost: agg.modelCost,
+                topModelId,
             };
-            const p = aggByProject[projectName];
-            for (const [modelId, stats] of Object.entries(models ?? {})) {
-                const s = stats;
-                const cost = Number(s.cost ?? 0);
-                const tokens = Number(s.totalTokens ?? 0);
-                const requests = Number(s.requests ?? 0);
-                const cacheHit = Number(s.cacheHitTokens ?? 0);
-                const cacheMiss = Number(s.cacheMissTokens ?? 0);
-                p.cost += cost;
-                p.tokens += tokens;
-                p.requests += requests;
-                p.modelCost[modelId] = (p.modelCost[modelId] ?? 0) + cost;
-                totalCost += cost;
-                totalTokens += tokens;
-                totalRequests += requests;
-                cacheHitTokens += cacheHit;
-                cacheMissTokens += cacheMiss;
-            }
-        }
+        })
+            .sort((a, b) => b.cost - a.cost);
     }
-    const projects = Object.values(aggByProject)
-        .map(p => {
-        let topModelId = '-';
-        let topModelCost = 0;
-        for (const [modelId, c] of Object.entries(p.modelCost)) {
-            if (c > topModelCost) {
-                topModelCost = c;
-                topModelId = modelId;
+    else {
+        const aggByProject = {};
+        let cacheHitTokens = 0;
+        let cacheMissTokens = 0;
+        for (const [date, dayData] of Object.entries(data.dailyData)) {
+            if (!date.startsWith(month))
+                continue;
+            for (const [projectName, models] of Object.entries(dayData ?? {})) {
+                aggByProject[projectName] ??= {
+                    name: projectName,
+                    shortName: data.projectDisplayNames?.[projectName] ?? resolveProjectName(projectName, data.workspaceMappings),
+                    cost: 0,
+                    tokens: 0,
+                    requests: 0,
+                    modelCost: {},
+                };
+                const p = aggByProject[projectName];
+                for (const [modelId, stats] of Object.entries(models ?? {})) {
+                    const s = stats;
+                    const cost = Number(s.cost ?? 0);
+                    const tokens = Number(s.totalTokens ?? 0);
+                    const requests = Number(s.requests ?? 0);
+                    const cacheHit = Number(s.cacheHitTokens ?? 0);
+                    const cacheMiss = Number(s.cacheMissTokens ?? 0);
+                    p.cost += cost;
+                    p.tokens += tokens;
+                    p.requests += requests;
+                    p.modelCost[modelId] = (p.modelCost[modelId] ?? 0) + cost;
+                    totalCost += cost;
+                    totalTokens += tokens;
+                    totalRequests += requests;
+                    cacheHitTokens += cacheHit;
+                    cacheMissTokens += cacheMiss;
+                }
             }
         }
-        return { ...p, topModelId };
-    })
-        .sort((a, b) => b.cost - a.cost);
+        projects = Object.values(aggByProject)
+            .map(p => {
+            let topModelId = '-';
+            let topModelCost = 0;
+            for (const [modelId, c] of Object.entries(p.modelCost)) {
+                if (c > topModelCost) {
+                    topModelCost = c;
+                    topModelId = modelId;
+                }
+            }
+            return { ...p, topModelId };
+        })
+            .sort((a, b) => b.cost - a.cost);
+        cacheHitRate = cacheHitTokens + cacheMissTokens > 0 ? cacheHitTokens / (cacheHitTokens + cacheMissTokens) : 0;
+    }
     if (!projects.length) {
         box.setContent(`{bold}${month}{/bold}\n\nNo data available for this month.`);
         return;
     }
-    const cacheHitRate = cacheHitTokens + cacheMissTokens > 0 ? cacheHitTokens / (cacheHitTokens + cacheMissTokens) : 0;
     // 根据宽度计算列宽
     const availableWidth = width - 6; // padding
     const costCol = 12;

package/dist/views/monthly.js

@@ -1,40 +1,42 @@
 import { resolveProjectName } from '../lib/workspace-resolver.js';
 import { formatCost, formatNumber, formatTokens, truncate } from '../lib/utils.js';
 export function renderMonthly(box, data, scrollOffset = 0, selectedIndex = 0, width, note, pageSize) {
-    const aggByMonth = {};
-    for (const [date, dayData] of Object.entries(data.dailyData)) {
-        const month = date.slice(0, 7);
-        if (!month)
-            continue;
-        aggByMonth[month] ??= {
-            cost: 0,
-            tokens: 0,
-            requests: 0,
-            cacheHitTokens: 0,
-            cacheMissTokens: 0,
-            modelCost: {},
-            projectCost: {},
-        };
-        const monthAgg = aggByMonth[month];
-        for (const [projectName, models] of Object.entries(dayData ?? {})) {
-            for (const [modelId, stats] of Object.entries(models ?? {})) {
-                const s = stats;
-                const cost = Number(s.cost ?? 0);
-                const tokens = Number(s.totalTokens ?? 0);
-                const requests = Number(s.requests ?? 0);
-                const cacheHit = Number(s.cacheHitTokens ?? 0);
-                const cacheMiss = Number(s.cacheMissTokens ?? 0);
-                monthAgg.cost += cost;
-                monthAgg.tokens += tokens;
-                monthAgg.requests += requests;
-                monthAgg.cacheHitTokens += cacheHit;
-                monthAgg.cacheMissTokens += cacheMiss;
-                monthAgg.modelCost[modelId] = (monthAgg.modelCost[modelId] ?? 0) + cost;
-                monthAgg.projectCost[projectName] = (monthAgg.projectCost[projectName] ?? 0) + cost;
+    const aggByMonth = data.monthlyAgg ?? {};
+    if (!data.monthlyAgg) {
+        for (const [date, dayData] of Object.entries(data.dailyData)) {
+            const month = date.slice(0, 7);
+            if (!month)
+                continue;
+            aggByMonth[month] ??= {
+                cost: 0,
+                tokens: 0,
+                requests: 0,
+                cacheHitTokens: 0,
+                cacheMissTokens: 0,
+                modelCost: {},
+                projectCost: {},
+            };
+            const monthAgg = aggByMonth[month];
+            for (const [projectName, models] of Object.entries(dayData ?? {})) {
+                for (const [modelId, stats] of Object.entries(models ?? {})) {
+                    const s = stats;
+                    const cost = Number(s.cost ?? 0);
+                    const tokens = Number(s.totalTokens ?? 0);
+                    const requests = Number(s.requests ?? 0);
+                    const cacheHit = Number(s.cacheHitTokens ?? 0);
+                    const cacheMiss = Number(s.cacheMissTokens ?? 0);
+                    monthAgg.cost += cost;
+                    monthAgg.tokens += tokens;
+                    monthAgg.requests += requests;
+                    monthAgg.cacheHitTokens += cacheHit;
+                    monthAgg.cacheMissTokens += cacheMiss;
+                    monthAgg.modelCost[modelId] = (monthAgg.modelCost[modelId] ?? 0) + cost;
+                    monthAgg.projectCost[projectName] = (monthAgg.projectCost[projectName] ?? 0) + cost;
+                }
             }
         }
     }
-    const sortedMonths = Object.keys(aggByMonth).sort().reverse();
+    const sortedMonths = data.sortedMonths ?? Object.keys(aggByMonth).sort().reverse();
     // 根据宽度计算列宽
     const availableWidth = width - 6; // padding
     const monthCol = 9;
@@ -73,7 +75,7 @@ export function renderMonthly(box, data, scrollOffset = 0, selectedIndex = 0, wi
             if (c > topProject.cost)
                 topProject = { name: projectName, cost: c };
         }
-        const shortProject = resolveProjectName(topProject.name, data.workspaceMappings);
+        const shortProject = data.projectDisplayNames?.[topProject.name] ?? resolveProjectName(topProject.name, data.workspaceMappings);
         const isSelected = scrollOffset + i === selectedIndex;
         const rowContent = month.padEnd(monthCol) +
             formatCost(m.cost).padStart(costCol) +

package/dist/views/overview.js

@@ -170,7 +170,7 @@ export function renderOverview(box, data, width, height, note) {
         }
         if (topProject) {
             const label = '{cyan-fg}Top project:{/cyan-fg} ';
-            const shortName = resolveProjectName(topProject.name, data.workspaceMappings);
+            const shortName = data.projectDisplayNames?.[topProject.name] ?? resolveProjectName(topProject.name, data.workspaceMappings);
             const tail = `(${formatCost(topProject.cost)})`;
             const maxNameLen = Math.max(4, maxW - visibleLen(label) - visibleLen(tail) - 2);
             lines.push(label + truncate(shortName, maxNameLen) + ' ' + tail);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codebuddy-stats",
-  "version": "1.3.1",
+  "version": "1.3.3",
   "type": "module",
   "main": "dist/index.js",
   "files": [