bencode helpers between dicts and string maps

1 files changed, 5 insertions, 9 deletions

diff --git a/lib/concepts/Tuple.ml b/lib/concepts/Tuple.ml
index dfb7cbe..943c1f6 100644
--- a/lib/concepts/Tuple.ml
+++ b/lib/concepts/Tuple.ml
@@ -23,19 +23,15 @@ and Body : Encoding.Record.BODY = struct
 
   let fields {type_; attributes} =
     [ "type", Encoding.Bencode.String type_;
-      ( "attributes",
-        Encoding.Bencode.Dict
-          ( BatMap.String.enum attributes
-          |> BatEnum.fold (fun acc (name, value) -> (name, bencode_of_value value) :: acc) [] ) ) ]
+      "attributes", Encoding.Bencode.of_map bencode_of_value attributes ]
 
   let of_fields data =
     let open Utilities.Result in
     let* type_ = Encoding.Bencode.field "type" data |>> Encoding.Bencode.as_string in
-    let* kvs = Encoding.Bencode.field "attributes" data |>> Encoding.Bencode.as_dict in
-    kvs
-    |> List.map (fun (name, v) -> value_of_bencode v |> Result.map (fun v -> name, v))
-    |> Utilities.List.sequence
-    |> Result.map (fun kvs -> {type_; attributes= BatMap.String.of_list kvs})
+    let* attributes =
+      Encoding.Bencode.field "attributes" data |>> Encoding.Bencode.to_map value_of_bencode
+    in
+    Ok {type_; attributes}
 end
 
 let hash tuple = Representation.to_blob tuple |> Hash.hash_of_blob