From a2988b650e39d7bf0cfed5972fe4d4a8cd16c636 Mon Sep 17 00:00:00 2001
From: Dan Cline <6798349+Rjected@users.noreply.github.com>
Date: Mon, 20 Jan 2025 11:40:13 -0500
Subject: [PATCH] feat: add state_hook_sender fn on StateRootTask

---
 crates/engine/tree/src/tree/root.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/crates/engine/tree/src/tree/root.rs b/crates/engine/tree/src/tree/root.rs
index 93cac7b435ed..717c75b04ac5 100644
--- a/crates/engine/tree/src/tree/root.rs
+++ b/crates/engine/tree/src/tree/root.rs
@@ -202,8 +202,8 @@ impl ProofSequencer {
 }
 
 /// A wrapper for the sender that signals completion when dropped
-#[derive(Deref)]
-pub(crate) struct StateHookSender<BPF: BlindedProviderFactory>(Sender<StateRootMessage<BPF>>);
+#[derive(Deref, Debug)]
+pub struct StateHookSender<BPF: BlindedProviderFactory>(Sender<StateRootMessage<BPF>>);
 
 impl<BPF: BlindedProviderFactory> StateHookSender<BPF> {
     pub(crate) const fn new(inner: Sender<StateRootMessage<BPF>>) -> Self {
@@ -325,9 +325,14 @@ where
         StateRootHandle::new(rx)
     }
 
+    /// Returns a [`StateHookSender`] that can be used to send state updates to this task.
+    pub fn state_hook_sender(&self) -> StateHookSender<BPF> {
+        StateHookSender::new(self.tx.clone())
+    }
+
     /// Returns a state hook to be used to send state updates to this task.
     pub fn state_hook(&self) -> impl OnStateHook {
-        let state_hook = StateHookSender::new(self.tx.clone());
+        let state_hook = self.state_hook_sender();
 
         move |state: &EvmState| {
             if let Err(error) = state_hook.send(StateRootMessage::StateUpdate(state.clone())) {