73 lines
2.7 KiB
Haskell
73 lines
2.7 KiB
Haskell
{-# 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
|