lanonna/pam/bin/main.ml

open Datatypes
open Batteries
open Utils

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


let generate_alert = function
  (*
   * if the deadline has passed: alert every day at noon
   * else parse the alert
   **)
  | {due_date=None; _} -> None
  | {due_date=Some date; body; _} ->
    let now = Ptime_clock.now () in
    let today = now |> Ptime.to_date in
    Ptime.of_rfc3339 date
    |> Ptime.rfc3339_string_error
    |> Result.map (fun (timestamp, _, _) -> Ptime.to_date timestamp = today)
    |> Option.some


let http_client (repos: Config.repo_data list) =
  let http_actor = Httpclient.init repos in
  let _ = print_endline "Initialized http client" in

  let rec loop () =
    let _ =
      match Riot.receive () with
      | ListIssues ->
        Httpclient.make_get_request http_actor
        |> Result.map Datatypes.forgejo_issues
        |> Result.map_error Datatypes.forgejo_error
        |> Result.fold ~ok:identity ~error:identity
        |> Riot.send_by_name ~name:_MQ_CLIENT


      | m -> unhandled m
    in
    loop ()
  in loop ()

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

  let call_consumer { Amqp_client_lwt.Message.message = (_content, body); _ } =
    Pamlog.error [%string "Received msg from rabbitmq: %{body}. PAM will ignore."]
  in

  let%lwt mq = Mq.init (mq_url, mq_user, mq_password) 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 _ = Pamlog.error [%string "Got error from Forgejo: %{err}"] in
          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 (config: Config.config) =
  let open Riot in
  let http_client_pid = spawn (fun () -> http_client config.repos) in  
  let mq_client_pid = spawn (fun () -> Lwt_main.run (mq_client (config.mq_url, config.mq_user, config.mq_password))) 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 config = Config.configuration () |> Result.fold ~error:exit2 ~ok:identity in
  Riot.run (fun () -> main config)