Add dump-data-json subcommand

ushitora-anqou · Sep 26, 2023 · 0a50e8b · 0a50e8b
1 parent ce4af95
commit 0a50e8b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/bin/main.ml b/bin/main.ml
@@ -39,6 +39,11 @@ let () =
             const generate
             $ Arg.(
                 required & pos 0 (some int) None & info ~docv:"NUM-ENTRIES" []));
+        v (info "dump-data-json")
+          Term.(
+            const dump_data_json
+            $ Arg.(
+                required & pos 0 (some string) None & info ~docv:"IN-FILE" []));
       ]
     |> eval)
   |> exit
diff --git a/lib/command.ml b/lib/command.ml
@@ -59,3 +59,8 @@ let serve =
 
 let generate num_entries =
   Generator.generate_sample num_entries |> print_endline
+
+let dump_data_json in_filename =
+  Lwt_main.run
+    (Web_server.generate_json in_filename
+    >|= Yojson.Safe.to_string >|= print_string)