Added the cli parser and a config file handler for hsbot-irc.

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 _ [] = []
+