#!/bin/bash
Green_font="\033[32m" && Yellow_font="\033[33m" && Red_font="\033[31m" && Font_suffix="\033[0m"
Info="${Green_font}[Info]${Font_suffix}"
Error="${Red_font}[Error]${Font_suffix}"
NGX_DIR=/www/server/nginx
NPS_VESION=1.13.35.2-stable

echo -e "${Green_font}
#=======================================
# Project:  nps-auto
# Platform: --Debian --Centos --Unbuntu
# requirement: root   gcc >= 4.8  bt.cn
# Version:  0.0.1
# Author:   madlifer
# Thanks:   nanqinlang / zhangge.net
# Copyright:   www.modpagespeed.com
#=======================================
${Font_suffix}"

download_ngx_pagespeed(){
	cd ${NGX_DIR}/src
	wget https://github.com/apache/incubator-pagespeed-ngx/archive/v${NPS_VESION}.zip
	unzip v${NPS_VESION}.zip
	rm v${NPS_VESION}.zip
	NPS_DIR=$(find . -name "*pagespeed-ngx-${NPS_VESION}" -type d)
	mv $NPS_DIR ngx_pagespeed
	cd ngx_pagespeed
	NPS_RELEASE_NUMBER=${NPS_VESION/beta/}
	NPS_RELEASE_NUMBER=${NPS_VESION/stable/}
	PSPL_URL=https://dl.google.com/dl/page-speed/psol/${NPS_RELEASE_NUMBER}.tar.gz
	[ -e scripts/format_binary_url.sh ]
	PSPL_URL=$(scripts/format_binary_url.sh PSOL_BINARY_URL)
	wget ${PSPL_URL}
	tar -xzvf $(basename ${PSPL_URL})
	rm $(basename ${PSPL_URL})
}

install_ngx_pagespeed(){
	cd ${NGX_DIR}/src
	NGX_CONF=`/usr/bin/nginx -V 2>&1 >/dev/null | grep 'configure' --color | awk -F':' '{print $2;}'`
	NGX_CONF="--add-module=${NGX_DIR}/src/ngx_pagespeed $NGX_CONF"
	./configure $NGX_CONF
	make
	make install
}

check_system() {
    if grep -Eqii "CentOS" /etc/issue || grep -Eq "CentOS" /etc/*-release; then
        DISTRO='CentOS'
        PM='yum'
    elif grep -Eqi "Red Hat Enterprise Linux Server" /etc/issue || grep -Eq "Red Hat Enterprise Linux Server" /etc/*-release; then
        DISTRO='RHEL'
        PM='yum'
    elif grep -Eqi "Aliyun" /etc/issue || grep -Eq "Aliyun" /etc/*-release; then
        DISTRO='Aliyun'
        PM='yum'
    elif grep -Eqi "Debian" /etc/issue || grep -Eq "Debian" /etc/*-release; then
        DISTRO='Debian'
        PM='apt'
    elif grep -Eqi "Ubuntu" /etc/issue || grep -Eq "Ubuntu" /etc/*-release; then
        DISTRO='Ubuntu'
        PM='sudo'
    else
        DISTRO='unknow'
    fi
}

install_basic(){
	case ${PM} in
		yum)
			yum -y install sudo
			yum -y update 
			sudo yum -y install gcc-c++ pcre-devel zlib-devel make unzip libuuid-devel
		;;
		apt)
			apt -y install sudo
			sudo apt -y update
			sudo apt-get -y install build-essential zlib1g-dev libpcre3 libpcre3-dev unzip uuid-dev
		;;
		sudo)
			sudo apt -y update
			sudo apt-get -y install build-essential zlib1g-dev libpcre3 libpcre3-dev unzip uuid-dev
		;;
		*)
			echo -e "${Error} Your system is not supported!"
		;;
	esac
	echo -e "${Info} The module dependency installation is complete!"
}

check_root(){
	[[ "`id -u`" != "0" ]] && echo -e "${Error}Please enter the root account first!"
}

check_gcc(){
	gcc --version  && echo -e "${Info} Please confirm gcc version>=4.8! Enter any key to confirm?"
	read aNum
}

restart_ngx(){
	service nginx restart
	echo -e "${Info} Nginx has been restarted!"
}

temp_swap_add(){
	sudo dd if=/dev/zero of=/swapfile bs=64M count=16
	sudo mkswap /swapfile
	sudo swapon /swapfile
	echo -e "${Info} temporarily increase Swap to solve the problem of insufficient memory during compilation!"	
}

temp_swap_del(){
	sudo swapoff /swapfile
	sudo rm /swapfile
	echo -e "${Info} delete temporarily increased swap space!"	
}

setup(){
	check_root
	check_system
	check_gcc
	install_basic
	temp_swap_add
	echo -e "${Info} The configuration before installation is complete!"	
}

install(){
	download_ngx_pagespeed
	install_ngx_pagespeed
	temp_swap_del
	restart_ngx
	echo -e "${Info} The ngx_pagespeed module is installed!"	
}

status(){
	NGX_CONF=`/usr/bin/nginx -V 2>&1 >/dev/null`
	echo $NGX_CONF | grep -q pagespeed
    if [ $? = 0 ]; then
        echo -e "${Info} Pagespeed is running !"
    else
    	echo -e "${Error} Pagespeed is not running !"
    fi
}

echo -e "${Info} Choose the function you want to use: "
echo -e "1. Configure before installation\n2. Install\n3. Check running status\n"
read -p "Enter numbers to select:" function

while [[ ! "${function}" =~ ^[1-4]$ ]]
	do
		echo -e "${Error} Invalid input"
		echo -e "${Info} Please select again" && read -p "Enter numbers to select:" function
	done

if [[ "${function}" == "1" ]]; then
	setup
elif [[ "${function}" == "2" ]]; then
	install
elif [[ "${function}" == "3" ]]; then
	status
fi