From d4103e3e18a36523735af38eace98a64e1068e6e Mon Sep 17 00:00:00 2001
From: Julien Dessaux
Date: Sun, 23 Aug 2009 23:07:43 +0200
Subject: Rebooting now works great, thanks to the communication channel
 preservation.

---
 Hsbot/Main.hs | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

(limited to 'Hsbot/Main.hs')

diff --git a/Hsbot/Main.hs b/Hsbot/Main.hs
index 4c006b5..8972022 100644
--- a/Hsbot/Main.hs
+++ b/Hsbot/Main.hs
@@ -14,28 +14,29 @@ import qualified Config as C
 import Hsbot.Core
 import Hsbot.IRC
 
-type Reboot = (Module -> Bot -> IO ())
+type Reboot = (Module -> Bot -> (Chan IrcLine) -> IO ())
 
 -- | Bot's first main entry point
 imain :: Module -> Reboot -> IO ()
-imain modul' reboot = imain' modul' reboot newbot
+imain modul' reboot = do
+    chan <- newChan :: IO (Chan IrcLine)
+    imain' modul' reboot newbot chan
 
 -- | Bot's main entry point
-imain' :: Module -> Reboot -> Bot -> IO ()
-imain' modul' reboot bot = do
+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
-    putStrLn "Connecting servers..."
     let newServers = filter (not . isConnected bot) (ircServers C.config)
+    putStrLn $ "Connecting servers : " ++ show (map address newServers)
     newServers' <- mapM connectServer newServers
-    putStrLn "Joining channels..."
+    putStrLn $ "Joining channels : " ++ show (map channels newServers) 
     mapM_ initServer newServers'
     putStrLn "Spawning threads..."
     let bot'  = saveServersStates newServers' bot
         Bot x = bot'
-    chan <- newChan :: IO (Chan IrcLine)
-    mapM_ (forkIO . listener chan) (M.toList x)
+    mapM_ (forkIO . listener chan) newServers' -- (M.toList x)
     bot'' <- monitor chan bot'
-    reboot modul' bot''
+    reboot modul' bot'' chan
 
 -- | Bot main loop, monitors the threads states and handle reboot
 monitor :: (Chan IrcLine) -> Bot -> IO Bot
-- 
cgit v1.2.3