#!/bin/bash

# --------------------------------------------
# Quake 3 Server Script                      |
# Created by q3aql (q3aql@duck.com)          |
# Licensed by GPL v2.0                       |
# Last update: 21-08-2021                    |
# --------------------------------------------
VERSION="4.3"

# 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