diff options
author | Julien Dessaux | 2010-08-01 00:22:58 +0200 |
---|---|---|
committer | Julien Dessaux | 2010-08-01 00:22:58 +0200 |
commit | 6888950b606430f3779507bf8106df168a52d0f2 (patch) | |
tree | f99b81ef1ede28027a3aba588d10aca80d6129e9 /HsbotIrcBot/Hsbot/Irc | |
parent | Got rid of the pseudo dynamic starting stuff, wrote a decent CLI parameters h... (diff) | |
download | hsbot-6888950b606430f3779507bf8106df168a52d0f2.tar.gz hsbot-6888950b606430f3779507bf8106df168a52d0f2.tar.bz2 hsbot-6888950b606430f3779507bf8106df168a52d0f2.zip |
Added the cli parser and a config file handler for hsbot-irc.
Diffstat (limited to 'HsbotIrcBot/Hsbot/Irc')
-rw-r--r-- | HsbotIrcBot/Hsbot/Irc/Config.hs | 78 |
1 files changed, 74 insertions, 4 deletions
diff --git a/HsbotIrcBot/Hsbot/Irc/Config.hs b/HsbotIrcBot/Hsbot/Irc/Config.hs index 0d1e5a2..e0b8c64 100644 --- a/HsbotIrcBot/Hsbot/Irc/Config.hs +++ b/HsbotIrcBot/Hsbot/Irc/Config.hs @@ -1,14 +1,18 @@ module Hsbot.Irc.Config ( IrcConfig(..) , ircDefaultConfig + , getIrcConfig ) where +import qualified Data.ConfigFile as C +import Data.Either.Utils import Network +import System.Exit +import System.Posix.Files -- | Configuration data type data IrcConfig = IrcConfig - { ircConfigName :: String -- The configuration name - , ircConfigAddress :: String -- the server's address + { ircConfigAddress :: String -- the server's address , ircConfigPort :: PortID -- the server's port , ircConfigChannels :: [String] -- the Channels to join on start , ircConfigNickname :: String -- the hsbot's nickname @@ -21,8 +25,7 @@ data IrcConfig = IrcConfig -- | User configuration ircDefaultConfig :: IrcConfig ircDefaultConfig = IrcConfig - { ircConfigName = "irc-alocalhost" - , ircConfigAddress = "localhost" + { ircConfigAddress = "localhost" , ircConfigPort = PortNumber 6667 , ircConfigChannels = ["#hsbot"] , ircConfigNickname = "hsbot" @@ -32,3 +35,70 @@ ircDefaultConfig = IrcConfig , ircConfigPlugins = ["Ping"] } +-- | config file retrieving +getIrcConfig :: Maybe String -> IO (IrcConfig) +getIrcConfig maybePath = + case maybePath of + Just path -> do + doesFileExists <- fileExist path + case doesFileExists of + True -> do + fileStatus <- getFileStatus path + case isRegularFile $ fileStatus of + True -> compileIrcConfig ircDefaultConfig path + False -> do + putStrLn "Invalid configuration file path." + exitWith $ ExitFailure 1 + False -> do + putStrLn "The specified configuration file does not exists." + exitWith $ ExitFailure 1 + Nothing -> return ircDefaultConfig -- TODO : try defaults like $HOME/.hsbotrc, /etc/hsbotrc or /usr/local/etc/hsbotrc + +-- | config file parsing +compileIrcConfig :: IrcConfig -> String -> IO (IrcConfig) +compileIrcConfig ircConfig path = do + val <- C.readfile C.emptyCP path + let cp = forceEither val + let address = case C.get cp "IRC" "address" of + Right this -> this + Left _ -> ircConfigAddress ircConfig + let port = case C.get cp "IRC" "port" of + Right this -> PortNumber $ fromIntegral (read this :: Int) -- TODO error handling + Left _ -> ircConfigPort ircConfig + let channels = case C.get cp "IRC" "channels" of + Right this -> map (lstrip ' ') (split this ',') + Left _ -> ircConfigChannels ircConfig + let nickname = case C.get cp "IRC" "nickname" of + Right this -> this + Left _ -> ircConfigNickname ircConfig + let password = case C.get cp "IRC" "password" of + Right this -> this + Left _ -> ircConfigPassword ircConfig + let realname = case C.get cp "IRC" "realname" of + Right this -> this + Left _ -> ircConfigRealname ircConfig + let commandPrefix = case C.get cp "IRC" "commandPrefix" of + Right this -> head this -- TODO error handling + Left _ -> ircConfigCommandPrefix ircConfig + let plugins = case C.get cp "IRC" "plugins" of + Right this -> map (lstrip ' ') (split this ',') + Left _ -> ircConfigPlugins ircConfig + return ircConfig { ircConfigAddress = address + , ircConfigPort = port + , ircConfigChannels = channels + , ircConfigNickname = nickname + , ircConfigPassword = password + , ircConfigRealname = realname + , ircConfigCommandPrefix = commandPrefix + , ircConfigPlugins = plugins } + where + split :: String -> Char -> [String] + split [] _ = [""] + split (c:cs) delim + | c == delim = "" : rest + | otherwise = (c : head rest) : tail rest + where rest = split cs delim + lstrip :: Char -> String -> String + lstrip x (c:cs) = if (x == c) then (lstrip x cs) else c:(lstrip x cs) + lstrip _ [] = [] + |