#!/usr/bin/env bash
set -euo pipefail  # 严格的安全设置

# =============================================================================
# 颜色和样式配置
# =============================================================================
readonly RED_COLOR='\033[1;31m'
readonly GREEN_COLOR='\033[1;32m'
readonly YELLOW_COLOR='\033[1;33m'
readonly BLUE_COLOR='\033[1;34m'
readonly MAGENTA_COLOR='\033[1;35m'
readonly CYAN_COLOR='\033[1;36m'
readonly BOLD='\033[1m'
readonly RESET='\033[0m'

# =============================================================================
# 全局常量定义
# =============================================================================
readonly SCRIPT_NAME="OpenWRT 构建系统"
readonly SCRIPT_VERSION="1.0.0"
readonly AUTHOR="OPPEN321"
readonly BLOG="www.kejizero.online"
readonly MIRROR="https://raw.githubusercontent.com/BlueStack-Sky/QuickWrt/refs/heads/master"
readonly SUPPORTED_ARCHITECTURES=("rockchip" "x86_64")
readonly REQUIRED_USER="zhao"

# =============================================================================
# 全局变量
# =============================================================================
GROUP_FLAG=false
START_TIME=$(date +%s)
CPU_CORES=$(( $(nproc --all) + 1 ))

# =============================================================================
# 函数定义
# =============================================================================

# 打印带颜色的消息
print_color() {
    local color=$1
    shift
    echo -e "${color}$*${RESET}"
}

# 打印错误消息并退出
error_exit() {
    print_color "$RED_COLOR" "❌ 错误: $1"
    exit 1
}

# 打印警告消息
print_warning() {
    print_color "$YELLOW_COLOR" "⚠️  警告: $1"
}

# 打印成功消息
print_success() {
    print_color "$GREEN_COLOR" "✅ $1"
}

# 打印信息消息
print_info() {
    print_color "$BLUE_COLOR" "ℹ️  信息: $1"
}

# 验证必需的环境变量
validate_environment() {
    if [[ "$(whoami)" != "$REQUIRED_USER" ]] && [[ -z "${git_name:-}" || -z "${git_password:-}" ]]; then
        error_exit "未授权访问。请设置认证信息后再执行此脚本。"
    fi
}

# 显示使用帮助
show_usage() {
    echo -e "\n${BOLD}使用方法:${RESET}"
    echo -e "  bash $0 <version> <architecture>"
    echo -e "\n${BOLD}支持的架构:${RESET}"
    for arch in "${SUPPORTED_ARCHITECTURES[@]}"; do
        echo -e "  • ${GREEN_COLOR}$arch${RESET}"
    done
    echo -e "\n${BOLD}示例:${RESET}"
    echo -e "  bash $0 v24 x86_64"
    echo -e "  bash $0 v24 rockchip"
}

# 验证参数
validate_arguments() {
    local version="$1"
    local arch="$2"
    
    if [[ -z "$version" ]]; then
        error_exit "未指定版本号"
    fi
    
    if [[ -z "$arch" ]]; then
        error_exit "未指定目标架构"
    fi
    
    local valid_arch=false
    for supported_arch in "${SUPPORTED_ARCHITECTURES[@]}"; do
        if [[ "$arch" == "$supported_arch" ]]; then
            valid_arch=true
            break
        fi
    done
    
    if [[ "$valid_arch" == false ]]; then
        error_exit "不支持的架构: '$arch'"
    fi
}

# 显示横幅
show_banner() {
    clear
    echo -e ""
    echo -e "${BOLD}${BLUE_COLOR}╔══════════════════════════════════════════════════════════════════╗${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}║${RESET}                       OpenWRT 自动化构建系统                     ${BOLD}${BLUE_COLOR}║${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}╠══════════════════════════════════════════════════════════════════╣${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}║${RESET}  🛠️   ${BOLD}开发者:${RESET} $AUTHOR                                            ${BOLD}${BLUE_COLOR}║${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}║${RESET}  🌐   ${BOLD}博客:${RESET} $BLOG                                  ${BOLD}${BLUE_COLOR}║${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}║${RESET}  💡   ${BOLD}理念:${RESET} 开源 · 定制化 · 高性能                               ${BOLD}${BLUE_COLOR}║${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}║${RESET}  📦   ${BOLD}版本:${RESET} $SCRIPT_VERSION                                                ${BOLD}${BLUE_COLOR}║${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}╠══════════════════════════════════════════════════════════════════╣${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}║${RESET}  🔧 ${GREEN_COLOR}构建开始:${RESET} $(date '+%Y-%m-%d %H:%M:%S')                                ${BOLD}${BLUE_COLOR}║${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}║${RESET}  ⚡ ${GREEN_COLOR}处理器核心:${RESET} $CPU_CORES 个                                           ${BOLD}${BLUE_COLOR}║${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}║${RESET}  🐧 ${GREEN_COLOR}系统用户:${RESET} $(whoami)                                               ${BOLD}${BLUE_COLOR}║${RESET}"
    echo -e "${BOLD}${BLUE_COLOR}╚══════════════════════════════════════════════════════════════════╝${RESET}"
    echo -e ""
}

