#!/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

channel=${1}
version=${2}
postgresql_path="${setup_path}/server/postgresql"
j=$(calculate_j)

# 安装依赖
if [ ${OS} == "rhel" ]; then
    dnf makecache -y
    dnf groupinstall "Development Tools" -y
    dnf install pkg-config make meson ninja-build bison flex gettext zlib-devel readline-devel lz4-devel libzstd-devel liburing-devel libcurl-devel libxml2-devel libxslt-devel openssl-devel libicu-devel systemd-devel -y
    # RHEL 10+ 默认禁用 io_uring，需要手动开启，https://issues.redhat.com/browse/RHEL-65347
    echo 0 >/proc/sys/kernel/io_uring_disabled
elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
    apt-get update
    # Debian 13+ 及 Ubuntu 25+ 需要单独安装 systemd-dev，旧版系统不能安装否则会出现严重问题
    if { [ ${OS} == "debian" ] && [ ${VERSION} -ge 13 ]; } || { [ ${OS} == "ubuntu" ] && [ ${VERSION} -ge 25 ]; }; then
        apt-get install systemd-dev -y
    fi
    apt-get install build-essential pkg-config make meson ninja-build bison flex gettext zlib1g-dev libreadline-dev liblz4-dev libzstd-dev liburing-dev libcurl4-openssl-dev libxml2-dev libxslt-dev libssl-dev libicu-dev libsystemd-dev -y
else
    error "Unsupported operating system"
fi
if [ "$?" != "0" ]; then
    error "Failed to install dependencies"
fi

user_check=$(cat /etc/passwd | grep postgres)
if [ "${user_check}" == "" ]; then
    groupadd postgres
    useradd -g postgres postgres
fi

# 准备目录
rm -rf ${postgresql_path}
mkdir -p ${postgresql_path}
cd ${postgresql_path}

# 下载源码
dl "${postgresql_path}" "/postgresql/postgresql-${version}.7z"

7z x postgresql-${version}.7z
rm -f postgresql-${version}.7z
mv postgresql-${version} src

# 编译
cd src
meson setup build --buildtype=release --prefix=${postgresql_path} -Dnls=enabled -Dlz4=enabled -Dzstd=enabled -Dssl=openssl -Dsystemd=enabled -Dlibcurl=enabled -Dliburing=enabled -Dlibxml=enabled -Dlibxslt=enabled
if [ "$?" != "0" ]; then
    rm -rf ${postgresql_path}
    error "Compilation initialization failed"
fi
cd build
ninja
if [ "$?" != "0" ]; then
    rm -rf ${postgresql_path}
    error "Compilation failed"
fi
ninja install
if [ "$?" != "0" ]; then
    rm -rf ${postgresql_path}
    error "Installation failed"
fi

cd ${postgresql_path}
rm -rf ${postgresql_path}/src

# 配置
mkdir -p ${postgresql_path}/data
mkdir -p ${postgresql_path}/logs
chown -R postgres:postgres ${postgresql_path}
chmod -R 700 ${postgresql_path}

# 软链接
ln -sf ${postgresql_path}/bin/* /usr/local/bin/

mkdir -p /home/postgres
cd /home/postgres
if [ -f /home/postgres/.bash_profile ]; then
    echo "export PGHOME=${postgresql_path}" >>/home/postgres/.bash_profile
    echo "export PGDATA=${postgresql_path}/data" >>/home/postgres/.bash_profile
    echo "export PATH=${postgresql_path}/bin:\$PATH " >>/home/postgres/.bash_profile
    echo "MANPATH=$PGHOME/share/man:$MANPATH" >>/home/postgres/.bash_profile
    echo "LD_LIBRARY_PATH=$PGHOME/lib:$LD_LIBRARY_PATH" >>/home/postgres/.bash_profile
    source /home/postgres/.bash_profile
fi
if [ -f /home/postgres/.profile ]; then
    echo "export PGHOME=${postgresql_path}" >>/home/postgres/.profile
    echo "export PGDATA=${postgresql_path}/data" >>/home/postgres/.profile
    echo "export PATH=${postgresql_path}/bin:\$PATH " >>/home/postgres/.profile
    echo "MANPATH=$PGHOME/share/man:$MANPATH" >>/home/postgres/.profile
    echo "LD_LIBRARY_PATH=$PGHOME/lib:$LD_LIBRARY_PATH" >>/home/postgres/.profile
    source /home/postgres/.profile
fi

# 初始化
postgres_password=$(cat /dev/urandom | head -n 16 | sha256sum | head -c 16)
echo ${postgres_password} >/home/postgres/.postgres_password
su - postgres -c "${postgresql_path}/bin/initdb -D ${postgresql_path}/data \
  --encoding=UTF8 --locale=C.UTF-8 --builtin-locale=C.UTF-8 --locale-provider=builtin  \
  --auth-local=peer --auth-host=scram-sha-256 \
  --pwfile=/home/postgres/.postgres_password"
if [ "$?" != "0" ]; then
    rm -rf ${postgresql_path}
    error "PostgreSQL initialization failed"
fi
rm -f /home/postgres/.postgres_password

# 配置
cat >>${postgresql_path}/data/postgresql.conf <<EOF
io_method = 'io_uring'

logging_collector = on
log_destination = 'stderr'
log_directory = '${postgresql_path}/logs'
log_filename = 'postgresql-%Y-%m-%d.log'
log_statement = all
log_min_duration_statement = 5000
EOF

# 写入服务
cat >/etc/systemd/system/postgresql.service <<EOF
[Unit]
Description=PostgreSQL database server
Documentation=man:postgres(1)
After=network-online.target
Wants=network-online.target

[Service]
Type=notify
User=postgres
ExecStart=${postgresql_path}/bin/postgres -D ${postgresql_path}/data
ExecReload=/bin/kill -HUP \$MAINPID
KillMode=mixed
KillSignal=SIGINT
TimeoutSec=infinity
LimitNOFILE=500000
Restart=on-failure
RestartSec=5s

[Install]
WantedBy=multi-user.target
EOF

chmod 644 /etc/systemd/system/postgresql.service

# 在 /etc/systemd/logind.conf 设置 RemoveIPC=no，否则会删除 /dev/shm 下的共享内存文件
check=$(cat /etc/systemd/logind.conf | grep '^RemoveIPC=no.*$')
if [ "${check}" == "" ]; then
    echo "RemoveIPC=no" >>/etc/systemd/logind.conf
    systemctl restart systemd-logind
fi

# 启动服务
systemctl daemon-reload
systemctl enable --now postgresql
if [ "$?" != "0" ]; then
    error "Failed to start"
fi

acepanel app write postgresql ${channel} ${version}
acepanel setting write postgres_password ${postgres_password}
acepanel database add-server --type=postgresql --name=local_postgresql --host=127.0.0.1 --port=5432 --username=postgres --password=${postgres_password}

echo -e $HR
echo "Installation successful"
echo -e $HR