#!/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 -y
				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 -y
		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 -y
	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
}

read_checknum() {
	while :; do
	read -p "请输入您的登录验证码: " checknum
	if [ "$checknum" == "" ]; then
		continue
	fi
	read -p "请再次输入您的登录验证码：" checknum2
	if [ "$checknum" != "$checknum2" ]; then
		echo "两次验证码不一致！请重新输入您的登录验证码"
		continue

	else
		screen -x -S userbot -p 0 -X stuff "$checknum"
		screen -x -S userbot -p 0 -X stuff $'\n'
		break
	fi
done
	read -p "有没有二次登录验证码？ [Y/n]" choi
	if [ "$choi" == "y" ] || [ "$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
	
}

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'
	sleep 3
	if [ "$(ps -def | grep [p]agermaid | grep -v grep)" == "" ]; then
		echo "PagerMaid 运行时发生错误，错误信息："
		cd /var/lib/pagermaid && $PYV -m pagermaid >err.log
		cat err.log
		screen -S userbot -X quit >>/dev/null 2>&1
		exit 1
	fi
	while :; do
		read -p "请输入您的 Telegram 手机号码（带国际区号 如 +8618888888888）: " phonenum

		if [ "$phonenum" == "" ]; then
			continue
		fi

		screen -x -S userbot -p 0 -X stuff "$phonenum"
		screen -x -S userbot -p 0 -X stuff $'\n'

		if [ "$(ps -def | grep [p]agermaid | grep -v grep)" == "" ]; then
			echo "手机号输入错误！请确认您是否带了区号（中国号码为 +86 如 +8618888888888）"
			screen -x -S userbot -p 0 -X stuff "cd /var/lib/pagermaid && $PYV -m pagermaid"
			screen -x -S userbot -p 0 -X stuff $'\n'
			continue
		fi

		sleep 1
		if [ "$(ps -def | grep [p]agermaid | grep -v grep)" == "" ]; then
			echo "PagerMaid 运行时发生错误，可能是因为发送验证码失败，请检查您的 API_KEY 和 API_HASH"
			exit 1
		fi

		read -p "请输入您的登录验证码: " checknum
		if [ "$checknum" == "" ]; then
			read_checknum
			break
		fi

		read -p "请再次输入您的登录验证码：" checknum2
		if [ "$checknum" != "$checknum2" ]; then
			echo "两次验证码不一致！请重新输入您的登录验证码"
			read_checknum
			break
		else
			screen -x -S userbot -p 0 -X stuff "$checknum"
			screen -x -S userbot -p 0 -X stuff $'\n'
		fi

		read -p "有没有二次登录验证码？ [Y/n]" choi
		if [ "$choi" == "y" ] || [ "$choi" == "Y" ]; then
			read -p "请输入您的二次登录验证码: " twotimepwd
			screen -x -S userbot -p 0 -X stuff "$twotimepwd"
			screen -x -S userbot -p 0 -X stuff $'\n'
			break
		fi
	done
	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