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
75
76
77
78
79
80
|
{-# LANGUAGE QuasiQuotes #-}
module Main
( main
) where
import Data.ByteString.Char8 (pack)
import qualified Data.ConfigFile as Cf
import Data.Either.Utils (forceEither)
import Data.List (isPrefixOf)
import Data.Maybe (fromJust)
import Data.Version (showVersion)
import Database.MySQL.Base (ConnectInfo(..), defaultSSLInfo)
import Database.MySQL.Base.Types
(Option(ReadDefaultFile, ReadDefaultGroup))
import Paths_mywatch (getDataDir, version)
import qualified System.Console.Docopt.NoTH as O
import System.Environment (getArgs)
import Text.InterpolatedString.Perl6 (qc)
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
|
