Add initial files for IRC bot.
Add basic IRC bot main file. Add simple module to parse HMTL title from an url.
This commit is contained in:
parent
abb28aa95b
commit
f9a2640104
33
HTMLParsing.hs
Normal file
33
HTMLParsing.hs
Normal file
|
@ -0,0 +1,33 @@
|
||||||
|
module HTMLParsing
|
||||||
|
( getHTTPTitle
|
||||||
|
) where
|
||||||
|
|
||||||
|
import Data.Tree.NTree.TypeDefs
|
||||||
|
import Data.Maybe
|
||||||
|
import Text.XML.HXT.Core
|
||||||
|
import Control.Monad
|
||||||
|
import Control.Monad.Trans
|
||||||
|
import Control.Monad.Maybe
|
||||||
|
import Network.HTTP
|
||||||
|
import Network.URI
|
||||||
|
|
||||||
|
-- http://adit.io/posts/2012-04-14-working_with_HTML_in_haskell.html
|
||||||
|
|
||||||
|
openUrl :: String -> MaybeT IO String
|
||||||
|
openUrl url = case parseURI url of
|
||||||
|
Nothing -> fail ""
|
||||||
|
Just u -> liftIO (getResponseBody =<< simpleHTTP (mkRequest GET u))
|
||||||
|
|
||||||
|
css :: ArrowXml a => String -> a XmlTree XmlTree
|
||||||
|
css tag = multi (hasName tag)
|
||||||
|
|
||||||
|
get :: String -> IO (IOSArrow XmlTree (NTree XNode))
|
||||||
|
get url = do
|
||||||
|
contents <- runMaybeT $ openUrl url
|
||||||
|
return $ readString [withParseHTML yes, withWarnings no] (fromMaybe "" contents)
|
||||||
|
|
||||||
|
getHTTPTitle :: String -> IO String
|
||||||
|
getHTTPTitle url = do
|
||||||
|
httpBody <- get url
|
||||||
|
title <- runX $ httpBody >>> css "title" /> getText
|
||||||
|
return $ head title
|
130
main.hs
Normal file
130
main.hs
Normal file
|
@ -0,0 +1,130 @@
|
||||||
|
--
|
||||||
|
-- A simple, clean IRC bot in Haskell
|
||||||
|
--
|
||||||
|
-- http://www.haskell.org/haskellwiki/Roll_your_own_IRC_bot
|
||||||
|
|
||||||
|
import Data.List
|
||||||
|
import Network
|
||||||
|
import Network.HTTP
|
||||||
|
import System.IO
|
||||||
|
import System.Time
|
||||||
|
import System.Exit
|
||||||
|
import Control.Monad.Reader
|
||||||
|
--import Control.Exception -- for base-3, with base-4 use Control.OldException
|
||||||
|
import Control.OldException
|
||||||
|
import Text.Printf
|
||||||
|
import Prelude hiding (catch)
|
||||||
|
|
||||||
|
import HTMLParsing
|
||||||
|
|
||||||
|
server = "irc.freenode.net"
|
||||||
|
port = 6667
|
||||||
|
chan = "#cheeseburger"
|
||||||
|
nick = "cheesebot"
|
||||||
|
|
||||||
|
--
|
||||||
|
-- The 'Net' monad, a wrapper over IO, carrying the bot's immutable state.
|
||||||
|
-- A socket and the bot's start time.
|
||||||
|
--
|
||||||
|
type Net = ReaderT Bot IO
|
||||||
|
data Bot = Bot { socket :: Handle, starttime :: ClockTime }
|
||||||
|
|
||||||
|
--
|
||||||
|
-- Set up actions to run on start and end, and run the main loop
|
||||||
|
--
|
||||||
|
main :: IO ()
|
||||||
|
main = bracket connect disconnect loop
|
||||||
|
where
|
||||||
|
disconnect = hClose . socket
|
||||||
|
loop st = catch (runReaderT run st) (const $ return ())
|
||||||
|
|
||||||
|
--
|
||||||
|
-- Connect to the server and return the initial bot state
|
||||||
|
--
|
||||||
|
connect :: IO Bot
|
||||||
|
connect = notify $ do
|
||||||
|
t <- getClockTime
|
||||||
|
h <- connectTo server (PortNumber (fromIntegral Main.port))
|
||||||
|
hSetBuffering h NoBuffering
|
||||||
|
return (Bot h t)
|
||||||
|
where
|
||||||
|
notify a = bracket_
|
||||||
|
(printf "Connecting to %s ... " server >> hFlush stdout)
|
||||||
|
(putStrLn "done.")
|
||||||
|
a
|
||||||
|
|
||||||
|
--
|
||||||
|
-- We're in the Net monad now, so we've connected successfully
|
||||||
|
-- Join a channel, and start processing commands
|
||||||
|
--
|
||||||
|
run :: Net ()
|
||||||
|
run = do
|
||||||
|
write "NICK" nick
|
||||||
|
write "USER" (nick++" 0 * :tutorial bot")
|
||||||
|
write "JOIN" chan
|
||||||
|
asks socket >>= listen
|
||||||
|
|
||||||
|
--
|
||||||
|
-- Process each line from the server
|
||||||
|
--
|
||||||
|
listen :: Handle -> Net ()
|
||||||
|
listen h = forever $ do
|
||||||
|
s <- init `fmap` io (hGetLine h)
|
||||||
|
io (putStrLn s)
|
||||||
|
if ping s then pong s else eval (clean s)
|
||||||
|
where
|
||||||
|
forever a = a >> forever a
|
||||||
|
clean = drop 1 . dropWhile (/= ':') . dropWhile (/= ' ')
|
||||||
|
ping x = "PING :" `isPrefixOf` x
|
||||||
|
pong x = write "PONG" (':' : drop 6 x)
|
||||||
|
|
||||||
|
--
|
||||||
|
-- Dispatch a command
|
||||||
|
--
|
||||||
|
eval :: String -> Net ()
|
||||||
|
eval x | "cheesebot" `isPrefixOf` x = privmsg (answerRequest (drop 9 x))
|
||||||
|
eval x | isYouTubeURL x = do
|
||||||
|
youtubeTitle <- io $ getHTTPTitle x
|
||||||
|
privmsg youtubeTitle
|
||||||
|
eval _ = return () -- ignore everything else
|
||||||
|
|
||||||
|
--
|
||||||
|
-- Send a privmsg to the current chan + server
|
||||||
|
--
|
||||||
|
privmsg :: String -> Net ()
|
||||||
|
privmsg s = write "PRIVMSG" (chan ++ " :" ++ s)
|
||||||
|
|
||||||
|
--
|
||||||
|
-- Send a message out to the server we're currently connected to
|
||||||
|
--
|
||||||
|
write :: String -> String -> Net ()
|
||||||
|
write s t = do
|
||||||
|
h <- asks socket
|
||||||
|
io $ hPrintf h "%s %s\r\n" s t
|
||||||
|
io $ printf "> %s %s\n" s t
|
||||||
|
|
||||||
|
--
|
||||||
|
-- Answer simple request
|
||||||
|
--
|
||||||
|
answerRequest :: String -> String
|
||||||
|
answerRequest x
|
||||||
|
| "/quit" `isInfixOf` x = "Nice try"
|
||||||
|
| ":" `isPrefixOf` x = drop 1 x
|
||||||
|
| "," `isPrefixOf` x = drop 1 x
|
||||||
|
| otherwise = x
|
||||||
|
|
||||||
|
--
|
||||||
|
-- Youtube
|
||||||
|
--
|
||||||
|
isYouTubeURL :: String -> Bool
|
||||||
|
isYouTubeURL s =
|
||||||
|
"http://www.youtube.com/" `isPrefixOf` s
|
||||||
|
|| "https://www.youtube.com/" `isPrefixOf` s
|
||||||
|
|| "http://youtu.be/" `isPrefixOf` s
|
||||||
|
|| "https://youtu.be/" `isPrefixOf` s
|
||||||
|
|
||||||
|
--
|
||||||
|
-- Convenience.
|
||||||
|
--
|
||||||
|
io :: IO a -> Net a
|
||||||
|
io = liftIO
|
Loading…
Reference in a new issue