dstds_tools.sh

#!/bin/bash
_scriptname="dstds-tools.sh"

# Are we running with a non-GNU readlink? Use homebrew's gnu version explicitly:
g=""
${g}readlink -m / > /dev/null 2>&1 || g=g
# See if the GNU version is installed
${g}readlink -m / > /dev/null 2>&1 || {
	echo "FATAL: (g)readlink -m doesn't work on your system. Provide a GNU environment for this script and try again." >&2
	exit 1
}

# Manually set steamcmd path here only if it's not in the PATH. It's o
#_STEAMPATH="/home/steam/steamcmd/steamcmd.sh"

_STEAMUPDARGS="validate"
# On MacOS, provide path to the folder containing the app bundle
_DSROOT=${_DSROOT:="/home/dontstarve/DST"}
_DSBIN="dontstarve_dedicated_server_nullrenderer"
#_DSARGS="-console" # this arg is deprecated, console is configured in cluster.ini
_DSARGS=${_DSARGS="-skip_update_server_mods -ugc_directory ../mods_ugc"}
unset _PSR
unset _CONFDIR
unset _CLUSTER
unset _SHARDS
_SHARDS=()
unset _MASTERSHARD

# set -x
set -e
export _top_pid=$$
trap "exit 1" TERM

function parse_cluster_directory {
# $1 = path to cluster
	local _cluster_canonical="$(${g}readlink -m "$1")"
	_CLUSTER="$(basename "$_cluster_canonical")"
	_cluster_canonical="$(${g}readlink -m "${_cluster_canonical}/..")"
	_CONFDIR="$(basename "$_cluster_canonical")"
	_PSR="$(${g}readlink -m "${_cluster_canonical}/..")"
	if [[ "$_CONFDIR" == "/" ]]; then
		echo "WARN: Confdir ($_CONFDIR) must not contain slashes! Changing to (.)" >&2
		_CONFDIR="."
	fi
}

