gameServer/Actions.hs
author unc0rr
Sun, 04 Nov 2012 21:28:03 +0400
changeset 7947 0cf5277fef1a
parent 7945 4006d77e1a28
child 7972 0dffb91eeb82
permissions -rw-r--r--
Better place for SendUpdateOnThisRoom
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
6012
6bac93097da3 Store replays for further analysis
unc0rr
parents: 5996
diff changeset
     1
{-# LANGUAGE CPP, OverloadedStrings #-}
7766
98edc0724a28 Fix most of server warnings
unc0rr
parents: 7757
diff changeset
     2
{-# OPTIONS_GHC -fno-warn-orphans #-}
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
     3
module Actions where
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
     4
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
     5
import Control.Concurrent
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
     6
import qualified Data.Set as Set
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
     7
import qualified Data.Sequence as Seq
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
     8
import qualified Data.List as L
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
     9
import qualified Control.Exception as Exception
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    10
import System.Log.Logger
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    11
import Control.Monad
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    12
import Data.Time
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    13
import Data.Maybe
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    14
import Control.Monad.Reader
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    15
import Control.Monad.State.Strict
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    16
import qualified Data.ByteString.Char8 as B
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    17
import Control.DeepSeq
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    18
import Data.Unique
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    19
import Control.Arrow
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    20
import Control.Exception
5209
f7a610e2ef5f On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents: 5184
diff changeset
    21
import System.Process
f7a610e2ef5f On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents: 5184
diff changeset
    22
import Network.Socket
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    23
-----------------------------
7766
98edc0724a28 Fix most of server warnings
unc0rr
parents: 7757
diff changeset
    24
#if defined(OFFICIAL_SERVER)
5996
2c72fe81dd37 Convert boolean variable + a bunch of fields which make sense only while game is going on into Maybe + structure
unc0rr
parents: 5426
diff changeset
    25
import OfficialServer.GameReplayStore
7766
98edc0724a28 Fix most of server warnings
unc0rr
parents: 7757
diff changeset
    26
#endif
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    27
import CoreTypes
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    28
import Utils
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    29
import ClientIO
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    30
import ServerState
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    31
import Consts
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    32
import ConfigFile
6068
e18713ecf1e0 Introduce EngineInteraction module
unc0rr
parents: 6012
diff changeset
    33
import EngineInteraction
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    34
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    35
data Action =
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    36
    AnswerClients ![ClientChan] ![B.ByteString]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    37
    | SendServerMessage
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    38
    | SendServerVars
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    39
    | MoveToRoom RoomIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    40
    | MoveToLobby B.ByteString
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    41
    | RemoveTeam B.ByteString
6753
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
    42
    | SendTeamRemovalMessage B.ByteString
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    43
    | RemoveRoom
6758
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
    44
    | FinishGame
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    45
    | UnreadyRoomClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    46
    | JoinLobby
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    47
    | ProtocolError B.ByteString
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    48
    | Warning B.ByteString
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    49
    | NoticeMessage Notice
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    50
    | ByeClient B.ByteString
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    51
    | KickClient ClientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    52
    | KickRoomClient ClientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    53
    | BanClient NominalDiffTime B.ByteString ClientIndex
5426
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
    54
    | BanIP B.ByteString NominalDiffTime B.ByteString
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
    55
    | BanList
7748
f160fbc139b1 UNBAN implementation
unc0rr
parents: 7735
diff changeset
    56
    | Unban B.ByteString
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    57
    | ChangeMaster
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    58
    | RemoveClientTeams ClientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    59
    | ModifyClient (ClientInfo -> ClientInfo)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    60
    | ModifyClient2 ClientIndex (ClientInfo -> ClientInfo)
7757
c20e6c80e249 Don't accept ROUNDFINISHED message twice. Fixes game hangs when half of teams quit game.
unc0rr
parents: 7748
diff changeset
    61
    | ModifyRoomClients (ClientInfo -> ClientInfo)
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    62
    | ModifyRoom (RoomInfo -> RoomInfo)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    63
    | ModifyServerInfo (ServerInfo -> ServerInfo)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    64
    | AddRoom B.ByteString B.ByteString
7921
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
    65
    | SendUpdateOnThisRoom
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    66
    | CheckRegistered
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    67
    | ClearAccountsCache
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    68
    | ProcessAccountInfo AccountInfo
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    69
    | AddClient ClientInfo
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    70
    | DeleteClient ClientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    71
    | PingAll
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    72
    | StatsAction
5209
f7a610e2ef5f On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents: 5184
diff changeset
    73
    | RestartServer
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    74
    | AddNick2Bans B.ByteString B.ByteString UTCTime
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    75
    | AddIP2Bans B.ByteString B.ByteString UTCTime
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    76
    | CheckBanned
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    77
    | SaveReplay
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    78
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    79
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    80
type CmdHandler = [B.ByteString] -> Reader (ClientIndex, IRnC) [Action]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    81
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    82
instance NFData Action where
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    83
    rnf (AnswerClients chans msg) = chans `deepseq` msg `deepseq` ()
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    84
    rnf a = a `seq` ()
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    85
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    86
instance NFData B.ByteString
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    87
instance NFData (Chan a)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    88
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    89
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    90
othersChans :: StateT ServerState IO [ClientChan]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    91
othersChans = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    92
    cl <- client's id
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    93
    ri <- clientRoomA
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    94
    liftM (map sendChan . filter (/= cl)) $ roomClientsS ri
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    95
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    96
processAction :: Action -> StateT ServerState IO ()
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    97
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    98
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
    99
processAction (AnswerClients chans msg) =
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   100
    io $ mapM_ (`writeChan` (msg `deepseq` msg)) (chans `deepseq` chans)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   101
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   102
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   103
processAction SendServerMessage = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   104
    chan <- client's sendChan
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   105
    protonum <- client's clientProto
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   106
    si <- liftM serverInfo get
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   107
    let message = if protonum < latestReleaseVersion si then
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   108
            serverMessageForOldVersions si
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   109
            else
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   110
            serverMessage si
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   111
    processAction $ AnswerClients [chan] ["SERVER_MESSAGE", message]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   112
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   113
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   114
processAction SendServerVars = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   115
    chan <- client's sendChan
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   116
    si <- gets serverInfo
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   117
    io $ writeChan chan ("SERVER_VARS" : vars si)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   118
    where
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   119
        vars si = [
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   120
            "MOTD_NEW", serverMessage si,
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   121
            "MOTD_OLD", serverMessageForOldVersions si,
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   122
            "LATEST_PROTO", showB $ latestReleaseVersion si
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   123
            ]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   124
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   125
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   126
processAction (ProtocolError msg) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   127
    chan <- client's sendChan
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   128
    processAction $ AnswerClients [chan] ["ERROR", msg]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   129
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   130
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   131
processAction (Warning msg) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   132
    chan <- client's sendChan
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   133
    processAction $ AnswerClients [chan] ["WARNING", msg]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   134
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   135
processAction (NoticeMessage n) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   136
    chan <- client's sendChan
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   137
    processAction $ AnswerClients [chan] ["NOTICE", showB . fromEnum $ n]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   138
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   139
processAction (ByeClient msg) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   140
    (Just ci) <- gets clientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   141
    ri <- clientRoomA
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   142
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   143
    chan <- client's sendChan
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   144
    clNick <- client's nick
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   145
    loggedIn <- client's logonPassed
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   146
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   147
    when (ri /= lobbyId) $ do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   148
        processAction $ MoveToLobby ("quit: " `B.append` msg)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   149
        return ()
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   150
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   151
    clientsChans <- liftM (Prelude.map sendChan . Prelude.filter logonPassed) $! allClientsS
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   152
    io $
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   153
        infoM "Clients" (show ci ++ " quits: " ++ B.unpack msg)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   154
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   155
    when loggedIn $ processAction $ AnswerClients clientsChans ["LOBBY:LEFT", clNick, msg]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   156
7465
c2dcf97ca664 Okay, this is workaround over ping timeouts problem on the server. Could make server crash if recieve thread wakes up after second ping timeout event.
unc0rr
parents: 7351
diff changeset
   157
    mapM processAction
c2dcf97ca664 Okay, this is workaround over ping timeouts problem on the server. Could make server crash if recieve thread wakes up after second ping timeout event.
unc0rr
parents: 7351
diff changeset
   158
        [
c2dcf97ca664 Okay, this is workaround over ping timeouts problem on the server. Could make server crash if recieve thread wakes up after second ping timeout event.
unc0rr
parents: 7351
diff changeset
   159
        AnswerClients [chan] ["BYE", msg]
7735
4c7e282b5732 Reset nickname so it may be reused while old connection is still hanging
unc0rr
parents: 7710
diff changeset
   160
        , ModifyClient (\c -> c{nick = "", logonPassed = False}) -- this will effectively hide client from others while he isn't deleted from list
7465
c2dcf97ca664 Okay, this is workaround over ping timeouts problem on the server. Could make server crash if recieve thread wakes up after second ping timeout event.
unc0rr
parents: 7351
diff changeset
   161
        ]
c2dcf97ca664 Okay, this is workaround over ping timeouts problem on the server. Could make server crash if recieve thread wakes up after second ping timeout event.
unc0rr
parents: 7351
diff changeset
   162
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   163
    s <- get
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   164
    put $! s{removedClients = ci `Set.insert` removedClients s}
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   165
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   166
processAction (DeleteClient ci) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   167
    io $ debugM "Clients"  $ "DeleteClient: " ++ show ci
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   168
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   169
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   170
    io $ removeClient rnc ci
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   171
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   172
    s <- get
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   173
    put $! s{removedClients = ci `Set.delete` removedClients s}
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   174
5209
f7a610e2ef5f On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents: 5184
diff changeset
   175
    sp <- gets (shutdownPending . serverInfo)
f7a610e2ef5f On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents: 5184
diff changeset
   176
    cls <- allClientsS
f7a610e2ef5f On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents: 5184
diff changeset
   177
    io $ when (sp && null cls) $ throwIO ShutdownException
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   178
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   179
processAction (ModifyClient f) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   180
    (Just ci) <- gets clientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   181
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   182
    io $ modifyClient rnc f ci
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   183
    return ()
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   185
processAction (ModifyClient2 ci f) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   186
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   187
    io $ modifyClient rnc f ci
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   188
    return ()
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   189
7757
c20e6c80e249 Don't accept ROUNDFINISHED message twice. Fixes game hangs when half of teams quit game.
unc0rr
parents: 7748
diff changeset
   190
processAction (ModifyRoomClients f) = do
c20e6c80e249 Don't accept ROUNDFINISHED message twice. Fixes game hangs when half of teams quit game.
unc0rr
parents: 7748
diff changeset
   191
    rnc <- gets roomsClients
c20e6c80e249 Don't accept ROUNDFINISHED message twice. Fixes game hangs when half of teams quit game.
unc0rr
parents: 7748
diff changeset
   192
    ri <- clientRoomA
c20e6c80e249 Don't accept ROUNDFINISHED message twice. Fixes game hangs when half of teams quit game.
unc0rr
parents: 7748
diff changeset
   193
    roomClIDs <- io $ roomClientsIndicesM rnc ri
c20e6c80e249 Don't accept ROUNDFINISHED message twice. Fixes game hangs when half of teams quit game.
unc0rr
parents: 7748
diff changeset
   194
    io $ mapM_ (modifyClient rnc f) roomClIDs
c20e6c80e249 Don't accept ROUNDFINISHED message twice. Fixes game hangs when half of teams quit game.
unc0rr
parents: 7748
diff changeset
   195
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   196
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   197
processAction (ModifyRoom f) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   198
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   199
    ri <- clientRoomA
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   200
    io $ modifyRoom rnc f ri
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   201
    return ()
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   202
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   203
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   204
processAction (ModifyServerInfo f) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   205
    modify (\s -> s{serverInfo = f $ serverInfo s})
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   206
    si <- gets serverInfo
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   207
    io $ writeServerConfig si
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   208
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   209
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   210
processAction (MoveToRoom ri) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   211
    (Just ci) <- gets clientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   212
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   213
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   214
    io $ do
7898
ba735701ce7f Clear in-game flag when user joins room
unc0rr
parents: 7775
diff changeset
   215
        modifyClient rnc (\cl -> cl{teamsInGame = 0, isReady = False, isMaster = False, isInGame = False}) ci
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   216
        modifyRoom rnc (\r -> r{playersIn = playersIn r + 1}) ri
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   217
        moveClientToRoom rnc ri ci
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   218
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   219
    chans <- liftM (map sendChan) $ roomClientsS ri
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   220
    clNick <- client's nick
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   221
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   222
    processAction $ AnswerClients chans ["JOINED", clNick]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   223
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   224
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   225
processAction (MoveToLobby msg) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   226
    (Just ci) <- gets clientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   227
    ri <- clientRoomA
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   228
    rnc <- gets roomsClients
7766
98edc0724a28 Fix most of server warnings
unc0rr
parents: 7757
diff changeset
   229
    playersNum <- io $ room'sM rnc playersIn ri
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   230
    master <- client's isMaster
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   231
--    client <- client's id
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   232
    clNick <- client's nick
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   233
    chans <- othersChans
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   234
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   235
    if master then
7521
093ea41051c5 Keep room till last player quits
unc0rr
parents: 7498
diff changeset
   236
        if playersNum > 1 then
7351
34efdd1f230f - Check ready status only after deleting player's teams (should fix the bug when you're unable to start game)
unc0rr
parents: 7321
diff changeset
   237
            mapM_ processAction [ChangeMaster, NoticeMessage AdminLeft, RemoveClientTeams ci, AnswerClients chans ["LEFT", clNick, msg]]
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   238
            else
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   239
            processAction RemoveRoom
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   240
        else
7351
34efdd1f230f - Check ready status only after deleting player's teams (should fix the bug when you're unable to start game)
unc0rr
parents: 7321
diff changeset
   241
        mapM_ processAction [RemoveClientTeams ci, AnswerClients chans ["LEFT", clNick, msg]]
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   242
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   243
    -- when not removing room
