pipeline {
    agent any

    parameters {
        choice(
            name: 'DEPLOY_ENV',
            choices: ['test', 'uat', 'prod'],
            description: '选择部署环境'
        )
    }

    triggers {
        cron('0 10 * * *')
    }

    environment {
        PYTHON_VERSION = '3.8'
        PROJECT_NAME = 'api_test'
        DEPLOY_ENV = "${params.DEPLOY_ENV ?: 'test'}"
        IP_HOST_TEST = 'ai-test.hikoon.com'
        IP_HOST_UAT = 'ai-uat.hikoon.com'
        IP_HOST_PROD = 'api.hikoon.com'
    }

    stages {
        stage('检出代码') {
            steps {
                echo "开始检出代码..."
                checkout scm
            }
        }

        stage('环境准备') {
            steps {
                echo "准备Python环境..."
                sh '''
                    python3 --version
                    if [ ! -d "venv" ]; then
                        python3 -m venv venv
                    fi
                    . venv/bin/activate
                    pip install --upgrade pip -q
                    pip install -r requirements.txt -q
                    echo "依赖安装完成"
                '''
            }
        }

        stage('执行测试') {
            steps {
                echo "开始执行API自动化测试 (环境: ${DEPLOY_ENV})..."
                script {
                    def ipHost = 'ai-test.hikoon.com'
                    if (env.DEPLOY_ENV == 'uat') {
                        ipHost = env.IP_HOST_UAT
                    } else if (env.DEPLOY_ENV == 'prod') {
                        ipHost = env.IP_HOST_PROD
                    } else {
                        ipHost = env.IP_HOST_TEST
                    }
                    env.IP_HOST = ipHost
                }
                sh '''
                    . venv/bin/activate
                    export DEPLOY_ENV="${DEPLOY_ENV}"
                    export IP_HOST="${IP_HOST}"
                    echo "当前执行环境: ${DEPLOY_ENV}"
                    echo "IP_HOST: ${IP_HOST}"
                    python3 api_test.py
                '''
            }
        }

        stage('生成报告') {
            steps {
                echo "正在生成测试报告..."
                sh '''
                    ls -la api_test_report_* 2>/dev/null || echo "未找到报告文件"
                '''
            }
        }
    }

    post {
        always {
            echo "清理工作目录..."
            sh 'echo "测试执行完成"'
        }

        success {
            echo "✅ [${DEPLOY_ENV}环境] 所有测试用例运行成功！"
        }

        failure {
            echo "❌ [${DEPLOY_ENV}环境] 有测试用例执行失败！"
        }
    }
}