#!/bin/bash
#
# DongTool 一键安装器
# 使用: curl -sS https://sh.arkup.cc | bash
#

set -e

DOWNLOAD_URL="https://sh.arkup.cc/dongtool"
CHECKSUM_URL="https://sh.arkup.cc/dongtool.sha256"
INSTALL_PATH="/usr/local/bin/d"

echo ""
echo -e "\033[32m ____                    _____           _  \033[0m"
echo -e "\033[32m|  _ \\  ___  _ __   __ _|_   _|__   ___ | | \033[0m"
echo -e "\033[32m| | | |/ _ \\| '_ \\ / _\` | | |/ _ \\ / _ \\| | \033[0m"
echo -e "\033[32m| |_| | (_) | | | | (_| | | | (_) | (_) | | \033[0m"
echo -e "\033[32m|____/ \\___/|_| |_|\\__, | |_|\\___/ \\___/|_| \033[0m"
echo -e "\033[32m                   |___/                     \033[0m"
echo ""
echo -e "\033[37m站群管理工具箱 - 安装程序\033[0m"
echo ""

# 检查是否 root
if [ "$(id -u)" -ne 0 ]; then
    echo -e "\033[31m请使用 root 用户运行: sudo bash -c \"\$(curl -sS https://sh.arkup.cc)\"\033[0m"
    exit 1
fi

# 检查系统架构
ARCH=$(uname -m)
if [[ "$ARCH" != "x86_64" && "$ARCH" != "amd64" ]]; then
    echo -e "\033[31m暂不支持 $ARCH 架构，仅支持 x86_64\033[0m"
    exit 1
fi

echo -e "\033[96m正在下载 DongTool...\033[0m"
curl -sS -o "$INSTALL_PATH" "$DOWNLOAD_URL"

if [ $? -ne 0 ]; then
    echo -e "\033[31m下载失败，请检查网络\033[0m"
    exit 1
fi

# 校验文件完整性
echo -e "\033[96m正在校验文件完整性...\033[0m"
EXPECTED_SUM=$(curl -sS "$CHECKSUM_URL" 2>/dev/null | awk '{print $1}')
if [ -n "$EXPECTED_SUM" ]; then
    ACTUAL_SUM=$(sha256sum "$INSTALL_PATH" | awk '{print $1}')
    if [ "$EXPECTED_SUM" != "$ACTUAL_SUM" ]; then
        echo -e "\033[31m文件校验失败，可能已被篡改，已终止安装\033[0m"
        rm -f "$INSTALL_PATH"
        exit 1
    fi
    echo -e "\033[32m✓ 校验通过\033[0m"
else
    echo -e "\033[33m⚠ 无法获取校验值，跳过校验\033[0m"
fi

chmod +x "$INSTALL_PATH"

echo ""
echo -e "\033[32m✓ DongTool 安装成功！\033[0m"
echo ""
echo -e "  输入 \033[38;5;208md\033[0m 启动工具箱"
echo ""