#!/bin/bash
if [[ $EUID -ne 0 ]]; then
    clear
    echo "错误：本脚本需要 root 权限执行。" 1>&2
    exit 1
fi

check_sys(){
    if [[ -f /etc/redhat-release ]]; then
      release="centos"
    elif cat /etc/issue | grep -q -E -i "debian"; then
      release="debian"
    elif cat /etc/issue | grep -q -E -i "ubuntu"; then
      release="ubuntu"
    elif cat /etc/issue | grep -q -E -i "centos|red hat|redhat"; then
      release="centos"
    elif cat /proc/version | grep -q -E -i "debian"; then
      release="debian"
    elif cat /proc/version | grep -q -E -i "ubuntu"; then
      release="ubuntu"
    elif cat /proc/version | grep -q -E -i "centos|red hat|redhat"; then
      release="centos"
          fi
}

welcome() {
  echo ""
  echo "欢迎使用 PagerMaid-Modify 一键安装程序。"
  echo "安装即将开始"
  echo "如果您想取消安装，"
  echo "请在 5 秒钟内按 Ctrl+C 终止此脚本。"
  echo ""
  sleep 5
}

yum_update(){
	echo "正在优化 yum . . ."
	yum install yum-utils epel-release -y >> /dev/null 2>&1
}

yum_git_check() {
  echo "正在检查 Git 安装情况 . . ."
  if command -v git >> /dev/null 2>&1;
  then
    echo "Git 似乎存在，安装过程继续 . . ."
  else
    echo "Git 未安装在此系统上，正在进行安装"
    yum install git -y >> /dev/null 2>&1
  fi
}

yum_python_check() {
	echo "正在检查 python 安装情况 . . ."
	if command -v python3 >> /dev/null 2>&1;then
		U_V1=`python3 -V 2>&1|awk '{print $2}'|awk -F '.' '{print $1}'`
        U_V2=`python3 -V 2>&1|awk '{print $2}'|awk -F '.' '{print $2}'`
        if [ $U_V1 -gt 3 ];then
	    	echo 'Python 3.6+ 存在 . . .'
	    elif [ $U_V2 -ge 6 ];then
	    	echo 'Python 3.6+ 存在 . . .'
	    	PYV=$U_V1.$U_V2
	        PYV=`which python$PYV`
	    else
	    	if command -v python3.6 >> /dev/null 2>&1;then
	    		echo 'Python 3.6+ 存在 . . .'
	    		PYV=`which python3.6`
	    	else
	    	    echo "Python3.6 未安装在此系统上，正在进行安装"
	    	    yum install python3 -y >> /dev/null 2>&1
	    	    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.6 1 >> /dev/null 2>&1
	    	    PYV=`which python3.6`
	    	fi
	    fi
	else
		echo "Python3.6 未安装在此系统上，正在进行安装"
		yum install python3 -y >> /dev/null 2>&1
		update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.6 1 >> /dev/null 2>&1
	fi
	if command -v pip3 >> /dev/null 2>&1;then
		echo 'pip 存在 . . .'
	else
		echo "pip3 未安装在此系统上，正在进行安装"
		yum install -y python3-pip >> /dev/null 2>&1
	fi
}

yum_screen_check() {
  echo "正在检查 Screen 安装情况 . . ."
  if command -v screen >> /dev/null 2>&1;
  then
    echo "Screen 似乎存在, 安装过程继续 . . ."
  else
    echo "Screen 未安装在此系统上，正在进行安装"
    yum install screen -y >> /dev/null 2>&1
  fi
}

yum_require_install(){
	echo "正在安装系统所需依赖，可能需要几分钟的时间 . . ."
	yum install python-devel python3-devel zbar zbar-devel ImageMagick wget -y >> /dev/null 2>&1
	wget -T 2 -O /etc/yum.repos.d/konimex-neofetch-epel-7.repo https://copr.fedorainfracloud.org/coprs/konimex/neofetch/repo/epel-7/konimex-neofetch-epel-7.repo >> /dev/null 2>&1
	yum groupinstall "Development Tools" -y >> /dev/null 2>&1
    yum-config-manager --add-repo https://download.opensuse.org/repositories/home:/Alexander_Pozdnyakov/CentOS_7/ >> /dev/null 2>&1
    sudo rpm --import https://build.opensuse.org/projects/home:Alexander_Pozdnyakov/public_key >> /dev/null 2>&1
    yum list updates >> /dev/null 2>&1
    yum install neofetch figlet tesseract tesseract-langpack-chi-sim tesseract-langpack-eng -y >> /dev/null 2>&1
}

