#!/bin/bash

# === CONFIG ===
CONFIG_CHUNK=1000000

# === DO NOT EDIT BELOW THIS LINE ===

if [ ! -n "$BASH" ] ;then echo Please run this script $0 with bash; exit 1; fi

# === FUNCTIONS ===
database_exists()
{
	RESULT=`mysqlshow --defaults-extra-file=$MYCNF $@| grep -v Wildcard | grep -o $@`
	if [ "$RESULT" == "$@" ]; then
			echo YES
	else
			echo NO
	fi
}

f_log() 
{
	logger "RESTORE: $@"
}

restore()
{
  DIR=$@

  f_log "Check path $DIR"	
	
	f_log "** START **"
	
	f_log "Check runtime"
	for BIN in $BIN_DEPS; do
			which $BIN 1>/dev/null 2>&1
			if [ $? -ne 0 ]; then
					f_log "Error: Required file could not be found: $BIN"
					exit 1
			fi
	done

	f_log "Check backups folder"
	if [ "$(ls -1 $DIR/*/__create.sql 2>/dev/null | wc -l)" -le "0" ]; then
			f_log "Your must run script from backup directory"
			exit 1
	fi

	for i in $(ls -1 -d $DIR/*); do
	
					BDD=$(basename $i)
					
					for skip in "${DATABASES_SKIP[@]}"; do
						if [ $BDD = $skip ]; then
							f_log "Skip database $BDD"
							unset BDD
							break
						fi								
					done
					
					for select in "${DATABASES_SELECTED[@]}"; do
						if [ $BDD != $select ]; then
							f_log "Skip database $BDD"
							unset BDD
							break
						fi								
					done
				
					if [ $BDD ]; then
					
						if [ -f $DIR/$BDD/__create.sql ]; then
							f_log "Create database $BDD"
							time mysql --defaults-extra-file=$MYCNF < $DIR/$BDD/__create.sql 2>/dev/null
						fi
						
						if [ $(database_exists $BDD) != "YES" ]; then
							f_log "Error: Database $BDD dose not exists";
						else
						
							tables=$(ls -1 $i |  grep -v __ | awk -F. '{print $1}' | sort | uniq)
						
							f_log "Create tables in $BDD"
							for TABLE in $tables; do							
											f_log "Create table: $BDD/$TABLE"
											time mysql --defaults-extra-file=$MYCNF $BDD -e "SET foreign_key_checks = 0;
																			DROP TABLE IF EXISTS $TABLE;
																			SOURCE $DIR/$BDD/$TABLE.sql;
																			SET foreign_key_checks = 1;"
							done
							
							f_log "Import data into $BDD"		
							for TABLE in $tables; do									
								f_log "Import data into $BDD/$TABLE"
									
								if [ -f "$DIR/$BDD/$TABLE.txt.bz2" ]; then
									f_log "< $TABLE"
									if [ -f "$DIR/$BDD/$TABLE.txt" ]; then
										rm $DIR/$BDD/$TABLE.txt
									fi
									bunzip2 $DIR/$BDD/$TABLE.txt.bz2
								fi
								
								if [ -f "$DIR/$BDD/$TABLE.txt" ]; then
									f_log "+ $TABLE"
									split -l $CONFIG_CHUNK "$DIR/$TABLE.txt" ${TABLE}_part_
									for segment in ${TABLE}_part_*; do
										time mysql --defaults-extra-file=$MYCNF $BDD --local-infile -e "SET foreign_key_checks = 0; SET unique_checks = 0; SET sql_log_bin = 0;
																		LOAD DATA LOCAL INFILE '$DIR/$BDD/$TABLE.txt'
																		INTO TABLE $TABLE;
																		SET foreign_key_checks = 1; SET unique_checks = 1; SET sql_log_bin = 1;"
										rm $segment								
									done																		
									if [ ! -f "$DIR/$BDD/$TABLE.txt.bz2" ]; then
										f_log "> $TABLE"
										bzip2 $DIR/$BDD/$TABLE.txt
									fi
								fi
								
								if [ $DATABASES_TABLE_CHECK ]; then
									if [ -f "$DIR/$BDD/$TABLE.ibd" ]; then
										if [ ! $(innochecksum $DIR/$BDD/$TABLE.ibd) ]; then
											f_log "$TABLE [OK]"
										else
											f_log "$TABLE [ERR]"
										fi
									fi
								fi
									
							done						
							
							if [ -f "$DIR/$BDD/__routines.sql" ]; then
									f_log "Import routines into $BDD"
									time mysql --defaults-extra-file=$MYCNF $BDD < $DIR/$BDD/__routines.sql 2>/dev/null
							fi
							
							if [ -f "$DIR/$BDD/__views.sql" ]; then
									f_log "Import views into $BDD"
									time mysql --defaults-extra-file=$MYCNF $BDD < $DIR/$BDD/__views.sql 2>/dev/null
							fi
							
						fi
					fi				
	done

	f_log "Flush privileges;"
	time mysql --defaults-extra-file=$MYCNF -e "flush privileges;"

	f_log "** END **"
}

usage()
{
cat << EOF
usage: $0 options

This script restore databases.

OPTIONS:
   -e      Exclude databases
	 -s			 Selected databases
	 -c			 Check innochecksum of table after import
EOF
}

# === CHECKS ===
if [ $# = 0 ]; then
    usage;
    exit;
fi

BIN_DEPS="ls grep awk sort uniq bunzip2 bzip2 mysql"

if [ -f '/etc/debian_version' ]; then
    CONFIG_FILE='/etc/mysql/debian.cnf'
else
    CONFIG_FILE='~/mysql_utils/etc/mysql/debian.cnf'
fi

for BIN in $BIN_DEPS; do
    which $BIN 1>/dev/null 2>&1
    if [ $? -ne 0 ]; then
        echo "Error: Required command file not be found: $BIN"
        exit 1
    fi
done

for i in "$@"
do
    case $i in
        -e)
						DATABASES_SKIP=( "${i#*=}" )
						shift
				;;
        -s)
						DATABASES_SELECTED=( "${i#*=}" )
						shift
				;;
        -c)
						DATABASES_TABLE_CHECK=1
						shift
				;;								
        --config=*)
            CONFIG_FILE=( "${i#*=}" )
            shift # past argument=value
        ;;
        -v | --verbose)
            VERBOSE=1
            shift # past argument=value
        ;;
        -h | --help)
            usage
            exit
        ;;
        *)
        # unknown option
        ;;
    esac
done

# === AUTORUN ===
restore $(pwd)