#!/usr/bin/env bash

CDN=https://cdn.jsdelivr.net/gh/etherdream/jsproxy-bin@master/

JSPROXY_VER=0.0.2
PCRE_VER=8.43
ZLIB_VER=1.2.11
OPENSSL_VER=1.1.1b
OPENRESTY_VER=1.15.8.1
BROTLI_VER=1.0.7

SUPPORTED_OS="Linux-x86_64"
OS="$(uname)-$(uname -m)"
NGX_DIR="$HOME/openresty"

COLOR_RESET="\033[0m"
COLOR_RED="\033[31m"
COLOR_GREEN="\033[32m"
COLOR_YELLOW="\033[33m"

output() {
  color=$1
  shift 1
  sdata="$@"
  stime=$(date "+%H:%M:%S")
  printf "$color[jsproxy $stime]$COLOR_RESET $sdata\n"
}
log() {
  output $COLOR_GREEN $1
}
warn() {
  output $COLOR_YELLOW $1
}
err() {
  output $COLOR_RED $1
}

check_nginx() {
  NGX_EXE="$NGX_DIR/nginx/sbin/nginx"
  NGX_VER=$($NGX_EXE -v 2>&1)

  if [[ "$NGX_VER" != *"nginx version:"* ]]; then
    err "$NGX_EXE 无法执行！尝试编译安装"
    exit 1
  fi
  echo "nginx 安装完成。版本: $NGX_VER 路径: $NGX_DIR"
}

install_jsproxy() {
  log "下载 jsproxy ..."
  curl -s -O $CDN/server-$JSPROXY_VER.tar.gz
  tar zxf server-$JSPROXY_VER.tar.gz
  rm -f server-$JSPROXY_VER.tar.gz

  if [ -d "server" ]; then
    backup=bak_$(date +%Y_%m_%d_%H_%M_%S)
    warn "当前 server 目录备份到 $backup"
    mkdir -p $backup
    mv server $backup
  fi

  mv server-$JSPROXY_VER server

  log "启动服务 ..."
  ./server/run.sh

  log "检测状态 ..."
  RET=$(curl -s http://127.0.0.1:8080/)

  ERR_LOG=$PWD/server/logs/error.log

  if [[ "$RET" != *"origin"* ]]; then
    err "服务启动异常。错误日志:"
    tail $ERR_LOG -n100
    exit 1
  fi

  log "服务已开启。后续维护参考 https://github.com/EtherDream/jsproxy"
}

compile() {
  TMP_DIR="$PWD/__tmp__"

  mkdir -p $TMP_DIR
  cd $TMP_DIR

  log "下载 pcre 源码 ..."
  curl -O https://ftp.pcre.org/pub/pcre/pcre-$PCRE_VER.tar.gz
  tar zxf pcre-$PCRE_VER.tar.gz

  log "下载 zlib 源码 ..."
  curl -O https://zlib.net/zlib-$ZLIB_VER.tar.gz
  tar zxf zlib-$ZLIB_VER.tar.gz

  log "下载 openssl 源码 ..."
  curl -O https://www.openssl.org/source/openssl-$OPENSSL_VER.tar.gz
  tar zxf openssl-$OPENSSL_VER.tar.gz

  log "下载 nginx 源码 ..."
  curl -O https://openresty.org/download/openresty-$OPENRESTY_VER.tar.gz
  tar zxf openresty-$OPENRESTY_VER.tar.gz

  cd openresty-$OPENRESTY_VER

  export PATH=$PATH:/sbin

  log "配置中 ..."
  ./configure \
    --with-openssl=../openssl-$OPENSSL_VER \
    --with-pcre=../pcre-$PCRE_VER \
    --with-zlib=../zlib-$ZLIB_VER \
    --with-http_v2_module \
    --with-http_ssl_module \
    --with-pcre-jit \
    --prefix=$NGX_DIR

  log "编译中 ..."
  make
  make install

  log "编译完成"
  rm -rf $TMP_DIR

  check_nginx
  install_jsproxy
}

install() {
  log "下载 nginx ..."
  curl -s -O $CDN/$OS/openresty-$OPENRESTY_VER.tar.gz
  tar zxf openresty-$OPENRESTY_VER.tar.gz
  rm -f openresty-$OPENRESTY_VER.tar.gz

  check_nginx
  install_jsproxy
}

update() {
  if [ ! -d "server" ]; then
    echo "当前不存在 server 目录，切换到主目录再更新"
    exit 1
  fi
  install_jsproxy
}

pack() {
  log "压缩 openresty ..."
  GZIP=-9
  tar cvzf openresty.tar.gz openresty
}

main() {
  if [[ "$SUPPORTED_OS" != *"$OS"* ]]; then
    err "当前系统 $OS 不支持自动安装。尝试编译安装"
    exit 1
  fi

  if [[ "$USER" != "root" ]]; then
    err "自动安装需要 root 权限。如果无法使用 root，尝试编译安装"
    exit 1
  fi

  log "创建用户 jsproxy ..."
  groupadd nobody
  useradd jsproxy -g nobody --create-home

  log "切换用户 jsproxy，执行安装脚本 ..."
  chmod +x $0
  cp $0 /home/jsproxy/i.sh
  su - jsproxy -c "/home/jsproxy/i.sh install"
}


case "$1" in
"install") install
  exit;;
"compile") compile
  exit;;
"pack") pack
  exit;;
"update") update
  exit;;
*) main
  exit;;
esac