@winston.wan/burn-your-money 2.1.2 → 2.2.0

package/src/statusline.js

@@ -0,0 +1,264 @@
+#!/usr/bin/env node
+/**
+ * Burn Your Money - Claude Code 状态栏脚本 (Node.js 版本)
+ * 格式：[模型] 今日：token 费用 🔥速度 | 总计：token 费用
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+
+// 配置文件路径
+const CONFIG_FILE = path.join(os.homedir(), '.claude', 'burn-your-money-config.json');
+const SESSION_STATE = path.join(os.homedir(), '.claude', 'cache', 'session-state.json');
+const TODAY_STATE = path.join(os.homedir(), '.claude', 'cache', 'today-state.json');
+const HISTORY_CACHE = path.join(os.homedir(), '.claude', 'cache', 'history-cache.json');
+
+// 默认配置
+const DEFAULT_CONFIG = {
+    theme: 'fire',
+    alert_daily: 10.0,
+    alert_weekly: 50.0,
+    show_burn_rate: true,
+    show_eta: true,
+    show_trend: false
+};
+
+// 主题颜色定义
+const THEMES = {
+    fire: { TODAY: '31', COST: '31', BURN: '31' },
+    ocean: { TODAY: '34', COST: '36', BURN: '36' },
+    forest: { TODAY: '32', COST: '33', BURN: '32' },
+    golden: { TODAY: '33', COST: '33', BURN: '1;33' }
+};
+
+/**
+ * 读取配置文件
+ */
+function readConfig() {
+    try {
+        if (fs.existsSync(CONFIG_FILE)) {
+            const data = fs.readFileSync(CONFIG_FILE, 'utf8');
+            return { ...DEFAULT_CONFIG, ...JSON.parse(data) };
+        }
+    } catch (e) {
+        // 忽略错误，使用默认配置
+    }
+    return DEFAULT_CONFIG;
+}
+
+/**
+ * 格式化数字（添加 K/M 后缀）
+ */
+function formatNumber(num) {
+    if (num >= 1000000000) return (num / 1000000000).toFixed(1) + 'B';
+    if (num >= 1000000) return (num / 1000000).toFixed(1) + 'M';
+    if (num >= 1000) return (num / 1000).toFixed(1) + 'K';
+    return num.toString();
+}
+
+/**
+ * 格式化费用
+ */
+function formatCost(cost) {
+    return '$' + parseFloat(cost).toFixed(2);
+}
+
+/**
+ * 格式化燃烧速度
+ */
+function formatBurnRate(rate) {
+    if (rate >= 1000) return (rate / 1000).toFixed(1) + 'K tok/s';
+    return Math.round(rate).toString() + ' tok/s';
+}
+
+/**
+ * 读取 JSON 文件
+ */
+function readJsonFile(filePath, defaultValue = {}) {
+    try {
+        if (fs.existsSync(filePath)) {
+            const data = fs.readFileSync(filePath, 'utf8');
+            return JSON.parse(data);
+        }
+    } catch (e) {
+        // 忽略错误
+    }
+    return defaultValue;
+}
+
+/**
+ * 写入 JSON 文件
+ */
+function writeJsonFile(filePath, data) {
+    try {
+        const tmpPath = filePath + '.tmp.' + process.pid;
+        fs.writeFileSync(tmpPath, JSON.stringify(data, null, 2));
+        fs.renameSync(tmpPath, filePath);
+    } catch (e) {
+        // 忽略错误
+    }
+}
+
+/**
+ * 确保目录存在
+ */
+function ensureDir(filePath) {
+    const dir = path.dirname(filePath);
+    if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true });
+    }
+}
+
+/**
+ * 主函数
+ */
+function main() {
+    // 读取配置
+    const config = readConfig();
+    const theme = THEMES[config.theme] || THEMES.fire;
+
+    // 从 stdin 读取 Claude Code 传入的数据
+    let inputData = '';
+    process.stdin.on('data', (chunk) => {
+        inputData += chunk;
+    });
+
+    process.stdin.on('end', () => {
+        try {
+            const data = JSON.parse(inputData || '{}');
+
+            // 提取数据
+            const modelName = data.model?.display_name || 'Unknown';
+            const sessionId = data.session_id || 'default';
+            const inputTokens = data.context_window?.total_input_tokens || 0;
+            const outputTokens = data.context_window?.total_output_tokens || 0;
+            const cacheReadTokens = data.context_window?.current_usage?.cache_read_input_tokens || 0;
+            const currentCost = data.cost?.total_cost_usd || 0;
+
+            const currentSessionTokens = inputTokens + outputTokens + cacheReadTokens;
+
+            // 计算燃烧速度
+            const currentTime = Math.floor(Date.now() / 1000);
+            let startTime = currentTime;
+
+            // 读取 session state
+            const sessionState = readJsonFile(SESSION_STATE);
+            if (sessionState.session_id === sessionId && sessionState.start_time) {
+                startTime = sessionState.start_time;
+            } else {
+                // 新会话，记录开始时间
+                ensureDir(SESSION_STATE);
+                writeJsonFile(SESSION_STATE, {
+                    session_id: sessionId,
+                    start_time: currentTime
+                });
+            }
+
+            const elapsed = currentTime - startTime;
+            let burnRate = 0;
+            if (elapsed > 0 && currentSessionTokens > 0) {
+                burnRate = currentSessionTokens / elapsed;
+            }
+
+            // 读取历史缓存
+            ensureDir(HISTORY_CACHE);
+            const now = Math.floor(Date.now() / 1000);
+            let historyData = readJsonFile(HISTORY_CACHE, {});
+
+            // 检查缓存是否过期（30秒）
+            const cacheAge = now - (historyData._cache_time || 0);
+            if (cacheAge > 30 || !historyData.total_tokens_all) {
+                // 缓存过期，重新生成（这里简化处理，实际应该调用 token-history）
+                historyData = {
+                    total_tokens_all: 0,
+                    total_cost: 0,
+                    _cache_time: now
+                };
+            }
+
+            // 今日数据持久化
+            const todayDate = new Date().toISOString().split('T')[0];
+            ensureDir(TODAY_STATE);
+            const todayState = readJsonFile(TODAY_STATE, {
+                date: todayDate,
+                tokens: 0,
+                cost: 0,
+                last_session_tokens: 0,
+                last_session_cost: 0
+            });
+
+            // 如果是新的一天，重置
+            if (todayState.date !== todayDate) {
+                todayState.date = todayDate;
+                todayState.tokens = 0;
+                todayState.cost = 0;
+                todayState.last_session_tokens = 0;
+                todayState.last_session_cost = 0;
+            }
+
+            // 计算增量
+            let tokenIncrement = currentSessionTokens - (todayState.last_session_tokens || 0);
+            if (tokenIncrement < 0) tokenIncrement = 0;
+
+            let costIncrement = currentCost - (todayState.last_session_cost || 0);
+            if (costIncrement < 0) costIncrement = 0;
+
+            // 更新今日数据
+            const newTodayTokens = (todayState.tokens || 0) + tokenIncrement;
+            const newTodayCost = (todayState.cost || 0) + costIncrement;
+
+            writeJsonFile(TODAY_STATE, {
+                date: todayDate,
+                tokens: newTodayTokens,
+                cost: newTodayCost,
+                last_session_tokens: currentSessionTokens,
+                last_session_cost: currentCost
+            });
+
+            // 计算总计
+            const totalTokens = (historyData.total_tokens_all || 0) + currentSessionTokens;
+            const totalCost = (historyData.total_cost || 0) + currentCost;
+
+            // 检查是否超过警报阈值
+            const isAlert = newTodayCost >= config.alert_daily;
+
+            // 构建输出
+            let output = '';
+
+            // 模型名称
+            if (isAlert) {
+                output += `\x1b[1;31m[${modelName}]\x1b[0m `;
+            } else {
+                output += `\x1b[0;90m[${modelName}]\x1b[0m `;
+            }
+
+            // 今日数据
+            if (newTodayTokens > 0 || newTodayCost > 0) {
+                output += `\x1b[${theme.TODAY}m今日：${formatNumber(newTodayTokens)}\x1b[0m `;
+                output += `\x1b[${theme.COST}m${formatCost(newTodayCost)}\x1b[0m `;
+            } else {
+                output += `\x1b[0;90m今日：0\x1b[0m \x1b[0;90m$0.00\x1b[0m `;
+            }
+
+            // 燃烧速度
+            if (config.show_burn_rate && burnRate > 0) {
+                output += `\x1b[${theme.BURN}m🔥${formatBurnRate(burnRate)}\x1b[0m `;
+            }
+
+            // 总计数据
+            output += `| \x1b[0;90m总计：${formatNumber(totalTokens)}\x1b[0m `;
+            output += `\x1b[0;90m${formatCost(totalCost)}\x1b[0m `;
+
+            // 输出
+            process.stdout.write(output);
+
+        } catch (e) {
+            // 出错时输出简单错误信息
+            process.stdout.write('[BurnYourMoney] Error');
+        }
+    });
+}
+
+// 运行
+main();

