mte/tools/offline.ml

113 lines
3.1 KiB
OCaml

open Cmdliner
open Cmdliner.Term.Syntax
open Offline_impl
let default_master_offline_key_file = "data/master_offline_private_key"
let default_response_file = "response.json"
let default_request_file = "request.json"
(*let to_term_ret = function Error e -> `Error (false, e) | Ok () -> `Ok ()*)
let to_term_ret = Fun.id
let setup_cmd =
let doc = "Generate offline master keys" in
let output =
let doc = "output file" in
Arg.(
value
& opt filepath default_master_offline_key_file
& info [ "o"; "output" ] ~doc)
in
Cmd.make (Cmd.info "setup" ~doc)
@@
let+ output = output in
setup ~output |> to_term_ret
let download_cmd =
let doc = "GET /management/keys/" in
let man =
[ `S Manpage.s_description; `P "$(cmd) download /management/keys." ]
in
let output =
let doc = "output file" in
Arg.(
value & opt filepath default_response_file & info [ "o"; "output" ] ~doc)
in
Cmd.make (Cmd.info "download" ~doc ~man)
@@
let+ output = output in
download ~output |> to_term_ret
let upload_cmd =
let doc = "POST input.json data to url (default /management/keys)" in
let input =
let doc = "input file" in
Arg.(value & opt file default_request_file & info [ "i"; "input" ] ~doc)
in
let url =
let doc = "url" in
Arg.(value & opt string "/management/keys" & info [ "url" ] ~doc)
in
Cmd.make (Cmd.info "upload" ~doc)
@@
let+ input = input and+ url = url in
upload ~input ~url |> to_term_ret
let master_key =
let doc = "Master offline Eddsa private key file." in
Arg.(value & opt file default_master_offline_key_file & info [ "key" ] ~doc)
let input =
let doc = "input file" in
Arg.(value & opt file default_response_file & info [ "i"; "input" ] ~doc)
let output =
let doc = "output file" in
Arg.(value & opt filepath default_request_file & info [ "o"; "output" ] ~doc)
let sign_cmd =
let doc = "Sign FutureKeysResponse." in
Cmd.make (Cmd.info "sign" ~doc)
@@
let+ input = input and+ output = output and+ master_key = master_key in
sign ~input ~output ~master_key |> to_term_ret
let revoke_denom_cmd =
let doc = "Revoke denomination." in
let h_denom =
let doc = "hash of denomination public key" in
Arg.(required & pos 0 (some string) None & info [] ~doc)
in
Cmd.make (Cmd.info "revoke-denom" ~doc)
@@
let+ output = output and+ master_key = master_key and+ h_denom = h_denom in
revoke_denom ~output ~master_key ~h_denom |> to_term_ret
let revoke_signkey_cmd =
let doc = "Revoke signkey." in
let signkey =
let doc = "public signing key" in
Arg.(required & pos 0 (some string) None & info [] ~doc)
in
Cmd.make (Cmd.info "revoke-signkey" ~doc)
@@
let+ output = output and+ master_key = master_key and+ signkey = signkey in
revoke_signkey ~output ~master_key ~signkey |> to_term_ret
let cli =
let info =
let doc = "MTE Offline CLI tool" in
Cmd.info "mte-offline" ~doc
in
Cmd.group info
[
setup_cmd;
download_cmd;
sign_cmd;
upload_cmd;
revoke_denom_cmd;
revoke_signkey_cmd;
]
let main () = Cmd.eval_result cli
let () = if !Sys.interactive then () else exit (main ())