mte/src/devices.ml

(* TODO KV store *)

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

module Secmod_signkey = struct
  type t = {
    sm_key: Signkey.t;
    keys: Signkey.t list;
  }

  let v =
    let finally _key = () in
    Vif.Device.v ~name:"secmod_signkey" ~finally [] @@ fun (_env : env) ->
    let sm_key = Signkey.generate () in
    let keys = [ Signkey.generate () ] in
    { sm_key; keys }
end

module Secmod_denom = struct
  type t = {
    sm_key: Signkey.t;
    keys: Denomination.t list;
  }

  let v =
    let finally _key = () in
    Vif.Device.v ~name:"secmod_denom" ~finally [] @@ fun (_env : env) ->
    let sm_key = Signkey.generate () in
    let keys = List.map Denomination.make Config.Coin.all_coins in
    { sm_key; keys }
end

let secmod_signkey = Secmod_signkey.v
let secmod_denom = Secmod_denom.v

let caqti :
    ( env,
      (Caqti_miou.connection, Caqti_error.t) Caqti_miou_unix.Pool.t )
    Vif.Device.device =
  let finally pool = Caqti_miou_unix.Pool.drain pool in
  Vif.Device.v ~name:"caqti" ~finally [] @@ fun { caqti_switch; db_uri } ->
  match Caqti_miou_unix.connect_pool ~sw:caqti_switch db_uri with
  | Error err ->
      Fmt.failwith "Database connection failure: %a." Caqti_error.pp err
  | Ok pool -> pool