50 lines
1.7 KiB
OCaml
50 lines
1.7 KiB
OCaml
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) ->
|
|
match Secmod_signkey.load conn with
|
|
| Ok None ->
|
|
let t = Secmod_signkey.make_new () in
|
|
Logs.info (fun m -> m "secmod_signkey initialized with fresh keys");
|
|
t
|
|
| Ok (Some v) ->
|
|
Logs.info (fun m -> m "secmod_signkey initialized from storage");
|
|
v
|
|
| Error _ ->
|
|
(* TODO error: pretty print *)
|
|
Fmt.failwith "secmod_signkey init failure."
|
|
|
|
let secmod_denom =
|
|
let finally _key = () in
|
|
Vif.Device.v ~name:"secmod_denom" ~finally [ Vif.Device.value db_connection ]
|
|
@@ fun conn (_env : env) ->
|
|
match Secmod_denom.load conn with
|
|
| Ok None ->
|
|
let t = Secmod_denom.make_new () in
|
|
Logs.info (fun m -> m "secmod_denom initialized with fresh keys");
|
|
t
|
|
| Ok (Some v) ->
|
|
Logs.info (fun m -> m "secmod_denom initialized from storage");
|
|
v
|
|
| Error _ -> Fmt.failwith "secmod_denom init failure."
|