prepare.sh

#!/bin/bash

#fonts color
Green="\033[32m"
Red="\033[31m"
#Yellow="\033[33m"
GreenBG="\033[42;37m"
RedBG="\033[41;37m"
YellowBG="\033[43;37m"
Font="\033[0m"

#notification information
Info="${Green}[信息]${Font}"
OK="${Green}[OK]${Font}"
Error="${Red}[错误]${Font}"
Warning="${Red}[警告]${Font}"


source '/etc/os-release'
VERSION=$(echo "${VERSION}" | awk -F "[()]" '{print $2}')


is_root() {
  if [ 0 == $UID ]; then
    echo -e "${OK} ${GreenBG} 当前用户是root用户，进入安装流程 ${Font}"
    sleep 3
  else
    echo -e "${Error} ${RedBG} 当前用户不是root用户，请切换到root用户后重新执行脚本 ${Font}"
    exit 1
  fi
}

judge() {
  if [[ 0 -eq $? ]]; then
    echo -e "${OK} ${GreenBG} $1 完成 ${Font}"
    sleep 1
  else
    echo -e "${Error} ${RedBG} $1 失败${Font}"
    exit 1
  fi
}

check_system() {
  if [[ "${ID}" == "centos" && ${VERSION_ID} -ge 7 ]]; then
    echo -e "${OK} ${GreenBG} 当前系统为 Centos ${VERSION_ID} ${VERSION} ${Font}"
    INS="yum"
  elif [[ "${ID}" == "debian" && ${VERSION_ID} -ge 8 ]]; then
    echo -e "${OK} ${GreenBG} 当前系统为 Debian ${VERSION_ID} ${VERSION} ${Font}"
    INS="apt"
    $INS update
    ## 添加 Nginx apt源
  elif [[ "${ID}" == "ubuntu" && $(echo "${VERSION_ID}" | cut -d '.' -f1) -ge 16 ]]; then
    echo -e "${OK} ${GreenBG} 当前系统为 Ubuntu ${VERSION_ID} ${UBUNTU_CODENAME} ${Font}"
    INS="apt"
    rm /var/lib/dpkg/lock
    dpkg --configure -a
    rm /var/lib/apt/lists/lock
    rm /var/cache/apt/archives/lock
    $INS update
  else
    echo -e "${Error} ${RedBG} 当前系统为 ${ID} ${VERSION_ID} 不在支持的系统列表内，安装中断 ${Font}"
    exit 1
  fi

  $INS install dbus -y

  systemctl stop firewalld
  systemctl disable firewalld
  echo -e "${OK} ${GreenBG} firewalld 已关闭 ${Font}"

  systemctl stop ufw
  systemctl disable ufw
  echo -e "${OK} ${GreenBG} ufw 已关闭 ${Font}"
}

config_chrony() {
  if [[ "${ID}" == "centos" && ${VERSION_ID} -ge 7 ]]; then
    systemctl start chronyd
    judge "启动时间同步服务器"
    systemctl enable chronyd
    judge "设置时间同步服务器开机启动"
  elif [[ "${ID}" == "debian" && ${VERSION_ID} -ge 8 ]] || \
       [[ "${ID}" == "ubuntu" && $(echo "${VERSION_ID}" | cut -d '.' -f1) -ge 16 ]]; then
    if ! command -v chronyd &> /dev/null; then
      echo -e "${Error} ${RedBG} chrony 未安装，请先确保已安装 chrony ${Font}"
      return 1
    fi
    systemctl start chrony
    judge "启动时间同步服务器"
    systemctl enable chrony
    judge "设置时间同步服务器开机启动"
  else
    echo -e "${Error} ${RedBG} 当前系统(${ID} ${VERSION_ID})不支持自动配置chrony，请手动配置时间同步 ${Font}"
  fi
}


dependency_install() {
  ${INS} install -y ca-certificates chrony wget git lsof bc unzip curl
  judge "安装基础软件包"

  config_chrony
  judge "配置时间同步服务器"

  date=`date`
  echo "现在的时间是 ${date}"
  sleep 5

  if [[ "${ID}" == "centos" && ${VERSION_ID} -ge 7 ]]; then
    ${INS} -y install python36
    judge "安装 python36"

    ${INS} -y install gcc python3-devel -y
    judge "安装Python依赖"

  elif [[ "${ID}" == "debian" && ${VERSION_ID} -ge 8 ]]; then
     ${INS} -y install python3 python3-pip
  elif [[ "${ID}" == "ubuntu" && $(echo "${VERSION_ID}" | cut -d '.' -f1) -ge 16 ]]; then
    ${INS} -y install python3 python3-pip
  else
    echo -e "${Error} ${RedBG} 当前系统为 ${ID} ${VERSION_ID} 不在支持的系统列表内，安装中断 ${Font}"
    exit 1
  fi

  mkdir -p /usr/local/bin >/dev/null 2>&1
}

basic_optimization() {
  # 最大文件打开数
  sed -i '/^\*\ *soft\ *nofile\ *[[:digit:]]*/d' /etc/security/limits.conf
  sed -i '/^\*\ *hard\ *nofile\ *[[:digit:]]*/d' /etc/security/limits.conf
  echo '* soft nofile 65536' >>/etc/security/limits.conf
  echo '* hard nofile 65536' >>/etc/security/limits.conf

  ulimit -SHn 10240
  ulimit -SHs unlimited
  echo '500000' > /proc/sys/net/netfilter/nf_conntrack_max

  # 关闭 Selinux
  if [[ "${ID}" == "centos" ]]; then
    if command -v getenforce &> /dev/null; then
      sed -i 's/^SELINUX=.*/SELINUX=disabled/' /etc/selinux/config
      setenforce 0 && echo "已经关闭selinux" || echo "关闭selinux出错"
    else
      echo "SELinux 未安装，无需关闭"
    fi
  fi

}

python_requirements() {
  pip3 install -r requirements.txt || {
    case "$?" in
      1)
        echo -e "${Error} ${RedBG} pip3 安装依赖时遇到未知错误，请检查网络或依赖定义 ${Font}"
        ;;
      2)
        echo -e "${Error} ${RedBG} 未找到 requirements.txt 文件，请确保文件存在且路径正确 ${Font}"
        ;;
      *)
        echo -e "${Error} ${RedBG} pip3 安装过程中发生错误（错误码: $?），可能是依赖冲突或其他问题，请检查输出信息 ${Font}"
        ;;
    esac
    exit 1
  }
  judge "安装python3 依赖"
}

prepare_process(){
  is_root
  check_system
  dependency_install
  basic_optimization
  judge "系统准备"
  python_requirements
  sleep 5
  echo "请执行 python3 main.py --help 命令执行下一步 "
}

list(){
    case $1 in
    run)
        prepare_process
        ;;
    *)
        echo -e "请使用 ${RedBG} bash $0 run ${Font} 选项，以安装准备运行环境"
        exit 1
        ;;
    esac
}

list "$1"