1 files changed, 67 insertions, 0 deletions
diff --git a/src/Main.hs b/src/Main.hs
new file mode 100644
index 0000000..58ae99f
--- /dev/null
+++ b/src/Main.hs
@@ -0,0 +1,67 @@
+{-# LANGUAGE QuasiQuotes #-}
+
+module Main (
+  main
+) where
+
+import Data.ByteString.Char8 (pack)
+import Data.Maybe (fromJust)
+import Data.Version (showVersion)
+import Database.MySQL.Base (ConnectInfo(..))
+import Database.MySQL.Base.Types (Option(ReadDefaultFile, ReadDefaultGroup))
+import Paths_juandelacosa (version) -- from cabal
+import System.Environment (getArgs)
+import Text.RawString.QQ (r)
+import qualified System.Console.Docopt.NoTH as O
+
+import Server (server)
+
+usage :: String
+usage =  "juandelacosa " ++ showVersion version
+  ++ " manage MariaDB user and roles" ++ [r|
+
+Usage:
+  juandelacosa [options]
+
+Options:
+  -f, --file=MYCNF         Read this MySQL client config file
+  -g, --group=GROUP        Read this options group in the above file [default: client]
+
+  -s, --socket=SOCK        Listen on this UNIX-socket [default: /tmp/juandelacosa.sock]
+  -p, --port=PORT          Instead of UNIX-socket, listen on this TCP port (localhost)
+
+  -h, --help               Show this message
+
+|]
+
+main :: IO()
+main = do
+  doco <- O.parseUsageOrExit usage
+  args <- O.parseArgsOrExit doco =<< getArgs
+  if args `O.isPresent` O.longOption "help"
+  then putStrLn $ O.usage doco
+  else do
+    let
+      file = O.getArg args $ O.longOption "file"
+      group = fromJust $ O.getArg args $ O.longOption "group"
+      port = O.getArg args $ O.longOption "port"
+      socket = fromJust $ O.getArg args $ O.longOption "socket"
+    -- XXX: mysql package maps empty strings to NULL
+    -- which is what we need, see documentation for mysql_real_connect()
+    let myInfo = ConnectInfo {
+        connectDatabase = "",
+        connectHost     = "",
+        connectOptions  = case file of
+                          Nothing -> []
+                          Just f -> [ ReadDefaultFile f, ReadDefaultGroup (pack group) ],
+        connectPassword = "",
+        connectPath     = "",
+        connectPort     = 0,
+        connectSSL      = Nothing,
+        connectUser     = ""
+      }
+    let listen = case port of
+          Nothing -> Right socket
+          Just p -> Left $ read p
+    server listen myInfo
+