apt_update(){
	echo "正在优化 apt-get . . ."
	apt-get install sudo -y >> /dev/null 2>&1
	apt-get update >> /dev/null 2>&1
}

apt_git_check() {
  echo "正在检查 Git 安装情况 . . ."
  if command -v git >> /dev/null 2>&1;
  then
    echo "Git 似乎存在, 安装过程继续 . . ."
  else
    echo "Git 未安装在此系统上，正在进行安装"
    apt-get install git -y >> /dev/null 2>&1
  fi
}

apt_python_check() {
	echo "正在检查 python 安装情况 . . ."
	if command -v python3 >> /dev/null 2>&1;then
        U_V1=`python3 -V 2>&1|awk '{print $2}'|awk -F '.' '{print $1}'`
        U_V2=`python3 -V 2>&1|awk '{print $2}'|awk -F '.' '{print $2}'`
        if [ $U_V1 -gt 3 ];then
	    	echo 'Python 3.6+ 存在 . . .'
	    elif [ $U_V2 -ge 6 ];then
	    	echo 'Python 3.6+ 存在 . . .'
	    	PYV=$U_V1.$U_V2
	        PYV=`which python$PYV`
	    else
	    	if command -v python3.6 >> /dev/null 2>&1;then
	    		echo 'Python 3.6+ 存在 . . .'
	    		PYV=`which python3.6`
	    	else
	    	    echo "Python3.6 未安装在此系统上，正在进行安装"
	    	    add-apt-repository ppa:deadsnakes/ppa
	    	    apt-get update >> /dev/null 2>&1
	        	apt-get install python3.6 -y >> /dev/null 2>&1
	    	    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.6 1 >> /dev/null 2>&1
	    	    PYV=`which python3.6`
	    	fi
	    fi
	else
		echo "Python3.6 未安装在此系统上，正在进行安装"
		add-apt-repository ppa:deadsnakes/ppa
		apt-get update >> /dev/null 2>&1
		apt-get install python3.6 -y >> /dev/null 2>&1
		update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.6 1 >> /dev/null 2>&1
	fi
	if command -v pip3 >> /dev/null 2>&1;then
		echo 'pip 存在 . . .'
	else
		echo "pip3 未安装在此系统上，正在进行安装"
		apt-get install -y python3-pip >> /dev/null 2>&1
	fi
}

