#!/usr/bin/env bash
# MySQL-BU - MySQL Backup Scripts

action="$1"
FOLDER='/opt/mysqlbu'
VERS='4.11232018'

source /opt/mysqlbu/settings.conf
source /dev/stdin <<< "$(curl -sL https://git.schroedercity.com/voltron/Misc-Scripts/raw/master/colors.inc)"
source /dev/stdin <<< "$(curl -sL https://git.schroedercity.com/voltron/Misc-Scripts/raw/master/default.inc)"

# ========================================================= #

if [ $(id -u) != "0" ]; then
	echo "Sorry, I need root privileges!"
fi

MYSQLDATA="`grep --perl-regexp --only-match '(?<=datadir=).+' "$MYSQLCFG"`"
DAY_BACKUP_DIR="${BACKUP_PATH}`date +%Y-%m-%d`/"
BASE_BACKUP="`ls -1 $DAY_BACKUP_DIR 2> /dev/null | grep --max-count=1 --perl-regexp '^.+_BASE'`"

case $1 in
	'backup')
		if [ "$BASE_BACKUP" = "" ]; then
			$0 backup-base
		else
			$0 backup-incremental
		fi
		
		allfiles=`ls -1 ${BACKUP_PATH}`
		files=(`ls -1 ${BACKUP_PATH} | tail -${BACKUPSTOKEEP}`)
		for i in $allfiles; do 
			keep=0; 
			for a in ${files[@]}; do 
				if [ $i == $a ]; then 
					keep=1; 
				fi; 
			done; 
			if [ $keep == 0 ]; then 
				rm -rf ${BACKUP_PATH}$i;
			fi;
		done
;;
    'backup-base')
		DEST_DIR="${DAY_BACKUP_DIR}`date +%H-%M-%S`_BASE"
		
		if [ "$BASE_BACKUP" != "" ]; then
			echo -e "${idsCL[Green]}Daily base backup already done. Running incremental backup...${idsCL[Default]}"
			$0 backup-incremental
			exit 1
		fi
		
		if [ ! -d "$DAY_BACKUP_DIR" ]; then
			mkdir -p "$DAY_BACKUP_DIR"
		fi
		
		innobackupex --defaults-file="$MYSQLCFG" --user="$MYSQLUSER" --password="$MYSQLPASS" --no-timestamp "$DEST_DIR"
;;
    'backup-incremental')
		DEST_DIR="${DAY_BACKUP_DIR}`date +%H-%M-%S`"
		
		if [ "$BASE_BACKUP" = "" ]; then
			echo -e "${idsCL[Green]}Daily base backup not found. Running base backup...${idsCL[Default]}"
			$0 backup-base
			exit 1
		fi
		
		innobackupex --defaults-file="$MYSQLCFG" --user="$MYSQLUSER" --password="$MYSQLPASS" \
		--incremental-basedir="${DAY_BACKUP_DIR}${BASE_BACKUP}" \
		--incremental "$DEST_DIR" \
		--no-timestamp
;;
    'restore')
		if [ "`ps aux | grep --only-match mysqld`" != "" ]; then
			read -p "I think mysql-server still running. It's STRONGLY recommended to stop it before restoring any backup to avoid corrupted files. Do you want to continue anyway? [Y/N]" yn
			case $yn in
				[Yy]* ) break;;
				* ) exit;;
			esac
		fi
		
		REQUEST_BACKUP="${BACKUP_PATH}${2}/${3}"
		BASE_BACKUP="${BACKUP_PATH}${2}/`ls ${BACKUP_PATH}${2} | grep --max-count=1 --perl-regexp '^.+_BASE'`"
		TMP_BACKUP="${BACKUP_PATH}${2}/.current_backup"
		
		if [ ! -d "$REQUEST_BACKUP" ] || [ ! $2 ] || [ ! $3 ]; then
			echo -e "${idsCL[Red]}Backup not found!${idsCL[Default]}"
			echo "Usage: $0 restore 2019-01-05 13-00-10"
			exit 1
		fi
		
		if [ -d "$MYSQLDATA" ]; then
			mv "$MYSQLDATA" "`grep --perl-regexp --only-match '^.*(?=/$)' <<< ${MYSQLDATA}`_restore-`date +%Y-%m-%d_%H-%M-%S`"
			mkdir "$MYSQLDATA"
		fi
		
		if [ -d "$TMP_BACKUP" ]; then
			rm -rf "$TMP_BACKUP"
		fi
		
		cp --recursive "$BASE_BACKUP" "$TMP_BACKUP"
		innobackupex --apply-log --redo-only "$TMP_BACKUP"
		
		if [ REQUEST_BACKUP != BASE_BACKUP ]; then
			innobackupex --apply-log "$TMP_BACKUP" --incremental-dir="$REQUEST_BACKUP"
		fi
		
		innobackupex --apply-log "$TMP_BACKUP"
		innobackupex --copy-back "$TMP_BACKUP"
		chown -R mysql:mysql "$MYSQLDATA"
		
		rm -rf "$TMP_BACKUP"
;;
    'list')
		echo ""
		echo -e "${idsCL[Yellow]}============ Available Backups ============${idsCL[Default]}"
		for day in $BACKUP_PATH* ; do
			tdt=`sed 's/.*\///' <<< $day`
			echo -e "${idsBG[Blue]}${idsCL[White]}-> `date -d"${tdt}" +"%a, %b, %d %Y"`               ${idsCL[Default]}${idsBG[Default]}"
			
			for hour in $day/* ; do
				thr=`sed 's/.*\///' <<< ${hour//_BASE/}`
				if [ ${hour: -5} = "_BASE" ]; then
					echo -e "${idsCL[LightGreen]}	-> `date -d"${tdt} ${thr//-/:}" +"%I:%M %P"` - FULL${idsCL[Default]}"
				else
					echo -e "${idsCL[Cyan]}	-> `date -d"${tdt} ${thr//-/:}" +"%I:%M %P"` - Incremental${idsCL[Default]}"
				fi
			done
		done
		echo -e "${idsCL[Yellow]}===========================================${idsCL[Default]}"
;;
    *)
        echo "Usage: $0 { backup | backup-base | backup-incremental | restore | list }"
;;
esac

exit 0