#!/usr/bin/env bash

# 当前脚本版本号和新增功能
VERSION=1.0.0

declare -A T

T[E0]="Language:\n  1.English (default) \n  2.简体中文"
T[C0]="${T[E0]}"
T[E1]="Speed test and unlocking test is for reference only and does not represent the actual usage, due to network changes, Netflix blocking and ip replacement. Speed test is time-sensitive."
T[C1]="测速及解锁测试仅供参考,不代表实际使用情况,由于网络情况变化、Netflix 封锁及 ip 更换,测速具有时效性"
T[E2]="New Features: "
T[C2]="新特性: "
T[E3]="Choose:"
T[C3]="请选择:"
T[E4]="！ This cannot be empty！"
T[C4]="！ 此处不能为空！"
T[E5]="More than 5 errors have been entered, and the script exits."
T[C5]="输入错误超过5次, 脚本退出"
T[E6]="Please input a subscription url or a single node supported by v2ray (VLESS is not supported):"
T[C6]="请输入订阅链接或者 v2ray 支持的单节点 (不支持 VLESS):"
T[E7]="If there are more than 2 filters below, you can separate the keywords by spaces."
T[C7]="以下筛选条件如超过 2 个, 可以通过空格分隔关键词."
T[E8]="Filter nodes by remarks using keyword:"
T[C8]="使用关键字通过注释筛选节点:"
T[E9]="Exclude nodes by remarks using keyword:"
T[C9]="通过使用关键字的注释排除节点:"
T[E10]="Manually set group:"
T[C10]="请输入测速组名:"
T[E11]="Set the colors when exporting images:\n  1. origin (default)\n  2. poor"
T[C11]="导出图像时设置颜色:\n  1. origin (默认)\n  2. poor"
T[E12]="Select sort method, default not sorted:\n  1. Sort by [speed] from fast to slow\n  2. Sort by [speed] from slow to fast\n  3. Sort by [ping] from low to high\n  4. Sort by [ping] from high to low"
T[C12]="请选择排序方法, 默认不排序, 如默认请跳过:\n  1. 按 [速度] 从快到慢排序\n  2. 按 [速度] 从慢到快排序\n  3. 按 [ping] 从低到高排序\n  4. 按 [ping] 从高到低排序"
T[E13]="The script supports Debian, Ubuntu, CentOS, Arch or Alpine systems only."
T[C13]="本脚本只支持 Debian、Ubuntu、CentOS、Arch 或 Alpine 系统"
T[E14]="Step 1/3: Install dependence-list:"
T[C14]="进度 1/3: 安装依赖列表:"
T[E15]="Step 2/3: Update SSRSpeedN and dependencies."
T[C15]="进度 2/3: 更新 SSRSpeedN 和依赖"
T[E16]="Step 3/3: SSRSpeedN speed test."
T[C16]="进度 3/3: SSRSpeedN 测速"
T[E17]="Step 1/3: All dependencies already exist and do not need to be installed additionally."
T[C17]="进度 1/3: 所有依赖已存在，不需要额外安装"
T[E18]="Failed to download the client zip package."
T[C18]="下载客户端压缩包失败"
T[E19]="Client decompression failed."
T[C19]="客户端解压失败"
T[E20]="Whether to uninstall the following python3 dependencies:"
T[C20]="是否卸载以下 python3 依赖:"
T[E21]="The script supports AMD64 only."
T[C21]="本脚本只支持 AMD64 架构"
T[E22]=""
T[C22]=""
T[E23]="To uninstall the above dependencies, please press [y]. The default is not to uninstall:"
T[C23]="卸载以上依赖请按[y], 默认为不卸载:"
T[E24]="Uninstallation of SSRSpeedN is complete."
T[C24]="卸载 SSRSpeedN 已完成"
T[E25]="The SSRSpeedN installation folder cannot be found in the current path. Please check if it is already installed or the installation path."
T[C25]="当前路径下找不到 SSRSpeedN 安装文件夹, 请确认是否已安装或安装路径"

# 彩色 log 函数, read 函数, text 函数, 友道翻译函数
error() { echo -e "\033[31m\033[01m$1\033[0m" && exit 1; }
info() { echo -e "\033[32m\033[01m$1\033[0m"; }
warning() { echo -e "\033[33m\033[01m$1\033[0m"; }
reading() { read -rp "$(info "$1")" "$2"; }
text() { eval echo "\${T[${L}$1]}"; }
# translate() { [ -n "$1" ] && curl -ksm8 "http://fanyi.youdao.com/translate?&doctype=json&type=AUTO&i=${1//[[:space:]]/}" | cut -d \" -f18 2>/dev/null; }