7351
34efdd1f230f - Check ready status only after deleting player's teams (should fix the bug when you're unable to start game)
unc0rr
parents: 7321
diff changeset
   244
    ready <- client's isReady
7521
093ea41051c5 Keep room till last player quits
unc0rr
parents: 7498
diff changeset
   245
    when (not master || playersNum > 1) . io $ do
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   246
        modifyRoom rnc (\r -> r{
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   247
                playersIn = playersIn r - 1,
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   248
                readyPlayers = if ready then readyPlayers r - 1 else readyPlayers r
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   249
                }) ri
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   250
        moveClientToLobby rnc ci
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   251
7710
fd5bcbd698a5 - Keep track of room name so correct name is displayed when you become room admin
unc0rr
parents: 7682
diff changeset
   252
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   253
processAction ChangeMaster = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   254
    (Just ci) <- gets clientIndex
7710
fd5bcbd698a5 - Keep track of room name so correct name is displayed when you become room admin
unc0rr
parents: 7682
diff changeset
   255
    proto <- client's clientProto
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   256
    ri <- clientRoomA
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   257
    rnc <- gets roomsClients
7926
550083f61a0e oops, fix incorrect room owner name in ROOM UPD command again
unc0rr
parents: 7924
diff changeset
   258
    newMasterId <- liftM (last . filter (/= ci)) . io $ roomClientsIndicesM rnc ri
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   259
    newMaster <- io $ client'sM rnc id newMasterId
