#!/bin/bash
# -----------------------------------------------------------------------------------------------------------
# Copyright (c) 2026 Huawei Technologies Co., Ltd.
# This program is free software, you can redistribute it and/or modify it under the terms and conditions of
# CANN Open Software License Agreement Version 2.0 (the "License").
# Please refer to the License for details. You may not use this file except in compliance with the License.
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
# INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
# See LICENSE in the root of the software repository for the full text of the License.
# -----------------------------------------------------------------------------------------------------------

# ==============================================================================
# ge (Graph Engine) 编译环境依赖检查脚本
#
# 仓库:   https://gitcode.com/cann/ge
# 依据:   https://gitcode.com/cann/ge/blob/master/docs/zh/build.md
#         https://gitcode.com/cann/ge/blob/master/requirements.txt
# 用法:   bash scripts/check_env.sh
#
# 本脚本所有检查项和版本约束严格来源于 docs/zh/build.md 和 requirements.txt，
# 如文档更新，请同步修改本脚本。
# ==============================================================================

# 不使用 set -e, 避免 grep 无匹配时静默退出

# ==================== 颜色定义 ====================
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

# ==================== 计数器 ====================
ERROR_COUNT=0
WARNING_COUNT=0
PASS_COUNT=0

# ==============================================================================
# 版本要求 (严格来源于 docs/zh/build.md 和 requirements.txt)
#
# build.md 原文:
#   - GCC >= 7.3.x
#   - Python3 >= 3.9.x
#   - CMake >= 3.16.0 （建议使用3.20.0版本）
#   - bash >= 5.1.16
#   - ccache/asan/autoconf/automake/libtool/gperf/lcov/libasan/patch/graph-easy
#     (其中graph-easy可选)
#
# requirements.txt:
#   numpy
#   pybind11>=2.13.6,<3.0.0
#   jinja2
#   setuptools>=59.0.1
#   wheel>=0.37.1
#   coverage
#   cloudpickle
# ==============================================================================
REQUIRED_GCC_MIN="7.3.0"
REQUIRED_PYTHON_MIN="3.9.0"
REQUIRED_CMAKE_MIN="3.16.0"
REQUIRED_CMAKE_RECOMMEND="3.20.0"
REQUIRED_BASH_MIN="5.1.16"
REQUIRED_PYBIND11_MIN="2.13.6"
REQUIRED_SETUPTOOLS_MIN="59.0.1"
REQUIRED_WHEEL_MIN="0.37.1"

# ==================== 工具函数 ====================
log_pass() {
    PASS_COUNT=$((PASS_COUNT + 1))
    echo -e "  ${GREEN}[PASS]${NC}    $1"
}

log_warn() {
    WARNING_COUNT=$((WARNING_COUNT + 1))
    echo -e "  ${YELLOW}[WARNING]${NC} $1"
}

log_error() {
    ERROR_COUNT=$((ERROR_COUNT + 1))
    echo -e "  ${RED}[ERROR]${NC}   $1"
}

log_info() {
    echo -e "  ${BLUE}[INFO]${NC}    $1"
}

version_ge() {
    [ "$(printf '%s\n' "$1" "$2" | sort -V | head -n1)" = "$2" ]
}

version_lt() {
    [ "$(printf '%s\n' "$1" "$2" | sort -V | head -n1)" = "$1" ] && [ "$1" != "$2" ]
}

check_command() {
    command -v "$1" &>/dev/null
}

# 根据检测到的发行版输出安装命令提示
# 自动适配 Ubuntu/Debian(apt-get) 与 CentOS/RHEL/EulerOS/openEuler(yum/dnf)
# 参数 $1: apt 系包名 (Ubuntu/Debian)
# 参数 $2: yum/dnf 系包名 (CentOS/RHEL/EulerOS/openEuler)
# 依赖全局变量: DISTRO (发行版标识), PKG_MGR (包管理器: apt-get/yum/dnf)
pkg_install_hint() {
    local apt_pkg="$1"
    local yum_pkg="$2"
    case "$DISTRO" in
        ubuntu|debian)
            log_info "安装: sudo apt-get install $apt_pkg"
            ;;
        centos|rhel|fedora|euleros|openeuler)
            log_info "安装: sudo $PKG_MGR install $yum_pkg"
            ;;
        *)
            # 未识别发行版: 同时给出两种主流方案
            log_info "安装 (Ubuntu/Debian):   sudo apt-get install $apt_pkg"
            log_info "安装 (CentOS/EulerOS): sudo yum install $yum_pkg"
            ;;
    esac
}

