ligo/src/client_main.ml

86 lines
3.2 KiB
OCaml
Raw Normal View History

2016-09-08 21:13:10 +04:00
(**************************************************************************)
(* *)
(* Copyright (c) 2014 - 2016. *)
(* Dynamic Ledger Solutions, Inc. <contact@tezos.com> *)
(* *)
(* All rights reserved. No warranty, explicit or implicit, provided. *)
(* *)
(**************************************************************************)
(* Tezos Command line interface - Main Program *)
open Lwt
let () =
let startup =
CalendarLib.Printer.Precise_Calendar.sprint
"%Y-%m-%dT%H:%M:%SZ"
(CalendarLib.Calendar.Precise.now ()) in
let log channel msg = match channel with
| "stdout" ->
print_endline msg ;
Lwt.return ()
| "stderr" ->
prerr_endline msg ;
Lwt.return ()
| log ->
Utils.create_dir Client_config.(base_dir#get // "logs" // log) >>= fun () ->
Lwt_io.with_file
~flags: Unix.[ O_APPEND ; O_CREAT ; O_WRONLY ]
~mode: Lwt_io.Output
Client_config.(base_dir#get // "logs" // log // startup)
(fun chan -> Lwt_io.write chan msg) in
Cli_entries.log_hook := Some log
2016-09-08 21:13:10 +04:00
(* Main (lwt) entry *)
let main () =
Random.self_init () ;
Sodium.Random.stir () ;
catch
(fun () ->
let block = Client_config.preparse_args () in
Lwt.catch
(fun () ->
Client_node_rpcs.Blocks.protocol block)
(fun _ ->
Cli_entries.message "\n\
The connection to the RPC server failed, \
using the default protocol version.\n" >>= fun () ->
2016-09-08 21:13:10 +04:00
Lwt.return Client_bootstrap.Client_proto_main.protocol)
>>= fun version ->
let commands =
Client_generic_rpcs.commands @
Client_keys.commands () @
2016-10-25 21:00:03 +04:00
Client_protocols.commands () @
2016-10-16 23:57:56 +04:00
Client_helpers.commands () @
2016-09-08 21:13:10 +04:00
Client_version.commands_for_version version in
Client_config.parse_args ~version
(Cli_entries.usage commands)
(Cli_entries.inline_dispatch commands))
2016-09-08 21:13:10 +04:00
(function
| Arg.Help help ->
Format.printf "%s%!" help ;
Pervasives.exit 0
| Arg.Bad help ->
Format.eprintf "%s%!" help ;
Pervasives.exit 1
| Cli_entries.Command_not_found ->
Format.eprintf "Unkonwn command, try `-help`.\n%!" ;
Pervasives.exit 1
| Client_version.Version_not_found ->
Format.eprintf "Unkonwn protocol version, try `list versions`.\n%!" ;
Pervasives.exit 1
| Cli_entries.Bad_argument (idx, _n, v) ->
Format.eprintf "There's a problem with argument %d, %s.\n%!" idx v ;
Pervasives.exit 1
| Cli_entries.Command_failed message ->
Format.eprintf "Command failed, %s.\n%!" message ;
Pervasives.exit 1
| exn ->
Format.printf "Fatal internal error: %s\n%!"
(Printexc.to_string exn) ;
Pervasives.exit 1)
(* Where all the user friendliness starts *)
let () = Lwt_main.run (main ())