blob: dc6eb5f410fd4f5d50b499804f488d1f87092d58 (
plain) (
blame)
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
type t = (string, Value.value) BatMap.t
let empty = BatMap.empty
let is_empty = BatMap.is_empty
let set = BatMap.add
let find_opt = BatMap.find_opt
module Error = struct
open Condition
let unknown_attribute name =
condition "unknown-attribute" "No such attribute on this tuple"
("attribute" |=| Value.String name)
end
let get name tuple =
match BatMap.find_opt name tuple with
| Some v -> Ok v
| None -> Error (Error.unknown_attribute name)
let attributes tuple = BatMap.foldi (fun name _ acc -> BatSet.add name acc) tuple BatSet.empty
let project names tuple =
let open Utilities.Result in
BatFingerTree.fold_left
(fun acc name ->
let* acc = acc in
let* value = get name tuple in
Ok (set name value acc))
(Ok empty) names
let rename pairs tuple =
let open Utilities.Result in
BatFingerTree.fold_left
(fun acc (old_name, new_name) ->
let* acc = acc in
let* value = get old_name tuple in
Ok (set new_name value acc))
(Ok empty) pairs
let merge left right =
BatMap.foldi
(fun name value acc ->
match acc with
| None -> None
| Some acc -> (
match BatMap.find_opt name acc with
| None -> Some (BatMap.add name value acc)
| Some value' -> if Value.equal value value' then Some acc else None))
right (Some left)
(* Tuple's own wire format for a single value: kept local rather than
added to Concepts.Encoding.Value (which only ever encodes hashes and
options of them today) so this module owns its representation
end to end. *)
let bencode_of_value = function
| Value.String s -> Encoding.Bencode.Tagged ('s', Encoding.Bencode.String s)
| Value.Integer n -> Encoding.Bencode.Tagged ('i', Encoding.Bencode.Int n)
let malformed_value () =
Condition.condition "malformed-tuple-value"
"A tuple field did not conform to what was expected" Condition.empty
let value_of_bencode = function
| Encoding.Bencode.Tagged ('s', Encoding.Bencode.String s) -> Ok (Value.String s)
| Encoding.Bencode.Tagged ('i', Encoding.Bencode.Int n) -> Ok (Value.Integer n)
| _ -> Error (malformed_value ())
module rec Representation : (Encoding.Record.S with type t = t) = Encoding.Record.Make (Body)
and Body : Encoding.Record.BODY = struct
type nonrec t = t
let tag = 'T'
let malformed = malformed_value
let fields tuple =
BatMap.foldi (fun name value acc -> (name, bencode_of_value value) :: acc) tuple []
let of_fields data =
let open Utilities.Result in
let* kvs = Encoding.Bencode.as_dict data in
kvs
|> List.map (fun (name, v) -> value_of_bencode v |> Result.map (fun v -> (name, v)))
|> Utilities.List.sequence
|> Result.map (List.fold_left (fun acc (name, v) -> set name v acc) empty)
end
let hash tuple = Representation.to_blob tuple |> Hash.hash_of_blob
|
