mtg_install.sh

#!/usr/bin/env bash
# shellcheck disable=SC2268
BIN_PATH=${BIN_PATH:-/usr/local/share/mtg}
TOML_PATH=${TOML_PATH:-/usr/local/etc/mtg}
curl() {
  $(type -P curl) -L -q --retry 5 --retry-delay 10 --retry-max-time 60 "$@"
}
systemd_cat_config() {
  if systemd-analyze --help | grep -qw 'cat-config'; then
    systemd-analyze --no-pager cat-config "$@"
    echo
  else
    echo "${aoi}~~~~~~~~~~~~~~~~"
    cat "$@" "$1".d/*
    echo "${aoi}~~~~~~~~~~~~~~~~"
    echo "${red}warning: ${green}The systemd version on the current operating system is too low."
    echo "${red}warning: ${green}Please consider to upgrade the systemd or the operating system.${reset}"
    echo
  fi
}

check_if_running_as_root() {
  # If you want to run as another user, please modify $UID to be owned by this user
  if [[ "$UID" -ne '0' ]]; then
    echo "WARNING: The user currently executing this script is not root. You may encounter the insufficient privilege error."
    read -r -p "Are you sure you want to continue? [y/n] " cont_without_been_root
    if [[ x"${cont_without_been_root:0:1}" = x'y' ]]; then
      echo "Continuing the installation with current user..."
    else
      echo "Not running with root, exiting..."
      exit 1
    fi
  fi
}

identify_the_operating_system_and_architecture() {
  if [[ "$(uname)" == 'Linux' ]]; then
    case "$(uname -m)" in
      'i386' | 'i686')
        MACHINE='386'
        ;;
      'amd64' | 'x86_64')
        MACHINE='amd64'
        ;;
      *)
        echo "error: The architecture is not supported."
        exit 1
        ;;
    esac
    if [[ ! -f '/etc/os-release' ]]; then
      echo "error: Don't use outdated Linux distributions."
      exit 1
    fi
    if [[ -f /.dockerenv ]] || grep -q 'docker\|lxc' /proc/1/cgroup && [[ "$(type -P systemctl)" ]]; then
      true
    elif [[ -d /run/systemd/system ]] || grep -q systemd <(ls -l /sbin/init); then
      true
    else
      echo "error: Only Linux distributions using systemd are supported."
      exit 1
    fi
    if [[ "$(type -P apt)" ]]; then
      PACKAGE_MANAGEMENT_INSTALL='apt -y --no-install-recommends install'
      PACKAGE_MANAGEMENT_REMOVE='apt purge'
      package_provide_tput='ncurses-bin'
    elif [[ "$(type -P dnf)" ]]; then
      PACKAGE_MANAGEMENT_INSTALL='dnf -y install'
      PACKAGE_MANAGEMENT_REMOVE='dnf remove'
      package_provide_tput='ncurses'
    elif [[ "$(type -P yum)" ]]; then
      PACKAGE_MANAGEMENT_INSTALL='yum -y install'
      PACKAGE_MANAGEMENT_REMOVE='yum remove'
      package_provide_tput='ncurses'
    elif [[ "$(type -P zypper)" ]]; then
      PACKAGE_MANAGEMENT_INSTALL='zypper install -y --no-recommends'
      PACKAGE_MANAGEMENT_REMOVE='zypper remove'
      package_provide_tput='ncurses-utils'
    elif [[ "$(type -P pacman)" ]]; then
      PACKAGE_MANAGEMENT_INSTALL='pacman -Syu --noconfirm'
      PACKAGE_MANAGEMENT_REMOVE='pacman -Rsn'
      package_provide_tput='ncurses'
    else
      echo "error: The script does not support the package manager in this operating system."
      exit 1
    fi
  else
    echo "error: This operating system is not supported."
    exit 1
  fi
}
## Demo function for processing parameters
judgment_parameters() {
  while [[ "$#" -gt '0' ]]; do
    case "$1" in
      '--remove')
        if [[ "$#" -gt '1' ]]; then
          echo 'error: Please enter the correct parameters.'
          exit 1
        fi
        REMOVE='1'
        ;;
      '--version')
        VERSION="${2:?error: Please specify the correct version.}"
        break
        ;;
      '--domain')
        DOMAIN="${2:?error: Please input a domain.}"
        shift
        ;;
       '--port')
        PORT="${2:?error: Please input a port to listen.}"
        shift
        ;;
      '-c' | '--check')
        CHECK='1'
        break
        ;;
      '-f' | '--force')
        FORCE='1'
        break
        ;;
      '-h' | '--help')
        HELP='1'
        break
        ;;
      '-l' | '--local')
        LOCAL_INSTALL='1'
        LOCAL_FILE="${2:?error: Please specify the correct local file.}"
        break
        ;;
      '-p' | '--proxy')
        if [[ -z "${2:?error: Please specify the proxy server address.}" ]]; then
          exit 1
        fi
        PROXY="$2"
        shift
        ;;
      *)
        echo "$0: unknown option -- -"
        exit 1
        ;;
    esac
    shift
  done
}

install_software() {
  package_name="$1"
  file_to_detect="$2"
  type -P "$file_to_detect" > /dev/null 2>&1 && return
  if ${PACKAGE_MANAGEMENT_INSTALL} "$package_name"; then
    echo "info: $package_name is installed."
  else
    echo "error: Installation of $package_name failed, please check your network."
    exit 1
  fi
}

get_version() {
  # 0: Install or update mtg.
  # 1: Installed or no new version of mtg.
  # 2: Install the specified version of mtg.
  if [[ -n "$VERSION" ]]; then
    RELEASE_VERSION="v${VERSION#v}"
    return 2
  fi
  # Determine the version number for mtg installed from a local file
  if [[ -f '/usr/local/bin/mtg' ]]; then
    VERSION="$(/usr/local/bin/mtg --version | awk 'NR==1 {print $1}')"
    CURRENT_VERSION="v${VERSION#v}"
    if [[ "$LOCAL_INSTALL" -eq '1' ]]; then
      RELEASE_VERSION="$CURRENT_VERSION"
      return
    fi
  fi
  # Get mtg release version number
  TMP_FILE="$(mktemp)"
  if ! curl -x "${PROXY}" -sS -H "Accept: application/vnd.github.v3+json" -o "$TMP_FILE" 'https://api.github.com/repos/9seconds/mtg/releases/latest'; then
    "rm" "$TMP_FILE"
    echo 'error: Failed to get release list, please check your network.'
    exit 1
  fi

  RELEASE_LATEST="$(sed 'y/,/\n/' "$TMP_FILE" | grep 'tag_name' | awk -F '"' '{print $4}')"
  "rm" "$TMP_FILE"
  RELEASE_VERSION="v${RELEASE_LATEST#v}"
  # Compare mtg version numbers
  if [[ "$RELEASE_VERSION" != "$CURRENT_VERSION" ]]; then
    RELEASE_VERSIONSION_NUMBER="${RELEASE_VERSION#v}"
    RELEASE_MAJOR_VERSION_NUMBER="${RELEASE_VERSIONSION_NUMBER%%.*}"
    RELEASE_MINOR_VERSION_NUMBER="$(echo "$RELEASE_VERSIONSION_NUMBER" | awk -F '.' '{print $2}')"
    RELEASE_MINIMUM_VERSION_NUMBER="${RELEASE_VERSIONSION_NUMBER##*.}"
    # shellcheck disable=SC2001
    CURRENT_VERSIONSION_NUMBER="$(echo "${CURRENT_VERSION#v}" | sed 's/-.*//')"
    CURRENT_MAJOR_VERSION_NUMBER="${CURRENT_VERSIONSION_NUMBER%%.*}"
    CURRENT_MINOR_VERSION_NUMBER="$(echo "$CURRENT_VERSIONSION_NUMBER" | awk -F '.' '{print $2}')"
    CURRENT_MINIMUM_VERSION_NUMBER="${CURRENT_VERSIONSION_NUMBER##*.}"
    if [[ "$RELEASE_MAJOR_VERSION_NUMBER" -gt "$CURRENT_MAJOR_VERSION_NUMBER" ]]; then
      return 0
    elif [[ "$RELEASE_MAJOR_VERSION_NUMBER" -eq "$CURRENT_MAJOR_VERSION_NUMBER" ]]; then
      if [[ "$RELEASE_MINOR_VERSION_NUMBER" -gt "$CURRENT_MINOR_VERSION_NUMBER" ]]; then
        return 0
      elif [[ "$RELEASE_MINOR_VERSION_NUMBER" -eq "$CURRENT_MINOR_VERSION_NUMBER" ]]; then
        if [[ "$RELEASE_MINIMUM_VERSION_NUMBER" -gt "$CURRENT_MINIMUM_VERSION_NUMBER" ]]; then
          return 0
        else
          return 1
        fi
      else
        return 1
      fi
    else
      return 1
    fi
  elif [[ "$RELEASE_VERSION" == "$CURRENT_VERSION" ]]; then
    return 1
  fi
}