package/src/token-history.js

@@ -0,0 +1,234 @@
+#!/usr/bin/env node
+/**
+ * Burn Your Money - Token 历史统计脚本 (Node.js 版本)
+ * 从 stats-cache.json 读取历史数据，返回今日/本周/本月统计
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+
+// 文件路径
+const STATS_CACHE = path.join(os.homedir(), '.claude', 'stats-cache.json');
+const TODAY = new Date().toISOString().split('T')[0];
+const CURRENT_MONTH = new Date().toISOString().slice(0, 7); // YYYY-MM
+
+/**
+ * 获取本周一的日期
+ */
+function getWeekStart() {
+    const now = new Date();
+    const dayOfWeek = now.getDay();
+    const diff = now.getDate() - dayOfWeek + (dayOfWeek === 0 ? -6 : 1);
+    const weekStart = new Date(now.setDate(diff));
+    return weekStart.toISOString().split('T')[0];
+}
+
+/**
+ * 获取本月第一天
+ */
+function getMonthStart() {
+    const now = new Date();
+    return new Date(now.getFullYear(), now.getMonth(), 1).toISOString().split('T')[0];
+}
+
+/**
+ * 获取上月第一天
+ */
+function getLastMonthStart() {
+    const now = new Date();
+    return new Date(now.getFullYear(), now.getMonth() - 1, 1).toISOString().split('T')[0];
+}
+
+/**
+ * 读取统计数据
+ */
+function readStats() {
+    try {
+        if (!fs.existsSync(STATS_CACHE)) {
+            return {
+                today_tokens: 0,
+                today_cost: 0,
+                week_tokens: 0,
+                week_cost: 0,
+                month_tokens: 0,
+                month_cost: 0,
+                last_month_tokens: 0,
+                last_month_cost: 0,
+                total_tokens_all: 0,
+                total_cost: 0,
+                daily_data: []
+            };
+        }
+
+        const data = JSON.parse(fs.readFileSync(STATS_CACHE, 'utf8'));
+
+        // 计算所有模型的 input 和 output tokens
+        const totalInputTokens = Object.values(data.modelUsage || {}).reduce((sum, model) => sum + (model.inputTokens || 0), 0);
+        const totalOutputTokens = Object.values(data.modelUsage || {}).reduce((sum, model) => sum + (model.outputTokens || 0), 0);
+        const totalCacheReadTokens = Object.values(data.modelUsage || {}).reduce((sum, model) => sum + (model.cacheReadInputTokens || 0), 0);
+
+        // 计算今日 tokens（总计 - 历史每日总和，排除今天）
+        const dailyTokens = data.dailyModelTokens || [];
+        let historySum = 0;
+        dailyTokens.forEach(day => {
+            if (day.date !== TODAY) {
+                const models = day.tokensByModel || {};
+                historySum += Object.values(models).reduce((sum, tokens) => sum + (tokens || 0), 0);
+            }
+        });
+
+        const todayTokens = totalInputTokens + totalOutputTokens - historySum;
+        if (todayTokens < 0) todayTokens = 0;
+
+        // 本周 tokens（排除今天，用上面的 todayTokens）
+        const WEEK_START = getWeekStart();
+        let weekTokens = 0;
+        dailyTokens.forEach(day => {
+            if (day.date >= WEEK_START && day.date !== TODAY) {
+                const models = day.tokensByModel || {};
+                weekTokens += Object.values(models).reduce((sum, tokens) => sum + (tokens || 0), 0);
+            }
+        });
+        weekTokens += todayTokens;
+
+        // 本月 tokens
+        const MONTH_START = getMonthStart();
+        let monthTokens = 0;
+        dailyTokens.forEach(day => {
+            if (day.date >= MONTH_START && day.date !== TODAY) {
+                const models = day.tokensByModel || {};
+                monthTokens += Object.values(models).reduce((sum, tokens) => sum + (tokens || 0), 0);
+            }
+        });
+        monthTokens += todayTokens;
+
+        // 上月 tokens
+        const LAST_MONTH_START = getLastMonthStart();
+        let lastMonthTokens = 0;
+        dailyTokens.forEach(day => {
+            if (day.date >= LAST_MONTH_START && day.date < MONTH_START) {
+                const models = day.tokensByModel || {};
+                lastMonthTokens += Object.values(models).reduce((sum, tokens) => sum + (tokens || 0), 0);
+            }
+        });
+
+        // 计算费用（简化：按 3/15/0.3 规则）
+        const totalCost = (totalInputTokens / 1000000 * 3) +
+                          (totalOutputTokens / 1000000 * 15) +
+                          (totalCacheReadTokens / 1000000 * 0.3);
+
+        const totalTokensAll = totalInputTokens + totalOutputTokens + totalCacheReadTokens;
+
+        // 计算各周期费用（按比例分摊）
+        const todayCost = todayTokens > 0 ? (todayTokens / (totalInputTokens + totalOutputTokens) * totalCost) : 0;
+        const weekCost = weekTokens > 0 ? (weekTokens / (totalInputTokens + totalOutputTokens) * totalCost) : 0;
+        const monthCost = monthTokens > 0 ? (monthTokens / (totalInputTokens + totalOutputTokens) * totalCost) : 0;
+        const lastMonthCost = lastMonthTokens > 0 ? (lastMonthTokens / (totalInputTokens + totalOutputTokens) * totalCost) : 0;
+
+        // 获取每日数据（用于趋势图）
+        const dailyData = dailyTokens.slice(-7).map(day => ({
+            date: day.date,
+            tokens: Object.values(day.tokensByModel || {}).reduce((sum, tokens) => sum + (tokens || 0), 0)
+        }));
+
+        return {
+            today_tokens: todayTokens,
+            today_cost: todayCost.toFixed(2),
+            week_tokens: weekTokens,
+            week_cost: weekCost.toFixed(2),
+            month_tokens: monthTokens,
+            month_cost: monthCost.toFixed(2),
+            last_month_tokens: lastMonthTokens,
+            last_month_cost: lastMonthCost.toFixed(2),
+            total_tokens_all: totalTokensAll,
+            total_cost: totalCost.toFixed(2),
+            daily_data: dailyData
+        };
+    } catch (e) {
+        console.error('Error reading stats:', e.message);
+        return {
+            today_tokens: 0,
+            today_cost: '0.00',
+            week_tokens: 0,
+            week_cost: '0.00',
+            month_tokens: 0,
+            month_cost: '0.00',
+            last_month_tokens: 0,
+            last_month_cost: '0.00',
+            total_tokens_all: 0,
+            total_cost: '0.00',
+            daily_data: []
+        };
+    }
+}
+
+/**
+ * 主函数
+ */
+function main() {
+    const mode = process.argv[2] || 'summary';
+
+    const stats = readStats();
+
+    switch (mode) {
+        case 'today_tokens':
+            console.log(stats.today_tokens);
+            break;
+        case 'today_cost':
+            console.log(stats.today_cost);
+            break;
+        case 'week_tokens':
+            console.log(stats.week_tokens);
+            break;
+        case 'week_cost':
+            console.log(stats.week_cost);
+            break;
+        case 'month_tokens':
+            console.log(stats.month_tokens);
+            break;
+        case 'month_cost':
+            console.log(stats.month_cost);
+            break;
+        case 'last_month_tokens':
+            console.log(stats.last_month_tokens);
+            break;
+        case 'last_month_cost':
+            console.log(stats.last_month_cost);
+            break;
+        case 'trend':
+            console.log(JSON.stringify(stats.daily_data));
+            break;
+        case 'export':
+            const format = process.argv[3] || 'json';
+            if (format === 'csv') {
+                console.log('date,tokens,cost');
+                stats.daily_data.forEach(day => {
+                    const cost = (day.tokens * 0.000006).toFixed(2);
+                    console.log(`${day.date},${day.tokens},${cost}`);
+                });
+            } else {
+                console.log(JSON.stringify(stats, null, 2));
+            }
+            break;
+        case 'chart':
+            // 生成简单的趋势图
+            const maxTokens = Math.max(...stats.daily_data.map(d => d.tokens), 1);
+            console.log('\n📊 Token 使用趋势 (最近 7 天)\n');
+            stats.daily_data.forEach(day => {
+                const date = day.date.slice(5); // MM-DD
+                const tokens = day.tokens;
+                const barLength = Math.round((tokens / maxTokens) * 30);
+                const bar = '█'.repeat(barLength);
+                console.log(`  ${date}  ${bar} ${tokens.toLocaleString()}`);
+            });
+            console.log('');
+            break;
+        default:
+            // summary
+            console.log(JSON.stringify(stats, null, 2));
+    }
+}
+
+// 运行
+main();

package/uninstall.js

@@ -30,9 +30,8 @@ function getHomeDir() {
 function uninstallFiles() {
     const home = getHomeDir();
     const files = [
-        path.join(home, '.claude', 'statusline.sh'),
-        path.join(home, '.claude', 'scripts', 'token-history.sh'),
-        path.join(home, '.claude', 'statusline-wrapper.sh'),
+        path.join(home, '.claude', 'statusline.js'),
+        path.join(home, '.claude', 'scripts', 'token-history.js'),
         // 命令文件
         path.join(home, '.claude', 'commands', 'burn-your-money-stats.md'),
         path.join(home, '.claude', 'commands', 'burn-your-money-today.md'),