#!/bin/sh

# 自定义action函数，实现通用action功能
success() {
  echo -en "\\033[60G[\\033[1;32m  OK  \\033[0;39m]\r"
  return 0
}

failure() {
  rc=$?
  echo -en "\\033[60G[\\033[1;31mFAILED\\033[0;39m]\r"
  [ -x /bin/plymouth ] && /bin/plymouth --details
  return $rc
}

action() {
  STRING=$1
  echo -n "$STRING "
  shift
  "$@" && success || failure
  rc=$?
  echo
  return $rc
}

# 函数，判断命令是否正常执行
if_success() {
  ReturnStatus=$3
  if [ $ReturnStatus -eq 0 ]; then
    action "$1" /bin/true
  else
    action "$2" /bin/false
    exit 1
  fi
}

# 定义路径变量
Server_Dir=$(cd "$(dirname "$0")" && pwd)
Conf_Dir="$Server_Dir/conf"
Log_Dir="$Server_Dir/logs"

## 关闭clash服务
Text1="服务关闭成功！"
Text2="服务关闭失败！"
# 查询并关闭程序进程
PID=$(pgrep -f 'clash-linux-armv7')
if [ -n "$PID" ]; then
  kill -9 $PID
  ReturnStatus=$?
else
  ReturnStatus=1
fi
if_success "$Text1" "$Text2" $ReturnStatus

sleep 3

## 获取CPU架构
if /bin/arch >/dev/null 2>&1; then
  CpuArch=$(/bin/arch)
elif /usr/bin/arch >/dev/null 2>&1; then
  CpuArch=$(/usr/bin/arch)
elif /bin/uname -m >/dev/null 2>&1; then
  CpuArch=$(/bin/uname -m)
else
  echo -e "\033[31m\n[ERROR] Failed to obtain CPU architecture！\033[0m"
  exit 1
fi

## 重启启动clash服务
Text5="服务启动成功！"
Text6="服务启动失败！"
if echo "$CpuArch" | grep -q "x86_64"; then
  nohup "$Server_Dir/bin/clash-linux-amd64" -d "$Conf_Dir" > "$Log_Dir/clash.log" 2>&1 &
  ReturnStatus=$?
  if_success "$Text5" "$Text6" $ReturnStatus
elif echo "$CpuArch" | grep -q "aarch64\|arm64"; then
  nohup "$Server_Dir/bin/clash-linux-arm64" -d "$Conf_Dir" > "$Log_Dir/clash.log" 2>&1 &
  ReturnStatus=$?
  if_success "$Text5" "$Text6" $ReturnStatus
elif echo "$CpuArch" | grep -q "armv7"; then
  nohup "$Server_Dir/bin/clash-linux-armv7" -d "$Conf_Dir" > "$Log_Dir/clash.log" 2>&1 &
  ReturnStatus=$?
  if_success "$Text5" "$Text6" $ReturnStatus
else
  echo -e "\033[31m\n[ERROR] Unsupported CPU Architecture！\033[0m"
  exit 1
fi

