yc

#!/usr/bin/env bash
# YesCode CLI
# 依赖: curl, jq

set -e

# ============================================================
# 颜色定义
# ============================================================

# ANSI 颜色码
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
BLUE='\033[0;34m'
MAGENTA='\033[0;35m'
CYAN='\033[0;36m'
WHITE='\033[0;37m'
GRAY='\033[0;90m'
BOLD='\033[1m'
RESET='\033[0m'

# ============================================================
# 配置管理
# ============================================================

prompt_api_key() {
    local show_message="${1:-true}"  # 默认显示提示信息
    local config_dir="$HOME/.yescode"
    local config_file="$config_dir/config.json"

    if [[ "$show_message" == "true" ]]; then
        echo
        echo -e "${YELLOW}未配置 API Key${RESET}"
        echo
    fi

    # 创建配置目录
    mkdir -p "$config_dir"

    # 提示用户输入
    read -p "请输入您的 API Key: " api_key

    # 验证输入
    if [[ -z "$api_key" ]]; then
        echo -e "${RED}错误: API Key 不能为空${RESET}"
        exit 1
    fi

    # 保存配置
    echo "{\"api_key\": \"$api_key\"}" > "$config_file"
    chmod 600 "$config_file"

    echo -e "${GREEN}✓ 配置已保存到: $config_file${RESET}"
    echo
}

load_config() {
    local config_file="$HOME/.yescode/config.json"

    # 如果配置文件不存在，提示用户输入
    if [[ ! -f "$config_file" ]]; then
        prompt_api_key
    fi

    # 使用 jq 读取 API Key
    API_KEY=$(jq -r '.api_key' "$config_file" 2>/dev/null)

    if [[ $? -ne 0 ]]; then
        echo -e "${RED}错误: 配置文件格式错误${RESET}"
        echo "请确保 $config_file 是有效的 JSON 格式"
        exit 1
    fi

    # 如果 API Key 为空或 null，提示用户输入
    if [[ -z "$API_KEY" || "$API_KEY" == "null" ]]; then
        prompt_api_key
        # 重新读取配置
        API_KEY=$(jq -r '.api_key' "$config_file" 2>/dev/null)
    fi

    # 固定的 API Base URL
    API_BASE_URL="https://co.yes.vg"
}

# ============================================================
# Loading 动画函数
# ============================================================

show_loading() {
    local pid=$1
    local spin='⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏'
    local i=0

    # debug 模式下不显示动画，直接等待
    if [[ "$DEBUG" == "true" ]]; then
        wait $pid
        return
    fi

    # 隐藏光标
    tput civis

    while kill -0 $pid 2>/dev/null; do
        i=$(( (i+1) % 10 ))
        printf "\r${CYAN}${spin:$i:1}${RESET}"
        sleep 0.1
    done

    # 清除loading行
    printf "\r\033[K"

    # 恢复光标
    tput cnorm
}

# ============================================================
# Debug 模式显示函数
# ============================================================

display_raw_response() {
    # 只在 debug 模式下显示
    if [[ "$DEBUG" != "true" ]]; then
        return
    fi

    local method="$1"
    local url="$2"
    local api_key_masked="$3"
    local http_code="$4"
    local headers="$5"
    local body="$6"
    local request_body="$7"  # 可选的请求体

    # 所有输出重定向到 stderr，避免污染 API 函数的返回值
    {
        echo
        echo -e "${BOLD}════════════════════════ 原始 HTTP 请求/响应 ════════════════════════${RESET}"
        echo
        echo -e "${CYAN}${BOLD}📤 请求信息${RESET}"
        echo -e "  方法: ${YELLOW}${method}${RESET}"
        echo -e "  URL: ${BLUE}${url}${RESET}"
        echo -e "  请求头:"
        echo -e "    ${WHITE}accept: application/json${RESET}"
        echo -e "    ${WHITE}X-API-Key: ${api_key_masked}${RESET}"
        if [[ "$method" == "PUT" || "$method" == "POST" ]]; then
            echo -e "    ${WHITE}Content-Type: application/json${RESET}"
        fi

        # 显示请求体（如果有）
        if [[ -n "$request_body" ]]; then
            echo
            echo -e "${CYAN}${BOLD}📄 请求体（JSON）:${RESET}"
            if command -v jq &> /dev/null; then
                echo "$request_body" | jq -C . 2>/dev/null || echo "$request_body"
            else
                echo "$request_body"
            fi
        fi
        echo
        echo -e "${CYAN}${BOLD}📥 响应信息${RESET}"
        echo -e "  状态码: ${GREEN}${http_code}${RESET}"

        if [[ -n "$headers" && "$headers" != "null" ]]; then
            echo -e "  响应头:"
            echo "$headers" | while IFS= read -r line; do
                if [[ -n "$line" ]]; then
                    echo -e "    ${WHITE}${line}${RESET}"
                fi
            done
        fi

        echo
        echo -e "${CYAN}${BOLD}📄 响应体（JSON）:${RESET}"

        # 使用 jq 格式化 JSON（带颜色）
        if command -v jq &> /dev/null; then
            echo "$body" | jq -C . 2>/dev/null || echo "$body"
        else
            echo "$body"
        fi

        echo
        echo -e "${BOLD}═══════════════════════════════════════════════════════════════${RESET}"
        echo
    } >&2
}

# ============================================================
# 颜色工具函数
# ============================================================

get_balance_color() {
    local balance=$1
    local threshold_low=20
    local threshold_medium=50

    if (( $(echo "$balance < $threshold_low" | bc -l) )); then
        echo "$RED"
    elif (( $(echo "$balance < $threshold_medium" | bc -l) )); then
        echo "$YELLOW"
    else
        echo "$GREEN"
    fi
}

get_spending_color() {
    local spent_percent=$1

    if (( $(echo "$spent_percent >= 95" | bc -l) )); then
        echo "$RED"
    elif (( $(echo "$spent_percent >= 80" | bc -l) )); then
        echo "$YELLOW"
    else
        echo "$GREEN"
    fi
}

create_progress_bar() {
    local current=$1
    local total=$2
    local width=30

    if (( $(echo "$total == 0" | bc -l) )); then
        local percent=0
        local filled=0
    else
        local percent=$(echo "scale=1; ($current / $total) * 100" | bc)
        local filled=$(echo "scale=0; ($current / $total) * $width" | bc)
    fi

    local empty=$((width - filled))
    local bar=$(printf '█%.0s' $(seq 1 $filled))$(printf '░%.0s' $(seq 1 $empty))

    local color=$(get_spending_color "$percent")
    echo -e "[${color}${bar}${RESET}] ${color}${BOLD}${percent}%${RESET}"
}

# ============================================================
# 系统检测与依赖安装
# ============================================================

detect_package_manager() {
    # 检测系统的包管理器并返回安装命令
    if command -v apt &> /dev/null; then
        echo "apt"
    elif command -v dnf &> /dev/null; then
        echo "dnf"
    elif command -v yum &> /dev/null; then
        echo "yum"
    elif command -v brew &> /dev/null; then
        echo "brew"
    elif command -v pacman &> /dev/null; then
        echo "pacman"
    elif command -v zypper &> /dev/null; then
        echo "zypper"
    else
        echo "unknown"
    fi
}

auto_install_dependency() {
    local dep=$1
    local pkg_manager=$(detect_package_manager)

    if [[ "$pkg_manager" == "unknown" ]]; then
        return 1
    fi

    read -p "是否自动安装 $dep? (y/n): " -r

    if [[ ! $REPLY =~ ^[Yy]$ ]]; then
        return 1
    fi

    echo "正在安装 $dep..."

    local install_cmd=""
    case "$pkg_manager" in
        apt)
            install_cmd="sudo apt install -y $dep"
            ;;
        dnf)
            install_cmd="sudo dnf install -y $dep"
            ;;
        yum)
            install_cmd="sudo yum install -y $dep"
            ;;
        brew)
            install_cmd="brew install $dep"
            ;;
        pacman)
            install_cmd="sudo pacman -S --noconfirm $dep"
            ;;
        zypper)
            install_cmd="sudo zypper install -y $dep"
            ;;
    esac

    if eval "$install_cmd"; then
        echo -e "${GREEN}✓ $dep 安装成功${RESET}"
        return 0
    else
        echo -e "${RED}✗ $dep 安装失败${RESET}"
        return 1
    fi
}

# ============================================================
# API 调用函数
# ============================================================

