{-# LANGUAGE RecordWildCards, DeriveDataTypeable, RankNTypes, GeneralizedNewtypeDeriving #-}

module Network.IRC.Types where

import Control.Monad.Reader
import Control.Monad.State
import Data.Configurator.Types
import Data.Dynamic
import Data.Text (Text)
import System.IO
import System.Time

type Channel     = Text
type Nick        = Text
type HandlerName = Text

newtype Handler = Handler {
  runHandler :: forall m . (MonadIO m) => BotConfig -> Message -> m (Maybe Command)
}

data User = Self | User { userNick :: Nick, userServer :: Text }
            deriving (Show, Eq)

data Message =
  ChannelMsg { time   :: ClockTime, user   :: User, msg     :: Text }
  | PrivMsg    { time :: ClockTime, user   :: User, msg     :: Text }
  | Ping       { time :: ClockTime, msg    :: Text }
  | JoinMsg    { time :: ClockTime, user   :: User }
  | ModeMsg    { time :: ClockTime, user   :: User, target  :: Text , mode   :: Text, modeArgs :: [Text] }
  | NickMsg    { time :: ClockTime, user   :: User, nick    :: Text }
  | QuitMsg    { time :: ClockTime, user   :: User, msg     :: Text }
  | PartMsg    { time :: ClockTime, user   :: User, msg     :: Text }
  | KickMsg    { time :: ClockTime, user   :: User, msg     :: Text }
  | OtherMsg   { time :: ClockTime, source :: Text, command :: Text , target :: Text, msg      :: Text }
  deriving (Show, Eq)

data Command =
    Pong            { rmsg  :: Text }
  | ChannelMsgReply { rmsg  :: Text }
  | PrivMsgReply    { ruser :: User, rmsg :: Text }
  | NickCmd
  | UserCmd
  | JoinCmd
  deriving (Show, Eq)

data BotConfig = BotConfig { server     :: String
                           , port       :: Int
                           , channel    :: Text
                           , botNick    :: Text
                           , botTimeout :: Int
                           , handlers   :: [HandlerName]
                           , config     :: Config }

instance Show BotConfig where
  show BotConfig { .. } = "server = " ++ show server ++ "\n" ++
                          "port = " ++ show port ++ "\n" ++
                          "channel = " ++ show channel ++ "\n" ++
                          "nick = " ++ show botNick ++ "\n" ++
                          "timeout = " ++ show botTimeout ++ "\n" ++
                          "handlers = " ++ show handlers ++ "\n"

data Bot = Bot { botConfig :: BotConfig, socket :: Handle } deriving (Show)

data BotStatus = Connected | Disconnected | Joined | Kicked | Errored
                 deriving (Show, Eq)

newtype IRC a = IRC { _runIRC :: StateT BotStatus (ReaderT Bot IO) a }
                deriving (Functor, Monad, MonadIO, MonadReader Bot, MonadState BotStatus)

runIRC :: Bot -> BotStatus -> IRC a -> IO BotStatus
runIRC bot botStatus irc =
  fmap snd $ runReaderT (runStateT (_runIRC irc) Connected) bot