#!/bin/bash

#Script - Functions
NO_ARGS=0
CMD=$1
SRV=$2

# Text color variables
txtund=$(tput sgr 0 1)    # Underline
txtbld=$(tput bold)       # Bold
txtred=$(tput setaf 1)    # Red
txtgrn=$(tput setaf 2)    # Green
txtylw=$(tput setaf 3)    # Yellow
txtblu=$(tput setaf 4)    # Blue
txtpur=$(tput setaf 5)    # Purple
txtcyn=$(tput setaf 6)    # Cyan
txtwht=$(tput setaf 7)    # White
txtrst=$(tput sgr0)       # Text reset

txtund=$(tput sgr 0 1)          # Underline
txtbld=$(tput bold)             # Bold

#Counter-Strike: Source - Functions

#Minecraft - Functions
UpdateURL=http://ci.bukkit.org/job/dev-CraftBukkit/Recommended/artifact/target/craftbukkit-0.0.1-SNAPSHOT.jar

#Backup - Functions
files=minecraft/world
ext=.tar.gz
parameters=zcvf

hdateformat=$(date '+%d-%m-%Y-%H-%M')H$ext

backupdir=minecraft/backups
expire=3

#Global - Functions

#If the user inputted the detials incorrectly
usage() {
    echo "Usage of the TmF Engine Script"
    echo "./tmf.sh $FUNC {Server}"
}

#Single Startup
server_start() {
	START=`ps x | grep SCREEN | grep $SRV | cut -d '?' -f 1`
		if [ ! -n "$START" ]; then
            echo -e "${txtgrn}${txtbld}$SRV is starting up${txtrst}"
			if [ $SRV = "minecraft" ]; then
				cd $SRV
                ./$SRV.sh
                cd
			else
				cd $SRV/css
                ./$SRV.sh
                cd
			fi
		else
            echo -e "${txtred}${txtbld}Error: $SRV is already running!${txtrst}"
		fi
}

#Single Stop
server_stop() {
STOP=`ps x | grep SCREEN | grep $SRV | cut -d '?' -f 1`
    if [ "$(screen -ls | grep $SRV)" == "" ] ; then
        echo -e "${txtred}${txtbld}Error: $SRV is not running!${txtrst}"
    else
        if [ $SRV = "minecraft" ]; then
            screen -x minecraft -p 0 -X stuff "Server shutting down$(echo -ne '\r')"
            screen -x minecraft -p 0 -X stuff "stop$(echo -ne '\r')"
            rm -R minecraft/server.log
            rm -R minecraft/worldedit.log
            cd
            echo -e "${txtgrn}${txtbld}$SRV is now offline${txtrst}"
        else
            sleep 5
            screen -dr $STOP -X quit
            echo -e "${txtgrn}${txtbld}$SRV is now offline${txtrst}"
        fi
    fi
}

#Update Check - Updates the correct server with the right parameters
server_update() {
    if [ $SRV = "minecraft" ]; then
        echo -e "${txtblu}${txtbld}Prompting users that update is incoming. Waiting for 60secs${txtrst}"
        screen -x minecraft -p 0 -X stuff "say Going down for an update in 60 seconds.$(echo -ne '\r')"
        sleep 55
        screen -x minecraft -p 0 -X stuff "say Going down for an update in 5 seconds.$(echo -ne '\r')"
        sleep 5
        screen -x minecraft -p 0 -X stuff "stop$(echo -ne '\r')"
        sleep 5
        server_update_MINECRAFT
    else
        server_stop
        sleep 5
        server_update_CSS
    fi
}

#Minecraft Update - Function
server_update_MINECRAFT() {
    PRC=`ps x | grep $PRC | grep -v grep | cut -d '?' -f 1`
    STILLRUN=`ps x | grep $PRC | grep -v grep | cut -d '?' -f 1`
        if [ "$STILLRUN" = "$PRC" ]; then
            echo -e "${txtred}${txtbld}Error: $SRV was unable to shutdown!${txtrst}"
            exit
        else
            cd minecraft
            wget -N $UpdateURL
            cd
            echo -e "${txtgrn}${txtbld}$SRV has been updated successfully${txtrst}"
        fi
}

