complete_setup endpoint

common/proxies/src/enshrine_esdt_safe_proxy.rs

@@ -218,15 +218,6 @@ where
             .original_result()
     }
 
-    pub fn end_setup_phase(
-        self,
-    ) -> TxTypedCall<Env, From, To, NotPayable, Gas, ()> {
-        self.wrapped_tx
-            .payment(NotPayable)
-            .raw_call("endSetupPhase")
-            .original_result()
-    }
-
     /// Tokens in the whitelist can be transferred without fees 
     pub fn add_tokens_to_whitelist<
         Arg0: ProxyArg<MultiValueEncoded<Env::Api, TokenIdentifier<Env::Api>>>,

common/proxies/src/esdt_safe_proxy.rs

@@ -221,15 +221,6 @@ where
             .original_result()
     }
 
-    pub fn end_setup_phase(
-        self,
-    ) -> TxTypedCall<Env, From, To, NotPayable, Gas, ()> {
-        self.wrapped_tx
-            .payment(NotPayable)
-            .raw_call("endSetupPhase")
-            .original_result()
-    }
-
     /// Tokens in the whitelist can be transferred without fees 
     pub fn add_tokens_to_whitelist<
         Arg0: ProxyArg<MultiValueEncoded<Env::Api, TokenIdentifier<Env::Api>>>,

common/proxies/src/sovereign_forge_proxy.rs

@@ -117,6 +117,15 @@ where
             .original_result()
     }
 
+    pub fn complete_setup_phase(
+        self,
+    ) -> TxTypedCall<Env, From, To, NotPayable, Gas, ()> {
+        self.wrapped_tx
+            .payment(NotPayable)
+            .raw_call("completeSetupPhase")
+            .original_result()
+    }
+
     pub fn deploy_phase_one<
         Arg0: ProxyArg<u64>,
         Arg1: ProxyArg<u64>,

common/setup-phase/src/lib.rs

