codebuddy-stats 1.0.0 → 1.1.1

package/README.md

@@ -0,0 +1,120 @@
+# CodeBuddy Stats
+
+一个用于分析 CodeBuddy 系列产品使用成本的命令行工具，支持交互式 TUI 界面和纯文本输出。
+
+## 功能特性
+
+- **双数据源支持** - 同时支持 CodeBuddy Code 和 CodeBuddy IDE（VS Code 扩展）
+- **成本热力图** - 可视化每日 AI 使用成本分布
+- **模型统计** - 按模型分类的费用、请求数、Token 用量
+- **项目统计** - 按项目分类的费用汇总
+- **每日明细** - 查看每日详细使用情况
+- **缓存命中率** - 显示 prompt cache 命中率（Code 模式）
+- **多模型定价** - 支持 GPT-5.2、Claude 4.5、Gemini 等模型
+
+## 安装
+
+```bash
+npm install -g codebuddy-stats
+```
+
+## 使用方法
+
+```bash
+# 启动交互式 TUI 界面
+cbs
+
+# 或使用完整命令名
+codebuddy-stats
+
+# 纯文本输出模式
+cbs --no-tui
+
+# 只显示最近 7 天的数据
+cbs --days 7
+
+# 显示帮助
+cbs --help
+```
+
+## TUI 界面操作
+
+| 按键        | 功能                     |
+| ----------- | ------------------------ |
+| `Tab`       | 切换到下一个视图         |
+| `Shift+Tab` | 切换到上一个视图         |
+| `s`         | 切换数据源（Code / IDE） |
+| `↑` / `k`   | 向上滚动 (Daily 视图)    |
+| `↓` / `j`   | 向下滚动 (Daily 视图)    |
+| `r`         | 刷新数据                 |
+| `q`         | 退出                     |
+
+## 视图说明
+
+### Overview
+
+显示成本热力图和汇总统计，包括：
+
+- 总费用、总 Token 数、总请求数
+- 活跃天数、缓存命中率、日均费用
+- 使用最多的模型和项目
+
+### By Model
+
+按 AI 模型分类的详细统计表格，包含每个模型的费用、请求数、Token 数和平均每次请求费用。
+
+### By Project
+
+按项目分类的费用统计，方便了解不同项目的 AI 使用成本。
+
+工具会自动将项目标识解析为可读路径：
+- **Code 模式**: 将 `Users-anoti-Documents-project-xxx` 还原为 `~/Documents/project/xxx`
+- **IDE 模式**: 将 MD5 hash 映射为实际工作区路径
+
+### Daily
+
+每日使用明细，显示日期、费用、请求数以及当天使用最多的模型和项目。
+
+## 支持的模型
+
+| 模型            | 输入价格 | 输出价格 |
+| --------------- | -------- | -------- |
+| GPT-5.2         | $1.75/M  | $14.00/M |
+| GPT-5.1 / GPT-5 | $1.25/M  | $10.00/M |
+| GPT-5-mini      | $0.25/M  | $2.00/M  |
+| GPT-5-nano      | $0.05/M  | $0.40/M  |
+| Claude Opus 4.5 | $5.00/M  | $25.00/M |
+| Claude 4.5      | $3.00/M  | $15.00/M |
+| Gemini 3 Pro    | $2.00/M  | $12.00/M |
+| Gemini 2.5 Pro  | $1.25/M  | $10.00/M |
+
+_价格单位：USD / 1M tokens，部分模型支持分层定价_
+
+## 数据来源
+
+工具支持两种数据源，可在 TUI 界面中按 `s` 键切换：
+
+### CodeBuddy Code（CLI 版本）
+
+- **macOS**: `~/.codebuddy/projects/`
+- **Windows**: `%APPDATA%/CodeBuddy/projects/`
+- **Linux**: `$XDG_CONFIG_HOME/codebuddy/projects/` 或 `~/.codebuddy/projects/`
+
+特点：包含完整的缓存命中/写入 token 数据，可计算缓存命中率和精确成本。
+
+### CodeBuddy IDE（VS Code 扩展）
+
+- **macOS**: `~/Library/Application Support/CodeBuddyExtension/Data/`
+- **Windows**: `%APPDATA%/CodeBuddyExtension/Data/`
+- **Linux**: `$XDG_CONFIG_HOME/CodeBuddyExtension/Data/` 或 `~/.config/CodeBuddyExtension/Data/`
+
+特点：基于 input/output tokens 估算成本，不包含缓存相关数据。
+
+## 系统要求
+
+- Node.js >= 18
+- 终端支持 Unicode 字符（用于热力图显示）
+
+## License
+
+ISC

