#!/bin/bash

# --------------------------------------------
# Quake 3 Server Script                      |
# Created by q3aql (q3aql@openmailbox.org)   |
# Licensed by GPL v.3                        |
# Last update: 02-04-2018                    |
# --------------------------------------------
VERSION="4.2"

# Path with binary and game files
RUN_PATH="/opt/Games/quake3"
cd $RUN_PATH

# Binary of server
RUN_BINARY="ioq3ded.x86_64"

# Baseq3 folder path
BASEQ3_PATH="$HOME/.q3a/baseq3"
mkdir -p $HOME/.q3a/baseq3

# Basic configs
BOTS_LEVEL="2"
MIN_PLAYERS="5"
TIME_LIMIT="25"
FLAGS_LIMIT="10"
FRAGS_LIMIT="150"
TOU_FRAGS_LIMIT="50"
RCON_PASSWORD='"secret"'

# Status of server
STATUS_PATH=/var/www/html/quake3/status.txt
# Note: Change to '/dev/null' if you want disable it.

# Map selection
expr $2 + 1 &> /dev/null
mapSelection=$?
if [ -z "$2" ] ; then
	MAP_SELECT=$(expr $RANDOM % 5 + 1)
else
	if [ $mapSelection -ne 0 ] ; then
		echo ""
		echo "<<[Error: Map-selection only must be numbers between 1 and 5]>>"
		echo ""
		exit
	elif [ $2 -lt 1 ] || [ $2 -gt 5 ] ; then
		echo ""
		echo "<<[Error: Map-selection must be between 1 and 5]>>"
		echo ""
		exit
	else
		MAP_SELECT=$2
	fi
fi

# Function to stop a previous server and
# remove the config. files
function clear_server() {
	killall -9 $RUN_BINARY 2> /dev/null
	rm -rf ${BASEQ3_PATH}/basic.cfg 
	rm -rf ${BASEQ3_PATH}/server*.cfg 
	rm -rf ${BASEQ3_PATH}/bots*.cfg 
	rm -rf ${BASEQ3_PATH}/levels*.cfg
}

# Function to remove the server logs.
function clear_server_logs() {
	echo > ${BASEQ3_PATH}/qconsole.log
	echo > ${BASEQ3_PATH}/server.log
}

# Function to show correct syntax in
# custom game.
function custom_game_menu() {
	echo ""
	echo "Syntax: q3aServ.sh CG [map-selection] [gametype] [minplayers] [bot-level] [time-limit] [flags/frags-limit]"
	echo ""
	echo "Available options:"
	echo ""
	echo "* map-selection -> 1-5"
	echo "* gametype -> CTF/FFA/TD/TOU/FT/CTFI/TDI/FFAI/CA/CAI"
	echo "* minplayers -> 0-10"
	echo "* bot-level -> 1-5"
	echo "* time-limit -> 0-999"
	echo "* flags/frags-limit -> 0-999"
	echo ""
	echo "Abbreviations:"
	echo ""
	echo "- CTF  = Capture The Flag"
	echo "- FFA  = Free For All"
	echo "- TD   = Team Deathmatch"
	echo "- TOU  = Tournament"
	echo "- FT   = Freeze Tag TD (OSP)"
	echo "- CFTI = Capture The Flag Instagib (OSP)"
	echo "- TDI  = Team Deatchmatch Instagib (OSP)"
	echo "- FFAI = Free For All Instagib (OSP)"
	echo "- CA   = Clan Arena (OSP)"
	echo "- CAI  = Clan Arena Instagib (OSP)"
	echo ""
}

