module HWProtoLobbyState where

import qualified Data.Map as Map

import qualified Data.IntMap as IntMap

import qualified Data.IntSet as IntSet

import qualified Data.Foldable as Foldable

import Maybe

import Data.List

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

import CoreTypes

import Actions

import Utils

answerAllTeams teams = concatMap toAnswer teams

	where

		toAnswer team =

			[AnswerThisClient $ teamToNet team,

			AnswerThisClient ["TEAM_COLOR", teamname team, teamcolor team],

			AnswerThisClient ["HH_NUM", teamname team, show $ hhnum team]]

handleCmd_lobby :: CmdHandler

handleCmd_lobby clID clients rooms ["LIST"] =

	[AnswerThisClient ("ROOMS" : roomsInfoList)]

	where

		roomsList = IntMap.elems rooms

		protocol = clientProto client

		client = clients IntMap.! clID

		roomInfo room = [

				name room,

				show $ gameinprogress room

				]

handleCmd_lobby clID clients _ ["CHAT", msg] =

	[AnswerOthersInRoom ["CHAT", clientNick, msg]]

	where

		clientNick = nick $ clients IntMap.! clID

		[RoomRemoveThisClient "", -- leave lobby

		AddRoom newRoom roomPassword,

		AnswerThisClient ["NOT_READY", clientNick]

		]

	where

		clientNick = nick $ clients IntMap.! clID

		haveSameRoom = isJust $ find (\room -> newRoom == name room) $ IntMap.elems rooms

handleCmd_lobby clID clients rooms ["CREATE_ROOM", newRoom] =

	handleCmd_lobby clID clients rooms ["CREATE_ROOM", newRoom, ""]

		[RoomRemoveThisClient "", -- leave lobby

		RoomAddThisClient rID] -- join room

		++ answerNicks

		++ answerReady

		++ [AnswerThisRoom ["NOT_READY", nick client]]

		++ answerFullConfig

		++ answerTeams

		++ watchRound

	where

		noSuchRoom = isNothing mbRoom

		jRoom = fromJust mbRoom

		rID = roomUID jRoom

		client = clients IntMap.! clID

		roomClientsIDs = IntSet.elems $ playersIDs jRoom

		toAnswer (paramName, paramStrs) = AnswerThisClient $ "CFG" : paramName : paramStrs

		answerFullConfig = map toAnswer (leftConfigPart ++ rightConfigPart)

		(leftConfigPart, rightConfigPart) = partition (\(p, _) -> p /= "MAP") (Map.toList $ params jRoom)

		watchRound = if not $ gameinprogress jRoom then

					[]

				else

					[AnswerThisClient  ["RUN_GAME"],

		answerTeams = if gameinprogress jRoom then

				answerAllTeams (teamsAtStart jRoom)

			else

				answerAllTeams (teams jRoom)

handleCmd_lobby clID clients rooms ["JOIN_ROOM", roomName] =

	handleCmd_lobby clID clients rooms ["JOIN_ROOM", roomName, ""]

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

	-- Administrator's stuff --

handleCmd_lobby clID clients rooms ["KICK", kickNick] =

	where

		client = clients IntMap.! clID

		maybeClient = Foldable.find (\cl -> kickNick == nick cl) clients

		noSuchClient = isNothing maybeClient

		kickID = clientUID $ fromJust maybeClient

handleCmd_lobby clID clients rooms ["BAN", banNick] =

	if not $ isAdministrator client then

		[]

	else

		BanClient banNick : handleCmd_lobby clID clients rooms ["KICK", banNick]

	where

		client = clients IntMap.! clID

handleCmd_lobby clID clients rooms ["SET_SERVER_MESSAGE", newMessage] =

	where

		client = clients IntMap.! clID

handleCmd_lobby clID clients rooms ["CLEAR_ACCOUNTS_CACHE"] =

	where

		client = clients IntMap.! clID

handleCmd_lobby clID _ _ _ = [ProtocolError "Incorrect command (state: in lobby)"]