92 lines
2.6 KiB
OCaml
92 lines
2.6 KiB
OCaml
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)
|