api_get_balance() {
    local url="${API_BASE_URL}/api/v1/user/balance"
    local method="GET"

    # 根据 debug 模式选择不同的 curl 参数
    local response
    if [[ "$DEBUG" == "true" ]]; then
        # debug 模式：获取响应头
        response=$(curl -s -i -w "\n%{http_code}" \
            -X GET \
            -H "accept: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    else
        # 普通模式：不获取响应头
        response=$(curl -s -w "\n%{http_code}" \
            -X GET \
            -H "accept: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    fi

    if [[ $? -ne 0 ]]; then
        echo -e "${RED}错误: 请求失败${RESET}" >&2
        return 1
    fi

    # 分离响应体和状态码
    local http_code=$(echo "$response" | tail -n1)
    local full_response=$(echo "$response" | head -n -1)

    # 在 debug 模式下分离响应头和响应体
    local headers=""
    local body=""
    if [[ "$DEBUG" == "true" ]]; then
        # 删除回车符（HTTP 响应使用 CRLF）
        full_response=$(echo "$full_response" | tr -d '\r')
        # 查找空行位置（响应头和响应体的分隔）
        local empty_line=$(echo "$full_response" | grep -n "^$" | head -1 | cut -d: -f1)
        if [[ -n "$empty_line" ]]; then
            headers=$(echo "$full_response" | head -n $((empty_line - 1)))
            body=$(echo "$full_response" | tail -n +$((empty_line + 1)))
        else
            # 如果没有找到空行，可能响应格式异常，全部作为 body
            body="$full_response"
        fi
    else
        body="$full_response"
    fi

    # 处理认证错误
    if [[ "$http_code" == "401" ]]; then
        echo "AUTH_ERROR_401"
        return 2
    fi

    # 处理其他错误
    if [[ "$http_code" != "200" ]]; then
        echo -e "${RED}获取余额失败: HTTP $http_code${RESET}" >&2
        echo "响应内容: $body" >&2
        return 1
    fi

    # 显示原始响应（仅在 debug 模式）
    if [[ "$DEBUG" == "true" ]]; then
        local api_key_masked="${API_KEY:0:8}********"
        display_raw_response "$method" "$url" "$api_key_masked" "$http_code" "$headers" "$body"
    fi

    echo "$body"
}

api_get_providers() {
    local url="${API_BASE_URL}/api/v1/user/available-providers"
    local method="GET"

    # 根据 debug 模式选择不同的 curl 参数
    local response
    if [[ "$DEBUG" == "true" ]]; then
        response=$(curl -s -i -w "\n%{http_code}" \
            -X GET \
            -H "accept: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    else
        response=$(curl -s -w "\n%{http_code}" \
            -X GET \
            -H "accept: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    fi

    if [[ $? -ne 0 ]]; then
        echo -e "${RED}错误: 请求失败${RESET}" >&2
        return 1
    fi

    # 分离响应体和状态码
    local http_code=$(echo "$response" | tail -n1)
    local full_response=$(echo "$response" | head -n -1)

    # 在 debug 模式下分离响应头和响应体
    local headers=""
    local body=""
    if [[ "$DEBUG" == "true" ]]; then
        full_response=$(echo "$full_response" | tr -d '\r')
        local empty_line=$(echo "$full_response" | grep -n "^$" | head -1 | cut -d: -f1)
        if [[ -n "$empty_line" ]]; then
            headers=$(echo "$full_response" | head -n $((empty_line - 1)))
            body=$(echo "$full_response" | tail -n +$((empty_line + 1)))
        else
            body="$full_response"
        fi
    else
        body="$full_response"
    fi

    # 处理认证错误
    if [[ "$http_code" == "401" ]]; then
        echo "AUTH_ERROR_401"
        return 2
    fi

    # 处理其他错误
    if [[ "$http_code" != "200" ]]; then
        echo -e "${RED}获取分组信息失败: HTTP $http_code${RESET}" >&2
        echo "响应内容: $body" >&2
        return 1
    fi

    # 显示原始响应（仅在 debug 模式）
    if [[ "$DEBUG" == "true" ]]; then
        local api_key_masked="${API_KEY:0:8}********"
        display_raw_response "$method" "$url" "$api_key_masked" "$http_code" "$headers" "$body"
    fi

    echo "$body"
}

api_get_provider_alternatives() {
    local provider_id=$1

    if [[ -z "$provider_id" ]]; then
        echo -e "${RED}错误: provider_id 不能为空${RESET}" >&2
        return 1
    fi

    local url="${API_BASE_URL}/api/v1/user/provider-alternatives/${provider_id}"
    local method="GET"

    # 根据 debug 模式选择不同的 curl 参数
    local response
    if [[ "$DEBUG" == "true" ]]; then
        response=$(curl -s -i -w "\n%{http_code}" \
            -X GET \
            -H "accept: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    else
        response=$(curl -s -w "\n%{http_code}" \
            -X GET \
            -H "accept: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    fi

    if [[ $? -ne 0 ]]; then
        echo -e "${RED}错误: 请求失败${RESET}" >&2
        return 1
    fi

    # 分离响应体和状态码
    local http_code=$(echo "$response" | tail -n1)
    local full_response=$(echo "$response" | head -n -1)

    # 在 debug 模式下分离响应头和响应体
    local headers=""
    local body=""
    if [[ "$DEBUG" == "true" ]]; then
        full_response=$(echo "$full_response" | tr -d '\r')
        local empty_line=$(echo "$full_response" | grep -n "^$" | head -1 | cut -d: -f1)
        if [[ -n "$empty_line" ]]; then
            headers=$(echo "$full_response" | head -n $((empty_line - 1)))
            body=$(echo "$full_response" | tail -n +$((empty_line + 1)))
        else
            body="$full_response"
        fi
    else
        body="$full_response"
    fi

    # 处理认证错误
    if [[ "$http_code" == "401" ]]; then
        echo "AUTH_ERROR_401"
        return 2
    fi

    # 处理其他错误
    if [[ "$http_code" != "200" ]]; then
        echo -e "${RED}获取分组失败: HTTP $http_code${RESET}" >&2
        echo "响应内容: $body" >&2
        return 1
    fi

    # 显示原始响应（仅在 debug 模式）
    if [[ "$DEBUG" == "true" ]]; then
        local api_key_masked="${API_KEY:0:8}********"
        display_raw_response "$method" "$url" "$api_key_masked" "$http_code" "$headers" "$body"
    fi

    echo "$body"
}

api_get_provider_selection() {
    local provider_id=$1

    if [[ -z "$provider_id" ]]; then
        echo -e "${RED}错误: provider_id 不能为空${RESET}" >&2
        return 1
    fi

    local url="${API_BASE_URL}/api/v1/user/provider-alternatives/${provider_id}/selection"
    local method="GET"

    # 根据 debug 模式选择不同的 curl 参数
    local response
    if [[ "$DEBUG" == "true" ]]; then
        response=$(curl -s -i -w "\n%{http_code}" \
            -X GET \
            -H "accept: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    else
        response=$(curl -s -w "\n%{http_code}" \
            -X GET \
            -H "accept: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    fi

    if [[ $? -ne 0 ]]; then
        echo -e "${RED}错误: 请求失败${RESET}" >&2
        return 1
    fi

    # 分离响应体和状态码
    local http_code=$(echo "$response" | tail -n1)
    local full_response=$(echo "$response" | head -n -1)

    # 在 debug 模式下分离响应头和响应体
    local headers=""
    local body=""
    if [[ "$DEBUG" == "true" ]]; then
        full_response=$(echo "$full_response" | tr -d '\r')
        local empty_line=$(echo "$full_response" | grep -n "^$" | head -1 | cut -d: -f1)
        if [[ -n "$empty_line" ]]; then
            headers=$(echo "$full_response" | head -n $((empty_line - 1)))
            body=$(echo "$full_response" | tail -n +$((empty_line + 1)))
        else
            body="$full_response"
        fi
    else
        body="$full_response"
    fi

    # 处理认证错误
    if [[ "$http_code" == "401" ]]; then
        echo "AUTH_ERROR_401"
        return 2
    fi

    # 处理其他错误
    if [[ "$http_code" != "200" ]]; then
        echo -e "${RED}获取当前选择失败: HTTP $http_code${RESET}" >&2
        echo "响应内容: $body" >&2
        return 1
    fi

    # 显示原始响应（仅在 debug 模式）
    if [[ "$DEBUG" == "true" ]]; then
        local api_key_masked="${API_KEY:0:8}********"
        display_raw_response "$method" "$url" "$api_key_masked" "$http_code" "$headers" "$body"
    fi

    echo "$body"
}

api_set_provider_selection() {
    local provider_id=$1
    local selected_alternative_id=$2

    if [[ -z "$provider_id" ]]; then
        echo -e "${RED}错误: provider_id 不能为空${RESET}" >&2
        return 1
    fi

    if [[ -z "$selected_alternative_id" ]]; then
        echo -e "${RED}错误: selected_alternative_id 不能为空${RESET}" >&2
        return 1
    fi

    local url="${API_BASE_URL}/api/v1/user/provider-alternatives/${provider_id}/selection"
    local method="PUT"
    local json_data="{\"selected_alternative_id\": ${selected_alternative_id}}"

    # 根据 debug 模式选择不同的 curl 参数
    local response
    if [[ "$DEBUG" == "true" ]]; then
        response=$(curl -s -i -w "\n%{http_code}" \
            -X PUT \
            -H "accept: application/json" \
            -H "Content-Type: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            -d "$json_data" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    else
        response=$(curl -s -w "\n%{http_code}" \
            -X PUT \
            -H "accept: application/json" \
            -H "Content-Type: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            -d "$json_data" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    fi

    if [[ $? -ne 0 ]]; then
        echo -e "${RED}错误: 请求失败${RESET}" >&2
        return 1
    fi

    # 分离响应体和状态码
    local http_code=$(echo "$response" | tail -n1)
    local full_response=$(echo "$response" | head -n -1)

    # 在 debug 模式下分离响应头和响应体
    local headers=""
    local body=""
    if [[ "$DEBUG" == "true" ]]; then
        full_response=$(echo "$full_response" | tr -d '\r')
        local empty_line=$(echo "$full_response" | grep -n "^$" | head -1 | cut -d: -f1)
        if [[ -n "$empty_line" ]]; then
            headers=$(echo "$full_response" | head -n $((empty_line - 1)))
            body=$(echo "$full_response" | tail -n +$((empty_line + 1)))
        else
            body="$full_response"
        fi
    else
        body="$full_response"
    fi

    # 处理认证错误
    if [[ "$http_code" == "401" ]]; then
        echo "AUTH_ERROR_401"
        return 2
    fi

    # 处理其他错误
    if [[ "$http_code" != "200" ]]; then
        echo -e "${RED}设置选择失败: HTTP $http_code${RESET}" >&2
        echo "响应内容: $body" >&2
        return 1
    fi

    # 显示原始响应（仅在 debug 模式）
    if [[ "$DEBUG" == "true" ]]; then
        local api_key_masked="${API_KEY:0:8}********"
        display_raw_response "$method" "$url" "$api_key_masked" "$http_code" "$headers" "$body" "$json_data"
    fi

    echo "$body"
}

api_set_balance_preference() {
    local balance_preference=$1

    if [[ -z "$balance_preference" ]]; then
        echo -e "${RED}错误: balance_preference 不能为空${RESET}" >&2
        return 1
    fi

    # 验证参数值
    if [[ "$balance_preference" != "subscription_first" && "$balance_preference" != "payg_only" ]]; then
        echo -e "${RED}错误: balance_preference 只能是 'subscription_first' 或 'payg_only'${RESET}" >&2
        return 1
    fi

    local url="${API_BASE_URL}/api/v1/user/balance-preference"
    local method="PUT"
    local json_data="{\"balance_preference\": \"${balance_preference}\"}"

    # 根据 debug 模式选择不同的 curl 参数
    local response
    if [[ "$DEBUG" == "true" ]]; then
        response=$(curl -s -i -w "\n%{http_code}" \
            -X PUT \
            -H "accept: application/json" \
            -H "Content-Type: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            -d "$json_data" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    else
        response=$(curl -s -w "\n%{http_code}" \
            -X PUT \
            -H "accept: application/json" \
            -H "Content-Type: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            -d "$json_data" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    fi

    if [[ $? -ne 0 ]]; then
        echo -e "${RED}错误: 请求失败${RESET}" >&2
        return 1
    fi

    # 分离响应体和状态码
    local http_code=$(echo "$response" | tail -n1)
    local full_response=$(echo "$response" | head -n -1)

    # 在 debug 模式下分离响应头和响应体
    local headers=""
    local body=""
    if [[ "$DEBUG" == "true" ]]; then
        full_response=$(echo "$full_response" | tr -d '\r')
        local empty_line=$(echo "$full_response" | grep -n "^$" | head -1 | cut -d: -f1)
        if [[ -n "$empty_line" ]]; then
            headers=$(echo "$full_response" | head -n $((empty_line - 1)))
            body=$(echo "$full_response" | tail -n +$((empty_line + 1)))
        else
            body="$full_response"
        fi
    else
        body="$full_response"
    fi

    # 处理认证错误
    if [[ "$http_code" == "401" ]]; then
        echo "AUTH_ERROR_401"
        return 2
    fi

    # 处理其他错误
    if [[ "$http_code" != "200" ]]; then
        echo -e "${RED}设置余额使用偏好失败: HTTP $http_code${RESET}" >&2
        echo "响应内容: $body" >&2
        return 1
    fi

    # 显示原始响应（仅在 debug 模式）
    if [[ "$DEBUG" == "true" ]]; then
        local api_key_masked="${API_KEY:0:8}********"
        display_raw_response "$method" "$url" "$api_key_masked" "$http_code" "$headers" "$body" "$json_data"
    fi

    echo "$body"
}

api_get_profile() {
    local url="${API_BASE_URL}/api/v1/auth/profile"
    local method="GET"

    # 根据 debug 模式选择不同的 curl 参数
    local response
    if [[ "$DEBUG" == "true" ]]; then
        response=$(curl -s -i -w "\n%{http_code}" \
            -X GET \
            -H "accept: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    else
        response=$(curl -s -w "\n%{http_code}" \
            -X GET \
            -H "accept: application/json" \
            -H "X-API-Key: ${API_KEY}" \
            --connect-timeout 10 \
            --max-time 30 \
            "$url" 2>&1)
    fi

    if [[ $? -ne 0 ]]; then
        echo -e "${RED}错误: 请求失败${RESET}" >&2
        return 1
    fi

    # 分离响应体和状态码
    local http_code=$(echo "$response" | tail -n1)
    local full_response=$(echo "$response" | head -n -1)

    # 在 debug 模式下分离响应头和响应体
    local headers=""
    local body=""
    if [[ "$DEBUG" == "true" ]]; then
        full_response=$(echo "$full_response" | tr -d '\r')
        local empty_line=$(echo "$full_response" | grep -n "^$" | head -1 | cut -d: -f1)
        if [[ -n "$empty_line" ]]; then
            headers=$(echo "$full_response" | head -n $((empty_line - 1)))
            body=$(echo "$full_response" | tail -n +$((empty_line + 1)))
        else
            body="$full_response"
        fi
    else
        body="$full_response"
    fi

    # 处理认证错误
    if [[ "$http_code" == "401" ]]; then
        echo "AUTH_ERROR_401"
        return 2
    fi

    # 处理其他错误
    if [[ "$http_code" != "200" ]]; then
        echo -e "${RED}获取用户资料失败: HTTP $http_code${RESET}" >&2
        echo "响应内容: $body" >&2
        return 1
    fi

    # 显示原始响应（仅在 debug 模式）
    if [[ "$DEBUG" == "true" ]]; then
        local api_key_masked="${API_KEY:0:8}********"
        display_raw_response "$method" "$url" "$api_key_masked" "$http_code" "$headers" "$body"
    fi

    echo "$body"
}

# ============================================================
# 显示余额信息
# ============================================================

display_balance_info() {
    local data=$1

    # 使用 jq 提取数据
    local subscription_balance=$(echo "$data" | jq -r '.subscription_balance // 0')
    local payg_balance=$(echo "$data" | jq -r '.pay_as_you_go_balance // 0')
    local total_balance=$(echo "$data" | jq -r '.total_balance // 0')
    local credit_balance=$(echo "$data" | jq -r '.credit_balance // 0')
    local weekly_limit=$(echo "$data" | jq -r '.weekly_limit // 0')
    local weekly_spent=$(echo "$data" | jq -r '.weekly_spent_balance // 0')

    # 计算剩余额度和消费百分比
    local weekly_remaining=$(echo "$weekly_limit - $weekly_spent" | bc)
    if (( $(echo "$weekly_limit > 0" | bc -l) )); then
        local spent_percent=$(echo "scale=1; ($weekly_spent / $weekly_limit) * 100" | bc)
        local remaining_percent=$(echo "scale=1; 100 - $spent_percent" | bc)
    else
        local spent_percent=0
        local remaining_percent=0
    fi

    # 显示余额信息
    echo -e "${CYAN}${BOLD}💰 余额信息${RESET}"

    if [[ "$subscription_balance" != "0" ]]; then
        local color=$(get_balance_color "$subscription_balance")
        echo -e "订阅余额：${color}\$$(printf "%.2f" $subscription_balance)${RESET}"
    fi

    if [[ "$payg_balance" != "0" ]]; then
        local color=$(get_balance_color "$payg_balance")
        echo -e "按需付费余额：${color}\$$(printf "%.2f" $payg_balance)${RESET}"
    fi

    if [[ "$total_balance" != "0" ]]; then
        local color=$(get_balance_color "$total_balance")
        echo -e "${WHITE}${BOLD}总余额：${RESET}${color}${BOLD}\$$(printf "%.2f" $total_balance)${RESET}"
    fi

    if (( $(echo "$credit_balance > 0" | bc -l) )); then
        echo -e "信用额度：${MAGENTA}\$$(printf "%.2f" $credit_balance)${RESET}"
    fi

    # 显示消费信息
    echo
    echo -e "${CYAN}${BOLD}📊 本周消费${RESET}"

    if (( $(echo "$weekly_limit > 0" | bc -l) )); then
        echo -e "周限额：${CYAN}\$$(printf "%.2f" $weekly_limit)${RESET}"

        local color=$(get_spending_color "$spent_percent")
        echo -e "已消费：${color}\$$(printf "%.2f" $weekly_spent)${RESET} ${color}(${spent_percent}%)${RESET}"

        local remaining_color=$(get_balance_color "$weekly_remaining")
        echo -e "剩余额度：${remaining_color}\$$(printf "%.2f" $weekly_remaining)${RESET} ${remaining_color}(${remaining_percent}%)${RESET}"

        echo -e "消费进度：$(create_progress_bar "$weekly_spent" "$weekly_limit")"
    fi
}

display_providers_info() {
    local data=$1

    # 提取账户信息
    local has_payg=$(echo "$data" | jq -r '.has_payg_balance')
    local has_subscription=$(echo "$data" | jq -r '.has_subscription')

    # 显示账户状态
    echo -e "${CYAN}${BOLD}🔌 可用分组${RESET}"
    echo

    # 使用 jq 解析 providers 数组
    local providers_count=$(echo "$data" | jq '.providers | length')

    for ((i=0; i<$providers_count; i++)); do
        local provider_data=$(echo "$data" | jq -r ".providers[$i]")

        local provider_id=$(echo "$provider_data" | jq -r '.provider.id')
        local display_name=$(echo "$provider_data" | jq -r '.provider.display_name')
        local type=$(echo "$provider_data" | jq -r '.provider.type')
        local rate=$(echo "$provider_data" | jq -r '.rate_multiplier')
        local is_default=$(echo "$provider_data" | jq -r '.is_default')
        local source=$(echo "$provider_data" | jq -r '.source')

        # 转换 source 为中文
        local source_text=""
        if [[ "$source" == "subscription" ]]; then
            source_text="${CYAN}订阅${RESET}"
        else
            source_text="${YELLOW}按需${RESET}"
        fi

        # 显示默认标记
        local default_mark=""
        if [[ "$is_default" == "true" ]]; then
            default_mark=" ${GREEN}[默认]${RESET}"
        fi

        # 显示分组信息
        echo -e "${WHITE}${BOLD}${display_name}${RESET}${default_mark}"
        echo -e "ID: ${BLUE}${provider_id}${RESET} | 类型: ${type} | 费率: ${MAGENTA}×${rate}${RESET} | 来源: ${source_text}"
        echo
    done
}

display_provider_alternatives() {
    local data=$1
    local provider_id=$2

    # 检查是否有数据
    local data_count=$(echo "$data" | jq '.data | length')
    if [[ "$data_count" == "0" || "$data_count" == "null" ]]; then
        echo -e "${RED}错误: 分组不存在${RESET}"
        echo
        echo -e "${YELLOW}提示: 请先运行 ${WHITE}yc providers${YELLOW} 查看可用的分组 ID${RESET}"
        return 1
    fi

    # 显示标题
    echo -e "${CYAN}${BOLD}分组 ${provider_id} 可用提供商：${RESET}"
    echo

    # 遍历所有方案（包括原方案和替代方案）
    for ((i=0; i<$data_count; i++)); do
        local item=$(echo "$data" | jq -r ".data[$i]")

        local is_self=$(echo "$item" | jq -r '.is_self')
        local provider_id=$(echo "$item" | jq -r '.alternative.id')
        local display_name=$(echo "$item" | jq -r '.alternative.display_name')
        local type=$(echo "$item" | jq -r '.alternative.type')
        local rate=$(echo "$item" | jq -r '.alternative.rate_multiplier')

        # 根据是否为原方案添加不同的标记和颜色
        if [[ "$is_self" == "true" ]]; then
            # 原方案：在名称后添加[默认]标签
            echo -e "  ID: ${BLUE}${provider_id}${RESET} | ${WHITE}${display_name}${RESET} ${GREEN}[默认]${RESET} | 费率: ${MAGENTA}×${rate}${RESET} | 类型: ${type}"
        else
            # 替代方案：普通显示
            echo -e "  ID: ${BLUE}${provider_id}${RESET} | ${WHITE}${display_name}${RESET} | 费率: ${MAGENTA}×${rate}${RESET} | 类型: ${type}"
        fi
    done
}

display_provider_selection() {
    local data=$1

    # 检查是否有数据
    local selection_data=$(echo "$data" | jq -r '.data')
    if [[ "$selection_data" == "null" ]]; then
        echo -e "${RED}错误: 分组不存在${RESET}"
        echo
        echo -e "${YELLOW}提示: 请先运行 ${WHITE}yc providers${YELLOW} 查看可用的分组 ID${RESET}"
        return 1
    fi

    local provider_id=$(echo "$selection_data" | jq -r '.provider_id')
    local selected_alt_id=$(echo "$selection_data" | jq -r '.selected_alternative_id')
    local selected_alt=$(echo "$selection_data" | jq -r '.selected_alternative')

    local display_name=$(echo "$selected_alt" | jq -r '.display_name')
    local type=$(echo "$selected_alt" | jq -r '.type')
    local rate=$(echo "$selected_alt" | jq -r '.rate_multiplier')
    local description=$(echo "$selected_alt" | jq -r '.description')

    # 优化输出格式
    echo -e "分组 ${BLUE}${provider_id}${RESET} 当前生效的提供商："
    echo -e "ID: ${BLUE}${selected_alt_id}${RESET} | ${GREEN}${BOLD}${display_name}${RESET} | 费率: ${MAGENTA}×${rate}${RESET} | 类型: ${type}"
}

display_profile() {
    local data=$1

    # 提取用户资料信息
    local email=$(echo "$data" | jq -r '.email // "N/A"')
    local username=$(echo "$data" | jq -r '.username // "N/A"')

    # 提取余额信息
    local balance=$(echo "$data" | jq -r '.balance // "0"')
    local subscription_balance=$(echo "$data" | jq -r '.subscription_balance // "0"')
    local payg_balance=$(echo "$data" | jq -r '.pay_as_you_go_balance // "0"')
    local balance_preference=$(echo "$data" | jq -r '.balance_preference // "N/A"')
    local subscription_expiry=$(echo "$data" | jq -r '.subscription_expiry // "N/A"')

    # 提取消费信息
    local current_week_spend=$(echo "$data" | jq -r '.current_week_spend // "0"')
    local current_month_spend=$(echo "$data" | jq -r '.current_month_spend // "0"')

    # 提取订阅套餐信息
    local plan_name=$(echo "$data" | jq -r '.subscription_plan.name // "N/A"')
    local plan_price=$(echo "$data" | jq -r '.subscription_plan.price // "0"')
    local plan_is_active=$(echo "$data" | jq -r '.subscription_plan.is_active // "false"')
    local plan_daily_balance=$(echo "$data" | jq -r '.subscription_plan.daily_balance // "0"')
    local plan_weekly_limit=$(echo "$data" | jq -r '.subscription_plan.weekly_limit // "0"')
    local plan_monthly_limit=$(echo "$data" | jq -r '.subscription_plan.monthly_spend_limit // "0"')

    # 显示用户资料
    echo -e "${CYAN}${BOLD}👤 用户资料${RESET}"
    if [[ "$username" != "N/A" ]]; then
        echo -e "用户名：${WHITE}${BOLD}${username}${RESET}"
    fi

    if [[ "$email" != "N/A" ]]; then
        echo -e "邮箱：${WHITE}${email}${RESET}"
    fi

    # 显示余额信息
    echo
    echo -e "${CYAN}${BOLD}💰 余额信息${RESET}"

    if [[ "$subscription_balance" != "0" && "$subscription_balance" != "N/A" ]]; then
        local color=$(get_balance_color "$subscription_balance")
        echo -e "订阅余额：${color}\$$(printf "%.2f" $subscription_balance)${RESET}"
    fi

    if [[ "$payg_balance" != "0" && "$payg_balance" != "N/A" ]]; then
        local color=$(get_balance_color "$payg_balance")
        echo -e "按需付费余额：${color}\$$(printf "%.2f" $payg_balance)${RESET}"
    fi

    if [[ "$balance" != "0" && "$balance" != "N/A" ]]; then
        local color=$(get_balance_color "$balance")
        echo -e "总余额：${color}\$$(printf "%.2f" $balance)${RESET}"
    fi

    # 显示余额使用偏好
    if [[ "$balance_preference" != "N/A" ]]; then
        local preference_text=""
        if [[ "$balance_preference" == "subscription_first" ]]; then
            preference_text="${CYAN}优先订阅${RESET}"
        elif [[ "$balance_preference" == "payg_only" ]]; then
            preference_text="${YELLOW}仅按需付费${RESET}"
        else
            preference_text="${WHITE}${balance_preference}${RESET}"
        fi
        echo -e "余额使用偏好：${preference_text}"
    fi

    # 显示订阅套餐信息
    if [[ "$plan_name" != "N/A" ]]; then
        echo
        echo -e "${CYAN}${BOLD}📦 订阅套餐${RESET}"

        echo -e "套餐名称：${WHITE}${BOLD}${plan_name}${RESET}"

        if [[ "$plan_price" != "0" && "$plan_price" != "N/A" ]]; then
            echo -e "套餐价格：${GREEN}\$$(printf "%.2f" $plan_price)${RESET}"
        fi

        if [[ "$plan_is_active" == "true" ]]; then
            echo -e "套餐状态：${GREEN}${BOLD}有效${RESET}"
        else
            echo -e "套餐状态：${RED}无效${RESET}"
        fi

        # 订阅到期时间（人类易读）
        if [[ "$subscription_expiry" != "N/A" && "$subscription_expiry" != "null" ]]; then
            # 尝试转换为人类易读格式
            local expiry_readable=""
            if command -v date &> /dev/null; then
                expiry_readable=$(date -d "$subscription_expiry" "+%Y年%m月%d日 %H:%M" 2>/dev/null || echo "$subscription_expiry")
            else
                expiry_readable="$subscription_expiry"
            fi
            echo -e "订阅到期时间：${YELLOW}${expiry_readable}${RESET}"
        fi

        if [[ "$plan_daily_balance" != "0" && "$plan_daily_balance" != "N/A" ]]; then
            echo -e "日消费限额：${CYAN}\$$(printf "%.2f" $plan_daily_balance)${RESET}"
        fi

        # 周消费限额（整合已消费和百分比）
        if [[ "$plan_weekly_limit" != "0" && "$plan_weekly_limit" != "N/A" ]]; then
            local output="周消费限额：${CYAN}\$$(printf "%.2f" $plan_weekly_limit)${RESET}"

            if [[ "$current_week_spend" != "0" && "$current_week_spend" != "N/A" ]]; then
                local week_percent=$(echo "scale=10; ($current_week_spend / $plan_weekly_limit) * 100" | bc)
                local week_percent_formatted=$(printf "%.2f" $week_percent)
                local week_color=$(get_spending_color "$week_percent")
                output="${output} ${MAGENTA}(已消费\$$(printf "%.2f" $current_week_spend), ${week_color}${week_percent_formatted}%${RESET}${MAGENTA})${RESET}"
            fi
            echo -e "$output"
        fi

        # 月消费限额（整合已消费和百分比）
        if [[ "$plan_monthly_limit" != "0" && "$plan_monthly_limit" != "N/A" ]]; then
            local output="月消费限额：${CYAN}\$$(printf "%.2f" $plan_monthly_limit)${RESET}"

            if [[ "$current_month_spend" != "0" && "$current_month_spend" != "N/A" ]]; then
                local month_percent=$(echo "scale=10; ($current_month_spend / $plan_monthly_limit) * 100" | bc)
                local month_percent_formatted=$(printf "%.2f" $month_percent)
                local month_color=$(get_spending_color "$month_percent")
                output="${output} ${MAGENTA}(已消费\$$(printf "%.2f" $current_month_spend), ${month_color}${month_percent_formatted}%${RESET}${MAGENTA})${RESET}"
            fi
            echo -e "$output"
        fi
    fi
}

# ============================================================
# 交互式选择函数
# ============================================================

interactive_select_provider() {
    local providers_data=$1

    echo -e "${CYAN}${BOLD}📋 选择要切换提供商的分组：${RESET}" >&2
    echo >&2

    # 解析并显示分组列表
    local providers_count=$(echo "$providers_data" | jq '.providers | length')

    if [[ "$providers_count" == "0" ]]; then
        echo -e "${RED}错误: 没有可用的分组${RESET}" >&2
        return 1
    fi

    # 创建序号数组和ID映射
    declare -a provider_ids
    declare -a provider_names

    for ((i=0; i<$providers_count; i++)); do
        local provider_data=$(echo "$providers_data" | jq -r ".providers[$i]")
        local provider_id=$(echo "$provider_data" | jq -r '.provider.id')
        local display_name=$(echo "$provider_data" | jq -r '.provider.display_name')
        local rate=$(echo "$provider_data" | jq -r '.rate_multiplier')
        local source=$(echo "$provider_data" | jq -r '.source')

        provider_ids[$i]=$provider_id
        provider_names[$i]=$display_name

        # 转换 source 为中文
        local source_text=""
        if [[ "$source" == "subscription" ]]; then
            source_text="${CYAN}订阅${RESET}"
        else
            source_text="${YELLOW}按需${RESET}"
        fi

        # 构建显示行
        local display_line="  ${WHITE}${BOLD}[$((i+1))]${RESET} ${display_name} | ID: ${BLUE}${provider_id}${RESET}"

        # 按需付费显示费率
        if [[ "$source" == "pay_as_you_go" ]]; then
            display_line="${display_line} | 费率: ${MAGENTA}×${rate}${RESET}"
        fi

        display_line="${display_line} | 来源: ${source_text}"

        echo -e "$display_line" >&2
    done

    echo >&2

    # 获取用户输入
    while true; do
        read -p "请输入分组序号 [1-$providers_count] (输入 q 退出): " choice

        # 支持退出
        if [[ "$choice" == "q" || "$choice" == "Q" ]]; then
            return 1
        fi

        # 验证输入是否为数字
        if ! [[ "$choice" =~ ^[0-9]+$ ]]; then
            echo -e "${RED}错误: 请输入有效的序号${RESET}" >&2
            continue
        fi

        # 验证范围
        if [[ "$choice" -lt 1 || "$choice" -gt "$providers_count" ]]; then
            echo -e "${RED}错误: 序号超出范围，请输入 1-$providers_count${RESET}" >&2
            continue
        fi

        # 返回选中的 provider_id 和名称，用 | 分隔
        local selected_index=$((choice - 1))
        echo "${provider_ids[$selected_index]}|${provider_names[$selected_index]}"
        return 0
    done
}

interactive_select_alternative() {
    local provider_id=$1
    local alternatives_data=$2
    local current_selection=$3
    local provider_name=$4

    echo >&2
    echo -e "${CYAN}${BOLD}📋 ${provider_name} 分组的可选提供商：${RESET}" >&2
    echo >&2

    # 解析方案数据
    local data_count=$(echo "$alternatives_data" | jq '.data | length')

    if [[ "$data_count" == "0" || "$data_count" == "null" ]]; then
        echo -e "${RED}错误: 没有可用的提供商${RESET}" >&2
        return 1
    fi

    # 提取当前选择的方案ID
    local current_alt_id=$(echo "$current_selection" | jq -r '.data.selected_alternative_id')

    # 创建方案数组
    declare -a alternative_ids
    declare -a alternative_names

    for ((i=0; i<$data_count; i++)); do
        local item=$(echo "$alternatives_data" | jq -r ".data[$i]")
        local alt_id=$(echo "$item" | jq -r '.alternative.id')
        local display_name=$(echo "$item" | jq -r '.alternative.display_name')
        local rate=$(echo "$item" | jq -r '.alternative.rate_multiplier')
        local is_self=$(echo "$item" | jq -r '.is_self')

        alternative_ids[$i]=$alt_id
        alternative_names[$i]=$display_name

        # 判断是否为当前方案或官方方案
        local prefix_mark=""
        local suffix_mark=""

        if [[ "$alt_id" == "$current_alt_id" ]]; then
            prefix_mark="${GREEN}${BOLD}[当前]${RESET} "
        fi

        if [[ "$is_self" == "true" ]]; then
            suffix_mark=" ${YELLOW}(官方)${RESET}"
        fi

        # 显示选项
        echo -e "  ${WHITE}${BOLD}[$((i+1))]${RESET} ${prefix_mark}${display_name}${suffix_mark} | 费率: ${MAGENTA}×${rate}${RESET}" >&2
    done

    echo >&2

    # 获取用户输入
    while true; do
        read -p "请输入要切换到的方案序号 [1-$data_count] (输入 q 退出): " choice

        # 支持退出
        if [[ "$choice" == "q" || "$choice" == "Q" ]]; then
            return 1
        fi

        # 验证输入是否为数字
        if ! [[ "$choice" =~ ^[0-9]+$ ]]; then
            echo -e "${RED}错误: 请输入有效的序号${RESET}" >&2
            continue
        fi

        # 验证范围
        if [[ "$choice" -lt 1 || "$choice" -gt "$data_count" ]]; then
            echo -e "${RED}错误: 序号超出范围，请输入 1-$data_count${RESET}" >&2
            continue
        fi

        # 返回选中的 alternative_id
        local selected_index=$((choice - 1))
        local selected_id="${alternative_ids[$selected_index]}"

        # 检查是否选择了当前方案
        if [[ "$selected_id" == "$current_alt_id" ]]; then
            echo -e "${YELLOW}⚠ 已经是当前方案，无需切换${RESET}" >&2
            return 1
        fi

        # 直接返回选中的方案 ID
        echo "$selected_id"
        return 0
    done
}

# ============================================================
# 命令处理
# ============================================================

cmd_balance() {
    local balance_data
    local exit_code
    local temp_file=$(mktemp)

    # 临时禁用 set -e，允许捕获非零退出码
    set +e

    # 在后台执行 API 调用（不重定向 stderr，让 verbose 输出直接显示）
    (api_get_balance > "$temp_file"; echo $? > "${temp_file}.exit") &
    local api_pid=$!

    # 显示 loading 动画
    show_loading $api_pid

    # 等待后台任务完成
    wait $api_pid

    # 读取结果
    balance_data=$(cat "$temp_file")
    exit_code=$(cat "${temp_file}.exit")

    # 清理临时文件
    rm -f "$temp_file" "${temp_file}.exit"

    set -e

    # 如果是认证错误（401），提示用户重新输入 API Key
    if [[ $exit_code -eq 2 && "$balance_data" == "AUTH_ERROR_401" ]]; then
        echo -e "${RED}✗ API Key 无效（认证失败）${RESET}"
        prompt_api_key false  # 不显示重复提示信息

        # 重新加载配置
        load_config

        # 重试请求（带loading）
        set +e
        temp_file=$(mktemp)
        (api_get_balance > "$temp_file"; echo $? > "${temp_file}.exit") &
        api_pid=$!
        show_loading $api_pid
        wait $api_pid
        balance_data=$(cat "$temp_file")
        exit_code=$(cat "${temp_file}.exit")
        rm -f "$temp_file" "${temp_file}.exit"
        set -e
    fi

    # 检查最终结果
    if [[ $exit_code -eq 0 && -n "$balance_data" && "$balance_data" != "AUTH_ERROR_401" ]]; then
        echo
        display_balance_info "$balance_data"
        echo
    else
        echo -e "${RED}✗ 获取余额失败${RESET}"
        exit 1
    fi
}

cmd_profile() {
    local profile_data
    local exit_code
    local temp_file=$(mktemp)

    # 临时禁用 set -e，允许捕获非零退出码
    set +e

    # 在后台执行 API 调用
    (api_get_profile > "$temp_file"; echo $? > "${temp_file}.exit") &
    local api_pid=$!

    # 显示 loading 动画
    show_loading $api_pid

    # 等待后台任务完成
    wait $api_pid

    # 读取结果
    profile_data=$(cat "$temp_file")
    exit_code=$(cat "${temp_file}.exit")

    # 清理临时文件
    rm -f "$temp_file" "${temp_file}.exit"

    set -e

    # 如果是认证错误（401），提示用户重新输入 API Key
    if [[ $exit_code -eq 2 && "$profile_data" == "AUTH_ERROR_401" ]]; then
        echo -e "${RED}✗ API Key 无效（认证失败）${RESET}"
        prompt_api_key false

        # 重新加载配置
        load_config

        # 重试请求（带loading）
        set +e
        temp_file=$(mktemp)
        (api_get_profile > "$temp_file"; echo $? > "${temp_file}.exit") &
        api_pid=$!
        show_loading $api_pid
        wait $api_pid
        profile_data=$(cat "$temp_file")
        exit_code=$(cat "${temp_file}.exit")
        rm -f "$temp_file" "${temp_file}.exit"
        set -e
    fi

    # 检查最终结果
    if [[ $exit_code -eq 0 && -n "$profile_data" && "$profile_data" != "AUTH_ERROR_401" ]]; then
        echo
        display_profile "$profile_data"
        echo
    else
        echo -e "${RED}✗ 获取用户资料失败${RESET}"
        exit 1
    fi
}

cmd_providers() {
    local providers_data
    local exit_code
    local temp_file=$(mktemp)

    # 临时禁用 set -e，允许捕获非零退出码
    set +e

    # 在后台执行 API 调用
    (api_get_providers > "$temp_file"; echo $? > "${temp_file}.exit") &
    local api_pid=$!

    # 显示 loading 动画
    show_loading $api_pid

    # 等待后台任务完成
    wait $api_pid

    # 读取结果
    providers_data=$(cat "$temp_file")
    exit_code=$(cat "${temp_file}.exit")

    # 清理临时文件
    rm -f "$temp_file" "${temp_file}.exit"

    set -e

    # 如果是认证错误（401），提示用户重新输入 API Key
    if [[ $exit_code -eq 2 && "$providers_data" == "AUTH_ERROR_401" ]]; then
        echo -e "${RED}✗ API Key 无效（401 认证失败）${RESET}"
        prompt_api_key false  # 不显示重复提示信息

        # 重新加载配置
        load_config

        # 重试请求（带loading）
        set +e
        temp_file=$(mktemp)
        (api_get_providers > "$temp_file"; echo $? > "${temp_file}.exit") &
        api_pid=$!
        show_loading $api_pid
        wait $api_pid
        providers_data=$(cat "$temp_file")
        exit_code=$(cat "${temp_file}.exit")
        rm -f "$temp_file" "${temp_file}.exit"
        set -e
    fi

    # 检查最终结果
    if [[ $exit_code -eq 0 && -n "$providers_data" && "$providers_data" != "AUTH_ERROR_401" ]]; then
        echo
        display_providers_info "$providers_data"
        echo
    else
        echo -e "${RED}✗ 获取分组信息失败${RESET}"
        exit 1
    fi
}

cmd_provider_alternatives() {
    local provider_id=$1

    if [[ -z "$provider_id" ]]; then
        echo -e "${RED}错误: 请提供 provider_id${RESET}"
        echo "用法: yc provider-alternatives <provider_id>"
        exit 1
    fi

    local alternatives_data
    local exit_code
    local temp_file=$(mktemp)

    # 临时禁用 set -e，允许捕获非零退出码
    set +e

    # 在后台执行 API 调用
    (api_get_provider_alternatives "$provider_id" > "$temp_file"; echo $? > "${temp_file}.exit") &
    local api_pid=$!

    # 显示 loading 动画
    show_loading $api_pid

    # 等待后台任务完成
    wait $api_pid

    # 读取结果
    alternatives_data=$(cat "$temp_file")
    exit_code=$(cat "${temp_file}.exit")

    # 清理临时文件
    rm -f "$temp_file" "${temp_file}.exit"

    set -e

    # 如果是认证错误（401），提示用户重新输入 API Key
    if [[ $exit_code -eq 2 && "$alternatives_data" == "AUTH_ERROR_401" ]]; then
        echo -e "${RED}✗ API Key 无效（401 认证失败）${RESET}"
        prompt_api_key false  # 不显示重复提示信息

        # 重新加载配置
        load_config

        # 重试请求（带loading）
        set +e
        temp_file=$(mktemp)
        (api_get_provider_alternatives "$provider_id" > "$temp_file"; echo $? > "${temp_file}.exit") &
        api_pid=$!
        show_loading $api_pid
        wait $api_pid
        alternatives_data=$(cat "$temp_file")
        exit_code=$(cat "${temp_file}.exit")
        rm -f "$temp_file" "${temp_file}.exit"
        set -e
    fi

    # 检查最终结果
    if [[ $exit_code -eq 0 && -n "$alternatives_data" && "$alternatives_data" != "AUTH_ERROR_401" ]]; then
        echo
        display_provider_alternatives "$alternatives_data" "$provider_id"
        echo
    else
        echo -e "${RED}✗ 获取可用提供商失败${RESET}"
        exit 1
    fi
}

cmd_provider_selection() {
    local provider_id=$1

    if [[ -z "$provider_id" ]]; then
        echo -e "${RED}错误: 请提供 provider_id${RESET}"
        echo "用法: yc provider-selection <provider_id>"
        exit 1
    fi

    local selection_data
    local exit_code
    local temp_file=$(mktemp)

    # 临时禁用 set -e，允许捕获非零退出码
    set +e

    # 在后台执行 API 调用
    (api_get_provider_selection "$provider_id" > "$temp_file"; echo $? > "${temp_file}.exit") &
    local api_pid=$!

    # 显示 loading 动画
    show_loading $api_pid

    # 等待后台任务完成
    wait $api_pid

    # 读取结果
    selection_data=$(cat "$temp_file")
    exit_code=$(cat "${temp_file}.exit")

    # 清理临时文件
    rm -f "$temp_file" "${temp_file}.exit"

    set -e

    # 如果是认证错误（401），提示用户重新输入 API Key
    if [[ $exit_code -eq 2 && "$selection_data" == "AUTH_ERROR_401" ]]; then
        echo -e "${RED}✗ API Key 无效（401 认证失败）${RESET}"
        prompt_api_key false  # 不显示重复提示信息

        # 重新加载配置
        load_config

        # 重试请求（带loading）
        set +e
        temp_file=$(mktemp)
        (api_get_provider_selection "$provider_id" > "$temp_file"; echo $? > "${temp_file}.exit") &
        api_pid=$!
        show_loading $api_pid
        wait $api_pid
        selection_data=$(cat "$temp_file")
        exit_code=$(cat "${temp_file}.exit")
        rm -f "$temp_file" "${temp_file}.exit"
        set -e
    fi

    # 检查最终结果
    if [[ $exit_code -eq 0 && -n "$selection_data" && "$selection_data" != "AUTH_ERROR_401" ]]; then
        echo
        display_provider_selection "$selection_data"
        echo
    else
        echo -e "${RED}✗ 获取当前生效的提供商失败${RESET}"
        exit 1
    fi
}

cmd_set_provider_selection() {
    local provider_id=$1
    local selected_alternative_id=$2

    if [[ -z "$provider_id" ]]; then
        echo -e "${RED}错误: 请提供 provider_id${RESET}"
        echo "用法: yc set-provider-selection <provider_id> <alternative_id>"
        exit 1
    fi

    if [[ -z "$selected_alternative_id" ]]; then
        echo -e "${RED}错误: 请提供 alternative_id${RESET}"
        echo "用法: yc set-provider-selection <provider_id> <alternative_id>"
        exit 1
    fi

    local result_data
    local exit_code
    local temp_file=$(mktemp)

    # 临时禁用 set -e，允许捕获非零退出码
    set +e

    # 在后台执行 API 调用
    (api_set_provider_selection "$provider_id" "$selected_alternative_id" > "$temp_file"; echo $? > "${temp_file}.exit") &
    local api_pid=$!

    # 显示 loading 动画
    show_loading $api_pid

    # 等待后台任务完成
    wait $api_pid

    # 读取结果
    result_data=$(cat "$temp_file")
    exit_code=$(cat "${temp_file}.exit")

    # 清理临时文件
    rm -f "$temp_file" "${temp_file}.exit"

    set -e

    # 如果是认证错误（401），提示用户重新输入 API Key
    if [[ $exit_code -eq 2 && "$result_data" == "AUTH_ERROR_401" ]]; then
        echo -e "${RED}✗ API Key 无效（401 认证失败）${RESET}"
        prompt_api_key false  # 不显示重复提示信息

        # 重新加载配置
        load_config

        # 重试请求（带loading）
        set +e
        temp_file=$(mktemp)
        (api_set_provider_selection "$provider_id" "$selected_alternative_id" > "$temp_file"; echo $? > "${temp_file}.exit") &
        api_pid=$!
        show_loading $api_pid
        wait $api_pid
        result_data=$(cat "$temp_file")
        exit_code=$(cat "${temp_file}.exit")
        rm -f "$temp_file" "${temp_file}.exit"
        set -e
    fi

    # 检查最终结果
    if [[ $exit_code -eq 0 && -n "$result_data" && "$result_data" != "AUTH_ERROR_401" ]]; then
        echo

        # 提取并显示选择信息
        local selected_alt=$(echo "$result_data" | jq -r '.data.selected_alternative')
        local display_name=$(echo "$selected_alt" | jq -r '.display_name')
        local alt_id=$(echo "$selected_alt" | jq -r '.id')
        local type=$(echo "$selected_alt" | jq -r '.type')
        local rate=$(echo "$selected_alt" | jq -r '.rate_multiplier')

        echo -e "分组 ${BLUE}${provider_id}${RESET} 提供商已切换为：${WHITE}${BOLD}${display_name}${RESET}"
        echo -e "ID: ${BLUE}${alt_id}${RESET} | 类型: ${type} | 费率: ${MAGENTA}×${rate}${RESET}"
        echo
    else
        echo -e "${RED}✗ 提供商设置失败${RESET}"
        exit 1
    fi
}

cmd_switch() {
    local provider_id=$1
    local provider_name=""
    local providers_data
    local alternatives_data
    local current_selection
    local exit_code
    local temp_file

    # 如果没有提供 provider_id，则进入交互式选择
    if [[ -z "$provider_id" ]]; then
        # 获取所有分组
        temp_file=$(mktemp)
        set +e
        (api_get_providers > "$temp_file"; echo $? > "${temp_file}.exit") &
        local api_pid=$!
        show_loading $api_pid
        wait $api_pid
        providers_data=$(cat "$temp_file")
        exit_code=$(cat "${temp_file}.exit")
        rm -f "$temp_file" "${temp_file}.exit"
        set -e

        # 处理认证错误
        if [[ $exit_code -eq 2 && "$providers_data" == "AUTH_ERROR_401" ]]; then
            echo -e "${RED}✗ API Key 无效（401 认证失败）${RESET}"
            prompt_api_key false
            load_config

            # 重试
            temp_file=$(mktemp)
            set +e
            (api_get_providers > "$temp_file"; echo $? > "${temp_file}.exit") &
            api_pid=$!
            show_loading $api_pid
            wait $api_pid
            providers_data=$(cat "$temp_file")
            exit_code=$(cat "${temp_file}.exit")
            rm -f "$temp_file" "${temp_file}.exit"
            set -e
        fi

        if [[ $exit_code -ne 0 || -z "$providers_data" || "$providers_data" == "AUTH_ERROR_401" ]]; then
            echo -e "${RED}✗ 获取分组列表失败${RESET}"
            exit 1
        fi

        echo
        # 交互式选择分组（返回格式: id|name）
        local selection_result
        selection_result=$(interactive_select_provider "$providers_data")
        if [[ $? -ne 0 || -z "$selection_result" ]]; then
            echo
            echo -e "${YELLOW}已取消操作${RESET}"
            exit 0
        fi

        # 拆分 id 和 name
        provider_id=$(echo "$selection_result" | cut -d'|' -f1)
        provider_name=$(echo "$selection_result" | cut -d'|' -f2)
    else
        # 如果直接指定了 provider_id，需要查询分组名称
        temp_file=$(mktemp)
        set +e
        (api_get_providers > "$temp_file"; echo $? > "${temp_file}.exit") &
        local api_pid=$!
        show_loading $api_pid
        wait $api_pid
        providers_data=$(cat "$temp_file")
        exit_code=$(cat "${temp_file}.exit")
        rm -f "$temp_file" "${temp_file}.exit"
        set -e

        if [[ $exit_code -eq 0 && -n "$providers_data" ]]; then
            # 从 providers_data 中查找分组名称
            provider_name=$(echo "$providers_data" | jq -r ".providers[] | select(.provider.id == $provider_id) | .provider.display_name")
            if [[ -z "$provider_name" || "$provider_name" == "null" ]]; then
                provider_name="分组 ${provider_id}"
            fi
        else
            provider_name="分组 ${provider_id}"
        fi
    fi

    # 并行获取可用方案和当前选择
    local temp_alternatives=$(mktemp)
    local temp_selection=$(mktemp)

    set +e
    (api_get_provider_alternatives "$provider_id" > "$temp_alternatives"; echo $? > "${temp_alternatives}.exit") &
    local pid_alternatives=$!
    (api_get_provider_selection "$provider_id" > "$temp_selection"; echo $? > "${temp_selection}.exit") &
    local pid_selection=$!

    # 显示 loading 动画
    show_loading $pid_alternatives

    # 等待两个 API 调用完成
    wait $pid_alternatives
    wait $pid_selection

    alternatives_data=$(cat "$temp_alternatives")
    local exit_code_alt=$(cat "${temp_alternatives}.exit")
    current_selection=$(cat "$temp_selection")
    local exit_code_sel=$(cat "${temp_selection}.exit")

    rm -f "$temp_alternatives" "${temp_alternatives}.exit" "$temp_selection" "${temp_selection}.exit"
    set -e

    # 检查结果
    if [[ $exit_code_alt -ne 0 || $exit_code_sel -ne 0 ]]; then
        echo -e "${RED}✗ 获取方案信息失败${RESET}"
        echo -e "${YELLOW}提示: 请检查 provider_id 是否正确${RESET}"
        exit 1
    fi

    # 交互式选择方案
    local selected_alt_id
    selected_alt_id=$(interactive_select_alternative "$provider_id" "$alternatives_data" "$current_selection" "$provider_name")
    if [[ $? -ne 0 || -z "$selected_alt_id" ]]; then
        echo
        exit 0
    fi

    # 执行切换
    echo
    local result_data
    temp_file=$(mktemp)
    set +e
    (api_set_provider_selection "$provider_id" "$selected_alt_id" > "$temp_file"; echo $? > "${temp_file}.exit") &
    local api_pid=$!
    show_loading $api_pid
    wait $api_pid
    result_data=$(cat "$temp_file")
    exit_code=$(cat "${temp_file}.exit")
    rm -f "$temp_file" "${temp_file}.exit"
    set -e

    # 检查切换结果
    if [[ $exit_code -eq 0 && -n "$result_data" && "$result_data" != "AUTH_ERROR_401" ]]; then
        echo
        local selected_alt=$(echo "$result_data" | jq -r '.data.selected_alternative')
        local display_name=$(echo "$selected_alt" | jq -r '.display_name')
        local rate=$(echo "$selected_alt" | jq -r '.rate_multiplier')

        echo -e "${GREEN}${BOLD}✓ 切换成功！${RESET}"
        echo
        echo -e "${provider_name} 的提供商已切换为："
        echo -e "${WHITE}${BOLD}${display_name}${RESET} | 费率: ${MAGENTA}×${rate}${RESET}"
        echo
    else
        echo -e "${RED}✗ 切换失败${RESET}"
        exit 1
    fi
}

cmd_balance_preference() {
    local preference=$1

    # 如果没有提供参数，则进入交互式选择
    if [[ -z "$preference" ]]; then
        # 获取当前余额偏好设置
        local profile_data
        local exit_code
        local temp_file=$(mktemp)

        set +e
        (api_get_profile > "$temp_file"; echo $? > "${temp_file}.exit") &
        local api_pid=$!
        show_loading $api_pid
        wait $api_pid
        profile_data=$(cat "$temp_file")
        exit_code=$(cat "${temp_file}.exit")
        rm -f "$temp_file" "${temp_file}.exit"
        set -e

        # 处理认证错误
        if [[ $exit_code -eq 2 && "$profile_data" == "AUTH_ERROR_401" ]]; then
            echo -e "${RED}✗ API Key 无效（认证失败）${RESET}"
            prompt_api_key false
            load_config

            # 重试
            temp_file=$(mktemp)
            set +e
            (api_get_profile > "$temp_file"; echo $? > "${temp_file}.exit") &
            api_pid=$!
            show_loading $api_pid
            wait $api_pid
            profile_data=$(cat "$temp_file")
            exit_code=$(cat "${temp_file}.exit")
            rm -f "$temp_file" "${temp_file}.exit"
            set -e
        fi

        # 提取当前余额偏好
        local current_preference="N/A"
        if [[ $exit_code -eq 0 && -n "$profile_data" && "$profile_data" != "AUTH_ERROR_401" ]]; then
            current_preference=$(echo "$profile_data" | jq -r '.balance_preference // "N/A"')
        fi

        echo
        echo -e "${CYAN}${BOLD}💳 选择余额使用偏好：${RESET}"
        echo

        # 显示选项1（优先订阅）
        local option1_prefix=""
        if [[ "$current_preference" == "subscription_first" ]]; then
            option1_prefix="${GREEN}${BOLD}[当前]${RESET} "
        fi
        echo -e "  ${WHITE}${BOLD}[1]${RESET} ${option1_prefix}优先订阅 (subscription_first)"
        echo -e "      ${WHITE}先使用订阅余额，然后使用按需付费${RESET}"
        echo -e "      ${YELLOW}⚠ OPUS使用限制适用${RESET}"
        echo

        # 显示选项2（仅按需付费）
        local option2_prefix=""
        if [[ "$current_preference" == "payg_only" ]]; then
            option2_prefix="${GREEN}${BOLD}[当前]${RESET} "
        fi
        echo -e "  ${WHITE}${BOLD}[2]${RESET} ${option2_prefix}仅按需付费 (payg_only)"
        echo -e "      ${WHITE}始终使用按需付费余额${RESET}"
        echo -e "      ${GREEN}✓ 无OPUS使用限制${RESET}"
        echo

        # 获取用户输入
        while true; do
            read -p "请输入选项 [1-2] (输入 q 退出): " choice

            # 支持退出
            if [[ "$choice" == "q" || "$choice" == "Q" ]]; then
                echo -e "${YELLOW}已取消操作${RESET}"
                exit 0
            fi

            # 验证输入
            case "$choice" in
                1)
                    preference="subscription_first"
                    break
                    ;;
                2)
                    preference="payg_only"
                    break
                    ;;
                *)
                    echo -e "${RED}错误: 请输入 1 或 2${RESET}"
                    continue
                    ;;
            esac
        done

        # 检查是否选择了当前选项
        if [[ "$preference" == "$current_preference" ]]; then
            echo -e "${YELLOW}⚠ 已经是当前设置，无需切换${RESET}"
            exit 0
        fi
    else
        # 验证直接提供的参数
        if [[ "$preference" != "subscription_first" && "$preference" != "payg_only" ]]; then
            echo -e "${RED}错误: preference 必须是 'subscription_first' 或 'payg_only'${RESET}"
            echo "用法: yc balance-preference [subscription_first|payg_only]"
            exit 1
        fi
    fi

    # 执行设置
    local result_data
    local exit_code
    local temp_file=$(mktemp)

    set +e
    (api_set_balance_preference "$preference" > "$temp_file"; echo $? > "${temp_file}.exit") &
    local api_pid=$!
    show_loading $api_pid
    wait $api_pid
    result_data=$(cat "$temp_file")
    exit_code=$(cat "${temp_file}.exit")
    rm -f "$temp_file" "${temp_file}.exit"
    set -e

    # 处理认证错误
    if [[ $exit_code -eq 2 && "$result_data" == "AUTH_ERROR_401" ]]; then
        echo -e "${RED}✗ API Key 无效（401 认证失败）${RESET}"
        prompt_api_key false
        load_config

        # 重试
        temp_file=$(mktemp)
        set +e
        (api_set_balance_preference "$preference" > "$temp_file"; echo $? > "${temp_file}.exit") &
        api_pid=$!
        show_loading $api_pid
        wait $api_pid
        result_data=$(cat "$temp_file")
        exit_code=$(cat "${temp_file}.exit")
        rm -f "$temp_file" "${temp_file}.exit"
        set -e
    fi

    # 检查设置结果
    if [[ $exit_code -eq 0 && -n "$result_data" && "$result_data" != "AUTH_ERROR_401" ]]; then
        echo
        echo -e "${GREEN}${BOLD}✓ 余额使用偏好设置成功！${RESET}"
        echo

        # 显示设置的偏好
        local preference_text=""
        local preference_desc=""
        if [[ "$preference" == "subscription_first" ]]; then
            preference_text="${CYAN}优先订阅${RESET}"
            preference_desc="先使用订阅余额，然后使用按需付费 ${YELLOW}(OPUS使用限制适用)${RESET}"
        else
            preference_text="${YELLOW}仅按需付费${RESET}"
            preference_desc="始终使用按需付费余额 ${GREEN}(无OPUS使用限制)${RESET}"
        fi

        echo -e "当前余额使用偏好：${WHITE}${BOLD}${preference_text}${RESET}"
        echo -e "${preference_desc}"
        echo
    else
        echo -e "${RED}✗ 设置余额使用偏好失败${RESET}"
        exit 1
    fi
}

# ============================================================
# 交互式菜单
# ============================================================

cmd_interactive_menu() {
    while true; do
        echo -e "${CYAN}${BOLD}📋 YesCode CLI 主菜单${RESET}"
        echo
        echo -e "  ${WHITE}${BOLD}[1]${RESET} 查询用户资料"
        echo -e "  ${WHITE}${BOLD}[2]${RESET} 查询余额信息"
        echo -e "  ${WHITE}${BOLD}[3]${RESET} 交互式切换提供商"
        echo -e "  ${WHITE}${BOLD}[4]${RESET} 设置余额使用偏好"
        echo -e "  ${WHITE}${BOLD}[5]${RESET} 查询提供商分组"
        echo -e "  ${WHITE}${BOLD}[6]${RESET} 安装到系统"
        echo -e "  ${WHITE}${BOLD}[7]${RESET} 从系统卸载"
        echo -e "  ${WHITE}${BOLD}[8]${RESET} 显示完整帮助"
        echo -e "  ${WHITE}${BOLD}[q]${RESET} 退出"
        echo

        read -p "请输入选项 [1-8/q]: " choice
        echo

        case "$choice" in
            1)
                cmd_profile
                ;;
            2)
                cmd_balance
                ;;
            3)
                cmd_switch ""
                ;;
            4)
                cmd_balance_preference ""
                ;;
            5)
                cmd_providers
                ;;
            6)
                cmd_install
                # install 命令会直接退出，不会返回
                exit 0
                ;;
            7)
                cmd_uninstall
                # uninstall 命令会直接退出，不会返回
                exit 0
                ;;
            8)
                show_help
                ;;
            q|Q)
                echo -e "${GREEN}再见！${RESET}"
                exit 0
                ;;
            *)
                echo -e "${RED}✗ 无效选项，请输入 1-8 或 q${RESET}"
                continue
                ;;
        esac

        # 命令执行完毕后，等待用户按 Esc 返回菜单
        echo
        echo -e "${CYAN}按 Esc 返回主菜单...${RESET}"
        while true; do
            read -rsn1 key
            if [[ $key == $'\e' ]]; then
                break
            fi
        done

        # 添加分隔线，保留之前的输出
        echo
        echo -e "${GRAY}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${RESET}"
        echo
    done
}

