diff options
Diffstat (limited to 'Hsbot/Irc/Plugin.hs')
| -rw-r--r-- | Hsbot/Irc/Plugin.hs | 32 |
1 files changed, 20 insertions, 12 deletions
diff --git a/Hsbot/Irc/Plugin.hs b/Hsbot/Irc/Plugin.hs index 3e6bef0..d972db2 100644 --- a/Hsbot/Irc/Plugin.hs +++ b/Hsbot/Irc/Plugin.hs @@ -5,7 +5,7 @@ module Hsbot.Irc.Plugin , loadIrcPlugin , sendToPlugin , spawnIrcPlugins - , unloadPlugin + , unloadIrcPlugin ) where import Control.Concurrent @@ -45,16 +45,20 @@ loadIrcPlugin pluginName = do "Ping" -> ircBotPluginPing "Quote" -> ircBotPluginQuote _ -> ircBotPluginDummy - let oldPlugins = ircBotPlugins ircbot + let oldPlugins = ircBotPlugins ircbot + oldResumeData = ircBotResumeData ircbot -- We check for unicity case M.lookup pluginName oldPlugins of Just _ -> return () Nothing -> do threadId <- liftIO $ forkIO (entryPoint pluginChan masterChan) - let plugin = IrcPluginState { ircPluginName = pluginName - , ircPluginChan = pluginChan - , ircPluginMasterChan = masterChan } - put $ ircbot { ircBotPlugins = M.insert pluginName (plugin, threadId) oldPlugins } + let plugin = IrcPluginState { ircPluginName = pluginName + , ircPluginChan = pluginChan + , ircPluginMasterChan = masterChan } + newPlugins = M.insert pluginName (plugin, threadId) oldPlugins + newResumeData = M.insert "PLUGINS" (show $ M.keys newPlugins) oldResumeData + put $ ircbot { ircBotPlugins = newPlugins + , ircBotResumeData = newResumeData } -- | Sends a list of loaded plugins listPlugins :: IrcMsg -> String -> IrcBot () @@ -66,14 +70,18 @@ listPlugins originalRequest dest = do Nothing -> return () -- | Unloads a plugin -unloadPlugin :: String -> IrcBot () -unloadPlugin name = do - bot <- get - let oldPlugins = ircBotPlugins bot +unloadIrcPlugin :: String -> IrcBot () +unloadIrcPlugin name = do + ircbot <- get + let oldPlugins = ircBotPlugins ircbot + oldResumeData = ircBotResumeData ircbot + -- We check if the plugin exists case M.lookup name oldPlugins of Just (_, threadId) -> do - let newPlugins = M.delete name oldPlugins + let newPlugins = M.delete name oldPlugins + newResumeData = M.insert "PLUGINS" (show $ M.keys newPlugins) oldResumeData liftIO $ throwTo threadId UserInterrupt - put $ bot { ircBotPlugins = newPlugins } + put $ ircbot { ircBotPlugins = newPlugins + , ircBotResumeData = newResumeData } Nothing -> return () |