adyxax/hsbot
Archived
1
0
Fork 0
Code Issues Activity

Allow to change ducks' spawning frequency.

Browse source
This commit is contained in:
Julien Dessaux 2011-05-15 13:43:50 +02:00
parent 1995f4ddb8
commit 352887b330
2 changed files with 16 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
Hsbot/Plugin/Duck.hs
View file

@ -15,7 +15,6 @@ import Data.Maybe
import Data.SafeCopy import Data.SafeCopy
import Data.Typeable import Data.Typeable
import qualified Network.IRC as IRC import qualified Network.IRC as IRC
import Prelude hiding (catch)
import System.Environment.XDG.BaseDir import System.Environment.XDG.BaseDir
import System.Random import System.Random
@ -77,11 +76,12 @@ theDuck channel seconds = do
baseDir <- liftIO $ System.Environment.XDG.BaseDir.getUserDataDir "hsbot" baseDir <- liftIO $ System.Environment.XDG.BaseDir.getUserDataDir "hsbot"
statDB <- liftIO $ openAcidStateFrom (baseDir ++ "/duckDB/") emptyStatDB statDB <- liftIO $ openAcidStateFrom (baseDir ++ "/duckDB/") emptyStatDB
ducksMVar <- liftIO newEmptyMVar ducksMVar <- liftIO newEmptyMVar
timeMVar <- liftIO $ newMVar seconds
duckSpawner channel seconds ducksMVar duckSpawner channel seconds ducksMVar
forever $ readMsg >>= eval statDB ducksMVar forever $ readMsg >>= eval statDB ducksMVar timeMVar
where where
eval :: AcidState StatDB -> MVar Int -> Message -> Plugin (Env IO) () eval :: AcidState StatDB -> MVar Int -> MVar Int -> Message -> Plugin (Env IO) ()
eval statDB ducksMVar (IncomingMsg msg) eval statDB ducksMVar timeMVar (IncomingMsg msg)
| IRC.msg_command msg == "PRIVMSG" = do | IRC.msg_command msg == "PRIVMSG" = do
-- First we kill the ducks that we find in the message -- First we kill the ducks that we find in the message
let kills = howManyDucksInThere . concat $ IRC.msg_params msg let kills = howManyDucksInThere . concat $ IRC.msg_params msg
@ -96,17 +96,24 @@ theDuck channel seconds = do
_ <- update' statDB (ScoreAction (getSender msg) 0 0 (min ducksWaitingForDeath shots)) _ <- update' statDB (ScoreAction (getSender msg) 0 0 (min ducksWaitingForDeath shots))
when (shots >= ducksWaitingForDeath) $ do when (shots >= ducksWaitingForDeath) $ do
_ <- liftIO $ takeMVar ducksMVar _ <- liftIO $ takeMVar ducksMVar
duckSpawner channel seconds ducksMVar time <- liftIO $ readMVar timeMVar
duckSpawner channel time ducksMVar
_ <- update' statDB (ScoreAction (getSender msg) 1 0 0) _ <- update' statDB (ScoreAction (getSender msg) 1 0 0)
return () return ()
-- Finally we check if we received some command -- Finally we check if we received some command
cmdArgs <- lift $ getCommand msg cmdArgs <- lift $ getCommand msg
case cmdArgs of case cmdArgs of
"ducks":"stats":_ -> query' statDB GetDuckStats >>= printDuckStats channel "duck":"freq":time:_ -> do
"ducks":_ -> answerMsg msg "Invalid duck command." case reads time :: [(Int, String)] of
(secs,_):_ -> liftIO $ modifyMVar_ timeMVar (\_ -> return secs)
_ -> answerMsg msg "Invalid time value."
"duck":"freq":_ -> answerMsg msg $ "You must provide an amount of seconds the bot should wait before spawning "
++ "new ducks after the end of a round."
"duck":"stat":_ -> query' statDB GetDuckStats >>= printDuckStats channel
"duck":_ -> answerMsg msg "Invalid duck command."
_ -> return () _ -> return ()
| otherwise = return () | otherwise = return ()
eval _ _ _ = return () eval _ _ _ _ = return ()
-- | Regularly spawns ducks on a channel, just waiting to be shot -- | Regularly spawns ducks on a channel, just waiting to be shot
duckSpawner :: String -> Int -> MVar Int -> Plugin (Env IO) () duckSpawner :: String -> Int -> MVar Int -> Plugin (Env IO) ()

2
hsbot.cabal
View file

@ -1,5 +1,5 @@
Name: hsbot Name: hsbot
Version: 0.4.15 Version: 0.4.16
Cabal-version: >=1.2 Cabal-version: >=1.2
Synopsis: A multipurposes IRC bot Synopsis: A multipurposes IRC bot
Description: Description: