Fix bully build

aurora-is-near · Oct 28, 2021 · 3c7653a · 3c7653a
1 parent 7f17239
commit 3c7653a
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/engine/src/lib.rs b/engine/src/lib.rs
@@ -489,28 +489,31 @@ mod contract {
     #[cfg(feature = "evm_bully")]
     #[no_mangle]
     pub extern "C" fn begin_chain() {
-        let mut state = Engine::get_state().sdk_unwrap();
+        let mut io = Runtime;
+        let mut state = Engine::get_state(&io).sdk_unwrap();
         require_owner_only(&state);
-        let args: BeginChainArgs = sdk::read_input_borsh().sdk_unwrap();
+        let args: BeginChainArgs = io.read_input_borsh().sdk_unwrap();
         state.chain_id = args.chain_id;
-        Engine::set_state(state);
+        Engine::set_state(&mut io, state);
         // set genesis block balances
         for account_balance in args.genesis_alloc {
             Engine::set_balance(
+                &mut io,
                 &Address(account_balance.address),
                 &crate::prelude::types::Wei::new(U256::from(account_balance.balance)),
             )
         }
         // return new chain ID
-        sdk::return_output(&Engine::get_state().sdk_unwrap().chain_id)
+        io.return_output(&Engine::get_state(&io).sdk_unwrap().chain_id)
     }
 
     #[cfg(feature = "evm_bully")]
     #[no_mangle]
     pub extern "C" fn begin_block() {
-        let state = Engine::get_state().sdk_unwrap();
+        let io = Runtime;
+        let state = Engine::get_state(&io).sdk_unwrap();
         require_owner_only(&state);
-        let _args: BeginBlockArgs = sdk::read_input_borsh().sdk_unwrap();
+        let _args: BeginBlockArgs = io.read_input_borsh().sdk_unwrap();
         // TODO: https://github.com/aurora-is-near/aurora-engine/issues/2
     }