#!/bin/bash
export PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin:$PATH

source <(curl -f -s --connect-timeout 10 --retry 3 https://dl.acepanel.net/public.sh)
if [ $? -ne 0 ]; then
    echo "Download public.sh failed, please check the network or try again later."
    exit 1
fi

action=${1}
version=${2}
php_path="${setup_path}/server/php/${version}"
rdkafka_version="6.0.5"

Install() {
    # 检查是否已经安装
    is_install=$(cat ${php_path}/etc/php.ini | grep '^extension=rdkafka')
    if [ "${is_install}" != "" ]; then
        error "PHP-${version} rdkafka already installed"
    fi

    # 安装依赖
    if [ ${OS} == "rhel" ]; then
        dnf install librdkafka-devel -y
    elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
        apt-get install librdkafka-dev -y
    else
        error "Unsupported operating system"
    fi

    cd ${php_path}/src/ext
    rm -rf rdkafka
    rm -rf rdkafka-${rdkafka_version}.tgz
    dl "${php_path}/src/ext" "/php_exts/rdkafka-${rdkafka_version}.tgz"

    tar -xzf rdkafka-${rdkafka_version}.tgz
    mv rdkafka-${rdkafka_version} rdkafka
    rm -f package.xml
    rm -f rdkafka-${rdkafka_version}.tgz
    cd rdkafka
    ${php_path}/bin/phpize
    ./configure --with-php-config=${php_path}/bin/php-config
    make
    if [ "$?" != "0" ]; then
        rm -rf ${php_path}/src/ext/rdkafka
        error "PHP-${version} rdkafka compilation failed"
    fi
    make install
    if [ "$?" != "0" ]; then
        rm -rf ${php_path}/src/ext/rdkafka
        error "PHP-${version} rdkafka installation failed"
    fi

    sed -i '/;panel/a\extension=rdkafka' ${php_path}/etc/php.ini

    # 重载PHP
    systemctl reload php-fpm-${version}.service
    echo -e $HR
    echo "PHP-${version} rdkafka installation successful"
}

Uninstall() {
    # 检查是否已经安装
    is_install=$(cat ${php_path}/etc/php.ini | grep '^extension=rdkafka$')
    if [ "${is_install}" == "" ]; then
        error "PHP-${version} rdkafka not installed"
    fi

    sed -i '/extension=rdkafka/d' ${php_path}/etc/php.ini

    # 重载PHP
    systemctl reload php-fpm-${version}.service
    echo -e $HR
    echo "PHP-${version} rdkafka uninstall successful"
}

if [ "$action" == 'install' ]; then
    Install
fi
if [ "$action" == 'uninstall' ]; then
    Uninstall
fi