# 选择语言, 先判断 SSRSpeedN/data/language 里的语言选择, 没有的话再让用户选择, 默认英语
select_language() {
  if [[ "$L" != [CE] ]]; then
    if [ -e SSRSpeedN/data/language ]; then
      L=$(cat SSRSpeedN/data/language 2>&1)
    else
      L=E && warning "\n $(text 0) \n" && reading " $(text 3) " LNG_CHOICE
      [ "$LNG_CHOICE" = 2 ] && L=C
    fi
  fi
  [ "$L" = C ] && timedatectl set-timezone Asia/Shanghai || timedatectl set-timezone UTC
}

help() {
  if [ $L = C ] || grep -q 'C' SSRSpeedN/data/language; then
    echo "
用法: ssrspeed [-h] [--version] [-c GUICONFIG] [-u URL] [-m TEST_METHOD]
                   [-M TEST_MODE] [--include FILTER [FILTER ...]]
                   [--include-remark REMARKS [REMARKS ...]]
                   [--include-group GROUP [GROUP ...]]
                   [--exclude EFLITER [EFLITER ...]]
                   [--exclude-group EGFILTER [EGFILTER ...]]
                   [--exclude-remark ERFILTER [ERFILTER ...]] [--use-ssr-cs]
                   [-g GROUP_OVERRIDE] [-y] [-C RESULT_COLOR] [-s SORT_METHOD]
                   [-i IMPORT_FILE] [--skip-requirements-check] [--debug]
                   [--paolu]

选项:

 --version                               显示程序的版本号并退出
 -h,--help                               显示此帮助消息并退出
 -c GUICONFIG,--config = GUICONFIG       加载由 shadowsocksr-csharp 生成的配置。
 -u URL,--url = URL                      从订阅 URL 加载 ssr 配置。
 -m TEST_METHOD,--method = TEST_METHOD   在 [speedtestnet, fast, socket, stasync] 中选择测试方法。
 -M TEST_MODE,--mode = TEST_MODE         在 [all, pingonly, wps] 中选择测试模式。
 --include                               按组过滤节点,并使用关键字注释。
 --include-remark                        使用关键字通过注释过滤节点。
 --include-group                         使用关键字按组名过滤节点。
 --exclude                               按组排除节点,并使用关键字进行注释。
 --exclude-group                         使用关键字按组排除节点。
 --exclude-remark                        通过使用关键字的注释排除节点。
 --use-ssr-cs                            用 ShadowsocksR-C# 替换 ShadowsocksR-libev(仅 Windows)
 -g GROUP                                手动设置组。
 -y,--yes                                测试前跳过节点列表确认。
 -C RESULT_COLOR,--color = RESULT_COLOR  导出图像时设置颜色。
 -S SORT_METHOD,--sort = SORT_METHOD     在 [speed, rspeed, ping, rping] 中选择排序方法,默认不排序。
 -i IMPORT_FILE,--import = IMPORT_FILE   从 json 文件导入测试结果并导出。
 -skip-requirements-check                跳过要求检查。
 --debug                                 在调试模式下运行程序。

 测试模式
 模式                  备注
 TCP_PING             仅 tcp ping,无速度测试
 WEB_PAGE_SIMULATION  网页模拟测试
 ALL                  全速测试(不包括网页模拟)

 测试方法
 方法                  备注
 ST_ASYNC             单线程异步下载
 SOCKET               具有多线程的原始套接字
 SPEED_TEST_NET       SpeedTest.Net 速度测试
 FAST                 Fast.com 速度测试
"
  else

    echo "
usage: ssrspeed [-h] [--version] [-c GUICONFIG] [-u URL] [-m TEST_METHOD]
                  [-M TEST_MODE] [--include FILTER [FILTER ...]]
                  [--include-remark REMARKS [REMARKS ...]]
                  [--include-group GROUP [GROUP ...]]
                  [--exclude EFLITER [EFLITER ...]]
                  [--exclude-group EGFILTER [EGFILTER ...]]
                  [--exclude-remark ERFILTER [ERFILTER ...]] [--use-ssr-cs]
                  [-g GROUP_OVERRIDE] [-y] [-C RESULT_COLOR] [-s SORT_METHOD]
                  [-i IMPORT_FILE] [--skip-requirements-check] [--debug]
                  [--paolu]

Options:

 --version                              show program's version number and exit
 -h, --help                             show this help message and exit
 -c GUICONFIG, --config=GUICONFIG       Load config generated by shadowsocksr-csharp.
 -u URL, --url=URL                      Load ssr config from subscription url.
 -m TEST_METHOD, --method=TEST_METHOD   Select test method in in [speedtestnet, fast, socket, stasync].
 -M TEST_MODE, --mode=TEST_MODE         Select test mode in [all, pingonly, wps].
 --include                              Filter nodes by group and remarks using keyword.
 --include-remark                       Filter nodes by remarks using keyword.
 --include-group                        Filter nodes by group name using keyword.
 --exclude                              Exclude nodes by group and remarks using keyword.
 --exclude-group                        Exclude nodes by group using keyword.
 --exclude-remark                       Exclude nodes by remarks using keyword.
 --use-ssr-cs                           Replace the ShadowsocksR-libev with the ShadowsocksR-C# (Only Windows)
 -g GROUP                               Manually set group.
 -y, --yes                              Skip node list confirmation before test.
 -C RESULT_COLOR, --color=RESULT_COLOR  Set the colors when exporting images..
 -S SORT_METHOD, --sort=SORT_METHOD     Select sort method in [speed, rspeed, ping, rping], default not sorted.
 -i IMPORT_FILE, --import=IMPORT_FILE   Import test result from json file and export it.
 --skip-requirements-check              Skip requirements check.
 --debug                                Run program in debug mode.

 Test Modes
 Mode                 Remark
 TCP_PING             Only tcp ping, no speed test
 WEB_PAGE_SIMULATION  Web page simulation test
 ALL                  Full speed test (exclude web page simulation)

 Test Methods
 Methods              Remark
 ST_ASYNC             Asynchronous download with single thread
 SOCKET               Raw socket with multithreading
 SPEED_TEST_NET       Speed Test Net speed test
 FAST                 Fast.com speed test
"
  fi
  exit 0
}

