diff options
Diffstat (limited to 'hirc.hs')
| -rw-r--r-- | hirc.hs | 31 |
1 files changed, 20 insertions, 11 deletions
@@ -1,3 +1,4 @@ +{-# LANGUAGE LambdaCase #-} import Data.List import Data.Monoid import Network @@ -15,6 +16,8 @@ server = "irc.freenode.org" port = 6667 chan = "#hirc-testing" nick = "hirc" + +filename = server <> (':' : show port) -- The 'Net' monad, a wrapper over IO, carrying the bot's immutable state. type Net = ReaderT Bot IO @@ -45,7 +48,6 @@ run :: Net () run = do write "NICK" nick write "USER" (nick++" 0 * :hirc bot") - write "JOIN" chan asks socket >>= listen -- Process each line from the server @@ -53,20 +55,27 @@ listen :: Handle -> Net () listen h = forever $ do s <- init `fmap` io (hGetLine h) io (putStrLn s) - io (putStrLn $ show $ parse P.message filename s) - if ping s then pong s else eval (clean s) + case parse P.message filename s of + Right m -> eval m + x -> io $ putStrLn $ show x where forever a = a >> forever a - clean = drop 1 . dropWhile (/= ':') . drop 1 - ping x = "PING :" `isPrefixOf` x - pong x = write "PONG" (':' : drop 6 x) - filename = server <> (':' : show port) -- Dispatch a command -eval :: String -> Net () -eval "!quit" = write "QUIT" ":Exiting" >> io (exitWith ExitSuccess) -eval x | "!id " `isPrefixOf` x = privmsg (drop 4 x) -eval _ = return () -- ignore everything else +eval :: Message -> Net () +eval = \case + Message _ "PING" [x] -> + write "PONG" (':' : x) + Message _ "376" _ -> -- End of /MOTD command. + write "JOIN" chan + Message _ "PRIVMSG" [chan, "!quit"] -> do + write "QUIT" ":Exiting" + io (exitWith ExitSuccess) + Message _ "PRIVMSG" [chan, x] | "!id " `isPrefixOf` x -> do + privmsg (drop 4 x) + m -> do + io (putStrLn $ show m) + return () -- ignore everything else -- Send a privmsg to the current chan + server privmsg :: String -> Net () |