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

difftreelog

Apply suggestions from code review

Gregory Simonov2022-12-06parent: #01e5c45.patch.diff
in: master

3 files changed

modifiedpallets/common/src/lib.rsdiffbeforeafterboth
--- a/pallets/common/src/lib.rs
+++ b/pallets/common/src/lib.rs
@@ -472,7 +472,7 @@
 			u128,
 		),
 
-		/// Amount pieces of token owned by `sender` was approved for `spender`.
+		/// A `sender` approves operations on all owned tokens for `spender`.
 		ApprovedForAll(
 			/// Id of collection to which item is belong.
 			CollectionId,
@@ -480,7 +480,7 @@
 			T::CrossAccountId,
 			/// Id for which operator status was granted or rewoked.
 			T::CrossAccountId,
-			/// Is operator status was granted or rewoked.
+			/// Is operator status granted or revoked?
 			bool,
 		),
 
@@ -1847,7 +1847,7 @@
 	/// An operator is allowed to transfer all tokens of the sender on their behalf.
 	/// * `owner` - Token owner
 	/// * `operator` - Operator
-	/// * `approve` - Is operator enabled or disabled
+	/// * `approve` - Should operator status be granted or revoked?
 	fn set_approval_for_all(
 		&self,
 		owner: T::CrossAccountId,
@@ -1855,7 +1855,7 @@
 		approve: bool,
 	) -> DispatchResultWithPostInfo;
 
-	/// Tells whether an operator is approved by a given owner.
+	/// Tells whether the given `owner` approves the `operator`.
 	fn is_approved_for_all(&self, owner: T::CrossAccountId, operator: T::CrossAccountId) -> bool;
 }
 
modifiedpallets/nonfungible/src/erc.rsdiffbeforeafterboth
--- a/pallets/nonfungible/src/erc.rs
+++ b/pallets/nonfungible/src/erc.rs
@@ -470,7 +470,7 @@
 	}
 
 	/// @notice Sets or unsets the approval of a given operator.
-	///  An operator is allowed to transfer all tokens of the sender on their behalf.
+	/// The `operator` is allowed to transfer all tokens of the `caller` on their behalf.
 	/// @param operator Operator
 	/// @param approved Is operator enabled or disabled
 	#[weight(<SelfWeightOf<T>>::set_approval_for_all())]
