61 lines
1.3 KiB
Haskell
61 lines
1.3 KiB
Haskell
module Hsbot.Types
|
|
( Bot
|
|
, BotState (..)
|
|
, BotStatus (..)
|
|
, BotEnv (..)
|
|
, Env
|
|
, Message (..)
|
|
, Plugin
|
|
, PluginId (..)
|
|
, PluginState (..)
|
|
) where
|
|
|
|
import Control.Concurrent
|
|
import qualified Data.Map as M
|
|
import Control.Monad.Reader
|
|
import Control.Monad.State
|
|
import Network.TLS
|
|
import System.IO
|
|
|
|
import Hsbot.Config
|
|
import Hsbot.Message
|
|
|
|
-- The bot environment
|
|
type Env = ReaderT BotEnv
|
|
|
|
data BotEnv = BotEnv
|
|
{ envHandle :: Handle
|
|
, envChan :: Chan Message
|
|
, envQuitMv :: MVar (BotStatus)
|
|
, envThreadIdsMv :: MVar [ThreadId]
|
|
, envConfig :: Config
|
|
, envTLS :: Maybe TLSParams
|
|
, envTLSCtx :: Maybe TLSCtx
|
|
}
|
|
|
|
-- The bot monad
|
|
type Bot = StateT BotState
|
|
|
|
data BotState = BotState
|
|
{ botPlugins :: M.Map String (PluginState, MVar (), ThreadId)
|
|
, botHooks :: [Chan Message]
|
|
, botChannels :: [String]
|
|
, botNickname :: String
|
|
}
|
|
|
|
-- The Plugin monad
|
|
type Plugin = StateT PluginState
|
|
|
|
data PluginState = PluginState
|
|
{ pluginId :: PluginId
|
|
, pluginChan :: Chan Message
|
|
, pluginMaster :: Chan Message
|
|
}
|
|
|
|
data PluginId = PluginId
|
|
{ pluginName :: String
|
|
, pluginEp :: PluginState -> IO ()
|
|
}
|
|
|
|
data BotStatus = BotContinue | BotExit | BotReload | BotRestart deriving (Show)
|
|
|