netserver/HWProto.hs
author unc0rr
Thu, 09 Oct 2008 13:57:18 +0000
changeset 1332 19ff9bf7d69e
parent 1331 ae291cfd617a
child 1333 b0b0510eb82d
permissions -rw-r--r--
Fix that bug (lol, pretty fast, eh?)
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
890
1d8c4a5ec622 - Improve server core
unc0rr
parents:
diff changeset
     1
module HWProto where
1d8c4a5ec622 - Improve server core
unc0rr
parents:
diff changeset
     2
1d8c4a5ec622 - Improve server core
unc0rr
parents:
diff changeset
     3
import IO
896
93df8ac94382 Handle password parameter on JOIN
unc0rr
parents: 895
diff changeset
     4
import Data.List
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
     5
import Data.Word
890
1d8c4a5ec622 - Improve server core
unc0rr
parents:
diff changeset
     6
import Miscutils
1320
bffc7262e25e Optimize list lookups a bit
unc0rr
parents: 1317
diff changeset
     7
import Maybe
1317
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
     8
import qualified Data.Map as Map
890
1d8c4a5ec622 - Improve server core
unc0rr
parents:
diff changeset
     9
1331
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    10
teamToNet team = ["ADD_TEAM", teamname team, teamgrave team, teamfort team, show $ difficulty team] ++ hhsInfo
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    11
	where
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    12
		hhsInfo = concatMap (\(HedgehogInfo name hat) -> [name, hat]) $ hedgehogs team
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    13
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    14
answerBadCmd = [(clientOnly, ["ERROR", "Bad command, state or incorrect parameter"])]
1317
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
    15
answerNotMaster = [(clientOnly, ["ERROR", "You cannot configure room parameters"])]
1327
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
    16
answerBadParam = [(clientOnly, ["ERROR", "Bad parameter"])]
1309
1a38a967bd48 Fix a bug with 'ghosts' on server
unc0rr
parents: 1308
diff changeset
    17
answerQuit = [(clientOnly, ["off"])]
1327
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
    18
answerAbandoned = [(othersInRoom, ["BYE"])]
1309
1a38a967bd48 Fix a bug with 'ghosts' on server
unc0rr
parents: 1308
diff changeset
    19
answerQuitInform nick = [(othersInRoom, ["LEFT", nick])]
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    20
answerNickChosen = [(clientOnly, ["ERROR", "The nick already chosen"])]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    21
answerNickChooseAnother = [(clientOnly, ["WARNING", "Choose another nick"])]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    22
answerNick nick = [(clientOnly, ["NICK", nick])]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    23
answerProtocolKnown = [(clientOnly, ["ERROR", "Protocol number already known"])]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    24
answerBadInput = [(clientOnly, ["ERROR", "Bad input"])]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    25
answerProto protoNum = [(clientOnly, ["PROTO", show protoNum])]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    26
answerRoomsList list = [(clientOnly, ["ROOMS"] ++ list)]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    27
answerRoomExists = [(clientOnly, ["WARNING", "There's already a room with that name"])]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    28
answerJoined nick = [(sameRoom, ["JOINED", nick])]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    29
answerNoRoom = [(clientOnly, ["WARNING", "There's no room with that name"])]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    30
answerWrongPassword = [(clientOnly, ["WARNING", "Wrong password"])]
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    31
answerChatString nick msg = [(othersInRoom, ["CHAT_STRING", nick, msg])]
1317
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
    32
answerConfigParam paramName paramStrs = [(othersInRoom, "CONFIG_PARAM" : paramName : paramStrs)]
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
    33
answerFullConfig room = map toAnswer (Map.toList $ params room)
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
    34
	where
1321
d7dc4e86201e - Add protocol description (just started)
unc0rr
parents: 1320
diff changeset
    35
		toAnswer (paramName, paramStrs) =
1317
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
    36
			(clientOnly, "CONFIG_PARAM" : paramName : paramStrs)
1328
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
    37
answerCantAdd = [(clientOnly, ["WARNING", "Too many teams or hedgehogs, or same name team"])]
1325
c8994d47f41d Adding teams now works
unc0rr
parents: 1323
diff changeset
    38
