#!/bin/bash

# Hysteria2 节点信息显示脚本

# 从配置文件解析信息
parse_config_info() {
    local config_file="$CONFIG_PATH"
    local node_info=()
    
    if [[ ! -f "$config_file" ]]; then
        echo "配置文件不存在"
        return 1
    fi
    
    # 解析监听端口
    local port=$(grep -E "^listen:" "$config_file" | awk '{print $2}' | sed 's/://')
    if [[ -z "$port" ]]; then
        port="443"
    fi
    
    # 解析认证密码
    local auth_password=$(grep -A 2 "^auth:" "$config_file" | grep "password:" | awk '{print $2}')
    
    # 解析混淆密码
    local obfs_password=""
    if grep -q "^obfs:" "$config_file"; then
        obfs_password=$(grep -A 3 "^obfs:" "$config_file" | grep "password:" | awk '{print $2}')
    fi
    
    # 解析伪装域名
    local masquerade_url=$(grep -A 3 "masquerade:" "$config_file" | grep "url:" | awk '{print $2}')
    local sni_domain=""
    if [[ -n "$masquerade_url" ]]; then
        sni_domain=$(echo "$masquerade_url" | sed 's|https\?://||' | sed 's|/.*||')
    fi
    
    # 检查证书类型
    local cert_type="ACME"
    local insecure="false"
    if grep -q "^tls:" "$config_file"; then
        cert_type="自签名"
        insecure="true"
    fi
    
    echo "$port|$auth_password|$obfs_password|$sni_domain|$cert_type|$insecure"
}

# 获取服务器域名配置
get_server_domain() {
    if [[ -f "/etc/hysteria/server-domain.conf" ]]; then
        cat "/etc/hysteria/server-domain.conf"
    else
        echo ""
    fi
}

# 获取服务器IP
get_current_server_ip() {
    local ip=""

    # 尝试多种方法获取公网IP
    ip=$(curl -s --connect-timeout 5 ipv4.icanhazip.com 2>/dev/null) || \
    ip=$(curl -s --connect-timeout 5 ifconfig.me 2>/dev/null) || \
    ip=$(curl -s --connect-timeout 5 ip.sb 2>/dev/null) || \
    ip=$(curl -s --connect-timeout 5 checkip.amazonaws.com 2>/dev/null)

    if [[ -n "$ip" && "$ip" =~ ^[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}$ ]]; then
        echo "$ip"
    else
        # 如果无法获取公网IP，尝试获取本地IP
        ip=$(ip route get 8.8.8.8 2>/dev/null | grep -oP 'src \K\S+')
        echo "${ip:-127.0.0.1}"
    fi
}

# 获取服务器地址（优先使用域名）
get_server_address() {
    local configured_domain=$(get_server_domain)

    if [[ -n "$configured_domain" ]]; then
        echo "$configured_domain"
    else
        get_current_server_ip
    fi
}

# 获取端口跳跃信息
get_port_hopping_info() {
    if [[ -f "/etc/hysteria/port-hopping.conf" ]]; then
        source "/etc/hysteria/port-hopping.conf"
        echo "$START_PORT-$END_PORT"
    else
        echo "未配置"
    fi
}

# 生成节点链接
generate_node_link() {
    local server_ip="$1"
    local port="$2"
    local auth_password="$3"
    local obfs_password="$4"
    local sni_domain="$5"
    local insecure="$6"
    
    local link="hysteria2://$auth_password@$server_ip:$port"
    local params=""
    
    if [[ -n "$sni_domain" ]]; then
        params="${params}&sni=$sni_domain"
    fi
    
    if [[ "$insecure" == "true" ]]; then
        params="${params}&insecure=1"
    fi
    
    if [[ -n "$obfs_password" ]]; then
        params="${params}&obfs=salamander&obfs-password=$obfs_password"
    fi
    
    # 移除开头的&
    params="${params#&}"
    
    if [[ -n "$params" ]]; then
        link="${link}?${params}"
    fi
    
    link="${link}#Hysteria2-Server"
    
    echo "$link"
}

# 生成订阅链接
generate_subscription_link() {
    local node_link="$1"
    echo "$node_link" | base64 -w 0
}

