vkit.sh

#!/usr/bin/env bash
# bash <(curl -Lso- https://sh.vps.dance/vkit.sh)

# Colors
RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[0;33m'; BLUE='\033[34m'; CYAN='\033[0;36m'; PURPLE='\033[35m'; BOLD='\033[1m'; NC='\033[0m';
success() { printf "${GREEN}%b${NC} ${@:2}\n" "$1"; }
info() { printf "${CYAN}%b${NC} ${@:2}\n" "$1"; }
danger() { printf "\n${RED}[x] %b${NC}\n" "$@"; }
warn() { printf "${YELLOW}%b${NC}\n" "$@"; }

# Keep stdio on tty for interactive reads when piped
read_tty() { read -p "$1" "$2" </dev/tty; }

OS=$(uname -s) # Linux, FreeBSD, Darwin
ARCH=$(uname -m) # x86_64, arm64, aarch64
DISTRO=$( ([[ -e "/usr/bin/yum" ]] && echo 'CentOS') || ([[ -e "/usr/bin/apt" ]] && echo 'Debian') || echo 'unknown' )
debug=$( [[ $OS == "Darwin" ]] && echo true || echo false )
cnd=$( tr '[:upper:]' '[:lower:]' <<<"$1" )

SH_PORT="${SH_PORT:-}"; SH="https://sh.vps.dance${SH_PORT:+:${SH_PORT}}"
GH='https://ghfast.top'

CURR_USER="$(whoami)"
# with_sudo func; with_sudo ls /root;
with_sudo() {
  # check sudo availability
  if ! command -v sudo >/dev/null 2>&1; then
    warn "Error: sudo command not found"
    return 1
  fi

  local cmd
  if [[ "$(type -t "$1")" == "function" ]]; then
    local declare_vars="$(declare -p CURR_USER OS ARCH DISTRO debug cnd SH GH RED GREEN YELLOW BLUE CYAN PURPLE BOLD NC 2>/dev/null)"
    local declare_funcs="$(declare -f)"
    cmd="$declare_vars; $declare_funcs; $1 "'"${@:2}"'
  else
    cmd="$1 "'"${@:2}"' # cmd="$1 "'"${@:2}"'
  fi

  if [[ $EUID -ne 0 ]]; then
    # sudo bash -c "$cmd" -- "$@" 2> >(grep -v "unable to resolve host" >&2)
    sudo bash -c "$cmd" -- "$@" < /dev/tty
  else
    bash -c "$cmd" -- "$@"
  fi
}

raw() {
  RAW='https://raw.githubusercontent.com'
  if [[ "$cnd" =~ ^(ghproxy)$ ]]; then echo "${GH}/${RAW}"
  elif [[ "${1}" =~ ^(ghproxy)$ ]]; then echo "${GH}/${RAW}"
  elif [[ "${1}" =~ ^(sh)$ ]]; then echo "${SH}/raw"
  else echo $RAW
  fi
}
# echo $(raw 'ghproxy')
# curl -Ls "$(raw '')/VPSDance/vkit/main/ssh.sh"

run_remote() {
  local url="$1"; shift
  # Suppress stty errors when stdin isn't a tty (e.g., curl | bash)
  bash -c 'stty(){ command stty "$@" 2>/dev/null || true; }; source <(curl -Lso- "$1") "${@:2}"' -- "$url" "$@"
}

next() { printf "%-37s\n" "-" | sed 's/\s/-/g'; }

# if (ver_lte 3 3.0); then echo 3; else echo 2; fi # ver_lte 2.5.7 3 && echo "yes" || echo "no"
ver_lte() { # <=
  [  "$(printf '%s\n' "$@" | sort -V | head -n 1)" = "$1" ] && return 0 || return 1
}
# if (ver_lt 2.9 3); then echo 2; else echo 3; fi
ver_lt() { # <
  [ "$1" = "$2" ] && return 1 || ver_lte "$1" "$2"
}
python_version() {
  python -V 2>&1 | awk '{print $2}' # | awk -F '.' '{print $1}'
}

header() {
  next
  printf "%s\n" "[vkit] 目前支持: Ubuntu/Debian, Centos/Redhat"
  printf "%b\n" "${GREEN}VPS/IPLC测评:${NC} ${YELLOW}https://vps.dance/${NC}"
  printf "%b\n" "${GREEN}Telegram频道:${NC} ${YELLOW}https://t.me/vpsdance${NC}"
  next
}
footer() {
  BLUE="\033[34m"; NC='\033[0m'
  printf "%-37s\n" "-" | sed 's/\s/-/g'
  printf "%b\n" " Supported by: ${BLUE}https://vps.dance${NC}"
  printf "%-37s\n" "-" | sed 's/\s/-/g'
}

install_deps() {
  case "${DISTRO}" in
    Debian*|Ubuntu*)
      apt update -y;
      apt install -y curl wget htop zip unzip xz-utils gzip ca-certificates; # zip unzip xz-utils gzip
      # ifconfig/netstat dig/nslookup ping nc
      apt install -y net-tools dnsutils iputils-ping mtr traceroute telnet tcpdump netcat-openbsd; 
      apt install -y nmap; # nping
      apt install -y python3 python3-pip;
    ;;
    CentOS*|RedHat*)
      yum update -y;
      yum install -y epel-release which openssl curl wget htop zip unzip xz gzip ca-certificates;
      yum install -y net-tools bind-utils iputils mtr traceroute telnet tcpdump nc;
      yum install -y nmap;
      yum install -y python3 python3-pip;
    ;;
  esac
}
install_bbr() {
  local url
  url="$(raw 'sh')/teddysun/across/master/bbr.sh"
  info "bash <(curl -Lso- ${url})"
  run_remote "$url"
}
ssh_key() { bash <(curl -Lso- ${SH}/ssh.sh); }
bashrc() { bash <(curl -Lso- ${SH}/bashrc.sh); }
tuning() { bash <(curl -Lso- ${SH}/tuning.sh); }
ssh_port() { bash <(curl -Lso- ${SH}/ssh.sh) port; }
add_swap() { bash <(curl -Lso- ${SH}/swap.sh); }
ip46() { bash <(curl -Lso- ${SH}/ip46.sh); }
install_tool() {
  bash <(curl -Lso- ${SH}/tools.sh) "$@"
}
install_xray() {
  bash <(curl -fsSL $(raw 'sh')/XTLS/Xray-install/main/install-release.sh) install
  # 使用增强版的 geosite/geoip 规则
  wget -q -O /usr/local/share/xray/geoip.dat ${GH}/https://github.com/Loyalsoldier/v2ray-rules-dat/releases/latest/download/geoip.dat
  wget -q -O /usr/local/share/xray/geosite.dat ${GH}/https://github.com/Loyalsoldier/v2ray-rules-dat/releases/latest/download/geosite.dat
  info "configuration file: /usr/local/etc/xray/config.json"
  info "status: systemctl status xray"
  info "restart: systemctl restart xray"
}
install_wrap() {
  bash <(curl -fsSL https://gitlab.com/fscarmen/warp/-/raw/main/menu.sh)
}
install_wireguard(){
  curl -Ls $(raw 'ghproxy')/teddysun/across/master/wireguard.sh | bash -s -- -r
  # uninstall_wireguard
  # curl -Ls $(raw 'ghproxy')/teddysun/across/master/wireguard.sh | bash -s -- -n
}
unlock_test() {
  info "bash <(curl -L -s check.unlock.media)"
  bash <(curl -Lso- ${SH}/unlockTest.sh)
}
tiktok_test() {
  info "bash <(curl -Lso- "$(raw '')/lmc999/TikTokCheck/main/tiktok.sh")"
  bash <(curl -Lso- "$(raw 'sh')/lmc999/TikTokCheck/main/tiktok.sh")
}
ipquality() {
  info "bash <(curl -Ls IP.Check.Place)"
  bash <(curl -Ls IP.Check.Place) -y
}
netquality() {
  info "bash <(curl -Ls Net.Check.Place)"
  bash <(curl -Ls Net.Check.Place) -y
}
bench() { bash <(curl -Lso- ${SH}/bench.sh); }
# https://github.com/veoco/bim-core/
# hyperspeed() { bash <(curl -Lso- https://bench.im/hyperspeed); }
iabc_speedtest() { bash <(curl -sL https://jihulab.com/i-abc/Speedtest/-/raw/main/speedtest.sh); }
# lemon_bench() { curl -fsSL http://ilemonra.in/LemonBenchIntl | bash -s fast; }
yabs() {
  while :; do
    read_tty "输入Geekbench 版本 [默认=6, 可选:4/5/6]: " BENCH_VER
    BENCH_VER=${BENCH_VER:-6}
    [[ $BENCH_VER =~ ^[0-9]+$ ]] || {
      echo "invalid number"
      continue
    }
    break
  done
  local cmd="curl -sL yabs.sh | bash -s -- -$BENCH_VER"
  local AR=(
    [1]='仅Benchmark'
    [2]='仅fio'
    [3]='仅iperf'
    [4]='Benchmark+fio+iperf'
  )
  clear;
  info "Geekbench 版本 $BENCH_VER, 请选择测试目标: "
  for i in "${!AR[@]}"; do
    success "$i." "${AR[i]}"
  done
  while :; do
    read_tty "输入数字以选择: " snum
    [[ -n "${AR[snum]}" ]] || {
      danger "invalid"
      continue
    }
    break
  done
  if [[ "$snum" == "1" ]]; then cmd="$cmd -din"
  elif [[ "$snum" == "2" ]]; then cmd="$cmd  -ign" $log
  elif [[ "$snum" == "3" ]]; then cmd="$cmd  -dgn" $log
  elif [[ "$snum" == "4" ]]; then cmd="$cmd  -n" $log
  fi
  clear;
  info "$cmd";
  with_sudo bash -c "$cmd"
}
besttrace() { bash <(curl -Lso- ${SH}/autoBestTrace.sh); }
nexttrace() { bash <(curl -Lso- ${SH}/autoNexttrace.sh); }
unix_bench() { bash <(curl -Lso- $(raw 'ghproxy')/teddysun/across/master/unixbench.sh); }
reinstall() { bash <(curl -Lso- $(raw 'ghproxy')/hiCasper/Shell/master/AutoReinstall.sh); }
uninstall() {
  app="$1"
  if [[ "$app" == "hy2" ]]; then
    app="hysteria-server"
  fi
  un_service() { systemctl disable $app --now; rm -rf "/etc/systemd/system/$app.service"; }
  case "$app" in
    xray)
      un_service;
      rm -rf /etc/systemd/system/xray@.service /usr/local/bin/xray
      rm -rf /usr/local/etc/xray/ /usr/local/share/xray/ /var/log/xray/
    ;;
    ss)
      un_service;
      rm -rf /root/ss.json /usr/bin/ssserver /usr/bin/sslocal /usr/bin/ssurl /usr/bin/ssmanager /usr/bin/ssservice
    ;;
    snell)
      un_service;
      rm -rf /root/snell.conf /usr/bin/snell-server
    ;;
    realm)
      un_service;
      rm -rf /root/realm.toml /usr/bin/realm
    ;;
    gost)
      un_service;
      rm -rf /root/gost.yaml /root/gost.json /usr/bin/gost
    ;;
    hysteria-server)
      un_service;
      rm -rf /etc/hysteria/ /usr/local/bin/hysteria
    ;;
    nali)
      rm -rf ~/.config/nali ~/.local/share/nali /usr/bin/nali
    ;;
    ddns-go)
      un_service;
      rm -rf /root/ddns-go.yaml /usr/bin/ddns-go
    ;;
    miniserve)
      un_service;
      rm -rf /usr/bin/miniserve
    ;;
    nexttrace)
      rm -rf /usr/bin/nexttrace
    ;;
   *)
    echo "$@"; exit;
   ;;
  esac
  echo -e "\n${GREEN}Done${NC}" 
}