# Function to check correct gametypes.
function check_gametype() {
	if [ "$1" == "CTF" ] ; then
		echo "Gametype selected = CTF"
	elif [ "$1" == "FFA" ] ; then
		echo "Gametype selected = FFA"
	elif [ "$1" == "TD" ] ; then
		echo "Gametype selected = TD"
	elif [ "$1" == "TOU" ] ; then
		echo "Gametype selected = TOU"
	elif [ "$1" == "FT" ] ; then
		echo "Gametype selected = FT"
	elif [ "$1" == "CTFI" ] ; then
		echo "Gametype selected = CTFI"
	elif [ "$1" == "TDI" ] ; then
		echo "Gametype selected = TDI"
	elif [ "$1" == "FFAI" ] ; then
		echo "Gametype selected = FFAI"
	elif [ "$1" == "CA" ] ; then
		echo "Gametype selected = CA"
	elif [ "$1" == "CAI" ] ; then
		echo "Gametype selected = CAI"
	elif [ -z "$1" ] ; then
		echo "Empty" > /dev/null
	else
		echo ""
		echo "<<[Error: Gametype invalid]>>"
		custom_game_menu
		exit
	fi
}

# Available options
case $1 in

CTF|ctf)
	# Initialize "Capture The Flag" session
	clear_server
	cp server/basic.cfg $BASEQ3_PATH
	cp server/serverCTF.cfg $BASEQ3_PATH
	cp server/bots.cfg $BASEQ3_PATH
	cp server/levels/levelsCTF-${MAP_SELECT}.cfg $BASEQ3_PATH
	echo "seta g_spskill $BOTS_LEVEL" >> ${BASEQ3_PATH}/bots.cfg
	echo "seta bot_minplayers $MIN_PLAYERS" >> ${BASEQ3_PATH}/bots.cfg
	echo "seta timelimit $TIME_LIMIT" >> ${BASEQ3_PATH}/serverCTF.cfg
	echo "seta capturelimit $FLAGS_LIMIT" >> ${BASEQ3_PATH}/serverCTF.cfg
	echo "seta rconpassword ${RCON_PASSWORD}" >> ${BASEQ3_PATH}/serverCTF.cfg
	echo "Running Capture The Flag ($MAP_SELECT)" > $STATUS_PATH
	./${RUN_BINARY} +exec basic.cfg +exec serverCTF.cfg +exec levelsCTF-${MAP_SELECT}.cfg +exec bots.cfg
	#clear_server
	echo "Stopped" > $STATUS_PATH
;;
TD|td)
	# Initialize "Team Deathmatch" session
	clear_server
	cp server/basic.cfg $BASEQ3_PATH
	cp server/serverTD.cfg $BASEQ3_PATH
	cp server/bots.cfg $BASEQ3_PATH
	cp server/levels/levelsTD-${MAP_SELECT}.cfg $BASEQ3_PATH
	echo "seta g_spskill $BOTS_LEVEL" >> ${BASEQ3_PATH}/bots.cfg
	echo "seta bot_minplayers $MIN_PLAYERS" >> ${BASEQ3_PATH}/bots.cfg
	echo "seta timelimit $TIME_LIMIT" >> ${BASEQ3_PATH}/serverTD.cfg
	echo "seta fraglimit $FRAGS_LIMIT" >> ${BASEQ3_PATH}/serverTD.cfg
	echo "seta rconpassword ${RCON_PASSWORD}" >> ${BASEQ3_PATH}/serverTD.cfg
	echo "Running Team Deathmatch ($MAP_SELECT)" > $STATUS_PATH
	./${RUN_BINARY} +exec basic.cfg +exec serverTD.cfg +exec levelsTD-${MAP_SELECT}.cfg +exec bots.cfg
	#clear_server
	echo "Stopped" > $STATUS_PATH
;;
FFA|ffa)
	# Initialize "Free For All" session
	clear_server
	cp server/basic.cfg $BASEQ3_PATH
	cp server/serverFFA.cfg $BASEQ3_PATH
	cp server/bots.cfg $BASEQ3_PATH
	cp server/levels/levelsFFA-${MAP_SELECT}.cfg $BASEQ3_PATH
	echo "seta g_spskill $BOTS_LEVEL" >> ${BASEQ3_PATH}/bots.cfg
	echo "seta bot_minplayers $MIN_PLAYERS" >> ${BASEQ3_PATH}/bots.cfg
	echo "seta timelimit $TIME_LIMIT" >> ${BASEQ3_PATH}/serverFFA.cfg
	echo "seta fraglimit $FRAGS_LIMIT" >> ${BASEQ3_PATH}/serverFFA.cfg
	echo "seta rconpassword ${RCON_PASSWORD}" >> ${BASEQ3_PATH}/serverFFA.cfg
	echo "Running Free For All ($MAP_SELECT)" > $STATUS_PATH
	./${RUN_BINARY} +exec basic.cfg +exec serverFFA.cfg +exec levelsFFA-${MAP_SELECT}.cfg +exec bots.cfg
	#clear_server
	echo "Stopped" > $STATUS_PATH