6733
5abbc345a82f Handle ROOM* commands in rooms list model
unc0rr
parents: 6541
diff changeset
   260
    oldRoomName <- io $ room'sM rnc name ri
7682
f6bfbe829008 'h' status for room admins
unc0rr
parents: 7668
diff changeset
   261
    oldMaster <- client's nick
7668
4cb423f42105 Show who is the room admin on join (no tested, also I don't like how it is done via server warnings, but it seems there's no other solution compatible with .17)
unc0rr
parents: 7664
diff changeset
   262
    thisRoomChans <- liftM (map sendChan) $ roomClientsS ri
7710
fd5bcbd698a5 - Keep track of room name so correct name is displayed when you become room admin
unc0rr
parents: 7682
diff changeset
   263
    let newRoomName = if proto < 42 then nick newMaster else oldRoomName
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   264
    mapM_ processAction [
7775
835ad028fb66 Keep room admin ready status always set
unc0rr
parents: 7766
diff changeset
   265
        ModifyRoom (\r -> r{masterID = newMasterId
835ad028fb66 Keep room admin ready status always set
unc0rr
parents: 7766
diff changeset
   266
                , name = newRoomName
835ad028fb66 Keep room admin ready status always set
unc0rr
parents: 7766
diff changeset
   267
                , isRestrictedJoins = False
835ad028fb66 Keep room admin ready status always set
unc0rr
parents: 7766
diff changeset
   268
                , isRestrictedTeams = False
835ad028fb66 Keep room admin ready status always set
unc0rr
parents: 7766
diff changeset
   269
                , readyPlayers = if isReady newMaster then readyPlayers r else readyPlayers r + 1})
835ad028fb66 Keep room admin ready status always set
unc0rr
parents: 7766
diff changeset
   270
        , ModifyClient2 newMasterId (\c -> c{isMaster = True, isReady = True})
7682
f6bfbe829008 'h' status for room admins
unc0rr
parents: 7668
diff changeset
   271
        , AnswerClients [sendChan newMaster] ["ROOM_CONTROL_ACCESS", "1"]
f6bfbe829008 'h' status for room admins
unc0rr
parents: 7668
diff changeset
   272
        , AnswerClients thisRoomChans ["WARNING", "New room admin is " `B.append` nick newMaster]
f6bfbe829008 'h' status for room admins
unc0rr
parents: 7668
diff changeset
   273
        , AnswerClients thisRoomChans ["CLIENT_FLAGS", "-h", oldMaster]
7775
835ad028fb66 Keep room admin ready status always set
unc0rr
parents: 7766
diff changeset
   274
        , AnswerClients thisRoomChans ["CLIENT_FLAGS", "+hr", nick newMaster]
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   275
        ]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   276