menu() {
  header
  info "请选择要使用的功能"
  local AR=(
    [1]="[推荐] 配置SSH Public Key (SSH免密登录)"
    [2]="[推荐] 终端优化 (颜色美化/上下键查找历史)"
    [3]="[推荐] 安装并开启 BBR"
    [4]="[推荐] 系统优化 (TCP网络优化/资源限制优化)"
    [5]="[推荐] 安装常用软件 (mtr/traceroute/nping/nc/tcpdump/python3)"
    [6]="[推荐] 修改默认SSH Port端口 (减少被扫描风险)"
    [7]="增加 swap 分区 (虚拟内存)"
    [8]="调整 IPv4/IPv6 优先级, 启用/禁用IPv6"
    [10]="安装/卸载工具 (xray/ss/hy2/realm/ddns-go/...)"
    [11]="使用 CF WARP 添加 IPv4/IPv6 网络"
    # [18]="安装 wireguard"
    [20]="检测 IP质量 (IPQuality)"
    [21]="检测 VPS流媒体解锁 (RegionRestrictionCheck)"
    [22]="检测 VPS信息/IO/网速 (Bench.sh)"
    [23]="检测 性能/IO (YABS)"
    [24]="检测 单线程/多线程网速 (i-abc/Speedtest)"
    [25]="检测 网络质量 (NetQuality)"
    [26]="检测 回程路由 (BestTrace)"
    [27]="检测 回程路由 (NextTrace)"
    [28]="检测 Tiktok解锁 (TikTokCheck)"
    # [23]="检测 VPS到国内网速 (Superspeed)"
    # [23]="检测 VPS到国内网速 (HyperSpeed)"
    # [25]="检测 VPS信息/IO/路由 (LemonBench)"
    # [29]="性能测试 (UnixBench)"
    # [31]="DD重装Linux系统"
    # [100]=""
  )
  for i in "${!AR[@]}"; do
    success "$i." "${AR[i]}"
  done

  while :; do
    read_tty "输入数字以选择:" num
    [[ $num =~ ^[0-9]+$ ]] || { danger "请输入正确的数字"; continue; }
    break
  done
  main="${AR[num]}"
  
  if [[ "$num" == "10" ]]; then
    clear
    tools_menu
    return
  fi
}

