codebuddy-stats 1.0.0 → 1.1.0

package/dist/lib/workspace-resolver.js

@@ -0,0 +1,211 @@
+import fs from 'node:fs/promises';
+import fsSync from 'node:fs';
+import path from 'node:path';
+import os from 'node:os';
+import crypto from 'node:crypto';
+import { getWorkspaceStorageDir } from './paths.js';
+// 缓存已解析的 CodeBuddy Code 路径名
+const codePathCache = new Map();
+/**
+ * 从 folder URI 提取纯路径（用于计算 MD5）
+ */
+function extractPathFromUri(folderUri) {
+    // 处理本地文件路径: file:///path/to/folder
+    if (folderUri.startsWith('file://')) {
+        try {
+            const url = new URL(folderUri);
+            return decodeURIComponent(url.pathname);
+        }
+        catch {
+            return decodeURIComponent(folderUri.replace('file://', ''));
+        }
+    }
+    // 处理远程路径: vscode-remote://codebuddy-remote-ssh%2B.../path
+    if (folderUri.startsWith('vscode-remote://')) {
+        try {
+            const url = new URL(folderUri);
+            return decodeURIComponent(url.pathname);
+        }
+        catch {
+            const match = folderUri.match(/vscode-remote:\/\/[^/]+(.+)$/);
+            if (match?.[1]) {
+                return decodeURIComponent(match[1]);
+            }
+        }
+    }
+    return null;
+}
+/**
+ * 格式化路径用于显示（简化 home 目录）
+ */
+function formatDisplayPath(p) {
+    const home = os.homedir();
+    if (p.startsWith(home)) {
+        return '~' + p.slice(home.length);
+    }
+    return p;
+}
+/**
+ * 从 folder URI 生成用于显示的友好路径
+ */
+function getDisplayPath(folderUri) {
+    // 本地路径
+    if (folderUri.startsWith('file://')) {
+        const p = extractPathFromUri(folderUri);
+        if (p) {
+            return formatDisplayPath(p);
+        }
+    }
+    // 远程路径
+    if (folderUri.startsWith('vscode-remote://')) {
+        const p = extractPathFromUri(folderUri);
+        if (p) {
+            const hostMatch = folderUri.match(/vscode-remote:\/\/codebuddy-remote-ssh%2B([^/]+)/);
+            if (hostMatch?.[1]) {
+                let host = decodeURIComponent(hostMatch[1]);
+                host = host.replace(/_x([0-9a-fA-F]{2})_/g, (_, hex) => String.fromCharCode(parseInt(hex, 16)));
+                host = host.replace(/\\x([0-9a-fA-F]{2})/g, (_, hex) => String.fromCharCode(parseInt(hex, 16)));
+                if (host.includes('@')) {
+                    const parts = host.split('@');
+                    host = parts[parts.length - 1]?.split(':')[0] || host;
+                }
+                if (host.length > 20) {
+                    host = host.slice(0, 17) + '...';
+                }
+                return `[${host}]${p}`;
+            }
+            return `[remote]${p}`;
+        }
+    }
+    return folderUri;
+}
+/**
+ * 计算路径的 MD5 hash（CodeBuddyExtension 使用纯路径计算）
+ */
+function computePathHash(p) {
+    return crypto.createHash('md5').update(p).digest('hex');
+}
+/**
+ * 加载所有工作区映射
+ */
+export async function loadWorkspaceMappings() {
+    const mappings = new Map();
+    const storageDir = getWorkspaceStorageDir();
+    let entries = [];
+    try {
+        entries = await fs.readdir(storageDir, { withFileTypes: true });
+    }
+    catch {
+        return mappings;
+    }
+    for (const entry of entries) {
+        if (!entry.isDirectory())
+            continue;
+        const workspaceJsonPath = path.join(storageDir, entry.name, 'workspace.json');
+        try {
+            const content = await fs.readFile(workspaceJsonPath, 'utf8');
+            const data = JSON.parse(content);
+            const folderUri = data.folder;
+            if (!folderUri)
+                continue;
+            const extractedPath = extractPathFromUri(folderUri);
+            if (!extractedPath)
+                continue;
+            const hash = computePathHash(extractedPath);
+            const displayPath = getDisplayPath(folderUri);
+            mappings.set(hash, { hash, folderUri, displayPath });
+        }
+        catch {
+            // 跳过无法读取的文件
+        }
+    }
+    return mappings;
+}
+/**
+ * 检查路径是否存在（同步版本，用于路径探测）
+ */
+function pathExistsSync(p) {
+    try {
+        fsSync.accessSync(p);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+/**
+ * 尝试将 CodeBuddy Code 的项目名（路径中 / 替换为 -）还原为真实路径
+ * 使用回溯搜索，因为目录名本身可能包含 -
+ *
+ * 例如: "Users-anoti-Documents-project-codebudy-cost-analyzer"
+ *    -> "/Users/anoti/Documents/project/codebudy-cost-analyzer"
+ */
+function tryResolveCodePath(name) {
+    // 检查缓存
+    const cached = codePathCache.get(name);
+    if (cached !== undefined) {
+        return cached || null;
+    }
+    const parts = name.split('-');
+    if (parts.length < 2) {
+        codePathCache.set(name, '');
+        return null;
+    }
+    // 回溯搜索：尝试不同的分割方式
+    function backtrack(index, currentPath) {
+        if (index >= parts.length) {
+            // 检查完整路径是否存在
+            if (pathExistsSync(currentPath)) {
+                return currentPath;
+            }
+            return null;
+        }
+        // 尝试从当前位置开始，合并不同数量的 parts
+        for (let end = index; end < parts.length; end++) {
+            const segment = parts.slice(index, end + 1).join('-');
+            const newPath = currentPath ? `${currentPath}/${segment}` : `/${segment}`;
+            // 如果这不是最后一段，检查目录是否存在
+            if (end < parts.length - 1) {
+                if (pathExistsSync(newPath)) {
+                    const result = backtrack(end + 1, newPath);
+                    if (result)
+                        return result;
+                }
+            }
+            else {
+                // 最后一段，检查完整路径
+                if (pathExistsSync(newPath)) {
+                    return newPath;
+                }
+            }
+        }
+        return null;
+    }
+    const result = backtrack(0, '');
+    codePathCache.set(name, result || '');
+    return result;
+}
+/**
+ * 解析项目名称
+ * - MD5 hash (32位十六进制): 从 IDE workspaceMappings 查找
+ * - 路径格式 (包含 -): 尝试还原 CodeBuddy Code 的路径格式
+ */
+export function resolveProjectName(name, mappings) {
+    // IDE source: MD5 hash
+    if (mappings && /^[a-f0-9]{32}$/.test(name)) {
+        const mapping = mappings.get(name);
+        if (mapping) {
+            return mapping.displayPath;
+        }
+    }
+    // Code source: 路径中 / 替换为 - 的格式
+    // 特征：以大写字母开头（如 Users-、home-），包含 -
+    if (/^[A-Za-z]/.test(name) && name.includes('-')) {
+        const resolved = tryResolveCodePath(name);
+        if (resolved) {
+            return formatDisplayPath(resolved);
+        }
+    }
+    return name;
+}
+//# sourceMappingURL=workspace-resolver.js.map

package/dist/lib/workspace-resolver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace-resolver.js","sourceRoot":"","sources":["../../src/lib/workspace-resolver.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,kBAAkB,CAAA;AACjC,OAAO,MAAM,MAAM,SAAS,CAAA;AAC5B,OAAO,IAAI,MAAM,WAAW,CAAA;AAC5B,OAAO,EAAE,MAAM,SAAS,CAAA;AACxB,OAAO,MAAM,MAAM,aAAa,CAAA;AAEhC,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAA;AAQnD,4BAA4B;AAC5B,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAA;AAE/C;;GAEG;AACH,SAAS,kBAAkB,CAAC,SAAiB;IAC3C,mCAAmC;IACnC,IAAI,SAAS,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAA;YAC9B,OAAO,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,kBAAkB,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAA;QAC7D,CAAC;IACH,CAAC;IAED,0DAA0D;IAC1D,IAAI,SAAS,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAA;YAC9B,OAAO,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAA;YAC7D,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACf,OAAO,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;YACrC,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAA;AACb,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,CAAS;IAClC,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,EAAE,CAAA;IACzB,IAAI,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACvB,OAAO,GAAG,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IACnC,CAAC;IACD,OAAO,CAAC,CAAA;AACV,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,SAAiB;IACvC,OAAO;IACP,IAAI,SAAS,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QACpC,MAAM,CAAC,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAA;QACvC,IAAI,CAAC,EAAE,CAAC;YACN,OAAO,iBAAiB,CAAC,CAAC,CAAC,CAAA;QAC7B,CAAC;IACH,CAAC;IAED,OAAO;IACP,IAAI,SAAS,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC7C,MAAM,CAAC,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAA;QACvC,IAAI,CAAC,EAAE,CAAC;YACN,MAAM,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAA;YACrF,IAAI,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnB,IAAI,IAAI,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAA;gBAC3C,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,CAAA;gBAC/F,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,CAAA;gBAC/F,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;oBACvB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;oBAC7B,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAA;gBACvD,CAAC;gBACD,IAAI,IAAI,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;oBACrB,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,CAAA;gBAClC,CAAC;gBACD,OAAO,IAAI,IAAI,IAAI,CAAC,EAAE,CAAA;YACxB,CAAC;YACD,OAAO,WAAW,CAAC,EAAE,CAAA;QACvB,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAA;AAClB,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,CAAS;IAChC,OAAO,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;AACzD,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB;IACzC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA4B,CAAA;IACpD,MAAM,UAAU,GAAG,sBAAsB,EAAE,CAAA;IAE3C,IAAI,OAAO,GAAoB,EAAE,CAAA;IACjC,IAAI,CAAC;QACH,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAA;IACjE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,QAAQ,CAAA;IACjB,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE;YAAE,SAAQ;QAElC,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAA;QAC7E,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAA;YAC5D,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAwB,CAAA;YACvD,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAA;YAC7B,IAAI,CAAC,SAAS;gBAAE,SAAQ;YAExB,MAAM,aAAa,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAA;YACnD,IAAI,CAAC,aAAa;gBAAE,SAAQ;YAE5B,MAAM,IAAI,GAAG,eAAe,CAAC,aAAa,CAAC,CAAA;YAC3C,MAAM,WAAW,GAAG,cAAc,CAAC,SAAS,CAAC,CAAA;YAE7C,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAA;QACtD,CAAC;QAAC,MAAM,CAAC;YACP,YAAY;QACd,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAA;AACjB,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,CAAS;IAC/B,IAAI,CAAC;QACH,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAA;QACpB,OAAO,IAAI,CAAA;IACb,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAA;IACd,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,SAAS,kBAAkB,CAAC,IAAY;IACtC,OAAO;IACP,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;IACtC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,OAAO,MAAM,IAAI,IAAI,CAAA;IACvB,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAC7B,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrB,aAAa,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAA;QAC3B,OAAO,IAAI,CAAA;IACb,CAAC;IAED,iBAAiB;IACjB,SAAS,SAAS,CAAC,KAAa,EAAE,WAAmB;QACnD,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YAC1B,aAAa;YACb,IAAI,cAAc,CAAC,WAAW,CAAC,EAAE,CAAC;gBAChC,OAAO,WAAW,CAAA;YACpB,CAAC;YACD,OAAO,IAAI,CAAA;QACb,CAAC;QAED,0BAA0B;QAC1B,KAAK,IAAI,GAAG,GAAG,KAAK,EAAE,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC;YAChD,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;YACrD,MAAM,OAAO,GAAG,WAAW,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,OAAO,EAAE,CAAA;YAEzE,qBAAqB;YACrB,IAAI,GAAG,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,IAAI,cAAc,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC5B,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,GAAG,CAAC,EAAE,OAAO,CAAC,CAAA;oBAC1C,IAAI,MAAM;wBAAE,OAAO,MAAM,CAAA;gBAC3B,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,cAAc;gBACd,IAAI,cAAc,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC5B,OAAO,OAAO,CAAA;gBAChB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAA;IACb,CAAC;IAED,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;IAC/B,aAAa,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,IAAI,EAAE,CAAC,CAAA;IACrC,OAAO,MAAM,CAAA;AACf,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,IAAY,EAAE,QAAwC;IACvF,uBAAuB;IACvB,IAAI,QAAQ,IAAI,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5C,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QAClC,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC,WAAW,CAAA;QAC5B,CAAC;IACH,CAAC;IAED,+BAA+B;IAC/B,kCAAkC;IAClC,IAAI,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACjD,MAAM,QAAQ,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAA;QACzC,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,QAAQ,CAAC,CAAA;QACpC,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAA;AACb,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codebuddy-stats",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "type": "module",
   "main": "dist/index.js",
   "repository": {

package/src/index.ts

@@ -1,12 +1,21 @@
 #!/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 type { AnalysisData } 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')) as { version: string }
+const VERSION = pkg.version
+
 type CliOptions = {
   days: number | null
   noTui: boolean
@@ -73,7 +82,7 @@ function getHeatChar(cost: number, maxCost: number): string {
 }
 
 // 渲染 Overview 视图
-function renderOverview(box: any, data: AnalysisData, width: number): void {
+function renderOverview(box: any, data: AnalysisData, width: number, note: string): void {
   const { dailySummary, grandTotal, topModel, topProject, cacheHitRate, activeDays } = data
   const heatmap = generateHeatmapData(dailySummary)
 
@@ -159,15 +168,19 @@ function renderOverview(box: any, data: AnalysisData, width: number): void {
   }
   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: any, data: AnalysisData, width: number): void {
+function renderByModel(box: any, data: AnalysisData, width: number, note: string): void {
   const { modelTotals, grandTotal } = data
   const sorted = Object.entries(modelTotals).sort((a, b) => b[1].cost - a[1].cost)
 
@@ -207,11 +220,15 @@ function renderByModel(box: any, data: AnalysisData, width: number): void {
     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: any, data: AnalysisData, width: number): void {
+function renderByProject(box: any, data: AnalysisData, width: number, note: string): void {
   const { projectTotals, grandTotal } = data
   const sorted = Object.entries(projectTotals).sort((a, b) => b[1].cost - a[1].cost)
 
@@ -232,7 +249,7 @@ function renderByProject(box: any, data: AnalysisData, width: number): void {
 
   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) +
@@ -250,11 +267,15 @@ function renderByProject(box: any, data: AnalysisData, width: number): void {
     formatTokens(grandTotal.tokens).padStart(12) +
     '{/bold}\n'
 
+  if (note) {
+    content += `\n{gray-fg}备注：${note}{/gray-fg}\n`
+  }
+
   box.setContent(content)
 }
 
 // 渲染 Daily 视图
-function renderDaily(box: any, data: AnalysisData, scrollOffset = 0, width: number): void {
+function renderDaily(box: any, data: AnalysisData, scrollOffset = 0, width: number, note: string): void {
   const { dailySummary, dailyData } = data
   const sortedDates = Object.keys(dailySummary).sort().reverse()
 
@@ -303,7 +324,7 @@ function renderDaily(box: any, data: AnalysisData, scrollOffset = 0, width: numb
       }
     }
 
-    const shortProject = shortenProjectName(topProject.name)
+    const shortProject = resolveProjectName(topProject.name, data.workspaceMappings)
 
     content +=
       date.padEnd(dateCol) +
@@ -318,6 +339,10 @@ function renderDaily(box: any, data: AnalysisData, scrollOffset = 0, width: numb
     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)
 }
 
@@ -338,7 +363,8 @@ function printTextReport(data: AnalysisData): void {
     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)})`)
   }
 
@@ -353,7 +379,7 @@ function printTextReport(data: AnalysisData): void {
   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
   }
 
@@ -365,7 +391,8 @@ async function main(): Promise<void> {
   const options = parseArgs()
 
   console.log('Loading data...')
-  let data = await loadUsageData({ days: options.days })
+  let currentSource: 'code' | 'ide' = 'code'
+  let data = await loadUsageData({ days: options.days, source: currentSource })
 
   if (options.noTui) {
     printTextReport(data)
@@ -376,6 +403,8 @@ async function main(): Promise<void> {
   const screen = blessed.screen({
     smartCSR: true,
     title: 'CodeBuddy Cost Analyzer',
+    forceUnicode: true,
+    fullUnicode: true,
   })
 
   // Tab 状态
@@ -438,6 +467,18 @@ async function main(): Promise<void> {
   // 更新 Tab 栏
   function updateTabBar(): void {
     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} `
@@ -445,25 +486,31 @@ async function main(): Promise<void> {
         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(): void {
     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
     }
   }
@@ -471,9 +518,12 @@ async function main(): Promise<void> {
   // 更新状态栏
   function updateStatusBar(): void {
     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)
   }
 
   // 键盘事件
@@ -519,7 +569,7 @@ async function main(): Promise<void> {
     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()
@@ -530,6 +580,22 @@ async function main(): Promise<void> {
     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()