upgrade-cacti.sh

#!/bin/bash

# bash <(curl -s https://raw.githubusercontent.com/KnoAll/cacti-template/dev/upgrade-cacti.sh) dev debug
green=$(tput setaf 2)
red=$(tput setaf 1)
tan=$(tput setaf 3)
reset=$(tput sgr0)
errorcount=0
branch=master
param1=$1
param2=$2

printinfo() {
	if [ -z "$1" ]; then
		printf "${tan}::: ${green}%s${reset}\n" "$@"
	else
		printf "${tan}::: ${green}%s${reset}\n" "$(date +%a_%R): $@"
	fi	
}
printwarn() {
	printf "${tan}*** WARNING: %s${reset}\n" "$(date +%a_%R): $@"
}
printerror() {
	printf "${red}!!! ERROR: %s${reset}\n" "$(date +%a_%R): $@"
}
printNotices() {
	notices=$(curl -s http://kevinnoall.com/notices.txt) && printinfo "$notices" && printinfo
}

#ingest options
if [[ "$#" > 0 ]]; then
	for var in "$@"; do
	    case $var in
		debug|-debug|--debug)
			trap 'echo cmd: "$BASH_COMMAND" on line $LINENO exited with code: $?' DEBUG
			printwarn "Now DEBUGGING"
		;;
		dev|-dev|--dev)
			branch=dev
			printwarn "Now on DEV branch."
		;;
		*) 
	    esac
	done
else
	counter=$( curl -s http://www.kevinnoall.com/cgi-bin/counter/unicounter.pl?name=cacti-upgrade&write=0 )
	printinfo
fi

case $(whoami) in
        root)
		printerror "You ran me as root! Do not run me as root!"
		exit 1
		;;
        pi)
		printerror "You ran me as pi user! Do not run me as pi!"
		exit 1
                ;;
        cacti)
                ;;
        *)
		printerror "Uh-oh. You are not logged in as the cacti user. Exiting..."
		exit 1
                ;;
esac

# get the Cacti version
upgrade_version=1.1.6
# get ready for dynamic update
#prod_version=$( curl -s https://raw.githubusercontent.com/Cacti/cacti/master/include/cacti_version )
prod_version=1.2.27
symlink_cactidir=1.1.28
cactiver=$( cat /var/www/html/cacti/include/cacti_version )
config_path=/var/www/html/cacti/include/config.php
if [[ -z $cactiver ]];then
	printerror "Cacti is either not installed or we were not able to determine it's version. Cannot proceed..."
	exit 1
fi

# get latest version of cacti-upgrade script
if grep -q v1.2.24 cacti-upgrade.sh; then
	printinfo
else
	printinfo "Upgrading cacti-upgrade.sh"
	rm cacti-upgrade.sh
	wget -q https://raw.githubusercontent.com/KnoAll/cacti-template/$branch/cacti-upgrade.sh
	chmod +x cacti-upgrade.sh
fi

cd ~
file="template"
if [ -f "$file" ]
then
	counter=$( curl -s http://www.kevinnoall.com/cgi-bin/counter/unicounter.pl?name=cacti-template&write=0 )
	printinfo
	rm $file
else
	printinfo
fi

file=".install"
if [ -f "$file" ]
then
	counter=$( curl -s http://www.kevinnoall.com/cgi-bin/counter/unicounter.pl?name=cacti-install&write=0 )
	printinfo
	rm $file
else
	printinfo
fi

if grep -q "AlmaLinux" /etc/os-release; then
	pkg_mgr=dnf
	os_dist=almalinux
elif grep -q "Rocky Linux" /etc/os-release; then
	pkg_mgr=dnf
	os_dist=rockylinux
elif grep -q "Raspbian GNU" /etc/os-release; then
	pkg_mgr=apt
	os_dist=raspbian
elif grep -q "CentOS Linux" /etc/os-release; then
	pkg_mgr=yum
	os_dist=centos
else
	printerror "You seem to be on something other than CentOS or Alma/Rocky Linux, cannot proceed..."
	exit 1
fi

function version_ge() { test "$(echo "$@" | tr " " "\n" | sort -rV | head -n 1)" == "$1"; }
function version_lt() { test "$(echo "$@" | tr " " "\n" | sort -rV | head -n 1)" != "$1"; }

function check-smokeping () {
	if [ -f /opt/smokeping/bin/smokeping ];then
		bash <(curl -s https://raw.githubusercontent.com/KnoAll/cacti-template/$branch/upgrade-smokeping.sh) $branch $param2
		smokeping_onoff
	else
		smokever=nosmoke
		printinfo
	fi
}

function smokeping_onoff () {
	if [[ $smokever == "nosmoke" ]]; then
		printinfo
	else
		systemctl -q is-enabled smokeping.service
		if [ $? -ne 0 ];then
			# smokeping not enabled
			printinfo "Smokeping service is disabled, do you wish to enable?"
			read -n 1 -p "y/N: " smokeon
			case "$smokeon" in
				y | Y | yes | YES| Yes ) 
					printinfo "Enabling Smokeping service..."
					sudo systemctl enable smokeping.service
					sudo systemctl start smokeping.service
				;;
				* ) 
					printwarn "OK, no Smokeping today!"
				;;
			esac
		else
			# smokeping enabled
			printinfo "Smokeping service is enabled and running at http://localhost/smokeping/smokeping.cgi, do you wish to disable?"
			read -n 1 -p "y/N: " smokeoff
			case "$smokeoff" in
				y | Y | yes | YES| Yes ) 
					printinfo "Disabling Smokeping service..."
					sudo systemctl disable smokeping.service
					sudo systemctl stop smokeping.service
				;;
				* ) 
					printinfo "OK, leaving Smokeping enabled, you should check it out!"
				;;
			esac
		fi
	fi	
}