@@ -4,42 +4,29 @@ multiversx_sc::imports!();
 
 #[multiversx_sc::module]
 pub trait SetupPhaseModule {
-    #[endpoint(endSetupPhase)]
-    fn end_setup_phase(&self) {
-        self.require_caller_initiator();
-        self.require_setup_phase();
-
-        self.setup_phase_complete().set(true);
+    fn require_setup_complete(&self, caller_shard_id: u32) {
+        require!(
+            self.is_setup_phase_complete(),
+            "The setup is not completed in shard {}",
+            caller_shard_id
+        );
     }
 
     fn require_caller_initiator(&self) {
         let caller = self.blockchain().get_caller();
-        let initiator_address = self.initiator_address().get();
-        require!(caller == initiator_address, "Invalid caller");
-    }
-
-    fn require_setup_phase(&self) {
-        require!(
-            !self.is_setup_phase_complete(),
-            "Setup phase complete already"
-        );
-    }
+        let initiator = self.initiator_address().get();
 
-    fn require_setup_complete(&self) {
-        require!(
-            self.is_setup_phase_complete(),
-            "Setup phase must be completed first"
-        );
+        require!(caller == initiator, "Invalid caller");
     }
 
     #[inline]
     fn is_setup_phase_complete(&self) -> bool {
         self.setup_phase_complete().get()
     }
 
-    #[storage_mapper("initiatorAddress")]
-    fn initiator_address(&self) -> SingleValueMapper<ManagedAddress>;
-
     #[storage_mapper("setupPhaseComplete")]
     fn setup_phase_complete(&self) -> SingleValueMapper<bool>;
+
+    #[storage_mapper("initiatorAddress")]
+    fn initiator_address(&self) -> SingleValueMapper<ManagedAddress>;
 }

enshrine-esdt-safe/interactor/src/interactor_main.rs

@@ -50,7 +50,6 @@ async fn main() {
         "registerNewTokenID" => interact.register_new_token_id().await,
         "setMaxBridgedAmount" => interact.set_max_bridged_amount().await,
         "getMaxBridgedAmount" => interact.max_bridged_amount().await,
-        "endSetupPhase" => interact.end_setup_phase().await,
         "addTokensToWhitelist" => interact.add_tokens_to_whitelist(TOKEN_ID).await,
         "removeTokensFromWhitelist" => interact.remove_tokens_from_whitelist().await,
         "addTokensToBlacklist" => interact.add_tokens_to_blacklist().await,
@@ -514,22 +513,6 @@ impl ContractInteract {
         println!("Result: {result_value:?}");
     }
 
-    async fn end_setup_phase(&mut self) {
-        let response = self
-            .interactor
-            .tx()
-            .from(&self.wallet_address)
-            .to(self.state.current_address())
-            .gas(30_000_000u64)
-            .typed(enshrine_esdt_safe_proxy::EnshrineEsdtSafeProxy)
-            .end_setup_phase()
-            .returns(ReturnsResultUnmanaged)
-            .run()
-            .await;
-
-        println!("Result: {response:?}");
-    }
-
     async fn add_tokens_to_whitelist(&mut self, token_id: &[u8]) {
         let tokens;
 

enshrine-esdt-safe/interactor/src/proxy.rs

@@ -218,15 +218,6 @@ where
             .original_result()
     }
 
-    pub fn end_setup_phase(
-        self,
-    ) -> TxTypedCall<Env, From, To, NotPayable, Gas, ()> {
-        self.wrapped_tx
-            .payment(NotPayable)
-            .raw_call("endSetupPhase")
-            .original_result()
-    }
-
     /// Tokens in the whitelist can be transferred without fees 
     pub fn add_tokens_to_whitelist<
         Arg0: ProxyArg<MultiValueEncoded<Env::Api, TokenIdentifier<Env::Api>>>,

enshrine-esdt-safe/wasm-enshrine-esdt-safe-full/src/lib.rs

@@ -6,9 +6,9 @@
 
 // Init:                                 1
 // Upgrade:                              1
-// Endpoints:                           19
+// Endpoints:                           18
 // Async Callback:                       1
-// Total number of exported functions:  22
+// Total number of exported functions:  21
 
 #![no_std]
 
@@ -29,7 +29,6 @@ multiversx_sc_wasm_adapter::endpoints! {
         registerNewTokenID => register_new_token_id
         setMaxBridgedAmount => set_max_bridged_amount
         getMaxBridgedAmount => max_bridged_amount
-        endSetupPhase => end_setup_phase
         addTokensToWhitelist => add_tokens_to_whitelist
         removeTokensFromWhitelist => remove_tokens_from_whitelist
         addTokensToBlacklist => add_tokens_to_blacklist

enshrine-esdt-safe/wasm/src/lib.rs

@@ -6,9 +6,9 @@
 
 // Init:                                 1
 // Upgrade:                              1
-// Endpoints:                           19
+// Endpoints:                           18
 // Async Callback:                       1
-// Total number of exported functions:  22
+// Total number of exported functions:  21
 
 #![no_std]
 
@@ -29,7 +29,6 @@ multiversx_sc_wasm_adapter::endpoints! {
         registerNewTokenID => register_new_token_id
         setMaxBridgedAmount => set_max_bridged_amount
         getMaxBridgedAmount => max_bridged_amount
-        endSetupPhase => end_setup_phase
         addTokensToWhitelist => add_tokens_to_whitelist
         removeTokensFromWhitelist => remove_tokens_from_whitelist
         addTokensToBlacklist => add_tokens_to_blacklist

esdt-safe/interactor/src/interactor_main.rs

@@ -48,7 +48,6 @@ async fn main() {
         "registerToken" => interact.register_token().await,
         "setMaxBridgedAmount" => interact.set_max_bridged_amount().await,
         "getMaxBridgedAmount" => interact.max_bridged_amount().await,
-        "endSetupPhase" => interact.end_setup_phase().await,
         "addTokensToWhitelist" => interact.add_tokens_to_whitelist(b"").await,
         "removeTokensFromWhitelist" => interact.remove_tokens_from_whitelist().await,
         "addTokensToBlacklist" => interact.add_tokens_to_blacklist(b"").await,
@@ -502,22 +501,6 @@ impl ContractInteract {
         println!("Result: {result_value:?}");
     }
 
-    async fn end_setup_phase(&mut self) {
-        let response = self
-            .interactor
-            .tx()
-            .from(&self.wallet_address)
-            .to(self.state.current_address())
-            .gas(30_000_000u64)
-            .typed(EsdtSafeProxy)
-            .end_setup_phase()
-            .returns(ReturnsResultUnmanaged)
-            .run()
-            .await;
-
-        println!("Result: {response:?}");
-    }
-
     async fn add_tokens_to_whitelist(&mut self, token_id: &[u8]) {
         let tokens = MultiValueVec::from(vec![TokenIdentifier::from_esdt_bytes(token_id)]);
 

esdt-safe/interactor/src/proxy.rs

@@ -221,15 +221,6 @@ where
             .original_result()
     }
 
-    pub fn end_setup_phase(
-        self,
-    ) -> TxTypedCall<Env, From, To, NotPayable, Gas, ()> {
-        self.wrapped_tx
-            .payment(NotPayable)
-            .raw_call("endSetupPhase")
-            .original_result()
-    }
-
     /// Tokens in the whitelist can be transferred without fees 
     pub fn add_tokens_to_whitelist<
         Arg0: ProxyArg<MultiValueEncoded<Env::Api, TokenIdentifier<Env::Api>>>,

esdt-safe/wasm-esdt-safe-full/src/lib.rs

@@ -6,10 +6,10 @@
 
 // Init:                                 1
 // Upgrade:                              1
-// Endpoints:                           19
+// Endpoints:                           18
 // Async Callback (empty):               1
 // Promise callbacks:                    2
-// Total number of exported functions:  24
+// Total number of exported functions:  23
 
 #![no_std]
 
@@ -30,7 +30,6 @@ multiversx_sc_wasm_adapter::endpoints! {
         executeBridgeOps => execute_operations
         setMaxBridgedAmount => set_max_bridged_amount
         getMaxBridgedAmount => max_bridged_amount
-        endSetupPhase => end_setup_phase
         addTokensToWhitelist => add_tokens_to_whitelist
         removeTokensFromWhitelist => remove_tokens_from_whitelist
         addTokensToBlacklist => add_tokens_to_blacklist

esdt-safe/wasm/src/lib.rs

@@ -6,10 +6,10 @@
 
 // Init:                                 1
 // Upgrade:                              1
-// Endpoints:                           19
+// Endpoints:                           18
 // Async Callback (empty):               1
 // Promise callbacks:                    2
-// Total number of exported functions:  24
+// Total number of exported functions:  23
 
 #![no_std]
 
@@ -30,7 +30,6 @@ multiversx_sc_wasm_adapter::endpoints! {
         executeBridgeOps => execute_operations
         setMaxBridgedAmount => set_max_bridged_amount
         getMaxBridgedAmount => max_bridged_amount
-        endSetupPhase => end_setup_phase
         addTokensToWhitelist => add_tokens_to_whitelist
         removeTokensFromWhitelist => remove_tokens_from_whitelist
         addTokensToBlacklist => add_tokens_to_blacklist

sovereign-forge/Cargo.toml

@@ -25,3 +25,6 @@ path = "../common/proxies"
 
 [dependencies.transaction]
 path = "../common/transaction"
+
+[dependencies.setup-phase]
+path = "../common/setup-phase"

sovereign-forge/src/lib.rs

@@ -8,7 +8,10 @@ mod phases;
 
 #[multiversx_sc::contract]
 pub trait SovereignForge:
-    phases::PhasesModule + common::storage::StorageModule + common::utils::UtilsModule
+    phases::PhasesModule
+    + common::storage::StorageModule
+    + common::utils::UtilsModule
+    + setup_phase::SetupPhaseModule
 {
     #[init]
     fn init(&self, deploy_cost: BigUint) {

sovereign-forge/src/phases.rs

@@ -15,8 +15,34 @@ use crate::{
     err_msg,
 };
 
+const NUMBER_OF_SHARDS: u32 = 3;
+
 #[multiversx_sc::module]
-pub trait PhasesModule: common::utils::UtilsModule + common::storage::StorageModule {
+pub trait PhasesModule:
+    common::utils::UtilsModule + common::storage::StorageModule + setup_phase::SetupPhaseModule
+{
+    #[endpoint(completeSetupPhase)]
+    fn complete_setup_phase(&self) {
+        if !self.is_setup_phase_complete() {
+            return;
+        }
+
+        for shard_id in 1..=NUMBER_OF_SHARDS {
+            require!(
+                !self.chain_factories(shard_id).is_empty(),
+                "There is no Chain-Factory contract assigned for shard {}",
+                shard_id
+            );
+            require!(
+                !self.token_handlers(shard_id).is_empty(),
+                "There is no Token-Handler contract assigned for shard {}",
+                shard_id
+            );
+        }
+
+        self.setup_phase_complete().set(true);
+    }
+
     #[payable("EGLD")]
     #[endpoint(deployPhaseOne)]
     fn deploy_phase_one(
@@ -26,13 +52,15 @@ pub trait PhasesModule: common::utils::UtilsModule + common::storage::StorageMod
         min_stake: BigUint,
         additional_stake_required: MultiValueEncoded<StakeMultiArg<Self::Api>>,
     ) {
-        let call_value = self.call_value().egld_value();
-        self.require_correct_deploy_cost(call_value.deref());
-
         let blockchain_api = self.blockchain();
         let caller = blockchain_api.get_caller();
         let caller_shard_id = blockchain_api.get_shard_of_address(&caller);
 
+        self.require_setup_complete(caller_shard_id);
+
+        let call_value = self.call_value().egld_value();
+        self.require_correct_deploy_cost(call_value.deref());
+
         let chain_id = self.generate_chain_id();
 
         let chain_factories_mapper = self.chain_factories(caller_shard_id);