# 生成 Clash 配置
generate_clash_config() {
    local server_address="$1"
    local port="$2"
    local auth_password="$3"
    local obfs_password="$4"
    local sni_domain="$5"
    local insecure="$6"

    cat << EOF
# Clash 配置片段 (Hysteria2)
proxies:
  - name: "Hysteria2-Server"
    type: hysteria2
    server: $server_address
    port: $port
    password: $auth_password
EOF
    
    if [[ -n "$obfs_password" ]]; then
        cat << EOF
    obfs: salamander
    obfs-password: "$obfs_password"
EOF
    fi
    
    if [[ -n "$sni_domain" ]]; then
        cat << EOF
    sni: $sni_domain
EOF
    fi
    
    if [[ "$insecure" == "true" ]]; then
        cat << EOF
    skip-cert-verify: true
EOF
    fi
    
    cat << EOF
    alpn:
      - h3
EOF
}

# 生成 SingBox 配置
generate_singbox_config() {
    local server_address="$1"
    local port="$2"
    local auth_password="$3"
    local obfs_password="$4"
    local sni_domain="$5"
    local insecure="$6"

    cat << EOF
# SingBox 配置片段 (Hysteria2)
{
  "type": "hysteria2",
  "tag": "Hysteria2-Server",
  "server": "$server_address",
  "server_port": $port,
  "password": "$auth_password",
EOF
    
    if [[ -n "$obfs_password" ]]; then
        cat << EOF
  "obfs": {
    "type": "salamander",
    "password": "$obfs_password"
  },
EOF
    fi
    
    cat << EOF
  "tls": {
EOF
    
    if [[ -n "$sni_domain" ]]; then
        cat << EOF
    "server_name": "$sni_domain",
EOF
    fi
    
    if [[ "$insecure" == "true" ]]; then
        cat << EOF
    "insecure": true,
EOF
    fi
    
    cat << EOF
    "alpn": ["h3"]
  }
}
EOF
}

# 生成不同客户端的配置和订阅
generate_client_subscriptions() {
    local node_link="$1"
    local server_address="$2"
    local port="$3"
    local auth_password="$4"
    local obfs_password="$5"
    local sni_domain="$6"
    local insecure="$7"

    echo -e "${CYAN}=== 客户端配置和订阅 ===${NC}"
    echo ""

    # Hysteria2 通用订阅链接
    local hysteria2_sub=$(echo "$node_link" | base64 -w 0)
    echo -e "${YELLOW}1. Hysteria2 通用订阅链接:${NC}"
    echo "$hysteria2_sub"
    echo ""

    # Hysteria2 节点链接
    echo -e "${YELLOW}2. Hysteria2 节点链接:${NC}"
    echo "$node_link"
    echo ""

    # Clash 配置
    echo -e "${YELLOW}3. Clash 配置:${NC}"
    generate_clash_config "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure"
    echo ""

    # SingBox 配置
    echo -e "${YELLOW}4. SingBox 配置:${NC}"
    generate_singbox_config "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure"
    echo ""

    echo -e "${GREEN}=== 支持 Hysteria2 的客户端 ===${NC}"
    echo ""
    echo -e "${BLUE}桌面客户端:${NC}"
    echo "• Clash Verge Rev (推荐) - Windows/macOS/Linux"
    echo "• Clash Meta (ClashX Pro) - Windows/macOS/Linux"  
    echo "• SingBox (官方客户端) - Windows/macOS/Linux"
    echo "• Hiddify Next - Windows/macOS/Linux"
    echo "• NekoRay/NekoBox - Windows/Linux"
    echo "• v2rayN - Windows"
    echo "• V2rayU - macOS"
    echo ""
    echo -e "${BLUE}移动客户端:${NC}"
    echo "• v2rayNG - Android (推荐)"
    echo "• NekoBox for Android - Android"
    echo "• SingBox - Android"
    echo "• Hiddify Next - Android"
    echo "• Clash Meta for Android - Android"
    echo "• ShadowRocket - iOS (推荐)"
    echo "• Stash - iOS"
    echo "• QuantumultX - iOS"
    echo "• Loon - iOS"
    echo ""
    echo -e "${BLUE}路由器/OpenWrt:${NC}"
    echo "• OpenClash - 支持 Hysteria2"
    echo "• SingBox - 官方路由器版本"
    echo "• Clash Premium/Meta 核心"
    echo ""
    echo -e "${YELLOW}使用建议:${NC}"
    echo "• 优先选择支持 Hysteria2 的新版客户端"
    echo "• 推荐使用 Clash Verge Rev 或 v2rayNG"
    echo "• iOS 用户推荐 ShadowRocket"
    echo "• 节点链接和订阅链接都可使用"
}

