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

channel=${1}
version=${2}

if [ ${OS} == "rhel" ]; then
    dnf makecache -y
    dnf update docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin -y
elif [ ${OS} == "debian" ] || [ ${OS} == "ubuntu" ]; then
    apt-get update
    apt-get upgrade docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin -y
else
    error "不支持的操作系统"
fi

if [ $? -ne 0 ]; then
    if [ ${OS} == "rhel" ]; then
        dnf update -y docker
        if [ $? -ne 0 ]; then
            error "docker 升级失败"
        fi
        systemctl enable --now docker
    else
        error "不支持的操作系统"
    fi
fi

sleep 3
systemctl restart docker

# 创建编排目录
mkdir -p ${setup_path}/server/compose

panel-cli app write docker ${channel} ${version}

echo -e ${HR}
echo "升级完成"
echo -e ${HR}