;;
TOU|tou|tourney)
	# Initialize "Tournament" session
	clear_server
	cp server/basic.cfg $BASEQ3_PATH
	cp server/serverTOU.cfg $BASEQ3_PATH
	cp server/bots.cfg $BASEQ3_PATH
	cp server/levels/levelsTOU-${MAP_SELECT}.cfg $BASEQ3_PATH
	echo "seta g_spskill $BOTS_LEVEL" >> ${BASEQ3_PATH}/bots.cfg
	echo "seta bot_minplayers 2" >> ${BASEQ3_PATH}/bots.cfg
	echo "seta timelimit $TIME_LIMIT" >> ${BASEQ3_PATH}/serverTOU.cfg
	echo "seta fraglimit $TOU_FRAGS_LIMIT" >> ${BASEQ3_PATH}/serverTOU.cfg
	echo "seta rconpassword ${RCON_PASSWORD}" >> ${BASEQ3_PATH}/serverTOU.cfg
	echo "Running Tournament ($MAP_SELECT)" > $STATUS_PATH
	./${RUN_BINARY} +exec basic.cfg +exec serverTOU.cfg +exec levelsTOU-${MAP_SELECT}.cfg +exec bots.cfg
	#clear_server
	echo "Stopped" > $STATUS_PATH
