mte/src/devices.ml

type env = {
  caqti_switch: Caqti_miou.Switch.t;
  db_uri: Uri.t;
}

let db_connection : (env, Caqti_miou.connection) Vif.Device.device =
  let finally (module Conn : Caqti_miou.CONNECTION) = Conn.disconnect () in
  Vif.Device.v ~name:"db_connection" ~finally []
  @@ fun { caqti_switch; db_uri } ->
  match Caqti_miou_unix.connect ~sw:caqti_switch db_uri with
  | Error err ->
      Fmt.failwith "Database connection failure: %a." Caqti_error.pp err
  | Ok conn -> (
      match Pg.preflight conn with
      | Error err ->
          Fmt.failwith "Database preflight failure: %a." Caqti_error.pp err
      | Ok () ->
          Logs.info (fun m -> m "database connection initialized");
          conn)

let secmod_signkey =
  let finally _key = () in
  Vif.Device.v ~name:"secmod_signkey" ~finally
    [ Vif.Device.value db_connection ]
  @@ fun conn (_env : env) -> Secmod_signkey.init conn

let secmod_denom =
  let finally _key = () in
  Vif.Device.v ~name:"secmod_denom" ~finally [ Vif.Device.value db_connection ]
  @@ fun conn (_env : env) -> Secmod_denom.init conn