diff options
Diffstat (limited to 'Hsbot')
-rw-r--r-- | Hsbot/Command.hs | 14 | ||||
-rw-r--r-- | Hsbot/IRCPlugin.hs | 13 |
2 files changed, 19 insertions, 8 deletions
diff --git a/Hsbot/Command.hs b/Hsbot/Command.hs index ea372e4..711aa32 100644 --- a/Hsbot/Command.hs +++ b/Hsbot/Command.hs @@ -38,16 +38,16 @@ unregisterCommand cmd pluginName' = do -- | Dispatches an input message dispatchMessage :: BotMsg -> IrcBot () -dispatchMessage (InputMsg inputMsg) = do - plugins <- gets botPlugins - cmds <- gets botCommands - if isPluginCommand --TODO : how to get rid of this if? - then +dispatchMessage (InputMsg inputMsg) + | isPluginCommand = do + plugins <- gets botPlugins + cmds <- gets botCommands let key = tail $ head $ words getMsgContent pluginNames = fromMaybe [] $ M.lookup key cmds plugins' = fromMaybe [] $ mapM (flip M.lookup plugins) pluginNames - in mapM_ (sendRunCommand $ tail getMsgContent) plugins' - else + mapM_ (sendRunCommand $ tail getMsgContent) plugins' + | otherwise = do + plugins <- gets botPlugins mapM_ (sendToPlugin (InputMsg inputMsg)) (M.elems plugins) where isPluginCommand :: Bool diff --git a/Hsbot/IRCPlugin.hs b/Hsbot/IRCPlugin.hs index c32d24c..4707ce1 100644 --- a/Hsbot/IRCPlugin.hs +++ b/Hsbot/IRCPlugin.hs @@ -1,5 +1,6 @@ module Hsbot.IRCPlugin - ( readMsg + ( answerMsg + , readMsg , sendCommand , sendCommandWithRequest , sendRegisterCommand @@ -9,6 +10,7 @@ module Hsbot.IRCPlugin import Control.Concurrent.Chan import Control.Monad.State +import Data.Maybe(fromMaybe) import Hsbot.Types @@ -24,6 +26,15 @@ writeMsg botMsg = do serverChan <- gets instanceServerChan liftIO $ writeChan serverChan $ botMsg +answerMsg :: Maybe IrcMsg -> String -> IrcPlugin () +answerMsg request msg = do + let incoming = fromMaybe (IrcMsg Nothing "ARGH" []) request + chanOrigin = head $ parameters (incoming) + sender = takeWhile (/= '!') $ fromMaybe "ARGH" (prefix incoming) + case head chanOrigin of + '#' -> writeMsg $ OutputMsg $ IrcMsg Nothing "PRIVMSG" [chanOrigin, msg] + _ -> writeMsg $ OutputMsg $ IrcMsg Nothing "PRIVMSG" [sender, msg] + -- | Commands management sendCommand :: String -> String -> String -> IrcPlugin () sendCommand cmd to params = sendCommandWithRequest cmd to params Nothing |