#!/bin/bash

# 设置正确的PATH环境变量
export PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:$PATH"

# 检查Java是否已安装
check_java() {
    if command -v java &> /dev/null && [ -n "$JAVA_HOME" ]; then
        echo "Java已安装，版本信息："
        java -version
        return 0
    else
        return 1
    fi
}

# 检查Gradle是否已安装
check_gradle() {
    if command -v gradle &> /dev/null; then
        echo "Gradle已安装，版本信息："
        gradle -version
        return 0
    else
        return 1
    fi
}

# 安装Java
install_java() {
    echo "开始安装Java..."
    
    # 更新包列表
    apt-get update

    # 安装必要的系统工具
    apt-get install -y wget curl gnupg software-properties-common dpkg

    # 创建必要的目录
    mkdir -p /usr/local/sbin /usr/sbin /sbin

    # 创建JDK安装目录
    mkdir -p /usr/lib/jvm

    # 下载并安装 JDK 21
    cd /tmp
    wget https://download.oracle.com/java/21/latest/jdk-21_linux-x64_bin.deb

    # 确保dpkg可以正常工作
    ldconfig

    # 安装JDK
    dpkg -i jdk-21_linux-x64_bin.deb
    rm jdk-21_linux-x64_bin.deb

    # 获取实际的JDK安装目录
    JDK_REAL_PATH=$(find /usr/lib/jvm -maxdepth 1 -type d -name "jdk-21*" | head -n 1)

    if [ -z "$JDK_REAL_PATH" ]; then
        echo "错误：未找到JDK安装目录，请检查安装是否成功"
        exit 1
    fi

    # 创建软链接
    ln -sf "$JDK_REAL_PATH" /usr/lib/jvm/jdk-21

    # 设置 JAVA_HOME 环境变量
    cat > /etc/profile.d/jdk.sh << 'EOF'
export JAVA_HOME=/usr/lib/jvm/jdk-21
export PATH=$JAVA_HOME/bin:$PATH
EOF

    # 立即生效环境变量
    source /etc/profile.d/jdk.sh
    
    echo "Java安装完成！"
}

# 安装Gradle
install_gradle() {
    echo "开始安装Gradle..."
    
    # 下载并安装 Gradle
    GRADLE_VERSION="8.5"
    cd /tmp
    wget -O gradle-${GRADLE_VERSION}-bin.zip https://mirrors.cloud.tencent.com/gradle/gradle-${GRADLE_VERSION}-bin.zip

    unzip gradle-${GRADLE_VERSION}-bin.zip -d /opt
    rm gradle-${GRADLE_VERSION}-bin.zip

    # 设置 Gradle 环境变量
    cat > /etc/profile.d/gradle.sh << EOF
export GRADLE_HOME=/opt/gradle-${GRADLE_VERSION}
export PATH=\$GRADLE_HOME/bin:\$PATH
EOF

    # 立即生效环境变量
    source /etc/profile.d/gradle.sh
    
    echo "Gradle安装完成！"
}

# 主程序开始
echo "检查Java安装状态..."
if ! check_java; then
    install_java
else
    echo "Java已安装，跳过安装步骤"
fi

echo "检查Gradle安装状态..."
if ! check_gradle; then
    install_gradle
else
    echo "Gradle已安装，跳过安装步骤"
fi

# 验证安装
echo "验证系统PATH："
echo $PATH
echo ""
echo "验证Java安装："
which java
java -version
echo ""
echo "验证JAVA_HOME设置："
echo $JAVA_HOME
echo ""
echo "验证Gradle安装："
which gradle
gradle -version