alias l='ls -lhF'
alias la='ls -AlhF'
alias ll=l
alias open=explorer

export LC_ALL=zh_CN.utf-8
export http_proxyserver="http://127.0.0.1:10809"

function set_proxy() {
  if [[ $1 == "on" ]]; then
    export HTTP_PROXY=$http_proxyserver
    export HTTPS_PROXY=$http_proxyserver
  elif [[ $1 == "off" ]]; then
    unset HTTP_PROXY HTTPS_PROXY
  fi

  echo
  echo "HTTP_PROXY=$HTTP_PROXY"
  echo "HTTPS_PROXY=$HTTPS_PROXY"
  echo
}

function c3tool() {
  # check `cmake`
  if command -v cmake &>/dev/null; then
    echo "[CHECK] cmake 已安装"
  else
    echo "[CHECK] FAIL: cmake 未安装，请先安装 cmake"
    return 0
  fi

  if command -v mingw32-make &>/dev/null; then
    echo "[CHECK] mingw32-make 已安装"
  else
    echo "[CHECK] FAIL: mingw32-make 未安装，请先安装 mingw32-make"
    return 0
  fi

  # check if augustus or julius build folder
  if [[ $PWD =~ "augustus" || $PWD =~ "julius" ]]; then
    # 判断 basename 是否为 build
    if [ "$(basename "$PWD")" = "build" ]; then
      echo "[CHECK] Working directory: $PWD"
    else
      echo "[CHECK] FAIL: 进入 'augustus' 或者 'julius' 的 'build' 目录"
      return 1
    fi
  else
    echo "[CHECK] FAIL: 进入 'augustus' 或者 'julius' 的 'build' 目录"
    return 1
  fi

  echo

  cmd=""
  if [[ $1 == "" || "$1" == -* ]]; then
    cmd="build"
  else
    cmd=$1
    shift
  fi

  force=false
  update_res=false
  update_libs=false

  # Parse args...
  for arg in "$@"; do
    case "$arg" in
    -r | --update-resources)
      update_res=true
      ;;
    -l | --update-libs)
      update_libs=true
      ;;
    -f | --force)
      force=true
      ;;
    -a | --all)
      force=true
      update_res=true
      update_libs=true
      ;;
    --release)
      export CMAKE_BUILD_TYPE=release
      ;;
    esac
  done

  # Execute cmd...
  case "$cmd" in
  b | build)
    if $force; then
      printf "[BUILD] 清理make文件:" && rm -rf CMakeCache.txt CMakeFiles Makefile cmake_install.cmake compile_commands.json augustus* .cmake &&
        echo "Done"
    fi

    printf "evn: - CMAKE_BUILD_TYPE=$CMAKE_BUILD_TYPE\n\n"

    # cmake
    if [[ ! -f "Makefile" ]]; then
      exec_str="cmake .. -G \"MinGW Makefiles\"" && eval $exec_str
      if [ $? -ne 0 ]; then
        echo "[BUILD] FAIL. Execute string: ${exec_str}"
        return 1
      fi
    fi

    # mingw32-make
    echo "开始编译..."
    if command -v nproc &>/dev/null; then
      makenum=$(nproc)
    else
      makenum=2
    fi
    exec_str="mingw32-make -j$makenum" && eval $exec_str
    if [ $? -ne 0 ]; then
      echo "[BUILD] FAIL. Execute string: ${exec_str}"
      return 1
    fi

    if $update_res; then
      printf "[BUILD] 清理res..." && rm -rf assets maps && echo "Done"
    fi

    if $update_libs; then
      printf "[BUILD] 清理libs..." && rm -f SDL2.dll SDL2_mixer.dll && echo "Done"
    fi

    # Updating res:
    if [[ ! -d assets || ! -d maps ]]; then
      printf "[BUILD] Updating assets ..." && cp -r ../res/assets . && echo "Done"
      printf "[BUILD] Updating maps ..." && cp -r ../res/maps . && echo "Done"
    fi

    # Updating libs:
    if [[ ! -f SDL2.dll || ! -f SDL2_mixer.dll ]]; then
      printf "[BUILD] Updating libs ..." && cp ../ext/SDL2/SDL2-2.32.4/x86_64-w64-mingw32/bin/SDL2.dll . &&
        cp ../ext/SDL2/SDL2_mixer-2.8.1/x86_64-w64-mingw32/bin/SDL2_mixer.dll . &&
        echo "Done"
    fi

    echo "[BUILD] success"
    ;;
  c | clean)
    printf "开始清理..."
    rm -rf CMakeCache.txt CMakeFiles Makefile cmake_install.cmake compile_commands.json augustus* &&
      rm -rf assets maps &&
      rm -f SDL2* &&
      echo "Done"
    ;;
  *)
    echo "[Err] Invalid cmd: $cmd"
    echo
    echo "Usage:"
    echo " - c3tool build|b [-r -l -f -a --release]"
    echo " - c3tool clean|c"
    return 1
    ;;
  esac
}