function upgrade-plugins() {
	printinfo "Would you like to check your Cacti plugins for updates?"
	read -p "Y/n: " plugup
	plugup=${plugup:-Y}
	case "$plugup" in
		y | Y | yes | YES| Yes ) 
			printinfo
			bash <(curl -s https://raw.githubusercontent.com/KnoAll/cacti-template/$branch/upgrade-plugins.sh) $branch $param2
		;;
		n | N | no | NO | No )
			printinfo "OK, no plug-up today..."
		;;
		* ) 
			printwarn "You have entered an invallid selection!"
			printinfo "Please try again!"
			upgrade-plugins
		;;
	esac
}

function upgradeAsk () {
	printinfo "Found compatible Cacti v$cactiver installed, do you want to upgrade to v$prod_version?"
	read -p "y/N: " upAsk
	upAsk=${upAsk:-N}
		case "$upAsk" in
		y | Y | yes | YES| Yes ) 
			printinfo "Ok, let's go!"
		;;
		n | N | no | NO | No )
			printwarn "OK, maybe next time..."
			check-smokeping
			exit 1
		;;
		* ) 
			printwarn "You have entered an invallid selection!"
			printinfo "Please try again!"
			upgradeAsk
		;;
		esac
}


if version_ge $cactiver $upgrade_version; then
        if version_ge $cactiver $prod_version; then
                printinfo "Cacti v$cactiver is up to date with production v$prod_version, nothing to do!"
		counter=$( curl -s http://www.kevinnoall.com/cgi-bin/counter/unicounter.pl?name=$cactiver-current&write=0 )
		printinfo
		printNotices
		#check for PHP version upgrade
		bash <(curl -s https://raw.githubusercontent.com/KnoAll/cacti-template/$branch/upgrade-php.sh) $param1 $param2
		upgrade-plugins
		check-smokeping
		printinfo "All done!"
                exit 0
        else
		printNotices
		#check for PHP version upgrade
		bash <(curl -s https://raw.githubusercontent.com/KnoAll/cacti-template/$branch/upgrade-php.sh) $param1 $param2
		phpStatus=$?
		case "$phpStatus" in
		167 )
			printerror "PHP is not at minimum version, cannot proceed. To automatically upgrade Cacti please upgrade PHP to v7.2 or higher. Exiting."
			exit 1
		;;
		* )
			printinfo "PHP seems to be at minimum version, proceeding"
		;;
		esac
		upgradeAsk
        fi
else
	printerror "Cacti v$cactiver is less than upgrade version v$upgrade_version cannot install, exiting..."
	exit 1
fi

printinfo "Welcome to Kevin's Cacti Template upgrade script!"
sudo echo

function update-php () {
	if version_ge $prod_version 1.2.0; then
		if version_ge $cactiver 1.2.0; then
			printinfo
		else
			printinfo "Updating php settings for cacti v1.2.x..."
			if [[ $pkg_mgr == "yum" ]]; then
				phpini_path=/etc/php.ini
				webserver=httpd
			else
				phpini_path=/etc/php/7.0/apache2/php.ini
				phpclini_path=/etc/php/7.0/cli/php.ini
				webserver=apache2
			fi
			grep -q -w "memory_limit = 128M" $phpini_path
			if [ $? -ne 0 ];then
				grep -q -w "memory_limit = 800M" $phpini_path
				if [ $? -ne 0 ];then
					printwarn "php memory_limit neither 128 or 800, cannot update..."
				else
					printinfo "php memory_limit already = 800."
				fi
			else
				sudo sed -i 's/memory_limit = 128M/memory_limit = 800M/g' $phpini_path
				if [ $? -ne 0 ];then
					printerror "php memory_limit NOT updated."
				else
					printinfo "php memory_limit updated to 800."
				fi
			fi
			grep -q -w "max_execution_time = 30" $phpini_path
			if [ $? -ne 0 ];then
				#NOT 128, check for 800
				grep -q -w "max_execution_time = 60" $phpini_path
				if [ $? -ne 0 ];then
					printwarn "php max_execution_time neither 30 or 60, cannot update..."
				else
					printinfo "php max_execution_time already = 60."
				fi
			else
				sudo sed -i 's/max_execution_time = 30/max_execution_time = 60/g' $phpini_path
						if [ $? -ne 0 ];then
							printwarn "php max_execution_time NOT updated."
						else
							printinfo "php max_execution_time updated to 60."
						fi	
			fi
			sudo systemctl restart $webserver.service
		fi
	fi
}

function update-mysqld () {
	case "$pkg_mgr" in
		dnf|yum)
			mycnf_path=/etc/my.cnf
		;;
		apt)
			mycnf_path=/etc/mysql/my.cnf
		;;
	esac
	if version_ge $prod_version 1.2.0; then
		if version_ge $cactiver 1.2.0; then
			printinfo
		else
			printinfo "updating mysqld settings for cacti v1.2.x..."
			grep -q -w "mysqld" $mycnf_path
			if [ $? -ne 0 ]; then
				#Fugly but works for now...
				sudo sed  -i '$ a [mysqld]' $mycnf_path
				sudo sed  -i '$ a max_allowed_packet=16M' $mycnf_path
				sudo sed  -i '$ a innodb_additional_mem_pool_size=80M' $mycnf_path 
				sudo sed  -i '$ a innodb_flush_log_at_timeout=3' $mycnf_path 
				sudo sed  -i '$ a innodb_read_io_threads=32' $mycnf_path
				sudo sed  -i '$ a innodb_write_io_threads=16' $mycnf_path 
				sudo sed  -i '$ a max_heap_table_size=30M' $mycnf_path 
				sudo sed  -i '$ a tmp_table_size=30M' $mycnf_path 
				sudo sed  -i '$ a join_buffer_size=58M' $mycnf_path 
				sudo sed  -i '$ a innodb_buffer_pool_size=450M' $mycnf_path 
				sudo sed  -i '$ a character-set-server=utf8mb4' $mycnf_path 
				sudo sed  -i '$ a collation-server=utf8mb4_unicode_ci' $mycnf_path 
				sudo sed  -i '$ a max_allowed_packet=16M' $mycnf_path
				sudo sed  -i '$ a innodb_file_format=Barracuda' $mycnf_path
			fi
		sudo systemctl restart mysqld.service
		fi
	fi

	#Update mysql for large_prefix
	grep -q -w "innodb_large_prefix" $mycnf_path
		if [ $? -ne 0 ];then
			sudo sed  -i '$ a innodb_large_prefix=1' $mycnf_path
			sudo systemctl restart mysqld.service
		else
			printinfo
		fi
	#Barracuda file format update
	grep -q -w "innodb_file_format" $mycnf_path
	if [ $? -ne 0 ];then
			sudo sed  -i '$ a innodb_file_format=Barracuda' $mycnf_path
			sudo systemctl restart mysqld.service
		else
			printinfo
		fi

	mysql -u root -pcacti cacti -s -e "ALTER DATABASE cacti CHARACTER SET = utf8mb4 COLLATE utf8mb4_unicode_ci;"

}