cmd_uninstall() {
    # 检测是否从项目目录运行（./yc 或相对路径）
    if [[ "$0" == ./* || "$0" == */* ]] && command -v yc &> /dev/null; then
        # 从项目目录运行，但系统中有安装的 yc 命令
        local installed_yc=$(which yc 2>/dev/null)
        local current_script=$(realpath "$0" 2>/dev/null || readlink -f "$0" 2>/dev/null || echo "$0")

        # 如果两者不同，说明用户在项目目录运行，但系统有安装的版本
        if [[ "$installed_yc" != "$current_script" ]]; then
            echo -e "${YELLOW}检测到您正在项目目录中运行此脚本${RESET}"
            echo -e "要卸载系统安装的 yc 命令，请直接运行："
            echo -e "  ${WHITE}${BOLD}yc uninstall${RESET}"
            echo
            echo -e "系统安装路径: ${CYAN}$installed_yc${RESET}"
            echo -e "当前脚本路径: ${CYAN}$current_script${RESET}"
            exit 1
        fi
    fi

    # 获取 yc 命令的实际路径
    local yc_path=""

    # 优先查找系统安装的 yc 命令
    if command -v yc &> /dev/null; then
        yc_path=$(which yc)
    fi

    # 如果没找到，使用当前脚本路径
    if [[ -z "$yc_path" ]]; then
        yc_path=$(realpath "$0" 2>/dev/null || readlink -f "$0" 2>/dev/null || echo "$0")
    fi

    # 检查是否有权限删除
    if [[ ! -w "$yc_path" && ! -w "$(dirname "$yc_path")" ]]; then
        echo -e "${RED}✗ 没有权限删除 $yc_path${RESET}"
        echo "请使用 sudo 运行或手动删除"
        exit 1
    fi

    # 询问是否删除配置文件
    local delete_config=false
    if [[ -f "$HOME/.yescode/config.json" ]]; then
        read -p "是否同时删除配置文件? (y/n): " -r
        if [[ $REPLY =~ ^[Yy]$ ]]; then
            delete_config=true
        fi
    fi

    # 删除 yc 命令
    if rm -f "$yc_path" 2>/dev/null; then
        echo -e "${GREEN}✓ 已卸载 yc${RESET}"
    else
        echo -e "${RED}✗ 卸载失败${RESET}"
        exit 1
    fi

    # 删除配置文件（如果用户选择）
    if [[ "$delete_config" == true ]]; then
        rm -rf "$HOME/.yescode" 2>/dev/null
        echo -e "${GREEN}✓ 已删除配置文件${RESET}"
    fi
}

