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

package/install.sh

@@ -1,271 +0,0 @@
-#!/bin/bash
-# Burn Your Money - 一键安装脚本
-# 用法: curl -fsSL https://raw.githubusercontent.com/winston-wwzhen/burn-your-money/main/install.sh | bash
-
-set -e
-
-# 颜色定义
-RED='\033[0;31m'
-GREEN='\033[0;32m'
-YELLOW='\033[1;33m'
-BLUE='\033[0;34m'
-PURPLE='\033[0;35m'
-CYAN='\033[0;36m'
-NC='\033[0m' # No Color
-
-# 打印带颜色的消息
-print_header() {
-    echo ""
-    echo -e "${PURPLE}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
-    echo -e "${PURPLE}  $1${NC}"
-    echo -e "${PURPLE}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
-    echo ""
-}
-
-print_step() {
-    echo -e "${CYAN}▶ $1${NC}"
-}
-
-print_success() {
-    echo -e "${GREEN}✓ $1${NC}"
-}
-
-print_warning() {
-    echo -e "${YELLOW}⚠ $1${NC}"
-}
-
-print_error() {
-    echo -e "${RED}✗ $1${NC}"
-}
-
-# 显示 Logo
-show_logo() {
-    echo ""
-    echo -e "${RED}"
-    echo "  ██╗    ██╗███████╗██████╗  ██████╗ ███████╗██████╗ ██████╗ "
-    echo "  ██║    ██║██╔════╝██╔══██╗██╔═══██╗██╔════╝██╔══██╗██╔══██╗"
-    echo "  ██║ █╗ ██║█████╗  ██████╔╝██║   ██║█████╗  ██████╔╝██║  ██║"
-    echo "  ██║███╗██║██╔══╝  ██╔══██╗██║   ██║██╔══╝  ██╔══██╗██║  ██║"
-    echo "  ╚███╔███╔╝███████╗██████╔╝╚██████╔╝███████╗██████╔╝██████╔╝"
-    echo "   ╚══╝╚══╝ ╚══════╝╚═════╝  ╚═════╝ ╚══════╝╚═════╝ ╚═════╝ "
-    echo ""
-    echo -e "${CYAN}  💸 Burn Your Money - 看着你的钱包燃烧${NC}"
-    echo ""
-}
-
-# 检查依赖
-check_dependencies() {
-    print_step "检查依赖..."
-
-    # 检查 jq
-    if ! command -v jq &> /dev/null; then
-        print_error "jq 未安装！"
-        echo ""
-        echo "请先安装 jq："
-        echo "  Ubuntu/Debian: sudo apt-get install jq"
-        echo "  macOS: brew install jq"
-        echo "  Arch: sudo pacman -S jq"
-        echo ""
-        exit 1
-    fi
-    print_success "jq 已安装"
-
-    # 检查 Claude Code
-    if ! command -v claude &> /dev/null; then
-        print_warning "Claude Code 未找到"
-    else
-        CLAUDE_VERSION=$(claude --version 2>/dev/null || echo "unknown")
-        print_success "Claude Code 版本: $CLAUDE_VERSION"
-    fi
-}
-
-# 创建目录结构
-create_directories() {
-    print_step "创建目录结构..."
-
-    mkdir -p ~/.claude/scripts
-    mkdir -p ~/.claude/cache
-
-    print_success "目录已创建"
-}
-
-# 从 GitHub 下载文件
-download_file() {
-    local url="$1"
-    local dest="$2"
-    local max_retries=3
-    local retry=0
-
-    while [ $retry -lt $max_retries ]; do
-        if curl -fsSL "$url" -o "$dest" 2>/dev/null; then
-            return 0
-        fi
-        retry=$((retry + 1))
-        if [ $retry -lt $max_retries ]; then
-            sleep 1
-        fi
-    done
-    return 1
-}
-
-# 安装脚本文件
-install_scripts() {
-    print_step "安装脚本文件..."
-
-    # 获取脚本目录
-    SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
-
-    # 检查本地文件是否存在
-    local use_remote=false
-    if [ ! -f "$SCRIPT_DIR/src/statusline.sh" ] || [ ! -f "$SCRIPT_DIR/src/token-history.sh" ]; then
-        print_warning "本地脚本文件不存在，从 GitHub 下载..."
-        use_remote=true
-    fi
-
-    # GitHub 原始文件 URL 基础路径
-    local BASE_URL="https://raw.githubusercontent.com/winston-wwzhen/burn-your-money/main"
-
-    # 安装 statusline.sh
-    if [ "$use_remote" = true ]; then
-        print_step "下载 statusline.sh..."
-        if download_file "$BASE_URL/src/statusline.sh" ~/.claude/statusline.sh; then
-            chmod +x ~/.claude/statusline.sh
-            print_success "statusline.sh 已下载并安装"
-        else
-            print_error "下载 statusline.sh 失败，请检查网络连接"
-            exit 1
-        fi
-    else
-        cp "$SCRIPT_DIR/src/statusline.sh" ~/.claude/statusline.sh
-        chmod +x ~/.claude/statusline.sh
-        print_success "statusline.sh 已安装"
-    fi
-
-    # 安装 token-history.sh
-    mkdir -p ~/.claude/scripts
-    if [ "$use_remote" = true ]; then
-        print_step "下载 token-history.sh..."
-        if download_file "$BASE_URL/src/token-history.sh" ~/.claude/scripts/token-history.sh; then
-            chmod +x ~/.claude/scripts/token-history.sh
-            print_success "token-history.sh 已下载并安装"
-        else
-            print_error "下载 token-history.sh 失败，请检查网络连接"
-            exit 1
-        fi
-    else
-        cp "$SCRIPT_DIR/src/token-history.sh" ~/.claude/scripts/token-history.sh
-        chmod +x ~/.claude/scripts/token-history.sh
-        print_success "token-history.sh 已安装"
-    fi
-}
-
-# 配置 settings.json
-configure_settings() {
-    print_step "配置 Claude Code..."
-
-    SETTINGS_FILE="$HOME/.claude/settings.json"
-
-    # 如果文件不存在，创建一个
-    if [ ! -f "$SETTINGS_FILE" ]; then
-        echo "{}" > "$SETTINGS_FILE"
-    fi
-
-    # 使用 jq 添加 statusline 配置
-    if command -v jq &> /dev/null; then
-        tmp_file=$(mktemp)
-        jq '.statusLine = {"type": "command", "command": "~/.claude/statusline.sh"}' "$SETTINGS_FILE" > "$tmp_file"
-        mv "$tmp_file" "$SETTINGS_FILE"
-        print_success "settings.json 已更新"
-    else
-        print_warning "无法自动更新 settings.json，请手动添加："
-        echo ""
-        echo '{
-  "statusLine": {
-    "type": "command",
-    "command": "~/.claude/statusline.sh"
-  }
-}'
-        echo ""
-    fi
-}
-
-# 测试安装
-test_installation() {
-    print_step "测试安装..."
-
-    # 测试 statusline 脚本
-    test_json='{
-      "model": {"display_name": "Test"},
-      "context_window": {
-        "total_input_tokens": 10000,
-        "total_output_tokens": 2000,
-        "context_window_size": 200000,
-        "used_percentage": 6,
-        "remaining_percentage": 94
-      },
-      "cost": {"total_cost_usd": 0.05}
-    }'
-
-    if output=$(echo "$test_json" | ~/.claude/statusline.sh 2>/dev/null); then
-        print_success "statusline 脚本工作正常"
-        echo "  输出: $output"
-    else
-        print_error "statusline 脚本测试失败"
-        return 1
-    fi
-
-    # 测试 token-history 脚本
-    if ~/.claude/scripts/token-history.sh summary > /dev/null 2>&1; then
-        print_success "token-history 脚本工作正常"
-    else
-        print_warning "token-history 脚本测试失败（可能没有历史数据）"
-    fi
-}
-
-# 显示完成信息
-show_completion() {
-    print_header "安装完成！"
-
-    echo -e "${GREEN}Burn Your Money 已成功安装！${NC}"
-    echo ""
-    echo "📝 已安装的文件："
-    echo "  • ~/.claude/statusline.sh"
-    echo "  • ~/.claude/scripts/token-history.sh"
-    echo ""
-    echo "🔧 已更新配置："
-    echo "  • ~/.claude/settings.json"
-    echo ""
-    echo -e "${YELLOW}下一步：${NC}"
-    echo "  1. 重启 Claude Code"
-    echo "  2. 观察状态栏是否显示 Token 信息"
-    echo ""
-    echo -e "${CYAN}如果状态栏没有显示，请检查：${NC}"
-    echo "  • Claude Code 版本是否 >= 2.1.6"
-    echo "  • 运行: claude --version"
-    echo ""
-    echo -e "${PURPLE}反馈与支持：${NC}"
-    echo "  • GitHub: https://github.com/winston-wwzhen/burn-your-money"
-    echo "  • Issues: https://github.com/winston-wwzhen/burn-your-money/issues"
-    echo ""
-}
-
-# 主函数
-main() {
-    show_logo
-
-    print_header "开始安装 Burn Your Money..."
-
-    check_dependencies
-    create_directories
-    install_scripts
-    configure_settings
-    test_installation
-
-    show_completion
-
-    echo -e "${GREEN}开始烧钱吧！🔥💸${NC}"
-    echo ""
-}
-
-# 运行主函数
-main

package/src/statusline.sh

@@ -1,250 +0,0 @@
-#!/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"