answerTeamAccepted team = [(clientOnly, ["TEAM_ACCEPTED", teamname team])]
1331
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    39
answerAddTeam team = [(othersInRoom, teamToNet team)]
1327
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
    40
answerHHNum teamName hhNumber = [(othersInRoom, ["HH_NUM", teamName, show hhNumber])]
1328
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
    41
answerRemoveTeam teamName = [(othersInRoom, ["REMOVE_TEAM", teamName])]
1329
69ddc231a911 - Only team owner can remove team from the list
unc0rr
parents: 1328
diff changeset
    42
answerNotOwner = [(clientOnly, ["ERROR", "You do not own this team"])]
1330
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
    43
answerTeamColor teamName newColor = [(othersInRoom, ["TEAM_COLOR", teamName, newColor])]
1331
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    44
answerAllTeams room = concatMap toAnswer (teams room)
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    45
	where
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    46
		toAnswer team =
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    47
			[(clientOnly, teamToNet team),
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    48
			(clientOnly, ["TEAM_COLOR", teamname team, teamcolor team]),
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
    49
			(clientOnly, ["HH_NUM", teamname team, show $ hhnum team])]
1307
ce26e16d18ab - Now actually fix
unc0rr
parents: 1305
diff changeset
    50
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    51
-- Main state-independent cmd handler
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    52
handleCmd :: CmdHandler
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    53
handleCmd client _ rooms ("QUIT":xs) =
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    54
	if null (room client) then
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    55
		(noChangeClients, noChangeRooms, answerQuit)
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    56
	else if isMaster client then
1327
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
    57
		(noChangeClients, removeRoom (room client), answerQuit ++ answerAbandoned) -- core disconnects clients on ROOMABANDONED answer
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    58
	else
1309
1a38a967bd48 Fix a bug with 'ghosts' on server
unc0rr
parents: 1308
diff changeset
    59
		(noChangeClients, noChangeRooms, answerQuit ++ (answerQuitInform $ nick client))
895
6aee2f335726 - Remove old hwserv code
unc0rr
parents: 894
diff changeset
    60
1307
ce26e16d18ab - Now actually fix
unc0rr
parents: 1305
diff changeset
    61
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    62
-- check state and call state-dependent commmand handlers
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    63
handleCmd client clients rooms cmd =
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    64
	if null (nick client) || protocol client == 0 then
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    65
		handleCmd_noInfo client clients rooms cmd
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    66
	else if null (room client) then
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    67
		handleCmd_noRoom client clients rooms cmd
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    68
	else
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    69
		handleCmd_inRoom client clients rooms cmd
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    70
1307
ce26e16d18ab - Now actually fix
unc0rr
parents: 1305
diff changeset
    71
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    72
-- 'no info' state - need to get protocol number and nickname
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    73
handleCmd_noInfo :: CmdHandler
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    74
handleCmd_noInfo client clients _ ["NICK", newNick] =
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    75
	if not . null $ nick client then
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    76
		(noChangeClients, noChangeRooms, answerNickChosen)
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    77
	else if haveSameNick then
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    78
		(noChangeClients, noChangeRooms, answerNickChooseAnother)
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    79
	else
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    80
		(modifyClient client{nick = newNick}, noChangeRooms, answerNick newNick)
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    81
	where
1320
bffc7262e25e Optimize list lookups a bit
unc0rr
parents: 1317
diff changeset
    82
		haveSameNick = isJust $ find (\cl -> newNick == nick cl) clients
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    83
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    84
handleCmd_noInfo client _ _ ["PROTO", protoNum] =
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    85
	if protocol client > 0 then
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    86
		(noChangeClients, noChangeRooms, answerProtocolKnown)
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    87
	else if parsedProto == 0 then
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    88
		(noChangeClients, noChangeRooms, answerBadInput)
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    89
	else
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    90
		(modifyClient client{protocol = parsedProto}, noChangeRooms, answerProto parsedProto)
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    91
	where
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    92
		parsedProto = fromMaybe 0 (maybeRead protoNum :: Maybe Word16)
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    93
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
    94
handleCmd_noInfo _ _ _ _ = (noChangeClients, noChangeRooms, answerBadCmd)
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    95
1307
ce26e16d18ab - Now actually fix
unc0rr
parents: 1305
diff changeset
    96
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
    97
