summaryrefslogtreecommitdiff
path: root/Hsbot/Main.hs
blob: 9edd807ad1e4c86f2b5628f3d3be1d8379f42dad (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
module Hsbot.Main
    ( imain
    , imain'
    ) where

import Control.Concurrent
import Control.Concurrent.Chan
import Control.Monad
import System.IO
import System.Plugins

import qualified Config as C
import Hsbot.Core
import Hsbot.IRC

type Reboot = (Module -> Bot -> IO ())

-- | Bot's first main entry point
imain :: Module -> Reboot -> IO ()
imain modul' reboot = imain' modul' reboot newbot

-- | Bot's main entry point
imain' :: Module -> Reboot -> Bot -> IO ()
imain' modul' reboot bot = do
    putStrLn "yeah"
    putStrLn "Connecting servers..."
    servers' <- mapM connectServer (ircServers C.config)
    putStrLn "Joining channels..."
    mapM_ initServer servers'
    putStrLn "Spawning threads..."
    chan <- newChan :: IO (Chan IrcLine)
    mapM_ (forkIO . listener chan) servers'
    state <- monitor chan bot
    reboot modul' bot

-- | 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
                case input of
                    Reboot ->do
                        putStrLn "Got reboot message, rebooting"
                        return bot'
                    Str str -> putStrLn ("received : " ++ str) >> loop bot'

-- | Thread entry point for socket listeners
listener :: (Chan IrcLine) -> (IrcServer, Handle) -> IO ()
listener chan (server, handle) = forever $ do
    str <- hGetLine handle
    writeChan chan (Str str)
    if ping str then pong handle str
      else eval (parseIrcMsg str)
    where
        eval str
            | (Cmd user channel (cmd, args)) <- str = do
                let cmd' = tail cmd
                unless (null cmd') (parseCmds user cmd' args channel)
        parseCmds user cmd args channel
            | cmd == "reboot" = writeChan chan Reboot
            | otherwise       = do
                putStrLn $"Unknown command : " ++ cmd