#!/bin/bash

# --- 样式定义 ---
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
BLUE='\033[0;34m'
PLAIN='\033[0m'

# --- 变量配置 ---
UA_Browser="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36"

clear
echo -e "${BLUE}=======================================${PLAIN}"
echo -e "${BLUE}          Bleki-流媒体解锁脚本   ${PLAIN}"
echo -e "${BLUE}=======================================${PLAIN}"

# 通用对齐输出函数
print_result() {
    printf " %-20s\t\t%b%s%b\n" "$1" "$2" "$3" "$PLAIN"
}

# 1. YouTube Premium (保留地区判断)
check_youtube() {
    local res=$(curl -4 -sL --user-agent "${UA_Browser}" "https://www.youtube.com/premium" 2>&1)
    if [[ "$res" == *"Premium is not available in your country"* ]]; then
        print_result "YouTube Premium:" "$RED" "未支持 (Not Available)"
    else
        local region=$(echo "$res" | grep -oP '"gl":"\K[A-Z]{2}' | head -n1)
        print_result "YouTube Premium:" "$GREEN" "支持解锁 (Region: ${region:-Unknown})"
    fi
}

# 2. Netflix (保留地区判断)
check_netflix() {
    local test_url="https://www.netflix.com/title/70143836"
    local res=$(curl -4 -sSL -m 10 -D - "$test_url" -o /dev/null)
    local http_code=$(echo "$res" | grep -i "HTTP/" | awk '{print $2}' | tail -n1)
    if [[ "$http_code" == "404" ]]; then
        print_result "Netflix:" "$YELLOW" "仅限自制剧 (Originals Only)"
    elif [[ "$http_code" == "200" ]]; then
        local region=$(curl -4 -sL -m 10 "$test_url" | grep -oP '"country":\s*"\K[A-Z]{2}' | head -n1)
        print_result "Netflix:" "$GREEN" "完整解锁 (Region: ${region:-Unknown})"
    else
        print_result "Netflix:" "$RED" "失败 (Not Available)"
    fi
}

# 3. Disney+ (仅判断解锁)
check_disney() {
    local res=$(curl -4 -sL -m 10 -o /dev/null -w "%{url_effective}" "https://www.disneyplus.com" -H "User-Agent: ${UA_Browser}")
    if [[ "$res" == *"unavailable"* ]]; then
        print_result "Disney+:" "$RED" "不支持 (Unavailable)"
    else
        print_result "Disney+:" "$GREEN" "支持解锁"
    fi
}

# 4. HBO Max (Max) (仅判断解锁)
check_max() {
    local res=$(curl -4 -sI -m 10 "https://www.max.com/" | grep -i "location")
    if [[ "$res" == *"geo-unavailable"* ]]; then
        print_result "HBO Max:" "$RED" "不支持 (Geo-Blocked)"
    else
        print_result "HBO Max:" "$GREEN" "支持解锁"
    fi
}

# 5. Amazon Prime Video (仅判断解锁)
check_primevideo() {
    local res=$(curl -4 -sL -m 10 -o /dev/null -w "%{url_effective}" "https://www.primevideo.com" -H "User-Agent: ${UA_Browser}")
    if [[ "$res" == *"region-restriction"* ]] || [[ "$res" == *"notavailable"* ]]; then
        print_result "Prime Video:" "$RED" "不支持 (Restricted)"
    else
        print_result "Prime Video:" "$GREEN" "支持解锁"
    fi
}

# 6. TikTok (仅判断解锁)
check_tiktok() {
    local http_code=$(curl -4 -sL -o /dev/null -w "%{http_code}" -m 10 "https://www.tiktok.com/" -H "User-Agent: ${UA_Browser}")
    if [[ "$http_code" == "200" ]] || [[ "$http_code" == "302" ]]; then
        print_result "TikTok:" "$GREEN" "支持解锁"
    else
        print_result "TikTok:" "$RED" "失败 (Blocked)"
    fi
}

echo -e "${BLUE}---------------------------------------${PLAIN}"

# 7. OpenAI / ChatGPT (仅判断解锁)
check_openai() {
    local res=$(curl -4 -sL -m 10 https://chat.openai.com/cdn-cgi/trace | grep "loc=")
    if [[ -n "$res" ]]; then
        print_result "OpenAI/ChatGPT:" "$GREEN" "支持解锁"
    else
        print_result "OpenAI/ChatGPT:" "$RED" "失败 (Blocked)"
    fi
}

# 8. Google Gemini (仅判断解锁)
check_gemini() {
    local http_code=$(curl -4 -sL -m 10 -o /dev/null -w "%{http_code}" -H "User-Agent: ${UA_Browser}" "https://gemini.google.com/app")
    if [[ "$http_code" == "200" ]] || [[ "$http_code" == "302" ]]; then
        print_result "Google Gemini:" "$GREEN" "支持解锁"
    else
        print_result "Google Gemini:" "$RED" "未支持 (Not Supported)"
    fi
}

# 9. Microsoft Copilot (仅判断解锁)
check_copilot() {
    local res=$(curl -4 -I -s -m 10 "https://copilot.microsoft.com/" | head -n 1)
    if [[ "$res" == *"200"* ]] || [[ "$res" == *"302"* ]]; then
        print_result "MS Copilot:" "$GREEN" "支持解锁"
    else
        print_result "MS Copilot:" "$RED" "失败 (Restricted)"
    fi
}

# --- 执行检测 ---
check_youtube
check_netflix
check_disney
check_max
check_primevideo
check_tiktok
echo -e "${BLUE}---------------------------------------${PLAIN}"
check_openai
check_gemini
check_copilot

echo -e "${BLUE}=======================================${PLAIN}"
echo -e " 测试完成时间: $(date '+%Y-%m-%d %H:%M:%S')"