#Counter-Strike: Source update function
server_update_CSS() {
    PRC=`ps x | grep SCREEN | grep $SRV | cut -d '?' -f 1`
        if [ ! -n "$PRC" ]; then
            PRC=999999999999
        else
            kill -9 $PRC
            echo -e "${txtgrn}${txtbld}$SRV has been successfully shutdown${txtrst}"
        fi
    STILLRUN=`ps x | grep $PRC | grep -v grep | cut -d '?' -f 1`
        if [ "$STILLRUN" = "$PRC" ]; then
            echo -e "${txtred}${txtbld}Error: $SRV was unable to shutdown!${txtrst}"
            exit
        else
            echo "Updating $SRV now"
            if [[ $SRV = 'badwater' || $SRV = 'controlpoint' ]]; then
				/home/server/1hldsupdatetool/./steam -command update -game tf -dir "/home/server/$SRV"
			else
				/home/server/1hldsupdatetool/./steam -command update -game "counter-strike source" -dir "/home/server/$SRV"
			fi
			echo -e "${txtgrn}${txtbld}$SRV has been updated successfully${txtrst}"
			server_start
		fi
}

#Single Restart
server_restart() {
    server_stop
    sleep 10
    server_start
}

#Minecraft Backup - Function
server_minecraft_backup() {
        #Does the backup folder exist if not create it
    if [ -d $BackupDIR ]; then
        sleep 0
    else
        mkdir -p $BackupDIR
    fi

    #is the server online, if not dont backup
    if [ "$(screen -ls | grep minecraft)" == "" ] ; then
        exit
    else
        #Minecraft Server messages (Backup Starting and stopping world saving)
        screen -x minecraft -p 0 -X stuff "say Backup Started, Saving turned off.$(echo -ne '\r')"
        screen -x minecraft -p 0 -X stuff "save-off$(echo -ne '\r')"
        
        #Compressions or the Files
        tar $parameters $hdateformat $files
        
        #Move compressed files to the backups folder
        mv $hdateformat $backupdir
        
        #Remove old backups
        find $backupdir/ -name *H$ext -mtime +$expire -exec rm {} \;
        
        #Minecraft Server messages (Backup Finished and starting world saving)
        screen -x minecraft -p 0 -X stuff "say Backup Finished, Saving turned on.$(echo -ne '\r')"
        screen -x minecraft -p 0 -X stuff "save-on$(echo -ne '\r')"
    fi
}

server_status() {
if [ "$(screen -ls | grep aim)" == "" ] ; then
    echo -e "Aim --- ${txtred}${txtbld}Offline${txtrst}"
else
    echo -e "Aim --- ${txtgrn}${txtbld}Online${txtrst}"
fi
sleep 1
if [ "$(screen -ls | grep badwater)" == "" ] ; then
    echo -e "Badwater --- ${txtred}${txtbld}Offline${txtrst}"
else
    echo -e "Badwater --- ${txtgrn}${txtbld}Online${txtrst}"
fi
sleep 1
if [ "$(screen -ls | grep deathrun)" == "" ] ; then
    echo -e "Deathrun --- ${txtred}${txtbld}Offline${txtrst}"
else
    echo -e "Deathrun --- ${txtgrn}${txtbld}Online${txtrst}"
fi
sleep 1
if [ "$(screen -ls | grep event)" == "" ] ; then
    echo -e "Event --- ${txtred}${txtbld}Offline${txtrst}"
else
    echo -e "Event --- ${txtgrn}${txtbld}Online${txtrst}"
fi
sleep 1
if [ "$(screen -ls | grep jailbreak)" == "" ] ; then
    echo -e "Jailbreak --- ${txtred}${txtbld}Offline${txtrst}"
else
    echo -e "Jailbreak --- ${txtgrn}${txtbld}Online${txtrst}"
fi
sleep 1
if [ "$(screen -ls | grep left4dead2)" == "" ] ; then
    echo -e "Left4Dead2 --- ${txtred}${txtbld}Offline${txtrst}"
else
    echo -e "Left4Dead2 --- ${txtgrn}${txtbld}Online${txtrst}"
fi
sleep 1
if [ "$(screen -ls | grep minecraft)" == "" ] ; then
    echo -e "Minecraft --- ${txtred}${txtbld}Offline${txtrst}"
else
    echo -e "Minecraft --- ${txtgrn}${txtbld}Online${txtrst}"
fi
sleep 1
if [ "$(screen -ls | grep skillsurf)" == "" ] ; then
    echo -e "SkillSurf --- ${txtred}${txtbld}Offline${txtrst}"
else
    echo -e "SkillSurf --- ${txtgrn}${txtbld}Online${txtrst}"
fi
}

case $CMD in
	stop)
		server_stop
	;;
	start)
		server_start
	;;
	update)
		server_update
    ;;
	status)
		server_status
	;;
    restart)
        server_restart
    ;;
    backup)
        server_minecraft_backup
    ;;
	*)
		server_usage 
esac

exit 0