# 工具二级菜单
tools_menu() {
  info "安装/卸载工具，请选择:"
  local ToolsAR=(
    [0]="返回上级菜单"
    [1]="xray"
    [2]="shadowsocks"
    [3]="snell 3"
    [4]="hysteria 2"
    [5]="realm (端口转发工具)"
    [6]="gost (隧道/端口转发工具)"
    [7]="ddns-go (DDNS工具)"
    [8]="nali (IP查询工具)"
    [9]="miniserve (HTTP 文件服务器)"
    [10]="nexttrace (路由测试)"
  )
  for i in "${!ToolsAR[@]}"; do
    success "$i." "${ToolsAR[i]}"
  done

  while :; do
    read_tty "输入数字以选择工具:" tool_num
    [[ $tool_num =~ ^[0-9]+$ ]] || { danger "请输入正确的数字"; continue; }
    [[ -n "${ToolsAR[tool_num]}" ]] || { danger "无效的选项"; continue; }
    break
  done
  
  if [[ "$tool_num" == "0" ]]; then
    clear
    menu
    return
  fi
  
  local tool_name="${ToolsAR[tool_num]}"
  tool_name=${tool_name%% (*}  # 移除括号内的描述
  clear
  info "$tool_name, 请选择:"
  local ActionAR=(
    [1]='安装'
    [2]='卸载'
    [0]='返回'
  )
  for i in "${!ActionAR[@]}"; do
    success "$i." "${ActionAR[i]}"
  done
  
  while :; do
    read_tty "输入操作选择:" action_num
    [[ $action_num =~ ^[0-9]+$ ]] || { danger "请输入正确的数字"; continue; }
    [[ -n "${ActionAR[action_num]}" ]] || { danger "无效的选项"; continue; }
    break
  done
  
  if [[ "$action_num" == "0" ]]; then
    clear
    tools_menu
    return
  fi
  case "$tool_num" in
    1)
      [[ "$action_num" == "1" ]] && with_sudo install_xray || with_sudo uninstall "xray"
      ;;
    2)
      [[ "$action_num" == "1" ]] && install_tool "ss" || with_sudo uninstall "ss"
      ;;
    3)
      [[ "$action_num" == "1" ]] && install_tool "snell" || with_sudo uninstall "snell"
      ;;
    4)
      [[ "$action_num" == "1" ]] && install_tool "hy2" || with_sudo uninstall "hy2"
      ;;
    5)
      [[ "$action_num" == "1" ]] && install_tool "realm" || with_sudo uninstall "realm"
      ;;
    6)
      [[ "$action_num" == "1" ]] && install_tool "gost" || with_sudo uninstall "gost"
      ;;
    7)
      [[ "$action_num" == "1" ]] && install_tool "ddns-go" || with_sudo uninstall "ddns-go"
      ;;
    8)
      [[ "$action_num" == "1" ]] && install_tool "nali" || with_sudo uninstall "nali"
      ;;
    9)
      [[ "$action_num" == "1" ]] && install_tool "miniserve" || with_sudo uninstall "miniserve"
      ;;
    10)
      [[ "$action_num" == "1" ]] && install_tool "nexttrace" || with_sudo uninstall "nexttrace"
      ;;
  esac
  exit;
}