;;
CG|cg|custom)
	# Initialize "Custom Game" session
	check_gametype $3
	expr $4 + 1 &> /dev/null
	minPlayers=$?
	expr $5 + 1 &> /dev/null
	botLevel=$?
	expr $6 + 1 &> /dev/null
	timeLimit=$?
	expr $7 + 1 &> /dev/null
	FFLimit=$?
	if [ -z "$2" ] ; then
		custom_game_menu
	elif [ "$2" == "-h" ] || [ "$2" == "--help" ] ; then
		custom_game_menu
	elif [ -z "$4" ] ; then
		echo ""
		echo "<<[Error: Minplayers must be specified]>>"
		custom_game_menu
	elif [ $minPlayers -ne 0 ] ; then
		echo ""
		echo "<<[Error: Minplayers only must be numbers between 0 and 10]>>"
		custom_game_menu
	elif [ $4 -lt 0 ] || [ $4 -gt 10 ] ; then
		echo ""
		echo "<<[Error: Minplayers must be between 0 and 10]>>"
		custom_game_menu
	elif [ -z "$5" ] ; then
		echo ""
		echo "<<[Error: Bot-level must be specified]>>"
		custom_game_menu
	elif [ $botLevel -ne 0 ] ; then
		echo ""
		echo "<<[Error: Bot-level only must be numbers between 1 and 5]>>"
		custom_game_menu
	elif [ $5 -lt 1 ] || [ $5 -gt 5 ] ; then
		echo ""
		echo "<<[Error: Bot-level must be between 1 and 5]>>"
		custom_game_menu
	elif [ -z "$6" ] ; then
		echo ""
		echo "<<[Error: Time-limit must be specified]>>"
		custom_game_menu
	elif [ $timeLimit -ne 0 ] ; then
		echo ""
		echo "<<[Error: Time-limit only must be numbers between 0 and 999]>>"
		custom_game_menu
	elif [ $6 -lt 0 ] || [ $6 -gt 999 ] ; then
		echo ""
		echo "<<[Error: Time-limit must be between 0 and 999]>>"
		custom_game_menu
	elif [ -z "$7" ] ; then
		echo ""
		echo "<<[Error: Flags/frags-limit must be specified]>>"
		custom_game_menu
	elif [ $FFLimit -ne 0 ] ; then
		echo ""
		echo "<<[Error: Flags/frags-limit only must be numbers between 0 and 999]>>"
		custom_game_menu
	elif [ $7 -lt 0 ] || [ $7 -gt 999 ] ; then
		echo ""
		echo "<<[Error: Flags/frags-limit must be between 0 and 999]>>"
		custom_game_menu
	else
		clear_server
		cp server/basic.cfg $BASEQ3_PATH
		cp server/server${3}.cfg $BASEQ3_PATH
		cp server/bots.cfg $BASEQ3_PATH
		cp server/levels/levels${3}-${2}.cfg $BASEQ3_PATH
		echo "seta g_spskill $5" >> ${BASEQ3_PATH}/bots.cfg
		echo "seta bot_minplayers $4" >> ${BASEQ3_PATH}/bots.cfg
		echo "seta timelimit $6" >> ${BASEQ3_PATH}/server${3}.cfg
		if [ "$3" == "CTF" ] ; then
			echo "seta capturelimit $7" >> ${BASEQ3_PATH}/server${3}.cfg
		elif [ "$3" == "CTFI" ] ; then
			echo "seta capturelimit $7" >> ${BASEQ3_PATH}/server${3}.cfg
		else
			echo "seta fraglimit $7" >> ${BASEQ3_PATH}/server${3}.cfg
		fi
		echo "seta rconpassword ${RCON_PASSWORD}" >> ${BASEQ3_PATH}/server${3}.cfg
		echo "Running $3 (Custom Game) ($2)" > $STATUS_PATH
		if [ "$3" == "FT" ] ; then
			./${RUN_BINARY} +set fs_game osp +exec basic.cfg +exec server${3}.cfg +exec levels${3}-${2}.cfg +exec bots.cfg
		elif [ "$3" == "CTFI" ] ; then
			./${RUN_BINARY} +set fs_game osp +exec basic.cfg +exec server${3}.cfg +exec levels${3}-${2}.cfg +exec bots.cfg
		elif [ "$3" == "TDI" ] ; then
			./${RUN_BINARY} +set fs_game osp +exec basic.cfg +exec server${3}.cfg +exec levels${3}-${2}.cfg +exec bots.cfg
		elif [ "$3" == "FFAI" ] ; then
			./${RUN_BINARY} +set fs_game osp +exec basic.cfg +exec server${3}.cfg +exec levels${3}-${2}.cfg +exec bots.cfg
		elif [ "$3" == "CA" ] ; then
			./${RUN_BINARY} +set fs_game osp +exec basic.cfg +exec server${3}.cfg +exec levels${3}-${2}.cfg +exec bots.cfg
		elif [ "$3" == "CAI" ] ; then
			./${RUN_BINARY} +set fs_game osp +exec basic.cfg +exec server${3}.cfg +exec levels${3}-${2}.cfg +exec bots.cfg
		else
			./${RUN_BINARY} +exec basic.cfg +exec server${3}.cfg +exec levels${3}-${2}.cfg +exec bots.cfg
		fi
		#clear_server
		echo "Stopped" > $STATUS_PATH
	fi
;;
ST|st|stop|STOP)
	# Close previous sessions
	clear_server
	echo "Stopped" > $STATUS_PATH
;;
CL|cl)
	# Clear all server logs
	clear_server_logs
;;
*)
	# Show menu
	echo ""
	echo "Quake 3 Server Script v${VERSION}"
	echo ""
	echo "Config.files: ${RUN_PATH}/server/*.cfg"
	echo "Log.files: ${BASEQ3_PATH}/*.log"
	echo "Status: $STATUS_PATH"
	echo ""
	echo "Syntax: q3aServ.sh <option> [1-5]"
	echo ""
	echo "Available options:"
	echo ""
	echo "CTF - Start Capture The Flag"
	echo "TD  - Start Team Deathmatch"
	echo "FFA - Start Free For All"
	echo "TOU - Start Tournament"
	echo "CG  - Custom Game"
	echo "CL  - Clear Logs"
	echo "ST  - Stop Current Server"
	echo ""
esac