install.sh

#!/bin/bash

#bash <(curl -s -L https://raw.githubusercontent.com/VIPORMiner/RMS/main/install.sh)
#bash <(curl -s -L -k https://cdn.jsdelivr.net/gh/VIPORMiner/RMS@main/install.sh)
#bash <(curl -s -L -k http://saz1122.oss-cn-hongkong.aliyuncs.com/ViporMiner/RMS/raw/main/install.sh)
#bash <(curl -s -L -k https://raw.nuaa.cf/VIPORMiner/RMS/main/install.sh)
clear

[ $(id -u) != "0" ] && { echo "请使用ROOT用户进行安装, 输入sudo -i切换。"; exit 1; }

if command -v systemctl &> /dev/null; then
    echo "check systemctl..."
    clear
else
    echo "当前系统不支持systemctl服务, 请先安装systemctl."
    exit 1;
fi

SERVICE_NAME="rmservice"

PATH_RMS="/root/rms"
PATH_EXEC="rms"
PATH_NOHUP="${PATH_RMS}/nohup.out"
PATH_ERR="${PATH_RMS}/err.log"

ROUTE_1="https://github.com"
ROUTE_2="http://vippool.cn"
# ROUTE_2="https://hub.njuu.cf"
# ROUTE_3="https://hub.yzuu.cf"
# ROUTE_4="https://hub.nuaa.cf"

ROUTE_EXEC_1="/ViporMiner/RMS/raw/main/x86_64-musl/rms"
ROUTE_EXEC_2="/ViporMiner/RMS/raw/main/x86_64-android/rms"
ROUTE_EXEC_3="/ViporMiner/RMS/raw/main/arm-musleabi/rms"
ROUTE_EXEC_4="/ViporMiner/RMS/raw/main/arm-musleabihf/rms"
ROUTE_EXEC_5="/ViporMiner/RMS/raw/main/armv7-musleabi/rms"
ROUTE_EXEC_6="/ViporMiner/RMS/raw/main/armv7-musleabihf/rms"
ROUTE_EXEC_7="/ViporMiner/RMS/raw/main/i586-musl/rms"
ROUTE_EXEC_8="/ViporMiner/RMS/raw/main/i686-android/rms"
ROUTE_EXEC_9="/ViporMiner/RMS/raw/main/aarch64-musl/rms"

TARGET_ROUTE=""
TARGET_ROUTE_EXEC=""

UNAME=`uname -m`

filterResult() {
    if [ $1 -eq 0 ]; then
        echo ""
    else
        echo "!!!!!!!!!!!!!!!ERROR!!!!!!!!!!!!!!!!"
        echo "【${2}】失败。"
	
        if [ ! $3 ];then
            echo "!!!!!!!!!!!!!!!ERROR!!!!!!!!!!!!!!!!"
            exit 1
        fi
    fi
    echo -e
}

disable_firewall() {
    os_name=$(grep "^ID=" /etc/os-release | cut -d "=" -f 2 | tr -d '"')
    echo "关闭防火墙"

    if [ "$os_name" == "ubuntu" ]; then
        sudo ufw disable
    elif [ "$os_name" == "centos" ]; then
        sudo systemctl stop firewalld
        sudo systemctl disable firewalld
    else
        echo "未知的操作系统, 关闭防火墙失败"
    fi
}

check_process() {
    if [[ $(uname) == "Linux" ]]; then
        if pgrep -x "$1" >/dev/null; then
            return 0
        else
            return 1
        fi
    else
        if ps aux | grep -v grep | grep "$1" >/dev/null; then
            return 0
        else
            return 1
        fi
    fi
}

# 设置开机启动且进程守护
enable_autostart() {
    echo "${m_14}"
    if [ "$(command -v systemctl)" ]; then
        sudo tee /etc/systemd/system/$SERVICE_NAME.service > /dev/null <<EOF
[Unit]
Description=My Program
After=network.target

[Service]
Type=simple
ExecStart=$PATH_RMS/$PATH_EXEC
WorkingDirectory=$PATH_RMS/
Restart=always
StandardOutput=file:$PATH_RMS/nohup.out
StandardError=file:$PATH_RMS/err.log
TimeoutStopSec=5

[Install]
WantedBy=multi-user.target
EOF
        sudo systemctl daemon-reload
        sudo systemctl enable $SERVICE_NAME.service
        sudo systemctl start $SERVICE_NAME.service
    else
        sudo sh -c "echo '${PATH_RMS}/${PATH_EXEC} &' >> /etc/rc.local"
        sudo chmod +x /etc/rc.local
    fi
}

# 禁用开机启动函数
disable_autostart() {
    echo "关闭开机启动..."
    if [ "$(command -v systemctl)" ]; then
        sudo systemctl stop $SERVICE_NAME.service
        sudo systemctl disable $SERVICE_NAME.service
        sudo rm /etc/systemd/system/$SERVICE_NAME.service
        sudo systemctl daemon-reload
    else # 系统使用的是SysVinit
        sudo sed -i '/\/root\/rustminersystem\/rustminersystem\ &/d' /etc/rc.local
    fi

    sleep 1
}

