summaryrefslogtreecommitdiff
path: root/HsbotIrcBot/Hsbot/Irc
diff options
context:
space:
mode:
Diffstat (limited to 'HsbotIrcBot/Hsbot/Irc')
-rw-r--r--HsbotIrcBot/Hsbot/Irc/Config.hs78
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 _ [] = []
+