# 安全提取版本号 (支持 1/2/3/4 段, 失败返回 0.0.0)
extract_version() {
    local input="$1"
    local ver

    # 移除前缀 v 或 V
    input=$(echo "$input" | sed 's/^[vV]//')
    # 移除后缀（-rc, -alpha, -beta 等）
    input=$(echo "$input" | sed 's/-.*$//')

    ver=$(echo "$input" | grep -oP '\d+\.\d+(\.\d+)*' | head -n1 || true)

    if [ -z "$ver" ]; then
        ver=$(echo "$input" | grep -oP '\d+' | head -n1 || true)
        if [ -n "$ver" ]; then
            ver="${ver}.0.0"
        else
            echo "0.0.0"
            return
        fi
    fi

    if [[ "$ver" =~ ^[0-9]+\.[0-9]+$ ]]; then
        ver="${ver}.0"
    fi

    echo "$ver"
}

print_header() {
    echo ""
    echo -e "${BLUE}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
    echo -e "${BLUE}  $1${NC}"
    echo -e "${BLUE}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
}

# ==============================================================================
echo ""
echo "=================================================================="
echo "  ge (Graph Engine) 编译环境依赖检查"
echo "  仓库: https://gitcode.com/cann/ge"
echo "  依据: docs/zh/build.md + requirements.txt"
echo "  时间: $(date '+%Y-%m-%d %H:%M:%S')"
echo "  系统: $(uname -s) $(uname -m)"
echo "=================================================================="

# ==================== 1. 操作系统 ====================
print_header "1. 操作系统 [可选]"

OS_NAME=$(uname -s)
OS_ARCH=$(uname -m)

if [ "$OS_NAME" = "Linux" ]; then
    log_info "操作系统: Linux"
else
    log_info "其他操作系统: $OS_NAME"
fi

if [ "$OS_ARCH" = "x86_64" ] || [ "$OS_ARCH" = "aarch64" ]; then
    log_info "CPU 架构: $OS_ARCH"
else
    log_info "其他CPU架构: $OS_ARCH (build.md 验证过 x86_64/aarch64)"
fi

if [ -f /etc/os-release ]; then
    DISTRO=$(grep "^ID=" /etc/os-release | cut -d'=' -f2 | tr -d '"')
    DISTRO_VER=$(grep "^VERSION_ID=" /etc/os-release | cut -d'=' -f2 | tr -d '"')
    log_info "发行版: $DISTRO $DISTRO_VER"
fi

# 根据发行版确定包管理器 (用于安装提示)，自动适配 Ubuntu/Debian/CentOS/EulerOS/openEuler
# 说明: openEuler 优先使用 dnf，EulerOS 2.x 及 CentOS/RHEL 使用 yum
case "$DISTRO" in
    ubuntu|debian)
        PKG_MGR="apt-get"
        ;;
    centos|rhel|fedora)
        PKG_MGR="yum"
        ;;
    euleros|openeuler)
        # openEuler 默认提供 dnf，EulerOS 2.x 仅提供 yum
        if check_command dnf; then
            PKG_MGR="dnf"
        else
            PKG_MGR="yum"
        fi
        ;;
    *)
        PKG_MGR="yum"
        log_info "未识别的发行版, 安装提示将同时给出 apt/yum 两种方案"
        ;;
esac
if [ -n "$DISTRO" ]; then
    log_info "包管理器: $PKG_MGR"
fi

# ==================== 2. GCC ====================
# build.md: GCC >= 7.3.x
print_header "2. GCC/G++ [build.md: >= $REQUIRED_GCC_MIN]"

if check_command gcc; then
    GCC_RAW=$(gcc -dumpversion)
    GCC_VER=$(extract_version "$GCC_RAW")
    log_info "gcc -dumpversion: $GCC_RAW → $GCC_VER"

    if version_ge "$GCC_VER" "$REQUIRED_GCC_MIN"; then
        log_pass "GCC $GCC_VER (>= $REQUIRED_GCC_MIN)"
    else
        log_error "GCC 版本过低: $GCC_VER (build.md 要求 >= $REQUIRED_GCC_MIN)"
        pkg_install_hint "build-essential" "gcc gcc-c++"
        # CentOS 7 默认 GCC 4.8 过旧, 需借助 devtoolset 安装高版本
        case "$DISTRO" in
            centos|rhel) log_info "  (CentOS 7 可用 SCL: sudo yum install devtoolset-7-gcc devtoolset-7-gcc-c++)" ;;
        esac
    fi