debian_python_check() {
	echo "正在检查 python 安装情况 . . ."
	if command -v python3 >> /dev/null 2>&1;then
        U_V1=`python3 -V 2>&1|awk '{print $2}'|awk -F '.' '{print $1}'`
        U_V2=`python3 -V 2>&1|awk '{print $2}'|awk -F '.' '{print $2}'`
        if [ $U_V1 -gt 3 ];then
	    	echo 'Python 3.6+ 存在 . . .'
	    elif [ $U_V2 -ge 6 ];then
	    	echo 'Python 3.6+ 存在 . . .'
	    	PYV=$U_V1.$U_V2
	        PYV=`which python$PYV`
	    else
	    	if command -v python3.6 >> /dev/null 2>&1;then
	    		echo 'Python 3.6+ 存在 . . .'
	    		PYV=`which python3.6`
	    	else
	    		echo "Python3.6 未安装在此系统上，正在进行安装"
	    		apt-get update -y >> /dev/null 2>&1
	    		apt-get install -y build-essential tk-dev libncurses5-dev libncursesw5-dev libreadline6-dev libdb5.3-dev libgdbm-dev libsqlite3-dev libssl-dev libbz2-dev libexpat1-dev liblzma-dev zlib1g-dev >> /dev/null 2>&1
	    		wget https://www.python.org/ftp/python/3.6.5/Python-3.6.5.tgz  >> /dev/null 2>&1
	    		tar -xvf Python-3.6.5.tgz  >> /dev/null 2>&1
	    		chmod -R +x Python-3.6.5  >> /dev/null 2>&1
	    		cd Python-3.6.5 >> /dev/null 2>&1
	    		./configure >> /dev/null 2>&1
	    		make&& make install >> /dev/null 2>&1
	    		cd .. >> /dev/null 2>&1
	    		rm -rf Python-3.6.5 Python-3.6.5.tar.gz >> /dev/null 2>&1
	    		PYP=`which python3.6`
	    		update-alternatives --install $PYP python3 $PYV 1 >> /dev/null 2>&1
	    	fi
	    fi
	else
		echo "Python3.6 未安装在此系统上，正在进行安装"
		apt-get update -y >> /dev/null 2>&1
        apt-get install -y build-essential tk-dev libncurses5-dev libncursesw5-dev libreadline6-dev libdb5.3-dev libgdbm-dev libsqlite3-dev libssl-dev libbz2-dev libexpat1-dev liblzma-dev zlib1g-dev >> /dev/null 2>&1
        wget https://www.python.org/ftp/python/3.6.5/Python-3.6.5.tgz  >> /dev/null 2>&1
        tar -xvf Python-3.6.5.tgz  >> /dev/null 2>&1
        chmod -R +x Python-3.6.5  >> /dev/null 2>&1
        cd Python-3.6.5 >> /dev/null 2>&1
        ./configure >> /dev/null 2>&1
        make&& make install >> /dev/null 2>&1
        cd .. >> /dev/null 2>&1
        rm -rf Python-3.6.5 Python-3.6.5.tar.gz >> /dev/null 2>&1
        PYP=`which python3`
		update-alternatives --install $PYP python3 $PYV 1 >> /dev/null 2>&1
	fi
	echo "正在检查 pip3 安装情况 . . ."
	if command -v pip3 >> /dev/null 2>&1;then
		echo 'pip 存在 . . .'
	else
		echo "pip3 未安装在此系统上，正在进行安装"
		apt-get install -y python3-pip >> /dev/null 2>&1
	fi
}

apt_screen_check() {
  echo "正在检查 Screen 安装情况 . . ."
  if command -v screen >> /dev/null 2>&1;
  then
    echo "Screen 似乎存在, 安装过程继续 . . ."
  else
    echo "Screen 未安装在此系统上，正在进行安装"
    apt-get install screen -y >> /dev/null 2>&1
  fi
}

apt_require_install(){
	echo "正在安装系统所需依赖，可能需要几分钟的时间 . . ."
	apt-get install python3.6-dev python3-dev imagemagick software-properties-common tesseract-ocr tesseract-ocr-chi-sim libzbar-dev -y >> /dev/null 2>&1
    add-apt-repository ppa:dawidd0811/neofetch
    apt-get install neofetch -y >> /dev/null 2>&1
}

debian_require_install(){
	echo "正在安装系统所需依赖，可能需要几分钟的时间 . . ."
	apt-get install imagemagick software-properties-common tesseract-ocr tesseract-ocr-chi-sim libzbar-dev neofetch -y >> /dev/null 2>&1
}

download_repo() {
  echo "下载 repository 中 . . ."
  rm -rf /var/lib/pagermaid >> /dev/null 2>&1
  git clone https://github.com/xtaodada/PagerMaid-Modify.git /var/lib/pagermaid >> /dev/null 2>&1
  cd /var/lib/pagermaid >> /dev/null 2>&1
  echo "Hello World!">/var/lib/pagermaid/public.lock
}

pypi_install(){
  echo "下载安装 pypi 依赖中 . . ."
  $PYV -m pip install --upgrade pip >> /dev/null 2>&1
  $PYV -m pip install -r requirements.txt >> /dev/null 2>&1
  sudo -H $PYV -m pip install --ignore-installed PyYAML >> /dev/null 2>&1
}

