Source file `yojson_utils.ml`

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
(**
   This module contains utility functions for writing Yojson codec generators.
*)

open Batteries;;

let list_to_yojson element_to_yojson lst =
  `List (List.map element_to_yojson lst)
;;

let set_to_yojson element_to_yojson enumerator set =
  `Assoc
    [ ( "type"
      , `String "set"
      )
    ; ( "elements"
      , `List
          (
            set
            |> enumerator
            |> Enum.map element_to_yojson
            |> List.of_enum
          )
      )
    ]
;;

let map_to_yojson key_to_yojson value_to_yojson enumerator map =
  `Assoc
    [ ( "type"
      , `String "map"
      )
    ; ( "mappings"
      , `List
        (
          map
          |> enumerator
          |> Enum.map
            (fun (k,v) ->
               `Assoc
                 [ ( "key" , key_to_yojson k )
                 ; ( "value", value_to_yojson v )
                 ]
            )
          |> List.of_enum
        )
      )
    ]
;;

module type To_yojson_type =
sig
  type t
  val to_yojson : t -> Yojson.Safe.t
end;;

module Set_to_yojson(S : Set.S)(Y : To_yojson_type with type t = S.elt) =
struct
  let to_yojson = set_to_yojson Y.to_yojson S.enum;;
end;;

module Map_to_yojson(M : Map.S)(Y : To_yojson_type with type t = M.key) =
struct
  let to_yojson value_to_yojson =
    map_to_yojson Y.to_yojson value_to_yojson M.enum
  ;;
end;;