# -*- coding: utf-8 -*-

# Modular system for querying game servers.
#
# Author:  Roman Vasiyarov <rvasiyarov@gmail.com>
# License: GPLv3
# $Id$

#
# Nexuiz module
#


import sys
import struct
from datetime import datetime


NAMES = [
	"gamename",
	"modname",
	"gameversion",
	"sv_maxclients",
	"clients",
	"bots",
	"mapname",
	"hostname",
	"protocol",
	"qcstatus",

	# should be last
	"ping"
]

NEXUIZ_MAGIC = 0xffffffff


def process(pkt):
	n   = 4  # "header" size
	seq = None
	out = ""

	try:
		seq = struct.unpack("<I", pkt[:n])
		out = pkt[n:].decode("utf-8")
	except:
		pass

	res = "infoResponse\n"
	try:
		if seq != NEXUIZ_MAGIC and not out.startswith(res):
			raise

		# format: \gamename\Nexuiz\modname\data...
		lst = out.split("\\")
		out = lst[2::2]
		return out
	except:
		return [None] * (len(NAMES) - 1)


def query(sock, servers, verbose = False):
	if not sock or not len(servers):
		return

	info = {}
	pkt  = struct.pack("<I", NEXUIZ_MAGIC)
	pkt += b"getinfo"

	# send queries
	for sv in servers:
		if verbose:
			sys.stderr.write("sending query to %s...\n" % repr(sv))
		info[sv] = [None, datetime.utcnow()]
		sock.sendto(pkt, 0, sv)

	return info