download_mtg() {
  RELEASE_VERSION_NUM=${RELEASE_VERSION#*v}
  DOWNLOAD_LINK="https://github.com/9seconds/mtg/releases/download/$RELEASE_VERSION/mtg-$RELEASE_VERSION_NUM-linux-$MACHINE.tar.gz"
  echo "Downloading mtg archive: $DOWNLOAD_LINK"
  if ! curl -x "${PROXY}" -R -H 'Cache-Control: no-cache' -o "$TAR_GZ_FILE" "$DOWNLOAD_LINK"; then
    echo 'error: Download failed! Please check your network or try again.'
    return 1
  fi 
}

decompression() {
  if ! tar -xzvf "$1" -C "$TMP_DIRECTORY"; then
    echo 'error: mtg decompression failed.'
    "rm" -r "$TMP_DIRECTORY"
    echo "removed: $TMP_DIRECTORY"
    exit 1
  fi
  echo "info: Extract the mtg package to $TMP_DIRECTORY and prepare it for installation."
}

install_file() {
  NAME="$1"
  if [[ "$NAME" == 'mtg' ]] ; then
    install -m 755 "${TMP_DIRECTORY}/mtg-$RELEASE_VERSION_NUM-linux-$MACHINE/$NAME" "/usr/local/bin/$NAME"
  fi
}

install_mtg() {
  # Install mtg binary to /usr/local/bin/ and $BIN_PATH
  install_file mtg
  install -d "$BIN_PATH"

  # Install mtg configuration file to $TOML_PATH
  # shellcheck disable=SC2153
#   if [[ ! -d "$TOML_PATH" ]]; then
#     install -d "$TOML_PATH"

#     echo "mtg generate-secret $DOMAIN"
#     MTG_SECRET = $(mtg generate-secret $DOMAIN)    
#     echo "secret = \"$MTG_SECRET\"
# bind-to = \"0.0.0.0:8443\"" > "${TOML_PATH}/mtg.toml"
#     CONFIG_NEW='1'
#   fi

    install -d "$TOML_PATH"

    if [[ -z "$DOMAIN" ]]; then
        echo "DOMAIN is empty"
        read -p "input a domain to use(fake):" DOMAIN
    fi
    if [[ -z $PORT ]]; then
        echo "PORT is empty"
        read -p "input a port to listen:" PORT
    fi
    echo "mtg generate-secret $DOMAIN"
    MTG_SECRET=$(mtg generate-secret $DOMAIN)    
    echo "secret = \"$MTG_SECRET\"
bind-to = \"0.0.0.0:$PORT\"" > "${TOML_PATH}/mtg.toml"
}

install_startup_service_file() {
  echo "[Unit]
Description=mtg - MTProto proxy server
Documentation=https://github.com/9seconds/mtg
After=network.target

[Service]
ExecStart=/usr/local/bin/mtg run ${TOML_PATH}/mtg.toml
Restart=always
RestartSec=3
DynamicUser=true
AmbientCapabilities=CAP_NET_BIND_SERVICE

[Install]
WantedBy=multi-user.target" > /etc/systemd/system/mtg.service
  systemd_cat_config /etc/systemd/system/mtg.service
  # shellcheck disable=SC2154

  systemctl daemon-reload
  SYSTEMD='1'
}

start_mtg() {
  if [[ -f '/etc/systemd/system/mtg.service' ]]; then
    if systemctl start "${mtg_CUSTOMIZE:-mtg}"; then
      echo 'info: Start the mtg service.'
    else
      echo 'error: Failed to start mtg service.'
      exit 1
    fi
  fi
}

stop_mtg() {
  mtg_CUSTOMIZE="$(systemctl list-units | grep 'mtg' | awk -F ' ' '{print $1}')"
  if [[ -z "$mtg_CUSTOMIZE" ]]; then
    local mtg_daemon_to_stop='mtg.service'
  else
    local mtg_daemon_to_stop="$mtg_CUSTOMIZE"
  fi
  if ! systemctl stop "$mtg_daemon_to_stop"; then
    echo 'error: Stopping the mtg service failed.'
    exit 1
  fi
  echo 'info: Stop the mtg service.'
}

check_update() {
  if [[ -f '/etc/systemd/system/mtg.service' ]]; then
    get_version
    local get_ver_exit_code=$?
    if [[ "$get_ver_exit_code" -eq '0' ]]; then
      echo "info: Found the latest release of mtg $RELEASE_VERSION . (Current release: $CURRENT_VERSION)"
    elif [[ "$get_ver_exit_code" -eq '1' ]]; then
      echo "info: No new version. The current version of mtg is $CURRENT_VERSION ."
    fi
    exit 0
  else
    echo 'error: mtg is not installed.'
    exit 1
  fi
}

remove_mtg() {
  if systemctl list-unit-files | grep -qw 'mtg'; then
    if [[ -n "$(pidof mtg)" ]]; then
      stop_mtg
    fi
    if ! ("rm" -r '/usr/local/bin/mtg' \
      "$BIN_PATH" \
      '/etc/systemd/system/mtg.service' ); then
      echo 'error: Failed to remove mtg.'
      exit 1
    else
      echo 'removed: /usr/local/bin/mtg'
      echo "removed: $BIN_PATH"
      echo 'removed: /etc/systemd/system/mtg.service'
      echo 'Please execute the command: systemctl disable mtg'
      echo "You may need to execute a command to remove dependent software: $PACKAGE_MANAGEMENT_REMOVE curl unzip"
      echo 'info: mtg has been removed.'
      echo 'info: If necessary, manually delete the configuration and log files.'
      exit 0
    fi
  else
    echo 'error: mtg is not installed.'
    exit 1
  fi
}

# Explanation of parameters in the script
show_help() {
  echo "usage: $0 [--remove | --version number | -c | -f | -h | -l | -p]"
  echo '  [-p address] [--version number | -c | -f]'
  echo '  --remove        Remove mtg'
  echo '  --version       Install the specified version of mtg, e.g., --version v4.18.0'
  echo '  -c, --check     Check if mtg can be updated'
  echo '  -f, --force     Force installation of the latest version of mtg'
  echo '  -h, --help      Show help'
  echo '  -l, --local     Install mtg from a local file'
  echo '  -p, --proxy     Download through a proxy server, e.g., -p http://127.0.0.1:8118 or -p socks5://127.0.0.1:1080'
  echo '  --domain        Fake Tls Domain, e.g., --domain baidu.com'
  echo '  --port          port to listen , e.g., --port 443'
  exit 0
}

main() {
  check_if_running_as_root
  identify_the_operating_system_and_architecture
  judgment_parameters "$@"

  install_software "$package_provide_tput" 'tput'
  red=$(tput setaf 1)
  green=$(tput setaf 2)
  aoi=$(tput setaf 6)
  reset=$(tput sgr0)

  # Parameter information
  [[ "$HELP" -eq '1' ]] && show_help
  [[ "$CHECK" -eq '1' ]] && check_update
  [[ "$REMOVE" -eq '1' ]] && remove_mtg

  # Two very important variables
  TMP_DIRECTORY="$(mktemp -d)"
  TAR_GZ_FILE="${TMP_DIRECTORY}/mtg-linux-$MACHINE.tar.gz"

  # Install mtg from a local file, but still need to make sure the network is available
  if [[ "$LOCAL_INSTALL" -eq '1' ]]; then
    echo 'warn: Install mtg from a local file, but still need to make sure the network is available.'
    echo -n 'warn: Please make sure the file is valid because we cannot confirm it. (Press any key) ...'
    read -r
    install_software 'tar' 'tar'
    decompression "$LOCAL_FILE"
  else
    # Normal way
    install_software 'curl' 'curl'
    get_version
    NUMBER="$?"
    if [[ "$NUMBER" -eq '0' ]] || [[ "$FORCE" -eq '1' ]] || [[ "$NUMBER" -eq 2 ]]; then
      echo "info: Installing mtg $RELEASE_VERSION for $(uname -m)"
      download_mtg
      if [[ "$?" -eq '1' ]]; then
        "rm" -r "$TMP_DIRECTORY"
        echo "removed: $TMP_DIRECTORY"
        exit 1
      fi
      install_software 'tar' 'tar'
      decompression "$TAR_GZ_FILE"
    elif [[ "$NUMBER" -eq '1' ]]; then
      echo "info: No new version. The current version of mtg is $CURRENT_VERSION ."
      exit 0
    fi
  fi

  # Determine if mtg is running
  if systemctl list-unit-files | grep -qw 'mtg'; then
    if [[ -n "$(pidof mtg)" ]]; then
      stop_mtg
      mtg_RUNNING='1'
    fi
  fi
  install_mtg
  install_startup_service_file
  echo 'installed: /usr/local/bin/mtg'
  if [[ "$CONFIG_NEW" -eq '1' ]]; then
    echo "installed: ${TOML_PATH}/mtg.toml"
  fi
  if [[ "$SYSTEMD" -eq '1' ]]; then
    echo 'installed: /etc/systemd/system/mtg.service'
  fi
  "rm" -r "$TMP_DIRECTORY"
  echo "removed: $TMP_DIRECTORY"
  if [[ "$LOCAL_INSTALL" -eq '1' ]]; then
    get_version
  fi
  echo "info: mtg $RELEASE_VERSION is installed."
  echo "You may need to execute a command to remove dependent software: $PACKAGE_MANAGEMENT_REMOVE curl tar"
  if [[ "$mtg_RUNNING" -eq '1' ]]; then
    start_mtg
  else
    echo 'Please execute the command: systemctl enable mtg; systemctl start mtg'
    start_mtg
  fi
  mtg access "${TOML_PATH}/mtg.toml"
}

main "$@"