#Insert the line "import dc_query" at the top of you __init__.py file
#in your BF2's admin/standard_admin directory, then use the bf2_rcon_class
#DC_query extension to utilize this file's functionality

import bf2
import host
import new, default
from bf2.stats.constants import *

def registerRConCommand(command, function): 
	# Put the function into a new instancemethod object bound to the AdminServer
	newMethod = new.instancemethod(function, default.server, default.AdminServer)
	
	# Create a new class attribute in AdminServer that's wired to our function
	setattr(default.AdminServer, command, newMethod)
	
	# Add the new command into the rcon dispatch table, pointing to our function
	default.server.rcon_cmds[command] = newMethod
	
def getPlayerList(self, ctx, cmd):
	output = ('player_id\tname\tteam\tsquad\tsquadleader\tcommander\tip\t'
	'connected\tvalid\tremote\tai\tping\talive\tmandown\tprofile_id\t'
	'flagholder\tsuicide\tspawngroup\ttimetospawn\tscore_kills\tscore_deaths\t'
	'score_tks\tscore_suicides\tscore_score\tscore_heals\tscore_ammos\t'
	'score_repairs\tscore_damageassists\tscore_passengerassists\t'
	'score_driverassists\tscore_targetassists\tscore_revives\t'
	'score_teamdamage\tscore_teamvehicledamage\tscore_cpcaptures\t'
	'score_cpdefends\tscore_cpassists\tscore_cpneutralizes\t'
	'score_cpneutralizeassets\tscore_skillscore\tscore_rplscore\t'
	'score_cmdscore\tscore_fracscore\trank\tvehicletype\tvehicle\t'
	'kittype\tkit\n')
	
	try:
		for p in bf2.playerManager.getPlayers():
			output += str(p.index)
			output += '\t' + p.getName()
			output += '\t' + str(p.getTeam())
			output += '\t' + str(p.getSquadId())
			output += '\t' + str(p.isSquadLeader())
			output += '\t' + str(p.isCommander())
			output += '\t' + str(p.getAddress())
			output += '\t' + str(p.isConnected())
			output += '\t' + str(p.isValid())
			output += '\t' + str(p.isRemote())
			output += '\t' + str(p.isAIPlayer())
			output += '\t' + str(p.getPing())
			output += '\t' + str(p.isAlive())
			output += '\t' + str(p.isManDown())
			output += '\t' + str(p.getProfileId())
			output += '\t' + str(p.isFlagHolder())
			output += '\t' + str(p.getSuicide())
			output += '\t' + str(p.getSpawnGroup())
			output += '\t' + str(p.getTimeToSpawn())		
			output += '\t' + str(p.score.kills)
			output += '\t' + str(p.score.deaths)
			output += '\t' + str(p.score.TKs)
			output += '\t' + str(p.score.suicides)
			output += '\t' + str(p.score.score)
			output += '\t' + str(p.score.heals)
			output += '\t' + str(p.score.ammos)
			output += '\t' + str(p.score.repairs)
			output += '\t' + str(p.score.damageAssists)
			output += '\t' + str(p.score.passengerAssists)
			output += '\t' + str(p.score.driverAssists)
			output += '\t' + str(p.score.targetAssists)
			output += '\t' + str(p.score.revives)
			output += '\t' + str(p.score.teamDamages)
			output += '\t' + str(p.score.teamVehicleDamages)
			output += '\t' + str(p.score.cpCaptures)
			output += '\t' + str(p.score.cpDefends)
			output += '\t' + str(p.score.cpAssists)
			output += '\t' + str(p.score.cpNeutralizes)
			output += '\t' + str(p.score.cpNeutralizeAssists)
			output += '\t' + str(p.score.skillScore)
			output += '\t' + str(p.score.rplScore)
			output += '\t' + str(p.score.cmdScore)
			output += '\t' + str(p.score.fracScore)
			output += '\t' + str(p.score.rank)

			try:
				output += '\t' + str(getVehicleType(p.getVehicle().templateName))
				output += '\t' + str(p.getVehicle().templateName)
			except:
				output += '\t' + str(VEHICLE_TYPE_UNKNOWN)
				output += '\t' + 'unknown'

			try:
				output += '\t' + str(getKitType(p.getKit().templateName))
				output += '\t' + str(p.getKit().templateName)
			except:
				output += '\t' + str(KIT_TYPE_UNKNOWN)
				output += '\t' + 'unknown'
			output += '\n'
	except:
		output = 'Error retriving player information.'

	ctx.write(output)

def getServerInfo(self, ctx, cmd):
	output = ('servername\tplayers_connected\tplayers_joining\tmod_dir\t'
	'mapname\tnextmap\tteam1_name\tteam2_name\tteam1_tickets\tteam2_tickets\t'
	'team1_defaulttickets\tteam2_defaulttickets\tticketratio\tteamratio\t'
	'numplayers\tmaxplayers\tgamemode\ttimelimit\tscorelimit\t'
	'autobalanceteam\ttkpunishenabled\ttknumpunishtokick\ttkpunishbydefault\t'
	'useglobalrank\tuseglobalunlocks\n')
	
	connected = 0
	joining = 0
	try:
		for p in bf2.playerManager.getPlayers():
			if p.isConnected():
				connected += 1
			else:
				joining += 1
	except:
		pass
			
	numplayers = joining + connected
	try:
		output += host.rcon_invoke('sv.serverName').strip()
		output += '\t' + str(connected)
		output += '\t' + str(joining)
		output += '\t' + str(bf2.gameLogic.getModDir())
		output += '\t' + str(bf2.gameLogic.getMapName().replace('_', ' ').title())
		output += '\t' + str(host.rcon_invoke('maplist.list').splitlines()[int(host.rcon_invoke('admin.nextLevel').strip())].split()[1].strip('"').replace('_', ' ').title())
		output += '\t' + str(bf2.gameLogic.getTeamName(1))
		output += '\t' + str(bf2.gameLogic.getTeamName(2))
		output += '\t' + str(bf2.gameLogic.getTickets(1))
		output += '\t' + str(bf2.gameLogic.getTickets(2))
		output += '\t' + str(bf2.gameLogic.getDefaultTickets(1))
		output += '\t' + str(bf2.gameLogic.getDefaultTickets(2))
		output += '\t' + str(bf2.serverSettings.getTicketRatio())
		output += '\t' + str(bf2.serverSettings.getTeamRatioPercent())
		output += '\t' + str(numplayers)
		output += '\t' + str(bf2.serverSettings.getMaxPlayers())
		output += '\t' + str(bf2.serverSettings.getGameMode())
		output += '\t' + str(bf2.serverSettings.getTimeLimit())
		output += '\t' + str(bf2.serverSettings.getScoreLimit())
		output += '\t' + str(bf2.serverSettings.getAutoBalanceTeam())
		output += '\t' + str(bf2.serverSettings.getTKPunishEnabled())
		output += '\t' + str(bf2.serverSettings.getTKNumPunishToKick())
		output += '\t' + str(bf2.serverSettings.getTKPunishByDefault())
		output += '\t' + str(bf2.serverSettings.getUseGlobalRank())
		output += '\t' + str(bf2.serverSettings.getUseGlobalUnlocks())
		output += '\n'
	except:
		output = 'Error retriving server information.'

	ctx.write(output)

registerRConCommand('dc_pl', getPlayerList)
registerRConCommand('dc_si', getServerInfo)