# 生成客户端配置
generate_client_config() {
    local server_address="$1"
    local port="$2"
    local auth_password="$3"
    local obfs_password="$4"
    local sni_domain="$5"
    local insecure="$6"

    cat << EOF
# Hysteria2 官方客户端配置
server: $server_address:$port
auth: $auth_password

tls:
  sni: $sni_domain
  insecure: $insecure

EOF
    
    if [[ -n "$obfs_password" ]]; then
        cat << EOF
obfs:
  type: salamander
  salamander:
    password: $obfs_password

EOF
    fi
    
    cat << EOF
socks5:
  listen: 127.0.0.1:1080

http:
  listen: 127.0.0.1:8080

bandwidth:
  up: 100 mbps
  down: 100 mbps

# 可选: UDP 转发
# udpForwarding:
#   - listen: 127.0.0.1:5353
#     remote: 8.8.8.8:53

# 可选: TCP 转发
# tcpForwarding:
#   - listen: 127.0.0.1:6666
#     remote: www.google.com:80
EOF
}

# 生成二维码 (如果有 qrencode)
generate_qrcode() {
    local content="$1"
    
    if command -v qrencode &> /dev/null; then
        echo -e "${BLUE}二维码:${NC}"
        qrencode -t ANSIUTF8 "$content"
        echo ""
    else
        echo -e "${YELLOW}提示: 安装 qrencode 可生成二维码${NC}"
        echo "Ubuntu/Debian: sudo apt install qrencode"
        echo "CentOS/RHEL: sudo yum install qrencode"
        echo ""
    fi
}

# 显示节点信息
display_node_info() {
    echo -e "${BLUE}Hysteria2 节点信息${NC}"
    echo ""
    
    # 检查服务状态
    if ! systemctl is-active --quiet hysteria-server.service; then
        echo -e "${RED}警告: Hysteria2 服务未运行${NC}"
        echo "请先启动服务"
        echo ""
        read -p "按回车键继续..."
        return
    fi
    
    # 检查配置文件
    if [[ ! -f "$CONFIG_PATH" ]]; then
        echo -e "${RED}错误: 配置文件不存在${NC}"
        read -p "按回车键继续..."
        return
    fi
    
    # 获取服务器信息
    local server_address=$(get_server_address)
    local server_ip=$(get_current_server_ip)
    local configured_domain=$(get_server_domain)
    local config_info=$(parse_config_info)

    if [[ -z "$config_info" ]]; then
        echo -e "${RED}错误: 无法解析配置文件${NC}"
        read -p "按回车键继续..."
        return
    fi

    # 解析配置信息
    IFS='|' read -r port auth_password obfs_password sni_domain cert_type insecure <<< "$config_info"

    # 获取端口跳跃信息
    local port_hopping=$(get_port_hopping_info)

    # 显示基本信息
    echo -e "${CYAN}=== 服务器信息 ===${NC}"
    if [[ -n "$configured_domain" ]]; then
        echo -e "${YELLOW}服务器域名:${NC} $configured_domain:$port"
        echo -e "${YELLOW}服务器IP:${NC} $server_ip:$port"
    else
        echo -e "${YELLOW}服务器地址:${NC} $server_ip:$port"
    fi
    echo -e "${YELLOW}认证密码:${NC} $auth_password"
    if [[ -n "$obfs_password" ]]; then
        echo -e "${YELLOW}混淆密码:${NC} $obfs_password"
        echo -e "${YELLOW}混淆类型:${NC} Salamander"
    else
        echo -e "${YELLOW}混淆配置:${NC} 未启用"
    fi
    echo -e "${YELLOW}SNI域名:${NC} ${sni_domain:-未设置}"
    echo -e "${YELLOW}证书类型:${NC} $cert_type"
    echo -e "${YELLOW}端口跳跃:${NC} $port_hopping"
    echo ""

    # 生成链接（使用服务器地址）
    local node_link=$(generate_node_link "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure")
    
    while true; do
        echo -e "${CYAN}=== 节点信息选项 ===${NC}"
        echo -e "${GREEN}1.${NC} 显示节点链接和二维码"
        echo -e "${GREEN}2.${NC} 显示客户端配置和订阅"
        echo -e "${GREEN}3.${NC} 显示 Hysteria2 官方客户端配置"
        echo -e "${GREEN}4.${NC} 显示 Clash 配置"
        echo -e "${GREEN}5.${NC} 显示 SingBox 配置"
        echo -e "${GREEN}6.${NC} 保存完整信息到文件"
        echo -e "${GREEN}7.${NC} 刷新信息"
        echo -e "${RED}0.${NC} 返回主菜单"
        echo ""
        echo -n -e "${BLUE}请选择操作 [0-7]: ${NC}"
        read -r choice
        
        case $choice in
            1)
                echo ""
                echo -e "${CYAN}=== 节点链接 ===${NC}"
                echo "$node_link"
                echo ""
                generate_qrcode "$node_link"
                echo ""
                read -p "按回车键继续..."
                ;;
            2)
                echo ""
                generate_client_subscriptions "$node_link" "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure"
                echo ""
                read -p "按回车键继续..."
                ;;
            3)
                echo ""
                echo -e "${CYAN}=== Hysteria2 官方客户端配置 ===${NC}"
                echo ""
                generate_client_config "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure"
                echo ""
                read -p "按回车键继续..."
                ;;
            4)
                echo ""
                echo -e "${CYAN}=== Clash 配置 ===${NC}"
                echo ""
                generate_clash_config "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure"
                echo ""
                read -p "按回车键继续..."
                ;;
            5)
                echo ""
                echo -e "${CYAN}=== SingBox 配置 ===${NC}"
                echo ""
                generate_singbox_config "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure"
                echo ""
                read -p "按回车键继续..."
                ;;
            6)
                save_node_info_to_file "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure" "$port_hopping" "$node_link"
                ;;
            7)
                # 刷新信息
                server_address=$(get_server_address)
                server_ip=$(get_current_server_ip)
                configured_domain=$(get_server_domain)
                config_info=$(parse_config_info)
                IFS='|' read -r port auth_password obfs_password sni_domain cert_type insecure <<< "$config_info"
                port_hopping=$(get_port_hopping_info)
                node_link=$(generate_node_link "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure")
                echo -e "${GREEN}信息已刷新${NC}"
                sleep 1
                ;;
            0)
                break
                ;;
            *)
                echo -e "${RED}无效选项${NC}"
                sleep 1
                ;;
        esac
    done
}