else
    log_error "未安装 GCC (build.md 要求 >= $REQUIRED_GCC_MIN)"
    pkg_install_hint "build-essential" "gcc gcc-c++"
fi

if check_command g++; then
    GPP_RAW=$(g++ -dumpversion)
    GPP_VER=$(extract_version "$GPP_RAW")
    log_pass "G++ $GPP_VER"
else
    log_error "未安装 G++"
    pkg_install_hint "g++" "gcc-c++"
fi

# ==================== 3. Python3 ====================
# build.md: Python3 >= 3.9.x
# build.md: 还需要额外安装 coverage, 并将 Python3 的 bin 路径添加到 PATH
print_header "3. Python3 [build.md: >= $REQUIRED_PYTHON_MIN]"

PYTHON_CMD=""
if check_command python3; then
    PYTHON_CMD="python3"
    PYTHON_RAW=$($PYTHON_CMD --version 2>&1)
    PYTHON_VER=$(extract_version "$PYTHON_RAW")
elif check_command python; then
    PYTHON_CMD="python"
    PYTHON_RAW=$($PYTHON_CMD --version 2>&1)
    PYTHON_VER=$(extract_version "$PYTHON_RAW")
fi

if [ -n "$PYTHON_CMD" ]; then
    log_info "$PYTHON_CMD --version: $PYTHON_RAW → $PYTHON_VER"

    if version_ge "$PYTHON_VER" "$REQUIRED_PYTHON_MIN"; then
        log_pass "Python $PYTHON_VER (>= $REQUIRED_PYTHON_MIN)"
    else
        log_error "Python 版本过低: $PYTHON_VER (build.md 要求 >= $REQUIRED_PYTHON_MIN)"
        log_info "请安装 Python 3.9 或更高版本"
    fi

    # Python 开发头文件
    PYTHON_INCLUDE=$($PYTHON_CMD -c "import sysconfig; print(sysconfig.get_path('include'))" 2>/dev/null || true)
    if [ -n "$PYTHON_INCLUDE" ] && [ -f "$PYTHON_INCLUDE/Python.h" ]; then
        log_pass "Python 开发头文件: $PYTHON_INCLUDE/Python.h"
    else
        log_error "缺少 Python.h"
        pkg_install_hint "python3-dev" "python3-devel"
    fi

    # build.md: 将 Python3 的 bin 路径添加到 PATH
    PYTHON_BIN_DIR=$($PYTHON_CMD -c "import sys; print(sys.prefix + '/bin')" 2>/dev/null || true)
    if [ -n "$PYTHON_BIN_DIR" ] && echo "$PATH" | grep -q "$PYTHON_BIN_DIR" 2>/dev/null; then
        log_pass "Python bin 路径已在 PATH 中"
    else
        log_warn "Python bin 路径可能未在 PATH 中"
        log_info "build.md: export PATH=\$PATH:\$PYTHON3_HOME/bin"
    fi

    # pip
    if $PYTHON_CMD -m pip --version &>/dev/null; then
        PIP_RAW=$($PYTHON_CMD -m pip --version)
        PIP_VER=$(extract_version "$PIP_RAW")
        log_pass "pip $PIP_VER"
    else
        log_error "pip 未安装 (安装 requirements.txt 需要)"
        log_info "安装: $PYTHON_CMD -m ensurepip --upgrade"
    fi
else
    log_error "未安装 Python3 (build.md 要求 >= $REQUIRED_PYTHON_MIN)"
    pkg_install_hint "python3 python3-dev python3-pip" "python3 python3-devel python3-pip"
fi

# ==================== 4. CMake ====================
# build.md: CMake >= 3.16.0 (建议使用 3.20.0 版本)
print_header "4. CMake [build.md: >= $REQUIRED_CMAKE_MIN, 建议 $REQUIRED_CMAKE_RECOMMEND]"