function backup-db () {
	printinfo "Backing up DB..."
	mysqldump --user=cacti --password=cacti -l --add-drop-table cacti |gzip > /var/www/html/cacti/mysql.cacti_$(date +\%Y\%m\%d).sql.gz
 		if [[ $? -ne 0 ]];then
			printwarn "Error backing up default Cacti db, trying to backup alternate db cactimain"
			mysqldump --user=cacti --password=cacti -l --add-drop-table cactimain |gzip > /var/www/html/cacti/mysql.cacti_$(date +\%Y\%m\%d).sql.gz
   		fi
	printinfo
}

function check-permissions () {
	touch /var/www/html/perm
	if [ $? -ne 0 ];then
		printwarn "File permissions not sufficient, attempting to repair..."
		update-permissions
	else
		rm /var/www/html/perm
		printinfo
	fi
}

# Function to disable the Cacti poller during the upgrade so that the poller does not try running while something is being updated.
function cron () {
	case $1 in
		disable )
			printwarn "Disabling Cacti cronjob"
			crontab -l | sed '/\/cacti\/poller\.php/s/^/#/' | crontab -
		;;
		enable )
			printwarn "Enabling Cacti cronjob"
			crontab -l | sed '/\/cacti\/poller\.php/s/^#//' | crontab -
				if [ $? -ne 0 ];then
					printerror "Re-Enabling cronjob failed, please check crontab -e"
				fi
		;;
	esac
}

