Getting listed regions by account

RegionX-Labs · Feb 24, 2024 · 581e47d · 581e47d
1 parent c21ac8c
commit 581e47d
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/contracts/coretime_market/src/lib.rs b/contracts/coretime_market/src/lib.rs
@@ -131,8 +131,19 @@ pub mod coretime_market {
 		}
 
 		#[ink(message)]
-		pub fn listed_regions(&self) -> Vec<RawRegionId> {
-			self.listed_regions.clone()
+		pub fn listed_regions(&self, maybe_who: Option<AccountId>) -> Vec<RawRegionId> {
+			if let Some(who) = maybe_who {
+				self.listed_regions
+					.clone()
+					.into_iter()
+					.filter(|region_id| {
+						let Some(listing) = self.listings.get(region_id) else { return false };
+						listing.seller == who
+					})
+					.collect()
+			} else {
+				self.listed_regions.clone()
+			}
 		}
 
 		#[ink(message)]