OGP-Agent-Windows/bin/ogp_agent

#!/bin/bash
#
#
#	A wrapper script for the OGP agent perl script.
#	Performs auto-restarting of the agent on crash. You can
#	extend this to log crashes and more.
#
# The ogp_agent_run script should be at the top level of the agent tree
# Make sure we are in that directory since the script assumes this is the case
AGENTDIR="/OGP"
BASH_PREFS_CONF="$AGENTDIR/Cfg/bash_prefs.cfg"
chmod -Rf 770 $AGENTDIR
find $AGENTDIR -type f -print | xargs chmod ug=rw 
if [ -d "$AGENTDIR/steamcmd" ]; then
	find $AGENTDIR/steamcmd -iname \*.dll -exec chmod +x {} \;
	find $AGENTDIR/steamcmd -iname \*.exe -exec chmod +x {} \;
fi
if [ -d "$AGENTDIR/screenlogs" ]; then
	chmod -Rf ug=rwx $AGENTDIR/screenlogs
fi
chmod +x $AGENTDIR/ogp_agent.pl &> /dev/null
chmod +x $AGENTDIR/agent_conf.sh &> /dev/null
chmod +x /usr/bin/ogp_agent &> /dev/null

# Should we perform an automatic update?
if [ -e $BASH_PREFS_CONF ]
then
	source "$BASH_PREFS_CONF"
	if [ "X$agent_auto_update" == "X1" ]
	then
		AUTO_UPDATE="yes"
	fi
	
	if [ "X$run_pureftpd" == "X0" ]
	then
		RUN_PUREFTPD="no"
	else
		RUN_PUREFTPD="yes"
	fi
	
	if [ "X$ftp_ip" != "X" ]
	then
		FTP_IP="$ftp_ip"
	else
		FTP_IP="0.0.0.0"
	fi
	
	if [ "X$ftp_port" != "X" ]
	then
		FTP_PORT="$ftp_port"
	else
		FTP_PORT="21"
	fi
	
	if [ "X$ftp_pasv_range" != "X" ]
	then
		FTP_PASV_STRING="-p $ftp_pasv_range"
	else
		FTP_PASV_STRING=""
	fi
else
	AUTO_UPDATE="yes"
	RUN_PUREFTPD="yes"
	FTP_IP="0.0.0.0"
	FTP_PORT="21"
	FTP_PASV_STING=""
fi

if test `id -u` -eq 0; then
	echo
	echo
	echo "************** WARNING ***************"
	echo "Running the OGP agent as root  "
	echo "is highly discouraged. It is generally"
	echo "unnecessary to use root privileges to "
	echo "execute the agent.         "
	echo "**************************************"
	echo
	echo
	timeout=10
	while test $timeout -gt 0; do
		echo -n "The agent will continue to launch in $timeout seconds\r"
		timeout=`expr $timeout - 1`
		sleep 1
	done
fi

init() {
	RESTART="yes"
	AGENT="$AGENTDIR/ogp_agent.pl"
	TIMEOUT=10 # time to wait after a crash (in seconds)
	PID_FILE=""
	while test $# -gt 0; do
		case "$1" in
		"-pidfile")
			PID_FILE="$2"
			PID_FILE_SET=1
			echo $$ > $PID_FILE
			shift ;;			
		esac
		shift
	done

	if test ! -f "$AGENT"; then
		echo "ERROR: '$AGENT' not found, exiting"
		quit 1
	elif test ! -x "$AGENT"; then
		# Could try chmod but dont know what we will be
		# chmoding so just fail.
		echo "ERROR: '$AGENT' not executable, exiting"
		quit 1
	fi
}

syntax () {
	# Prints script syntax

	echo "Syntax:"
	echo "$0"
}

checkDepends() {
	CURL=`which curl 2>/dev/null`
	if test "$?" -gt 0; then
		echo "WARNING: Failed to locate curl binary."
	else
		echo "INFO: Located curl: $CURL"
	fi
	UNZIP=`which unzip 2>/dev/null`
	if test "$?" -gt 0; then
		echo "WARNING: Failed to locate unzip binary."
	else
		echo "INFO: Located unzip: $UNZIP"
	fi
}