kill_process() {
    local process_name="$1"
  local pids=($(pgrep "$process_name"))
  if [ ${#pids[@]} -eq 0 ]; then
    echo "未发现 $process_name 进程."
    return 1
  fi
  for pid in "${pids[@]}"; do
    echo "Stopping process $pid ..."
    kill -TERM "$pid"
  done
  echo "终止 $process_name ."

  sleep 1
}

install() {
    chown root:root /mnt -R
    chown root:root /etc -R
    chown root:root /usr -R
    chown man:root /var/cache/man -R
    chmod g+s /var/cache/man -R

    disable_firewall

    check_process $PATH_EXEC

    if [ $? -eq 0 ]; then
        echo "发现正在运行的${PATH_EXEC}需要停止才可继续安装。"
        echo "输入1停止正在运行的${PATH_EXEC}并且继续安装, 输入2取消安装。"

        read -p "$(echo -e "请选择[1-2]：")" choose
        case $choose in
        1)
            stop
            ;;
        2)
            echo "取消安装"
            return
            ;;
        *)
            echo "输入错误, 取消安装。"
            return
            ;;
        esac
    fi

    if [[ ! -d $PATH_RMS ]];then
        mkdir $PATH_RMS
        chmod 777 -R $PATH_RMS
    else
        echo "目录已存在, 无需重复创建, 继续执行安装。"
    fi

    if [[ ! -d $PATH_NOHUP ]];then
        touch $PATH_NOHUP
        touch $PATH_ERR

        chmod 777 -R $PATH_NOHUP
        chmod 777 -R $PATH_ERR
    fi

    echo "开始下载程序..."

    wget -P $PATH_RMS "${TARGET_ROUTE}${TARGET_ROUTE_EXEC}" -O "${PATH_RMS}/${PATH_EXEC}" 1>/dev/null

    filterResult $? "下载程序"

    chmod 777 -R "${PATH_RMS}/${PATH_EXEC}"

    start
}

restart() {
    stop

    start
}

uninstall() {
    stop

    rm -rf ${PATH_RMS}

    disable_autostart

    echo "卸载成功"
}

start() {
    echo $BLUE "启动程序..."
    check_process $PATH_EXEC

    if [ $? -eq 0 ]; then
        echo "程序已经启动，请不要重复启动。"
        return
    else
        # cd $PATH_RUST

        # nohup "${PATH_RUST}/${PATH_EXEC}" 2>$PATH_ERR &

        enable_autostart

        sleep 1

        check_process $PATH_EXEC

        if [ $? -eq 0 ]; then
            echo "|----------------------------------------------------------------|"
            echo "程序启动成功, 访问此地址: 局域网IP:42703"
            echo "|----------------------------------------------------------------|"
        else
            echo "程序启动失败!!!"
        fi
    fi
}

stop() {
    sleep 1

    disable_autostart

    sleep 1

    echo "终止进程..."

    kill_process $PATH_EXEC

    sleep 1
}

echo "------RMS Linux------"
echo "1. 安装"
echo "2. 停止运行RMS"
echo "3. 重启RMS"
echo "4. 卸载RMS"
echo "---------------------"

read -p "$(echo -e "[1-4]：")" comm

if [ "$comm" = "1" ]; then
    clear
elif [ "$comm" = "2" ]; then
    stop
    exit 1
elif [ "$comm" = "3" ]; then
    restart
    exit 1
elif [ "$comm" = "4" ]; then
    uninstall
    exit 1
fi


echo "------RMS Linux------"
echo "当前CPU架构【${UNAME}】"
echo 请选择对应架构安装选项。
echo "---------------------"
echo "1. x86-64"
echo "2. x86-64-android"
echo "3. arm-musleabi"
echo "4. arm-musleabihf"
echo "5. armv7-musleabi"
echo "6. armv7-musleabihf"
echo "7. i586"
echo "8. i686-android"
echo "9. aarch64"
echo ""

read -p "$(echo -e "[1-9]：")" targetExec

VARNAME="ROUTE_EXEC_${targetExec}"
TARGET_ROUTE_EXEC="${!VARNAME}"

clear

echo "------RMS Linux------"
echo "请选择下载线路:"
echo "1. 线路1（github官方地址, 如无法下载请选择其他线路）"
echo "2. 线路2"
# echo "3. 线路3"
# echo "4. 线路4"
echo "---------------------"

read -p "$(echo -e "[1-2]：")" targetRoute

VARNAME="ROUTE_${targetRoute}"
TARGET_ROUTE="${!VARNAME}"

[ ! $TARGET_ROUTE ] && { echo "错误的线路选择命令"; exit 1; }
[ ! $TARGET_ROUTE_EXEC ] && { echo "错误的架构选择命令"; exit 1; }

echo "${TARGET_ROUTE}${TARGET_ROUTE_EXEC}"

install