Source file block_directory_helpers.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
(*****************************************************************************)
(*                                                                           *)
(* Open Source License                                                       *)
(* Copyright (c) 2023 Nomadic Labs, <contact@nomadic-labs.com>               *)
(*                                                                           *)
(* Permission is hereby granted, free of charge, to any person obtaining a   *)
(* copy of this software and associated documentation files (the "Software"),*)
(* to deal in the Software without restriction, including without limitation *)
(* the rights to use, copy, modify, merge, publish, distribute, sublicense,  *)
(* and/or sell copies of the Software, and to permit persons to whom the     *)
(* Software is furnished to do so, subject to the following conditions:      *)
(*                                                                           *)
(* The above copyright notice and this permission notice shall be included   *)
(* in all copies or substantial portions of the Software.                    *)
(*                                                                           *)
(* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR*)
(* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,  *)
(* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL   *)
(* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER*)
(* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING   *)
(* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER       *)
(* DEALINGS IN THE SOFTWARE.                                                 *)
(*                                                                           *)
(*****************************************************************************)

(* Conveniences to construct RPC directory
   against a subcontext of the Node_context *)

let get_head store =
  let open Lwt_result_syntax in
  let* head = Node_context.last_processed_head_opt store in
  match head with
  | None -> failwith "No head"
  | Some {header = {block_hash; _}; _} -> return block_hash

let get_head_level store =
  let open Lwt_result_syntax in
  let* head = Node_context.last_processed_head_opt store in
  match head with
  | None -> failwith "No head"
  | Some {header = {level; _}; _} -> return level

let get_finalized node_ctxt =
  let open Lwt_result_syntax in
  let* level = Node_context.get_finalized_level node_ctxt in
  Node_context.hash_of_level node_ctxt level

let get_last_cemented (node_ctxt : _ Node_context.t) =
  protect @@ fun () ->
  let lcc = Reference.get node_ctxt.lcc in
  Node_context.hash_of_level node_ctxt lcc.level

let block_of_prefix node_ctxt block =
  match block with
  | `Head -> get_head node_ctxt
  | `Hash b -> Lwt_result.return b
  | `Level l -> Node_context.hash_of_level node_ctxt l
  | `Finalized -> get_finalized node_ctxt
  | `Cemented -> get_last_cemented node_ctxt

let block_level_of_id node_ctxt block =
  match block with
  | `Head -> get_head_level node_ctxt
  | `Hash b -> Node_context.level_of_hash node_ctxt b
  | `Level l -> Lwt_result.return l
  | `Finalized -> Node_context.get_finalized_level node_ctxt
  | `Cemented -> Lwt_result.return (Reference.get node_ctxt.lcc).level