adyxax/hsbot
Archived
1
0
Fork 0
Code Issues Activity

Added exception handling, an autorestart when that happens and output in case of restart/reload

Browse source
This commit is contained in:
Julien Dessaux 2011-08-08 20:56:20 +02:00
parent fe1acc3db5
commit 3b914c1b77
6 changed files with 61 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

15
Hsbot/Utils.hs
View file

@ -8,12 +8,14 @@ module Hsbot.Utils
) where
import Control.Concurrent
import Control.Exception (IOException, catch)
import Control.Monad.Reader
import Control.Monad.State
import qualified Data.ByteString.Lazy.UTF8 as L
import qualified Data.List as L
import qualified Network.IRC as IRC
import Network.TLS
import Prelude hiding (catch)
import System.IO
import Hsbot.Types
@ -47,9 +49,16 @@ hasAccess (Just mask) right = do
| otherwise = False
-- Helpers
sendStr :: Handle -> Maybe TLSCtx -> String -> IO ()
sendStr _ (Just ctx) msg = sendData ctx . L.fromString $ msg ++ "\r\n"
sendStr handle Nothing msg = hPutStrLn handle $ msg ++ "\r\n"
sendStr :: BotEnv -> Handle -> Maybe TLSCtx -> String -> IO ()
sendStr env _ (Just ctx) msg = sendData ctx (L.fromString $ msg ++ "\r\n") `catch` handleIOException env ("sendStr " ++ msg)
sendStr env handle Nothing msg = hPutStrLn handle (msg ++ "\r\n") `catch` handleIOException env ("sendStr " ++ msg)
handleIOException :: BotEnv -> String -> IOException -> IO ()
handleIOException env msg ioException = do
runReaderT (setGlobalQuitMVar $ BotRestart (show ioException, Just msg)) env
myId <- myThreadId
killThread myId
return ()
-- TLS utils
initTLSEnv :: TLSConfig -> IO TLSParams