-- 'noRoom' clients state command handlers
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    98
handleCmd_noRoom :: CmdHandler
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
    99
handleCmd_noRoom client _ rooms ["LIST"] =
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
   100
		(noChangeClients, noChangeRooms, answerRoomsList $ map name rooms)
903
d4e5d8cbe449 Implement LIST command
unc0rr
parents: 902
diff changeset
   101
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
   102
handleCmd_noRoom client _ rooms ["CREATE", newRoom, roomPassword] =
895
6aee2f335726 - Remove old hwserv code
unc0rr
parents: 894
diff changeset
   103
	if haveSameRoom then
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
   104
		(noChangeClients, noChangeRooms, answerRoomExists)
895
6aee2f335726 - Remove old hwserv code
unc0rr
parents: 894
diff changeset
   105
	else
1317
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
   106
		(modifyClient client{room = newRoom, isMaster = True}, addRoom (RoomInfo newRoom roomPassword (protocol client) [] Map.empty), answerJoined $ nick client)
895
6aee2f335726 - Remove old hwserv code
unc0rr
parents: 894
diff changeset
   107
	where
1320
bffc7262e25e Optimize list lookups a bit
unc0rr
parents: 1317
diff changeset
   108
		haveSameRoom = isJust $ find (\room -> newRoom == name room) rooms
895
6aee2f335726 - Remove old hwserv code
unc0rr
parents: 894
diff changeset
   109
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
   110
handleCmd_noRoom client clients rooms ["CREATE", newRoom] =
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
   111
	handleCmd_noRoom client clients rooms ["CREATE", newRoom, ""]
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
   112
	
1308
d5dcd6cfa5e2 Fix another server failure (when second client in room disconnects)
unc0rr
parents: 1307
diff changeset
   113
handleCmd_noRoom client clients rooms ["JOIN", roomName, roomPassword] =
902
3cc10f0aae37 Finish conversion
unc0rr
parents: 901
diff changeset
   114
	if noSuchRoom then
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
   115
		(noChangeClients, noChangeRooms, answerNoRoom)
1321
d7dc4e86201e - Add protocol description (just started)
unc0rr
parents: 1320
diff changeset
   116
	else if roomPassword /= password clRoom then
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
   117
		(noChangeClients, noChangeRooms, answerWrongPassword)
895
6aee2f335726 - Remove old hwserv code
unc0rr
parents: 894
diff changeset
   118
	else
1331
ae291cfd617a Send teams info to newly connected client (has a bug with team sequence, need to discover)
unc0rr
parents: 1330
diff changeset
   119
		(modifyClient client{room = roomName}, noChangeRooms, (answerJoined $ nick client) ++ answerNicks ++ answerFullConfig clRoom ++ answerAllTeams clRoom)
895
6aee2f335726 - Remove old hwserv code
unc0rr
parents: 894
diff changeset
   120
	where
1320
bffc7262e25e Optimize list lookups a bit
unc0rr
parents: 1317
diff changeset
   121
		noSuchRoom = isNothing $ find (\room -> roomName == name room) rooms
1308
d5dcd6cfa5e2 Fix another server failure (when second client in room disconnects)
unc0rr
parents: 1307
diff changeset
   122
		answerNicks = [(clientOnly, ["JOINED"] ++ (map nick $ filter (\ci -> room ci == roomName) clients))]
1321
d7dc4e86201e - Add protocol description (just started)
unc0rr
parents: 1320
diff changeset
   123
		clRoom = roomByName roomName rooms
895
6aee2f335726 - Remove old hwserv code
unc0rr
parents: 894
diff changeset
   124
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
   125
handleCmd_noRoom client clients rooms ["JOIN", roomName] =
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
   126
	handleCmd_noRoom client clients rooms ["JOIN", roomName, ""]
894
2ca76a7f3121 - Fixed some bugs
unc0rr
parents: 893
diff changeset
   127
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
   128
handleCmd_noRoom _ _ _ _ = (noChangeClients, noChangeRooms, answerBadCmd)
895
6aee2f335726 - Remove old hwserv code
unc0rr
parents: 894
diff changeset
   129
1307
ce26e16d18ab - Now actually fix
unc0rr
parents: 1305
diff changeset
   130
