Evaluate parsed messages

1 files changed, 20 insertions, 11 deletions

diff --git a/hirc.hs b/hirc.hs
index 82b6cd2..46a91fb 100644
--- a/hirc.hs
+++ b/hirc.hs
@@ -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 ()