package/dist/index.js

@@ -1,8 +1,16 @@
 #!/usr/bin/env node
+import fs from 'node:fs';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
 import blessed from 'blessed';
 import { loadUsageData } from './lib/data-loader.js';
-import { shortenProjectName } from './lib/paths.js';
+import { resolveProjectName } from './lib/workspace-resolver.js';
 import { formatCost, formatNumber, formatPercent, formatTokens, truncate } from './lib/utils.js';
+// 读取 package.json 获取版本号
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const pkgPath = path.resolve(__dirname, '../package.json');
+const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf8'));
+const VERSION = pkg.version;
 // 解析命令行参数
 function parseArgs() {
     const args = process.argv.slice(2);
@@ -59,7 +67,7 @@ function getHeatChar(cost, maxCost) {
     return '█';
 }
 // 渲染 Overview 视图
-function renderOverview(box, data, width) {
+function renderOverview(box, data, width, note) {
     const { dailySummary, grandTotal, topModel, topProject, cacheHitRate, activeDays } = data;
     const heatmap = generateHeatmapData(dailySummary);
     // 根据宽度计算热力图周数
@@ -135,13 +143,16 @@ function renderOverview(box, data, width) {
     }
     if (topProject) {
         const projectMaxLen = width >= 100 ? 60 : 35;
-        const shortName = shortenProjectName(topProject.name);
+        const shortName = resolveProjectName(topProject.name, data.workspaceMappings);
         content += `{cyan-fg}Top project:{/cyan-fg}      ${truncate(shortName, projectMaxLen)} (${formatCost(topProject.cost)})\n`;
     }
+    if (note) {
+        content += `\n{gray-fg}备注：${note}{/gray-fg}\n`;
+    }
     box.setContent(content);
 }
 // 渲染 By Model 视图
-function renderByModel(box, data, width) {
+function renderByModel(box, data, width, note) {
     const { modelTotals, grandTotal } = data;
     const sorted = Object.entries(modelTotals).sort((a, b) => b[1].cost - a[1].cost);
     // 根据宽度计算列宽
@@ -176,10 +187,13 @@ function renderByModel(box, data, width) {
             formatNumber(grandTotal.requests).padStart(12) +
             formatTokens(grandTotal.tokens).padStart(12) +
             '{/bold}\n';
+    if (note) {
+        content += `\n{gray-fg}备注：${note}{/gray-fg}\n`;
+    }
     box.setContent(content);
 }
 // 渲染 By Project 视图
-function renderByProject(box, data, width) {
+function renderByProject(box, data, width, note) {
     const { projectTotals, grandTotal } = data;
     const sorted = Object.entries(projectTotals).sort((a, b) => b[1].cost - a[1].cost);
     // 根据宽度计算列宽
@@ -197,7 +211,7 @@ function renderByProject(box, data, width) {
             '{/underline}\n';
     for (const [projectName, stats] of sorted) {
         // 简化项目名
-        const shortName = shortenProjectName(projectName);
+        const shortName = resolveProjectName(projectName, data.workspaceMappings);
         content +=
             truncate(shortName, projectCol - 1).padEnd(projectCol) +
                 formatCost(stats.cost).padStart(12) +
@@ -213,10 +227,13 @@ function renderByProject(box, data, width) {
             formatNumber(grandTotal.requests).padStart(12) +
             formatTokens(grandTotal.tokens).padStart(12) +
             '{/bold}\n';
+    if (note) {
+        content += `\n{gray-fg}备注：${note}{/gray-fg}\n`;
+    }
     box.setContent(content);
 }
 // 渲染 Daily 视图
-function renderDaily(box, data, scrollOffset = 0, width) {
+function renderDaily(box, data, scrollOffset = 0, width, note) {
     const { dailySummary, dailyData } = data;
     const sortedDates = Object.keys(dailySummary).sort().reverse();
     // 根据宽度计算列宽
@@ -259,7 +276,7 @@ function renderDaily(box, data, scrollOffset = 0, width) {
                 topProject = { name: project, cost: projectCost };
             }
         }
-        const shortProject = shortenProjectName(topProject.name);
+        const shortProject = resolveProjectName(topProject.name, data.workspaceMappings);
         content +=
             date.padEnd(dateCol) +
                 formatCost(daySummary.cost).padStart(costCol) +
@@ -271,6 +288,9 @@ function renderDaily(box, data, scrollOffset = 0, width) {
     if (sortedDates.length > 20) {
         content += `\n{gray-fg}Showing ${scrollOffset + 1}-${Math.min(scrollOffset + 20, sortedDates.length)} of ${sortedDates.length} days (↑↓ to scroll){/gray-fg}`;
     }
+    if (note) {
+        content += `\n\n{gray-fg}备注：${note}{/gray-fg}\n`;
+    }
     box.setContent(content);
 }
 // 纯文本输出模式
@@ -287,7 +307,8 @@ function printTextReport(data) {
         console.log(`\nTop model:         ${topModel.id} (${formatCost(topModel.cost)})`);
     }
     if (topProject) {
-        console.log(`Top project:       ${topProject.name}`);
+        const shortName = resolveProjectName(topProject.name, data.workspaceMappings);
+        console.log(`Top project:       ${shortName}`);
         console.log(`                   (${formatCost(topProject.cost)})`);
     }
     console.log('\n' + '-'.repeat(50));
@@ -300,7 +321,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 = shortenProjectName(project);
+        const shortName = resolveProjectName(project, data.workspaceMappings);
         console.log(`  ${truncate(shortName, 40)}: ${formatCost(stats.cost)}`); // eslint-disable-line no-console
     }
     console.log('\n' + '='.repeat(50) + '\n');
@@ -309,7 +330,8 @@ function printTextReport(data) {
 async function main() {
     const options = parseArgs();
     console.log('Loading data...');
-    let data = await loadUsageData({ days: options.days });
+    let currentSource = 'code';
+    let data = await loadUsageData({ days: options.days, source: currentSource });
     if (options.noTui) {
         printTextReport(data);
         return;
@@ -318,6 +340,8 @@ async function main() {
     const screen = blessed.screen({
         smartCSR: true,
         title: 'CodeBuddy Cost Analyzer',
+        forceUnicode: true,
+        fullUnicode: true,
     });
     // Tab 状态
     const tabs = ['Overview', 'By Model', 'By Project', 'Daily'];
@@ -374,6 +398,16 @@ async function main() {
     // 更新 Tab 栏
     function updateTabBar() {
         let content = ' Cost Analysis  ';
+        content += '{gray-fg}Source:{/gray-fg} ';
+        if (currentSource === 'code') {
+            content += '{black-fg}{green-bg} Code {/green-bg}{/black-fg} ';
+            content += '{gray-fg}IDE{/gray-fg}  ';
+        }
+        else {
+            content += '{gray-fg}Code{/gray-fg} ';
+            content += '{black-fg}{green-bg} IDE {/green-bg}{/black-fg}  ';
+        }
+        content += '{gray-fg}Views:{/gray-fg} ';
         for (let i = 0; i < tabs.length; i++) {
             if (i === currentTab) {
                 content += `{black-fg}{green-bg} ${tabs[i]} {/green-bg}{/black-fg} `;
@@ -382,31 +416,39 @@ async function main() {
                 content += `{gray-fg}${tabs[i]}{/gray-fg} `;
             }
         }
-        content += ' {gray-fg}(Tab to switch){/gray-fg}';
+        content += ' {gray-fg}(Tab view, s source){/gray-fg}';
         tabBar.setContent(content);
     }
     // 更新内容
     function updateContent() {
         const width = Number(screen.width) || 80;
+        const note = currentSource === 'code'
+            ? `针对 CodeBuddy Code ≤ 2.20.0 版本产生的数据，由于没有请求级别的 model ID，用量是基于当前 CodeBuddy Code 设置的 model ID（${data.defaultModelId}）计算价格的`
+            : 'IDE 的 usage 不包含缓存命中/写入 tokens，无法计算缓存相关价格与命中率；成本按 input/output tokens 估算';
         switch (currentTab) {
             case 0:
-                renderOverview(contentBox, data, width);
+                renderOverview(contentBox, data, width, note);
                 break;
             case 1:
-                renderByModel(contentBox, data, width);
+                renderByModel(contentBox, data, width, note);
                 break;
             case 2:
-                renderByProject(contentBox, data, width);
+                renderByProject(contentBox, data, width, note);
                 break;
             case 3:
-                renderDaily(contentBox, data, dailyScrollOffset, width);
+                renderDaily(contentBox, data, dailyScrollOffset, width, note);
                 break;
         }
     }
     // 更新状态栏
     function updateStatusBar() {
         const daysInfo = options.days ? `Last ${options.days} days` : 'All time';
-        statusBar.setContent(` ${daysInfo} | Total: ${formatCost(data.grandTotal.cost)} | q quit, Tab switch, r refresh`);
+        const sourceInfo = currentSource === 'code' ? 'Code' : 'IDE';
+        const leftContent = ` ${daysInfo} | Source: ${sourceInfo} | Total: ${formatCost(data.grandTotal.cost)} | q quit, Tab view, s source, r refresh`;
+        const rightContent = `v${VERSION} `;
+        const width = Number(screen.width) || 80;
+        const padding = Math.max(0, width - leftContent.length - rightContent.length);
+        statusBar.setContent(leftContent + ' '.repeat(padding) + rightContent);
     }
     // 键盘事件
     screen.key(['tab'], () => {
@@ -446,7 +488,7 @@ async function main() {
         statusBar.setContent(' {yellow-fg}Reloading...{/yellow-fg}');
         screen.render();
         try {
-            data = await loadUsageData({ days: options.days });
+            data = await loadUsageData({ days: options.days, source: currentSource });
             dailyScrollOffset = 0;
             updateTabBar();
             updateContent();
@@ -457,6 +499,22 @@ async function main() {
         }
         screen.render();
     });
+    screen.key(['s'], async () => {
+        statusBar.setContent(' {yellow-fg}Switching source...{/yellow-fg}');
+        screen.render();
+        try {
+            currentSource = currentSource === 'code' ? 'ide' : 'code';
+            data = await loadUsageData({ days: options.days, source: currentSource });
+            dailyScrollOffset = 0;
+            updateTabBar();
+            updateContent();
+            updateStatusBar();
+        }
+        catch (err) {
+            statusBar.setContent(` {red-fg}Switch source failed: ${String(err)}{/red-fg}`);
+        }
+        screen.render();
+    });
     // 监听窗口大小变化
     screen.on('resize', () => {
         updateContent();
@@ -472,4 +530,3 @@ main().catch(err => {
     console.error('Error:', err);
     process.exit(1);
 });
-//# sourceMappingURL=index.js.map