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

package/src/token-history.sh

@@ -1,268 +0,0 @@
-#!/bin/bash
-# Burn Your Money - Token 历史统计脚本
-# 从 stats-cache.json 读取历史数据，返回今日/本周/本月统计
-
-set -e
-
-# 设置 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
-    echo "Error: jq not found. Please install jq first:"
-    echo "  Windows: npm install -g @winston.wan/burn-your-money"
-    echo "  macOS: brew install jq"
-    echo "  Linux: sudo apt-get install jq"
-    exit 1
-fi
-
-# 文件路径
-STATS_CACHE="$HOME/.claude/stats-cache.json"
-TODAY=$(date +%Y-%m-%d)
-CURRENT_MONTH=$(date +%Y-%m)
-LAST_MONTH=$(date -d "1 month ago" +%Y-%m 2>/dev/null || python3 -c "from datetime import datetime; print((datetime.now().replace(day=1) - __import__('datetime').timedelta(days=1)).strftime('%Y-%m'))")
-
-# 获取本周一的日期
-get_week_start() {
-    local day_of_week=$(date +%u)  # 1=Monday, 7=Sunday
-    local days_to_subtract=$((day_of_week - 1))
-    date -d "$days_to_subtract days ago" +%Y-%m-%d 2>/dev/null || \
-    python3 -c "from datetime import datetime, timedelta; d = datetime.now() - timedelta(days=$(date +%u) - 1); print(d.strftime('%Y-%m-%d'))"
-}
-
-# 获取本月第一天
-get_month_start() {
-    date +%Y-%m-01
-}
-
-WEEK_START=$(get_week_start)
-MONTH_START=$(get_month_start)
-LAST_MONTH_START=$(date -d "$LAST_MONTH-01" +%Y-%m-%d 2>/dev/null || echo "$LAST_MONTH-01")
-
-# 从 stats-cache.json 读取统计数据
-read_stats() {
-    if [ ! -f "$STATS_CACHE" ]; then
-        "$JQ_CMD" -n '{
-            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: []
-        }'
-        return
-    fi
-
-    # 获取今日 token：总计 - 历史每日总和（排除今天）
-    # 修改：聚合所有模型的 input 和 output tokens
-    local total_io_tokens=$(cat "$STATS_CACHE" | $JQ_CMD -r '[.modelUsage[] | (.inputTokens + .outputTokens)] | add // 0' 2>/dev/null || echo "0")
-    
-    # 修改：聚合每日数据中所有模型的 token
-    local history_sum=$(cat "$STATS_CACHE" | $JQ_CMD -r "[.dailyModelTokens[] | select(.date != \"$TODAY\") | (.tokensByModel[] // 0)] | add // 0" 2>/dev/null || echo "0")
-    local today_data=$((total_io_tokens - history_sum))
-    [ "$today_data" -lt 0 ] && today_data=0
-
-    # 本周 token（排除今天，用上面的 today_data）
-    local week_tokens=$(cat "$STATS_CACHE" | $JQ_CMD -r "[.dailyModelTokens[] | select(.date >= \"$WEEK_START\" and .date != \"$TODAY\") | (.tokensByModel[] // 0)] | add // 0" 2>/dev/null || echo "0")
-    week_tokens=$((week_tokens + today_data))
-
-    # 本月 token
-    local month_tokens=$(cat "$STATS_CACHE" | $JQ_CMD -r "[.dailyModelTokens[] | select(.date >= \"$MONTH_START\" and .date != \"$TODAY\") | (.tokensByModel[] // 0)] | add // 0" 2>/dev/null || echo "0")
-    month_tokens=$((month_tokens + today_data))
-
-    # 上月 token
-    local last_month_tokens=$(cat "$STATS_CACHE" | $JQ_CMD -r "[.dailyModelTokens[] | select(.date >= \"$LAST_MONTH_START\" and .date < \"$MONTH_START\") | (.tokensByModel[] // 0)] | add // 0" 2>/dev/null || echo "0")
-
-    # 计算全部历史费用（使用 jq 进行精确计算，避免 awk 大数字精度问题）
-    # 修改：聚合所有模型的 usage
-    local total_input=$(cat "$STATS_CACHE" | $JQ_CMD -r '[.modelUsage[].inputTokens // 0] | add // 0' 2>/dev/null || echo "0")
-    local total_output=$(cat "$STATS_CACHE" | $JQ_CMD -r '[.modelUsage[].outputTokens // 0] | add // 0' 2>/dev/null || echo "0")
-    local total_cache_read=$(cat "$STATS_CACHE" | $JQ_CMD -r '[.modelUsage[].cacheReadInputTokens // 0] | add // 0' 2>/dev/null || echo "0")
-    
-    # 估算总费用 (简单按 3/15/0.3 规则，虽然不同模型费率不同，但作为估算足够)
-    # 如果要精确需针对模型 loop，这里简化为聚合
-    local total_cost=$(cat "$STATS_CACHE" | $JQ_CMD -r '
-        [.modelUsage[]] | map(
-            ((.inputTokens // 0) / 1000000 * 3) +
-            ((.outputTokens // 0) / 1000000 * 15) +
-            ((.cacheReadInputTokens // 0) / 1000000 * 0.3)
-        ) | add | floor * 100 / 100
-    ' 2>/dev/null || echo "0")
-
-    # 总 token 数量（包含 cache）
-    local total_tokens_all=$((total_input + total_output + total_cache_read))
-
-    # 计算各周期费用（按比例分摊）
-    local week_cost="0.00"
-    local month_cost="0.00"
-    local last_month_cost="0.00"
-    local today_cost="0.00"
-
-    if [ "$total_io_tokens" != "0" ] && [ "$total_io_tokens" != "" ]; then
-        week_cost=$(awk "BEGIN {printf \"%.2f\", ${week_tokens} / ${total_io_tokens} * ${total_cost}}")
-        month_cost=$(awk "BEGIN {printf \"%.2f\", ${month_tokens} / ${total_io_tokens} * ${total_cost}}")
-        if [ "$last_month_tokens" -gt 0 ]; then
-            last_month_cost=$(awk "BEGIN {printf \"%.2f\", ${last_month_tokens} / ${total_io_tokens} * ${total_cost}}")
-        fi
-    else
-        week_cost=$total_cost
-        month_cost=$total_cost
-    fi
-
-    if [ "$week_tokens" != "0" ] && [ "$week_tokens" != "" ]; then
-        today_cost=$(awk "BEGIN {printf \"%.2f\", ${today_data} / ${week_tokens} * ${week_cost}}")
-    fi
-
-    # 获取每日数据（用于趋势图）
-    # 修改：聚合每日所有模型的 tokens
-    local daily_data=$(cat "$STATS_CACHE" | $JQ_CMD -c "[.dailyModelTokens[-7:][] | {date: .date, tokens: ([.tokensByModel[]] | add // 0)}]" 2>/dev/null || echo "[]")
-
-    # 输出 JSON
-    "$JQ_CMD" -n \
-        --arg today_tokens "$today_data" \
-        --arg today_cost "$today_cost" \
-        --arg week_tokens "$week_tokens" \
-        --arg week_cost "$week_cost" \
-        --arg month_tokens "$month_tokens" \
-        --arg month_cost "$month_cost" \
-        --arg last_month_tokens "$last_month_tokens" \
-        --arg last_month_cost "$last_month_cost" \
-        --arg total_tokens_all "$total_tokens_all" \
-        --arg total_cost "$total_cost" \
-        --argjson daily_data "$daily_data" \
-        '{
-            today_tokens: ($today_tokens | tonumber),
-            today_cost: ($today_cost | tonumber),
-            week_tokens: ($week_tokens | tonumber),
-            week_cost: ($week_cost | tonumber),
-            month_tokens: ($month_tokens | tonumber),
-            month_cost: ($month_cost | tonumber),
-            last_month_tokens: ($last_month_tokens | tonumber),
-            last_month_cost: ($last_month_cost | tonumber),
-            total_tokens_all: ($total_tokens_all | tonumber),
-            total_cost: ($total_cost | tonumber),
-            daily_data: $daily_data
-        }'
-}
-
-# 生成 ASCII 趋势图
-generate_trend() {
-    local data="$1"
-    local max_tokens=0
-
-    # 找到最大值
-    for row in $(echo "$data" | $JQ_CMD -r '.[].tokens'); do
-        if [ "$row" -gt "$max_tokens" ]; then
-            max_tokens=$row
-        fi
-    done
-
-    # 生成图表
-    echo ""
-    echo "📊 过去 7 天 Token 趋势:"
-    echo ""
-
-    local count=0
-    for entry in $(echo "$data" | $JQ_CMD -c '.[]'); do
-        local date=$(echo "$entry" | $JQ_CMD -r '.date' | cut -c 6-)
-        local tokens=$(echo "$entry" | $JQ_CMD -r '.tokens')
-
-        # 计算条形长度
-        if [ "$max_tokens" -gt 0 ]; then
-            local bars=$(awk "BEGIN {printf \"%.0f\", $tokens / $max_tokens * 20}")
-        else
-            local bars=0
-        fi
-
-        # 生成条形
-        local bar=""
-        for i in $(seq 1 20); do
-            if [ "$i" -le "$bars" ]; then
-                bar+="█"
-            else
-                bar+="░"
-            fi
-        done
-
-        # 格式化 token 数量
-        local tokens_display=$(echo "$tokens" | awk '{
-            if ($1 >= 1000000) printf "%.1fM", $1/1000000
-            else if ($1 >= 1000) printf "%.1fK", $1/1000
-            else printf "%d", $1
-        }')
-
-        printf "  %s  │ %s │ %s\n" "$date" "$bar" "$tokens_display"
-        count=$((count + 1))
-    done
-    echo ""
-}
-
-# 导出数据
-export_data() {
-    local format="$1"
-    local stats=$(read_stats)
-
-    case "$format" in
-        csv)
-            echo "date,tokens,cost"
-            echo "$stats" | $JQ_CMD -r '.daily_data[] | "\(.date),\(.tokens),\(.tokens * 0.000006)"'
-            ;;
-        json)
-            echo "$stats" | $JQ_CMD '.'
-            ;;
-        *)
-            echo "支持的格式: csv, json"
-            exit 1
-            ;;
-    esac
-}
-
-# 根据 mode 返回不同数据
-MODE=${1:-summary}
-
-case "$MODE" in
-    today_tokens)
-        read_stats | $JQ_CMD -r '.today_tokens'
-        ;;
-    today_cost)
-        read_stats | $JQ_CMD -r '.today_cost'
-        ;;
-    week_tokens)
-        read_stats | $JQ_CMD -r '.week_tokens'
-        ;;
-    week_cost)
-        read_stats | $JQ_CMD -r '.week_cost'
-        ;;
-    month_tokens)
-        read_stats | $JQ_CMD -r '.month_tokens'
-        ;;
-    month_cost)
-        read_stats | $JQ_CMD -r '.month_cost'
-        ;;
-    last_month_tokens)
-        read_stats | $JQ_CMD -r '.last_month_tokens'
-        ;;
-    last_month_cost)
-        read_stats | $JQ_CMD -r '.last_month_cost'
-        ;;
-    trend)
-        read_stats | $JQ_CMD -r '.daily_data'
-        ;;
-    chart)
-        generate_trend "$(read_stats | $JQ_CMD -r '.daily_data')"
-        ;;
-    export)
-        export_data "$2"
-        ;;
-    summary|json|*)
-        read_stats
-        ;;
-esac