Rationalized the way bot configs are handled.

Hsbot/Config.hs

@@ -1,18 +1,9 @@
 module Hsbot.Config
-    ( Config(..)
+    ( BotConfig (..)
-    , defaultConfig
     ) where
 
-import Hsbot.Irc.Config (IrcConfig)
+import Hsbot.Irc.Config
 
 -- | Configuration data type
-data Config = Config
+data BotConfig = IrcBotConfig IrcConfig
-    { ircConfigs :: [IrcConfig]
-    }
-
--- | User configuration
-defaultConfig :: Config
-defaultConfig = Config
-    { ircConfigs = []
-    }
 

Hsbot/Core.hs

@@ -17,14 +17,14 @@ import Hsbot.Plugin
 import Hsbot.Types
 
 -- | Bot's main entry point
-hsbot :: Config -> IO ()
+hsbot :: [BotConfig] -> IO ()
 hsbot config = do
     startTime <- getCurrentTime
     putStrLn "[Hsbot] Opening communication channel... "
     chan <- newChan :: IO (Chan BotMsg)
     mvar <- newMVar M.empty :: IO (MVar BotResumeData)
     putStrLn "[Hsbot] Spawning IrcBot plugins... "
-    botState <- execStateT spawnIrcPlugins BotState { botStartTime  = startTime
+    botState <- execStateT spawnPlugins BotState { botStartTime  = startTime
                                                  , botPlugins    = M.empty
                                                  , botChan       = chan
                                                  , botConfig     = config

Hsbot/Irc/Config.hs

@@ -21,7 +21,7 @@ data IrcConfig = IrcConfig
 -- | User configuration
 ircDefaultConfig :: IrcConfig
 ircDefaultConfig = IrcConfig
-    { ircConfigName          = "localhost"
+    { ircConfigName          = "irc-alocalhost"
     , ircConfigAddress       = "localhost"
     , ircConfigPort          = PortNumber 6667
     , ircConfigChannels      = ["#hsbot"]

Hsbot/Plugin.hs

@@ -1,7 +1,7 @@
 module Hsbot.Plugin
     ( killPlugin
-    , spawnIrcPlugins
+    , spawnPlugins
-    , spawnIrcPlugin
+    , spawnPlugin
     , unloadPlugin
     ) where
 
@@ -17,24 +17,26 @@ import Hsbot.Irc.Config
 import Hsbot.Irc.Core
 import Hsbot.Types
 
--- | spawns IrcPlugins
+-- | spawns plugins
-spawnIrcPlugins :: Bot ()
+spawnPlugins :: Bot ()
-spawnIrcPlugins = do
+spawnPlugins = do
     config <- gets botConfig
-    mapM_ (spawnIrcPlugin) (ircConfigs config)
+    mapM_ (spawnPlugin) config
 
--- | spawns a single irc plugin
+-- | spawns a single plugin
-spawnIrcPlugin :: IrcConfig -> Bot ()
+spawnPlugin :: BotConfig -> Bot ()
-spawnIrcPlugin config = do
+spawnPlugin (IrcBotConfig ircConfig) = do
     bot <- get
     let chan  = botChan bot
     pchan <- liftIO (newChan :: IO (Chan BotMsg))
-    threadId <- liftIO $ forkIO (startIrcbot config chan pchan)
+    threadId <- liftIO $ forkIO (startIrcbot ircConfig chan pchan)
-    let plugin  = PluginState { pluginName    = ircConfigName config
+    let plugin  = PluginState { pluginName    = ircConfigName ircConfig
                               , pluginChan    = pchan
                               , pluginHandles = M.empty }
         plugins = botPlugins bot
     put $ bot { botPlugins = M.insert (pluginName plugin) (plugin, threadId) plugins }
+    resumeData <- gets botResumeData
+    liftIO $ modifyMVar_ resumeData (\oldData -> return $ M.insert (ircConfigName ircConfig) M.empty oldData)
 
 -- | Unloads a plugin
 unloadPlugin :: String -> Bot ()

Hsbot/Types.hs

@@ -27,7 +27,7 @@ data BotState = BotState
     { botStartTime  :: UTCTime      -- the bot's uptime
     , botPlugins    :: M.Map String (PluginState, ThreadId) -- Loaded plugins
     , botChan       :: Chan BotMsg  -- the bot's communication channel
-    , botConfig     :: Config       -- the bot's starting config
+    , botConfig     :: [BotConfig]  -- the bot's starting config
     , botResumeData :: MVar BotResumeData   -- the necessary data to resume the bot's operations on reboot
     }