open Vmm_core
open Albatross_stats_pure

let old_rt = ref 0L

let timer pid vmmapi interval =
  let rusage, kinfo_mem = sysctl_kinfo_proc pid in
  Logs.app (fun m -> m "sysctl rusage: %a" Stats.pp_rusage_mem rusage) ;
  Logs.app (fun m -> m "kinfo mem: %a" Stats.pp_kinfo_mem kinfo_mem) ;
  let delta, pct =
    if !old_rt = 0L then
      let delta = kinfo_mem.Stats.runtime in
      let now = Unix.gettimeofday () in
      let start =
        Int64.to_float (fst kinfo_mem.start) +.
        ((float_of_int (snd kinfo_mem.start)) /. 1_000_000_000.)
      in
      delta, Int64.to_float delta /. ((now -. start) *. 1_000_000.)
    else
      let delta = Int64.sub kinfo_mem.Stats.runtime !old_rt in
      delta, Int64.to_float delta /. (interval *. 1_000_000.)
  in
  Logs.app (fun m -> m "delta %Lu pct %f" delta pct) ;
  old_rt := kinfo_mem.Stats.runtime;
  match vmmapi with
  | None -> ()
  | Some vmctx -> match wrap vmmapi_stats vmctx with
    | None -> Logs.app (fun m -> m "no vmctx stats")
    | Some st ->
      let all = List.combine !descr st in
      Logs.app (fun m -> m "bhyve stats %a" Stats.pp_vmm_mem all)

let jump _ pid name interval tmpdir =
  Sys.(set_signal sigpipe Signal_ignore) ;
  Albatross_cli.set_tmpdir tmpdir;
  let interval = Duration.(to_f (of_sec interval)) in
  Lwt_main.run (
    let vmmapi = match name with
      | None ->
        Logs.warn (fun m -> m "no name, no vmmapi") ;
        None
      | Some name -> match wrap vmmapi_open name with
        | None ->
          Logs.warn (fun m -> m "vmmapi_open failed for %s" name) ;
          None
        | Some vmctx ->
          Logs.info (fun m -> m "vmmapi_open succeeded for %s" name) ;
          fill_descr vmctx ;
          Some vmctx
    in
    let _ev = Lwt_engine.on_timer interval true (fun _e -> timer pid vmmapi interval) in
    let t, _u = Lwt.task () in
    t)

open Cmdliner
open Albatross_cli

let interval =
  let doc = "Interval between statistics gatherings (in seconds)" in
  Arg.(value & opt int 10 & info [ "interval" ] ~doc)

let pid =
  let doc = "Process id (defaults to own pid)" in
  Arg.(value & opt int (Unix.getpid ()) & info [ "pid" ] ~doc)

let vmname =
  let doc = "VM name" in
  Arg.(value & opt (some string) None & info [ "name" ] ~doc)

let cmd =
  Term.(term_result (const jump $ setup_log $ pid $ vmname $ interval $ tmpdir)),
  Term.info "albatross_stat_client" ~version

let () = match Term.eval cmd with `Ok () -> exit 0 | _ -> exit 1