check_operating_system(){
  [ "$(uname -m)"  != "x86_64" ] && error " $(text 21) "

  CMD=( "$(grep -i pretty_name /etc/os-release 2>/dev/null | cut -d \" -f2)"
        "$(hostnamectl 2>/dev/null | grep -i system | cut -d : -f2)"
        "$(lsb_release -sd 2>/dev/null)"
        "$(grep -i description /etc/lsb-release 2>/dev/null | cut -d \" -f2)"
        "$(grep . /etc/redhat-release 2>/dev/null)"
        "$(grep . /etc/issue 2>/dev/null | cut -d \\ -f1 | sed '/^[ ]*$/d')"
      )

  for i in "${CMD[@]}"; do SYS="$i" && [ -n "$SYS" ] && break; done
  
  REGEX=("debian" "ubuntu" "centos|red hat|kernel|oracle linux|alma|rocky" "'amazon linux'" "alpine" "arch linux")
  RELEASE=("Debian" "Ubuntu" "CentOS" "CentOS" "Alpine" "Arch")
  EXCLUDE=("bookworm")
  PACKAGE_UPDATE=("apt -y update" "apt -y update" "yum -y update" "yum -y update" "apk update -f" "pacman -Sy")
  PACKAGE_INSTALL=("apt -y install" "apt -y install" "yum -y install" "yum -y install" "apk add -f" "pacman -S --noconfirm")
  PACKAGE_UNINSTALL=("apt -y autoremove" "apt -y autoremove" "yum -y autoremove" "yum -y autoremove" "apk del -f" "pacman -Rcnsu --noconfirm")
 
  for ((int=0; int<${#REGEX[@]}; int++)); do
    echo "$SYS" | grep -iq "${REGEX[int]}" && SYSTEM="${RELEASE[int]}" && [ -n "$SYSTEM" ] && break
  done
  [ -z "$SYSTEM" ] && error " $(text 13) "
}

input() {
  local i=0
  while [ -z "$URL" ]; do
    ((i++)) || true
    [ "$i" -gt 1 ] && NOT_BLANK="$(text 4) " && [ "$i" = 6 ] && error "\n $(text 5) "
    reading "\n ${NOT_BLANK}$(text 6) " URL
  done
  [ -n "$URL" ] && URL="-u $URL"
  warning "\n $(text 7) "
  reading "\n $(text 8) " INCLUDE_REMARK
  [ -n "$INCLUDE_REMARK" ] && INCLUDE_REMARK="--include-remark $INCLUDE_REMARK"
  reading "\n $(text 9) " EXCLUDE_REMARK
  [ -n "$EXCLUDE_REMARK" ] && EXCLUDE_REMARK="--exclude-remark $EXCLUDE_REMARK"
  reading "\n $(text 10) " GROUP
  [ -n "$GROUP" ] && GROUP="-g $GROUP"
  RESULT_COLOR="--color=origin"
  #  RESULT_COLOR="--color=origin" && warning "\n $(text 11) " && reading " $(text 3) " CHOOSE_COLOR && [ "$CHOOSE_COLOR" = 2 ] && RESULT_COLOR="--color=poor"
  warning "\n $(text 12) " && reading " $(text 3) " METHOD_CHOICE
  case "$METHOD_CHOICE" in 1) SORT_METHOD="--sort=speed" ;; 2) SORT_METHOD="--sort=rspeed" ;; 3) SORT_METHOD="--sort=ping" ;; 4) SORT_METHOD="--sort=rping" ;; esac
}