configure() {
  config_file=config.yml
  echo "生成配置文件中 . . ."
  cp config.gen.yml config.yml
  echo "api_key、api_hash 申请地址： https://my.telegram.org/"
  printf "请输入应用程序 api_key："
  read -r api_key <&1
  sed -i "s/ID_HERE/$api_key/" $config_file
  printf "请输入应用程序 api_hash："
  read -r api_hash <&1
  sed -i "s/HASH_HERE/$api_hash/" $config_file
  printf "请输入应用程序语言（默认：zh-cn）："
  read -r application_language <&1
  if [ -z "$application_language" ]
  then
  	echo "语言设置为 简体中文"
  else
    sed -i "s/zh-cn/$application_language/" $config_file
  fi
  printf "请输入应用程序地区（默认：China）："
  read -r application_region <&1
  if [ -z "$application_region" ]
  then
  	echo "地区设置为 中国"
  else
    sed -i "s/China/$application_region/" $config_file
  fi
  printf "请输入 Google TTS 语言（默认：zh-CN）："
  read -r application_tts <&1
  if [ -z "$application_tts" ]
  then
  	echo "tts发音语言设置为 简体中文"
  else
    sed -i "s/zh-CN/$application_tts/" $config_file
  fi
  printf "启用日志记录？ [Y/n]"
  read -r logging_confirmation <&1
  case $logging_confirmation in
      [yY][eE][sS]|[yY])
		    printf "请输入您的日志记录群组/频道的 ChatID （如果要发送给 原 PagerMaid 作者 ，请按Enter）："
		    read -r log_chatid <&1
		    if [ -z "$log_chatid" ]
		    then
		      echo "LOG 将发送到 原 PagerMaid 作者."
		    else
		      sed -i "s/503691334/$log_chatid/" $config_file
		    fi
		    sed -i "s/log: False/log: True/" $config_file
		    ;;
      [nN][oO]|[nN])
		    echo "安装过程继续 . . ."
        ;;
      *)
	  echo "输入错误 . . ."
	  exit 1
	  ;;
  esac
}

login_screen(){
	screen -S userbot -X quit >> /dev/null 2>&1
	screen -dmS userbot
	sleep 1
    screen -x -S userbot -p 0 -X stuff "cd /var/lib/pagermaid && $PYV -m pagermaid"
    screen -x -S userbot -p 0 -X stuff $'\n'
    read -p "请输入您的 Telegram 手机号码: " phonenum
    screen -x -S userbot -p 0 -X stuff "$phonenum"
    screen -x -S userbot -p 0 -X stuff $'\n'
    read -p "请输入您的登录验证码: " checknum
    screen -x -S userbot -p 0 -X stuff "$checknum"
    screen -x -S userbot -p 0 -X stuff $'\n'
    read -p "您是否有二次登录验证码(y或n): " choi
    if [ "$choi" == "y" ]; then
     	read -p "请输入您的二次登录验证码: " twotimepwd
	    screen -x -S userbot -p 0 -X stuff "$twotimepwd"
    	screen -x -S userbot -p 0 -X stuff $'\n'
    fi
    sleep 5
    screen -S userbot -X quit >> /dev/null 2>&1
}

systemctl_reload(){
	echo "正在写入系统进程守护 . . ."
    echo "[Unit]
    Description=PagerMaid-Modify telegram utility daemon
    After=network.target
    [Install]
    WantedBy=multi-user.target
    [Service]
    Type=simple
    WorkingDirectory=/var/lib/pagermaid
    ExecStart=$PYV -m pagermaid
    Restart=always
    ">/etc/systemd/system/pagermaid.service
    chmod 755 pagermaid.service >> /dev/null 2>&1
    systemctl daemon-reload >> /dev/null 2>&1
    systemctl start pagermaid >> /dev/null 2>&1
    systemctl enable pagermaid >> /dev/null 2>&1
}

start_installation() {
  if [ "$release" = "centos" ]; then
    echo "系统检测通过。"
    welcome
    yum_update
    yum_git_check
    yum_python_check
    yum_screen_check
    yum_require_install
    download_repo
    pypi_install
    configure
    login_screen
    systemctl_reload
    echo "PagerMaid 已经安装完毕 在telegram对话框中输入 -help 并发送查看帮助列表"
  elif [ "$release" = "ubuntu" ]; then
  	echo "系统检测通过。"
    welcome
    apt_update
    apt_git_check
    apt_python_check
    apt_screen_check
    apt_require_install
    download_repo
    pypi_install
    configure
    login_screen
    systemctl_reload
    echo "PagerMaid 已经安装完毕 在telegram对话框中输入 -help 并发送查看帮助列表"
   elif [ "$release" = "debian" ]; then
   	echo "系统检测通过。"
   	welcome
    apt_update
    apt_git_check
    debian_python_check
    apt_screen_check
    debian_require_install
    download_repo
    pypi_install
    configure
    login_screen
    systemctl_reload
    echo "PagerMaid 已经安装完毕 在telegram对话框中输入 -help 并发送查看帮助列表"
  else
    echo "目前暂时不支持此系统。"
    fi
  exit 1
}

