#!/bin/bash

#   Copyright 2016 Christian 'Alloc' Illy
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.


# Provides common functions for 7dtd-scripts. Not intended to be run directly.

# Check if the script is run as root (exit otherwise) and load global config
checkRootLoadConf() {
	if [ `id -u` -ne 0 ]; then
		echo "This script has to be run as root!"
		exit 10
	fi
	. /etc/7dtd.conf
}

# Get the config path for the given instance
# Params:
#   1: Instance name
# Returns:
#   Config path for instance
getInstancePath() {
	echo $SDTD_BASE/instances/$1
}

# Check if the given instance name is valid (no blanks, no special chars,
# only letters, digits, underscore, hyphen -> [A-Za-z0-9_\-])
# Params:
#   1: Instance name
# Returns:
#   0/1 instance not valid/valid
isValidInstanceName() {
	if [[ "$1" =~ ^[A-Za-z0-9_\-]+$ ]]; then
		echo 1
	else
		echo 0
	fi
}

# Check if the given instance name is an existing instance
# Params:
#   1: Instance name
# Returns:
#   0/1 instance not valid/valid
isValidInstance() {
	if [ ! -z "$1" ]; then
		if [ $(isValidInstanceName "$1") -eq 1 ]; then
			if [ -d $(getInstancePath "$1") ]; then
				if [ -f $(getInstancePath "$1")/config.xml ]; then
					echo 1
					return
				fi
			fi
		fi
	fi
	echo 0
}

# Check if the given instance is currently running
# Params:
#   1: Instance name
# Returns:
#   0 = not running
#   1 = running
isRunning() {
	$SSD --status --pidfile $(getInstancePath $1)/7dtd.pid
	if [ $? -eq 0 ]; then
		echo 1
	else
		echo 0
	fi
}