# 初始化构建环境
setup_build_environment() {
    if [[ "$(id -u)" == "0" ]]; then
        export FORCE_UNSAFE_CONFIGURE=1
        export FORCE=1
        print_warning "以 root 权限运行，已启用强制不安全配置"
    fi
}

# 设置下载进度条
setup_curl_progress() {
    if curl --help | grep -q progress-bar; then
        CURL_OPTIONS="--progress-bar"
    else
        CURL_OPTIONS="--silent"
    fi
    export CURL_OPTIONS
}

# 编译脚本
compilation_script() {
    print_info "开始查询最新 OpenWRT 版本..."
    tag_version="$(curl -s https://github.com/openwrt/openwrt/tags | grep -Eo "v[0-9\.]+\-*r*c*[0-9]*.tar.gz" | sed -n '/[2-9][4-9]/p' | sed -n 1p | sed 's/v//g' | sed 's/.tar.gz//g')"
    export tag_version="$tag_version"
    print_success "检测到最新版本: $tag_version"

    print_info "开始克隆源代码仓库..."
    git -c advice.detachedHead=false clone --depth=1 --quiet https://github.com/openwrt/openwrt -b "v$tag_version"
    git clone --depth=1 --quiet -b openwrt-24.10 https://github.com/immortalwrt/immortalwrt
    git clone --depth=1 --quiet -b openwrt-24.10 https://github.com/openwrt/openwrt openwrt_snap

    print_info "正在进行源代码处理..."
    find openwrt/package/* -maxdepth 0 ! -name 'firmware' ! -name 'kernel' ! -name 'base-files' ! -name 'Makefile' -exec rm -rf {} +
    rm -rf ./openwrt_snap/package/firmware ./openwrt_snap/package/kernel ./openwrt_snap/package/base-files ./openwrt_snap/package/Makefile
    cp -rf ./openwrt_snap/package/* ./openwrt/package/
    cp -rf ./openwrt_snap/feeds.conf.default ./openwrt/feeds.conf.default

    print_info "为 Rockchip 架构替换 ImmortalWRT 组件以增强设备兼容性..."
    rm -rf openwrt/package/boot/{rkbin,uboot-rockchip,arm-trusted-firmware-rockchip}
    rm -rf openwrt/target/linux/rockchip
    cp -rf immortalwrt/target/linux/rockchip openwrt/target/linux/rockchip
    cp -rf immortalwrt/package/boot/uboot-rockchip openwrt/package/boot/uboot-rockchip
    cp -rf immortalwrt/package/boot/arm-trusted-firmware-rockchip openwrt/package/boot/arm-trusted-firmware-rockchip

    print_info "正在克隆 OpenBox 仓库以支持后续编译"
    git clone --depth=1 --quiet -b main https://github.com/BlueStack-Sky/OpenBox

    print_info "正在复制密钥文件..."
    if [ -d "openwrt" ]; then
        cd openwrt || { printf "%b\n" "${RED_COLOR}进入 openwrt 目录失败${RES}"; exit 1; }

        if cp -rf ../OpenBox/key.tar.gz ./key.tar.gz; then
            if tar zxf key.tar.gz; then
                rm -f key.tar.gz
                print_info "密钥已复制并解压完成"
            else
                printf "%b\n" "${RED_COLOR}解压 key.tar.gz 失败${RES}"
                exit 1
            fi
        else
            printf "%b\n" "${RED_COLOR}复制 key.tar.gz 失败${RES}"
            exit 1
        fi
    else
        printf "%b\n" "${RED_COLOR}未找到 openwrt 源码目录，下载源码失败${RES}"
        exit 1
    fi

    print_info "正在更新软件源 feeds..."
    echo -e "${BLUE_COLOR}├─ 更新软件包列表...${RESET}"
    if ./scripts/feeds update -a > /dev/null 2>&1; then
        echo -e "${GREEN_COLOR}├─ 软件包列表更新成功${RESET}"
    else
        error_exit "feeds 更新失败"
    fi
    
    echo -e "${BLUE_COLOR}├─ 安装软件包依赖...${RESET}"
    if ./scripts/feeds install -a > /dev/null 2>&1; then
        echo -e "${GREEN_COLOR}└─ 软件包依赖安装完成${RESET}"
        print_success "Feeds 更新和安装完成"
    else
        error_exit "feeds 安装失败"
    fi
    
    print_info "下载并执行构建脚本..."
    local scripts=(
        00-prepare_base.sh
        01-prepare_package.sh
        02-rockchip_target_only.sh
        02-x86_64_target_only.sh
    )

    # 下载所有脚本
    echo -e "${BLUE_COLOR}├─ 下载构建脚本...${RESET}"
    for script in "${scripts[@]}"; do
        if curl -sO "$MIRROR/scripts/$script"; then
            echo -e "${GREEN_COLOR}│   ✓ 已下载: $script${RESET}"
        else
            error_exit "下载脚本 $script 失败"
        fi
    done

    echo -e "${BLUE_COLOR}├─ 设置脚本执行权限...${RESET}"
    if chmod 0755 ./*.sh; then
        echo -e "${GREEN_COLOR}│   ✓ 权限设置完成${RESET}"
    else
        error_exit "设置脚本权限失败"
    fi

    # 执行基础准备脚本
    echo -e "${BLUE_COLOR}├─ 执行基础环境准备...${RESET}"
    local base_scripts=(
        "00-prepare_base.sh"
        "01-prepare_package.sh" 
    )

    for script in "${base_scripts[@]}"; do
        echo -e "${BLUE_COLOR}│   ├─ 执行: $script${RESET}"
        if bash "$script" > /dev/null 2>&1; then
            echo -e "${GREEN_COLOR}│   │   ✓ 完成${RESET}"
        else
            error_exit "脚本 $script 执行失败"
        fi
    done

    # 执行架构特定脚本
    echo -e "${BLUE_COLOR}├─ 执行架构特定配置...${RESET}"
    if [[ "$1" == "rockchip" ]]; then
        echo -e "${BLUE_COLOR}│   ├─ 配置 Rockchip 架构${RESET}"
        if bash 02-rockchip_target_only.sh > /dev/null 2>&1; then
            export core=arm64
            echo -e "${GREEN_COLOR}│   │   ✓ Rockchip 架构配置完成${RESET}"
            print_success "Rockchip 架构配置完成"
        else
            error_exit "Rockchip 架构配置脚本执行失败"
        fi
    elif [[ "$1" == "x86_64" ]]; then
        echo -e "${BLUE_COLOR}│   ├─ 配置 x86_64 架构${RESET}"
        if bash 02-x86_64_target_only.sh > /dev/null 2>&1; then
            export core=amd64
            echo -e "${GREEN_COLOR}│   │   ✓ x86_64 架构配置完成${RESET}"
            print_success "x86_64 架构配置完成"
        else
            error_exit "x86_64 架构配置脚本执行失败"
        fi
    fi

    # 清理临时脚本文件
    echo -e "${BLUE_COLOR}├─ 清理临时文件...${RESET}"
    if rm -f 0*-*.sh; then
        echo -e "${GREEN_COLOR}└─ ✓ 临时文件清理完成${RESET}"
    else
        print_warning "清理临时文件时出现警告，但可继续执行"
    fi

    print_success "构建环境准备完成"
}

# =============================================================================
# 主程序逻辑
# =============================================================================
main() {
    local version="${1:-}"
    local architecture="${2:-}"
    
    # 参数验证
    validate_arguments "$version" "$architecture"
    
    # 显示横幅
    show_banner
    
    # 环境验证
    validate_environment
    
    # 环境设置
    setup_build_environment
    setup_curl_progress
    
    print_success "初始化完成，开始构建 $architecture 架构的 $version 版本"
    
    # 记录开始时间
    START_TIME=$(date +%s)
    
    # 执行编译脚本
    compilation_script "$architecture"
    
    # 计算构建时间
    local end_time=$(date +%s)
    local duration=$((end_time - START_TIME))
    print_success "构建完成！总耗时: $((duration / 60)) 分 $((duration % 60)) 秒"
}

# 脚本入口点
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
    # 如果没有提供足够的参数，显示使用帮助
    if [[ $# -lt 2 ]]; then
        show_usage
        error_exit "参数不足，需要指定版本号和目标架构"
    fi
    
    # 执行主程序
    main "$@"
fi