Source file role_set.ml

module type S = sig
  include Set.S

  val to_yojson : t -> Yojson.Safe.t
  val of_yojson : Yojson.Safe.t -> (t, string) CCResult.t
  val pp : Format.formatter -> t -> unit
end

module Make (R : Role.Sig) : S with type elt = R.t = struct
  include Set.Make (R)

  let to_yojson t = `List (CCList.map R.to_yojson (elements t))

  let of_yojson =
    let open CCResult in
    function
    | `List items ->
      CCList.fold_left
        (fun acc x ->
          acc >>= fun acc' -> x |> R.of_yojson >|= CCFun.flip add acc')
        (Ok empty)
        items
    | _ -> Error "Invalid role set"
  ;;

  let pp fmt t =
    let show = [%show: R.t list] in
    Format.fprintf fmt "[%s]" (show (elements t))
  ;;
end