#!/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}
frp_path="${setup_path}/server/frp"

if [ ! -d "${frp_path}" ]; then
    mkdir -p ${frp_path}
fi

# 架构判断
if [ ${ARCH} == "x86_64" ]; then
    frpFile="frp_${version}_linux_amd64.7z"
elif [ ${ARCH} == "aarch64" ]; then
    frpFile="frp_${version}_linux_arm64.7z"
else
    error "Unsupported architecture"
fi

# 备份配置
if [ -f "${frp_path}/frps.toml" ]; then
    \cp -f ${frp_path}/frps.toml ${frp_path}/frps.toml.bak
fi
if [ -f "${frp_path}/frpc.toml" ]; then
    \cp -f ${frp_path}/frpc.toml ${frp_path}/frpc.toml.bak
fi

# 下载frp
cd ${frp_path}
dl "${frp_path}" "/frp/${frpFile}"

# 解压frp
cd ${frp_path}
7z x ${frpFile}
chown -R www:www ${frp_path}
chmod -R 700 ${frp_path}

# 还原配置
if [ -f "${frp_path}/frps.toml.bak" ]; then
    \cp -f ${frp_path}/frps.toml.bak ${frp_path}/frps.toml
fi
if [ -f "${frp_path}/frpc.toml.bak" ]; then
    \cp -f ${frp_path}/frpc.toml.bak ${frp_path}/frpc.toml
fi

systemctl restart frps
if [ "$?" != "0" ]; then
    error "Failed to start"
fi
systemctl restart frpc
if [ "$?" != "0" ]; then
    error "Failed to start"
fi

acepanel app write frp ${channel} ${version}

echo -e ${HR}
echo "Upgrade successful"
echo -e ${HR}