git.delta.rocks / unique-network / refs/commits / e0035410299d

--- a/pallets/common/src/lib.rs
+++ b/pallets/common/src/lib.rs
@@ -155,6 +155,11 @@
 }
 
 impl<T: Config> CollectionHandle<T> {
+	/// Get the mode of the collection: NFT/FT/RFT.
+	pub fn mode(&self) -> CollectionMode {
+		self.mode
+	}
+
 	/// Same as [CollectionHandle::new] but with an explicit gas limit.
 	pub fn new_with_gas_limit(id: CollectionId, gas_limit: u64) -> Option<Self> {
 		<CollectionById<T>>::get(id).map(|collection| Self {
@@ -1872,6 +1877,9 @@
 /// It wraps methods in Fungible, Nonfungible and Refungible pallets
 /// and adds weight info.
 pub trait CommonCollectionOperations<T: Config> {
+	/// Get the mode of the collection: NFT/FT/RFT.
+	fn mode(&self) -> CollectionMode;
+
 	/// Create token.
 	///
 	/// * `sender` - The user who mint the token and pays for the transaction.

--- a/pallets/fungible/src/common.rs
+++ b/pallets/fungible/src/common.rs
@@ -125,6 +125,10 @@
 /// Implementation of `CommonCollectionOperations` for `FungibleHandle`. It wraps FungibleHandle Pallete
 /// methods and adds weight info.
 impl<T: Config> CommonCollectionOperations<T> for FungibleHandle<T> {
+	fn mode(&self) -> up_data_structs::CollectionMode {
+		self.0.mode()
+	}
+
 	fn create_item(
 		&self,
 		sender: T::CrossAccountId,

--- a/pallets/nonfungible/src/common.rs
+++ b/pallets/nonfungible/src/common.rs
@@ -152,6 +152,10 @@
 /// Implementation of `CommonCollectionOperations` for `NonfungibleHandle`. It wraps Nonfungible Pallete
 /// methods and adds weight info.
 impl<T: Config> CommonCollectionOperations<T> for NonfungibleHandle<T> {
+	fn mode(&self) -> up_data_structs::CollectionMode {
+		self.0.mode()
+	}
+
 	fn create_item(
 		&self,
 		sender: T::CrossAccountId,

--- a/pallets/proxy-rmrk-core/src/lib.rs
+++ b/pallets/proxy-rmrk-core/src/lib.rs
@@ -741,7 +741,8 @@
 						Some((collection_id, nft_id)),
 						&target_nft_budget,
 					)
-					.map_err(Self::map_unique_err_to_proxy)?;
+					.map_err(Self::map_unique_err_to_proxy)?
+					.ok_or::<DispatchError>(<Error<T>>::NoPermission.into())?;
 
 					approval_required = cross_sender != target_nft_owner;
 
@@ -989,7 +990,8 @@
 
 			let nft_owner =
 				<PalletStructure<T>>::find_topmost_owner(collection_id, nft_id, &budget)
-					.map_err(|_| <Error<T>>::ResourceDoesntExist)?;
+					.map_err(|_| <Error<T>>::ResourceDoesntExist)?
+					.ok_or::<DispatchError>(<Error<T>>::NoPermission.into())?;
 
 			Self::try_mutate_resource_info(collection_id, nft_id, resource_id, |res| {
 				ensure!(res.pending, <Error<T>>::ResourceNotPending);
@@ -1044,7 +1046,8 @@
 
 			let nft_owner =
 				<PalletStructure<T>>::find_topmost_owner(collection_id, nft_id, &budget)
-					.map_err(|_| <Error<T>>::ResourceDoesntExist)?;
+					.map_err(|_| <Error<T>>::ResourceDoesntExist)?
+					.ok_or::<DispatchError>(<Error<T>>::NoPermission.into())?;
 
 			ensure!(cross_sender == nft_owner, <Error<T>>::NoPermission);
 
@@ -1666,7 +1669,8 @@
 		let budget = budget::Value::new(NESTING_BUDGET);
 
 		let nft_owner = <PalletStructure<T>>::find_topmost_owner(collection_id, nft_id, &budget)
-			.map_err(Self::map_unique_err_to_proxy)?;
+			.map_err(Self::map_unique_err_to_proxy)?
+			.ok_or::<DispatchError>(<Error<T>>::NoPermission.into())?;
 
 		let pending = sender != nft_owner;
 
@@ -1720,7 +1724,8 @@
 
 		let budget = up_data_structs::budget::Value::new(NESTING_BUDGET);
 		let topmost_owner =
-			<PalletStructure<T>>::find_topmost_owner(collection_id, nft_id, &budget)?;
+			<PalletStructure<T>>::find_topmost_owner(collection_id, nft_id, &budget)?
+				.ok_or::<DispatchError>(<Error<T>>::NoPermission.into())?;
 
 		let sender = T::CrossAccountId::from_sub(sender);
 		if topmost_owner == sender {

--- a/pallets/refungible/src/common.rs
+++ b/pallets/refungible/src/common.rs
@@ -186,6 +186,10 @@
 /// Implementation of `CommonCollectionOperations` for `RefungibleHandle`. It wraps Refungible Pallete
 /// methods and adds weight info.
 impl<T: Config> CommonCollectionOperations<T> for RefungibleHandle<T> {
+	fn mode(&self) -> up_data_structs::CollectionMode {
+		self.0.mode()
+	}
+
 	fn create_item(
 		&self,
 		sender: T::CrossAccountId,

61use frame_support::fail;61use frame_support::fail;
62pub use pallet::*;62pub use pallet::*;
63use pallet_common::{dispatch::CollectionDispatch, CollectionHandle};63use pallet_common::{dispatch::CollectionDispatch, CollectionHandle};
64use up_data_structs::CollectionMode;
64use up_data_structs::{CollectionId, TokenId, mapping::TokenAddressMapping, budget::Budget};65use up_data_structs::{CollectionId, TokenId, mapping::TokenAddressMapping, budget::Budget};
6566
66#[cfg(feature = "runtime-benchmarks")]67#[cfg(feature = "runtime-benchmarks")]
135	User(CrossAccountId),136	User(CrossAccountId),
136	/// Could not find the token provided as the owner.137	/// Could not find the token provided as the owner.
137	TokenNotFound,138	TokenNotFound,
139	/// Nested token has multiple owners.
140	MultipleOwners,
138	/// Token owner is another token (still, the target token may not exist).141	/// Token owner is another token (still, the target token may not exist).
139	Token(CollectionId, TokenId),142	Token(CollectionId, TokenId),
140}143}
163				Some((collection, token)) => Parent::Token(collection, token),166				Some((collection, token)) => Parent::Token(collection, token),
164				None => Parent::User(owner),167				None => Parent::User(owner),
165			},168			},
169			None if handle.mode() == CollectionMode::ReFungible => handle
170				.total_pieces(token)
171				.map(|_| Parent::MultipleOwners)
172				.unwrap_or(Parent::TokenNotFound),
166			None => Parent::TokenNotFound,173			None => Parent::TokenNotFound,
167		})174		})
168	}175	}
202	/// Try to dereference address, until finding top level owner209	/// Try to dereference address, until finding top level owner
203	///210	///
204	/// May return token address if parent token not yet exists211	/// May return token address if parent token not yet exists
212	///
213	/// Returns `None` if the token has multiple owners.
205	///214	///
206	/// - `budget`: Limit for searching parents in depth.215	/// - `budget`: Limit for searching parents in depth.
207	pub fn find_topmost_owner(216	pub fn find_topmost_owner(
208		collection: CollectionId,217		collection: CollectionId,
209		token: TokenId,218		token: TokenId,
210		budget: &dyn Budget,219		budget: &dyn Budget,
211	) -> Result<T::CrossAccountId, DispatchError> {220	) -> Result<Option<T::CrossAccountId>, DispatchError> {
212		let owner = Self::parent_chain(collection, token)221		let owner = Self::parent_chain(collection, token)
213			.take_while(|_| budget.consume())222			.take_while(|_| budget.consume())
214			.find(|p| matches!(p, Ok(Parent::User(_) | Parent::TokenNotFound)))223			.find(|p| {
224				matches!(
225					p,
226					Ok(Parent::User(_) | Parent::TokenNotFound | Parent::MultipleOwners)
227				)
228			})
215			.ok_or(<Error<T>>::DepthLimit)??;229			.ok_or(<Error<T>>::DepthLimit)??;
216230
217		Ok(match owner {231		Ok(match owner {
218			Parent::User(v) => v,232			Parent::User(v) => Some(v),
233			Parent::MultipleOwners => None,
219			_ => fail!(<Error<T>>::TokenNotFound),234			_ => fail!(<Error<T>>::TokenNotFound),
220		})235		})
221	}236	}
222237
223	/// Find the topmost parent and check that assigning `for_nest` token as a child for238	/// Find the topmost parent and check that assigning `for_nest` token as a child for
224	/// `token` wouldn't create a cycle.239	/// `token` wouldn't create a cycle.
240	///
241	/// Returns `None` if the token has multiple owners.
225	///242	///
226	/// - `budget`: Limit for searching parents in depth.243	/// - `budget`: Limit for searching parents in depth.
227	pub fn get_checked_topmost_owner(244	pub fn get_checked_topmost_owner(
228		collection: CollectionId,245		collection: CollectionId,
229		token: TokenId,246		token: TokenId,
230		for_nest: Option<(CollectionId, TokenId)>,247		for_nest: Option<(CollectionId, TokenId)>,
231		budget: &dyn Budget,248		budget: &dyn Budget,
232	) -> Result<T::CrossAccountId, DispatchError> {249	) -> Result<Option<T::CrossAccountId>, DispatchError> {
233		// Tried to nest token in itself250		// Tried to nest token in itself
234		if Some((collection, token)) == for_nest {251		if Some((collection, token)) == for_nest {
235			return Err(<Error<T>>::OuroborosDetected.into());252			return Err(<Error<T>>::OuroborosDetected.into());
242					return Err(<Error<T>>::OuroborosDetected.into())259					return Err(<Error<T>>::OuroborosDetected.into())
243				}260				}
244				// Token is owned by other user261				// Token is owned by other user
245				Parent::User(user) => return Ok(user),262				Parent::User(user) => return Ok(Some(user)),
246				Parent::TokenNotFound => return Err(<Error<T>>::TokenNotFound.into()),263				Parent::TokenNotFound => return Err(<Error<T>>::TokenNotFound.into()),
264				Parent::MultipleOwners => return Ok(None),
247				// Continue parent chain265				// Continue parent chain
248				Parent::Token(_, _) => {}266				Parent::Token(_, _) => {}
249			}267			}
284		budget: &dyn Budget,302		budget: &dyn Budget,
285	) -> Result<bool, DispatchError> {303	) -> Result<bool, DispatchError> {
286		let target_parent = match T::CrossTokenAddressMapping::address_to_token(&user) {304		let target_parent = match T::CrossTokenAddressMapping::address_to_token(&user) {
287			Some((collection, token)) => Self::find_topmost_owner(collection, token, budget)?,305			Some((collection, token)) => match Self::find_topmost_owner(collection, token, budget)?
306			{
307				Some(topmost_owner) => topmost_owner,
308				None => return Ok(false),
309			},
288			None => user,310			None => user,
289		};311		};
290312
291		Self::get_checked_topmost_owner(collection, token, for_nest, budget)313		Self::get_checked_topmost_owner(collection, token, for_nest, budget).map(|indirect_owner| {
292			.map(|indirect_owner| indirect_owner == target_parent)314			indirect_owner.map_or(false, |indirect_owner| indirect_owner == target_parent)
315		})
293	}316	}
294317
295	/// Checks that `under` is valid token and that `token_id` could be nested under it318	/// Checks that `under` is valid token and that `token_id` could be nested under it

--- a/primitives/data-structs/src/lib.rs
+++ b/primitives/data-structs/src/lib.rs
@@ -252,7 +252,7 @@
 /// Collection can represent various types of tokens.
 /// Each collection can contain only one type of tokens at a time.
 /// This type helps to understand which tokens the collection contains.
-#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq, TypeInfo, MaxEncodedLen)]
+#[derive(Encode, Decode, Eq, Debug, Clone, Copy, PartialEq, TypeInfo, MaxEncodedLen)]
 #[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]
 pub enum CollectionMode {
 	/// Non fungible tokens.

--- a/runtime/common/runtime_apis.rs
+++ b/runtime/common/runtime_apis.rs
@@ -83,7 +83,7 @@
                 fn topmost_token_owner(collection: CollectionId, token: TokenId) -> Result<Option<CrossAccountId>, DispatchError> {
                     let budget = up_data_structs::budget::Value::new(10);
 
-                    Ok(Some(<pallet_structure::Pallet<Runtime>>::find_topmost_owner(collection, token, &budget)?))
+                    Ok(<pallet_structure::Pallet<Runtime>>::find_topmost_owner(collection, token, &budget)?)
                 }
                 fn token_children(collection: CollectionId, token: TokenId) -> Result<Vec<TokenChild>, DispatchError> {
                     Ok(<pallet_nonfungible::Pallet<Runtime>>::token_children_ids(collection, token))