Signer: do not depend on Client_main_run

This commit is contained in:
Grégoire Henry 2018-05-26 12:50:38 +02:00
parent 0dbe24290f
commit e8fb2bf515

View File

@ -109,12 +109,104 @@ let commands =
Https_daemon.run cctxt ~host ~port ~cert ~key) ; Https_daemon.run cctxt ~host ~port ~cert ~key) ;
] ]
let home = try Sys.getenv "HOME" with Not_found -> "/root"
let default_base_dir =
Filename.concat home ".tezos-signer"
let (//) = Filename.concat
let string_parameter () : (string, _) parameter =
parameter (fun _ x -> return x)
let base_dir_arg () =
arg
~long:"base-dir"
~short:'d'
~placeholder:"path"
~doc:("signer data directory\n\
The directory where the Tezos client will store all its data.\n\
By default: '" ^ default_base_dir ^"'.")
(string_parameter ())
let global_options () =
args1
(base_dir_arg ())
(* Main (lwt) entry *)
let main () =
let executable_name = Filename.basename Sys.executable_name in
let original_args, autocomplete =
(* for shell aliases *)
let rec move_autocomplete_token_upfront acc = function
| "bash_autocomplete" :: prev_arg :: cur_arg :: script :: args ->
let args = List.rev acc @ args in
args, Some (prev_arg, cur_arg, script)
| x :: rest -> move_autocomplete_token_upfront (x :: acc) rest
| [] -> List.rev acc, None in
match Array.to_list Sys.argv with
| _ :: args -> move_autocomplete_token_upfront [] args
| [] -> [], None in
Random.self_init () ;
ignore Clic.(setup_formatter Format.std_formatter
(if Unix.isatty Unix.stdout then Ansi else Plain) Short) ;
ignore Clic.(setup_formatter Format.err_formatter
(if Unix.isatty Unix.stderr then Ansi else Plain) Short) ;
begin
begin
parse_global_options
(global_options ()) () original_args >>=? fun (base_dir, remaining) ->
let base_dir = Option.unopt ~default:default_base_dir base_dir in
let cctxt = object
inherit Client_context_unix.unix_logger ~base_dir
inherit Client_context_unix.unix_prompter
inherit Client_context_unix.unix_wallet ~base_dir
end in
Client_keys.register_signer
(module Tezos_signer_backends.Encrypted) ;
Client_keys.register_signer
(module Tezos_signer_backends.Unencrypted) ;
let commands =
Clic.add_manual
~executable_name
~global_options:(global_options ())
(if Unix.isatty Unix.stdout then Clic.Ansi else Clic.Plain)
Format.std_formatter
commands in
begin match autocomplete with
| Some (prev_arg, cur_arg, script) ->
Clic.autocompletion
~script ~cur_arg ~prev_arg ~args:original_args
~global_options:(global_options ())
commands cctxt >>=? fun completions ->
List.iter print_endline completions ;
return ()
| None ->
Clic.dispatch commands cctxt remaining
end
end >>= function
| Ok () ->
Lwt.return 0
| Error [ Clic.Help command ] ->
Clic.usage
Format.std_formatter
~executable_name
~global_options:(global_options ())
(match command with None -> [] | Some c -> [ c ]) ;
Lwt.return 0
| Error errs ->
Clic.pp_cli_errors
Format.err_formatter
~executable_name
~global_options:(global_options ())
~default:Error_monad.pp
errs ;
Lwt.return 1
end >>= fun retcode ->
Format.pp_print_flush Format.err_formatter () ;
Format.pp_print_flush Format.std_formatter () ;
Logging_unix.close () >>= fun () ->
Lwt.return retcode
let () = let () =
Client_main_run.run Pervasives.exit (Lwt_main.run (main ()))
(fun _ _ ->
return @@
List.map
(Clic.map_command
(fun (o : Client_context.full) -> (o :> Client_context.io_wallet)))
commands)