# Get list of defined instances
# Returns:
#   List of instances
getInstanceList() {
	local IF
	for IF in $SDTD_BASE/instances/*; do
		local I=`basename $IF`
		if [ $(isValidInstance $I) -eq 1 ]; then
			echo $I
		fi
	done
}

# Get the PID of the instance if it is running, 0 otherwise
# Params:
#   1: Instance name
# Returns:
#   0 if not running
#   PID otherwise
getInstancePID() {
	if [ $(isRunning $1) -eq 1 ]; then
		cat $(getInstancePath $1)/7dtd.pid
	else
		echo 0
	fi
}

# Get the installed branch name
# Returns:
#   "public" if no engine installed or no appmanifest found or buildid could not be read
#   Branch name
getLocalBranch() {
	local LOCAL="public"
	if [ -d "$SDTD_BASE/engine" ]; then
		local APPMANIFEST=$(find $SDTD_BASE/engine -type f -name "appmanifest_294420.acf")
		if [ -f "$APPMANIFEST" ]; then
			LOCAL=$(grep betakey "$APPMANIFEST" | tr '[:blank:]"' ' ' | tr -s ' ' | cut -d\  -f3)
			if [[ -z $LOCAL ]]; then
				LOCAL="public"
			else
				echo $LOCAL
				return
			fi
		fi
	fi
	echo $LOCAL
}

# Get the local engine version number (i.e. build id)
# Returns:
#   0 if no engine installed or no appmanifest found or buildid could not be read
#   Build Id otherwise
getLocalEngineVersion() {
	local LOCAL=0
	if [ -d "$SDTD_BASE/engine" ]; then
		local APPMANIFEST=$(find $SDTD_BASE/engine -type f -name "appmanifest_294420.acf")
		if [ -f "$APPMANIFEST" ]; then
			LOCAL=$(grep buildid "$APPMANIFEST" | tr '[:blank:]"' ' ' | tr -s ' ' | cut -d\  -f3)
			if [ $(isANumber "$LOCAL") -eq 0 ]; then
				LOCAL=0
			fi
		fi
	fi
	echo $LOCAL
}

# Get the local engine update time
# Returns:
#   0 if no engine installed or no appmanifest found or buildid could not be read
#   Update time otherwise
getLocalEngineUpdateTime() {
	local LOCAL=0
	if [ -d "$SDTD_BASE/engine" ]; then
		local APPMANIFEST=$(find $SDTD_BASE/engine -type f -name "appmanifest_294420.acf")
		if [ -f "$APPMANIFEST" ]; then
			LOCAL=$(grep LastUpdated "$APPMANIFEST" | tr '[:blank:]"' ' ' | tr -s ' ' | cut -d\  -f3)
			if [ $(isANumber "$LOCAL") -eq 0 ]; then
				LOCAL=0
			else
				date --date="@${LOCAL}" "+%Y-%m-%d %H:%M:%S"
				return
			fi
		fi
	fi
	echo $LOCAL
}


# Get the latest remote (on Steam) engine version numbers etc
updateRemoteEngineInfo() {
	local DOCHECK=no
	if [ ! -e /tmp/7dtd-appinfo ]; then
		DOCHECK=yes
	else
		AGE=$((`date +%s` - `stat -L --format %Y /tmp/7dtd-appinfo`))
		if [ $AGE -gt 300 ]; then
			DOCHECK=yes
		fi
	fi
	if [ "$DOCHECK" = "yes" ]; then
		echo "Updating version information..."
		rm -f /root/Steam/appcache/appinfo.vdf
		cd $SDTD_BASE/steamcmd
		
		expect /usr/local/lib/7dtd/steamcmd.exp $SDTD_BASE/steamcmd/steamcmd.sh 294420 2>/dev/null > /tmp/7dtd-appinfo

		local BUILDID=$(getRemoteBuildId public)

		if [ $(isANumber "$BUILDID") -eq 0 ]; then
			rm -f /tmp/7dtd-appinfo
		fi
	fi
}

# Get the latest build id (on Steam)
# Params:
#   1. Branch name
# Returns:
#   "?" if data could not be retrieved
#   BuildId otherwise
getRemoteBuildId() {
	if [ ! -f "/tmp/7dtd-appinfo" ]; then
		echo "?"
		return
	fi
	
	local BUILDID=$(grep -A 1000 \"branches\" /tmp/7dtd-appinfo | grep -A 1000 \"$1\" | grep -B 10 \} --max-count=1 | grep \"buildid\" | cut -d\" -f4)

	if [ $(isANumber "$BUILDID") -eq 0 ]; then
		echo "?"
	else
		echo $BUILDID
	fi
}

# Get the update time of the latest build (on Steam)
# Params:
#   1. Branch name
# Returns:
#   "?" if data could not be retrieved
#   Update timestamp otherwise
getRemoteBuildUpdateTime() {
	if [ ! -f "/tmp/7dtd-appinfo" ]; then
		echo "?"
		return
	fi
	
	local TIMESTAMP=$(grep -A 1000 \"branches\" /tmp/7dtd-appinfo | grep -A 1000 \"$1\" | grep -B 10 \} --max-count=1 | grep \"timeupdated\" | cut -d\" -f4)
	
	if [ $(isANumber "$TIMESTAMP") -eq 0 ]; then
		echo "?"
	else
		date --date="@${TIMESTAMP}" "+%Y-%m-%d %H:%M:%S"
	fi
}

# Get a list of available branch names, blank separated
# Returns:
#   Blank separated list of branch names (can be empty if an error occured)
getBranchNames() {
	if [ ! -f "/tmp/7dtd-appinfo" ]; then
		return
	fi
	
	grep -A 1000 \"branches\" /tmp/7dtd-appinfo | grep -E '^[[:space:]]*"[^"]+"[[:space:]]*$' | tail --lines=+2 | cut -d\" -f2
}

# Check if a given port range (baseport, baseport+1, baseport+2 each udp)
# is already in use by any other instance
# Params:
#   1: Baseport
#   2: Current instance (ignored)
# Returns:
#   0/1 not in use/in use
checkGamePortUsed() {
	local PORTMIN=$1
	local PORTMAX=$(( $1 + 2 ))
	local I
	for I in $(getInstanceList); do
		if [ "$2" != "$I" ]; then
			local CURPORTMIN=$(getConfigValue $I "ServerPort")
			local CURPORTMAX=$(( $CURPORTMIN + 2 ))
			if [ $PORTMAX -ge $CURPORTMIN -a $PORTMIN -le $CURPORTMAX ]; then
				echo 1
				return
			fi
		fi
	done
	echo 0
}

# Check if a given TCP port is already in use by any instance (either by control
# panel or telnet)
# Params:
#   1: Port
# Returns:
#   0/1 not in use/in use
checkTCPPortUsed() {
	local I
	for I in $(getInstanceList); do
		if [ "$2" != "$I" ]; then
			local CURENABLED=$(getConfigValue $I "TelnetEnabled")
			local CURPORT=$(getConfigValue $I "TelnetPort")
			if [ "$CURENABLED" = "true" -a $CURPORT -eq $1 ]; then
				echo 1
				return
			fi
			CURENABLED=$(getConfigValue $I "ControlPanelEnabled")
			CURPORT=$(getConfigValue $I "ControlPanelPort")
			if [ "$CURENABLED" = "true" -a $CURPORT -eq $1 ]; then
				echo 1
				return
			fi
		fi
	done
	echo 0
}

# Send a single command to the telnet port
# Params:
#   1: Instance name
#   2: Command
#   3: (Optional) Timeout in sec, defaulting to 1
# Returns:
#   String of telnet output
telnetCommand() {
	local TEL_ENABLED=$(getConfigValue $1 TelnetEnabled)
	local TEL_PORT=$(getConfigValue $1 TelnetPort)
	local TEL_PASS=$(getConfigValue $1 TelnetPassword)	
	if [ "$TEL_ENABLED" = "true" ]; then
		local TEMPFILE=$(mktemp)
		rm -f $TEMPFILE
		mkfifo $TEMPFILE
		exec 3<> $TEMPFILE
		nc 127.0.0.1 $TEL_PORT <&3 &
		local NCPID=$!
		disown
		if [ -n "$TEL_PASS" ]; then
			printf "$TEL_PASS\n$2\n" >&3
		else
			printf "$2\n" >&3
		fi
		sleep ${3:-1}
		printf "exit\n" >&3
		sleep 0.2
		kill -9 $NCPID > /dev/null 2>&1
		exec 3>&-
		rm -f $TEMPFILE
	else
		echo "Telnet not enabled."
	fi
}

# Get all hook files for the given hook-name
# Params:
#   1: Hook name
#   2: Instance name
# Returns:
#   Names of hook files
getHooksFor() {
	if [ -n "$2" ]; then
		if [ -d $(getInstancePath $2)/hooks/$1 ]; then
			local H
			for H in $(getInstancePath $2)/hooks/$1/*.sh; do
				echo "$H"
			done
		fi
	fi
	if [ -d $SDTD_BASE/hooks/$1 ]; then
		local H
		for H in $SDTD_BASE/hooks/$1/*.sh; do
			echo "$H"
		done
	fi
}

# Lowercase passed string
# Params:
#   1: String
# Returns:
#   Lowercased string
lowercase() {
	echo "${1}" | tr "[:upper:]" "[:lower:]"
}

# Prepare passed string as part of camelcase, i.e. first char upper case, others
# lowercase
# Params:
#   1: String
# Returns:
#   Transformed string
camelcasePrep() {
	echo $(echo "${1:0:1}" | tr "[:lower:]" "[:upper:]")$(echo "${1:1}" | tr "[:upper:]" "[:lower:]")
}

# Check if given value is a (integer) number
# Params:
#   1: Value
# Returns:
#   0/1 for NaN / is a number
isANumber() {
	if [[ $1 =~ ^[0-9]+$ ]] ; then
		echo "1"
	else
		echo "0"
	fi
}

# Check if given value is a boolean (true/false, yes/no, y/n)
# Params:
#   1: Value
# Returns:
#   0/1
isABool() {
	local LOW=$(lowercase "$1")
	if [ "$LOW" = "false" -o "$LOW" = "true"\
		-o "$LOW" = "yes" -o "$LOW" = "y"\
		-o "$LOW" = "no" -o "$LOW" = "n" ]; then
		echo 1
	else
		echo 0
	fi
}

# Convert the given value to a boolean 0/1
# Params:
#   1: Value
# Returns:
#   0/1 as false/true
getBool() {
	if [ $(isABool "$1") -eq 0 ]; then
		echo 0
	else
		local LOW=$(lowercase "$1")
		if [ "$LOW" = "true" -o "$LOW" = "yes" -o "$LOW" = "y" ]; then
			echo 1
		else
			echo 0
		fi
	fi
}

listCommands() {
	local C
	for C in $(declare -F | cut -d\  -f3 | grep "^sdtdCommand"\
			| grep -v "Help$"\
			| grep -v "Description$"\
			| grep -v "Expects$"); do
		local CMD=$(lowercase "${C#sdtdCommand}")
		printf "%s " "$CMD"
	done
}

. /usr/local/lib/7dtd/help.sh
. /usr/local/lib/7dtd/serverconfig.sh
for M in /usr/local/lib/7dtd/commands/*.sh; do
	. $M
done

checkRootLoadConf