check_dependencies() {
  for j in {" sudo"," wget"," git"," python3"," unzip"}; do ! type -p $j >/dev/null 2>&1 && DEPS+=$j; done
  if [ -n "$DEPS" ]; then
    info "\n $(text 14) $DEPS \n"
    ${PACKAGE_UPDATE[int]}
    ${PACKAGE_INSTALL[int]} $DEPS
  else
    info "\n $(text 17) \n"
  fi
}

check_ssrspeedn() {
  info "\n $(text 15) \n"
  [ ! -e SSRSpeedN ] && sudo git clone https://github.com/Oreomeow/SSRSpeedN
  if [ ! -e SSRSpeedN/resources/clients ]; then
    LATEST=$(sudo wget -qO- "https://api.github.com/repos/Oreomeow/SSRSpeedN/releases/latest" | grep tag_name | sed -E 's/.*"([^"]+)".*/\1/' | cut -c 2-)
    LATEST=${LATEST:-'1.1.1'}
    sudo wget -O SSRSpeedN/resources/clients_linux_amd64.zip https://github.com/OreosLab/SSRSpeedN/releases/download/v"$LATEST"/clients_linux_amd64.zip
    [ ! -e SSRSpeedN/resources/clients_linux_amd64.zip ] && error " $(text 18) " || sudo unzip -d SSRSpeedN/resources/ SSRSpeedN/resources/clients_linux_amd64.zip
    [ ! -e SSRSpeedN/resources/clients ] && error " $(text 19) " || sudo rm -f SSRSpeedN/resources/clients_linux_amd64.zip
  fi
  sudo chmod -R +x SSRSpeedN
  cd SSRSpeedN || exit 1
  sudo git pull || sudo git fetch --all && sudo git reset --hard origin/main
  echo "$L" | sudo tee data/language >/dev/null 2>&1
  sudo pip3 install --upgrade pip
  sudo pip3 install six
  sudo pip3 install -r requirements.txt
  [ ! -e data/ssrspeed.json ] && sudo cp -f data/ssrspeed.example.json data/ssrspeed.json
}

test() {
  info "\n $(text 16) \n"
  sudo python3 -m ssrspeed $URL $INCLUDE_REMARK $EXCLUDE_REMARK $GROUP $RESULT_COLOR $SORT_METHOD --skip-requirements-check --yes
}

uninstall() {
  if [ -e SSRSpeedN ]; then
    REQS=$(sed "/^$/d" SSRSpeedN/requirements.txt)
    REQS="${REQS//[[:space:]]/, }"
    warning "\n $(text 20)\n $REQS " && reading " $(text 23) " UNINSTALL_REQS
    cd SSRSpeedN || exit 1
    [[ "$UNINSTALL_REQS" =~ [Yy] ]] && sudo pip3 uninstall -yr requirements.txt
    cd ..
    sudo rm -rf SSRSpeedN
    info " $(text 24) "
    exit 0
  else
    error " $(text 25) "
  fi
}

## Main ##

# 传参 1/2
[[ "$*" =~ -[Ee] ]] && L=E
[[ "$*" =~ -[Cc] ]] && L=C

select_language

# 传参 2/2
while getopts ":HhUuR:r:" OPTNAME; do
  case "$OPTNAME" in
  [Hh]) help ;;
  [Uu]) uninstall ;;
  [Rr]) URL=$OPTARG ;;
  esac
done

check_operating_system
input
check_dependencies
check_ssrspeedn
test