if check_command cmake; then
    CMAKE_RAW=$(cmake --version | head -n1)
    CMAKE_VER=$(extract_version "$CMAKE_RAW")
    log_info "cmake --version: $CMAKE_RAW → $CMAKE_VER"

    if version_ge "$CMAKE_VER" "$REQUIRED_CMAKE_RECOMMEND"; then
        log_pass "CMake $CMAKE_VER (>= 建议版本 $REQUIRED_CMAKE_RECOMMEND)"
    elif version_ge "$CMAKE_VER" "$REQUIRED_CMAKE_MIN"; then
        log_pass "CMake $CMAKE_VER (>= $REQUIRED_CMAKE_MIN)"
        log_warn "build.md 建议使用 CMake $REQUIRED_CMAKE_RECOMMEND, 当前 $CMAKE_VER"
    else
        log_error "CMake 版本过低: $CMAKE_VER (build.md 要求 >= $REQUIRED_CMAKE_MIN)"
        log_info "升级 (pip):  pip install cmake --upgrade"
        log_info "升级 (源码): https://cmake.org/download/"
    fi
else
    log_error "未安装 CMake (build.md 要求 >= $REQUIRED_CMAKE_MIN)"
    pkg_install_hint "cmake" "cmake"
    log_info "安装 (pip):    pip install cmake"
fi

# ==================== 5. bash ====================
# build.md: bash >= 5.1.16
print_header "5. bash [build.md: >= $REQUIRED_BASH_MIN]"

if check_command bash; then
    BASH_RAW=$(bash --version | head -n1)
    BASH_VER=$(extract_version "$BASH_RAW")
    log_info "bash --version: $BASH_RAW → $BASH_VER"

    if version_ge "$BASH_VER" "$REQUIRED_BASH_MIN"; then
        log_pass "bash $BASH_VER (>= $REQUIRED_BASH_MIN)"
    else
        log_error "bash 版本过低: $BASH_VER (build.md 要求 >= $REQUIRED_BASH_MIN)"
        log_info "升级 bash 通常需要从源码编译或升级操作系统"
        log_info "源码: https://ftp.gnu.org/gnu/bash/"
    fi
else
    log_error "未安装 bash"
fi

# ==================== 6. 系统工具 (build.md 列表) ====================
# build.md: ccache/asan/autoconf/automake/libtool/gperf/lcov/libasan/patch/graph-easy
print_header "6. 系统工具 [build.md: 必需]"

# --- ccache ---
if check_command ccache; then
    CCACHE_RAW=$(ccache --version | head -n1)
    CCACHE_VER=$(extract_version "$CCACHE_RAW")
    log_pass "ccache $CCACHE_VER"
else
    log_error "未安装 ccache (build.md 要求)"
    pkg_install_hint "ccache" "ccache"
fi

# --- autoconf ---
if check_command autoconf; then
    AUTOCONF_RAW=$(autoconf --version | head -n1)
    AUTOCONF_VER=$(extract_version "$AUTOCONF_RAW")
    log_pass "autoconf $AUTOCONF_VER"
else
    log_error "未安装 autoconf (build.md 要求)"
    pkg_install_hint "autoconf" "autoconf"
fi

# --- automake ---
if check_command automake; then
    AUTOMAKE_RAW=$(automake --version | head -n1)
    AUTOMAKE_VER=$(extract_version "$AUTOMAKE_RAW")
    log_pass "automake $AUTOMAKE_VER"
else
    log_error "未安装 automake (build.md 要求)"
    pkg_install_hint "automake" "automake"
fi

# --- libtool ---
# Ubuntu/Debian 中 libtool 包安装后，可执行文件可能叫 libtoolize 而不是 libtool，或需要额外安装 libtool-bin
if check_command libtool || check_command libtoolize; then
    if command -v libtool &> /dev/null; then
        LIBTOOL_RAW=$(libtool --version | head -n1)
    else
        LIBTOOL_RAW=$(libtoolize --version | head -n1)
    fi
    LIBTOOL_VER=$(extract_version "$LIBTOOL_RAW")
    log_pass "libtool $LIBTOOL_VER"
else
    log_error "未安装 libtool (build.md 要求)"
    pkg_install_hint "libtool libtool-bin" "libtool"
fi

