#!/bin/sh
#################################################################
# Title : Qstat plugin for Munin                                #
# Author : Benjamin DUPUIS - Poil				#
# Email : poil@quake.fr						#
# First release :  18/10/2007					#
#---------------------------------------------------------------#

#################################################################
# Variable :							#
#---------------------------------------------------------------#
qstat_exe="${qstat_exe:-/usr/local/bin/qstat}"
bot_prefix="${bot_prefix:-}"

#---------------------------------------------------------------#
# End of config
script_name=$(basename "$0")
#################################################################

#################################################################
#       Help		                                        #
#---------------------------------------------------------------#
usage() {
	echo 'For testing the script, run qstat_ GameType IP Port'
	echo ' - GameType : q3s, q4s ... run qstat for seeing available gametype'
	echo 'For munin you must ln -s /usr/share/munin/plugins/qstat_ /etc/munin/plugins/qstat_GameType_IP2test_Port'
	echo 'Perhaps you must have to set qstat_exe path, actually on' "${qstat_exe}";
	echo 'For some GameTypes you have to specify bot_prefix';
	echo 'Have Fun'
}

config() {
	if [ "${script_name}" != "qstat_" ]; then
                gametype=$(echo "$script_name" | cut -d_ -f2)
                ip=$(echo "$script_name" | cut -d_ -f3)
                port=$(echo "$script_name" | cut -d_ -f4)
        else
                gametype=$1
                ip=$2
                port=$3
        fi

echo "graph_title Number of players on ${gametype} - ${ip}:${port}
graph_vlabel players
graph_args --base 1000 -r --lower-limit 0
graph_category games
maxplayer.label max players
player.label players
player.draw AREASTACK
bot.label bots
bot.draw AREASTACK"
}

#################################################################
#       Quake Stat, call qstat	                                #
#---------------------------------------------------------------#
qstat_run() {
	if [ "${script_name}" != "qstat_" ]; then
                gametype=$(echo "$script_name" | cut -d_ -f2)
                ip=$(echo "$script_name" | cut -d_ -f3)
                port=$(echo "$script_name" | cut -d_ -f4)
	else
		gametype=$1
		ip=$2
		port=$3
	fi

	if [ -n "$gametype" ] && [ -n "$ip" ] && [ -n "$port" ]; then
		rawstats=$("$qstat_exe" -raw ";" -nh "-$gametype" "${ip}:${port}")

		playervalue=$(echo "$rawstats" | cut -d\; -f6)
		maxplayervalue=$(echo "$rawstats" | cut -d\; -f5)
		
		if [ -n "$bot_prefix" ]; then
			botvalue=$("$qstat_exe" -P -pa -sort P "-$gametype" "${ip}:${port}" | grep frags | grep -c "$bot_prefix")
		else
			# Assume that bots have a ping time of 0 miliseconds
			botvalue=$("$qstat_exe" -P -pa -sort P "-$gametype" "${ip}:${port}" | grep -c 0ms)
		fi

		playervalue=$((playervalue-botvalue))

		if [ -z "${playervalue}" ]; then
			playervalue=0
		fi

		if [ -z "${botvalue}" ]; then
			botvalue=0
		fi

		if [ -z "${maxplayervalue}" ]; then
		        maxplayervalue=0
		fi


		echo "maxplayer.value ${maxplayervalue}";
		echo "player.value ${playervalue}";
		echo "bot.value ${botvalue}";
	else
		echo "maxplayer.value U"
		echo "player.value U"
		echo "bot.value U"
	fi
}

#################################################################
#	Main 							#
#---------------------------------------------------------------#
case $1 in
	config)
		config "$@"
		exit 0
   	;;
	help | ?)
		usage
		exit 0
	;;
	autoconf)
		echo "no (edit the script for set qstat path)"
	;;
	*)
		qstat_run "$@"
		exit 0
	;;
esac