use cmdliner

This commit is contained in:
Hannes Mehnert 2022-01-27 12:58:01 +01:00
parent 9bee347c17
commit 82f99de39c
2 changed files with 51 additions and 42 deletions

View file

@ -1,4 +1,4 @@
(executable
(name main)
(public_name opam_graph)
(libraries cmdliner logs logs.fmt opam-graph))
(libraries cmdliner logs logs.fmt fmt.cli fmt.tty logs.cli opam-graph))

View file

@ -11,45 +11,54 @@ let read_file file =
invalid_arg ("Error reading file: " ^ file)
with _ -> invalid_arg ("Error opening file " ^ file)
let () =
match Sys.argv with
| [| _ ; output_format; file |] -> (
let jump () transitive file output_format =
let switch = read_file file in
let data = OpamFile.SwitchExport.read_from_string switch in
match output_format with
| "text" ->
let graph = Opam_graph.dependencies data in
| `Text ->
let graph = Opam_graph.dependencies ~transitive data in
Format.printf "%a" Opam_graph.pp_graph graph
| "text_transitive" ->
let graph = Opam_graph.dependencies ~transitive:true data in
Format.printf "%a" Opam_graph.pp_graph graph
| "dot" ->
let graph = Opam_graph.dependencies data in
| `Dot ->
let graph = Opam_graph.dependencies ~transitive data in
let dot = Opam_graph.Render.Dot.of_graph graph in
Format.printf "%a" Opam_graph.Render.Dot.pp dot
| "dot_ui" ->
let transitive = false in
| `Dot_ui ->
let graph = Opam_graph.Ui.dependencies ~transitive data in
let dot = Opam_graph.Render.Dot.of_assoc graph in
Format.printf "%a" Opam_graph.Render.Dot.pp dot
| "dot_ui_transitive" ->
let transitive = true in
let graph = Opam_graph.Ui.dependencies ~transitive data in
let dot = Opam_graph.Render.Dot.of_assoc graph in
Format.printf "%a" Opam_graph.Render.Dot.pp dot
| "html_ui" ->
let transitive = false in
| `Html ->
let graph = Opam_graph.Ui.dependencies ~transitive data in
let html = Opam_graph.Render.Html.of_assoc graph in
Format.printf "%a" Opam_graph.Render.Html.pp html
| "html_ui_transitive" ->
let transitive = true in
let graph = Opam_graph.Ui.dependencies ~transitive data in
let html = Opam_graph.Render.Html.of_assoc graph in
Format.printf "%a" Opam_graph.Render.Html.pp html
| _ -> failwith "Unsupported output format"
)
| _ ->
print_endline "expecting exactly one argument";
exit 1
let setup_log style_renderer level =
Fmt_tty.setup_std_outputs ?style_renderer ();
Logs.set_level level;
Logs.set_reporter (Logs_fmt.reporter ~dst:Format.std_formatter ())
open Cmdliner
let setup_log =
Term.(const setup_log
$ Fmt_cli.style_renderer ()
$ Logs_cli.level ())
let transitive =
let doc = "Transitive dependencies" in
Arg.(value & flag & info [ "transitive" ] ~doc)
let formats = [ "html", `Html ; "dot", `Dot ; "dot-ui", `Dot_ui ; "text", `Text ]
let output_format =
let doc = "Output format" in
Arg.(value & opt (Arg.enum formats) `Text & info [ "output-format" ] ~doc)
let file =
let doc = "The opam switch export to graph" in
Arg.(required & pos 0 (some file) None & info [ ] ~doc ~docv:"FILE")
let cmd =
Term.(const jump $ setup_log $ transitive $ file $ output_format),
Term.info "opam_graph" ~version:"%%VERSION%%"
let () = match Term.eval cmd with `Ok () -> exit 0 | _ -> exit 1