blob: b5adb474693a74d5ab18dce51667328781066e3d (
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
75
76
77
|
open Lwt
open Defaults
open Vyconf_config
let () = Lwt_log.add_rule "*" Lwt_log.Info
(* Default VyConf configuration *)
let daemonize = ref true
let config_file = ref defaults.config_file
let log_file = ref None
let log_template = ref "$(date): $(message)"
(* Global data *)
(* Command line arguments *)
let args = [
("--no-daemon", Arg.Unit (fun () -> daemonize := false),
"Do not daemonize");
("--config", Arg.String (fun s -> config_file := s),
(Printf.sprintf "<string> Configuration file, default is %s" defaults.config_file));
("--log-file", Arg.String (fun s -> log_file := Some s),
"<string> Log file");
("--version", Arg.Unit (fun () -> print_endline @@ Version.version_info (); exit 0), "Print version and exit")
]
let usage = "Usage: " ^ Sys.argv.(0) ^ " [options]"
let panic msg =
Lwt_log.fatal msg |> Lwt.ignore_result;
exit 1
let load_config path =
let result = Vyconf_config.load path in
match result with
| Ok cfg -> cfg
| Error err ->
panic (Printf.sprintf "Could not load the configuration file %s" err)
let check_dirs dirs =
let res = Directories.test dirs in
match res with
| Ok _ -> ()
| Error err -> panic err
let setup_logger daemonize log_file template =
(*
If log file is specified, log to the file whether we are a daemon or not
If we are a daemon and log file is not specified, log to syslog
If we are not a daemon and log file is not specified, log to stderr
*)
match log_file with
| None ->
if daemonize then
begin
Lwt_log.default := Lwt_log.syslog ~template:template ~facility:`Daemon ();
Lwt.return_unit
end
else
begin
Lwt_log.default := Lwt_log.channel ~template:template ~close_mode:`Keep ~channel:Lwt_io.stderr ();
Lwt.return_unit
end
| Some file ->
let%lwt l = Lwt_log.file ~template:template ~mode:`Append ~file_name:file () in
Lwt_log.default := l; Lwt.return_unit
let main_loop config () =
let%lwt () = setup_logger !daemonize !log_file !log_template in
let%lwt () = Lwt_log.notice @@ Printf.sprintf "Starting VyConf for %s" config.app_name in
Lwt.return_unit
let () =
let () = Arg.parse args (fun f -> ()) usage in
let config = load_config !config_file in
let dirs = Directories.make config in
check_dirs dirs;
Lwt_main.run @@ main_loop config ()
|
