{-# LANGUAGE OverloadedStrings, DeriveDataTypeable #-}

module CoreTypes where

import Control.Concurrent

import Data.Word

import qualified Data.Map as Map

import Data.Sequence(Seq, empty)

import Data.Time

import Network

import Data.Function

import Data.ByteString.Char8 as B

import Data.Unique

import Control.Exception

import Data.Typeable

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

import RoomsAndClients

type ClientChan = Chan [B.ByteString]

data ClientInfo =

    ClientInfo

    {

        clUID :: Unique,

        sendChan :: ClientChan,

        clientSocket :: Socket,

        host :: B.ByteString,

        connectTime :: UTCTime,

        nick :: B.ByteString,

        webPassword :: B.ByteString,

        logonPassed :: Bool,

        clientProto :: !Word16,

        roomID :: RoomIndex,

        pingsQueue :: !Word,

        isMaster :: Bool,

        isReady :: !Bool,

        isAdministrator :: Bool,

        clientClan :: Maybe B.ByteString,

        teamsInGame :: Word

    }

instance Show ClientInfo where

    show ci = " nick: " ++ unpack (nick ci) ++ " host: " ++ unpack (host ci)

instance Eq ClientInfo where

    (==) = (==) `on` clientSocket

data HedgehogInfo =

    HedgehogInfo B.ByteString B.ByteString

data TeamInfo =

    TeamInfo

    {

        teamownerId :: ClientIndex,

        teamowner :: B.ByteString,

        teamname :: B.ByteString,

        teamcolor :: B.ByteString,

        teamgrave :: B.ByteString,

        teamfort :: B.ByteString,

        teamvoicepack :: B.ByteString,

        teamflag :: B.ByteString,

        difficulty :: Int,

        hhnum :: Int,

        hedgehogs :: [HedgehogInfo]

    }

instance Show TeamInfo where

    show ti = "owner: " ++ unpack (teamowner ti)

            ++ "name: " ++ unpack (teamname ti)

            ++ "color: " ++ unpack (teamcolor ti)

data RoomInfo =

    RoomInfo

    {

        masterID :: ClientIndex,

        name :: B.ByteString,

        password :: B.ByteString,

        roomProto :: Word16,

        teams :: [TeamInfo],

        gameinprogress :: Bool,

        playersIn :: !Int,

        readyPlayers :: !Int,

        isRestrictedJoins :: Bool,

        isRestrictedTeams :: Bool,

        roundMsgs :: Seq B.ByteString,

        leftTeams :: [B.ByteString],

        teamsAtStart :: [TeamInfo],

        mapParams :: Map.Map B.ByteString B.ByteString,

        params :: Map.Map B.ByteString [B.ByteString]

    }

instance Show RoomInfo where

    show ri = ", players: " ++ show (playersIn ri)

            ++ ", ready: " ++ show (readyPlayers ri)

            ++ ", teams: " ++ show (teams ri)

newRoom :: RoomInfo

newRoom =

    RoomInfo

        ""

        ""

        0

        []

        False

        0

        0

        False

        False

        Data.Sequence.empty

        []

        []

        (

            Map.fromList $ Prelude.zipWith (,)

                ["MAP", "MAPGEN", "MAZE_SIZE", "SEED", "TEMPLATE"]

                ["+rnd+", "0", "0", "seed", "0"]

        )

        (Map.singleton "SCHEME" ["Default"])

data StatisticsInfo =

    StatisticsInfo

    {

        playersNumber :: Int,

        roomsNumber :: Int

    }

data ServerInfo c =

    ServerInfo

    {

        isDedicated :: Bool,

        serverMessage :: B.ByteString,

        serverMessageForOldVersions :: B.ByteString,

        latestReleaseVersion :: Word16,

        listenPort :: PortNumber,

        nextRoomID :: Int,

        dbHost :: B.ByteString,

        dbName :: B.ByteString,

        dbLogin :: B.ByteString,

        dbPassword :: B.ByteString,

        lastLogins :: [(B.ByteString, (UTCTime, B.ByteString))],

        restartPending :: Bool,

        coreChan :: Chan CoreMessage,

        dbQueries :: Chan DBQuery,

        serverConfig :: Maybe c

    }

instance Show (ServerInfo c) where

    show _ = "Server Info"

newServerInfo :: Chan CoreMessage -> Chan DBQuery -> Maybe c -> ServerInfo c

newServerInfo =

    ServerInfo

        True

        "<h2><p align=center><a href=\"http://www.hedgewars.org/\">http://www.hedgewars.org/</a></p></h2>"

        "<font color=yellow><h3 align=center>Hedgewars 0.9.14.1 is out! Please update.</h3><p align=center><a href=http://hedgewars.org/download.html>Download page here</a></font>"

        35

        46631

        0

        ""

        ""

        ""

        ""

        []

        False

data AccountInfo =

    HasAccount B.ByteString Bool

    | Guest

    | Admin

    deriving (Show, Read)

data DBQuery =

    CheckAccount ClientIndex Int B.ByteString B.ByteString

    | ClearCache

    | SendStats Int Int

    deriving (Show, Read)

data CoreMessage =

    Accept ClientInfo

    | ClientMessage (ClientIndex, [B.ByteString])

    | ClientAccountInfo ClientIndex Int AccountInfo

    | TimerAction Int

    | Remove ClientIndex

instance Show CoreMessage where

    show (Accept _) = "Accept"

    show (ClientMessage _) = "ClientMessage"

    show (ClientAccountInfo {}) = "ClientAccountInfo"

    show (TimerAction _) = "TimerAction"

    show (Remove _) = "Remove"

type MRnC = MRoomsAndClients RoomInfo ClientInfo

type IRnC = IRoomsAndClients RoomInfo ClientInfo

data Notice =

    NickAlreadyInUse

    | AdminLeft

    deriving Enum

data ShutdownException =

    ShutdownException

    | RestartException

     deriving (Show, Typeable)

instance Exception ShutdownException