function upgrade-cacti () {
printinfo "Beginning Cacti upgrade..."
cd /var/www/html/
wget -q https://github.com/Cacti/cacti/archive/release/$prod_version.tar.gz
if [ $? -ne 0 ];then
		printerror "Cacti download error cannot install, exiting..."
		exit 1
else
	tar -xzf $prod_version.tar.gz
	if [ $? -ne 0 ];then
		printerror "Cacti unpack error cannot install, exiting..."
		exit 1
	else
		sudo $pkg_mgr install -y -q php-gmp sendmail php-intl
		sudo systemctl restart httpd
		mv cacti/ cacti_$cactiver/
		rm $prod_version.tar.gz
		mv cacti-release-$prod_version cacti
		cp cacti_$cactiver/include/config.php cacti/include/
	fi
fi
printinfo "Restoring data"
rsync -raq --ignore-existing cacti_$cactiver/rra cacti
rsync -raq --ignore-existing cacti_$cactiver/scripts cacti
rsync -raq --ignore-existing cacti_$cactiver/resource cacti
rsync -raq --ignore-existing cacti_$cactiver/plugins cacti
rsync -raq --ignore-existing cacti_$cactiver/include/themes cacti/include
#update-config
update-permissions
sudo systemctl restart $webserver
printinfo
}

function update-config () {
printinfo "Updating cacti config..."
if [ -f  $config_path ];
then
	sed -i 's/cactiuser/cacti/g' $config_path
else
	mv cacti/include/config.php.dist $config_path
	sed -i 's/cactiuser/cacti/g' $config_path
fi
}

function update-permissions () {
	bash <(curl -s https://raw.githubusercontent.com/KnoAll/cacti-template/$branch/update-permissions.sh) $param1 $param2
}

function compress-delete () {
	printinfo "Do you want to archive the original Cacti directory?"
	read -n 3 -p "Y/n: " cleanup
	cleanup=${cleanup:-Y}
	case "$cleanup" in
		y | Y | yes | YES | Yes ) 
			printinfo
			printinfo "Creating compressed archive..."
			rm -rf /var/www/html/cacti_$cactiver/log/*log*
   			tar -pczf ~/backup_cacti-$cactiver.tar.gz -C /var/www/html/ cacti_$cactiver
			if [ $? -ne 0 ];then
				printwarn "Archive creation failed."
			else
				rm -rf /var/www/html/cacti_$cactiver
				printinfo "Archive created in home directory ~/backup_cacti-$cactiver.tar.gz..."
			fi
		;;
		n | N | no | NO | No )
			printwarn "Original Cacti directory will be left at /var/www/html/cacti_$cactiver. This could take up a lot of space."
		;;
		* ) 
			printwarn "You have entered an invallid selection!"
			printinfo "Please try again!"
			compress-delete
		;;
	esac
}

function update-cactidir () {
if version_lt $cactiver $symlink_cactidir; then
	printinfo "Legacy cacti install directory found, updating..."
	rm /var/www/html/cacti
	if [ $? -ne 0 ];then
		printerror "Cacti directory update failed, exiting."
		exit 1
	else
	mv /var/www/html/cacti-$cactiver /var/www/html/cacti
	fi
fi
}

check-permissions
cron disable
backup-db
update-cactidir
upgrade-cacti $param2
update-php
update-mysqld
bash <(curl -s https://raw.githubusercontent.com/KnoAll/cacti-template/$branch/upgrade-spine.sh) $param1 $param2
bash <(curl -s https://raw.githubusercontent.com/KnoAll/cacti-template/$branch/fping.sh) $param1 $param2
cron enable
upgrade-plugins
update-permissions
compress-delete

if [[ $branch == dev ]]; then
	printinfo
else
	counter=$( curl -s http://www.kevinnoall.com/cgi-bin/counter/unicounter.pl?name=cacti-$cactiver-$prod_version&write=0 )
	printinfo
	counter=$( curl -s http://www.kevinnoall.com/cgi-bin/counter/unicounter.pl?name=cacti-$os_dist&write=0 )
	printinfo
fi

check-smokeping

printinfo "Cacti upgraded to v$prod_version. Proceed to the web interface to complete upgrade..."
printinfo "For script errors or troubleshooting please check the Github page at https://github.com/KnoAll/cacti-template. "
exit 0