{-

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2004-2015 Andrey Korotaev <unC0Rr@gmail.com>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; version 2 of the License

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 \-}

{-# LANGUAGE OverloadedStrings #-}

module HWProtoCore where

import Control.Monad.Reader

import Data.Maybe

import qualified Data.ByteString.Char8 as B

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

import CoreTypes

import HWProtoNEState

import HWProtoLobbyState

import HWProtoInRoomState

import HWProtoChecker

import HandlerUtils

import RoomsAndClients

import Utils

import Consts

handleCmd ["PING"] = answerClient ["PONG"]

handleCmd ("QUIT" : xs) = return [ByeClient msg]

    where

        -- "User quit: " is a special string parsed by frontend, do not localize.

        -- It denotes when the /quit command has been used with message parameter.

        -- "bye" is also a special string.

        msg = if not $ null xs then "User quit: " `B.append` (head xs) else "bye"

handleCmd ["PONG"] = do

    cl <- thisClient

    if pingsQueue cl == 0 then

        return [ProtocolError "Protocol violation"]

        else

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

handleCmd cmd = do

    (ci, irnc) <- ask

    let cl = irnc `client` ci

    if logonPassed cl then

        if isChecker cl then

            handleCmd_checker cmd

            else

            handleCmd_loggedin cmd

        else

        handleCmd_NotEntered cmd

unknownCmdWarningText :: B.ByteString

unknownCmdWarningText = loc "Unknown command or invalid parameters. Say '/help' in chat for a list of commands."

handleCmd_loggedin ["CMD"] = return [Warning unknownCmdWarningText]

handleCmd_loggedin ["CMD", parameters] = uncurry h $ extractParameters parameters

    where

        h "QUIT" m | not $ B.null m = handleCmd ["QUIT", m]

                   | otherwise = handleCmd ["QUIT"]

        h "RND" p = handleCmd ("RND" : B.words p)

        h "GLOBAL" p = serverAdminOnly $ do

            rnc <- liftM snd ask

            let chans = map (sendChan . client rnc) $ allClients rnc

            return [AnswerClients chans ["CHAT", nickGlobal, p]]

        h "WATCH" f = return [QueryReplay f]

        h "INFO" n | not $ B.null n = handleCmd ["INFO", n]

        h "HELP" _ = handleCmd ["HELP"]

        h "REGISTERED_ONLY" _ = serverAdminOnly $ do

            cl <- thisClient

            return

                [ModifyServerInfo(\s -> s{isRegisteredUsersOnly = not $ isRegisteredUsersOnly s})

                -- TODO: Say whether 'registered only' state is on or off

                , AnswerClients [sendChan cl] ["CHAT", nickServer, loc "'Registered only' state toggled."]

                ]

        h "SUPER_POWER" _ = serverAdminOnly $ do

            cl <- thisClient

            return

                [ModifyClient (\c -> c{hasSuperPower = True})

                , AnswerClients [sendChan cl] ["CHAT", nickServer, loc "Super power activated."]

                ]

        h _ _ = return [Warning unknownCmdWarningText]

        extractParameters p = let (a, b) = B.break (== ' ') p in (upperCase a, B.dropWhile (== ' ') b)

handleCmd_loggedin ["INFO", asknick] = do

    (_, rnc) <- ask

    maybeClientId <- clientByNick asknick

    isAdminAsking <- liftM isAdministrator thisClient

    let noSuchClient = isNothing maybeClientId

    let clientId = fromJust maybeClientId

    let cl = rnc `client` fromJust maybeClientId

    let roomId = clientRoom rnc clientId

    let clRoom = room rnc roomId

    let roomMasterSign = if isMaster cl then "+" else ""

    let adminSign = if isAdministrator cl then "@" else ""

    let rInfo = if roomId /= lobbyId then B.concat [adminSign, roomMasterSign, loc "room", " ", name clRoom] else adminSign `B.append` (loc "lobby")

    let roomStatus = if isJust $ gameInfo clRoom then

            if teamsInGame cl > 0 then (loc "(playing)") else (loc "(spectating)")

            else

            ""

    let hostStr = if isAdminAsking then host cl else B.empty

    if noSuchClient then

        answerClient [ "CHAT", nickServer, loc "Player is not online." ]

        else

        answerClient [

            "INFO",

            nick cl,

            B.concat ["[", hostStr, "]"],

            protoNumber2ver $ clientProto cl,

            B.concat ["[", rInfo, "]", roomStatus]

            ]

handleCmd_loggedin cmd = do

    (ci, rnc) <- ask

    if clientRoom rnc ci == lobbyId then

        handleCmd_lobby cmd

        else

        handleCmd_inRoom cmd