# --- gperf ---
if check_command gperf; then
    GPERF_RAW=$(gperf --version | head -n1)
    GPERF_VER=$(extract_version "$GPERF_RAW")
    log_pass "gperf $GPERF_VER"
else
    log_error "未安装 gperf (build.md 要求)"
    pkg_install_hint "gperf" "gperf"
fi

# --- lcov (非必需, 仅用于代码覆盖率统计) ---
if check_command lcov; then
    LCOV_RAW=$(lcov --version 2>&1 | head -n1)
    LCOV_VER=$(extract_version "$LCOV_RAW")
    log_pass "lcov $LCOV_VER"
else
    log_warn "未安装 lcov (非必需, 仅用于代码覆盖率统计, 不影响编译)"
    pkg_install_hint "lcov" "lcov"
fi

# --- patch ---
if check_command patch; then
    log_pass "patch 已安装"
else
    log_error "未安装 patch (build.md 要求)"
    pkg_install_hint "patch" "patch"
fi

# --- libasan ---
# build.md: "asan以gcc 7.5.0版本为例安装的是libasan4，其他版本请安装对应版本asan"
ASAN_FOUND=false
for path in /usr/lib /usr/lib64 /usr/local/lib /usr/local/lib64 \
            /usr/lib/x86_64-linux-gnu /usr/lib/aarch64-linux-gnu \
            /usr/lib/gcc/x86_64-linux-gnu/*/  /usr/lib/gcc/aarch64-linux-gnu/*/ \
            /usr/lib/gcc/x86_64-linux-gnu/*/*/  /usr/lib/gcc/aarch64-linux-gnu/*/*/; do
    if ls "$path"/libasan.so* &>/dev/null 2>&1; then
        ASAN_FOUND=true
        ASAN_PATH="$path"
        break
    fi
done

if $ASAN_FOUND; then
    log_pass "libasan 已安装 ($ASAN_PATH)"
else
    log_error "未安装 libasan (build.md 要求)"
    log_info "build.md: 'asan以gcc 7.5.0版本为例安装的是libasan4，其他版本请安装对应版本asan'"

    # 提取当前 GCC 主版本号 (用于 apt 系精确推荐 libasan 版本)
    if check_command gcc; then
        GCC_MAJOR=$(gcc -dumpversion | cut -d'.' -f1)
    else
        GCC_MAJOR=""
    fi

    # 根据发行版给出安装建议:
    #   apt 系 (Ubuntu/Debian): libasan 包名随 GCC 主版本变化 (libasan4/5/6/8)
    #   yum/dnf 系 (CentOS/EulerOS/openEuler): 统一为 libasan, 由包管理器匹配版本
    case "$DISTRO" in
        ubuntu|debian)
            case "$GCC_MAJOR" in
                7)       apt_asan="libasan4" ;;
                8|9)     apt_asan="libasan5" ;;
                10|11)   apt_asan="libasan6" ;;
                12|13)   apt_asan="libasan8" ;;
                *)       apt_asan="libasan" ;;
            esac
            log_info ">>> 当前 GCC ${GCC_MAJOR:-未知}, 建议: sudo apt-get install $apt_asan"
            ;;
        centos|rhel|fedora|euleros|openeuler)
            log_info ">>> 当前 GCC ${GCC_MAJOR:-未知}, 建议: sudo $PKG_MGR install libasan"
            ;;
        *)
            # 未识别发行版: 同时给出两种方案
            log_info "安装 (Ubuntu/Debian):   sudo apt-get install libasan4 (随 GCC 版本: 4/5/6/8)"
            log_info "安装 (CentOS/EulerOS): sudo yum install libasan"
            ;;
    esac
fi

# --- graph-easy (可选) ---
if check_command graph-easy; then
    log_pass "graph-easy 已安装"
else
    log_info "graph-easy 未安装 (build.md 标记为可选, 不影响编译)"
    pkg_install_hint "libgraph-easy-perl" "perl-Graph-Easy"
fi

# ==================== 7. CANN Toolkit ====================
# build.md 步骤一: 安装社区版 CANN Toolkit 包
print_header "7. CANN Toolkit [build.md: 步骤一, 必需]"

ASCEND_HOME=""
if [ -n "$ASCEND_HOME_PATH" ]; then
    ASCEND_HOME="$ASCEND_HOME_PATH"
