#!/bin/bash
# ch6 测试脚本
#
# 用法：
#   ./test.sh          # 运行全部测试（等价于 ./test.sh all）
#   ./test.sh base     # 仅运行基础测试
#   ./test.sh exercise # 仅运行练习测试
#   ./test.sh all      # 运行基础 + 练习测试

set -e

SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
"$SCRIPT_DIR/scripts/ensure-vendor.sh"
cd "$SCRIPT_DIR"

HOST_TARGET="$(rustc -vV | sed -n 's/^host: //p')"

GREEN='\033[0;32m'
RED='\033[0;31m'
YELLOW='\033[0;33m'
NC='\033[0m'

# 检查并安装 tg-rcore-tutorial-checker
ensure_tg_checker() {
    if ! command -v tg-rcore-tutorial-checker &> /dev/null; then
        if [ -f vendor/tg-rcore-tutorial-checker/Cargo.toml ]; then
            echo -e "${YELLOW}tg-rcore-tutorial-checker 未安装，使用本地 vendor 版本...${NC}"
            CHECKER_CMD=(cargo run --quiet --target "$HOST_TARGET" --manifest-path vendor/tg-rcore-tutorial-checker/Cargo.toml --)
            echo -e "${GREEN}✓ 使用本地 tg-rcore-tutorial-checker${NC}"
        else
            echo -e "${YELLOW}tg-rcore-tutorial-checker 未安装，正在安装...${NC}"
            if cargo install tg-rcore-tutorial-checker; then
                CHECKER_CMD=(tg-rcore-tutorial-checker)
                echo -e "${GREEN}✓ tg-rcore-tutorial-checker 安装成功${NC}"
            else
                echo -e "${RED}✗ tg-rcore-tutorial-checker 安装失败${NC}"
                exit 1
            fi
        fi
    else
        CHECKER_CMD=(tg-rcore-tutorial-checker)
    fi
}

ensure_tg_checker

# 使用 pipefail 确保管道中任意命令失败都能被捕获
set -o pipefail

run_base() {
    echo "运行 ch6 基础测试..."
    cargo clean
    export CHAPTER=-6
    export TG_USER_DIR="$PWD/vendor/tg-rcore-tutorial-user"
    echo -e "${YELLOW}────────── cargo run 输出 ──────────${NC}"

    # 使用 tee 将 cargo run 的输出同时显示在终端和传递给 tg-rcore-tutorial-checker
    if cargo run 2>&1 | tee /dev/stderr | "${CHECKER_CMD[@]}" --ch 6; then
        echo ""
        echo -e "${YELLOW}────────── 测试结果 ──────────${NC}"
        echo -e "${GREEN}✓ ch6 基础测试通过${NC}"
        cargo clean
        return 0
    else
        echo ""
        echo -e "${YELLOW}────────── 测试结果 ──────────${NC}"
        echo -e "${RED}✗ ch6 基础测试失败${NC}"
        cargo clean
        return 1
    fi
}

run_exercise() {
    echo "运行 ch6 练习测试..."
    cargo clean
    export CHAPTER=6
    export TG_USER_DIR="$PWD/vendor/tg-rcore-tutorial-user"
    echo -e "${YELLOW}────────── cargo run --features exercise 输出 ──────────${NC}"

    if cargo run --features exercise 2>&1 | tee /dev/stderr | "${CHECKER_CMD[@]}" --ch 6 --exercise; then
        echo ""
        echo -e "${YELLOW}────────── 测试结果 ──────────${NC}"
        echo -e "${GREEN}✓ ch6 练习测试通过${NC}"
        cargo clean
        return 0
    else
        echo ""
        echo -e "${YELLOW}────────── 测试结果 ──────────${NC}"
        echo -e "${RED}✗ ch6 练习测试失败${NC}"
        cargo clean
        return 1
    fi
}

case "${1:-all}" in
    base)
        run_base
        ;;
    exercise)
        run_exercise
        ;;
    all)
        run_base
        echo ""
        run_exercise
        ;;
    *)
        echo "用法: $0 [base|exercise|all]"
        exit 1
        ;;
esac