aboutsummaryrefslogtreecommitdiff
path: root/src/Main.hs
blob: e9a29b7e425837d7a8abf8f7ba297793489154d0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
{-# LANGUAGE QuasiQuotes #-}

module Main (
  main
) where

import Data.ByteString.Char8 (pack)
import Data.Either.Utils (forceEither)
import Data.Maybe (fromJust)
import Data.List (isPrefixOf)
import Data.Version (showVersion)
import Database.MySQL.Base (ConnectInfo(..), defaultSSLInfo)
import Database.MySQL.Base.Types (Option(ReadDefaultFile, ReadDefaultGroup))
import Paths_mywatch (getDataDir, version) -- from cabal
import System.Environment (getArgs)
import Text.InterpolatedString.Perl6 (qc)
import qualified Data.ConfigFile as Cf
import qualified System.Console.Docopt.NoTH as O

import Server (server)

usage :: IO String
usage = do
  dataDir <- getDataDir
  return $
    "mywatch " ++ showVersion version
    ++ " view queries on many MySQL servers" ++ [qc|

Usage:
  mywatch [options] MYCNF

Options:

  -d, --datadir=DIR        Data directory including static files [default: {dataDir}]

  -s, --socket=SOCK        Listen on this UNIX-socket [default: /tmp/mywatch.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 = fromJust $ O.getArg args $ O.argument "MYCNF"
      port = O.getArg args $ O.longOption "port"
      socket = fromJust $ O.getArg args $ O.longOption "socket"
      datadir = fromJust $ O.getArg args $ O.longOption "datadir"

    cf <- forceEither <$> Cf.readfile Cf.emptyCP file
    let
      servers = filter ("client" /=) . Cf.sections $ cf
      myInfo = map (\g -> ConnectInfo {
        connectDatabase = "",
        connectHost     = "",
        connectPassword = "",
        connectPath     = "",
        connectPort     = 0,
        -- FIXME: https://jira.mariadb.org/browse/MDEV-10246
        connectSSL      = if any (isPrefixOf "ssl") (forceEither $ Cf.options cf g)
                          then Just defaultSSLInfo else Nothing,
        connectUser     = "",
        -- FIXME: Work aroung buggy mysql: unsafeUseAsCString creates garbage.
        connectOptions  = [ ReadDefaultFile file, ReadDefaultGroup (pack $ g ++ "\0") ]
      }) servers
      listen = maybe (Right socket) (Left . read) port
    server listen myInfo datadir