elif [ -d "/usr/local/Ascend/ascend-toolkit/latest" ]; then
    ASCEND_HOME="/usr/local/Ascend/ascend-toolkit/latest"
elif [ -d "$HOME/Ascend/ascend-toolkit/latest" ]; then
    ASCEND_HOME="$HOME/Ascend/ascend-toolkit/latest"
elif [ -d "/usr/local/Ascend/latest" ]; then
    ASCEND_HOME="/usr/local/Ascend/latest"
fi

if [ -n "$ASCEND_HOME" ] && [ -d "$ASCEND_HOME" ]; then
    log_pass "CANN Toolkit 路径: $ASCEND_HOME"

    if [ -f "$ASCEND_HOME/version.cfg" ]; then
        CANN_VER=$(head -n1 "$ASCEND_HOME/version.cfg")
        log_info "CANN 版本: $CANN_VER"
    fi

    # 环境变量 (build.md 步骤三: source set_env.sh)
    SET_ENV_FOUND=false
    for env_script in \
        "/usr/local/Ascend/cann/set_env.sh" \
        "$HOME/Ascend/cann/set_env.sh" \
        "$ASCEND_HOME/bin/setenv.bash" \
        "$ASCEND_HOME/../set_env.sh"; do
        if [ -f "$env_script" ]; then
            SET_ENV_FOUND=true
            log_pass "set_env.sh: $env_script"
            break
        fi
    done

    if ! $SET_ENV_FOUND; then
        log_warn "未找到 set_env.sh"
        log_info "build.md: source /usr/local/Ascend/cann/set_env.sh"
    fi

    if [ -n "$ASCEND_TOOLKIT_HOME" ]; then
        log_pass "ASCEND_TOOLKIT_HOME: $ASCEND_TOOLKIT_HOME"
    else
        log_warn "ASCEND_TOOLKIT_HOME 未设置 (请先 source set_env.sh)"
    fi

    if echo "$LD_LIBRARY_PATH" | grep -q "Ascend" 2>/dev/null; then
        log_pass "LD_LIBRARY_PATH 包含 Ascend"
    else
        log_warn "LD_LIBRARY_PATH 未包含 Ascend (请先 source set_env.sh)"
    fi
else
    log_error "未检测到 CANN Toolkit"
    log_info "build.md: 安装社区版 CANN Toolkit 包"
    log_info "下载: https://ascend.devcloud.huaweicloud.com/artifactory/cann-run-mirror/software/master/"
    log_info "安装: bash Ascend-cann-toolkit_xxx_linux-xxx.run --full --quiet"
    log_info "环境: source /usr/local/Ascend/cann/set_env.sh"
fi

# ==================== 8. git 和 make ====================
print_header "8. git / make [build.md: 步骤三需要 git clone]"

if check_command git; then
    GIT_RAW=$(git --version)
    GIT_VER=$(extract_version "$GIT_RAW")
    log_pass "git $GIT_VER"
else
    log_error "未安装 git (build.md 步骤三需要 git clone)"
    pkg_install_hint "git" "git"
fi

if check_command make; then
    MAKE_RAW=$(make --version | head -n1)
    MAKE_VER=$(extract_version "$MAKE_RAW")
    log_pass "make $MAKE_VER"
else
    log_error "未安装 make"
    pkg_install_hint "make" "make"
fi

# ==================== 9. Python 包 (requirements.txt) ====================
# requirements.txt:
#   numpy
#   pybind11>=2.13.6,<3.0.0
#   jinja2
#   setuptools>=59.0.1
#   wheel>=0.37.1
#   coverage          (build.md 也单独提及: pip3 install coverage)
#   cloudpickle
print_header "9. Python 包 [requirements.txt]"