update() {
	# Run the git update
	if test -n "$AUTO_UPDATE"; then
		if [ -z "$CURL" -o -z "$UNZIP" ]; then
			checkDepends
		fi
		if [ -f "$CURL" -a -x "$CURL" ] && [ -f "$UNZIP" -a -x "$UNZIP" ]; then
			cd $AGENTDIR
			if [ ! -d tmp ]; then
				mkdir tmp
			fi
			cd tmp
			REPONAME=OGP-Agent-Windows
			REVISION=`curl -Lks https://github.com/OpenGamePanel/${REPONAME}/commits/master.atom | grep -Eo "([a-f0-9]{40})" | head -n 1`
			curl -Os https://raw.githubusercontent.com/OpenGamePanel/${REPONAME}/${REVISION}/bin/ogp_agent
			diff ./ogp_agent /bin/ogp_agent &>/dev/null
			if test $? -ne 0; then
				cp -f ./ogp_agent /bin/ogp_agent &> /dev/null
				if test $? -eq 0; then
					cd /bin
					chmod ugo+x ogp_agent 2>/dev/null
					echo "`date`: The agent updater has been changed, relaunching..."
					rm -Rf tmp
					/bin/ogp_agent
					exit 0
				fi
			fi
			CURRENT=$(cat $AGENTDIR/Cfg/Config.pm | grep version | grep -Eo '[0-9a-f]{40}')
			if [ "$CURRENT" == "$REVISION" ]; then
				echo "The agent is up to date."
			else
				URL=https://github.com/OpenGamePanel/${REPONAME}/archive/${REVISION}.zip
				HEAD=$(curl -L -Os --head -w "%{http_code}" "$URL")
				if [ "$HEAD" == "200" ]; then
					echo "Updating agent using curl."
					curl -L -Os $URL
					if test $? -ne 0; then
						echo "`date`: curl failed to download the update package."
					else
						unzip -oq "${REVISION}.zip"
						if test $? -ne 0; then
							echo "`date`: Unable to unzip the update package."
							echo "cleaning up..."
							rm -Rf ${REPONAME}-* &> /dev/null
						else
							cd ${REPONAME}-${REVISION}
							cp -avf OGP/* $AGENTDIR/. &> /dev/null
							CP_APP_RET=$?
							cp -avf bin/* /usr/bin/. &> /dev/null
							CP_BIN_RET=$?
							if [ $CP_APP_RET -ne 0 -o $CP_BIN_RET -ne 0 ]; then
								echo "`date`: The agent files cannot be overwritten."
								echo "Cleaning up..."
								cd ..
								rm -Rf ${REPONAME}-* &> /dev/null
								echo "Agent update failed."
							else
								if test ! -e "$AGENTDIR/Cfg/Preferences.pm"; then
									cp -f Cfg/Preferences.pm $AGENTDIR/Cfg/Preferences.pm &> /dev/null
								fi
								echo "Fixing permissions..."
								chmod +x $AGENTDIR/ogp_agent.pl &> /dev/null
								chmod +x $AGENTDIR/agent_conf.sh &> /dev/null
								chmod +x /usr/bin/ogp_agent &> /dev/null
								echo "Cleaning up..."
								cd ..
								rm -Rf ${REPONAME}-* &> /dev/null
								sed -i "s/version.*/version => '${REVISION}',/" $AGENTDIR/Cfg/Config.pm
								echo "Agent updated successfully."
							fi
						fi
					fi
				else
					echo "There is a update available (${REVISION}) but the download source is not ready.";
					echo "Try again later."
				fi
			fi
			
		else
			echo "Update failed."
		fi
	fi
	
	return 0
}
	
run() {
	# Runs the update and agent
	update
	if test -n "$RESTART" ; then
		echo "Agent will auto-restart if there is a crash."
		#loop forever
		while true
		do
			# Run
			cd $AGENTDIR
			./ogp_agent.pl
			echo "`date`: Agent restart in $TIMEOUT seconds"
			# don't thrash the hard disk if the agent dies, wait a little
			sleep $TIMEOUT
		done # while true
	else
		cd $AGENTDIR
		./ogp_agent.pl
	fi
}

quit() {
	# Exits with the give error code, 1
	# if none specified.
	# exit code 2 also prints syntax
	exitcode="$1"

	# default to failure
	if test -z "$exitcode"; then
		exitcode=1
	fi

	case "$exitcode" in
	0)
		echo "`date`: OGP Agent Quit" ;;
	2)
		syntax ;;
	*)
		echo "`date`: OGP Agent Failed" ;;
	esac

	# Remove pid file
	if test -n "$PID_FILE" && test -f "$PID_FILE" ; then
		# The specified pid file
		rm -f $PID_FILE
	fi

	# reset SIGINT and then kill ourselves properly
	trap - 2
	kill -2 $$
}

PATH=/usr/local/bin:/usr/bin:${PATH}

# Start PureFTPD if OGP has been configured to manage FTP users
if [ ! -z "$RUN_PUREFTPD" ] && [ "$RUN_PUREFTPD" == "yes" ]; then
	/usr/sbin/pure-ftpd.exe -S ${FTP_IP},${FTP_PORT} ${FTP_PASV_STRING} -lpuredb:/etc/pureftpd.pdb -g /var/run/pure-ftpd.pid &
fi

# Initialise
init $*

# Run
run

# Quit normally
quit 0