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

: '
Copyright (C) 2022 - now  HaoZi Technology Co., Ltd.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published
by the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>.
'

source <(curl -f -s --connect-timeout 10 --retry 3 https://dl.cdn.haozi.net/panel/public.sh)
if [ $? -ne 0 ]; then
    echo "下载 public.sh 失败，请检查网络或稍后重试。"
    echo "Download public.sh failed, please check the network or try again later."
    exit 1
fi

action=${1}
version=${2}
extension=${3} # 扩展名称
php_path="${setup_path}/server/php/${version}"
add_args="" # 附加参数

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

    # 安装依赖
    if [ "${extension}" == "snmp" ]; then
        if [ ${OS} == "rhel" ]; then
            dnf install -y net-snmp-devel
        elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
            apt-get install -y libsnmp-dev
        fi
    fi
    if [ "${extension}" == "ldap" ]; then
        if [ ${OS} == "rhel" ]; then
            dnf install -y openldap-devel
            ln -sf /usr/lib64/libldap* /usr/lib
        elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
            apt-get install -y libldap2-dev
            ln -sf /usr/lib/x86_64-linux-gnu/libldap* /usr/lib
        fi
    fi
    if [ "${extension}" == "imap" ]; then
        if [ ${OS} == "rhel" ]; then
            # RHEL 9 的仓库中没有 libc-client-devel，待考虑
            dnf install -y libc-client-devel
        elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
            apt-get install -y libc-client-dev
        fi
        add_args="--with-imap --with-imap-ssl --with-kerberos"
    fi
    if [ "${extension}" == "enchant" ]; then
        if [ ${OS} == "rhel" ]; then
            dnf install -y enchant-devel
        elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
            apt-get install -y libenchant-2-dev
        fi
    fi
    if [ "${extension}" == "pspell" ]; then
        if [ ${OS} == "rhel" ]; then
            dnf install -y aspell-devel
        elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
            apt-get install -y libpspell-dev
        fi
    fi
    if [ "${extension}" == "gmp" ]; then
        if [ ${OS} == "rhel" ]; then
            dnf install -y gmp-devel
        elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
            apt-get install -y libgmp-dev
        fi
    fi
    if [ "${extension}" == "gettext" ]; then
        if [ ${OS} == "rhel" ]; then
            dnf install -y gettext-devel
        elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
            apt-get install -y libgettextpo-dev
        fi
    fi
    if [ "${extension}" == "bz2" ]; then
        if [ ${OS} == "rhel" ]; then
            dnf install -y bzip2-devel
        elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
            apt-get install -y libbz2-dev
        fi
    fi
    if [ "${extension}" == "zip" ]; then
        if [ ${OS} == "rhel" ]; then
            dnf install -y libzip-devel
        elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
            apt-get install -y libzip-dev
        fi
    fi

    if [ "$?" != "0" ]; then
        error "安装依赖软件失败"
    fi

    if [ "${extension}" == "pdo_pgsql" ]; then
        add_args="--with-pdo-pgsql=${setup_path}/server/postgresql"
    fi
    if [ "${extension}" == "pgsql" ]; then
        add_args="--with-pgsql=${setup_path}/server/postgresql"
    fi

    # 安装扩展
    if [ ! -d ${php_path}/src/ext/${extension} ]; then
        error "PHP-${version} ${extension} 不存在"
    fi
    cd ${php_path}/src/ext/${extension}
    ${php_path}/bin/phpize
    ./configure --with-php-config=${php_path}/bin/php-config ${add_args}
    make
    if [ "$?" != "0" ]; then
        error "PHP-${version} ${extension} 编译失败"
    fi
    make install
    if [ "$?" != "0" ]; then
        error "PHP-${version} ${extension} 安装失败"
    fi

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

    # 重载PHP
    systemctl reload php-fpm-${version}.service
    echo -e $HR
    echo "PHP-${version} ${extension} 安装成功"
}

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

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

    # 重载PHP
    systemctl reload php-fpm-${version}.service
    echo -e $HR
    echo "PHP-${version} ${extension} 卸载成功"
}

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