function collect_shards {
# $1 = path to cluster
	local _shard
	while IFS= read -r -d $'\n' _shard; do
		echo "Found shard: $_shard"
		_SHARDS+=( "$_shard" )
	done < <(find "$1" -type f -name server.ini -print | sed -E 's|.*/([^/]+)/[^/]+$|\1|')
	[ ${#_SHARDS[@]} -gt 0 ] || { \
		echo "ERROR: No shards found." >&2; kill -s TERM $_top_pid; }
	_MASTERSHARD="$(find "$1" -type f -name server.ini -exec grep -qE "^[[:space:]]*is_master[[:space:]]*\=[[:space:]]*true[[:space:]]*" {} \; -print | sed -E 's|.*/([^/]+)/[^/]+$|\1|')"
	_MASTERSHARD="${_MASTERSHARD##*/}"
	echo "Detected master shard: $_MASTERSHARD"
	[ -n "$_MASTERSHARD" ] || { \
		echo "ERROR: Master shard not found!" >&2 ; kill -s TERM $_top_pid; }
}

function create_cluster {
# $1 = path to cluster
#	TODO
	echo TODO
}

function validate_cluster {
# $1 = path to cluster
	[ -f "$1/cluster.ini" ] || { echo "ERROR: cluster.ini not found." >&2; kill -s TERM $_top_pid; }
# TODO?
}

function start_cluster {
# $1 = path to cluster
	# cluster named by its directory name
	parse_cluster_directory "$1"
	local _cluster_canonical="$(${g}readlink -m "$1")"
	screen -S "$_CLUSTER" -Q select . && {\
		echo "ERROR: screen session already exists, cluster is already running." >&2 ; kill -s TERM $_top_pid; }
	# check if we should be using shards
	collect_shards "$1"
	echo "Starting master shard: $_MASTERSHARD"
	set -x
	(
		cd "${_DSROOT}/bin" || cd "${_DSROOT}/dontstarve_dedicated_server_nullrenderer.app/Contents/MacOS" || cd "${_DSROOT}/Contents/MacOS"
		screen -dm -S "$_CLUSTER" -p + -t "$_MASTERSHARD" ./$_DSBIN $_DSARGS -persistent_storage_root $_PSR -conf_dir "$_CONFDIR" -cluster "$_CLUSTER" -shard "$_MASTERSHARD"
		sleep 4
	)
	for ((i=0;i<${#_SHARDS[@]};i++)); do
		local _shard=${_SHARDS[$i]}
		if [[ "$_shard" != "$_MASTERSHARD" ]]; then
			echo "Starting slave: $_shard"
			(
			screen -S "$_CLUSTER" -X screen -t "$_shard" ./$_DSBIN $_DSARGS -persistent_storage_root $_PSR -conf_dir "$_CONFDIR" -cluster "$_CLUSTER" -shard "$_shard"
			)
		fi
	done
	set +x
}

function stop_cluster {
# $1 = cluster name (NOT PATH!)
# $2 = announce message (optional)
# $3 = timeout (optional)
	# Stop if screen with such name isn't found
	screen -S "$1" -Q select . || {\
		echo "ERROR: screen session with specified name not found." >&2 ; kill -s TERM $_top_pid; }
	# Get list of windows in the session and parse it
	local _swindows=()
	while IFS= read -r -d '' _window; do
		echo "Found screen window: $_window"
		_swindows+=("$_window")
	done < <( ( screen -S "$1" -Q windows; echo -ne "\0" ) | sed -E "s/  /"$'\n'"/g"  | sed -E "s/^[[:digit:]]\+ //")
	local _skip="yes"
	local _timeout=${3:-"90"}
	for ((i=0;i<${#_swindows[@]};i++)); do
		local WINDOW="${_swindows[$i]}"
		if [[ $i == 0 ]]; then
			# First window is always the master shard. For now, just announce the server is going down.
			echo "Announcing shutdown..."
			screen -S "$1" -p "$WINDOW" -X stuff $'c_announce("'"${2:-"Server is shutting down in $_timeout seconds."}"'",nil,"leave_game")\r'
			sleep $_timeout
		else
			# Other windows are all slaves - we've already waited, it's time to shut down slaves
			echo "Shutting down slave: $WINDOW"
			(
				# Ctrl+C (SIGINT)
				screen -S "$1" -p "$WINDOW" -X stuff $'\cc'
				# c_shutdown() results in duplicated snapshots
				# screen -S $1 -p $WINDOW -X stuff $'c_shutdown()\r'
			) & sleep 2
		fi
	done
	wait
	# Finally, shut down master shard
	echo "Shutting down master: ${_swindows[0]}"
	screen -S "$1" -p "${_swindows[0]}" -X stuff $'\cc'
}

function usage {
	cat << EOF
Usage:				dstds-tools.sh [OPTION] [CLUSTER]

This script is a tool to control the execution of the dedicated server executable for Don\'t Starve Together.

Available options are:
  -s CLUSTERPATH		Start DS server cluster located in CLUSTERPATH
  -n CLUSTERPATH		Create new DS server cluster in CLUSTERPATH (start without checking for existence)
  -t CLUSTER			Terminate cluster with specified name
  -b MESSAGE			Broadcast message to all running clusters
  -u				Update infrastructure:
				 0. Notify all clusters about upcoming downtime (90 seconds)
				 1. Terminate all clusters
				 2. Update DST DS steam app.
				 3. Update mods.
				 4. Restart previously running clusters.

EOF
}

function update_all {
# TODO
	echo TODO
#	exit 1
	# Work-in-progress
	# TODO - note and stop all running clusters
	"$_STEAMPATH" \
		+@ShutdownOnFailedCommand 1 \
		+@NoPromptForPassword 1 \
		+login anonymous \
		+force_install_dir "$_DSROOT" \
		+app_update 343050 \
		+quit
	sleep 10
#	(
#		cd "${_DSROOT}/bin"
#		./$_DSBIN -only_update_server_mods
#	)
	# TODO - restart noted clusters
}

while getopts "s:n:t:b:u" OPT; do
	case $OPT in
	  s)
		# Start existing cluster
		validate_cluster "$OPTARG"
		start_cluster "$OPTARG"
		;;
	  n)
		# Create new cluster
		start_cluster "$OPTARG"
		;;
	  t)
		# Terminate cluster by name
		stop_cluster "$OPTARG" '' "15"
		;;
	  b)
		# TODO
		echo "NOT IMPLEMENTED"
		;;
	  u)
		update_all
		;;
	  \?)
		echo "Invalid option: -$OPTARG" >&2
		usage
		;;
	  :)
		echo "Option $OPTARG requires an argument." >&2
		;;
	  *)
		usage
		;;
	esac
done