7766
98edc0724a28 Fix most of server warnings
unc0rr
parents: 7757
diff changeset
   277
    newRoom' <- io $ room'sM rnc id ri
6541
08ed346ed341 Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents: 6191
diff changeset
   278
    chans <- liftM (map sendChan) $! sameProtoClientsS proto
7766
98edc0724a28 Fix most of server warnings
unc0rr
parents: 7757
diff changeset
   279
    processAction $ AnswerClients chans ("ROOM" : "UPD" : oldRoomName : roomInfo newRoomName newRoom')
6541
08ed346ed341 Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents: 6191
diff changeset
   280
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   281
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   282
processAction (AddRoom roomName roomPassword) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   283
    Just clId <- gets clientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   284
    rnc <- gets roomsClients
6541
08ed346ed341 Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents: 6191
diff changeset
   285
    proto <- client's clientProto
08ed346ed341 Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents: 6191
diff changeset
   286
    n <- client's nick
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   287
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   288
    let rm = newRoom{
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   289
            masterID = clId,
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   290
            name = roomName,
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   291
            password = roomPassword,
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   292
            roomProto = proto
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   293
            }
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   294
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   295
    rId <- io $ addRoom rnc rm
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   296
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   297
    processAction $ MoveToRoom rId
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   298
6541
08ed346ed341 Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents: 6191
diff changeset
   299
    chans <- liftM (map sendChan) $! sameProtoClientsS proto
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   300
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   301
    mapM_ processAction [
7945
4006d77e1a28 Send notification about 1 player in room on room creation
unc0rr
parents: 7926
diff changeset
   302
      AnswerClients chans ("ROOM" : "ADD" : roomInfo n rm{playersIn = 1})
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   303
        ]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   304
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   305
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   306
processAction RemoveRoom = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   307
    Just clId <- gets clientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   308
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   309
    ri <- io $ clientRoomM rnc clId
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   310
    roomName <- io $ room'sM rnc name ri
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   311
    others <- othersChans
6541
08ed346ed341 Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents: 6191
diff changeset
   312
    proto <- client's clientProto
08ed346ed341 Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents: 6191
diff changeset
   313
    chans <- liftM (map sendChan) $! sameProtoClientsS proto
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   314
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   315
    mapM_ processAction [
6541
08ed346ed341 Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents: 6191
diff changeset
   316
            AnswerClients chans ["ROOM", "DEL", roomName],
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   317
            AnswerClients others ["ROOMABANDONED", roomName]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   318
        ]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   319
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   320
    io $ removeRoom rnc ri
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   321
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   322
7921
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
   323
processAction SendUpdateOnThisRoom = do
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
   324
    Just clId <- gets clientIndex
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
   325
    proto <- client's clientProto
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
   326
    rnc <- gets roomsClients
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
   327
    ri <- io $ clientRoomM rnc clId
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
   328
    rm <- io $ room'sM rnc id ri
7926
550083f61a0e oops, fix incorrect room owner name in ROOM UPD command again
unc0rr
parents: 7924
diff changeset
   329
    n <- io $ client'sM rnc nick (masterID rm)
7921
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
   330
    chans <- liftM (map sendChan) $! sameProtoClientsS proto
7924
351f970c60e1 oops, fix incorrect room owner name in ROOM UPD command
unc0rr
parents: 7921
diff changeset
   331
    processAction $ AnswerClients chans ("ROOM" : "UPD" : name rm : roomInfo n rm)
7921
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
   332
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
   333
6758
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   334
processAction UnreadyRoomClients = do
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   335
    ri <- clientRoomA
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   336
    roomPlayers <- roomClientsS ri
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   337
    pr <- client's clientProto
7757
c20e6c80e249 Don't accept ROUNDFINISHED message twice. Fixes game hangs when half of teams quit game.
unc0rr
parents: 7748
diff changeset
   338
    mapM_ processAction [
7775
835ad028fb66 Keep room admin ready status always set
unc0rr
parents: 7766
diff changeset
   339
        AnswerClients (map sendChan roomPlayers) $ notReadyMessage pr . map nick . filter (not . isMaster) $ roomPlayers
835ad028fb66 Keep room admin ready status always set
unc0rr
parents: 7766
diff changeset
   340
        , ModifyRoomClients (\cl -> cl{isReady = isMaster cl})
835ad028fb66 Keep room admin ready status always set
unc0rr
parents: 7766
diff changeset
   341
        , ModifyRoom (\r -> r{readyPlayers = 1})
7757
c20e6c80e249 Don't accept ROUNDFINISHED message twice. Fixes game hangs when half of teams quit game.
unc0rr
parents: 7748
diff changeset
   342
        ]
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   343
    where
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   344
        notReadyMessage p nicks = if p < 38 then "NOT_READY" : nicks else "CLIENT_FLAGS" : "-r" : nicks
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   345
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   346
6758
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   347
processAction FinishGame = do
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   348
    rnc <- gets roomsClients
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   349
    ri <- clientRoomA
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   350
    thisRoomChans <- liftM (map sendChan) $ roomClientsS ri
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   351
    answerRemovedTeams <- io $
7126
8daa5c8e84c0 Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents: 7124
diff changeset
   352
         room'sM rnc (map (\t -> AnswerClients thisRoomChans ["REMOVE_TEAM", t]) . leftTeams . fromJust . gameInfo) ri
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   353
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   354
    mapM_ processAction $
7124
cfee05712896 - Restore pre-r9257cf8e7af2 behavior
unc0rr
parents: 7120
diff changeset
   355
        SaveReplay
7126
8daa5c8e84c0 Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents: 7124
diff changeset
   356
        : ModifyRoom
6758
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   357
            (\r -> r{
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   358
                gameInfo = Nothing,
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   359
                readyPlayers = 0
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   360
                }
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   361
            )
7126
8daa5c8e84c0 Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents: 7124
diff changeset
   362
        : UnreadyRoomClients
7921
6b074de32bea Send ROOM UPD message when team is added/deleted from room, and when game starts or finishes
unc0rr
parents: 7898
diff changeset
   363
        : SendUpdateOnThisRoom
7126
8daa5c8e84c0 Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents: 7124
diff changeset
   364
        : answerRemovedTeams
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   365
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   366
6753
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   367
processAction (SendTeamRemovalMessage teamName) = do
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   368
    chans <- othersChans
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   369
    mapM_ processAction [
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   370
        AnswerClients chans ["EM", rmTeamMsg],
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   371
        ModifyRoom (\r -> r{
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   372
                gameInfo = liftM (\g -> g{
7124
cfee05712896 - Restore pre-r9257cf8e7af2 behavior
unc0rr
parents: 7120
diff changeset
   373
                    teamsInGameNumber = teamsInGameNumber g - 1
cfee05712896 - Restore pre-r9257cf8e7af2 behavior
unc0rr
parents: 7120
diff changeset
   374
                    , roundMsgs = roundMsgs g Seq.|> rmTeamMsg
6753
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   375
                }) $ gameInfo r
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   376
            })
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   377
        ]
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   378
6758
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   379
    rnc <- gets roomsClients
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   380
    ri <- clientRoomA
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   381
    gi <- io $ room'sM rnc gameInfo ri
26bf919aeb57 Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents: 6756
diff changeset
   382
    when (isJust gi && 0 == teamsInGameNumber (fromJust gi)) $
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   383
        processAction FinishGame
6753
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   384
    where
e95b1f62d0de Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents: 6733
diff changeset
   385
        rmTeamMsg = toEngineMsg $ 'F' `B.cons` teamName
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   386
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   387
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   388
processAction (RemoveTeam teamName) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   389
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   390
    ri <- clientRoomA
5996
2c72fe81dd37 Convert boolean variable + a bunch of fields which make sense only while game is going on into Maybe + structure
unc0rr
parents: 5426
diff changeset
   391
    inGame <- io $ room'sM rnc (isJust . gameInfo) ri
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   392
    chans <- othersChans
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   393
    mapM_ processAction $
7126
8daa5c8e84c0 Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents: 7124
diff changeset
   394
        ModifyRoom (\r -> r{
8daa5c8e84c0 Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents: 7124
diff changeset
   395
            teams = Prelude.filter (\t -> teamName /= teamname t) $ teams r
8daa5c8e84c0 Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents: 7124
diff changeset
   396
            , gameInfo = liftM (\g -> g{leftTeams = teamName : leftTeams g}) $ gameInfo r
8daa5c8e84c0 Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents: 7124
diff changeset
   397
            })
7947
0cf5277fef1a Better place for SendUpdateOnThisRoom
unc0rr
parents: 7945
diff changeset
   398
        : SendUpdateOnThisRoom
7124
cfee05712896 - Restore pre-r9257cf8e7af2 behavior
unc0rr
parents: 7120
diff changeset
   399
        : AnswerClients chans ["REMOVE_TEAM", teamName]
cfee05712896 - Restore pre-r9257cf8e7af2 behavior
unc0rr
parents: 7120
diff changeset
   400
        : [SendTeamRemovalMessage teamName | inGame]
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   401
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   402
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   403
processAction (RemoveClientTeams clId) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   404
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   405
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   406
    removeTeamActions <- io $ do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   407
        clNick <- client'sM rnc nick clId
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   408
        rId <- clientRoomM rnc clId
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   409
        roomTeams <- room'sM rnc teams rId
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   410
        return . Prelude.map (RemoveTeam . teamname) . Prelude.filter (\t -> teamowner t == clNick) $ roomTeams
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   411
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   412
    mapM_ processAction removeTeamActions
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   413
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   414
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   415
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   416
processAction CheckRegistered = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   417
    (Just ci) <- gets clientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   418
    n <- client's nick
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   419
    h <- client's host
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   420
    p <- client's clientProto
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   421
    uid <- client's clUID
6191
190a8e5d9956 Case-insensitive comparison of nicks
unc0rr
parents: 6068
diff changeset
   422
    haveSameNick <- liftM (not . null . tail . filter (\c -> caseInsensitiveCompare (nick c) n)) allClientsS
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   423
    if haveSameNick then
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   424
        if p < 38 then
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   425
            mapM_ processAction [ByeClient "Nickname is already in use", removeNick]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   426
            else
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   427
            mapM_ processAction [NoticeMessage NickAlreadyInUse, removeNick]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   428
        else
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   429
        do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   430
        db <- gets (dbQueries . serverInfo)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   431
        io $ writeChan db $ CheckAccount ci (hashUnique uid) n h
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   432
        return ()
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   433
   where
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   434
       removeNick = ModifyClient (\c -> c{nick = ""})
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   435
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   436
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   437
processAction ClearAccountsCache = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   438
    dbq <- gets (dbQueries . serverInfo)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   439
    io $ writeChan dbq ClearCache
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   440
    return ()
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   441
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   442
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   443
processAction (ProcessAccountInfo info) =
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   444
    case info of
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   445
        HasAccount passwd isAdmin -> do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   446
            chan <- client's sendChan
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   447
            mapM_ processAction [AnswerClients [chan] ["ASKPASSWORD"], ModifyClient (\c -> c{webPassword = passwd, isAdministrator = isAdmin})]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   448
        Guest ->
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   449
            processAction JoinLobby
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   450
        Admin -> do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   451
            mapM_ processAction [ModifyClient (\cl -> cl{isAdministrator = True}), JoinLobby]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   452
            chan <- client's sendChan
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   453
            processAction $ AnswerClients [chan] ["ADMIN_ACCESS"]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   454
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   455
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   456
processAction JoinLobby = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   457
    chan <- client's sendChan
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   458
    clientNick <- client's nick
7498
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   459
    isAuthenticated <- liftM (not . B.null) $ client's webPassword
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   460
    isAdmin <- client's isAdministrator
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   461
    loggedInClients <- liftM (Prelude.filter logonPassed) $! allClientsS
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   462
    let (lobbyNicks, clientsChans) = unzip . L.map (nick &&& sendChan) $ loggedInClients
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   463
    let authenticatedNicks = L.map nick . L.filter (not . B.null . webPassword) $ loggedInClients
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   464
    let adminsNicks = L.map nick . L.filter isAdministrator $ loggedInClients
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   465
    let clFlags = B.concat . L.concat $ [["u" | isAuthenticated], ["a" | isAdmin]]
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   466
    mapM_ processAction . concat $ [
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   467
        [AnswerClients clientsChans ["LOBBY:JOINED", clientNick]]
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   468
        , [AnswerClients [chan] ("LOBBY:JOINED" : clientNick : lobbyNicks)]
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   469
        , [AnswerClients [chan] ("CLIENT_FLAGS" : "+u" : authenticatedNicks) | not $ null authenticatedNicks]
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   470
        , [AnswerClients [chan] ("CLIENT_FLAGS" : "+a" : adminsNicks) | not $ null adminsNicks]
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   471
        , [AnswerClients (chan : clientsChans) ["CLIENT_FLAGS",  B.concat["+" , clFlags], clientNick] | not $ B.null clFlags]
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   472
        , [ModifyClient (\cl -> cl{logonPassed = True})]
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   473
        , [SendServerMessage]
86984f6fa1b9 Introduce 'a' and 'u' client flags to mark admins and authenticated users
unc0rr
parents: 7465
diff changeset
   474
        ]
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   475
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   476
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   477
processAction (KickClient kickId) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   478
    modify (\s -> s{clientIndex = Just kickId})
5214
d2ad737891b0 60 seconds cooldown ban on kick
unc0rr
parents: 5212
diff changeset
   479
    clHost <- client's host
d2ad737891b0 60 seconds cooldown ban on kick
unc0rr
parents: 5212
diff changeset
   480
    currentTime <- io getCurrentTime
d2ad737891b0 60 seconds cooldown ban on kick
unc0rr
parents: 5212
diff changeset
   481
    mapM_ processAction [
d2ad737891b0 60 seconds cooldown ban on kick
unc0rr
parents: 5212
diff changeset
   482
        AddIP2Bans clHost "60 seconds cooldown after kick" (addUTCTime 60 currentTime),
d2ad737891b0 60 seconds cooldown ban on kick
unc0rr
parents: 5212
diff changeset
   483
        ByeClient "Kicked"
d2ad737891b0 60 seconds cooldown ban on kick
unc0rr
parents: 5212
diff changeset
   484
        ]
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   485
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   486
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   487
processAction (BanClient seconds reason banId) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   488
    modify (\s -> s{clientIndex = Just banId})
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   489
    clHost <- client's host
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   490
    currentTime <- io getCurrentTime
5426
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   491
    let msg = B.concat ["Ban for ", B.pack . show $ seconds, " (", reason, ")"]
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   492
    mapM_ processAction [
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   493
        AddIP2Bans clHost msg (addUTCTime seconds currentTime)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   494
        , KickClient banId
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   495
        ]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   496
5426
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   497
processAction (BanIP ip seconds reason) = do
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   498
    currentTime <- io getCurrentTime
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   499
    let msg = B.concat ["Ban for ", B.pack . show $ seconds, " (", reason, ")"]
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   500
    processAction $
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   501
        AddIP2Bans ip msg (addUTCTime seconds currentTime)
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   502
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   503
processAction BanList = do
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   504
    ch <- client's sendChan
7766
98edc0724a28 Fix most of server warnings
unc0rr
parents: 7757
diff changeset
   505
    b <- gets (B.pack . unlines . map show . bans . serverInfo)
5426
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   506
    processAction $
7766
98edc0724a28 Fix most of server warnings
unc0rr
parents: 7757
diff changeset
   507
        AnswerClients [ch] ["BANLIST", b]
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   508
7748
f160fbc139b1 UNBAN implementation
unc0rr
parents: 7735
diff changeset
   509
processAction (Unban entry) = do
f160fbc139b1 UNBAN implementation
unc0rr
parents: 7735
diff changeset
   510
    processAction $ ModifyServerInfo (\s -> s{bans = filter f $ bans s})
f160fbc139b1 UNBAN implementation
unc0rr
parents: 7735
diff changeset
   511
    where
f160fbc139b1 UNBAN implementation
unc0rr
parents: 7735
diff changeset
   512
        f (BanByIP bip _ _) = bip == entry
f160fbc139b1 UNBAN implementation
unc0rr
parents: 7735
diff changeset
   513
        f (BanByNick bn _ _) = bn == entry
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   514
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   515
processAction (KickRoomClient kickId) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   516
    modify (\s -> s{clientIndex = Just kickId})
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   517
    ch <- client's sendChan
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   518
    mapM_ processAction [AnswerClients [ch] ["KICKED"], MoveToLobby "kicked"]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   519
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   520
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   521
processAction (AddClient cl) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   522
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   523
    si <- gets serverInfo
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   524
    newClId <- io $ do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   525
        ci <- addClient rnc cl
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   526
        _ <- Exception.mask (forkIO . clientRecvLoop (clientSocket cl) (coreChan si) (sendChan cl) ci)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   527
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   528
        infoM "Clients" (show ci ++ ": New client. Time: " ++ show (connectTime cl))
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   529
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   530
        return ci
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   531
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   532
    modify (\s -> s{clientIndex = Just newClId})
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   533
    mapM_ processAction
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   534
        [
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   535
            AnswerClients [sendChan cl] ["CONNECTED", "Hedgewars server http://www.hedgewars.org/", serverVersion]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   536
            , CheckBanned
6809
unc0rr
parents: 6805
diff changeset
   537
            , AddIP2Bans (host cl) "Reconnected too fast" (addUTCTime 10 $ connectTime cl)
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   538
        ]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   539
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   540
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   541
processAction (AddNick2Bans n reason expiring) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   542
    processAction $ ModifyServerInfo (\s -> s{bans = BanByNick n reason expiring : bans s})
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   543
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   544
processAction (AddIP2Bans ip reason expiring) = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   545
    (Just ci) <- gets clientIndex
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   546
    rc <- gets removedClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   547
    when (not $ ci `Set.member` rc)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   548
        $ processAction $ ModifyServerInfo (\s -> s{bans = BanByIP ip reason expiring : bans s})
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   549
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   550
processAction CheckBanned = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   551
    clTime <- client's connectTime
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   552
    clNick <- client's nick
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   553
    clHost <- client's host
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   554
    si <- gets serverInfo
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   555
    let validBans = filter (checkNotExpired clTime) $ bans si
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   556
    let ban = L.find (checkBan clHost clNick) $ validBans
5426
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   557
    mapM_ processAction $
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   558
        ModifyServerInfo (\s -> s{bans = validBans})
5426
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   559
        : [ByeClient (getBanReason $ fromJust ban) | isJust ban]
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   560
    where
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   561
        checkNotExpired testTime (BanByIP _ _ time) = testTime `diffUTCTime` time <= 0
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   562
        checkNotExpired testTime (BanByNick _ _ time) = testTime `diffUTCTime` time <= 0
5426
109e9b5761c2 Implement command for banning by ip and a command for bans list
unc0rr
parents: 5214
diff changeset
   563
        checkBan ip _ (BanByIP bip _ _) = bip `B.isPrefixOf` ip
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   564
        checkBan _ n (BanByNick bn _ _) = bn == n
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   565
        getBanReason (BanByIP _ msg _) = msg
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   566
        getBanReason (BanByNick _ msg _) = msg
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   567
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   568
processAction PingAll = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   569
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   570
    io (allClientsM rnc) >>= mapM_ (kickTimeouted rnc)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   571
    cis <- io $ allClientsM rnc
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   572
    chans <- io $ mapM (client'sM rnc sendChan) cis
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   573
    io $ mapM_ (modifyClient rnc (\cl -> cl{pingsQueue = pingsQueue cl + 1})) cis
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   574
    processAction $ AnswerClients chans ["PING"]
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   575
    where
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   576
        kickTimeouted rnc ci = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   577
            pq <- io $ client'sM rnc pingsQueue ci
7465
c2dcf97ca664 Okay, this is workaround over ping timeouts problem on the server. Could make server crash if recieve thread wakes up after second ping timeout event.
unc0rr
parents: 7351
diff changeset
   578
            when (pq > 0) $ do
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   579
                withStateT (\as -> as{clientIndex = Just ci}) $
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   580
                    processAction (ByeClient "Ping timeout")
7600
31a177d2856c Disable workaround, as it still makes server crash and hung clients are hidden from players anyway
unc0rr
parents: 7537
diff changeset
   581
--                when (pq > 1) $
31a177d2856c Disable workaround, as it still makes server crash and hung clients are hidden from players anyway
unc0rr
parents: 7537
diff changeset
   582
--                    processAction $ DeleteClient ci -- smth went wrong with client io threads, issue DeleteClient here
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   583
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   584
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   585
processAction StatsAction = do
5211
8ebf92014447 Don't send stats after spawning new server
unc0rr
parents: 5210
diff changeset
   586
    si <- gets serverInfo
8ebf92014447 Don't send stats after spawning new server
unc0rr
parents: 5210
diff changeset
   587
    when (not $ shutdownPending si) $ do
5212
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   588
        rnc <- gets roomsClients
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   589
        (roomsNum, clientsNum) <- io $ withRoomsAndClients rnc st
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   590
        io $ writeChan (dbQueries si) $ SendStats clientsNum (roomsNum - 1)
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   591
    where
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   592
          st irnc = (length $ allRooms irnc, length $ allClients irnc)
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   593
7321
57bd4f201401 - Try sending remove message in 'finally' as a last resort
unc0rr
parents: 7126
diff changeset
   594
processAction RestartServer = do
5212
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   595
    sp <- gets (shutdownPending . serverInfo)
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   596
    when (not sp) $ do
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   597
        sock <- gets (fromJust . serverSocket . serverInfo)
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   598
        args <- gets (runArgs . serverInfo)
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   599
        io $ do
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   600
            noticeM "Core" "Closing listening socket"
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   601
            sClose sock
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   602
            noticeM "Core" "Spawning new server"
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   603
            _ <- createProcess (proc "./hedgewars-server" args)
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   604
            return ()
eaffb02f0053 Don't perform RestartServer action when already did it once
unc0rr
parents: 5211
diff changeset
   605
        processAction $ ModifyServerInfo (\s -> s{shutdownPending = True})
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   606
6012
6bac93097da3 Store replays for further analysis
unc0rr
parents: 5996
diff changeset
   607
#if defined(OFFICIAL_SERVER)
5184
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   608
processAction SaveReplay = do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   609
    ri <- clientRoomA
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   610
    rnc <- gets roomsClients
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   611
    io $ do
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   612
        r <- room'sM rnc id ri
bf7bba60ed93 update iphone too
nemo
parents: 5143
diff changeset
   613
        saveReplay r
6012
6bac93097da3 Store replays for further analysis
unc0rr
parents: 5996
diff changeset
   614
#else
6bac93097da3 Store replays for further analysis
unc0rr
parents: 5996
diff changeset
   615
processAction SaveReplay = return ()
6bac93097da3 Store replays for further analysis
unc0rr
parents: 5996
diff changeset
   616
#endif