main() {
  clear
  if [[ "$num" == "1" ]]; then ssh_key
  elif [[ "$num" == "2" ]]; then bashrc
  elif [[ "$num" == "3" ]]; then with_sudo install_bbr
  elif [[ "$num" == "4" ]]; then tuning
  elif [[ "$num" == "5" ]]; then with_sudo install_deps
  elif [[ "$num" == "6" ]]; then ssh_port
  elif [[ "$num" == "7" ]]; then add_swap
  elif [[ "$num" == "8" ]]; then ip46
  # 10 tools
  elif [[ "$num" == "11" ]]; then install_wrap
  elif [[ "$num" == "20" ]]; then ipquality
  elif [[ "$num" == "21" ]]; then unlock_test
  elif [[ "$num" == "22" ]]; then with_sudo bench
  elif [[ "$num" == "23" ]]; then yabs
  elif [[ "$num" == "24" ]]; then with_sudo iabc_speedtest
  elif [[ "$num" == "25" ]]; then netquality
  elif [[ "$num" == "26" ]]; then with_sudo besttrace
  elif [[ "$num" == "27" ]]; then with_sudo nexttrace
  elif [[ "$num" == "28" ]]; then tiktok_test
  # elif [[ "$num" == "29" ]]; then unix_bench
  # elif [[ "$num" == "31" ]]; then reinstall
  else exit
  fi
}

menu
main