cmd_install() {
    # 检查依赖
    local missing_deps=()

    if ! command -v curl &> /dev/null; then
        missing_deps+=("curl")
    fi

    if ! command -v jq &> /dev/null; then
        missing_deps+=("jq")
    fi

    # 如果有缺失的依赖，尝试自动安装
    if [ ${#missing_deps[@]} -gt 0 ]; then
        echo -e "${YELLOW}⚠ 缺少依赖: ${missing_deps[*]}${RESET}"

        local pkg_manager=$(detect_package_manager)

        if [[ "$pkg_manager" != "unknown" ]]; then
            # 尝试自动安装每个缺失的依赖
            for dep in "${missing_deps[@]}"; do
                if auto_install_dependency "$dep"; then
                    # 安装成功，从缺失列表中移除
                    missing_deps=("${missing_deps[@]/$dep}")
                fi
            done

            # 过滤空元素
            missing_deps=($(printf '%s\n' "${missing_deps[@]}" | grep -v '^$'))
        fi

        # 重新检查是否还有缺失
        if [ ${#missing_deps[@]} -gt 0 ]; then
            echo
            echo -e "${YELLOW}⚠ 依赖仍然缺失: ${missing_deps[*]}${RESET}"
            echo "请手动安装："
            echo "  Ubuntu/Debian: sudo apt install ${missing_deps[*]}"
            echo "  RHEL/CentOS: sudo yum install ${missing_deps[*]}"
            echo "  Fedora: sudo dnf install ${missing_deps[*]}"
            echo "  macOS: brew install ${missing_deps[*]}"
            echo "  Arch Linux: sudo pacman -S ${missing_deps[*]}"
            echo
            read -p "是否继续安装? (y/n): " -r
            if [[ ! $REPLY =~ ^[Yy]$ ]]; then
                echo "已取消"
                exit 1
            fi
        fi
    fi

    # 获取当前脚本路径
    local script_path=$(realpath "$0" 2>/dev/null || readlink -f "$0" 2>/dev/null || echo "$0")

    # 确定安装目标目录
    local install_dir=""
    if [ -w "/usr/local/bin" ]; then
        install_dir="/usr/local/bin"
    elif [ -d "$HOME/.local/bin" ]; then
        install_dir="$HOME/.local/bin"
    else
        install_dir="$HOME/.local/bin"
        mkdir -p "$install_dir"
    fi

    # 检查是否已安装
    local install_path="$install_dir/yc"
    if [ -f "$install_path" ]; then
        if [ "$script_path" = "$install_path" ]; then
            echo -e "${CYAN}ℹ yc 已安装${RESET}"
            return 0
        fi
    fi

    # 安装脚本文件
    if cp "$script_path" "$install_path" 2>/dev/null; then
        chmod +x "$install_path"
        echo -e "${GREEN}✓ 已安装到 $install_path${RESET}"
    else
        echo -e "${RED}✗ 安装失败，没有权限写入 $install_dir${RESET}"
        echo "请使用 sudo 运行"
        exit 1
    fi

    # 配置 API Key
    local config_dir="$HOME/.yescode"
    local config_file="$config_dir/config.json"

    if [ ! -f "$config_file" ]; then
        mkdir -p "$config_dir"
        read -p "配置 API Key: " api_key

        # 验证输入
        if [[ -z "$api_key" ]]; then
            echo -e "${RED}错误: API Key 不能为空${RESET}"
            exit 1
        fi

        # 保存配置
        echo "{\"api_key\": \"$api_key\"}" > "$config_file"
        chmod 600 "$config_file"

        echo -e "${GREEN}✓ 配置已保存${RESET}"
    fi

    # 检查 PATH
    if [[ ":$PATH:" != *":$install_dir:"* ]]; then
        echo
        echo -e "${YELLOW}⚠ $install_dir 不在 PATH 中${RESET}"
        echo "请添加到 ~/.bashrc 或 ~/.zshrc:"
        echo "  export PATH=\"\$HOME/.local/bin:\$PATH\""
    fi

    # 显示帮助信息
    echo
    show_help
}

show_help() {
    cat << EOF
usage: yc [-h] [-d] [command] [args...]

YesCode API CLI 工具

说明:
  无参数运行 'yc' 将进入交互式菜单模式 ★

主要命令:
  menu                              显示交互式菜单 ★
  profile                           查询用户资料
  balance                           仅查询余额信息
  balance-preference [preference]   设置余额使用偏好 ★
                                      preference: subscription_first | payg_only
  switch [id]                       交互式切换提供商 ★
                                      id: 分组 ID（可选，留空则交互式选择）

其他命令:
  install                           安装 yc 到系统
  uninstall                         卸载 yc
  providers                         查询提供商分组
  provider-alternatives <id>        查询指定分组的所有可用提供商
                                      id: 分组 ID
  provider-selection <id>           查询指定分组当前生效的提供商
                                      id: 分组 ID
  set-provider-selection <id> <alt> 设置指定分组的生效提供商
                                      id: 分组 ID, alt: 提供商 ID

选项:
  -h, --help                        显示帮助信息
  -d, --debug                       显示调试信息（请求/响应详情）
EOF
}

# ============================================================
# 主程序
# ============================================================

main() {
    # 全局 debug 标志
    DEBUG=false

    # 解析全局选项
    while [[ $# -gt 0 ]]; do
        case "$1" in
            -d|--debug)
                DEBUG=true
                shift
                ;;
            *)
                # 遇到非选项参数，结束循环
                break
                ;;
        esac
    done

    # 解析命令行参数
    if [[ $# -eq 0 ]]; then
        load_config
        cmd_interactive_menu
        exit 0
    fi

    # 处理不需要依赖检查的命令
    case "$1" in
        -h|--help)
            show_help
            exit 0
            ;;
        install)
            cmd_install
            exit 0
            ;;
        uninstall)
            cmd_uninstall
            exit 0
            ;;
    esac

    # 检查依赖（balance 等命令需要）
    if ! command -v jq &> /dev/null; then
        echo -e "${RED}错误: 未找到 jq 命令${RESET}"
        echo "请安装 jq: apt install jq (Ubuntu/Debian) 或 brew install jq (macOS)"
        exit 1
    fi

    if ! command -v curl &> /dev/null; then
        echo -e "${RED}错误: 未找到 curl 命令${RESET}"
        echo "请安装 curl: apt install curl (Ubuntu/Debian) 或 brew install curl (macOS)"
        exit 1
    fi

    # 处理需要依赖的命令
    case "$1" in
        balance)
            load_config
            cmd_balance
            ;;
        profile)
            load_config
            cmd_profile
            ;;
        providers)
            load_config
            cmd_providers
            ;;
        provider-alternatives)
            load_config
            cmd_provider_alternatives "$2"
            ;;
        provider-selection)
            load_config
            cmd_provider_selection "$2"
            ;;
        set-provider-selection)
            load_config
            cmd_set_provider_selection "$2" "$3"
            ;;
        switch)
            load_config
            cmd_switch "$2"
            ;;
        balance-preference)
            load_config
            cmd_balance_preference "$2"
            ;;
        menu)
            load_config
            cmd_interactive_menu
            ;;
        *)
            echo -e "${RED}错误: 未知命令 '$1'${RESET}"
            echo
            show_help
            exit 1
            ;;
    esac
}

main "$@"