Expose protocol_states_for_root_scan_state in transition frontier

Author: georgeee

src/lib/transition_frontier/frontier_base/frontier_intf.ml

@@ -43,4 +43,11 @@ module type S = sig
   val precomputed_values : t -> Precomputed_values.t
 
   val genesis_constants : t -> Genesis_constants.t
+
+  module Protocol_states_for_root_scan_state : sig
+    type t
+  end
+
+  val protocol_states_for_root_scan_state :
+    t -> Protocol_states_for_root_scan_state.t
 end

src/lib/transition_frontier/full_frontier/full_frontier.mli

@@ -24,8 +24,6 @@ module type CONTEXT = sig
   val consensus_constants : Consensus.Constants.t
 end
 
-include Frontier_intf.S
-
 module Protocol_states_for_root_scan_state : sig
   type t = Protocol_state.value State_hash.With_state_hashes.t State_hash.Map.t
 
@@ -36,6 +34,10 @@ module Protocol_states_for_root_scan_state : sig
     -> Protocol_state.value State_hash.With_state_hashes.t list
 end
 
+include
+  Frontier_intf.S
+    with module Protocol_states_for_root_scan_state := Protocol_states_for_root_scan_state
+
 val create :
      context:(module CONTEXT)
   -> root_data:Root_data.t
@@ -54,9 +56,6 @@ val root_data : t -> Root_data.t
 
 val calculate_diffs : t -> Breadcrumb.t -> Diff.Full.E.t list
 
-val protocol_states_for_root_scan_state :
-  t -> Protocol_states_for_root_scan_state.t
-
 val apply_diffs :
      t
   -> Diff.Full.E.t list

src/lib/transition_frontier/transition_frontier.ml

@@ -9,13 +9,13 @@ open Mina_base
 module Ledger = Mina_ledger.Ledger
 module Root_ledger = Mina_ledger.Root
 include Frontier_base
-module Full_frontier = Full_frontier
 module Extensions = Extensions
 module Persistent_root = Persistent_root
 module Persistent_frontier = Persistent_frontier
 module Catchup_state = Catchup_state
 module Full_catchup_tree = Full_catchup_tree
 module Catchup_hash_tree = Catchup_hash_tree
+module Util = Full_frontier.Util
 
 module type CONTEXT = sig
   val logger : Logger.t
@@ -564,6 +564,12 @@ include struct
   (* why can't this one be proxied? *)
   let path_map ?max_length { full_frontier; _ } breadcrumb ~f =
     path_map ?max_length full_frontier breadcrumb ~f
+
+  module Protocol_states_for_root_scan_state =
+    Protocol_states_for_root_scan_state
+
+  let protocol_states_for_root_scan_state =
+    proxy1 protocol_states_for_root_scan_state
 end
 
 module For_tests = struct

src/lib/transition_frontier/transition_frontier.mli

@@ -18,6 +18,7 @@ module Catchup_state = Catchup_state
 module Full_catchup_tree = Full_catchup_tree
 module Catchup_hash_tree = Catchup_hash_tree
 module Root_ledger = Mina_ledger.Root
+module Util = Full_frontier.Util
 
 module type CONTEXT = sig
   val logger : Logger.t
@@ -33,7 +34,10 @@ module type CONTEXT = sig
   val signature_kind : Mina_signature_kind.t
 end
 
-include Frontier_intf.S
+include
+  Frontier_intf.S
+    with module Protocol_states_for_root_scan_state := Full_frontier
+                                                       .Protocol_states_for_root_scan_state
 
 type Structured_log_events.t += Added_breadcrumb_user_commands
   [@@deriving register_event]