open Riot
open Datatypes
open Batteries
open Utils

let _MQ_CLIENT = "mq_client"
let _HTTP_CLIENT = "http_client"

let http_client base64password =
  let http_actor = Httpclient.init base64password in
  let _ = print_endline "Initialized http client" in

  let rec loop () =
    let _ =
      match Riot.receive () with
      | ListIssues ->
        let reminders = Httpclient.make_get_request http_actor in
        let msg = result_unpack reminders
        in
        Riot.send_by_name ~name:_MQ_CLIENT msg 
      | m -> unhandled m
    in
    loop ()
  in loop ()

let mq_client (mq_url, mq_user, mq_password) consumer =
  let pprint rem =
    [%string "%{r.title}"]
  in

  let call_consumer pid { Amqp_client_lwt.Message.message = (_content, body); _ } =
    Riot.send pid (Datatypes.ReceivedFromMq ("GOT RABBITs: "^body)) (*TODO: log*)
  in

  let%lwt mq = Mq.init (mq_url, mq_user, mq_password) consumer call_consumer in

  let rec loop () =
    let%lwt _ = Lwt_unix.sleep 1.0 in
    let _ =  
      try%lwt
        match Riot.receive ~after:one_second () with 
        | ForgejoError err ->
          let _ = print_endline [%string "Got error from Forgejo: %{err}"] in
          (* Lwt.return (Mq.mq_publish mq err ) *)
          Mq.mq_publish mq err 
        | ForgejoIssues reminders ->
          let _ = [%string "Got reminders: %{Batteries.dump reminders}"] |> print_endline in
          let rems = List.map pprint reminders in
          Mq.mq_publish_all mq rems
        | m -> unhandled m
      with | Riot.Receive_timeout -> Lwt.return_unit

    in
    loop ()
  in
  loop ()

let main (mq_url, mq_user, mq_password, base64_creds) =
  let http_client_pid = spawn (fun () -> http_client base64_creds) in  
  let mq_client_pid = spawn (fun () -> Lwt_main.run (mq_client (mq_url, mq_user, mq_password) http_client_pid)) in
  let _ = Riot.register _HTTP_CLIENT http_client_pid in
  let _ = Riot.register _MQ_CLIENT mq_client_pid
  in
  let timeout = 6.0 in
  let rec loop_ () =
    let _ = send http_client_pid ListIssues in
    sleep timeout |> loop_
  in
  sleep timeout |> loop_
    


let () =
  let (mq_url, mq_user, mq_password, base64_creds) = Config.configuration () in
  Riot.run (fun () -> main (mq_url, mq_user, mq_password, base64_creds))