function connect() {
  GREEN='\033[0;32m'
  RED='\033[0;31m'
  BLUE='\033[0;34m'
  YELLOW='\033[0;33m'
  UNDERLINE='\033[4m' # 下划线
  RESET_COLOR='\033[0m'        # No Color
  NC='\033[0m'        # No Color

  STR_BASH="[${GREEN}Bash${RESET_COLOR}]"
  STR_ZSH="[${GREEN}Zsh${RESET_COLOR}]"

  STR_ERR_PREFIX="[${RED}Err${RESET_COLOR}]"
  STR_INFO_PREFIX="[${GREEN}INFO${NC}]"

  if [[ -n "$BASH_VERSION" ]]; then
    printf "$STR_BASH Detected\n\n"
    declare -A servers
  elif [[ -n "$ZSH_VERSION" ]]; then
    printf "$STR_ZSH Detected\n\n"
    typeset -A servers
    setopt KSH_ARRAYS
  else
    echo -e "[${RED}Err${RESET_COLOR}] Running another shell. Only ${STR_BASH} or ${STR_ZSH} supported."
    return
  fi

  servers=(
    # ["yact"]="ssh root@172.50.10.83"
    ["yacloud"]="ssh -p 7189 gaohengyi@jumpserver.yacloud.net"
    ["u1"]="ssh root@u1"
    ["dev"]="ssh root@125.64.33.85"
    ["tw"]="ssh root@tw.xmdgg.xyz"
    ["hk"]="ssh root@hk.xmdgg.xyz"
    ["us"]="ssh root@us.xmdgg.xyz"
    # ["us1"]="ssh root@us.xmdgg.xyz"
    # ["us2"]="ssh root@us.xmdgg.xyz"
    # ["us3"]="ssh root@us.xmdgg.xyz"
    # ["us4"]="ssh root@us.xmdgg.xyz"
    # ["us5"]="ssh root@us.xmdgg.xyz"
    # ["us6"]="ssh root@us.xmdgg.xyz"
    # ["us7"]="ssh root@us.xmdgg.xyz"
    # ["us8"]="ssh root@us.xmdgg.xyz"
    # ["us9"]="ssh root@us.xmdgg.xyz"
    # ["us10"]="ssh root@us.xmdgg.xyz"
  )

  servers_index=()
  if [[ -n "$BASH_VERSION" ]]; then
    for key in "${!servers[@]}"; do
      servers_index+=("$key")
    done
  elif [[ -n "$ZSH_VERSION" ]]; then
    for key in "${(k)servers[@]}"; do
      servers_index+=("$key")
    done
  else
    return
  fi


  show_help=false
  server_name=""
  if [ -z "$1" ] || [ "$1" = "-h" ] || [ "$1" = "--help" ]; then
    show_help=true
  elif [[ $1 =~ ^[0-9]+$ ]]; then
    server_index=$1
    server_name=${servers_index[$server_index]}
    if [ -z $server_name ]; then
      printf "${STR_ERR_PREFIX} Invalid index: $server_index\n\n"
      show_help=true
    fi
  else
    server_name=$1
    if [[ " ${servers_index[*]} " != *" $server_name "* ]]; then
      printf "${STR_ERR_PREFIX} Invalid name: $server_name"
      show_help=true
    fi
  fi

  if $show_help; then
    echo -e "Usage:\n"
    echo -e " - connect [${BLUE}ID${NC}] | [${BLUE}NAME${NC}]"
    echo -e "\nServers:\n"
    if [[ -n "$BASH_VERSION" ]]; then
      for i in ${!servers_index[@]}; do
        name=${servers_index[$i]}
        ssh_cmd=${servers[$name]}
        printf "${YELLOW}%2d${RESET_COLOR}|" $i
        printf "%-8s -> " $name
        printf "%s\n" "$ssh_cmd"
      done
    elif [[ -n "$ZSH_VERSION" ]]; then
      length=${#servers_index[@]}
      max=$((length-1))
      for i in {0..${max}}; do
        name=${servers_index[$i]}
        ssh_cmd=${servers[$name]}
        printf "${YELLOW}%2d${RESET_COLOR}|" $i
        printf "%-8s -> " $name
        printf "%s\n" "$ssh_cmd"
      done
      echo
    fi

    return
  fi

  ssh_cmd=${servers[$server_name]}
  echo -e "${STR_INFO_PREFIX} Connect to Server:"
  echo -e "${STR_INFO_PREFIX} NAME: $server_name"
  echo -e "${STR_INFO_PREFIX} CMD: ${YELLOW}$ssh_cmd${NC}"
  echo -e "${STR_INFO_PREFIX} Connecting..."
  err_msg=$(eval $ssh_cmd 2>&1 >/dev/tty)
  if [[ $? != 0 ]]; then
    echo -e "${STR_ERR_PREFIX} $err_msg"
  fi
  echo
}