897
35d91fa3753b 'In room' state stub
unc0rr
parents: 896
diff changeset
   131
-- 'inRoom' clients state command handlers
1082
596b1dcdc1df - Modify network protocol to use new delimiter
unc0rr
parents: 903
diff changeset
   132
handleCmd_inRoom :: CmdHandler
1322
c624b04699fb Fix protocol implementation to conform documentation
unc0rr
parents: 1321
diff changeset
   133
handleCmd_inRoom client _ _ ["CHAT_STRING", msg] =
1317
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
   134
	(noChangeClients, noChangeRooms, answerChatString (nick client) msg)
897
35d91fa3753b 'In room' state stub
unc0rr
parents: 896
diff changeset
   135
1327
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   136
handleCmd_inRoom client _ rooms ("CONFIG_PARAM" : paramName : paramStrs) =
1317
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
   137
	if isMaster client then
1322
c624b04699fb Fix protocol implementation to conform documentation
unc0rr
parents: 1321
diff changeset
   138
		(noChangeClients, modifyRoom clRoom{params = Map.insert paramName paramStrs (params clRoom)}, answerConfigParam paramName paramStrs)
1317
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
   139
	else
13cf8c5a7428 Server now fully supports game options
unc0rr
parents: 1309
diff changeset
   140
		(noChangeClients, noChangeRooms, answerNotMaster)
1321
d7dc4e86201e - Add protocol description (just started)
unc0rr
parents: 1320
diff changeset
   141
	where
d7dc4e86201e - Add protocol description (just started)
unc0rr
parents: 1320
diff changeset
   142
		clRoom = roomByName (room client) rooms
d7dc4e86201e - Add protocol description (just started)
unc0rr
parents: 1320
diff changeset
   143
1327
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   144
handleCmd_inRoom client _ rooms ("ADD_TEAM" : name : color : grave : fort : difStr : hhsInfo)
1323
d166f9069c2b Build neccessary structures in memory on ADDTEAM message, but don't send answer yet (need to review team id concept)
unc0rr
parents: 1322
diff changeset
   145
	| length hhsInfo == 16 =
1328
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   146
	if length (teams clRoom) == 6 || canAddNumber <= 0 || isJust findTeam then
1323
d166f9069c2b Build neccessary structures in memory on ADDTEAM message, but don't send answer yet (need to review team id concept)
unc0rr
parents: 1322
diff changeset
   147
		(noChangeClients, noChangeRooms, answerCantAdd)
d166f9069c2b Build neccessary structures in memory on ADDTEAM message, but don't send answer yet (need to review team id concept)
unc0rr
parents: 1322
diff changeset
   148
	else
1332
19ff9bf7d69e Fix that bug (lol, pretty fast, eh?)
unc0rr
parents: 1331
diff changeset
   149
		(noChangeClients, modifyRoom clRoom{teams = teams clRoom ++ [newTeam]}, answerTeamAccepted newTeam ++ answerAddTeam newTeam)
1323
d166f9069c2b Build neccessary structures in memory on ADDTEAM message, but don't send answer yet (need to review team id concept)
unc0rr
parents: 1322
diff changeset
   150
	where
d166f9069c2b Build neccessary structures in memory on ADDTEAM message, but don't send answer yet (need to review team id concept)
unc0rr
parents: 1322
diff changeset
   151
		clRoom = roomByName (room client) rooms
1329
69ddc231a911 - Only team owner can remove team from the list
unc0rr
parents: 1328
diff changeset
   152
		newTeam = (TeamInfo (nick client) name color grave fort difficulty newTeamHHNum (hhsList hhsInfo))
1328
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   153
		findTeam = find (\t -> name == teamname t) $ teams clRoom
1323
d166f9069c2b Build neccessary structures in memory on ADDTEAM message, but don't send answer yet (need to review team id concept)
unc0rr
parents: 1322
diff changeset
   154
		difficulty = fromMaybe 0 (maybeRead difStr :: Maybe Int)
1325
c8994d47f41d Adding teams now works
unc0rr
parents: 1323
diff changeset
   155
		hhsList [] = []
