codebuddy-stats 1.1.2 → 1.1.4

package/dist/index.js

@@ -40,19 +40,6 @@ Options:
     }
     return options;
 }
-// 生成热力图数据
-function generateHeatmapData(dailySummary) {
-    const sortedDates = Object.keys(dailySummary).sort();
-    if (sortedDates.length === 0)
-        return { dates: [], costs: [], maxCost: 0 };
-    const costs = sortedDates.map(d => dailySummary[d]?.cost ?? 0);
-    const maxCost = Math.max(...costs);
-    return {
-        dates: sortedDates,
-        costs,
-        maxCost,
-    };
-}
 // 获取热力图字符
 function getHeatChar(cost, maxCost) {
     if (cost === 0)
@@ -69,7 +56,6 @@ function getHeatChar(cost, maxCost) {
 // 渲染 Overview 视图
 function renderOverview(box, data, width, note) {
     const { dailySummary, grandTotal, topModel, topProject, cacheHitRate, activeDays } = data;
-    const heatmap = generateHeatmapData(dailySummary);
     // 根据宽度计算热力图周数
     const availableWidth = width - 10;
     const maxWeeks = Math.min(Math.floor(availableWidth / 2), 26); // 最多 26 周 (半年)
@@ -96,7 +82,43 @@ function renderOverview(box, data, width, note) {
         }
         weeks.push(week);
     }
-    const maxCost = heatmap.maxCost || 1;
+    // 以“当前热力图窗口”的最大值做归一化（避免历史极值导致近期全是浅色）
+    const visibleCosts = [];
+    for (const week of weeks) {
+        for (const date of week) {
+            if (!date || date > todayStr)
+                continue;
+            visibleCosts.push(dailySummary[date]?.cost ?? 0);
+        }
+    }
+    const maxCost = Math.max(...visibleCosts, 0) || 1;
+    // 月份标尺（在列上方标注月份变化）
+    const monthNames = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'];
+    const colWidth = 2; // 每周一列：字符 + 空格
+    const heatStartCol = 4; // 左侧周几标签宽度
+    const headerLen = heatStartCol + weeks.length * colWidth;
+    const monthHeader = Array.from({ length: headerLen }, () => ' ');
+    let lastMonth = -1;
+    let lastPlacedAt = -999;
+    for (let i = 0; i < weeks.length; i++) {
+        const week = weeks[i];
+        const repDate = week.find(d => d && d <= todayStr) ?? week[0];
+        if (!repDate)
+            continue;
+        const m = new Date(repDate).getMonth();
+        if (m !== lastMonth) {
+            const label = monthNames[m];
+            const pos = heatStartCol + i * colWidth;
+            // 避免月份标签过于拥挤/相互覆盖
+            if (pos - lastPlacedAt >= 4 && pos + label.length <= monthHeader.length) {
+                for (let k = 0; k < label.length; k++)
+                    monthHeader[pos + k] = label[k];
+                lastPlacedAt = pos;
+            }
+            lastMonth = m;
+        }
+    }
+    content += `{gray-fg}${monthHeader.join('').trimEnd()}{/gray-fg}\n`;
     const dayLabels = ['Sun', 'Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat'];
     for (let dayOfWeek = 0; dayOfWeek < 7; dayOfWeek++) {
         let row = dayLabels[dayOfWeek].padEnd(4);
@@ -114,6 +136,8 @@ function renderOverview(box, data, width, note) {
         }
         content += row + '\n';
     }
+    const rangeStart = weeks[0]?.[0] ?? todayStr;
+    content += `{gray-fg}Range: ${rangeStart} → ${todayStr}{/gray-fg}\n`;
     content += '    Less {gray-fg}·░▒▓{/gray-fg}{white-fg}█{/white-fg} More\n\n';
     // 汇总指标 - 根据宽度决定布局
     const avgDailyCost = activeDays > 0 ? grandTotal.cost / activeDays : 0;
@@ -164,7 +188,7 @@ function renderByModel(box, data, width, note) {
     content +=
         '{underline}' +
             'Model'.padEnd(modelCol) +
-            'Cost'.padStart(12) +
+            '~Cost'.padStart(12) +
             'Requests'.padStart(12) +
             'Tokens'.padStart(12) +
             'Avg/Req'.padStart(10) +
@@ -205,7 +229,7 @@ function renderByProject(box, data, width, note) {
     content +=
         '{underline}' +
             'Project'.padEnd(projectCol) +
-            'Cost'.padStart(12) +
+            '~Cost'.padStart(12) +
             'Requests'.padStart(12) +
             'Tokens'.padStart(12) +
             '{/underline}\n';
@@ -240,8 +264,9 @@ function renderDaily(box, data, scrollOffset = 0, width, note) {
     const availableWidth = width - 6; // padding
     const dateCol = 12;
     const costCol = 12;
+    const tokensCol = 10;
     const reqCol = 10;
-    const fixedCols = dateCol + costCol + reqCol;
+    const fixedCols = dateCol + costCol + tokensCol + reqCol;
     const remainingWidth = availableWidth - fixedCols;
     const modelCol = Math.max(15, Math.min(25, Math.floor(remainingWidth * 0.4)));
     const projectCol = Math.max(20, remainingWidth - modelCol);
@@ -249,7 +274,8 @@ function renderDaily(box, data, scrollOffset = 0, width, note) {
     content +=
         '{underline}' +
             'Date'.padEnd(dateCol) +
-            'Cost'.padStart(costCol) +
+            '~Cost'.padStart(costCol) +
+            'Tokens'.padStart(tokensCol) +
             'Requests'.padStart(reqCol) +
             'Top Model'.padStart(modelCol) +
             'Top Project'.padStart(projectCol) +
@@ -280,6 +306,7 @@ function renderDaily(box, data, scrollOffset = 0, width, note) {
         content +=
             date.padEnd(dateCol) +
                 formatCost(daySummary.cost).padStart(costCol) +
+                formatTokens(daySummary.tokens).padStart(tokensCol) +
                 formatNumber(daySummary.requests).padStart(reqCol) +
                 truncate(topModel.id, modelCol - 1).padStart(modelCol) +
                 truncate(shortProject, projectCol - 1).padStart(projectCol) +
@@ -444,10 +471,29 @@ async function main() {
     function updateStatusBar() {
         const daysInfo = options.days ? `Last ${options.days} days` : 'All time';
         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 rightContent = `v${VERSION}`;
         const width = Number(screen.width) || 80;
-        const padding = Math.max(0, width - leftContent.length - rightContent.length);
+        // 根据剩余宽度决定左侧内容详细程度（预留版本号空间）
+        const reservedForRight = rightContent.length + 2; // 版本号 + 两侧空格
+        const availableForLeft = width - reservedForRight;
+        let leftContent;
+        const fullContent = ` ${daysInfo} | Source: ${sourceInfo} | Total: ${formatCost(data.grandTotal.cost)} | q quit, Tab view, s source, r refresh`;
+        const mediumContent = ` ${daysInfo} | ${sourceInfo} | ${formatCost(data.grandTotal.cost)} | q/Tab/s/r`;
+        const shortContent = ` ${sourceInfo} | ${formatCost(data.grandTotal.cost)} | q/Tab/s/r`;
+        const minContent = ` ${formatCost(data.grandTotal.cost)}`;
+        if (fullContent.length <= availableForLeft) {
+            leftContent = fullContent;
+        }
+        else if (mediumContent.length <= availableForLeft) {
+            leftContent = mediumContent;
+        }
+        else if (shortContent.length <= availableForLeft) {
+            leftContent = shortContent;
+        }
+        else {
+            leftContent = minContent;
+        }
+        const padding = Math.max(1, width - leftContent.length - rightContent.length);
         statusBar.setContent(leftContent + ' '.repeat(padding) + rightContent);
     }
     // 键盘事件
@@ -517,7 +563,9 @@ async function main() {
     });
     // 监听窗口大小变化
     screen.on('resize', () => {
+        updateTabBar();
         updateContent();
+        updateStatusBar();
         screen.render();
     });
     // 初始渲染

package/dist/lib/paths.js

@@ -24,7 +24,7 @@ export function getConfigDir() {
  * 获取 CodeBuddy IDE (CodeBuddyExtension) 数据目录
  * - macOS: ~/Library/Application Support/CodeBuddyExtension/Data
  * - Windows: %APPDATA%/CodeBuddyExtension/Data
- * - Linux: $XDG_CONFIG_HOME/CodeBuddyExtension/Data 或 ~/.config/CodeBuddyExtension/Data
+ * - Linux: $XDG_DATA_HOME/CodeBuddyExtension/Data 或 ~/.local/share/CodeBuddyExtension/Data
  */
 export function getIdeDataDir() {
     if (process.platform === 'darwin') {
@@ -34,8 +34,8 @@ export function getIdeDataDir() {
         const appData = process.env.APPDATA || path.join(os.homedir(), 'AppData', 'Roaming');
         return path.join(appData, 'CodeBuddyExtension', 'Data');
     }
-    const xdgConfigHome = process.env.XDG_CONFIG_HOME || path.join(os.homedir(), '.config');
-    return path.join(xdgConfigHome, 'CodeBuddyExtension', 'Data');
+    const xdgDataHome = process.env.XDG_DATA_HOME || path.join(os.homedir(), '.local', 'share');
+    return path.join(xdgDataHome, 'CodeBuddyExtension', 'Data');
 }
 /**
  * 获取 CodeBuddy IDE 的 workspaceStorage 目录

package/dist/lib/workspace-resolver.js

@@ -3,7 +3,7 @@ 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';
+import { getIdeDataDir, getWorkspaceStorageDir } from './paths.js';
 // 缓存已解析的 CodeBuddy Code 路径名
 const codePathCache = new Map();
 /**
@@ -86,38 +86,381 @@ function computePathHash(p) {
     return crypto.createHash('md5').update(p).digest('hex');
 }
 /**
- * 加载所有工作区映射
+ * 从 CodeBuddyIDE 的 file-tree 数据中反推出 workspace hash -> 真实路径
+ *
+ * 背景：Remote SSH 场景下，server 侧 `workspaceStorage/<hash>/workspace.json` 可能不存在，
+ * 但 CodeBuddyExtension 会把对话关联的文件路径写入 `file-tree.json`，可据此还原工作区根目录。
  */
-export async function loadWorkspaceMappings() {
-    const mappings = new Map();
-    const storageDir = getWorkspaceStorageDir();
-    let entries = [];
+async function loadWorkspaceMappingsFromIdeFileTree() {
+    const out = new Map();
+    const root = getIdeDataDir();
+    async function pathExists(p) {
+        try {
+            await fs.access(p);
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    function collectFilePaths(node, acc, depth = 0) {
+        if (depth > 10)
+            return;
+        if (typeof node === 'string')
+            return;
+        if (Array.isArray(node)) {
+            for (const item of node)
+                collectFilePaths(item, acc, depth + 1);
+            return;
+        }
+        if (!node || typeof node !== 'object')
+            return;
+        for (const [k, v] of Object.entries(node)) {
+            if (k === 'filePath' && typeof v === 'string' && v) {
+                acc.push(v);
+                continue;
+            }
+            if (k === 'createdEntries' && Array.isArray(v)) {
+                for (const e of v) {
+                    if (typeof e === 'string' && e)
+                        acc.push(e);
+                }
+                continue;
+            }
+            collectFilePaths(v, acc, depth + 1);
+        }
+    }
+    function resolveWorkspacePathFromFilePaths(hash, filePaths) {
+        for (const raw of filePaths) {
+            const p = extractPathFromUri(raw) || (path.isAbsolute(raw) ? raw : null);
+            if (!p)
+                continue;
+            let cur = p;
+            for (let i = 0; i < 50; i++) {
+                if (computePathHash(cur) === hash)
+                    return cur;
+                const parent = path.dirname(cur);
+                if (parent === cur)
+                    break;
+                cur = parent;
+            }
+        }
+        return null;
+    }
+    // 扫描 `Data/*/CodeBuddyIDE/*/file-tree/*/<convId>/file-tree.json`
+    let level1 = [];
     try {
-        entries = await fs.readdir(storageDir, { withFileTypes: true });
+        level1 = await fs.readdir(root, { withFileTypes: true });
     }
     catch {
-        return mappings;
+        return out;
     }
-    for (const entry of entries) {
-        if (!entry.isDirectory())
+    for (const dirent of level1) {
+        if (!dirent.isDirectory())
             continue;
-        const workspaceJsonPath = path.join(storageDir, entry.name, 'workspace.json');
+        const codeBuddyIdeRoot = path.join(root, dirent.name, 'CodeBuddyIDE');
+        if (!(await pathExists(codeBuddyIdeRoot)))
+            continue;
+        // 兼容两种结构：
+        // 1) CodeBuddyIDE/file-tree
+        // 2) CodeBuddyIDE/<profile>/file-tree（当前主流）
+        const profileDirs = [];
+        if (await pathExists(path.join(codeBuddyIdeRoot, 'file-tree'))) {
+            profileDirs.push(codeBuddyIdeRoot);
+        }
+        let nested = [];
         try {
-            const content = await fs.readFile(workspaceJsonPath, 'utf8');
-            const data = JSON.parse(content);
-            const folderUri = data.folder;
-            if (!folderUri)
+            nested = await fs.readdir(codeBuddyIdeRoot, { withFileTypes: true });
+        }
+        catch {
+            nested = [];
+        }
+        for (const child of nested) {
+            if (!child.isDirectory())
                 continue;
-            const extractedPath = extractPathFromUri(folderUri);
-            if (!extractedPath)
+            const profileDir = path.join(codeBuddyIdeRoot, child.name);
+            if (await pathExists(path.join(profileDir, 'file-tree'))) {
+                profileDirs.push(profileDir);
+            }
+        }
+        for (const profileDir of profileDirs) {
+            const fileTreeDir = path.join(profileDir, 'file-tree');
+            let workspaces = [];
+            try {
+                workspaces = await fs.readdir(fileTreeDir, { withFileTypes: true });
+            }
+            catch {
                 continue;
-            const hash = computePathHash(extractedPath);
-            const displayPath = getDisplayPath(folderUri);
-            mappings.set(hash, { hash, folderUri, displayPath });
+            }
+            for (const ws of workspaces) {
+                if (!ws.isDirectory())
+                    continue;
+                const workspaceHash = ws.name;
+                if (!/^[a-f0-9]{32}$/.test(workspaceHash))
+                    continue;
+                if (out.has(workspaceHash))
+                    continue;
+                const workspaceDir = path.join(fileTreeDir, workspaceHash);
+                let convDirs = [];
+                try {
+                    convDirs = await fs.readdir(workspaceDir, { withFileTypes: true });
+                }
+                catch {
+                    continue;
+                }
+                let resolved = null;
+                for (const conv of convDirs) {
+                    if (!conv.isDirectory())
+                        continue;
+                    const fileTreeJson = path.join(workspaceDir, conv.name, 'file-tree.json');
+                    try {
+                        const raw = await fs.readFile(fileTreeJson, 'utf8');
+                        const parsed = JSON.parse(raw);
+                        const paths = [];
+                        collectFilePaths(parsed, paths);
+                        resolved = resolveWorkspacePathFromFilePaths(workspaceHash, paths);
+                        if (resolved)
+                            break;
+                    }
+                    catch {
+                        // ignore
+                    }
+                }
+                if (!resolved)
+                    continue;
+                const folderUri = 'file://' + resolved;
+                out.set(workspaceHash, {
+                    hash: workspaceHash,
+                    folderUri,
+                    displayPath: getDisplayPath(folderUri),
+                });
+            }
+        }
+    }
+    return out;
+}
+/**
+ * 从 CodeBuddyIDE 的 history/messages 中反推出 workspace hash -> 真实路径
+ *
+ * 场景：有些对话没有触发 file-tree 落盘，但 messages 里常包含 tool-result 的绝对路径。
+ */
+async function loadWorkspaceMappingsFromIdeHistory() {
+    const out = new Map();
+    const root = getIdeDataDir();
+    async function pathExists(p) {
+        try {
+            await fs.access(p);
+            return true;
         }
         catch {
-            // 跳过无法读取的文件
+            return false;
+        }
+    }
+    async function readFileHeadUtf8(filePath, bytes = 64 * 1024) {
+        const fh = await fs.open(filePath, 'r');
+        try {
+            const buf = Buffer.alloc(bytes);
+            const { bytesRead } = await fh.read(buf, 0, buf.length, 0);
+            return buf.toString('utf8', 0, bytesRead);
         }
+        finally {
+            await fh.close();
+        }
+    }
+    function extractCandidatePathsFromText(text) {
+        const set = new Set();
+        const uriRe = /(vscode-remote:\/\/[^\s"']+|file:\/\/[^\s"']+)/g;
+        for (const m of text.matchAll(uriRe)) {
+            const v = m[1];
+            if (v)
+                set.add(v);
+        }
+        const absPathRe = /\/[A-Za-z0-9._~@%+=:,\-]+(?:\/[A-Za-z0-9._~@%+=:,\-]+)+/g;
+        for (const m of text.matchAll(absPathRe)) {
+            const v = m[0];
+            if (v)
+                set.add(v);
+        }
+        return [...set];
+    }
+    function resolveWorkspacePathFromCandidates(hash, candidates) {
+        for (const raw of candidates) {
+            const p = extractPathFromUri(raw) || (path.isAbsolute(raw) ? raw : null);
+            if (!p)
+                continue;
+            let cur = p;
+            for (let i = 0; i < 50; i++) {
+                if (computePathHash(cur) === hash)
+                    return cur;
+                const parent = path.dirname(cur);
+                if (parent === cur)
+                    break;
+                cur = parent;
+            }
+        }
+        return null;
+    }
+    let level1 = [];
+    try {
+        level1 = await fs.readdir(root, { withFileTypes: true });
+    }
+    catch {
+        return out;
+    }
+    for (const dirent of level1) {
+        if (!dirent.isDirectory())
+            continue;
+        const codeBuddyIdeRoot = path.join(root, dirent.name, 'CodeBuddyIDE');
+        if (!(await pathExists(codeBuddyIdeRoot)))
+            continue;
+        const profileDirs = [];
+        if (await pathExists(path.join(codeBuddyIdeRoot, 'history'))) {
+            profileDirs.push(codeBuddyIdeRoot);
+        }
+        let nested = [];
+        try {
+            nested = await fs.readdir(codeBuddyIdeRoot, { withFileTypes: true });
+        }
+        catch {
+            nested = [];
+        }
+        for (const child of nested) {
+            if (!child.isDirectory())
+                continue;
+            const profileDir = path.join(codeBuddyIdeRoot, child.name);
+            if (await pathExists(path.join(profileDir, 'history'))) {
+                profileDirs.push(profileDir);
+            }
+        }
+        for (const profileDir of profileDirs) {
+            const historyDir = path.join(profileDir, 'history');
+            let workspaces = [];
+            try {
+                workspaces = await fs.readdir(historyDir, { withFileTypes: true });
+            }
+            catch {
+                continue;
+            }
+            for (const ws of workspaces) {
+                if (!ws.isDirectory())
+                    continue;
+                const workspaceHash = ws.name;
+                if (!/^[a-f0-9]{32}$/.test(workspaceHash))
+                    continue;
+                if (out.has(workspaceHash))
+                    continue;
+                const workspaceDir = path.join(historyDir, workspaceHash);
+                let convDirs = [];
+                try {
+                    convDirs = await fs.readdir(workspaceDir, { withFileTypes: true });
+                }
+                catch {
+                    continue;
+                }
+                let resolved = null;
+                for (const conv of convDirs) {
+                    if (!conv.isDirectory())
+                        continue;
+                    const messagesDir = path.join(workspaceDir, conv.name, 'messages');
+                    if (!(await pathExists(messagesDir)))
+                        continue;
+                    let msgFiles = [];
+                    try {
+                        msgFiles = await fs.readdir(messagesDir, { withFileTypes: true });
+                    }
+                    catch {
+                        continue;
+                    }
+                    let tried = 0;
+                    for (const msg of msgFiles) {
+                        if (!msg.isFile() || !msg.name.endsWith('.json'))
+                            continue;
+                        const msgPath = path.join(messagesDir, msg.name);
+                        tried += 1;
+                        if (tried > 30)
+                            break;
+                        try {
+                            const head = await readFileHeadUtf8(msgPath);
+                            const candidates = extractCandidatePathsFromText(head);
+                            resolved = resolveWorkspacePathFromCandidates(workspaceHash, candidates);
+                            if (resolved)
+                                break;
+                        }
+                        catch {
+                            // ignore
+                        }
+                    }
+                    if (resolved)
+                        break;
+                }
+                if (!resolved)
+                    continue;
+                const folderUri = 'file://' + resolved;
+                out.set(workspaceHash, {
+                    hash: workspaceHash,
+                    folderUri,
+                    displayPath: getDisplayPath(folderUri),
+                });
+            }
+        }
+    }
+    return out;
+}
+/**
+ * 加载所有工作区映射
+ */
+export async function loadWorkspaceMappings() {
+    const mappings = new Map();
+    // 1) 尝试从客户端 workspaceStorage/workspace.json 解析
+    const storageDir = getWorkspaceStorageDir();
+    try {
+        const entries = await fs.readdir(storageDir, { withFileTypes: true });
+        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 {
+                // ignore
+            }
+        }
+    }
+    catch {
+        // ignore
+    }
+    // 2) Remote SSH 等场景的兜底：从 CodeBuddyIDE file-tree 反推
+    try {
+        const ideMappings = await loadWorkspaceMappingsFromIdeFileTree();
+        for (const [hash, mapping] of ideMappings) {
+            if (!mappings.has(hash))
+                mappings.set(hash, mapping);
+        }
+    }
+    catch {
+        // ignore
+    }
+    // 3) 进一步兜底：从 CodeBuddyIDE history/messages 反推（tool-result 常带绝对路径）
+    try {
+        const ideMappings = await loadWorkspaceMappingsFromIdeHistory();
+        for (const [hash, mapping] of ideMappings) {
+            if (!mappings.has(hash))
+                mappings.set(hash, mapping);
+        }
+    }
+    catch {
+        // ignore
     }
     return mappings;
 }

package/package.json

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