module HWProtoCore where

import qualified Data.IntMap as IntMap

import Data.Foldable

import Maybe

--------------------------------------

import CoreTypes

import Actions

import Utils

import HWProtoNEState

import HWProtoLobbyState

import HWProtoInRoomState

handleCmd, handleCmd_loggedin :: CmdHandler

handleCmd clID _ _ ["PING"] = [AnswerThisClient ["PONG"]]

handleCmd clID clients rooms ("QUIT" : xs) =

	[ByeClient msg]

	where

		msg = if not $ null xs then head xs else ""

handleCmd clID clients _ ["PONG"] =

	if pingsQueue client == 0 then

		[ProtocolError "Protocol violation"]

	else

		[ModifyClient (\cl -> cl{pingsQueue = pingsQueue cl - 1})]

	where

		client = clients IntMap.! clID

handleCmd clID clients rooms cmd =

	if not $ logonPassed client then

		handleCmd_NotEntered clID clients rooms cmd

	else

		handleCmd_loggedin clID clients rooms cmd

	where

		client = clients IntMap.! clID

handleCmd_loggedin clID clients rooms ["INFO", asknick] =

	if noSuchClient then

		[]

	else

		[AnswerThisClient

			["INFO",

			nick client,

			"[" ++ host client ++ "]",

			protoNumber2ver $ clientProto client,

			"[" ++ roomInfo ++ "]" ++ roomStatus]]

	where

		maybeClient = find (\cl -> asknick == nick cl) clients

		noSuchClient = isNothing maybeClient

		client = fromJust maybeClient

		room = rooms IntMap.! roomID client

		roomInfo = if roomID client /= 0 then roomMasterSign ++ "room " ++ (name room) else adminSign ++ "lobby"

		roomMasterSign = if isMaster client then "@" else ""

		adminSign = if isAdministrator client then "@" else ""

		roomStatus =

			if gameinprogress room

			then if teamsInGame client > 0 then "(playing)" else "(spectating)"

			else ""

handleCmd_loggedin clID clients rooms ["FOLLOW", asknick] =

	if inLobby || noSuchClient then

		[]

	else

		handleCmd_lobby clID clients rooms ["JOIN_ROOM", roomname]

	where

		maybeClient = find (\cl -> asknick == nick cl) clients

		noSuchClient = isNothing maybeClient

		client = fromJust maybeClient

		room = rooms IntMap.! roomID client

		roomname = (name room)

		inLobby = roomname == ""

handleCmd_loggedin clID clients rooms cmd =

	if roomID client == 0 then

		handleCmd_lobby clID clients rooms cmd

	else

		handleCmd_inRoom clID clients rooms cmd

	where

		client = clients IntMap.! clID