AccountStore in-memory state initialization and synchronization

parachain/pallets/omni-account/src/lib.rs

@@ -132,13 +132,25 @@ pub mod pallet {
 	#[pallet::generate_deposit(pub(super) fn deposit_event)]
 	pub enum Event<T: Config> {
 		/// An account store is created
-		AccountStoreCreated { who: T::AccountId },
+		AccountStoreCreated { who: T::AccountId, account_store: MemberAccounts<T> },
 		/// Some member account is added
-		AccountAdded { who: T::AccountId, member_account_hash: H256 },
+		AccountAdded {
+			who: T::AccountId,
+			member_account_hash: H256,
+			account_store: MemberAccounts<T>,
+		},
 		/// Some member accounts are removed
-		AccountRemoved { who: T::AccountId, member_account_hashes: Vec<H256> },
+		AccountRemoved {
+			who: T::AccountId,
+			member_account_hashes: Vec<H256>,
+			account_store: MemberAccounts<T>,
+		},
 		/// Some member account is made public
-		AccountMadePublic { who: T::AccountId, member_account_hash: H256 },
+		AccountMadePublic {
+			who: T::AccountId,
+			member_account_hash: H256,
+			account_store: MemberAccounts<T>,
+		},
 		/// An account store is updated
 		AccountStoreUpdated { who: T::AccountId },
 		/// Some call is dispatched as omni-account origin
@@ -244,7 +256,11 @@ pub mod pallet {
 			MemberAccountHash::<T>::insert(hash, who.clone());
 			AccountStore::<T>::insert(who.clone(), member_accounts.clone());
 
-			Self::deposit_event(Event::AccountAdded { who, member_account_hash: hash });
+			Self::deposit_event(Event::AccountAdded {
+				who,
+				member_account_hash: hash,
+				account_store: member_accounts,
+			});
 
 			Ok(())
 		}
@@ -275,10 +291,14 @@ pub mod pallet {
 			if member_accounts.is_empty() {
 				AccountStore::<T>::remove(&who);
 			} else {
-				AccountStore::<T>::insert(who.clone(), member_accounts);
+				AccountStore::<T>::insert(who.clone(), member_accounts.clone());
 			}
 
-			Self::deposit_event(Event::AccountRemoved { who, member_account_hashes });
+			Self::deposit_event(Event::AccountRemoved {
+				who,
+				member_account_hashes,
+				account_store: member_accounts,
+			});
 
 			Ok(())
 		}
@@ -299,9 +319,13 @@ pub mod pallet {
 				.ok_or(Error::<T>::AccountNotFound)?;
 			*m = member_account.into();
 
-			AccountStore::<T>::insert(who.clone(), member_accounts);
+			AccountStore::<T>::insert(who.clone(), member_accounts.clone());
 
-			Self::deposit_event(Event::AccountMadePublic { who, member_account_hash: hash });
+			Self::deposit_event(Event::AccountMadePublic {
+				who,
+				member_account_hash: hash,
+				account_store: member_accounts,
+			});
 
 			Ok(())
 		}
@@ -385,7 +409,10 @@ pub mod pallet {
 			MemberAccountHash::<T>::insert(hash, omni_account.clone());
 			AccountStore::<T>::insert(omni_account.clone(), member_accounts.clone());
 
-			Self::deposit_event(Event::AccountStoreCreated { who: omni_account });
+			Self::deposit_event(Event::AccountStoreCreated {
+				who: omni_account,
+				account_store: member_accounts.clone(),
+			});
 
 			Ok(member_accounts)
 		}

parachain/pallets/omni-account/src/tests.rs

@@ -57,7 +57,16 @@ fn create_account_store_works() {
 			alice().identity,
 		));
 
-		System::assert_last_event(Event::AccountStoreCreated { who: alice().omni_account }.into());
+		let member_accounts: MemberAccounts<TestRuntime> =
+			vec![public_member_account(alice())].try_into().unwrap();
+
+		System::assert_last_event(
+			Event::AccountStoreCreated {
+				who: alice().omni_account,
+				account_store: member_accounts,
+			}
+			.into(),
+		);
 
 		// create it the second time will fail
 		assert_noop!(
@@ -111,6 +120,7 @@ fn add_account_works() {
 			Event::AccountAdded {
 				who: alice().omni_account.clone(),
 				member_account_hash: bob.hash(),
+				account_store: expected_member_accounts.clone(),
 			}
 			.into(),
 		);
@@ -126,20 +136,29 @@ fn add_account_works() {
 			alice().identity.hash(),
 			call
 		));
+		let expected_member_accounts: MemberAccounts<TestRuntime> =
+			BoundedVec::truncate_from(vec![
+				public_member_account(alice()),
+				bob.clone(),
+				charlie.clone(),
+			]);
 
 		System::assert_has_event(
-			Event::AccountAdded { who: alice().omni_account, member_account_hash: charlie.hash() }
-				.into(),
+			Event::AccountAdded {
+				who: alice().omni_account,
+				member_account_hash: charlie.hash(),
+				account_store: expected_member_accounts.clone(),
+			}
+			.into(),
 		);
 
-		let expected_member_accounts: MemberAccounts<TestRuntime> =
-			vec![public_member_account(alice()), bob.clone(), charlie.clone()]
-				.try_into()
-				.unwrap();
-
-		assert_eq!(
-			AccountStore::<TestRuntime>::get(alice().omni_account).unwrap(),
-			expected_member_accounts
+		System::assert_has_event(
+			Event::AccountAdded {
+				who: alice().omni_account,
+				member_account_hash: charlie.hash(),
+				account_store: expected_member_accounts.clone(),
+			}
+			.into(),
 		);
 
 		assert!(MemberAccountHash::<TestRuntime>::contains_key(bob.hash()));
@@ -324,17 +343,18 @@ fn remove_account_works() {
 			.into(),
 		);
 
+		let expected_member_accounts: MemberAccounts<TestRuntime> =
+			BoundedVec::truncate_from(vec![public_member_account(alice())]);
+
 		System::assert_has_event(
 			Event::AccountRemoved {
-				who: alice().omni_account,
+				who: alice().omni_account.clone(),
 				member_account_hashes: vec![bob.hash()],
+				account_store: expected_member_accounts.clone(),
 			}
 			.into(),
 		);
 
-		let expected_member_accounts: MemberAccounts<TestRuntime> =
-			vec![public_member_account(alice())].try_into().unwrap();
-
 		assert_eq!(
 			AccountStore::<TestRuntime>::get(alice().omni_account).unwrap(),
 			expected_member_accounts
@@ -432,16 +452,18 @@ fn publicize_account_works() {
 			.into(),
 		);
 
+		let expected_member_accounts: MemberAccounts<TestRuntime> =
+			BoundedVec::truncate_from(vec![public_member_account(alice()), public_bob]);
+
 		System::assert_has_event(
 			Event::AccountMadePublic {
-				who: alice().omni_account,
-				member_account_hash: public_bob.hash(),
+				who: alice().omni_account.clone(),
+				member_account_hash: bob().identity.hash(),
+				account_store: expected_member_accounts.clone(),
 			}
 			.into(),
 		);
 
-		let expected_member_accounts: MemberAccounts<TestRuntime> =
-			vec![public_member_account(alice()), public_bob].try_into().unwrap();
 		assert_eq!(
 			AccountStore::<TestRuntime>::get(alice().omni_account).unwrap(),
 			expected_member_accounts

tee-worker/Cargo.toml

@@ -301,7 +301,7 @@ lc-stf-task-sender = { path = "identity/litentry/core/stf-task/sender", default-
 lc-stf-task-receiver = { path = "identity/litentry/core/stf-task/receiver", default-features = false }
 lc-vc-task-sender = { path = "identity/litentry/core/vc-task/sender", default-features = false }
 lc-vc-task-receiver = { path = "identity/litentry/core/vc-task/receiver", default-features = false }
-lc-omni-account = { path = "identity/app-libs/omni-account", default-features = false }
+lc-omni-account = { path = "identity/litentry/core/omni-account", default-features = false }
 lc-native-task-sender = { path = "identity/litentry/core/native-task/sender", default-features = false }
 lc-native-task-receiver = { path = "identity/litentry/core/native-task/receiver", default-features = false }
 

tee-worker/bitacross/app-libs/parentchain-interface/src/integritee/event_filter.rs

@@ -117,4 +117,20 @@ impl FilterEvents for FilterableEvents {
 	fn get_btc_wallet_generated_events(&self) -> Result<Vec<BtcWalletGenerated>, Self::Error> {
 		self.filter()
 	}
+
+	fn get_account_store_created_events(&self) -> Result<Vec<AccountStoreCreated>, Self::Error> {
+		self.filter()
+	}
+
+	fn get_account_added_events(&self) -> Result<Vec<AccountAdded>, Self::Error> {
+		self.filter()
+	}
+
+	fn get_account_removed_events(&self) -> Result<Vec<AccountRemoved>, Self::Error> {
+		self.filter()
+	}
+
+	fn get_account_made_public_events(&self) -> Result<Vec<AccountMadePublic>, Self::Error> {
+		self.filter()
+	}
 }

tee-worker/bitacross/app-libs/parentchain-interface/src/integritee/event_handler.rs

@@ -32,6 +32,7 @@ use itp_types::{
 use litentry_primitives::{Address32, Identity};
 use log::*;
 use sp_core::{blake2_256, H256};
+use sp_runtime::traits::{Block as ParentchainBlock, Header as ParentchainHeader};
 use sp_std::vec::Vec;
 use std::string::ToString;
 
@@ -129,11 +130,15 @@ where
 {
 	type Output = Vec<H256>;
 
-	fn handle_events(
+	fn handle_events<Block>(
 		&self,
 		executor: &Executor,
 		events: impl FilterEvents,
-	) -> Result<Vec<H256>, Error> {
+		_block_number: <<Block as ParentchainBlock>::Header as ParentchainHeader>::Number,
+	) -> Result<Vec<H256>, Error>
+	where
+		Block: ParentchainBlock,
+	{
 		let mut handled_events: Vec<H256> = Vec::new();
 
 		if let Ok(events) = events.get_relayer_added_events() {

tee-worker/bitacross/app-libs/parentchain-interface/src/target_a/event_filter.rs

@@ -115,4 +115,20 @@ impl FilterEvents for FilterableEvents {
 	fn get_btc_wallet_generated_events(&self) -> Result<Vec<BtcWalletGenerated>, Self::Error> {
 		self.filter()
 	}
+
+	fn get_account_store_created_events(&self) -> Result<Vec<AccountStoreCreated>, Self::Error> {
+		self.filter()
+	}
+
+	fn get_account_added_events(&self) -> Result<Vec<AccountAdded>, Self::Error> {
+		self.filter()
+	}
+
+	fn get_account_removed_events(&self) -> Result<Vec<AccountRemoved>, Self::Error> {
+		self.filter()
+	}
+
+	fn get_account_made_public_events(&self) -> Result<Vec<AccountMadePublic>, Self::Error> {
+		self.filter()
+	}
 }

tee-worker/bitacross/app-libs/parentchain-interface/src/target_a/event_handler.rs

@@ -28,6 +28,7 @@ use itp_types::{
 	H256,
 };
 use log::*;
+use sp_runtime::traits::{Block as ParentchainBlock, Header as ParentchainHeader};
 use std::vec::Vec;
 
 pub struct ParentchainEventHandler {}
@@ -52,11 +53,15 @@ where
 {
 	type Output = Vec<H256>;
 
-	fn handle_events(
+	fn handle_events<Block>(
 		&self,
 		_executor: &Executor,
 		_events: impl FilterEvents,
-	) -> Result<Vec<H256>, Error> {
+		_block_number: <<Block as ParentchainBlock>::Header as ParentchainHeader>::Number,
+	) -> Result<Vec<H256>, Error>
+	where
+		Block: ParentchainBlock,
+	{
 		debug!("not handling any events for target a");
 		Ok(Vec::new())
 	}

tee-worker/bitacross/app-libs/parentchain-interface/src/target_b/event_filter.rs

@@ -115,4 +115,20 @@ impl FilterEvents for FilterableEvents {
 	fn get_btc_wallet_generated_events(&self) -> Result<Vec<BtcWalletGenerated>, Self::Error> {
 		self.filter()
 	}
+
+	fn get_account_store_created_events(&self) -> Result<Vec<AccountStoreCreated>, Self::Error> {
+		self.filter()
+	}
+
+	fn get_account_added_events(&self) -> Result<Vec<AccountAdded>, Self::Error> {
+		self.filter()
+	}
+
+	fn get_account_removed_events(&self) -> Result<Vec<AccountRemoved>, Self::Error> {
+		self.filter()
+	}
+
+	fn get_account_made_public_events(&self) -> Result<Vec<AccountMadePublic>, Self::Error> {
+		self.filter()
+	}
 }

tee-worker/bitacross/app-libs/parentchain-interface/src/target_b/event_handler.rs

@@ -28,6 +28,7 @@ use itp_types::{
 	H256,
 };
 use log::*;
+use sp_runtime::traits::{Block as ParentchainBlock, Header as ParentchainHeader};
 use std::vec::Vec;
 
 pub struct ParentchainEventHandler {}
@@ -52,11 +53,15 @@ where
 {
 	type Output = Vec<H256>;
 
-	fn handle_events(
+	fn handle_events<Block>(
 		&self,
 		_executor: &Executor,
 		_events: impl FilterEvents,
-	) -> Result<Vec<H256>, Error> {
+		_block_number: <<Block as ParentchainBlock>::Header as ParentchainHeader>::Number,
+	) -> Result<Vec<H256>, Error>
+	where
+		Block: ParentchainBlock,
+	{
 		debug!("not handling any events for target B");
 		Ok(Vec::new())
 	}

tee-worker/bitacross/core/parentchain/indirect-calls-executor/src/executor.rs

@@ -195,7 +195,11 @@ impl<
 			})?
 			.ok_or_else(|| Error::Other("Could not create events from metadata".into()))?;
 
-		let processed_events = self.parentchain_event_handler.handle_events(self, events)?;
+		let processed_events = self.parentchain_event_handler.handle_events::<ParentchainBlock>(
+			self,
+			events,
+			block_number,
+		)?;
 
 		if self.parentchain_id == ParentchainId::Litentry {
 			// Include a processed parentchain block confirmation for each block.