cleanup() {
  if [ ! -x "/var/lib/pagermaid" ]; then
	echo "目录不存在不需要卸载。"
  else
  	echo "正在关闭 PagerMaid . . ."
    systemctl disable pagermaid >> /dev/null 2>&1
    systemctl stop pagermaid >> /dev/null 2>&1
    echo "正在删除 PagerMaid 文件 . . ."
    rm -rf /etc/systemd/system/pagermaid.service >> /dev/null 2>&1
    rm -rf /var/lib/pagermaid >> /dev/null 2>&1
    echo "卸载完成 . . ."
  fi
}

reinstall(){
	cleanup
	start_installation
}

cleansession(){
	if [ ! -x "/var/lib/pagermaid" ]; then
	    echo "目录不存在请重新安装 PagerMaid。"
	    exit 1
    fi
	echo "正在关闭 PagerMaid . . ."
    systemctl stop pagermaid >> /dev/null 2>&1
	echo "正在删除账户授权文件 . . ."
	rm -rf /var/lib/pagermaid/pagermaid.session >> /dev/null 2>&1
	echo "请进行重新登陆. . ."
	if [ "$release" = "centos" ]; then
		yum_python_check
	    yum_screen_check
	elif [ "$release" = "ubuntu" ]; then
		apt_python_check
		apt_screen_check
	elif [ "$release" = "debian" ]; then
		debian_python_check
		apt_screen_check
	else
		echo "目前暂时不支持此系统。"
	fi
	login_screen
	systemctl start pagermaid >> /dev/null 2>&1
}

stop_pager(){
	echo ""
	echo "正在关闭 PagerMaid . . ."
	systemctl stop pagermaid >> /dev/null 2>&1
	echo ""
	sleep 3
	shon_online
}

start_pager(){
	echo ""
	echo "正在启动 PagerMaid . . ."
	systemctl start pagermaid >> /dev/null 2>&1
	echo ""
	sleep 3
	shon_online
}

restart_pager(){
	echo ""
	echo "正在重新启动 PagerMaid . . ."
	systemctl restart pagermaid >> /dev/null 2>&1
	echo ""
	sleep 3
	shon_online
}

install_require(){
  if [ "$release" = "centos" ]; then
    echo "系统检测通过。"
    yum_update
    yum_git_check
    yum_python_check
    yum_screen_check
    yum_require_install
    pypi_install
    systemctl_reload
    shon_online
  elif [ "$release" = "ubuntu" ]; then
  	echo "系统检测通过。"
    apt_update
    apt_git_check
    apt_python_check
    apt_screen_check
    apt_require_install
    pypi_install
    systemctl_reload
    shon_online
   elif [ "$release" = "debian" ]; then
   	echo "系统检测通过。"
   	welcome
    apt_update
    apt_git_check
    debian_python_check
    apt_screen_check
    debian_require_install
    pypi_install
    systemctl_reload
    shon_online
  else
    echo "目前暂时不支持此系统。"
    fi
  exit 1
}

shon_online(){
echo "请选择您需要进行的操作:"
echo "  1) 安装 PagerMaid"
echo "  2) 卸载 PagerMaid"
echo "  3) 重新安装 PagerMaid"
echo "  4) 重新登陆 PagerMaid"
echo "  5) 关闭 PagerMaid"
echo "  6) 启动 PagerMaid"
echo "  7) 重新启动 PagerMaid"
echo "  8) 重新安装 PagerMaid 依赖"
echo "  9) 退出脚本"
echo ""
echo "     Version：0.1.3"
echo ""
echo -n "请输入编号: "
read N
case $N in
  1) start_installation ;;
  2) cleanup ;;
  3) reinstall ;;
  4) cleansession ;;
  5) stop_pager ;;
  6) start_pager ;;
  7) restart_pager ;;
  8) install_require ;;
  9) exit ;;
  *) echo "Wrong input!" ;;
esac 
}

check_sys
shon_online