modifiedpallets/nonfungible/src/lib.rsdiffbeforeafterboth
before · pallets/nonfungible/src/lib.rs
1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617//! # Nonfungible Pallet18//!19//! The Nonfungible pallet provides functionality for handling nonfungible collections and tokens.20//!21//! - [`Config`]22//! - [`NonfungibleHandle`]23//! - [`Pallet`]24//! - [`CommonWeights`](common::CommonWeights)25//!26//! ## Overview27//!28//! The Nonfungible pallet provides functions for:29//!30//! - NFT collection creation and removal31//! - Minting and burning of NFT tokens32//! - Retrieving account balances33//! - Transfering NFT tokens34//! - Setting and checking allowance for NFT tokens35//! - Setting properties and permissions for NFT collections and tokens36//! - Nesting and unnesting tokens37//!38//! ### Terminology39//!40//! - **NFT token:** Non fungible token.41//!42//! - **NFT Collection:** A collection of NFT tokens. All NFT tokens are part of a collection.43//!   Each collection can define it's own properties, properties for it's tokens and set of permissions.44//!45//! - **Balance:** Number of NFT tokens owned by an account46//!47//! - **Allowance:** NFT tokens owned by one account that another account is allowed to make operations on48//!49//! - **Burning:** The process of “deleting” a token from a collection and from50//!   an account balance of the owner.51//!52//! - **Nesting:** Setting up parent-child relationship between tokens. Nested tokens are inhereting53//!   owner from their parent. There could be multiple levels of nesting. Token couldn't be nested in54//!   it's child token i.e. parent-child relationship graph shouldn't have cycles.55//!56//! - **Properties:** Key-Values pairs. Token properties are attached to a token. Collection properties are57//!   attached to a collection. Set of permissions could be defined for each property.58//!59//! ### Implementations60//!61//! The Nonfungible pallet provides implementations for the following traits. If these traits provide62//! the functionality that you need, then you can avoid coupling with the Nonfungible pallet.63//!64//! - [`CommonWeightInfo`](pallet_common::CommonWeightInfo): Functions for retrieval of transaction weight65//! - [`CommonCollectionOperations`](pallet_common::CommonCollectionOperations): Functions for dealing66//!   with collections67//!68//! ## Interface69//!70//! ### Dispatchable Functions71//!72//! - `init_collection` - Create NFT collection. NFT collection can be configured to allow or deny access for73//!   some accounts.74//! - `destroy_collection` - Destroy exising NFT collection. There should be no tokens in the collection.75//! - `burn` - Burn NFT token owned by account.76//! - `transfer` - Transfer NFT token. Transfers should be enabled for NFT collection.77//!   Nests the NFT token if it is sent to another token.78//! - `create_item` - Mint NFT token in collection. Sender should have permission to mint tokens.79//! - `set_allowance` - Set allowance for another account.80//! - `set_token_property` - Set token property value.81//! - `delete_token_property` - Remove property from the token.82//! - `set_collection_properties` - Set collection properties.83//! - `delete_collection_properties` - Remove properties from the collection.84//! - `set_property_permission` - Set collection property permission.85//! - `set_token_property_permissions` - Set token property permissions.86//!87//! ## Assumptions88//!89//! * To perform operations on tokens sender should be in collection's allow list if collection access mode is `AllowList`.9091#![cfg_attr(not(feature = "std"), no_std)]9293use erc::ERC721Events;94use evm_coder::ToLog;95use frame_support::{96	BoundedVec, ensure, fail, transactional,97	storage::with_transaction,98	pallet_prelude::DispatchResultWithPostInfo,99	pallet_prelude::Weight,100	dispatch::{PostDispatchInfo, Pays},101};102use up_data_structs::{103	AccessMode, CollectionId, CollectionFlags, CustomDataLimit, TokenId, CreateCollectionData,104	CreateNftExData, mapping::TokenAddressMapping, budget::Budget, Property, PropertyPermission,105	PropertyKey, PropertyValue, PropertyKeyPermission, Properties, PropertyScope, TrySetProperty,106	TokenChild, AuxPropertyValue,107};108use pallet_evm::{account::CrossAccountId, Pallet as PalletEvm};109use pallet_common::{110	Error as CommonError, Pallet as PalletCommon, Event as CommonEvent, CollectionHandle,111	eth::collection_id_to_address,112};113use pallet_structure::{Pallet as PalletStructure, Error as StructureError};114use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};115use sp_core::H160;116use sp_runtime::{ArithmeticError, DispatchError, DispatchResult, TransactionOutcome};117use sp_std::{vec::Vec, vec, collections::btree_map::BTreeMap};118use core::ops::Deref;119use codec::{Encode, Decode, MaxEncodedLen};120use scale_info::TypeInfo;121122pub use pallet::*;123use weights::WeightInfo;124#[cfg(feature = "runtime-benchmarks")]125pub mod benchmarking;126pub mod common;127pub mod erc;128pub mod weights;129130pub type CreateItemData<T> = CreateNftExData<<T as pallet_evm::Config>::CrossAccountId>;131pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;132133/// Token data, stored independently from other data used to describe it134/// for the convenience of database access. Notably contains the owner account address.135#[struct_versioning::versioned(version = 2, upper)]136#[derive(Encode, Decode, TypeInfo, MaxEncodedLen)]137pub struct ItemData<CrossAccountId> {138	#[version(..2)]139	pub const_data: BoundedVec<u8, CustomDataLimit>,140141	#[version(..2)]142	pub variable_data: BoundedVec<u8, CustomDataLimit>,143144	pub owner: CrossAccountId,145}146147#[frame_support::pallet]148pub mod pallet {149	use super::*;150	use frame_support::{151		Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key, traits::StorageVersion,152	};153	use frame_system::pallet_prelude::*;154	use up_data_structs::{CollectionId, TokenId};155	use super::weights::WeightInfo;156157	#[pallet::error]158	pub enum Error<T> {159		/// Not Nonfungible item data used to mint in Nonfungible collection.160		NotNonfungibleDataUsedToMintFungibleCollectionToken,161		/// Used amount > 1 with NFT162		NonfungibleItemsHaveNoAmount,163		/// Unable to burn NFT with children164		CantBurnNftWithChildren,165	}166167	#[pallet::config]168	pub trait Config:169		frame_system::Config + pallet_common::Config + pallet_structure::Config + pallet_evm::Config170	{171		type WeightInfo: WeightInfo;172	}173174	const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);175176	#[pallet::pallet]177	#[pallet::storage_version(STORAGE_VERSION)]178	#[pallet::generate_store(pub(super) trait Store)]179	pub struct Pallet<T>(_);180181	/// Total amount of minted tokens in a collection.182	#[pallet::storage]183	pub type TokensMinted<T: Config> =184		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;185186	/// Amount of burnt tokens in a collection.187	#[pallet::storage]188	pub type TokensBurnt<T: Config> =189		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;190191	/// Token data, used to partially describe a token.192	#[pallet::storage]193	pub type TokenData<T: Config> = StorageNMap<194		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),195		Value = ItemData<T::CrossAccountId>,196		QueryKind = OptionQuery,197	>;198199	/// Map of key-value pairs, describing the metadata of a token.200	#[pallet::storage]201	#[pallet::getter(fn token_properties)]202	pub type TokenProperties<T: Config> = StorageNMap<203		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),204		Value = Properties,205		QueryKind = ValueQuery,206		OnEmpty = up_data_structs::TokenProperties,207	>;208209	/// Custom data of a token that is serialized to bytes,210	/// primarily reserved for on-chain operations,211	/// normally obscured from the external users.212	///213	/// Auxiliary properties are slightly different from214	/// usual [`TokenProperties`] due to an unlimited number215	/// and separately stored and written-to key-value pairs.216	///217	/// Currently used to store RMRK data.218	#[pallet::storage]219	#[pallet::getter(fn token_aux_property)]220	pub type TokenAuxProperties<T: Config> = StorageNMap<221		Key = (222			Key<Twox64Concat, CollectionId>,223			Key<Twox64Concat, TokenId>,224			Key<Twox64Concat, PropertyScope>,225			Key<Twox64Concat, PropertyKey>,226		),227		Value = AuxPropertyValue,228		QueryKind = OptionQuery,229	>;230231	/// Used to enumerate tokens owned by account.232	#[pallet::storage]233	pub type Owned<T: Config> = StorageNMap<234		Key = (235			Key<Twox64Concat, CollectionId>,236			Key<Blake2_128Concat, T::CrossAccountId>,237			Key<Twox64Concat, TokenId>,238		),239		Value = bool,240		QueryKind = ValueQuery,241	>;242243	/// Used to enumerate token's children.244	#[pallet::storage]245	#[pallet::getter(fn token_children)]246	pub type TokenChildren<T: Config> = StorageNMap<247		Key = (248			Key<Twox64Concat, CollectionId>,249			Key<Twox64Concat, TokenId>,250			Key<Twox64Concat, (CollectionId, TokenId)>,251		),252		Value = bool,253		QueryKind = ValueQuery,254	>;255256	/// Amount of tokens owned by an account in a collection.257	#[pallet::storage]258	pub type AccountBalance<T: Config> = StorageNMap<259		Key = (260			Key<Twox64Concat, CollectionId>,261			Key<Blake2_128Concat, T::CrossAccountId>,262		),263		Value = u32,264		QueryKind = ValueQuery,265	>;266267	/// Allowance set by a token owner for another user to perform one of certain transactions on a token.268	#[pallet::storage]269	pub type Allowance<T: Config> = StorageNMap<270		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),271		Value = T::CrossAccountId,272		QueryKind = OptionQuery,273	>;274275	/// Operator set by a wallet owner that could perform certain transactions on all tokens in the wallet.276	#[pallet::storage]277	pub type WalletOperator<T: Config> = StorageNMap<278		Key = (279			Key<Twox64Concat, CollectionId>,280			Key<Blake2_128Concat, T::CrossAccountId>,281			Key<Blake2_128Concat, T::CrossAccountId>,282		),283		Value = bool,284		QueryKind = OptionQuery,285	>;286287	/// Upgrade from the old schema to properties.288	#[pallet::hooks]289	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {290		fn on_runtime_upgrade() -> Weight {291			StorageVersion::new(1).put::<Pallet<T>>();292293			Weight::zero()294		}295	}296}297298pub struct NonfungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);299impl<T: Config> NonfungibleHandle<T> {300	pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {301		Self(inner)302	}303	pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {304		self.0305	}306	pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {307		&mut self.0308	}309}310311impl<T: Config> WithRecorder<T> for NonfungibleHandle<T> {312	fn recorder(&self) -> &SubstrateRecorder<T> {313		self.0.recorder()314	}315	fn into_recorder(self) -> SubstrateRecorder<T> {316		self.0.into_recorder()317	}318}319impl<T: Config> Deref for NonfungibleHandle<T> {320	type Target = pallet_common::CollectionHandle<T>;321322	fn deref(&self) -> &Self::Target {323		&self.0324	}325}326327impl<T: Config> Pallet<T> {328	/// Get number of NFT tokens in collection.329	pub fn total_supply(collection: &NonfungibleHandle<T>) -> u32 {330		<TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)331	}332333	/// Check that NFT token exists.334	///335	/// - `token`: Token ID.336	pub fn token_exists(collection: &NonfungibleHandle<T>, token: TokenId) -> bool {337		<TokenData<T>>::contains_key((collection.id, token))338	}339340	/// Set the token property with the scope.341	///342	/// - `property`: Contains key-value pair.343	pub fn set_scoped_token_property(344		collection_id: CollectionId,345		token_id: TokenId,346		scope: PropertyScope,347		property: Property,348	) -> DispatchResult {349		TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {350			properties.try_scoped_set(scope, property.key, property.value)351		})352		.map_err(<CommonError<T>>::from)?;353354		Ok(())355	}356357	/// Batch operation to set multiple properties with the same scope.358	pub fn set_scoped_token_properties(359		collection_id: CollectionId,360		token_id: TokenId,361		scope: PropertyScope,362		properties: impl Iterator<Item = Property>,363	) -> DispatchResult {364		TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {365			stored_properties.try_scoped_set_from_iter(scope, properties)366		})367		.map_err(<CommonError<T>>::from)?;368369		Ok(())370	}371372	/// Add or edit auxiliary data for the property.373	///374	/// - `f`: function that adds or edits auxiliary data.375	pub fn try_mutate_token_aux_property<R, E>(376		collection_id: CollectionId,377		token_id: TokenId,378		scope: PropertyScope,379		key: PropertyKey,380		f: impl FnOnce(&mut Option<AuxPropertyValue>) -> Result<R, E>,381	) -> Result<R, E> {382		<TokenAuxProperties<T>>::try_mutate((collection_id, token_id, scope, key), f)383	}384385	/// Remove auxiliary data for the property.386	pub fn remove_token_aux_property(387		collection_id: CollectionId,388		token_id: TokenId,389		scope: PropertyScope,390		key: PropertyKey,391	) {392		<TokenAuxProperties<T>>::remove((collection_id, token_id, scope, key));393	}394395	/// Get all auxiliary data in a given scope.396	///397	/// Returns iterator over Property Key - Data pairs.398	pub fn iterate_token_aux_properties(399		collection_id: CollectionId,400		token_id: TokenId,401		scope: PropertyScope,402	) -> impl Iterator<Item = (PropertyKey, AuxPropertyValue)> {403		<TokenAuxProperties<T>>::iter_prefix((collection_id, token_id, scope))404	}405406	/// Get ID of the last minted token407	pub fn current_token_id(collection_id: CollectionId) -> TokenId {408		TokenId(<TokensMinted<T>>::get(collection_id))409	}410}411412// unchecked calls skips any permission checks413impl<T: Config> Pallet<T> {414	/// Create NFT collection415	///416	/// `init_collection` will take non-refundable deposit for collection creation.417	///418	/// - `data`: Contains settings for collection limits and permissions.419	pub fn init_collection(420		owner: T::CrossAccountId,421		payer: T::CrossAccountId,422		data: CreateCollectionData<T::AccountId>,423		flags: CollectionFlags,424	) -> Result<CollectionId, DispatchError> {425		<PalletCommon<T>>::init_collection(owner, payer, data, flags)426	}427428	/// Destroy NFT collection429	///430	/// `destroy_collection` will throw error if collection contains any tokens.431	/// Only owner can destroy collection.432	pub fn destroy_collection(433		collection: NonfungibleHandle<T>,434		sender: &T::CrossAccountId,435	) -> DispatchResult {436		let id = collection.id;437438		if Self::collection_has_tokens(id) {439			return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());440		}441442		// =========443444		PalletCommon::destroy_collection(collection.0, sender)?;445446		let _ = <TokenData<T>>::clear_prefix((id,), u32::MAX, None);447		let _ = <TokenChildren<T>>::clear_prefix((id,), u32::MAX, None);448		let _ = <Owned<T>>::clear_prefix((id,), u32::MAX, None);449		<TokensMinted<T>>::remove(id);450		<TokensBurnt<T>>::remove(id);451		let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);452		let _ = <AccountBalance<T>>::clear_prefix((id,), u32::MAX, None);453		let _ = <WalletOperator<T>>::clear_prefix((id,), u32::MAX, None);454		Ok(())455	}456457	/// Burn NFT token458	///459	/// `burn` removes `token` from the `collection`, from it's owner and from the parent token460	/// if the token is nested.461	/// Only the owner can `burn` the token. The `token` shouldn't have any nested tokens.462	/// Also removes all corresponding properties and auxiliary properties.463	///464	/// - `token`: Token that should be burned465	/// - `collection`: Collection that contains the token466	pub fn burn(467		collection: &NonfungibleHandle<T>,468		sender: &T::CrossAccountId,469		token: TokenId,470	) -> DispatchResult {471		let token_data =472			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;473		ensure!(&token_data.owner == sender, <CommonError<T>>::NoPermission);474475		if collection.permissions.access() == AccessMode::AllowList {476			collection.check_allowlist(sender)?;477		}478479		if Self::token_has_children(collection.id, token) {480			return Err(<Error<T>>::CantBurnNftWithChildren.into());481		}482483		let burnt = <TokensBurnt<T>>::get(collection.id)484			.checked_add(1)485			.ok_or(ArithmeticError::Overflow)?;486487		let balance = <AccountBalance<T>>::get((collection.id, token_data.owner.clone()))488			.checked_sub(1)489			.ok_or(ArithmeticError::Overflow)?;490491		// =========492493		if balance == 0 {494			<AccountBalance<T>>::remove((collection.id, token_data.owner.clone()));495		} else {496			<AccountBalance<T>>::insert((collection.id, token_data.owner.clone()), balance);497		}498499		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);500501		<Owned<T>>::remove((collection.id, &token_data.owner, token));502		<TokensBurnt<T>>::insert(collection.id, burnt);503		<TokenData<T>>::remove((collection.id, token));504		<TokenProperties<T>>::remove((collection.id, token));505		let _ = <TokenAuxProperties<T>>::clear_prefix((collection.id, token), u32::MAX, None);506		let old_spender = <Allowance<T>>::take((collection.id, token));507508		if let Some(old_spender) = old_spender {509			<PalletCommon<T>>::deposit_event(CommonEvent::Approved(510				collection.id,511				token,512				token_data.owner.clone(),513				old_spender,514				0,515			));516		}517518		<PalletEvm<T>>::deposit_log(519			ERC721Events::Transfer {520				from: *token_data.owner.as_eth(),521				to: H160::default(),522				token_id: token.into(),523			}524			.to_log(collection_id_to_address(collection.id)),525		);526		<PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(527			collection.id,528			token,529			token_data.owner,530			1,531		));532		Ok(())533	}534535	/// Same as [`burn`] but burns all the tokens that are nested in the token first536	///537	/// - `self_budget`: Limit for searching children in depth.538	/// - `breadth_budget`: Limit of breadth of searching children.539	///540	/// [`burn`]: struct.Pallet.html#method.burn541	#[transactional]542	pub fn burn_recursively(543		collection: &NonfungibleHandle<T>,544		sender: &T::CrossAccountId,545		token: TokenId,546		self_budget: &dyn Budget,547		breadth_budget: &dyn Budget,548	) -> DispatchResultWithPostInfo {549		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);550551		let current_token_account =552			T::CrossTokenAddressMapping::token_to_address(collection.id, token);553554		let mut weight = Weight::zero();555556		// This method is transactional, if user in fact doesn't have permissions to remove token -557		// tokens removed here will be restored after rejected transaction558		for ((collection, token), _) in <TokenChildren<T>>::iter_prefix((collection.id, token)) {559			ensure!(breadth_budget.consume(), <StructureError<T>>::BreadthLimit,);560			let PostDispatchInfo { actual_weight, .. } =561				<PalletStructure<T>>::burn_item_recursively(562					current_token_account.clone(),563					collection,564					token,565					self_budget,566					breadth_budget,567				)?;568			if let Some(actual_weight) = actual_weight {569				weight = weight.saturating_add(actual_weight);570			}571		}572573		Self::burn(collection, sender, token)?;574		DispatchResultWithPostInfo::Ok(PostDispatchInfo {575			actual_weight: Some(weight + <SelfWeightOf<T>>::burn_item()),576			pays_fee: Pays::Yes,577		})578	}579580	/// Batch operation to add, edit or remove properties for the token581	///582	/// All affected properties should have mutable permission and sender should have583	/// permission to edit those properties.584	///585	/// - `nesting_budget`: Limit for searching parents in depth to check ownership.586	/// - `is_token_create`: Indicates that method is called during token initialization.587	///   Allows to bypass ownership check.588	#[transactional]589	fn modify_token_properties(590		collection: &NonfungibleHandle<T>,591		sender: &T::CrossAccountId,592		token_id: TokenId,593		properties: impl Iterator<Item = (PropertyKey, Option<PropertyValue>)>,594		is_token_create: bool,595		nesting_budget: &dyn Budget,596	) -> DispatchResult {597		let mut collection_admin_status = None;598		let mut token_owner_result = None;599600		let mut is_collection_admin =601			|| *collection_admin_status.get_or_insert_with(|| collection.is_owner_or_admin(sender));602603		let mut is_token_owner = || {604			*token_owner_result.get_or_insert_with(|| -> Result<bool, DispatchError> {605				let is_owned = <PalletStructure<T>>::check_indirectly_owned(606					sender.clone(),607					collection.id,608					token_id,609					None,610					nesting_budget,611				)?;612613				Ok(is_owned)614			})615		};616617		for (key, value) in properties {618			let permission = <PalletCommon<T>>::property_permissions(collection.id)619				.get(&key)620				.cloned()621				.unwrap_or_else(PropertyPermission::none);622623			let is_property_exists = TokenProperties::<T>::get((collection.id, token_id))624				.get(&key)625				.is_some();626627			match permission {628				PropertyPermission { mutable: false, .. } if is_property_exists => {629					return Err(<CommonError<T>>::NoPermission.into());630				}631632				PropertyPermission {633					collection_admin,634					token_owner,635					..636				} => {637					//TODO: investigate threats during public minting.638					if is_token_create && (collection_admin || token_owner) && value.is_some() {639						// Pass640					} else if collection_admin && is_collection_admin() {641						// Pass642					} else if token_owner && is_token_owner()? {643						// Pass644					} else {645						fail!(<CommonError<T>>::NoPermission);646					}647				}648			}649650			match value {651				Some(value) => {652					<TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {653						properties.try_set(key.clone(), value)654					})655					.map_err(<CommonError<T>>::from)?;656657					<PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertySet(658						collection.id,659						token_id,660						key,661					));662				}663				None => {664					<TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {665						properties.remove(&key)666					})667					.map_err(<CommonError<T>>::from)?;668669					<PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertyDeleted(670						collection.id,671						token_id,672						key,673					));674				}675			}676		}677678		Ok(())679	}680681	/// Batch operation to add or edit properties for the token682	///683	/// Same as [`modify_token_properties`] but doesn't allow to remove properties684	///685	/// [`modify_token_properties`]: struct.Pallet.html#method.modify_token_properties686	pub fn set_token_properties(687		collection: &NonfungibleHandle<T>,688		sender: &T::CrossAccountId,689		token_id: TokenId,690		properties: impl Iterator<Item = Property>,691		is_token_create: bool,692		nesting_budget: &dyn Budget,693	) -> DispatchResult {694		Self::modify_token_properties(695			collection,696			sender,697			token_id,698			properties.map(|p| (p.key, Some(p.value))),699			is_token_create,700			nesting_budget,701		)702	}703704	/// Add or edit single property for the token705	///706	/// Calls [`set_token_properties`] internally707	///708	/// [`set_token_properties`]: struct.Pallet.html#method.set_token_properties709	pub fn set_token_property(710		collection: &NonfungibleHandle<T>,711		sender: &T::CrossAccountId,712		token_id: TokenId,713		property: Property,714		nesting_budget: &dyn Budget,715	) -> DispatchResult {716		let is_token_create = false;717718		Self::set_token_properties(719			collection,720			sender,721			token_id,722			[property].into_iter(),723			is_token_create,724			nesting_budget,725		)726	}727728	/// Batch operation to remove properties from the token729	///730	/// Same as [`modify_token_properties`] but doesn't allow to add or edit properties731	///732	/// [`modify_token_properties`]: struct.Pallet.html#method.modify_token_properties733	pub fn delete_token_properties(734		collection: &NonfungibleHandle<T>,735		sender: &T::CrossAccountId,736		token_id: TokenId,737		property_keys: impl Iterator<Item = PropertyKey>,738		nesting_budget: &dyn Budget,739	) -> DispatchResult {740		let is_token_create = false;741742		Self::modify_token_properties(743			collection,744			sender,745			token_id,746			property_keys.into_iter().map(|key| (key, None)),747			is_token_create,748			nesting_budget,749		)750	}751752	/// Remove single property from the token753	///754	/// Calls [`delete_token_properties`] internally755	///756	/// [`delete_token_properties`]: struct.Pallet.html#method.delete_token_properties757	pub fn delete_token_property(758		collection: &NonfungibleHandle<T>,759		sender: &T::CrossAccountId,760		token_id: TokenId,761		property_key: PropertyKey,762		nesting_budget: &dyn Budget,763	) -> DispatchResult {764		Self::delete_token_properties(765			collection,766			sender,767			token_id,768			[property_key].into_iter(),769			nesting_budget,770		)771	}772773	/// Add or edit properties for the collection774	pub fn set_collection_properties(775		collection: &NonfungibleHandle<T>,776		sender: &T::CrossAccountId,777		properties: Vec<Property>,778	) -> DispatchResult {779		<PalletCommon<T>>::set_collection_properties(collection, sender, properties)780	}781782	/// Remove properties from the collection783	pub fn delete_collection_properties(784		collection: &CollectionHandle<T>,785		sender: &T::CrossAccountId,786		property_keys: Vec<PropertyKey>,787	) -> DispatchResult {788		<PalletCommon<T>>::delete_collection_properties(collection, sender, property_keys)789	}790791	/// Set property permissions for the token.792	///793	/// Sender should be the owner or admin of token's collection.794	pub fn set_token_property_permissions(795		collection: &CollectionHandle<T>,796		sender: &T::CrossAccountId,797		property_permissions: Vec<PropertyKeyPermission>,798	) -> DispatchResult {799		<PalletCommon<T>>::set_token_property_permissions(collection, sender, property_permissions)800	}801802	/// Set property permissions for the token with scope.803	///804	/// Sender should be the owner or admin of token's collection.805	pub fn set_scoped_token_property_permissions(806		collection: &CollectionHandle<T>,807		sender: &T::CrossAccountId,808		scope: PropertyScope,809		property_permissions: Vec<PropertyKeyPermission>,810	) -> DispatchResult {811		<PalletCommon<T>>::set_scoped_token_property_permissions(812			collection,813			sender,814			scope,815			property_permissions,816		)817	}818819	/// Set property permissions for the collection.820	///821	/// Sender should be the owner or admin of the collection.822	pub fn set_property_permission(823		collection: &CollectionHandle<T>,824		sender: &T::CrossAccountId,825		permission: PropertyKeyPermission,826	) -> DispatchResult {827		<PalletCommon<T>>::set_property_permission(collection, sender, permission)828	}829830	pub fn check_token_immediate_ownership(831		collection: &NonfungibleHandle<T>,832		token: TokenId,833		possible_owner: &T::CrossAccountId,834	) -> DispatchResult {835		let token_data =836			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;837		ensure!(838			&token_data.owner == possible_owner,839			<CommonError<T>>::NoPermission840		);841		Ok(())842	}843844	/// Transfer NFT token from one account to another.845	///846	/// `from` account stops being the owner and `to` account becomes the owner of the token.847	/// If `to` is token than `to` becomes owner of the token and the token become nested.848	/// Unnests token from previous parent if it was nested before.849	/// Removes allowance for the token if there was any.850	/// Throws if transfers aren't allowed for collection or if receiver reached token ownership limit.851	///852	/// - `nesting_budget`: Limit for token nesting depth853	pub fn transfer(854		collection: &NonfungibleHandle<T>,855		from: &T::CrossAccountId,856		to: &T::CrossAccountId,857		token: TokenId,858		nesting_budget: &dyn Budget,859	) -> DispatchResult {860		ensure!(861			collection.limits.transfers_enabled(),862			<CommonError<T>>::TransferNotAllowed863		);864865		let token_data =866			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;867		ensure!(&token_data.owner == from, <CommonError<T>>::NoPermission);868869		if collection.permissions.access() == AccessMode::AllowList {870			collection.check_allowlist(from)?;871			collection.check_allowlist(to)?;872		}873		<PalletCommon<T>>::ensure_correct_receiver(to)?;874875		let balance_from = <AccountBalance<T>>::get((collection.id, from))876			.checked_sub(1)877			.ok_or(<CommonError<T>>::TokenValueTooLow)?;878		let balance_to = if from != to {879			let balance_to = <AccountBalance<T>>::get((collection.id, to))880				.checked_add(1)881				.ok_or(ArithmeticError::Overflow)?;882883			ensure!(884				balance_to < collection.limits.account_token_ownership_limit(),885				<CommonError<T>>::AccountTokenLimitExceeded,886			);887888			Some(balance_to)889		} else {890			None891		};892893		<PalletStructure<T>>::nest_if_sent_to_token(894			from.clone(),895			to,896			collection.id,897			token,898			nesting_budget,899		)?;900901		// =========902903		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);904905		<TokenData<T>>::insert(906			(collection.id, token),907			ItemData {908				owner: to.clone(),909				..token_data910			},911		);912913		if let Some(balance_to) = balance_to {914			// from != to915			if balance_from == 0 {916				<AccountBalance<T>>::remove((collection.id, from));917			} else {918				<AccountBalance<T>>::insert((collection.id, from), balance_from);919			}920			<AccountBalance<T>>::insert((collection.id, to), balance_to);921			<Owned<T>>::remove((collection.id, from, token));922			<Owned<T>>::insert((collection.id, to, token), true);923		}924		Self::set_allowance_unchecked(collection, from, token, None, true);925926		<PalletEvm<T>>::deposit_log(927			ERC721Events::Transfer {928				from: *from.as_eth(),929				to: *to.as_eth(),930				token_id: token.into(),931			}932			.to_log(collection_id_to_address(collection.id)),933		);934		<PalletCommon<T>>::deposit_event(CommonEvent::Transfer(935			collection.id,936			token,937			from.clone(),938			to.clone(),939			1,940		));941		Ok(())942	}943944	/// Batch operation to mint multiple NFT tokens.945	///946	/// The sender should be the owner/admin of the collection or collection should be configured947	/// to allow public minting.948	/// Throws if amount of tokens reached it's limit for the collection or if caller reached949	/// token ownership limit.950	///951	/// - `data`: Contains list of token properties and users who will become the owners of the952	///   corresponging tokens.953	/// - `nesting_budget`: Limit for token nesting depth954	pub fn create_multiple_items(955		collection: &NonfungibleHandle<T>,956		sender: &T::CrossAccountId,957		data: Vec<CreateItemData<T>>,958		nesting_budget: &dyn Budget,959	) -> DispatchResult {960		if !collection.is_owner_or_admin(sender) {961			ensure!(962				collection.permissions.mint_mode(),963				<CommonError<T>>::PublicMintingNotAllowed964			);965			collection.check_allowlist(sender)?;966967			for item in data.iter() {968				collection.check_allowlist(&item.owner)?;969			}970		}971972		for data in data.iter() {973			<PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;974		}975976		let first_token = <TokensMinted<T>>::get(collection.id);977		let tokens_minted = first_token978			.checked_add(data.len() as u32)979			.ok_or(ArithmeticError::Overflow)?;980		ensure!(981			tokens_minted <= collection.limits.token_limit(),982			<CommonError<T>>::CollectionTokenLimitExceeded983		);984985		let mut balances = BTreeMap::new();986		for data in &data {987			let balance = balances988				.entry(&data.owner)989				.or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));990			*balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;991992			ensure!(993				*balance <= collection.limits.account_token_ownership_limit(),994				<CommonError<T>>::AccountTokenLimitExceeded,995			);996		}997998		for (i, data) in data.iter().enumerate() {999			let token = TokenId(first_token + i as u32 + 1);10001001			<PalletStructure<T>>::check_nesting(1002				sender.clone(),1003				&data.owner,1004				collection.id,1005				token,1006				nesting_budget,1007			)?;1008		}10091010		// =========10111012		with_transaction(|| {1013			for (i, data) in data.iter().enumerate() {1014				let token = first_token + i as u32 + 1;10151016				<TokenData<T>>::insert(1017					(collection.id, token),1018					ItemData {1019						// const_data: data.const_data.clone(),1020						owner: data.owner.clone(),1021					},1022				);10231024				<PalletStructure<T>>::nest_if_sent_to_token_unchecked(1025					&data.owner,1026					collection.id,1027					TokenId(token),1028				);10291030				if let Err(e) = Self::set_token_properties(1031					collection,1032					sender,1033					TokenId(token),1034					data.properties.clone().into_iter(),1035					true,1036					nesting_budget,1037				) {1038					return TransactionOutcome::Rollback(Err(e));1039				}1040			}1041			TransactionOutcome::Commit(Ok(()))1042		})?;10431044		<TokensMinted<T>>::insert(collection.id, tokens_minted);1045		for (account, balance) in balances {1046			<AccountBalance<T>>::insert((collection.id, account), balance);1047		}1048		for (i, data) in data.into_iter().enumerate() {1049			let token = first_token + i as u32 + 1;1050			<Owned<T>>::insert((collection.id, &data.owner, token), true);10511052			<PalletEvm<T>>::deposit_log(1053				ERC721Events::Transfer {1054					from: H160::default(),1055					to: *data.owner.as_eth(),1056					token_id: token.into(),1057				}1058				.to_log(collection_id_to_address(collection.id)),1059			);1060			<PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(1061				collection.id,1062				TokenId(token),1063				data.owner.clone(),1064				1,1065			));1066		}1067		Ok(())1068	}10691070	pub fn set_allowance_unchecked(1071		collection: &NonfungibleHandle<T>,1072		sender: &T::CrossAccountId,1073		token: TokenId,1074		spender: Option<&T::CrossAccountId>,1075		assume_implicit_eth: bool,1076	) {1077		if let Some(spender) = spender {1078			let old_spender = <Allowance<T>>::get((collection.id, token));1079			<Allowance<T>>::insert((collection.id, token), spender);1080			// In ERC721 there is only one possible approved user of token, so we set1081			// approved user to spender1082			<PalletEvm<T>>::deposit_log(1083				ERC721Events::Approval {1084					owner: *sender.as_eth(),1085					approved: *spender.as_eth(),1086					token_id: token.into(),1087				}1088				.to_log(collection_id_to_address(collection.id)),1089			);1090			// In Unique chain, any token can have any amount of approved users, so we need to1091			// set allowance of old owner to 0, and allowance of new owner to 11092			if old_spender.as_ref() != Some(spender) {1093				if let Some(old_owner) = old_spender {1094					<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1095						collection.id,1096						token,1097						sender.clone(),1098						old_owner,1099						0,1100					));1101				}1102				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1103					collection.id,1104					token,1105					sender.clone(),1106					spender.clone(),1107					1,1108				));1109			}1110		} else {1111			let old_spender = <Allowance<T>>::take((collection.id, token));1112			if !assume_implicit_eth {1113				// In ERC721 there is only one possible approved user of token, so we set1114				// approved user to zero address1115				<PalletEvm<T>>::deposit_log(1116					ERC721Events::Approval {1117						owner: *sender.as_eth(),1118						approved: H160::default(),1119						token_id: token.into(),1120					}1121					.to_log(collection_id_to_address(collection.id)),1122				);1123			}1124			// In Unique chain, any token can have any amount of approved users, so we need to1125			// set allowance of old owner to 01126			if let Some(old_spender) = old_spender {1127				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1128					collection.id,1129					token,1130					sender.clone(),1131					old_spender,1132					0,1133				));1134			}1135		}1136	}11371138	/// Set allowance for the spender to `transfer` or `burn` sender's token.1139	///1140	/// - `token`: Token the spender is allowed to `transfer` or `burn`.1141	pub fn set_allowance(1142		collection: &NonfungibleHandle<T>,1143		sender: &T::CrossAccountId,1144		token: TokenId,1145		spender: Option<&T::CrossAccountId>,1146	) -> DispatchResult {1147		if collection.permissions.access() == AccessMode::AllowList {1148			collection.check_allowlist(sender)?;1149			if let Some(spender) = spender {1150				collection.check_allowlist(spender)?;1151			}1152		}11531154		if let Some(spender) = spender {1155			<PalletCommon<T>>::ensure_correct_receiver(spender)?;1156		}11571158		let token_data =1159			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;1160		if &token_data.owner != sender {1161			ensure!(1162				collection.ignores_owned_amount(sender),1163				<CommonError<T>>::CantApproveMoreThanOwned1164			);1165		}11661167		// =========11681169		Self::set_allowance_unchecked(collection, sender, token, spender, false);1170		Ok(())1171	}11721173	/// Checks allowance for the spender to use the token.1174	fn check_allowed(1175		collection: &NonfungibleHandle<T>,1176		spender: &T::CrossAccountId,1177		from: &T::CrossAccountId,1178		token: TokenId,1179		nesting_budget: &dyn Budget,1180	) -> DispatchResult {1181		if spender.conv_eq(from) {1182			return Ok(());1183		}1184		if collection.permissions.access() == AccessMode::AllowList {1185			// `from`, `to` checked in [`transfer`]1186			collection.check_allowlist(spender)?;1187		}11881189		if collection.limits.owner_can_transfer() && collection.is_owner_or_admin(spender) {1190			return Ok(());1191		}11921193		if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {1194			ensure!(1195				<PalletStructure<T>>::check_indirectly_owned(1196					spender.clone(),1197					source.0,1198					source.1,1199					None,1200					nesting_budget1201				)?,1202				<CommonError<T>>::ApprovedValueTooLow,1203			);1204			return Ok(());1205		}1206		if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {1207			return Ok(());1208		}1209		if <WalletOperator<T>>::get((collection.id, from, spender)) == Some(true) {1210			return Ok(());1211		}1212		ensure!(1213			collection.ignores_allowance(spender),1214			<CommonError<T>>::ApprovedValueTooLow1215		);1216		Ok(())1217	}12181219	/// Transfer NFT token from one account to another.1220	///1221	/// Same as the [`transfer`] but spender doesn't needs to be the owner of the token.1222	/// The owner should set allowance for the spender to transfer token.1223	///1224	/// [`transfer`]: struct.Pallet.html#method.transfer1225	pub fn transfer_from(1226		collection: &NonfungibleHandle<T>,1227		spender: &T::CrossAccountId,1228		from: &T::CrossAccountId,1229		to: &T::CrossAccountId,1230		token: TokenId,1231		nesting_budget: &dyn Budget,1232	) -> DispatchResult {1233		Self::check_allowed(collection, spender, from, token, nesting_budget)?;12341235		// =========12361237		// Allowance is reset in [`transfer`]1238		Self::transfer(collection, from, to, token, nesting_budget)1239	}12401241	/// Burn NFT token for `from` account.1242	///1243	/// Same as the [`burn`] but spender doesn't need to be an owner of the token. The owner should1244	/// set allowance for the spender to burn token.1245	///1246	/// [`burn`]: struct.Pallet.html#method.burn1247	pub fn burn_from(1248		collection: &NonfungibleHandle<T>,1249		spender: &T::CrossAccountId,1250		from: &T::CrossAccountId,1251		token: TokenId,1252		nesting_budget: &dyn Budget,1253	) -> DispatchResult {1254		Self::check_allowed(collection, spender, from, token, nesting_budget)?;12551256		// =========12571258		Self::burn(collection, from, token)1259	}12601261	/// Check that `from` token could be nested in `under` token.1262	///1263	pub fn check_nesting(1264		handle: &NonfungibleHandle<T>,1265		sender: T::CrossAccountId,1266		from: (CollectionId, TokenId),1267		under: TokenId,1268		nesting_budget: &dyn Budget,1269	) -> DispatchResult {1270		let nesting = handle.permissions.nesting();12711272		#[cfg(not(feature = "runtime-benchmarks"))]1273		let permissive = false;1274		#[cfg(feature = "runtime-benchmarks")]1275		let permissive = nesting.permissive;12761277		if permissive {1278			// Pass1279		} else if nesting.token_owner1280			&& <PalletStructure<T>>::check_indirectly_owned(1281				sender.clone(),1282				handle.id,1283				under,1284				Some(from),1285				nesting_budget,1286			)? {1287			// Pass1288		} else if nesting.collection_admin && handle.is_owner_or_admin(&sender) {1289			// Pass1290		} else {1291			fail!(<CommonError<T>>::UserIsNotAllowedToNest);1292		}12931294		if let Some(whitelist) = &nesting.restricted {1295			ensure!(1296				whitelist.contains(&from.0),1297				<CommonError<T>>::SourceCollectionIsNotAllowedToNest1298			);1299		}1300		Ok(())1301	}13021303	fn nest(under: (CollectionId, TokenId), to_nest: (CollectionId, TokenId)) {1304		<TokenChildren<T>>::insert((under.0, under.1, (to_nest.0, to_nest.1)), true);1305	}13061307	fn unnest(under: (CollectionId, TokenId), to_unnest: (CollectionId, TokenId)) {1308		<TokenChildren<T>>::remove((under.0, under.1, to_unnest));1309	}13101311	fn collection_has_tokens(collection_id: CollectionId) -> bool {1312		<TokenData<T>>::iter_prefix((collection_id,))1313			.next()1314			.is_some()1315	}13161317	fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {1318		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1319			.next()1320			.is_some()1321	}13221323	pub fn token_children_ids(collection_id: CollectionId, token_id: TokenId) -> Vec<TokenChild> {1324		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1325			.map(|((child_collection_id, child_id), _)| TokenChild {1326				collection: child_collection_id,1327				token: child_id,1328			})1329			.collect()1330	}13311332	/// Mint single NFT token.1333	///1334	/// Delegated to [`create_multiple_items`]1335	///1336	/// [`create_multiple_items`]: struct.Pallet.html#method.create_multiple_items1337	pub fn create_item(1338		collection: &NonfungibleHandle<T>,1339		sender: &T::CrossAccountId,1340		data: CreateItemData<T>,1341		nesting_budget: &dyn Budget,1342	) -> DispatchResult {1343		Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1344	}13451346	/// Sets or unsets the approval of a given operator.1347	///1348	/// An operator is allowed to transfer all token pieces of the sender on their behalf.1349	/// - `owner`: Token owner1350	/// - `operator`: Operator1351	/// - `approve`: Is operator enabled or disabled1352	pub fn set_approval_for_all(1353		collection: &NonfungibleHandle<T>,1354		owner: &T::CrossAccountId,1355		operator: &T::CrossAccountId,1356		approve: bool,1357	) -> DispatchResult {1358		if collection.permissions.access() == AccessMode::AllowList {1359			collection.check_allowlist(owner)?;1360			collection.check_allowlist(operator)?;1361		}13621363		<PalletCommon<T>>::ensure_correct_receiver(operator)?;13641365		// =========13661367		<WalletOperator<T>>::insert((collection.id, owner, operator), approve);1368		<PalletEvm<T>>::deposit_log(1369			ERC721Events::ApprovalForAll {1370				owner: *owner.as_eth(),1371				operator: *operator.as_eth(),1372				approved: approve,1373			}1374			.to_log(collection_id_to_address(collection.id)),1375		);1376		<PalletCommon<T>>::deposit_event(CommonEvent::ApprovedForAll(1377			collection.id,1378			owner.clone(),1379			operator.clone(),1380			approve,1381		));1382		Ok(())1383	}13841385	/// Tells whether an operator is approved by a given owner.1386	pub fn is_approved_for_all(1387		collection: &NonfungibleHandle<T>,1388		owner: &T::CrossAccountId,1389		operator: &T::CrossAccountId,1390	) -> bool {1391		<WalletOperator<T>>::get((collection.id, owner, operator)).unwrap_or(false)1392	}1393}
after · pallets/nonfungible/src/lib.rs
1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617//! # Nonfungible Pallet18//!19//! The Nonfungible pallet provides functionality for handling nonfungible collections and tokens.20//!21//! - [`Config`]22//! - [`NonfungibleHandle`]23//! - [`Pallet`]24//! - [`CommonWeights`](common::CommonWeights)25//!26//! ## Overview27//!28//! The Nonfungible pallet provides functions for:29//!30//! - NFT collection creation and removal31//! - Minting and burning of NFT tokens32//! - Retrieving account balances33//! - Transfering NFT tokens34//! - Setting and checking allowance for NFT tokens35//! - Setting properties and permissions for NFT collections and tokens36//! - Nesting and unnesting tokens37//!38//! ### Terminology39//!40//! - **NFT token:** Non fungible token.41//!42//! - **NFT Collection:** A collection of NFT tokens. All NFT tokens are part of a collection.43//!   Each collection can define it's own properties, properties for it's tokens and set of permissions.44//!45//! - **Balance:** Number of NFT tokens owned by an account46//!47//! - **Allowance:** NFT tokens owned by one account that another account is allowed to make operations on48//!49//! - **Burning:** The process of “deleting” a token from a collection and from50//!   an account balance of the owner.51//!52//! - **Nesting:** Setting up parent-child relationship between tokens. Nested tokens are inhereting53//!   owner from their parent. There could be multiple levels of nesting. Token couldn't be nested in54//!   it's child token i.e. parent-child relationship graph shouldn't have cycles.55//!56//! - **Properties:** Key-Values pairs. Token properties are attached to a token. Collection properties are57//!   attached to a collection. Set of permissions could be defined for each property.58//!59//! ### Implementations60//!61//! The Nonfungible pallet provides implementations for the following traits. If these traits provide62//! the functionality that you need, then you can avoid coupling with the Nonfungible pallet.63//!64//! - [`CommonWeightInfo`](pallet_common::CommonWeightInfo): Functions for retrieval of transaction weight65//! - [`CommonCollectionOperations`](pallet_common::CommonCollectionOperations): Functions for dealing66//!   with collections67//!68//! ## Interface69//!70//! ### Dispatchable Functions71//!72//! - `init_collection` - Create NFT collection. NFT collection can be configured to allow or deny access for73//!   some accounts.74//! - `destroy_collection` - Destroy exising NFT collection. There should be no tokens in the collection.75//! - `burn` - Burn NFT token owned by account.76//! - `transfer` - Transfer NFT token. Transfers should be enabled for NFT collection.77//!   Nests the NFT token if it is sent to another token.78//! - `create_item` - Mint NFT token in collection. Sender should have permission to mint tokens.79//! - `set_allowance` - Set allowance for another account.80//! - `set_token_property` - Set token property value.81//! - `delete_token_property` - Remove property from the token.82//! - `set_collection_properties` - Set collection properties.83//! - `delete_collection_properties` - Remove properties from the collection.84//! - `set_property_permission` - Set collection property permission.85//! - `set_token_property_permissions` - Set token property permissions.86//!87//! ## Assumptions88//!89//! * To perform operations on tokens sender should be in collection's allow list if collection access mode is `AllowList`.9091#![cfg_attr(not(feature = "std"), no_std)]9293use erc::ERC721Events;94use evm_coder::ToLog;95use frame_support::{96	BoundedVec, ensure, fail, transactional,97	storage::with_transaction,98	pallet_prelude::DispatchResultWithPostInfo,99	pallet_prelude::Weight,100	dispatch::{PostDispatchInfo, Pays},101};102use up_data_structs::{103	AccessMode, CollectionId, CollectionFlags, CustomDataLimit, TokenId, CreateCollectionData,104	CreateNftExData, mapping::TokenAddressMapping, budget::Budget, Property, PropertyPermission,105	PropertyKey, PropertyValue, PropertyKeyPermission, Properties, PropertyScope, TrySetProperty,106	TokenChild, AuxPropertyValue,107};108use pallet_evm::{account::CrossAccountId, Pallet as PalletEvm};109use pallet_common::{110	Error as CommonError, Pallet as PalletCommon, Event as CommonEvent, CollectionHandle,111	eth::collection_id_to_address,112};113use pallet_structure::{Pallet as PalletStructure, Error as StructureError};114use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};115use sp_core::H160;116use sp_runtime::{ArithmeticError, DispatchError, DispatchResult, TransactionOutcome};117use sp_std::{vec::Vec, vec, collections::btree_map::BTreeMap};118use core::ops::Deref;119use codec::{Encode, Decode, MaxEncodedLen};120use scale_info::TypeInfo;121122pub use pallet::*;123use weights::WeightInfo;124#[cfg(feature = "runtime-benchmarks")]125pub mod benchmarking;126pub mod common;127pub mod erc;128pub mod weights;129130pub type CreateItemData<T> = CreateNftExData<<T as pallet_evm::Config>::CrossAccountId>;131pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;132133/// Token data, stored independently from other data used to describe it134/// for the convenience of database access. Notably contains the owner account address.135#[struct_versioning::versioned(version = 2, upper)]136#[derive(Encode, Decode, TypeInfo, MaxEncodedLen)]137pub struct ItemData<CrossAccountId> {138	#[version(..2)]139	pub const_data: BoundedVec<u8, CustomDataLimit>,140141	#[version(..2)]142	pub variable_data: BoundedVec<u8, CustomDataLimit>,143144	pub owner: CrossAccountId,145}146147#[frame_support::pallet]148pub mod pallet {149	use super::*;150	use frame_support::{151		Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key, traits::StorageVersion,152	};153	use frame_system::pallet_prelude::*;154	use up_data_structs::{CollectionId, TokenId};155	use super::weights::WeightInfo;156157	#[pallet::error]158	pub enum Error<T> {159		/// Not Nonfungible item data used to mint in Nonfungible collection.160		NotNonfungibleDataUsedToMintFungibleCollectionToken,161		/// Used amount > 1 with NFT162		NonfungibleItemsHaveNoAmount,163		/// Unable to burn NFT with children164		CantBurnNftWithChildren,165	}166167	#[pallet::config]168	pub trait Config:169		frame_system::Config + pallet_common::Config + pallet_structure::Config + pallet_evm::Config170	{171		type WeightInfo: WeightInfo;172	}173174	const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);175176	#[pallet::pallet]177	#[pallet::storage_version(STORAGE_VERSION)]178	#[pallet::generate_store(pub(super) trait Store)]179	pub struct Pallet<T>(_);180181	/// Total amount of minted tokens in a collection.182	#[pallet::storage]183	pub type TokensMinted<T: Config> =184		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;185186	/// Amount of burnt tokens in a collection.187	#[pallet::storage]188	pub type TokensBurnt<T: Config> =189		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;190191	/// Token data, used to partially describe a token.192	#[pallet::storage]193	pub type TokenData<T: Config> = StorageNMap<194		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),195		Value = ItemData<T::CrossAccountId>,196		QueryKind = OptionQuery,197	>;198199	/// Map of key-value pairs, describing the metadata of a token.200	#[pallet::storage]201	#[pallet::getter(fn token_properties)]202	pub type TokenProperties<T: Config> = StorageNMap<203		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),204		Value = Properties,205		QueryKind = ValueQuery,206		OnEmpty = up_data_structs::TokenProperties,207	>;208209	/// Custom data of a token that is serialized to bytes,210	/// primarily reserved for on-chain operations,211	/// normally obscured from the external users.212	///213	/// Auxiliary properties are slightly different from214	/// usual [`TokenProperties`] due to an unlimited number215	/// and separately stored and written-to key-value pairs.216	///217	/// Currently used to store RMRK data.218	#[pallet::storage]219	#[pallet::getter(fn token_aux_property)]220	pub type TokenAuxProperties<T: Config> = StorageNMap<221		Key = (222			Key<Twox64Concat, CollectionId>,223			Key<Twox64Concat, TokenId>,224			Key<Twox64Concat, PropertyScope>,225			Key<Twox64Concat, PropertyKey>,226		),227		Value = AuxPropertyValue,228		QueryKind = OptionQuery,229	>;230231	/// Used to enumerate tokens owned by account.232	#[pallet::storage]233	pub type Owned<T: Config> = StorageNMap<234		Key = (235			Key<Twox64Concat, CollectionId>,236			Key<Blake2_128Concat, T::CrossAccountId>,237			Key<Twox64Concat, TokenId>,238		),239		Value = bool,240		QueryKind = ValueQuery,241	>;242243	/// Used to enumerate token's children.244	#[pallet::storage]245	#[pallet::getter(fn token_children)]246	pub type TokenChildren<T: Config> = StorageNMap<247		Key = (248			Key<Twox64Concat, CollectionId>,249			Key<Twox64Concat, TokenId>,250			Key<Twox64Concat, (CollectionId, TokenId)>,251		),252		Value = bool,253		QueryKind = ValueQuery,254	>;255256	/// Amount of tokens owned by an account in a collection.257	#[pallet::storage]258	pub type AccountBalance<T: Config> = StorageNMap<259		Key = (260			Key<Twox64Concat, CollectionId>,261			Key<Blake2_128Concat, T::CrossAccountId>,262		),263		Value = u32,264		QueryKind = ValueQuery,265	>;266267	/// Allowance set by a token owner for another user to perform one of certain transactions on a token.268	#[pallet::storage]269	pub type Allowance<T: Config> = StorageNMap<270		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),271		Value = T::CrossAccountId,272		QueryKind = OptionQuery,273	>;274275	/// Operator set by a wallet owner that could perform certain transactions on all tokens in the wallet.276	#[pallet::storage]277	pub type WalletOperator<T: Config> = StorageNMap<278		Key = (279			Key<Twox64Concat, CollectionId>,280			Key<Blake2_128Concat, T::CrossAccountId>,281			Key<Blake2_128Concat, T::CrossAccountId>,282		),283		Value = bool,284		QueryKind = ValueQuery,285	>;286287	/// Upgrade from the old schema to properties.288	#[pallet::hooks]289	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {290		fn on_runtime_upgrade() -> Weight {291			StorageVersion::new(1).put::<Pallet<T>>();292293			Weight::zero()294		}295	}296}297298pub struct NonfungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);299impl<T: Config> NonfungibleHandle<T> {300	pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {301		Self(inner)302	}303	pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {304		self.0305	}306	pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {307		&mut self.0308	}309}310311impl<T: Config> WithRecorder<T> for NonfungibleHandle<T> {312	fn recorder(&self) -> &SubstrateRecorder<T> {313		self.0.recorder()314	}315	fn into_recorder(self) -> SubstrateRecorder<T> {316		self.0.into_recorder()317	}318}319impl<T: Config> Deref for NonfungibleHandle<T> {320	type Target = pallet_common::CollectionHandle<T>;321322	fn deref(&self) -> &Self::Target {323		&self.0324	}325}326327impl<T: Config> Pallet<T> {328	/// Get number of NFT tokens in collection.329	pub fn total_supply(collection: &NonfungibleHandle<T>) -> u32 {330		<TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)331	}332333	/// Check that NFT token exists.334	///335	/// - `token`: Token ID.336	pub fn token_exists(collection: &NonfungibleHandle<T>, token: TokenId) -> bool {337		<TokenData<T>>::contains_key((collection.id, token))338	}339340	/// Set the token property with the scope.341	///342	/// - `property`: Contains key-value pair.343	pub fn set_scoped_token_property(344		collection_id: CollectionId,345		token_id: TokenId,346		scope: PropertyScope,347		property: Property,348	) -> DispatchResult {349		TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {350			properties.try_scoped_set(scope, property.key, property.value)351		})352		.map_err(<CommonError<T>>::from)?;353354		Ok(())355	}356357	/// Batch operation to set multiple properties with the same scope.358	pub fn set_scoped_token_properties(359		collection_id: CollectionId,360		token_id: TokenId,361		scope: PropertyScope,362		properties: impl Iterator<Item = Property>,363	) -> DispatchResult {364		TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {365			stored_properties.try_scoped_set_from_iter(scope, properties)366		})367		.map_err(<CommonError<T>>::from)?;368369		Ok(())370	}371372	/// Add or edit auxiliary data for the property.373	///374	/// - `f`: function that adds or edits auxiliary data.375	pub fn try_mutate_token_aux_property<R, E>(376		collection_id: CollectionId,377		token_id: TokenId,378		scope: PropertyScope,379		key: PropertyKey,380		f: impl FnOnce(&mut Option<AuxPropertyValue>) -> Result<R, E>,381	) -> Result<R, E> {382		<TokenAuxProperties<T>>::try_mutate((collection_id, token_id, scope, key), f)383	}384385	/// Remove auxiliary data for the property.386	pub fn remove_token_aux_property(387		collection_id: CollectionId,388		token_id: TokenId,389		scope: PropertyScope,390		key: PropertyKey,391	) {392		<TokenAuxProperties<T>>::remove((collection_id, token_id, scope, key));393	}394395	/// Get all auxiliary data in a given scope.396	///397	/// Returns iterator over Property Key - Data pairs.398	pub fn iterate_token_aux_properties(399		collection_id: CollectionId,400		token_id: TokenId,401		scope: PropertyScope,402	) -> impl Iterator<Item = (PropertyKey, AuxPropertyValue)> {403		<TokenAuxProperties<T>>::iter_prefix((collection_id, token_id, scope))404	}405406	/// Get ID of the last minted token407	pub fn current_token_id(collection_id: CollectionId) -> TokenId {408		TokenId(<TokensMinted<T>>::get(collection_id))409	}410}411412// unchecked calls skips any permission checks413impl<T: Config> Pallet<T> {414	/// Create NFT collection415	///416	/// `init_collection` will take non-refundable deposit for collection creation.417	///418	/// - `data`: Contains settings for collection limits and permissions.419	pub fn init_collection(420		owner: T::CrossAccountId,421		payer: T::CrossAccountId,422		data: CreateCollectionData<T::AccountId>,423		flags: CollectionFlags,424	) -> Result<CollectionId, DispatchError> {425		<PalletCommon<T>>::init_collection(owner, payer, data, flags)426	}427428	/// Destroy NFT collection429	///430	/// `destroy_collection` will throw error if collection contains any tokens.431	/// Only owner can destroy collection.432	pub fn destroy_collection(433		collection: NonfungibleHandle<T>,434		sender: &T::CrossAccountId,435	) -> DispatchResult {436		let id = collection.id;437438		if Self::collection_has_tokens(id) {439			return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());440		}441442		// =========443444		PalletCommon::destroy_collection(collection.0, sender)?;445446		let _ = <TokenData<T>>::clear_prefix((id,), u32::MAX, None);447		let _ = <TokenChildren<T>>::clear_prefix((id,), u32::MAX, None);448		let _ = <Owned<T>>::clear_prefix((id,), u32::MAX, None);449		<TokensMinted<T>>::remove(id);450		<TokensBurnt<T>>::remove(id);451		let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);452		let _ = <AccountBalance<T>>::clear_prefix((id,), u32::MAX, None);453		let _ = <WalletOperator<T>>::clear_prefix((id,), u32::MAX, None);454		Ok(())455	}456457	/// Burn NFT token458	///459	/// `burn` removes `token` from the `collection`, from it's owner and from the parent token460	/// if the token is nested.461	/// Only the owner can `burn` the token. The `token` shouldn't have any nested tokens.462	/// Also removes all corresponding properties and auxiliary properties.463	///464	/// - `token`: Token that should be burned465	/// - `collection`: Collection that contains the token466	pub fn burn(467		collection: &NonfungibleHandle<T>,468		sender: &T::CrossAccountId,469		token: TokenId,470	) -> DispatchResult {471		let token_data =472			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;473		ensure!(&token_data.owner == sender, <CommonError<T>>::NoPermission);474475		if collection.permissions.access() == AccessMode::AllowList {476			collection.check_allowlist(sender)?;477		}478479		if Self::token_has_children(collection.id, token) {480			return Err(<Error<T>>::CantBurnNftWithChildren.into());481		}482483		let burnt = <TokensBurnt<T>>::get(collection.id)484			.checked_add(1)485			.ok_or(ArithmeticError::Overflow)?;486487		let balance = <AccountBalance<T>>::get((collection.id, token_data.owner.clone()))488			.checked_sub(1)489			.ok_or(ArithmeticError::Overflow)?;490491		// =========492493		if balance == 0 {494			<AccountBalance<T>>::remove((collection.id, token_data.owner.clone()));495		} else {496			<AccountBalance<T>>::insert((collection.id, token_data.owner.clone()), balance);497		}498499		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);500501		<Owned<T>>::remove((collection.id, &token_data.owner, token));502		<TokensBurnt<T>>::insert(collection.id, burnt);503		<TokenData<T>>::remove((collection.id, token));504		<TokenProperties<T>>::remove((collection.id, token));505		let _ = <TokenAuxProperties<T>>::clear_prefix((collection.id, token), u32::MAX, None);506		let old_spender = <Allowance<T>>::take((collection.id, token));507508		if let Some(old_spender) = old_spender {509			<PalletCommon<T>>::deposit_event(CommonEvent::Approved(510				collection.id,511				token,512				token_data.owner.clone(),513				old_spender,514				0,515			));516		}517518		<PalletEvm<T>>::deposit_log(519			ERC721Events::Transfer {520				from: *token_data.owner.as_eth(),521				to: H160::default(),522				token_id: token.into(),523			}524			.to_log(collection_id_to_address(collection.id)),525		);526		<PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(527			collection.id,528			token,529			token_data.owner,530			1,531		));532		Ok(())533	}534535	/// Same as [`burn`] but burns all the tokens that are nested in the token first536	///537	/// - `self_budget`: Limit for searching children in depth.538	/// - `breadth_budget`: Limit of breadth of searching children.539	///540	/// [`burn`]: struct.Pallet.html#method.burn541	#[transactional]542	pub fn burn_recursively(543		collection: &NonfungibleHandle<T>,544		sender: &T::CrossAccountId,545		token: TokenId,546		self_budget: &dyn Budget,547		breadth_budget: &dyn Budget,548	) -> DispatchResultWithPostInfo {549		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);550551		let current_token_account =552			T::CrossTokenAddressMapping::token_to_address(collection.id, token);553554		let mut weight = Weight::zero();555556		// This method is transactional, if user in fact doesn't have permissions to remove token -557		// tokens removed here will be restored after rejected transaction558		for ((collection, token), _) in <TokenChildren<T>>::iter_prefix((collection.id, token)) {559			ensure!(breadth_budget.consume(), <StructureError<T>>::BreadthLimit,);560			let PostDispatchInfo { actual_weight, .. } =561				<PalletStructure<T>>::burn_item_recursively(562					current_token_account.clone(),563					collection,564					token,565					self_budget,566					breadth_budget,567				)?;568			if let Some(actual_weight) = actual_weight {569				weight = weight.saturating_add(actual_weight);570			}571		}572573		Self::burn(collection, sender, token)?;574		DispatchResultWithPostInfo::Ok(PostDispatchInfo {575			actual_weight: Some(weight + <SelfWeightOf<T>>::burn_item()),576			pays_fee: Pays::Yes,577		})578	}579580	/// Batch operation to add, edit or remove properties for the token581	///582	/// All affected properties should have mutable permission and sender should have583	/// permission to edit those properties.584	///585	/// - `nesting_budget`: Limit for searching parents in depth to check ownership.586	/// - `is_token_create`: Indicates that method is called during token initialization.587	///   Allows to bypass ownership check.588	#[transactional]589	fn modify_token_properties(590		collection: &NonfungibleHandle<T>,591		sender: &T::CrossAccountId,592		token_id: TokenId,593		properties: impl Iterator<Item = (PropertyKey, Option<PropertyValue>)>,594		is_token_create: bool,595		nesting_budget: &dyn Budget,596	) -> DispatchResult {597		let mut collection_admin_status = None;598		let mut token_owner_result = None;599600		let mut is_collection_admin =601			|| *collection_admin_status.get_or_insert_with(|| collection.is_owner_or_admin(sender));602603		let mut is_token_owner = || {604			*token_owner_result.get_or_insert_with(|| -> Result<bool, DispatchError> {605				let is_owned = <PalletStructure<T>>::check_indirectly_owned(606					sender.clone(),607					collection.id,608					token_id,609					None,610					nesting_budget,611				)?;612613				Ok(is_owned)614			})615		};616617		for (key, value) in properties {618			let permission = <PalletCommon<T>>::property_permissions(collection.id)619				.get(&key)620				.cloned()621				.unwrap_or_else(PropertyPermission::none);622623			let is_property_exists = TokenProperties::<T>::get((collection.id, token_id))624				.get(&key)625				.is_some();626627			match permission {628				PropertyPermission { mutable: false, .. } if is_property_exists => {629					return Err(<CommonError<T>>::NoPermission.into());630				}631632				PropertyPermission {633					collection_admin,634					token_owner,635					..636				} => {637					//TODO: investigate threats during public minting.638					if is_token_create && (collection_admin || token_owner) && value.is_some() {639						// Pass640					} else if collection_admin && is_collection_admin() {641						// Pass642					} else if token_owner && is_token_owner()? {643						// Pass644					} else {645						fail!(<CommonError<T>>::NoPermission);646					}647				}648			}649650			match value {651				Some(value) => {652					<TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {653						properties.try_set(key.clone(), value)654					})655					.map_err(<CommonError<T>>::from)?;656657					<PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertySet(658						collection.id,659						token_id,660						key,661					));662				}663				None => {664					<TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {665						properties.remove(&key)666					})667					.map_err(<CommonError<T>>::from)?;668669					<PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertyDeleted(670						collection.id,671						token_id,672						key,673					));674				}675			}676		}677678		Ok(())679	}680681	/// Batch operation to add or edit properties for the token682	///683	/// Same as [`modify_token_properties`] but doesn't allow to remove properties684	///685	/// [`modify_token_properties`]: struct.Pallet.html#method.modify_token_properties686	pub fn set_token_properties(687		collection: &NonfungibleHandle<T>,688		sender: &T::CrossAccountId,689		token_id: TokenId,690		properties: impl Iterator<Item = Property>,691		is_token_create: bool,692		nesting_budget: &dyn Budget,693	) -> DispatchResult {694		Self::modify_token_properties(695			collection,696			sender,697			token_id,698			properties.map(|p| (p.key, Some(p.value))),699			is_token_create,700			nesting_budget,701		)702	}703704	/// Add or edit single property for the token705	///706	/// Calls [`set_token_properties`] internally707	///708	/// [`set_token_properties`]: struct.Pallet.html#method.set_token_properties709	pub fn set_token_property(710		collection: &NonfungibleHandle<T>,711		sender: &T::CrossAccountId,712		token_id: TokenId,713		property: Property,714		nesting_budget: &dyn Budget,715	) -> DispatchResult {716		let is_token_create = false;717718		Self::set_token_properties(719			collection,720			sender,721			token_id,722			[property].into_iter(),723			is_token_create,724			nesting_budget,725		)726	}727728	/// Batch operation to remove properties from the token729	///730	/// Same as [`modify_token_properties`] but doesn't allow to add or edit properties731	///732	/// [`modify_token_properties`]: struct.Pallet.html#method.modify_token_properties733	pub fn delete_token_properties(734		collection: &NonfungibleHandle<T>,735		sender: &T::CrossAccountId,736		token_id: TokenId,737		property_keys: impl Iterator<Item = PropertyKey>,738		nesting_budget: &dyn Budget,739	) -> DispatchResult {740		let is_token_create = false;741742		Self::modify_token_properties(743			collection,744			sender,745			token_id,746			property_keys.into_iter().map(|key| (key, None)),747			is_token_create,748			nesting_budget,749		)750	}751752	/// Remove single property from the token753	///754	/// Calls [`delete_token_properties`] internally755	///756	/// [`delete_token_properties`]: struct.Pallet.html#method.delete_token_properties757	pub fn delete_token_property(758		collection: &NonfungibleHandle<T>,759		sender: &T::CrossAccountId,760		token_id: TokenId,761		property_key: PropertyKey,762		nesting_budget: &dyn Budget,763	) -> DispatchResult {764		Self::delete_token_properties(765			collection,766			sender,767			token_id,768			[property_key].into_iter(),769			nesting_budget,770		)771	}772773	/// Add or edit properties for the collection774	pub fn set_collection_properties(775		collection: &NonfungibleHandle<T>,776		sender: &T::CrossAccountId,777		properties: Vec<Property>,778	) -> DispatchResult {779		<PalletCommon<T>>::set_collection_properties(collection, sender, properties)780	}781782	/// Remove properties from the collection783	pub fn delete_collection_properties(784		collection: &CollectionHandle<T>,785		sender: &T::CrossAccountId,786		property_keys: Vec<PropertyKey>,787	) -> DispatchResult {788		<PalletCommon<T>>::delete_collection_properties(collection, sender, property_keys)789	}790791	/// Set property permissions for the token.792	///793	/// Sender should be the owner or admin of token's collection.794	pub fn set_token_property_permissions(795		collection: &CollectionHandle<T>,796		sender: &T::CrossAccountId,797		property_permissions: Vec<PropertyKeyPermission>,798	) -> DispatchResult {799		<PalletCommon<T>>::set_token_property_permissions(collection, sender, property_permissions)800	}801802	/// Set property permissions for the token with scope.803	///804	/// Sender should be the owner or admin of token's collection.805	pub fn set_scoped_token_property_permissions(806		collection: &CollectionHandle<T>,807		sender: &T::CrossAccountId,808		scope: PropertyScope,809		property_permissions: Vec<PropertyKeyPermission>,810	) -> DispatchResult {811		<PalletCommon<T>>::set_scoped_token_property_permissions(812			collection,813			sender,814			scope,815			property_permissions,816		)817	}818819	/// Set property permissions for the collection.820	///821	/// Sender should be the owner or admin of the collection.822	pub fn set_property_permission(823		collection: &CollectionHandle<T>,824		sender: &T::CrossAccountId,825		permission: PropertyKeyPermission,826	) -> DispatchResult {827		<PalletCommon<T>>::set_property_permission(collection, sender, permission)828	}829830	pub fn check_token_immediate_ownership(831		collection: &NonfungibleHandle<T>,832		token: TokenId,833		possible_owner: &T::CrossAccountId,834	) -> DispatchResult {835		let token_data =836			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;837		ensure!(838			&token_data.owner == possible_owner,839			<CommonError<T>>::NoPermission840		);841		Ok(())842	}843844	/// Transfer NFT token from one account to another.845	///846	/// `from` account stops being the owner and `to` account becomes the owner of the token.847	/// If `to` is token than `to` becomes owner of the token and the token become nested.848	/// Unnests token from previous parent if it was nested before.849	/// Removes allowance for the token if there was any.850	/// Throws if transfers aren't allowed for collection or if receiver reached token ownership limit.851	///852	/// - `nesting_budget`: Limit for token nesting depth853	pub fn transfer(854		collection: &NonfungibleHandle<T>,855		from: &T::CrossAccountId,856		to: &T::CrossAccountId,857		token: TokenId,858		nesting_budget: &dyn Budget,859	) -> DispatchResult {860		ensure!(861			collection.limits.transfers_enabled(),862			<CommonError<T>>::TransferNotAllowed863		);864865		let token_data =866			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;867		ensure!(&token_data.owner == from, <CommonError<T>>::NoPermission);868869		if collection.permissions.access() == AccessMode::AllowList {870			collection.check_allowlist(from)?;871			collection.check_allowlist(to)?;872		}873		<PalletCommon<T>>::ensure_correct_receiver(to)?;874875		let balance_from = <AccountBalance<T>>::get((collection.id, from))876			.checked_sub(1)877			.ok_or(<CommonError<T>>::TokenValueTooLow)?;878		let balance_to = if from != to {879			let balance_to = <AccountBalance<T>>::get((collection.id, to))880				.checked_add(1)881				.ok_or(ArithmeticError::Overflow)?;882883			ensure!(884				balance_to < collection.limits.account_token_ownership_limit(),885				<CommonError<T>>::AccountTokenLimitExceeded,886			);887888			Some(balance_to)889		} else {890			None891		};892893		<PalletStructure<T>>::nest_if_sent_to_token(894			from.clone(),895			to,896			collection.id,897			token,898			nesting_budget,899		)?;900901		// =========902903		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);904905		<TokenData<T>>::insert(906			(collection.id, token),907			ItemData {908				owner: to.clone(),909				..token_data910			},911		);912913		if let Some(balance_to) = balance_to {914			// from != to915			if balance_from == 0 {916				<AccountBalance<T>>::remove((collection.id, from));917			} else {918				<AccountBalance<T>>::insert((collection.id, from), balance_from);919			}920			<AccountBalance<T>>::insert((collection.id, to), balance_to);921			<Owned<T>>::remove((collection.id, from, token));922			<Owned<T>>::insert((collection.id, to, token), true);923		}924		Self::set_allowance_unchecked(collection, from, token, None, true);925926		<PalletEvm<T>>::deposit_log(927			ERC721Events::Transfer {928				from: *from.as_eth(),929				to: *to.as_eth(),930				token_id: token.into(),931			}932			.to_log(collection_id_to_address(collection.id)),933		);934		<PalletCommon<T>>::deposit_event(CommonEvent::Transfer(935			collection.id,936			token,937			from.clone(),938			to.clone(),939			1,940		));941		Ok(())942	}943944	/// Batch operation to mint multiple NFT tokens.945	///946	/// The sender should be the owner/admin of the collection or collection should be configured947	/// to allow public minting.948	/// Throws if amount of tokens reached it's limit for the collection or if caller reached949	/// token ownership limit.950	///951	/// - `data`: Contains list of token properties and users who will become the owners of the952	///   corresponging tokens.953	/// - `nesting_budget`: Limit for token nesting depth954	pub fn create_multiple_items(955		collection: &NonfungibleHandle<T>,956		sender: &T::CrossAccountId,957		data: Vec<CreateItemData<T>>,958		nesting_budget: &dyn Budget,959	) -> DispatchResult {960		if !collection.is_owner_or_admin(sender) {961			ensure!(962				collection.permissions.mint_mode(),963				<CommonError<T>>::PublicMintingNotAllowed964			);965			collection.check_allowlist(sender)?;966967			for item in data.iter() {968				collection.check_allowlist(&item.owner)?;969			}970		}971972		for data in data.iter() {973			<PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;974		}975976		let first_token = <TokensMinted<T>>::get(collection.id);977		let tokens_minted = first_token978			.checked_add(data.len() as u32)979			.ok_or(ArithmeticError::Overflow)?;980		ensure!(981			tokens_minted <= collection.limits.token_limit(),982			<CommonError<T>>::CollectionTokenLimitExceeded983		);984985		let mut balances = BTreeMap::new();986		for data in &data {987			let balance = balances988				.entry(&data.owner)989				.or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));990			*balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;991992			ensure!(993				*balance <= collection.limits.account_token_ownership_limit(),994				<CommonError<T>>::AccountTokenLimitExceeded,995			);996		}997998		for (i, data) in data.iter().enumerate() {999			let token = TokenId(first_token + i as u32 + 1);10001001			<PalletStructure<T>>::check_nesting(1002				sender.clone(),1003				&data.owner,1004				collection.id,1005				token,1006				nesting_budget,1007			)?;1008		}10091010		// =========10111012		with_transaction(|| {1013			for (i, data) in data.iter().enumerate() {1014				let token = first_token + i as u32 + 1;10151016				<TokenData<T>>::insert(1017					(collection.id, token),1018					ItemData {1019						// const_data: data.const_data.clone(),1020						owner: data.owner.clone(),1021					},1022				);10231024				<PalletStructure<T>>::nest_if_sent_to_token_unchecked(1025					&data.owner,1026					collection.id,1027					TokenId(token),1028				);10291030				if let Err(e) = Self::set_token_properties(1031					collection,1032					sender,1033					TokenId(token),1034					data.properties.clone().into_iter(),1035					true,1036					nesting_budget,1037				) {1038					return TransactionOutcome::Rollback(Err(e));1039				}1040			}1041			TransactionOutcome::Commit(Ok(()))1042		})?;10431044		<TokensMinted<T>>::insert(collection.id, tokens_minted);1045		for (account, balance) in balances {1046			<AccountBalance<T>>::insert((collection.id, account), balance);1047		}1048		for (i, data) in data.into_iter().enumerate() {1049			let token = first_token + i as u32 + 1;1050			<Owned<T>>::insert((collection.id, &data.owner, token), true);10511052			<PalletEvm<T>>::deposit_log(1053				ERC721Events::Transfer {1054					from: H160::default(),1055					to: *data.owner.as_eth(),1056					token_id: token.into(),1057				}1058				.to_log(collection_id_to_address(collection.id)),1059			);1060			<PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(1061				collection.id,1062				TokenId(token),1063				data.owner.clone(),1064				1,1065			));1066		}1067		Ok(())1068	}10691070	pub fn set_allowance_unchecked(1071		collection: &NonfungibleHandle<T>,1072		sender: &T::CrossAccountId,1073		token: TokenId,1074		spender: Option<&T::CrossAccountId>,1075		assume_implicit_eth: bool,1076	) {1077		if let Some(spender) = spender {1078			let old_spender = <Allowance<T>>::get((collection.id, token));1079			<Allowance<T>>::insert((collection.id, token), spender);1080			// In ERC721 there is only one possible approved user of token, so we set1081			// approved user to spender1082			<PalletEvm<T>>::deposit_log(1083				ERC721Events::Approval {1084					owner: *sender.as_eth(),1085					approved: *spender.as_eth(),1086					token_id: token.into(),1087				}1088				.to_log(collection_id_to_address(collection.id)),1089			);1090			// In Unique chain, any token can have any amount of approved users, so we need to1091			// set allowance of old owner to 0, and allowance of new owner to 11092			if old_spender.as_ref() != Some(spender) {1093				if let Some(old_owner) = old_spender {1094					<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1095						collection.id,1096						token,1097						sender.clone(),1098						old_owner,1099						0,1100					));1101				}1102				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1103					collection.id,1104					token,1105					sender.clone(),1106					spender.clone(),1107					1,1108				));1109			}1110		} else {1111			let old_spender = <Allowance<T>>::take((collection.id, token));1112			if !assume_implicit_eth {1113				// In ERC721 there is only one possible approved user of token, so we set1114				// approved user to zero address1115				<PalletEvm<T>>::deposit_log(1116					ERC721Events::Approval {1117						owner: *sender.as_eth(),1118						approved: H160::default(),1119						token_id: token.into(),1120					}1121					.to_log(collection_id_to_address(collection.id)),1122				);1123			}1124			// In Unique chain, any token can have any amount of approved users, so we need to1125			// set allowance of old owner to 01126			if let Some(old_spender) = old_spender {1127				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1128					collection.id,1129					token,1130					sender.clone(),1131					old_spender,1132					0,1133				));1134			}1135		}1136	}11371138	/// Set allowance for the spender to `transfer` or `burn` sender's token.1139	///1140	/// - `token`: Token the spender is allowed to `transfer` or `burn`.1141	pub fn set_allowance(1142		collection: &NonfungibleHandle<T>,1143		sender: &T::CrossAccountId,1144		token: TokenId,1145		spender: Option<&T::CrossAccountId>,1146	) -> DispatchResult {1147		if collection.permissions.access() == AccessMode::AllowList {1148			collection.check_allowlist(sender)?;1149			if let Some(spender) = spender {1150				collection.check_allowlist(spender)?;1151			}1152		}11531154		if let Some(spender) = spender {1155			<PalletCommon<T>>::ensure_correct_receiver(spender)?;1156		}11571158		let token_data =1159			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;1160		if &token_data.owner != sender {1161			ensure!(1162				collection.ignores_owned_amount(sender),1163				<CommonError<T>>::CantApproveMoreThanOwned1164			);1165		}11661167		// =========11681169		Self::set_allowance_unchecked(collection, sender, token, spender, false);1170		Ok(())1171	}11721173	/// Checks allowance for the spender to use the token.1174	fn check_allowed(1175		collection: &NonfungibleHandle<T>,1176		spender: &T::CrossAccountId,1177		from: &T::CrossAccountId,1178		token: TokenId,1179		nesting_budget: &dyn Budget,1180	) -> DispatchResult {1181		if spender.conv_eq(from) {1182			return Ok(());1183		}1184		if collection.permissions.access() == AccessMode::AllowList {1185			// `from`, `to` checked in [`transfer`]1186			collection.check_allowlist(spender)?;1187		}11881189		if collection.limits.owner_can_transfer() && collection.is_owner_or_admin(spender) {1190			return Ok(());1191		}11921193		if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {1194			ensure!(1195				<PalletStructure<T>>::check_indirectly_owned(1196					spender.clone(),1197					source.0,1198					source.1,1199					None,1200					nesting_budget1201				)?,1202				<CommonError<T>>::ApprovedValueTooLow,1203			);1204			return Ok(());1205		}1206		if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {1207			return Ok(());1208		}1209		if <WalletOperator<T>>::get((collection.id, from, spender)) == Some(true) {1210			return Ok(());1211		}1212		ensure!(1213			collection.ignores_allowance(spender),1214			<CommonError<T>>::ApprovedValueTooLow1215		);1216		Ok(())1217	}12181219	/// Transfer NFT token from one account to another.1220	///1221	/// Same as the [`transfer`] but spender doesn't needs to be the owner of the token.1222	/// The owner should set allowance for the spender to transfer token.1223	///1224	/// [`transfer`]: struct.Pallet.html#method.transfer1225	pub fn transfer_from(1226		collection: &NonfungibleHandle<T>,1227		spender: &T::CrossAccountId,1228		from: &T::CrossAccountId,1229		to: &T::CrossAccountId,1230		token: TokenId,1231		nesting_budget: &dyn Budget,1232	) -> DispatchResult {1233		Self::check_allowed(collection, spender, from, token, nesting_budget)?;12341235		// =========12361237		// Allowance is reset in [`transfer`]1238		Self::transfer(collection, from, to, token, nesting_budget)1239	}12401241	/// Burn NFT token for `from` account.1242	///1243	/// Same as the [`burn`] but spender doesn't need to be an owner of the token. The owner should1244	/// set allowance for the spender to burn token.1245	///1246	/// [`burn`]: struct.Pallet.html#method.burn1247	pub fn burn_from(1248		collection: &NonfungibleHandle<T>,1249		spender: &T::CrossAccountId,1250		from: &T::CrossAccountId,1251		token: TokenId,1252		nesting_budget: &dyn Budget,1253	) -> DispatchResult {1254		Self::check_allowed(collection, spender, from, token, nesting_budget)?;12551256		// =========12571258		Self::burn(collection, from, token)1259	}12601261	/// Check that `from` token could be nested in `under` token.1262	///1263	pub fn check_nesting(1264		handle: &NonfungibleHandle<T>,1265		sender: T::CrossAccountId,1266		from: (CollectionId, TokenId),1267		under: TokenId,1268		nesting_budget: &dyn Budget,1269	) -> DispatchResult {1270		let nesting = handle.permissions.nesting();12711272		#[cfg(not(feature = "runtime-benchmarks"))]1273		let permissive = false;1274		#[cfg(feature = "runtime-benchmarks")]1275		let permissive = nesting.permissive;12761277		if permissive {1278			// Pass1279		} else if nesting.token_owner1280			&& <PalletStructure<T>>::check_indirectly_owned(1281				sender.clone(),1282				handle.id,1283				under,1284				Some(from),1285				nesting_budget,1286			)? {1287			// Pass1288		} else if nesting.collection_admin && handle.is_owner_or_admin(&sender) {1289			// Pass1290		} else {1291			fail!(<CommonError<T>>::UserIsNotAllowedToNest);1292		}12931294		if let Some(whitelist) = &nesting.restricted {1295			ensure!(1296				whitelist.contains(&from.0),1297				<CommonError<T>>::SourceCollectionIsNotAllowedToNest1298			);1299		}1300		Ok(())1301	}13021303	fn nest(under: (CollectionId, TokenId), to_nest: (CollectionId, TokenId)) {1304		<TokenChildren<T>>::insert((under.0, under.1, (to_nest.0, to_nest.1)), true);1305	}13061307	fn unnest(under: (CollectionId, TokenId), to_unnest: (CollectionId, TokenId)) {1308		<TokenChildren<T>>::remove((under.0, under.1, to_unnest));1309	}13101311	fn collection_has_tokens(collection_id: CollectionId) -> bool {1312		<TokenData<T>>::iter_prefix((collection_id,))1313			.next()1314			.is_some()1315	}13161317	fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {1318		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1319			.next()1320			.is_some()1321	}13221323	pub fn token_children_ids(collection_id: CollectionId, token_id: TokenId) -> Vec<TokenChild> {1324		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1325			.map(|((child_collection_id, child_id), _)| TokenChild {1326				collection: child_collection_id,1327				token: child_id,1328			})1329			.collect()1330	}13311332	/// Mint single NFT token.1333	///1334	/// Delegated to [`create_multiple_items`]1335	///1336	/// [`create_multiple_items`]: struct.Pallet.html#method.create_multiple_items1337	pub fn create_item(1338		collection: &NonfungibleHandle<T>,1339		sender: &T::CrossAccountId,1340		data: CreateItemData<T>,1341		nesting_budget: &dyn Budget,1342	) -> DispatchResult {1343		Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1344	}13451346	/// Sets or unsets the approval of a given operator.1347	///1348	/// An operator is allowed to transfer all token pieces of the sender on their behalf.1349	/// - `owner`: Token owner1350	/// - `operator`: Operator1351	/// - `approve`: Is operator enabled or disabled1352	pub fn set_approval_for_all(1353		collection: &NonfungibleHandle<T>,1354		owner: &T::CrossAccountId,1355		operator: &T::CrossAccountId,1356		approve: bool,1357	) -> DispatchResult {1358		if collection.permissions.access() == AccessMode::AllowList {1359			collection.check_allowlist(owner)?;1360			collection.check_allowlist(operator)?;1361		}13621363		<PalletCommon<T>>::ensure_correct_receiver(operator)?;13641365		// =========13661367		<WalletOperator<T>>::insert((collection.id, owner, operator), approve);1368		<PalletEvm<T>>::deposit_log(1369			ERC721Events::ApprovalForAll {1370				owner: *owner.as_eth(),1371				operator: *operator.as_eth(),1372				approved: approve,1373			}1374			.to_log(collection_id_to_address(collection.id)),1375		);1376		<PalletCommon<T>>::deposit_event(CommonEvent::ApprovedForAll(1377			collection.id,1378			owner.clone(),1379			operator.clone(),1380			approve,1381		));1382		Ok(())1383	}13841385	/// Tells whether an operator is approved by a given owner.1386	pub fn is_approved_for_all(1387		collection: &NonfungibleHandle<T>,1388		owner: &T::CrossAccountId,1389		operator: &T::CrossAccountId,1390	) -> bool {1391		<WalletOperator<T>>::get((collection.id, owner, operator)).unwrap_or(false)1392	}1393}