From b8b235f7de4fd0543e218eb616f8040d13e70100 Mon Sep 17 00:00:00 2001
From: swrup <swrup@protonmail.com>
Date: Fri, 28 Nov 2025 11:08:43 +0100
Subject: [PATCH]

---
 tools/dune       |  2 +-
 tools/offline.ml | 45 ++++++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 45 insertions(+), 2 deletions(-)

diff --git a/tools/dune b/tools/dune
index 3375a448..3366e550 100644
--- a/tools/dune
+++ b/tools/dune
@@ -2,4 +2,4 @@
  (public_name offline)
  (name offline)
  (modules offline)
- (libraries cmdliner bos fmt mirage-crypto ptime mte))
+ (libraries cmdliner bos fmt mirage-crypto ptime mte vif))
diff --git a/tools/offline.ml b/tools/offline.ml
index 5d06c89b..eef53279 100644
--- a/tools/offline.ml
+++ b/tools/offline.ml
@@ -1 +1,44 @@
-let () = Fmt.pr "hello~@."; ()
+let download ~output_file =
+  let output_file = Fpath.v output_file in
+  Miou_unix.run @@ fun () ->
+  let rng = Mirage_crypto_rng_miou_unix.(initialize (module Pfortuna)) in
+  let finally () = Mirage_crypto_rng_miou_unix.kill rng in
+  Fun.protect ~finally @@ fun () ->
+  let fn _meta _req _resp buf = function
+    | Some str -> Buffer.add_string buf str; buf
+    | None -> buf
+  in
+  let uri = "http://localhost:3434/management/keys/" in
+  let prm =
+    Miou.async @@ fun () -> Httpcats.request ~fn ~uri (Buffer.create 0x1000)
+  in
+  match Miou.await_exn prm with
+  | Error err -> Fmt.failwith "Download failure: `%a`@." Httpcats.pp_error err
+  | Ok (_resp, buf) -> (
+      let data = Buffer.to_bytes buf |> Bytes.to_string in
+      match Bos.OS.File.write output_file data with
+      | Error (`Msg err) -> Fmt.failwith "Download write failure: `%s`@." err
+      | Ok () -> ())
+
+(* CLI *)
+
+open Cmdliner
+open Cmdliner.Term.Syntax
+
+let output =
+  let doc = "File to save data to be signed." in
+  Arg.(
+    value & opt filepath "management_keys.json" & info [ "o"; "output" ] ~doc)
+
+let download_cmd =
+  let doc = "Downloads /management/keys/" in
+  let man =
+    [ `S Manpage.s_description; `P "$(cmd) download /management/keys." ]
+  in
+  Cmd.make (Cmd.info "offline" ~version:"%%VERSION%%" ~doc ~man)
+  @@
+  let+ output_file = output in
+  download ~output_file
+
+let main () = Cmd.eval download_cmd
+let () = if !Sys.interactive then () else exit (main ())