Archived
1
0
Fork 0

Added command registering and dispatching for plugins.

This commit is contained in:
Julien Dessaux 2010-02-04 20:40:16 +01:00
parent d2f40f6481
commit cf68de02be
6 changed files with 79 additions and 16 deletions

52
Hsbot/Command.hs Normal file
View file

@ -0,0 +1,52 @@
module Hsbot.Command
( dispatchCommand
, registerCommand
) where
import Control.Monad.State
import qualified Data.Map as M
import Data.Maybe
import Config
import Hsbot.Plugin
import Hsbot.Types
import Hsbot.Utils
-- TODO : unregister command
dispatchCommand :: BotMsg -> IrcBot ()
dispatchCommand (InputMsg inputMsg) = do
plugins <- gets botPlugins
cmds <- gets botCommands
if command inputMsg == "PRIVMSG"
then -- The first word matters as the command
let msg = (parameters inputMsg) !! 1
pfx = commandPrefix config
in if (head msg) == pfx
then
let key = tail msg -- all but the cmd prefix
pluginNames = fromMaybe [] $ M.lookup key cmds
plugins' = fromMaybe [] $ mapM (flip M.lookup plugins) pluginNames
in mapM_ (sendToPlugin (InputMsg inputMsg)) plugins'
else
mapM_ (sendToPlugin (InputMsg inputMsg)) (M.elems plugins)
else
return ()
dispatchCommand _ = return ()
registerCommand :: String -> String -> IrcBot ()
registerCommand cmd pluginName' = do
bot <- get
cmds <- gets botCommands
exists <- pluginExists pluginName'
if exists
then
let cmds' = if cmd `M.member` cmds
then cmds
else M.singleton cmd []
newCmds = M.adjust (++ [pluginName']) cmd cmds'
in put $ bot { botCommands = newCmds }
else
traceM $ inColor ("Couldn't register command \"" ++ cmd ++ "\" for plugin \""
++ pluginName' ++ "\" : plugin does not exists") [31]