# 保存节点信息到文件
save_node_info_to_file() {
    local server_address="$1"
    local port="$2"
    local auth_password="$3"
    local obfs_password="$4"
    local sni_domain="$5"
    local insecure="$6"
    local port_hopping="$7"
    local node_link="$8"

    local output_file="/etc/hysteria/node-info.txt"
    local configured_domain=$(get_server_domain)
    local server_ip=$(get_current_server_ip)

    cat > "$output_file" << EOF
# Hysteria2 节点信息
# 生成时间: $(date)

=== 服务器信息 ===
EOF

    if [[ -n "$configured_domain" ]]; then
        cat >> "$output_file" << EOF
服务器域名: $configured_domain:$port
服务器IP: $server_ip:$port
EOF
    else
        cat >> "$output_file" << EOF
服务器地址: $server_address:$port
EOF
    fi

    cat >> "$output_file" << EOF
认证密码: $auth_password
混淆密码: ${obfs_password:-未启用}
混淆类型: $([ -n "$obfs_password" ] && echo "Salamander" || echo "未启用")
SNI域名: ${sni_domain:-未设置}
证书验证: $([ "$insecure" == "true" ] && echo "忽略 (自签名)" || echo "验证 (ACME)")
端口跳跃: $port_hopping

=== 节点链接 ===
$node_link

=== Hysteria2 通用订阅链接 ===
$(echo "$node_link" | base64 -w 0)

=== Hysteria2 官方客户端配置 ===
$(generate_client_config "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure")

=== Clash 配置 ===
$(generate_clash_config "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure")

=== SingBox 配置 ===
$(generate_singbox_config "$server_address" "$port" "$auth_password" "$obfs_password" "$sni_domain" "$insecure")

=== 支持 Hysteria2 的客户端 ===

桌面客户端:
• Clash Verge Rev (推荐) - Windows/macOS/Linux
• Clash Meta (ClashX Pro) - Windows/macOS/Linux  
• SingBox (官方客户端) - Windows/macOS/Linux
• Hiddify Next - Windows/macOS/Linux
• NekoRay/NekoBox - Windows/Linux
• v2rayN - Windows
• V2rayU - macOS

移动客户端:
• v2rayNG - Android (推荐)
• NekoBox for Android - Android
• SingBox - Android
• Hiddify Next - Android
• Clash Meta for Android - Android
• ShadowRocket - iOS (推荐)
• Stash - iOS
• QuantumultX - iOS
• Loon - iOS

路由器/OpenWrt:
• OpenClash - 支持 Hysteria2
• SingBox - 官方路由器版本
• Clash Premium/Meta 核心

使用建议:
• 优先选择支持 Hysteria2 的新版客户端
• 推荐使用 Clash Verge Rev 或 v2rayNG
• iOS 用户推荐 ShadowRocket
• 节点链接和订阅链接都可使用
EOF

    echo ""
    echo -e "${GREEN}完整节点信息已保存到: $output_file${NC}"
    echo ""
    read -p "按回车键继续..."
}