#!/bin/bash

declare -a system_versions
system_versions=(
    [1]='海纳思NAS社区版固件 Ubuntu-20.04.5-arm64.gz'
    [2]='海纳思宝塔收费专用固件 Ubuntu-bt-arm64.gz'
)

echo "
+----------------------------------------------------------------------
| 海纳思系统固件切换专用脚本温馨提示：
+----------------------------------------------------------------------
| 切换后原系统将完全清除，请注意保存资料。
+----------------------------------------------------------------------
| 如果不确定，请按CTRL+C结束此脚本.
+----------------------------------------------------------------------
"

temp_file=$(mktemp)
curl -sSL https://hi.ecoo.top/mirror-utils.sh > $temp_file
if [ ! -f /etc/public-download.sh ] || [ "$(cat $temp_file|md5sum|awk '{print $1}')" != "$(cat /etc/public-download.sh|md5sum|awk '{print $1}')" ]; then
	cp $temp_file /etc/public-download.sh
fi
rm -f $temp_file
source /etc/public-download.sh

if [ $(whoami) != "root" ];then
	_exit 1 "请使用root权限切换系统命令！"
fi

date_20230818=$(date -d '20230818' +%s)
if [ $(date -d $(cat /etc/nasversion) +%s) -lt ${date_20230818} ]; then
	_exit 1 "请升级到 20230818 或更新的版本后再执行此命令！"
fi

if [ $(df -T|grep /tmp|awk '{print int($5)}') -lt 358400 ]; then
    rm -rf /tmp/*
fi

is64bit=$(getconf LONG_BIT)
if [ "${is64bit}" != '64' ];then
	_exit 1 "抱歉, 当前系统切换版本不支持32位系统, 请使用64位系统!";
fi

for index in "${!system_versions[@]}"; do
    printStr green "$index. $(echo ${system_versions[$index]}|awk '{print $1}')" 
done

while :; do
    printStr yellow "你想要安装哪个版本? 输入数字后按回车确定：" 
    read version_number
    for index in "${!system_versions[@]}"; do
        if [ "$version_number" = "$index" ]; then
            selected_version="$(echo ${system_versions[$index]}|awk '{print $1}')" 
            dlfile="$(echo ${system_versions[$index]}|awk '{print $2}')" 
            break 2
        fi
    done
done

printStr yellow "检测并选定下载节点..."
get_mirror_url

rm -rf /tmp/*
printStr yellow "下载刷机包..."
dl_get update/system/${dlfile} /root/${dlfile}
[ "$(get_md5 -c ${NODE_URL}/update/system/${dlfile}.md5sum ${dlfile})" != "$(get_md5 -l /root/${dlfile})" ] && _exit 1 "文件校验失败，请重试"
printStr green "正在配置系统，请稍后..."
mv /root/${dlfile} /root/backup-64.gz
mkdir -p /tmp/backup
mount /dev/mmcblk0p8 /tmp/backup
mv /root/backup-64.gz /tmp/backup
sync
umount /tmp/backup

printStr yellow "
==============================================================
配置已完成，盒子正在重启并准备初始化中，
请关闭电源重新上电，8分钟后手动重启电源第二次，并继续稍等初始化结束
=============================================================="
recoverbackup -a