#!/bin/bash
# LGSM command_stop.sh function
# Author: Daniel Gibbs
# Website: http://gameservermanagers.com
lgsm_version="271215"

# Description: Stops the server.

local modulename="Stopping"
function_selfname="$(basename $(readlink -f "${BASH_SOURCE[0]}"))"

# Attempts Graceful of source using rcon 'quit' command.
fn_stop_graceful_source(){
fn_print_dots "Graceful: rcon quit"
fn_scriptlog "Graceful: rcon quit"
# sends quit
tmux send -t "${servicename}" quit ENTER > /dev/null 2>&1
# waits up to 30 seconds giving the server time to shutdown gracefuly
for seconds in {1..30}; do
	pid=$(tmux list-sessions 2>&1|awk '{print $1}'|grep -Ec "^${servicename}:")
	if [ "${pid}" == "0" ]; then
		fn_print_ok_nl "Graceful: rcon quit: ${seconds}"
		fn_scriptlog "Graceful: rcon quit: OK: ${seconds} seconds"
		break
	fi
	sleep 1
	fn_print_dots "Graceful: rcon quit: ${seconds}"
done
if [ "${pid}" != "0" ]; then
	fn_print_fail_nl "Graceful: rcon quit"
	fn_scriptlog "Graceful: rcon quit: FAIL"
	fn_stop_tmux
fi
sleep 1
}

# Attempts Graceful of goldsource using rcon 'quit' command.
# Goldsource 'quit' command restarts rather than shutsdown
# this function will only wait 3 seconds then force a tmux shutdown.
# preventing the server from coming back online.
fn_stop_graceful_goldsource(){
fn_print_dots "Graceful: rcon quit"
fn_scriptlog "Graceful: rcon quit"
# sends quit
tmux send -t "${servicename}" quit ENTER > /dev/null 2>&1
# waits 3 seconds as goldsource servers restart with the quit command
for seconds in {1..3}; do
	sleep 1
	fn_print_dots "Graceful: rcon quit: ${seconds}"
done
fn_print_ok_nl "Graceful: rcon quit: ${seconds}"
sleep 1
fn_stop_tmux
}

# Attempts Graceful of 7 Days To Die using telnet.
fn_stop_telnet_sdtd(){
sdtdshutdown=$( expect -c '
proc abort {} {
	puts "Timeout or EOF\n"
	exit 1
}
spawn telnet '"${telnetip}"' '"${telnetport}"'
expect {
	"password:"     { send "'"${telnetpass}"'\r" }
	default         abort
}
expect {
	"session."  { send "shutdown\r" }
	default         abort
}
expect { eof }
puts "Completed.\n"
')
}

fn_stop_graceful_sdtd(){
# Gets server IP.
info_config.sh

fn_print_dots "Graceful: telnet"
fn_scriptlog "Graceful: telnet"
sleep 1

# uses localhost on first attempt.
telnetip=127.0.0.1
fn_print_dots "Graceful: telnet: ${telnetip}"
fn_scriptlog "Graceful: telnet: ${telnetip}"
fn_stop_telnet_sdtd
sleep 1

# falls back to the server ip if localhost fails.
refused=$(echo -en "\n ${sdtdshutdown}"| grep "Timeout or EOF")
if [ -n "${refused}" ]; then
	fn_print_warn_nl "Graceful: telnet: localhost: "
	fn_print_fail_eol
	fn_scriptlog "Graceful: telnet: localhost: FAIL"
	sleep 1

	telnetip=${ip}
	fn_print_dots "Graceful: telnet: ${telnetip}"
	fn_scriptlog "Graceful: telnet: ${telnetip}"
	fn_stop_telnet_sdtd
	refused=$(echo -en "\n ${sdtdshutdown}"| grep "Timeout or EOF")

	fn_print_warnnl "Graceful: telnet: ${telnetip}: "
	fn_print_fail_eol
	fn_scriptlog "Graceful: telnet: ${telnetip}: FAIL"
	sleep 1 
fi

# Checks if attempts have worked.
completed=$(echo -en "\n ${sdtdshutdown}"| grep "Completed.")
if [ -n "${completed}" ]; then
	fn_print_ok_nl "Graceful: telnet: "
	fn_print_ok_eol
	fn_scriptlog "Graceful: telnet: OK"
elif [ -n "${refused}" ]; then
	fn_print_fail_nl "Graceful: telnet: "
	fn_print_fail_eol
	fn_scriptlog "Graceful: telnet: ${telnetip}: FAIL"
	echo -en "\n\n" | tee -a "${scriptlog}"
	echo -en "Telnet output:" | tee -a "${scriptlog}"
	echo -en "\n ${sdtdshutdown}" | tee -a "${scriptlog}"
	echo -en "\n\n" | tee -a "${scriptlog}"
else
	fn_print_fail_nl "Graceful: telnet: Unknown error"
	fn_scriptlog "Graceful: telnet: Unknown error"
	echo -en "\n\n" | tee -a "${scriptlog}"
	echo -en "Telnet output:" | tee -a "${scriptlog}"
	echo -en "\n ${sdtdshutdown}" | tee -a "${scriptlog}"
	echo -en "\n\n" | tee -a "${scriptlog}" 
fi
}

fn_stop_graceful_select(){
if [ "${gamename}" == "7 Days to Die" ]; then
	fn_stop_graceful_sdtd
elif [ "${engine}" == "source" ]; then
	fn_stop_graceful_source
elif [ "${engine}" == "goldsource" ]; then
	fn_stop_graceful_goldsource
else
	fn_stop_tmux
fi		
}

fn_stop_teamspeak3(){
fn_print_dots "${servername}"
fn_scriptlog "${servername}"
sleep 1
${filesdir}/ts3server_startscript.sh stop > /dev/null 2>&1
# Remove lock file
rm -f "${rootdir}/${lockselfname}"
fn_print_ok_nl "${servername}"
fn_scriptlog "Stopped ${servername}"
}

fn_stop_tmux(){
fn_print_dots "${servername}"
fn_scriptlog "tmux kill-session: ${servername}"
sleep 1
# Kill tmux session
tmux kill-session -t "${servicename}" > /dev/null 2>&1
pid=$(tmux list-sessions 2>&1|awk '{print $1}'|grep -Ec "^${servicename}:")
if [ "${pid}" == "0" ]; then
	fn_print_ok_nl "${servername}"
	fn_scriptlog "Stopped ${servername}"
	sleep 1
	# Remove lock file
	rm -f "${rootdir}/${lockselfname}"
else
	fn_print_fail_nl "Unable to stop${servername}"
	fn_scriptlog "Unable to stop${servername}"
fi

}

# checks if the server is already stopped before trying to stop.
fn_stop_pre_check(){
if [ "${gamename}" == "Teamspeak 3" ]; then
	info_ts3status.sh
	if [ "${ts3status}" = "No server running (ts3server.pid is missing)" ]; then
		fn_print_ok_nl "${servername} is already stopped"
		fn_scriptlog "${servername} is already stopped"
	else
		fn_stop_teamspeak3
	fi      
else
	pid=$(tmux list-sessions 2>&1|awk '{print $1}'|grep -Ec "^${servicename}:")
	if [ "${pid}" == "0" ]; then
		fn_print_ok_nl "${servername} is already stopped"
		fn_scriptlog "${servername} is already stopped"
	else
		fn_stop_graceful_select
	fi
fi
}

check.sh
fn_print_dots "${servername}"
fn_scriptlog "${servername}"
sleep 1
fn_stop_pre_check