@winston.wan/burn-your-money 2.0.4 → 2.1.0

package/src/statusline.sh

@@ -1,205 +1,250 @@
-#!/bin/bash
-# Burn Your Money - Claude Code 状态栏脚本
-# 格式：[模型] 今日：token 费用 🔥速度 | 总计：token 费用
-
-# 配置文件
-CONFIG_FILE="$HOME/.claude/burn-your-money-config.json"
-SESSION_STATE="$HOME/.claude/cache/session-state.json"
-
-# 初始化配置
-init_config() {
-    if [ ! -f "$CONFIG_FILE" ]; then
-        jq -n '{
-            theme: "fire",
-            alert_daily: 10.0,
-            alert_weekly: 50.0,
-            show_burn_rate: true,
-            show_eta: true,
-            show_trend: false
-        }' > "$CONFIG_FILE"
-    fi
-}
-
-# 读取配置
-init_config
-THEME=$(jq -r '.theme // "fire"' "$CONFIG_FILE")
-ALERT_DAILY=$(jq -r '.alert_daily // 10' "$CONFIG_FILE")
-SHOW_BURN_RATE=$(jq -r '.show_burn_rate // true' "$CONFIG_FILE")
-SHOW_ETA=$(jq -r '.show_eta // true' "$CONFIG_FILE")
-
-# 主题颜色定义
-case "$THEME" in
-    fire)
-        TODAY_COLOR="0;31"    # 红色
-        COST_COLOR="0;31"     # 红色
-        BURN_COLOR="0;31"     # 红色
-        ;;
-    ocean)
-        TODAY_COLOR="0;34"    # 蓝色
-        COST_COLOR="0;36"     # 青色
-        BURN_COLOR="0;36"     # 青色
-        ;;
-    forest)
-        TODAY_COLOR="0;32"    # 绿色
-        COST_COLOR="0;33"     # 黄色
-        BURN_COLOR="0;32"     # 绿色
-        ;;
-    golden)
-        TODAY_COLOR="0;33"    # 黄色
-        COST_COLOR="0;33"     # 黄色
-        BURN_COLOR="1;33"     # 亮黄色
-        ;;
-    *)
-        TODAY_COLOR="0;36"    # 默认青色
-        COST_COLOR="0;34"     # 默认蓝色
-        BURN_COLOR="0;31"     # 默认红色
-        ;;
-esac
-
-# 格式化数字（添加 K/M 后缀）
-format_number() {
-    local num=$1
-    if [ "$num" -ge 1000000000 ]; then
-        awk "BEGIN {printf \"%.1fB\", $num/1000000000}"
-    elif [ "$num" -ge 1000000 ]; then
-        awk "BEGIN {printf \"%.1fM\", $num/1000000}"
-    elif [ "$num" -ge 1000 ]; then
-        awk "BEGIN {printf \"%.1fK\", $num/1000}"
-    else
-        echo "$num"
-    fi
-}
-
-# 格式化费用
-format_cost() {
-    local cost=$1
-    awk "BEGIN {printf \"\$%.2f\", $cost}"
-}
-
-# 格式化燃烧速度
-format_burn_rate() {
-    local rate=$1
-    if [ "$rate" -ge 1000 ]; then
-        awk "BEGIN {printf \"%.1fK tok/s\", $rate/1000}"
-    else
-        awk "BEGIN {printf \"%.0f tok/s\", $rate}"
-    fi
-}
-
-# 获取或创建会话状态
-get_session_start_time() {
-    local session_id=$(jq -r '.session_id // "default"' "$TEMP_JSON" 2>/dev/null || echo "default")
-    local now=$(date +%s)
-
-    if [ -f "$SESSION_STATE" ]; then
-        local cached_id=$(jq -r '.session_id // ""' "$SESSION_STATE" 2>/dev/null || echo "")
-        local cached_time=$(jq -r '.start_time // 0' "$SESSION_STATE" 2>/dev/null || echo "0")
-
-        if [ "$cached_id" = "$session_id" ] && [ "$cached_time" -gt 0 ]; then
-            echo "$cached_time"
-            return
-        fi
-    fi
-
-    # 新会话，记录开始时间
-    echo "$now"
-    jq -n --arg id "$session_id" --arg time "$now" '{session_id: $id, start_time: ($time | tonumber)}' > "$SESSION_STATE"
-    echo "$now"
-}
-
-# 主逻辑：使用临时文件存储 stdin
-TEMP_JSON=$(mktemp)
-trap "rm -f $TEMP_JSON" EXIT
-cat > "$TEMP_JSON"
-
-# 读取模型名称
-model=$(jq -r '.model.display_name // "Unknown"' "$TEMP_JSON" 2>/dev/null || echo "Unknown")
-
-# 读取当前会话的 token 数据（从 context_window）
-current_input=$(jq -r '.context_window.total_input_tokens // 0' "$TEMP_JSON" 2>/dev/null || echo "0")
-current_output=$(jq -r '.context_window.total_output_tokens // 0' "$TEMP_JSON" 2>/dev/null || echo "0")
-current_session_tokens=$((current_input + current_output))
-
-# 计算燃烧速度
-session_start=$(get_session_start_time)
-current_time=$(date +%s)
-elapsed=$((current_time - session_start))
-
-if [ "$elapsed" -gt 0 ] && [ "$current_session_tokens" -gt 0 ]; then
-    burn_rate=$(awk "BEGIN {printf \"%.0f\", ${current_session_tokens} / ${elapsed}}")
-else
-    burn_rate=0
-fi
-
-# 获取历史数据（缓存 30 秒）
-history_cache="$HOME/.claude/cache/history-cache.json"
-now=$(date +%s)
-cache_age=999999
-if [ -f "$history_cache" ]; then
-    cache_time=$(stat -c %Y "$history_cache" 2>/dev/null || stat -f %m "$history_cache" 2>/dev/null || echo "0")
-    cache_age=$((now - cache_time))
-fi
-
-if [ $cache_age -gt 30 ]; then
-    ~/.claude/scripts/token-history.sh summary > "$history_cache" 2>/dev/null || echo '{}' > "$history_cache"
-fi
-
-# 读取历史数据并加上当前会话
-history_today_tokens=$(jq -r '.today_tokens // 0' "$history_cache" 2>/dev/null || echo "0")
-history_today_cost=$(jq -r '.today_cost // 0' "$history_cache" 2>/dev/null || echo "0")
-
-# 今日数据 = 历史数据 + 当前会话
-today_tokens=$((history_today_tokens + current_session_tokens))
-# 简单估算今日费用（按比例）
-if [ "$history_today_tokens" -gt 0 ] && [ "$history_today_cost" != "0" ]; then
-    today_cost=$(awk "BEGIN {printf \"%.2f\", ${history_today_cost} / ${history_today_tokens} * ${today_tokens}}")
-elif [ "$today_tokens" -gt 0 ]; then
-    # 如果没有历史费用数据，用平均价格估算
-    today_cost=$(awk "BEGIN {printf \"%.2f\", ${today_tokens} * 0.000006}")
-else
-    today_cost="0.00"
-fi
-
-total_tokens_all=$(jq -r '.total_tokens_all // 0' "$history_cache" 2>/dev/null || echo "0")
-total_cost=$(jq -r '.week_cost // 0' "$history_cache" 2>/dev/null || echo "0")
-
-# 检查费用警报
-alert_needed=false
-if awk "BEGIN {exit !($today_cost >= $ALERT_DAILY)}"; then
-    alert_needed=true
-fi
-
-# 构建输出
-output=""
-
-# 模型名称
-if [ "$alert_needed" = true ]; then
-    output+="\\033[1;31m[$model]\\033[0m "  # 红色闪烁
-else
-    output+="\\033[0;90m[$model]\\033[0m "
-fi
-
-# 今日数据
-today_display=$(format_number $today_tokens)
-today_cost_display=$(format_cost $today_cost)
-if [ "$today_tokens" -gt 0 ] || awk "BEGIN {exit !($today_cost > 0)}"; then
-    output+="\\033[${TODAY_COLOR}m今日：${today_display}\\033[0m "
-    output+="\\033[${COST_COLOR}m${today_cost_display}\\033[0m "
-else
-    output+="\\033[0;90m今日：0\\033[0m "
-    output+="\\033[0;90m\$0.00\\033[0m "
-fi
-
-# 燃烧速度
-if [ "$SHOW_BURN_RATE" = true ] && [ "$burn_rate" -gt 0 ]; then
-    burn_display=$(format_burn_rate $burn_rate)
-    output+="\\033[${BURN_COLOR}m🔥${burn_display}\\033[0m "
-fi
-
-# 总计数据
-total_display=$(format_number $total_tokens_all)
-total_cost_display=$(format_cost $total_cost)
-output+="| \\033[0;90m总计：${total_display}\\033[0m "
-output+="\\033[0;90m${total_cost_display}\\033[0m "
-
-printf "%b" "$output"
+#!/bin/bash
+# Burn Your Money - Claude Code 状态栏脚本
+# 格式：[模型] 今日：token 费用 🔥速度 | 总计：token 费用
+
+# 设置 jq 命令路径（支持自定义安装的 jq）
+if [ -n "$JQ_PATH" ]; then
+    JQ_CMD="$JQ_PATH"
+elif [ -f "$HOME/.claude/bin/jq.exe" ]; then
+    JQ_CMD="$HOME/.claude/bin/jq.exe"
+elif [ -f "$HOME/.claude/bin/jq" ]; then
+    JQ_CMD="$HOME/.claude/bin/jq"
+else
+    JQ_CMD="jq"
+fi
+
+# 验证 jq 是否可用
+if ! command -v "$JQ_CMD" >/dev/null 2>&1; then
+    # jq 不可用，返回简单错误信息
+    printf "[BurnYourMoney] Error: jq not found. Run: npm install -g @winston.wan/burn-your-money"
+    exit 0
+fi
+
+# 配置文件
+CONFIG_FILE="$HOME/.claude/burn-your-money-config.json"
+SESSION_STATE="$HOME/.claude/cache/session-state.json"
+TODAY_STATE="$HOME/.claude/cache/today-state.json"
+HISTORY_CACHE="$HOME/.claude/cache/history-cache.json"
+
+# 初始化配置
+init_config() {
+    if [ ! -f "$CONFIG_FILE" ]; then
+        $JQ_CMD -n '{
+            theme: "fire",
+            alert_daily: 10.0,
+            alert_weekly: 50.0,
+            show_burn_rate: true,
+            show_eta: true,
+            show_trend: false
+        }' > "$CONFIG_FILE"
+    fi
+}
+
+# 1. 批量读取配置 (One JQ pass)
+init_config
+if [ -f "$CONFIG_FILE" ]; then
+    # 使用 eval 批量赋值变量
+    eval $("$JQ_CMD" -r '@sh "THEME=\(.theme // "fire") ALERT_DAILY=\(.alert_daily // 10) SHOW_BURN_RATE=\(.show_burn_rate // true)"' "$CONFIG_FILE" 2>/dev/null)
+else
+    THEME="fire"
+    ALERT_DAILY=10
+    SHOW_BURN_RATE=true
+fi
+
+# 主题颜色定义
+case "$THEME" in
+    fire)   TODAY_COLOR="0;31"; COST_COLOR="0;31"; BURN_COLOR="0;31" ;;
+    ocean)  TODAY_COLOR="0;34"; COST_COLOR="0;36"; BURN_COLOR="0;36" ;;
+    forest) TODAY_COLOR="0;32"; COST_COLOR="0;33"; BURN_COLOR="0;32" ;;
+    golden) TODAY_COLOR="0;33"; COST_COLOR="0;33"; BURN_COLOR="1;33" ;;
+    *)      TODAY_COLOR="0;36"; COST_COLOR="0;34"; BURN_COLOR="0;31" ;;
+esac
+
+# 格式化函数 (纯 Bash/Awk 实现，无外部进程调用)
+format_number() {
+    local num=$1
+    if [ "$num" -ge 1000000000 ]; then awk "BEGIN {printf \"%.1fB\", $num/1000000000}";
+    elif [ "$num" -ge 1000000 ]; then awk "BEGIN {printf \"%.1fM\", $num/1000000}";
+    elif [ "$num" -ge 1000 ]; then awk "BEGIN {printf \"%.1fK\", $num/1000}";
+    else echo "$num"; fi
+}
+
+format_cost() {
+    awk "BEGIN {printf \"\$%.2f\", $1}"
+}
+
+format_burn_rate() {
+    local rate=$1
+    if [ "$rate" -ge 1000 ]; then awk "BEGIN {printf \"%.1fK tok/s\", $rate/1000}";
+    else awk "BEGIN {printf \"%.0f tok/s\", $rate}"; fi
+}
+
+# 2. 核心数据解析 (One JQ pass from stdin)
+# 将 stdin 存入临时文件（避免大 JSON 管道卡顿）
+TEMP_JSON=$(mktemp)
+trap "rm -f $TEMP_JSON" EXIT
+cat > "$TEMP_JSON"
+
+# 一次性提取所有需要的字段
+# 注意：使用 @sh 确保字符串安全引用
+eval $("$JQ_CMD" -r '
+    @sh "model_name=\(.model.display_name // "Unknown")
+    session_id=\(.session_id // "default")
+    input=\(.context_window.total_input_tokens // 0)
+    output=\(.context_window.total_output_tokens // 0)
+    cache=\(.context_window.current_usage.cache_read_input_tokens // 0)
+    current_cost=\(.cost.total_cost_usd // 0)"
+' "$TEMP_JSON" 2>/dev/null)
+
+# 计算当前会话 Token 总量
+current_session_tokens=$((input + output + cache))
+
+# 3. 燃烧速度计算 (Burn Rate)
+current_time=$(date +%s)
+start_time_cache=0
+
+# 读取 Session State (尽可能不做写入，除非是新 Session)
+if [ -f "$SESSION_STATE" ]; then
+    eval $("$JQ_CMD" -r '@sh "cached_id=\(.session_id // "") cached_time=\(.start_time // 0)"' "$SESSION_STATE" 2>/dev/null)
+else
+    cached_id=""
+    cached_time=0
+fi
+
+if [ "$cached_id" = "$session_id" ] && [ "$cached_time" -ne 0 ]; then
+    start_time_cache=$cached_time
+else
+    # New session detected - 使用原子写入
+    start_time_cache=$current_time
+    SESSION_STATE_TMP="${SESSION_STATE}.tmp.$$"
+    $JQ_CMD -n --arg id "$session_id" --arg time "$current_time" '{session_id: $id, start_time: ($time | tonumber)}' > "$SESSION_STATE_TMP" 2>/dev/null && mv "$SESSION_STATE_TMP" "$SESSION_STATE"
+fi
+
+elapsed=$((current_time - start_time_cache))
+burn_rate=0
+if [ "$elapsed" -gt 0 ] && [ "$current_session_tokens" -gt 0 ]; then
+    burn_rate=$(awk "BEGIN {printf \"%.0f\", ${current_session_tokens} / ${elapsed}}")
+fi
+
+# 4. 历史数据缓存 (每 30 秒刷新)
+# 确保 cache 目录存在
+mkdir -p "$(dirname "$HISTORY_CACHE")" 2>/dev/null
+
+now=$(date +%s)
+cache_age=999999
+if [ -f "$HISTORY_CACHE" ]; then
+    # Portable modification time check
+    if [ "$(uname)" = "Darwin" ]; then
+        cache_time=$(stat -f %m "$HISTORY_CACHE" 2>/dev/null || echo 0)
+    else
+        cache_time=$(stat -c %Y "$HISTORY_CACHE" 2>/dev/null || echo 0)
+    fi
+    cache_age=$((now - cache_time))
+else
+    echo '{}' > "$HISTORY_CACHE"
+fi
+
+# 如果缓存太旧，并在后台刷新（虽然后台刷新可能在 statusline 不合适，但我们可以同步调用历史脚本）
+# 为了性能，这里我们仍然做同步调用，但减少频率
+if [ $cache_age -gt 30 ]; then
+    # 重新生成历史数据
+    ~/.claude/scripts/token-history.sh summary > "$HISTORY_CACHE" 2>/dev/null
+fi
+
+# 读取历史总计 (One pass)
+eval $("$JQ_CMD" -r '@sh "history_total_tokens=\(.total_tokens_all // 0) history_total_cost=\(.total_cost // 0)"' "$HISTORY_CACHE" 2>/dev/null)
+
+# 5. 今日数据持久化 (One-pass Read & Write)
+# 逻辑：读取旧状态 -> 计算差值 -> 写入新状态
+today_date=$(date +%Y-%m-%d)
+
+if [ -f "$TODAY_STATE" ]; then
+    eval $("$JQ_CMD" -r '@sh "saved_date=\(.date // "") saved_tokens=\(.tokens // 0) saved_cost=\(.cost // 0) last_session_tokens=\(.last_session_tokens // 0) last_session_cost=\(.last_session_cost // 0)"' "$TODAY_STATE" 2>/dev/null)
+else
+    saved_date=""
+    saved_tokens=0
+    saved_cost=0
+    last_session_tokens=0
+    last_session_cost=0
+fi
+
+if [ "$saved_date" != "$today_date" ]; then
+    # 新的一天，重置累计数据
+    saved_tokens=0
+    saved_cost=0
+    last_session_tokens=0
+    last_session_cost=0
+fi
+
+# 计算增量
+token_increment=$((current_session_tokens - last_session_tokens))
+if [ "$token_increment" -lt 0 ]; then token_increment=0; fi # 防御性编程
+
+# 浮点数增量计算
+cost_increment=$(awk "BEGIN {print ($current_cost - $last_session_cost < 0 ? 0 : $current_cost - $last_session_cost)}")
+
+# 计算新的今日累计
+new_today_tokens=$((saved_tokens + token_increment))
+new_today_cost=$(awk "BEGIN {printf \"%.5f\", $saved_cost + $cost_increment}")
+
+# 写入新状态 (一次性写入，使用原子写入避免并发问题)
+# 先写入临时文件，然后重命名，确保原子性
+TODAY_STATE_TMP="${TODAY_STATE}.tmp.$$"
+$JQ_CMD -n \
+    --arg date "$today_date" \
+    --arg tokens "$new_today_tokens" \
+    --arg cost "$new_today_cost" \
+    --arg last_sess_tok "$current_session_tokens" \
+    --arg last_sess_cost "$current_cost" \
+    '{
+        date: $date,
+        tokens: ($tokens|tonumber),
+        cost: ($cost|tonumber),
+        last_session_tokens: ($last_sess_tok|tonumber),
+        last_session_cost: ($last_sess_cost|tonumber)
+    }' > "$TODAY_STATE_TMP" 2>/dev/null && mv "$TODAY_STATE_TMP" "$TODAY_STATE"
+
+# 6. 最终计算与显示
+total_tokens_all=$((history_total_tokens + current_session_tokens))
+total_cost_all=$(awk "BEGIN {print $history_total_cost + $current_cost}")
+
+# 警报检查
+is_alert=false
+if awk "BEGIN {exit !($new_today_cost >= $ALERT_DAILY)}"; then
+    is_alert=true
+fi
+
+# 构建输出字符串
+output=""
+
+# [Model]
+if [ "$is_alert" = true ]; then
+    output+="\\033[1;31m[${model_name}]\\033[0m "
+else
+    output+="\\033[0;90m[${model_name}]\\033[0m "
+fi
+
+# Today
+d_today_tokens=$(format_number $new_today_tokens)
+d_today_cost=$(format_cost $new_today_cost)
+
+if [ "$new_today_tokens" -gt 0 ]; then
+    output+="\\033[${TODAY_COLOR}m今日：${d_today_tokens}\\033[0m "
+    output+="\\033[${COST_COLOR}m${d_today_cost}\\033[0m "
+else
+    output+="\\033[0;90m今日：0\\033[0m \\033[0;90m\$0.00\\033[0m "
+fi
+
+# Burn Rate
+if [ "$SHOW_BURN_RATE" = "true" ] && [ "$burn_rate" -gt 0 ]; then
+    d_burn=$(format_burn_rate $burn_rate)
+    output+="\\033[${BURN_COLOR}m🔥${d_burn}\\033[0m "
+fi
+
+# Total
+d_total_tokens=$(format_number $total_tokens_all)
+d_total_cost=$(format_cost $total_cost_all)
+output+="| \\033[0;90m总计：${d_total_tokens}\\033[0m \\033[0;90m${d_total_cost}\\033[0m "
+
+# Print final result
+printf "%b" "$output"