if [ -n "$PYTHON_CMD" ]; then

    # 通用检查函数 (无版本要求)
    check_py_pkg() {
        local pkg_name=$1
        local import_name=$2

        if $PYTHON_CMD -c "import $import_name" &>/dev/null; then
            local ver
            ver=$($PYTHON_CMD -c "import $import_name; print(getattr($import_name, '__version__', 'ok'))" 2>/dev/null || echo "ok")
            log_pass "$pkg_name ($ver) [requirements.txt]"
        else
            log_error "缺少: $pkg_name [requirements.txt]"
            log_info "安装: pip install $pkg_name"
        fi
    }

    # 带版本范围检查的函数
    check_py_pkg_ver() {
        local pkg_name=$1
        local import_name=$2
        local min_ver=$3
        local max_ver=$4  # 可选, 为空则不检查上限

        if $PYTHON_CMD -c "import $import_name" &>/dev/null; then
            local ver
            ver=$($PYTHON_CMD -c "import $import_name; print($import_name.__version__)" 2>/dev/null || echo "0.0.0")
            local parsed_ver
            parsed_ver=$(extract_version "$ver")

            local ok=true

            if [ -n "$min_ver" ] && ! version_ge "$parsed_ver" "$min_ver"; then
                ok=false
            fi

            if [ -n "$max_ver" ] && ! version_lt "$parsed_ver" "$max_ver"; then
                ok=false
            fi

            if $ok; then
                if [ -n "$max_ver" ]; then
                    log_pass "$pkg_name $ver (>= $min_ver, < $max_ver) [requirements.txt]"
                else
                    log_pass "$pkg_name $ver (>= $min_ver) [requirements.txt]"
                fi
            else
                if [ -n "$max_ver" ]; then
                    log_error "$pkg_name $ver 不满足版本要求 (>= $min_ver, < $max_ver)"
                else
                    log_error "$pkg_name $ver 不满足版本要求 (>= $min_ver)"
                fi
                log_info "安装: pip install '$pkg_name>=$min_ver${max_ver:+,<$max_ver}'"
            fi
        else
            log_error "缺少: $pkg_name [requirements.txt]"
            log_info "安装: pip install '$pkg_name>=$min_ver${max_ver:+,<$max_ver}'"
        fi
    }

    # --- 有版本要求的包 ---
    check_py_pkg_ver "pybind11"   "pybind11"   "$REQUIRED_PYBIND11_MIN"   "3.0.0"
    check_py_pkg_ver "setuptools" "setuptools"  "$REQUIRED_SETUPTOOLS_MIN" ""
    check_py_pkg_ver "wheel"      "wheel"       "$REQUIRED_WHEEL_MIN"      ""

    # --- 无版本要求的包 ---
    check_py_pkg "numpy"       "numpy"
    check_py_pkg "jinja2"      "jinja2"
    check_py_pkg "coverage"    "coverage"      # build.md 也单独提及
    check_py_pkg "cloudpickle" "cloudpickle"

    # --- 快捷安装提示 ---
    log_info "一键安装: pip3 install -r requirements.txt"

else
    log_error "Python 未安装, 无法检查 requirements.txt 依赖"
fi

# ==================== 10. 系统资源 ====================
print_header "10. 系统资源 [可选]"

# 磁盘
DISK_AVAIL=$(df -B1 . 2>/dev/null | tail -1 | awk '{printf "%.0f", $4/1024/1024/1024}' || echo "0")
if [ "$DISK_AVAIL" -gt 30 ] 2>/dev/null; then
    log_info "可用磁盘: ${DISK_AVAIL}GB"
elif [ "$DISK_AVAIL" -gt 15 ] 2>/dev/null; then
    log_info "磁盘偏少: ${DISK_AVAIL}GB (ge 编译建议 >30GB)"
else
    log_info "磁盘不足: ${DISK_AVAIL}GB (至少 15GB)"
fi

# 内存
if [ -f /proc/meminfo ]; then
    MEM_KB=$(grep MemTotal /proc/meminfo | awk '{print $2}')
    MEM_GB=$((MEM_KB / 1024 / 1024))
    if [ "$MEM_GB" -ge 16 ]; then
        log_info "内存: ${MEM_GB}GB"
    elif [ "$MEM_GB" -ge 8 ]; then
        log_info "内存偏少: ${MEM_GB}GB (ge 编译推荐 >=16GB)"
    else
        log_info "内存不足: ${MEM_GB}GB"
    fi
fi

# CPU
CPU_CORES=$(nproc 2>/dev/null || echo "unknown")
log_info "CPU 核心数: $CPU_CORES"

