blob: 897202246b4a0f00bf2a73d9f80630b0072ada9c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
module Hsbot.Main
( imain
, imain'
) where
import Control.Concurrent
import Control.Concurrent.Chan
import Control.Monad
import qualified Data.Map as M
import System.IO
import System.Plugins
import qualified Config as C
import Hsbot.Core
import Hsbot.IRC
type Reboot = (Module -> Bot -> (Chan IrcLine) -> IO ())
-- | Bot's first main entry point
imain :: Module -> Reboot -> IO ()
imain modul' reboot = do
chan <- newChan :: IO (Chan IrcLine)
imain' modul' reboot newbot chan
-- | Bot's main entry point
imain' :: Module -> Reboot -> Bot -> (Chan IrcLine) -> IO ()
imain' modul' reboot bot chan = do
-- The chan passing to reboot (or another way to keep it) is still missing
let newServers = filter (not . isConnected bot) (ircServers C.config)
putStrLn $ "Connecting servers : " ++ show (map address newServers)
newServers' <- mapM connectServer newServers
putStrLn $ "Joining channels : " ++ show (map channels newServers)
mapM_ initServer newServers'
putStrLn "Spawning threads..."
let bot' = saveServersStates newServers' bot
Bot x = bot'
mapM_ (forkIO . listener chan) newServers' -- (M.toList x)
bot'' <- monitor chan bot'
reboot modul' bot'' chan
-- | Bot main loop, monitors the threads states and handle reboot
monitor :: (Chan IrcLine) -> Bot -> IO Bot
monitor chan bot = do
loop bot
where
loop bot' = do
input <- readChan chan :: IO IrcLine
case input of
Reboot -> do
putStrLn "Got reboot message, rebooting"
return bot'
_ -> loop bot'
-- | Thread entry point for socket listeners
listener :: (Chan IrcLine) -> (IrcServer, Handle) -> IO ()
listener chan (server, handle) = forever $ do
str <- hGetLine handle
let msg = parseIrcMsg str
writeChan chan msg
eval msg
where
eval :: IrcLine -> IO ()
eval (Privmsg (statement, stuff')) = sendPrivmsg (server, handle) stuff'
eval (Quit (ircServer, handle')) = return ()
eval (Join (ircServer, handle')) = return ()
eval (Part (ircServer, handle')) = return ()
eval (Ping (string)) = do pong handle string
eval stuff' = case stuff' of
Reboot -> return ()
Nil -> return ()
|
