#!/usr/bin/env python3

import sys, socket, inspect

opensockets=[]

encoding="latin-1"

def connect():
	sock=socket.socket()
	sock.connect(("localhost",1423))
	opensockets.append(sock)
	return sock

def close(sock):
	sock.close()
	opensockets.remove(sock)

def sendraw(sock,text):
	sock.sendall(text.encode(encoding))

def sendline(sock,line):
	sendraw(sock,line+"\n")

def recvraw(sock,length):
	text=""
	while len(text)<length:
		text+=sock.recv(length-len(text)).decode(encoding)
	return text

def recvline(sock):
	line=""
	while True:
		c=sock.recv(1).decode(encoding)
		if c=="\n":
			break
		line+=c
	return line

def recvint(sock):
	line=recvline(sock)
	spl=line.split(" ")
	assert len(spl)==3 and spl[0]=="int"
	return int(spl[2])

def sendlineok(sock,line):
	sendline(sock,line)
	cmd=line.split(" ")[0]
	expect(sock,"ok "+cmd)

def sendlineerror(sock,line,msg):
	sendline(sock,line)
	cmd=line.split(" ")[0]
	expect(sock,"error "+cmd+" "+msg)

def sendlinelist(sock,line,lst):
	sendline(sock,line)
	cmd=line.split(" ")[0]
	expect(sock,"list "+cmd+" "+str(len(lst))+("" if len(lst)==0 else " "+" ".join(lst)))

def expectraw(sock,text):
	got=recvraw(sock,len(text))
	if got!=text:
		print("\nEXPECTRAW error: expected '"+text+"', got '"+got+"'")
		sys.exit(1)

def expect(sock,line):
	got=recvline(sock)
	if got!=line:
		print("\nEXPECT error: expected '"+line+"', got '"+got+"'")
		sys.exit(1)


testfunctions=[]

def testfunction(tag):
	def dec(func):
		def newfunc():
			global opensockets
			socks=[]
			for i in range(len(inspect.signature(func).parameters)):
				socks.append(connect())
			func(*socks)
			for sock in opensockets: sock.close()
			opensockets=[]
		testfunctions.append((tag,newfunc))
	return dec

def runtests():
	for (tag,f) in testfunctions:
		print("- "+tag+": ",end="")
		sys.stdout.flush()
		f()
		print("\x1B[32mOK\x1B[0m")
	print("\x1B[32m>> All OK!\x1B[0m")



@testfunction("Pingpong")
def testping(s):
	sendline(s,"ping")
	expect(s,"pong")

@testfunction("Create room, list, query")
def testroom(s,t):
	sendlinelist(s,"room_query",[])
	sendlinelist(t,"room_query",[])
	sendlinelist(s,"room_list game",[])
	sendlinelist(t,"room_list game",[])
	sendlineok(s,"room_create game room 1 2")
	sendlinelist(s,"room_query",["room"])
	sendlinelist(t,"room_query",[])
	sendlinelist(s,"room_list game",["room"])
	sendlinelist(t,"room_list game",["room"])

@testfunction("Create private room, list, query")
def testprivateroom(s,t):
	sendlineok(s,"room_create game room 0 2")
	sendlinelist(s,"room_query",["room"])
	sendlinelist(t,"room_query",[])
	sendlinelist(s,"room_list game",[])
	sendlinelist(t,"room_list game",[])

@testfunction("Join room")
def testjoin(s,t):
	sendlineok(s,"room_create game room 1 2")
	sendline(s,"id"); sid=recvint(s)
	sendline(t,"id"); tid=recvint(t)
	sendlineok(t,"room_join game room")
	expect(s,"room_join game room "+str(tid))
	sendlinelist(t,"room_query",["room"])

	sendlineok(t,"room_create game room2 1 1")
	sendlinelist(t,"room_query",["room","room2"])
	sendlineerror(s,"room_join game room2","Room full")

	sendlineok(s,"room_leave game room")
	expect(t,"room_leave game room "+str(sid))

@testfunction("Player list, messaging")
def testid(s,t):
	sendline(s,"id"); sid=recvint(s)
	sendline(t,"id"); tid=recvint(t)
	sendlineok(s,"room_create game room 0 2")
	sendlineok(t,"room_join game room")
	expect(s,"room_join game room "+str(tid))
	sendlinelist(s,"room_player_list game room",[str(sid),str(tid)])
	sendlinelist(t,"room_player_list game room",[str(sid),str(tid)])

	sendline(s,"room_message game room "+str(tid)+" 4")
	sendraw(s,"kaas")
	expect(s,"ok room_message")
	expect(t,"room_message game room "+str(sid)+" 4")
	expectraw(t,"kaas")

	sendline(t,"room_message game room "+str(sid)+" 256")
	sendraw(t,"".join(chr(i) for i in range(256)))
	expect(t,"ok room_message")
	expect(s,"room_message game room "+str(tid)+" 256")
	expectraw(s,"".join(chr(i) for i in range(256)))



if __name__=="__main__":
	runtests()