1323
d166f9069c2b Build neccessary structures in memory on ADDTEAM message, but don't send answer yet (need to review team id concept)
unc0rr
parents: 1322
diff changeset
   156
		hhsList (n:h:hhs) = HedgehogInfo n h : hhsList hhs
1327
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   157
		canAddNumber = 18 - (sum . map hhnum $ teams clRoom)
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   158
		newTeamHHNum = min 4 canAddNumber
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   159
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   160
handleCmd_inRoom client _ rooms ["HH_NUM", teamName, numberStr] =
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   161
	if not $ isMaster client then
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   162
		(noChangeClients, noChangeRooms, answerNotMaster)
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   163
	else
1329
69ddc231a911 - Only team owner can remove team from the list
unc0rr
parents: 1328
diff changeset
   164
		if hhNumber < 1 || hhNumber > 8 || noSuchTeam || hhNumber > (canAddNumber + (hhnum team)) then
1327
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   165
			(noChangeClients, noChangeRooms, answerBadParam)
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   166
		else
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   167
			(noChangeClients, modifyRoom $ modifyTeam clRoom team{hhnum = hhNumber}, answerHHNum teamName hhNumber)
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   168
	where
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   169
		hhNumber = fromMaybe 0 (maybeRead numberStr :: Maybe Int)
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   170
		noSuchTeam = isNothing findTeam
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   171
		team = fromJust findTeam
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   172
		findTeam = find (\t -> teamName == teamname t) $ teams clRoom
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   173
		clRoom = roomByName (room client) rooms
9d43a6e6b9ca Can choose hedgehogs number now
unc0rr
parents: 1325
diff changeset
   174
		canAddNumber = 18 - (sum . map hhnum $ teams clRoom)
1323
d166f9069c2b Build neccessary structures in memory on ADDTEAM message, but don't send answer yet (need to review team id concept)
unc0rr
parents: 1322
diff changeset
   175
1330
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   176
handleCmd_inRoom client _ rooms ["TEAM_COLOR", teamName, newColor] =
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   177
	if not $ isMaster client then
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   178
		(noChangeClients, noChangeRooms, answerNotMaster)
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   179
	else
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   180
		(noChangeClients, modifyRoom $ modifyTeam clRoom team{teamcolor = newColor}, answerTeamColor teamName newColor)
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   181
	where
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   182
		noSuchTeam = isNothing findTeam
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   183
		team = fromJust findTeam
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   184
		findTeam = find (\t -> teamName == teamname t) $ teams clRoom
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   185
		clRoom = roomByName (room client) rooms
12c13ffb426f - Allow team color changing
unc0rr
parents: 1329
diff changeset
   186
1328
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   187
handleCmd_inRoom client _ rooms ["REMOVE_TEAM", teamName] =
1329
69ddc231a911 - Only team owner can remove team from the list
unc0rr
parents: 1328
diff changeset
   188
	if noSuchTeam then
69ddc231a911 - Only team owner can remove team from the list
unc0rr
parents: 1328
diff changeset
   189
		(noChangeClients, noChangeRooms, answerBadParam)
1328
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   190
	else
1329
69ddc231a911 - Only team owner can remove team from the list
unc0rr
parents: 1328
diff changeset
   191
		if not $ nick client == teamowner team then
69ddc231a911 - Only team owner can remove team from the list
unc0rr
parents: 1328
diff changeset
   192
			(noChangeClients, noChangeRooms, answerNotOwner)
1328
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   193
		else
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   194
			(noChangeClients, modifyRoom clRoom{teams = filter (\t -> teamName /= teamname t) $ teams clRoom}, answerRemoveTeam teamName)
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   195
	where
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   196
		noSuchTeam = isNothing findTeam
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   197
		team = fromJust findTeam
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   198
		findTeam = find (\t -> teamName == teamname t) $ teams clRoom
c41344e3c236 Add support for removing team
unc0rr
parents: 1327
diff changeset
   199
		clRoom = roomByName (room client) rooms
1083
3448dd03483f Further work on dedicated server
unc0rr
parents: 1082
diff changeset
   200
1304
05cebf68ebd8 Start refactoring standalone server (prepare to change protocol)
unc0rr
parents: 1302
diff changeset
   201
handleCmd_inRoom _ _ _ _ = (noChangeClients, noChangeRooms, answerBadCmd)