# 根据当前 GCC 主版本计算 apt 系对应的 libasan 包名 (用于结论段安装命令)
# 说明: libasan 包名随 GCC 主版本变化 (libasan4/5/6/8), yum/dnf 系统一为 libasan
APT_ASAN_PKG="libasan4"
if check_command gcc; then
    GCC_MAJOR_FOR_ASAN=$(gcc -dumpversion | cut -d'.' -f1)
    case "$GCC_MAJOR_FOR_ASAN" in
        7)       APT_ASAN_PKG="libasan4" ;;
        8|9)     APT_ASAN_PKG="libasan5" ;;
        10|11)   APT_ASAN_PKG="libasan6" ;;
        12|13)   APT_ASAN_PKG="libasan8" ;;
        *)       APT_ASAN_PKG="libasan" ;;
    esac
fi

# ==============================================================================
#                            汇总报告
# ==============================================================================
echo ""
echo "=================================================================="
echo "  ge (Graph Engine) 环境检查完成"
echo "  依据: docs/zh/build.md + requirements.txt"
echo "=================================================================="
echo ""

[ $ERROR_COUNT -gt 0 ]   && echo -e "  ${RED}✗ ERRORS:   $ERROR_COUNT${NC}  (必须修复才能编译)"
[ $WARNING_COUNT -gt 0 ] && echo -e "  ${YELLOW}⚠ WARNINGS: $WARNING_COUNT${NC}  (建议修复)"
echo -e "  ${GREEN}✓ PASSED:   $PASS_COUNT${NC}"
echo ""

if [ $ERROR_COUNT -gt 0 ]; then
    echo -e "  ${RED}结论: 环境不满足 ge 编译要求，请修复上述 ERROR 项。${NC}"
    echo ""
    # 根据发行版给出对应的依赖安装命令 (build.md 仅提供 Ubuntu/Debian 示例)
    case "$DISTRO" in
        ubuntu|debian)
            echo "  build.md 安装命令 (Ubuntu/Debian, libasan 按 GCC 版本适配):"
            echo "    sudo apt-get install cmake ccache bash lcov $APT_ASAN_PKG \\"
            echo "      autoconf automake libtool gperf libgraph-easy-perl patch"
            echo ""
            echo "  完整依赖一键安装:"
            echo "    sudo apt-get update && sudo apt-get install -y \\"
            echo "      build-essential cmake ccache lcov patch gperf \\"
            echo "      autoconf automake libtool $APT_ASAN_PKG \\"
            echo "      python3 python3-dev python3-pip \\"
            echo "      git make"
            echo "    pip3 install -r requirements.txt"
            ;;
        centos|rhel|fedora|euleros|openeuler)
            echo "  完整依赖一键安装 (CentOS/EulerOS/openEuler):"
            echo "    sudo $PKG_MGR install -y \\"
            echo "      gcc gcc-c++ make cmake ccache bash lcov libasan \\"
            echo "      autoconf automake libtool gperf patch \\"
            echo "      python3 python3-devel python3-pip git"
            echo "    pip3 install -r requirements.txt"
            ;;
        *)
            echo "  build.md 安装命令 (Ubuntu/Debian, libasan 按 GCC 版本适配):"
            echo "    sudo apt-get install cmake ccache bash lcov $APT_ASAN_PKG \\"
            echo "      autoconf automake libtool gperf libgraph-easy-perl patch"
            echo ""
            echo "  完整依赖一键安装 (CentOS/EulerOS/openEuler):"
            echo "    sudo yum install -y gcc gcc-c++ make cmake ccache bash lcov libasan \\"
            echo "      autoconf automake libtool gperf patch \\"
            echo "      python3 python3-devel python3-pip git"
            echo "    pip3 install -r requirements.txt"
            ;;
    esac
    echo ""
    echo "  CANN Toolkit:"
    echo "    下载: https://ascend.devcloud.huaweicloud.com/artifactory/cann-run-mirror/software/master/"
    echo "    安装: bash Ascend-cann-toolkit_xxx_linux-xxx.run --full --quiet"
    echo "    环境: source /usr/local/Ascend/cann/set_env.sh"
    echo ""
    echo "  详细说明: docs/zh/build.md"
    echo ""
    exit 1
elif [ $WARNING_COUNT -gt 0 ]; then
    echo -e "  ${YELLOW}结论: 环境基本可用，建议关注上述 WARNING 项。${NC}"
    exit 0
else
    echo -e "  ${GREEN}结论: 环境检查全部通过！可以编译 ge。${NC}"
    echo -e "  ${GREEN}编译命令: bash build.sh${NC}"
    exit 0
fi