Update end_slot_hook

examples/demo-stf/src/runtime.rs

@@ -119,8 +119,13 @@ impl<C: Context, Cond: ValidityCondition> SlotHooks<Cond> for Runtime<C> {
 
     fn end_slot_hook(
         &self,
-        _working_set: &mut sov_state::WorkingSet<<Self::Context as sov_modules_api::Spec>::Storage>,
+        #[allow(unused_variables)] root_hash: [u8; 32],
+        #[allow(unused_variables)] working_set: &mut sov_state::WorkingSet<
+            <Self::Context as sov_modules_api::Spec>::Storage,
+        >,
     ) {
+        #[cfg(feature = "experimental")]
+        self.evm.end_slot_hook(root_hash, working_set);
     }
 }
 

module-system/module-implementations/integration-tests/src/chain_state/helpers.rs

@@ -72,7 +72,12 @@ impl<C: Context, Cond: ValidityCondition> SlotHooks<Cond> for TestRuntime<C, Con
         self.chain_state.begin_slot_hook(slot_data, working_set)
     }
 
-    fn end_slot_hook(&self, _working_set: &mut WorkingSet<<Self::Context as Spec>::Storage>) {}
+    fn end_slot_hook(
+        &self,
+        _root_hash: [u8; 32],
+        _working_set: &mut WorkingSet<<Self::Context as Spec>::Storage>,
+    ) {
+    }
 }
 
 impl<C: Context, Cond: ValidityCondition> BlobSelector for TestRuntime<C, Cond> {

module-system/module-implementations/sov-chain-state/src/hooks.rs

@@ -63,5 +63,10 @@ impl<Ctx: Context, Cond: ValidityCondition> SlotHooks<Cond> for ChainState<Ctx,
         );
     }
 
-    fn end_slot_hook(&self, _working_set: &mut WorkingSet<<Self::Context as Spec>::Storage>) {}
+    fn end_slot_hook(
+        &self,
+        _root_hash: [u8; 32],
+        _working_set: &mut WorkingSet<<Self::Context as Spec>::Storage>,
+    ) {
+    }
 }

module-system/module-implementations/sov-evm/src/hooks.rs

@@ -0,0 +1,9 @@
+use sov_state::WorkingSet;
+
+use crate::Evm;
+
+impl<C: sov_modules_api::Context> Evm<C> {
+    pub fn end_slot_hook(&self, _root_hash: [u8; 32], _working_set: &mut WorkingSet<C::Storage>) {
+        // TODO implement block creation logic.
+    }
+}

module-system/module-implementations/sov-evm/src/lib.rs

@@ -4,6 +4,8 @@ pub mod call;
 pub mod evm;
 #[cfg(feature = "experimental")]
 pub mod genesis;
+#[cfg(feature = "experimental")]
+pub mod hooks;
 #[cfg(feature = "native")]
 #[cfg(feature = "experimental")]
 pub mod query;

module-system/sov-modules-api/src/hooks.rs

@@ -63,5 +63,9 @@ pub trait SlotHooks<Condition: ValidityCondition> {
         working_set: &mut WorkingSet<<Self::Context as Spec>::Storage>,
     );
 
-    fn end_slot_hook(&self, working_set: &mut WorkingSet<<Self::Context as Spec>::Storage>);
+    fn end_slot_hook(
+        &self,
+        root_hash: [u8; 32],
+        working_set: &mut WorkingSet<<Self::Context as Spec>::Storage>,
+    );
 }

module-system/sov-modules-stf-template/src/lib.rs

@@ -99,8 +99,7 @@ where
             .expect("jellyfish merkle tree update must succeed");
 
         let mut working_set = WorkingSet::new(self.current_storage.clone());
-
-        self.runtime.end_slot_hook(&mut working_set);
+        self.runtime.end_slot_hook(root_hash, &mut working_set);
 
         (jmt::RootHash(root_hash), witness)
     }