git.delta.rocks / unique-network / refs/commits / 9e201d9e49be

difftreelog

Cargo fmt

Trubnikov Sergey2022-06-03parent: #85427bc.patch.diff
in: master

3 files changed

modifiedpallets/common/src/erc.rsdiffbeforeafterboth
--- a/pallets/common/src/erc.rs
+++ b/pallets/common/src/erc.rs
@@ -47,8 +47,8 @@
 }
 
 #[solidity_interface(name = "Collection")]
-impl<T: Config> CollectionHandle<T> 
-// where 
+impl<T: Config> CollectionHandle<T>
+// where
 // 	T::AccountId: From<H256>
 {
 	fn set_collection_property(&mut self, caller: caller, key: string, value: bytes) -> Result<()> {
@@ -207,8 +207,7 @@
 		self.check_is_owner_or_admin(&caller)
 			.map_err(dispatch_to_evm::<T>)?;
 		let admin = T::CrossAccountId::from_eth(admin);
-		<Pallet<T>>::toggle_admin(&self, &caller, &admin, false)
-			.map_err(dispatch_to_evm::<T>)?;
+		<Pallet<T>>::toggle_admin(&self, &caller, &admin, false).map_err(dispatch_to_evm::<T>)?;
 		Ok(())
 	}
 
@@ -226,12 +225,21 @@
 	}
 
 	#[solidity(rename_selector = "setCollectionNesting")]
-	fn set_nesting(&mut self, caller: caller, enable: bool, collections: Vec<address>) -> Result<void> {
+	fn set_nesting(
+		&mut self,
+		caller: caller,
+		enable: bool,
+		collections: Vec<address>,
+	) -> Result<void> {
 		if collections.is_empty() {
 			return Err("No addresses provided".into());
 		}
 		if collections.len() >= OwnerRestrictedSet::bound() {
-			return Err(Error::Revert(format!("Out of bound: {} >= {}", collections.len(), OwnerRestrictedSet::bound())));
+			return Err(Error::Revert(format!(
+				"Out of bound: {} >= {}",
+				collections.len(),
+				OwnerRestrictedSet::bound()
+			)));
 		}
 		let caller = T::CrossAccountId::from_eth(caller);
 		self.check_is_owner_or_admin(&caller)
@@ -241,12 +249,12 @@
 			true => {
 				let mut bv = OwnerRestrictedSet::new();
 				for i in collections {
-					bv.try_insert(
-						crate::eth::map_eth_to_id(&i)
-							.ok_or(Error::Revert("Can't convert address into collection id".into()))?
-					).map_err(|e| Error::Revert(format!("{:?}", e)))?;
+					bv.try_insert(crate::eth::map_eth_to_id(&i).ok_or(Error::Revert(
+						"Can't convert address into collection id".into(),
+					))?)
+					.map_err(|e| Error::Revert(format!("{:?}", e)))?;
 				}
-				NestingRule::OwnerRestricted (bv)
+				NestingRule::OwnerRestricted(bv)
 			}
 		});
 		save(self);
@@ -269,16 +277,14 @@
 	fn add_to_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {
 		let caller = check_is_owner_or_admin(caller, self)?;
 		let user = T::CrossAccountId::from_eth(user);
-		<Pallet<T>>::toggle_allowlist(self, &caller, &user, true)
-			.map_err(dispatch_to_evm::<T>)?;
+		<Pallet<T>>::toggle_allowlist(self, &caller, &user, true).map_err(dispatch_to_evm::<T>)?;
 		Ok(())
 	}
 
 	fn remove_from_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {
 		let caller = check_is_owner_or_admin(caller, self)?;
 		let user = T::CrossAccountId::from_eth(user);
-		<Pallet<T>>::toggle_allowlist(self, &caller, &user, false)
-			.map_err(dispatch_to_evm::<T>)?;
+		<Pallet<T>>::toggle_allowlist(self, &caller, &user, false).map_err(dispatch_to_evm::<T>)?;
 		Ok(())
 	}
 
@@ -298,7 +304,10 @@
 	Ok(())
 }
 
-fn check_is_owner_or_admin<T: Config>(caller: caller, collection: &CollectionHandle<T>) -> Result<T::CrossAccountId> {
+fn check_is_owner_or_admin<T: Config>(
+	caller: caller,
+	collection: &CollectionHandle<T>,
+) -> Result<T::CrossAccountId> {
 	let caller = T::CrossAccountId::from_eth(caller);
 	collection
 		.check_is_owner_or_admin(&caller)
modifiedpallets/nonfungible/src/erc.rsdiffbeforeafterboth
--- a/pallets/nonfungible/src/erc.rs
+++ b/pallets/nonfungible/src/erc.rs
@@ -82,8 +82,14 @@
 			.map_err(|_| "key too long")?;
 		let value = value.try_into().map_err(|_| "value too long")?;
 
-		<Pallet<T>>::set_token_property(self, &caller, TokenId(token_id), Property { key, value }, false)
-			.map_err(dispatch_to_evm::<T>)
+		<Pallet<T>>::set_token_property(
+			self,
+			&caller,
+			TokenId(token_id),
+			Property { key, value },
+			false,
+		)
+		.map_err(dispatch_to_evm::<T>)
 	}
 
 	fn delete_property(&mut self, token_id: uint256, caller: caller, key: string) -> Result<()> {
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#![cfg_attr(not(feature = "std"), no_std)]1819use erc::ERC721Events;20use evm_coder::ToLog;21use frame_support::{22	BoundedVec, ensure, fail, transactional,23	storage::with_transaction,24	pallet_prelude::DispatchResultWithPostInfo,25	pallet_prelude::Weight,26	weights::{PostDispatchInfo, Pays},27};28use up_data_structs::{29	AccessMode, CollectionId, CustomDataLimit, TokenId, CreateCollectionData, CreateNftExData,30	mapping::TokenAddressMapping, NestingRule, budget::Budget, Property, PropertyPermission,31	PropertyKey, PropertyKeyPermission, Properties, PropertyScope, TrySetProperty, TokenChild,32};33use pallet_evm::{account::CrossAccountId, Pallet as PalletEvm};34use pallet_common::{35	Error as CommonError, Pallet as PalletCommon, Event as CommonEvent, CollectionHandle,36	eth::collection_id_to_address,37};38use pallet_structure::{Pallet as PalletStructure, Error as StructureError};39use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};40use sp_core::H160;41use sp_runtime::{ArithmeticError, DispatchError, DispatchResult, TransactionOutcome};42use sp_std::{vec::Vec, vec, collections::btree_map::BTreeMap, collections::btree_set::BTreeSet};43use core::ops::Deref;44use codec::{Encode, Decode, MaxEncodedLen};45use scale_info::TypeInfo;4647pub use pallet::*;48use weights::WeightInfo;49#[cfg(feature = "runtime-benchmarks")]50pub mod benchmarking;51pub mod common;52pub mod erc;53pub mod weights;5455pub type CreateItemData<T> = CreateNftExData<<T as pallet_evm::account::Config>::CrossAccountId>;56pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;5758#[struct_versioning::versioned(version = 2, upper)]59#[derive(Encode, Decode, TypeInfo, MaxEncodedLen)]60pub struct ItemData<CrossAccountId> {61	#[version(..2)]62	pub const_data: BoundedVec<u8, CustomDataLimit>,6364	#[version(..2)]65	pub variable_data: BoundedVec<u8, CustomDataLimit>,6667	pub owner: CrossAccountId,68}6970#[frame_support::pallet]71pub mod pallet {72	use super::*;73	use frame_support::{74		Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key, traits::StorageVersion,75	};76	use frame_system::pallet_prelude::*;77	use up_data_structs::{CollectionId, TokenId};78	use super::weights::WeightInfo;7980	#[pallet::error]81	pub enum Error<T> {82		/// Not Nonfungible item data used to mint in Nonfungible collection.83		NotNonfungibleDataUsedToMintFungibleCollectionToken,84		/// Used amount > 1 with NFT85		NonfungibleItemsHaveNoAmount,86		/// Unable to burn NFT with children87		CantBurnNftWithChildren,88	}8990	#[pallet::config]91	pub trait Config:92		frame_system::Config + pallet_common::Config + pallet_structure::Config + pallet_evm::Config93	{94		type WeightInfo: WeightInfo;95	}9697	const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);9899	#[pallet::pallet]100	#[pallet::storage_version(STORAGE_VERSION)]101	#[pallet::generate_store(pub(super) trait Store)]102	pub struct Pallet<T>(_);103104	#[pallet::storage]105	pub type TokensMinted<T: Config> =106		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;107	#[pallet::storage]108	pub type TokensBurnt<T: Config> =109		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;110111	#[pallet::storage]112	pub type TokenData<T: Config> = StorageNMap<113		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),114		Value = ItemData<T::CrossAccountId>,115		QueryKind = OptionQuery,116	>;117118	#[pallet::storage]119	#[pallet::getter(fn token_properties)]120	pub type TokenProperties<T: Config> = StorageNMap<121		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),122		Value = Properties,123		QueryKind = ValueQuery,124		OnEmpty = up_data_structs::TokenProperties,125	>;126127	/// Used to enumerate tokens owned by account128	#[pallet::storage]129	pub type Owned<T: Config> = StorageNMap<130		Key = (131			Key<Twox64Concat, CollectionId>,132			Key<Blake2_128Concat, T::CrossAccountId>,133			Key<Twox64Concat, TokenId>,134		),135		Value = bool,136		QueryKind = ValueQuery,137	>;138139	/// Used to enumerate token's children140	#[pallet::storage]141	#[pallet::getter(fn token_children)]142	pub type TokenChildren<T: Config> = StorageNMap<143		Key = (144			Key<Twox64Concat, CollectionId>,145			Key<Twox64Concat, TokenId>,146			Key<Twox64Concat, (CollectionId, TokenId)>,147		),148		Value = bool,149		QueryKind = ValueQuery,150	>;151152	#[pallet::storage]153	pub type AccountBalance<T: Config> = StorageNMap<154		Key = (155			Key<Twox64Concat, CollectionId>,156			Key<Blake2_128Concat, T::CrossAccountId>,157		),158		Value = u32,159		QueryKind = ValueQuery,160	>;161162	#[pallet::storage]163	pub type Allowance<T: Config> = StorageNMap<164		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),165		Value = T::CrossAccountId,166		QueryKind = OptionQuery,167	>;168169	#[pallet::hooks]170	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {171		fn on_runtime_upgrade() -> Weight {172			if StorageVersion::get::<Pallet<T>>() < StorageVersion::new(1) {173				let mut had_consts = BTreeSet::new();174				<TokenData<T>>::translate::<ItemDataVersion1<T::CrossAccountId>, _>(175					|(collection, token), v| {176						let mut props = vec![];177						if !v.const_data.is_empty() {178							props.push(Property {179								key: b"_old_constData".to_vec().try_into().unwrap(),180								value: v181									.const_data182									.clone()183									.into_inner()184									.try_into()185									.expect("const too long"),186							});187							had_consts.insert(collection);188						}189						if !v.variable_data.is_empty() {190							props.push(Property {191								key: b"_old_variableData".to_vec().try_into().unwrap(),192								value: v193									.variable_data194									.clone()195									.into_inner()196									.try_into()197									.expect("variable too long"),198							})199						}200						if !props.is_empty() {201							Self::set_scoped_token_properties(202								collection,203								token,204								PropertyScope::None,205								props.into_iter(),206							)207							.expect("existing token data exceeds property storage");208						}209						Some(<ItemDataVersion2<T::CrossAccountId>>::from(v))210					},211				);212				for collection in had_consts {213					<PalletCommon<T>>::set_property_permission_unchecked(214						collection,215						PropertyKeyPermission {216							key: b"_old_constData".to_vec().try_into().unwrap(),217							permission: PropertyPermission {218								mutable: false,219								collection_admin: true,220								token_owner: false,221							},222						},223					)224					.expect("failed to configure permission");225				}226			}227228			0229		}230	}231}232233pub struct NonfungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);234impl<T: Config> NonfungibleHandle<T> {235	pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {236		Self(inner)237	}238	pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {239		self.0240	}241	pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {242		&mut self.0243	}244}245impl<T: Config> WithRecorder<T> for NonfungibleHandle<T> {246	fn recorder(&self) -> &SubstrateRecorder<T> {247		self.0.recorder()248	}249	fn into_recorder(self) -> SubstrateRecorder<T> {250		self.0.into_recorder()251	}252}253impl<T: Config> Deref for NonfungibleHandle<T> {254	type Target = pallet_common::CollectionHandle<T>;255256	fn deref(&self) -> &Self::Target {257		&self.0258	}259}260261impl<T: Config> Pallet<T> {262	pub fn total_supply(collection: &NonfungibleHandle<T>) -> u32 {263		<TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)264	}265	pub fn token_exists(collection: &NonfungibleHandle<T>, token: TokenId) -> bool {266		<TokenData<T>>::contains_key((collection.id, token))267	}268269	pub fn set_scoped_token_property(270		collection_id: CollectionId,271		token_id: TokenId,272		scope: PropertyScope,273		property: Property,274	) -> DispatchResult {275		TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {276			properties.try_scoped_set(scope, property.key, property.value)277		})278		.map_err(<CommonError<T>>::from)?;279280		Ok(())281	}282283	pub fn set_scoped_token_properties(284		collection_id: CollectionId,285		token_id: TokenId,286		scope: PropertyScope,287		properties: impl Iterator<Item = Property>,288	) -> DispatchResult {289		TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {290			stored_properties.try_scoped_set_from_iter(scope, properties)291		})292		.map_err(<CommonError<T>>::from)?;293294		Ok(())295	}296297	pub fn current_token_id(collection_id: CollectionId) -> TokenId {298		TokenId(<TokensMinted<T>>::get(collection_id))299	}300}301302// unchecked calls skips any permission checks303impl<T: Config> Pallet<T> {304	pub fn init_collection(305		owner: T::CrossAccountId,306		data: CreateCollectionData<T::AccountId>,307	) -> Result<CollectionId, DispatchError> {308		<PalletCommon<T>>::init_collection(owner, data)309	}310	pub fn destroy_collection(311		collection: NonfungibleHandle<T>,312		sender: &T::CrossAccountId,313	) -> DispatchResult {314		let id = collection.id;315316		if Self::collection_has_tokens(id) {317			return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());318		}319320		// =========321322		PalletCommon::destroy_collection(collection.0, sender)?;323324		<TokenData<T>>::remove_prefix((id,), None);325		<TokenChildren<T>>::remove_prefix((id,), None);326		<Owned<T>>::remove_prefix((id,), None);327		<TokensMinted<T>>::remove(id);328		<TokensBurnt<T>>::remove(id);329		<Allowance<T>>::remove_prefix((id,), None);330		<AccountBalance<T>>::remove_prefix((id,), None);331		Ok(())332	}333334	pub fn burn(335		collection: &NonfungibleHandle<T>,336		sender: &T::CrossAccountId,337		token: TokenId,338	) -> DispatchResult {339		let token_data =340			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;341		ensure!(342			&token_data.owner == sender343				|| (collection.limits.owner_can_transfer() && collection.is_owner_or_admin(sender)),344			<CommonError<T>>::NoPermission345		);346347		if collection.permissions.access() == AccessMode::AllowList {348			collection.check_allowlist(sender)?;349		}350351		if Self::token_has_children(collection.id, token) {352			return Err(<Error<T>>::CantBurnNftWithChildren.into());353		}354355		let burnt = <TokensBurnt<T>>::get(collection.id)356			.checked_add(1)357			.ok_or(ArithmeticError::Overflow)?;358359		let balance = <AccountBalance<T>>::get((collection.id, token_data.owner.clone()))360			.checked_sub(1)361			.ok_or(ArithmeticError::Overflow)?;362363		// =========364365		if balance == 0 {366			<AccountBalance<T>>::remove((collection.id, token_data.owner.clone()));367		} else {368			<AccountBalance<T>>::insert((collection.id, token_data.owner.clone()), balance);369		}370371		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);372373		<Owned<T>>::remove((collection.id, &token_data.owner, token));374		<TokensBurnt<T>>::insert(collection.id, burnt);375		<TokenData<T>>::remove((collection.id, token));376		<TokenProperties<T>>::remove((collection.id, token));377		let old_spender = <Allowance<T>>::take((collection.id, token));378379		if let Some(old_spender) = old_spender {380			<PalletCommon<T>>::deposit_event(CommonEvent::Approved(381				collection.id,382				token,383				token_data.owner.clone(),384				old_spender,385				0,386			));387		}388389		<PalletEvm<T>>::deposit_log(390			ERC721Events::Transfer {391				from: *token_data.owner.as_eth(),392				to: H160::default(),393				token_id: token.into(),394			}395			.to_log(collection_id_to_address(collection.id)),396		);397		<PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(398			collection.id,399			token,400			token_data.owner,401			1,402		));403		Ok(())404	}405406	#[transactional]407	pub fn burn_recursively(408		collection: &NonfungibleHandle<T>,409		sender: &T::CrossAccountId,410		token: TokenId,411		self_budget: &dyn Budget,412		breadth_budget: &dyn Budget,413	) -> DispatchResultWithPostInfo {414		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);415416		let current_token_account =417			T::CrossTokenAddressMapping::token_to_address(collection.id, token);418419		let mut weight = 0 as Weight;420421		// This method is transactional, if user in fact doesn't have permissions to remove token -422		// tokens removed here will be restored after rejected transaction423		for ((collection, token), _) in <TokenChildren<T>>::iter_prefix((collection.id, token)) {424			ensure!(breadth_budget.consume(), <StructureError<T>>::BreadthLimit,);425			let PostDispatchInfo { actual_weight, .. } =426				<PalletStructure<T>>::burn_item_recursively(427					current_token_account.clone(),428					collection,429					token,430					self_budget,431					breadth_budget,432				)?;433			if let Some(actual_weight) = actual_weight {434				weight = weight.saturating_add(actual_weight);435			}436		}437438		Self::burn(collection, sender, token)?;439		DispatchResultWithPostInfo::Ok(PostDispatchInfo {440			actual_weight: Some(weight + <SelfWeightOf<T>>::burn_item()),441			pays_fee: Pays::Yes,442		})443	}444445	pub fn set_token_property(446		collection: &NonfungibleHandle<T>,447		sender: &T::CrossAccountId,448		token_id: TokenId,449		property: Property,450		is_token_create: bool,451	) -> DispatchResult {452		Self::check_token_change_permission(collection, sender, token_id, &property.key, is_token_create)?;453454		<TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {455			let property = property.clone();456			properties.try_set(property.key, property.value)457		})458		.map_err(<CommonError<T>>::from)?;459460		<PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertySet(461			collection.id,462			token_id,463			property.key,464		));465466		Ok(())467	}468469	#[transactional]470	pub fn set_token_properties(471		collection: &NonfungibleHandle<T>,472		sender: &T::CrossAccountId,473		token_id: TokenId,474		properties: Vec<Property>,475		is_token_create: bool,476	) -> DispatchResult {477		for property in properties {478			Self::set_token_property(collection, sender, token_id, property, is_token_create)?;479		}480481		Ok(())482	}483484	pub fn delete_token_property(485		collection: &NonfungibleHandle<T>,486		sender: &T::CrossAccountId,487		token_id: TokenId,488		property_key: PropertyKey,489	) -> DispatchResult {490		Self::check_token_change_permission(collection, sender, token_id, &property_key, false)?;491492		<TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {493			properties.remove(&property_key)494		})495		.map_err(<CommonError<T>>::from)?;496497		<PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertyDeleted(498			collection.id,499			token_id,500			property_key,501		));502503		Ok(())504	}505506	fn check_token_change_permission(507		collection: &NonfungibleHandle<T>,508		sender: &T::CrossAccountId,509		token_id: TokenId,510		property_key: &PropertyKey,511		is_token_create: bool,512	) -> DispatchResult {513		let permission = <PalletCommon<T>>::property_permissions(collection.id)514			.get(property_key)515			.cloned()516			.unwrap_or_else(PropertyPermission::none);517518		let token_data = <TokenData<T>>::get((collection.id, token_id))519			.ok_or(<CommonError<T>>::TokenNotFound)?;520521		let check_token_owner = || -> DispatchResult {522			ensure!(&token_data.owner == sender, <CommonError<T>>::NoPermission);523			Ok(())524		};525526		let is_property_exists = TokenProperties::<T>::get((collection.id, token_id))527			.get(property_key)528			.is_some();529530		match permission {531			PropertyPermission { mutable: false, .. } if is_property_exists => {532				Err(<CommonError<T>>::NoPermission.into())533			}534535			PropertyPermission {536				collection_admin,537				token_owner,538				..539			} => {540				if is_token_create && (collection_admin || token_owner) {541					return Ok(());542				}543544				let mut check_result = Err(<CommonError<T>>::NoPermission.into());545546				if collection_admin {547					check_result = collection.check_is_owner_or_admin(sender);548				}549550				if token_owner {551					check_result.or_else(|_| check_token_owner())552				} else {553					check_result554				}555			}556		}557	}558559	#[transactional]560	pub fn delete_token_properties(561		collection: &NonfungibleHandle<T>,562		sender: &T::CrossAccountId,563		token_id: TokenId,564		property_keys: Vec<PropertyKey>,565	) -> DispatchResult {566		for key in property_keys {567			Self::delete_token_property(collection, sender, token_id, key)?;568		}569570		Ok(())571	}572573	pub fn set_collection_properties(574		collection: &NonfungibleHandle<T>,575		sender: &T::CrossAccountId,576		properties: Vec<Property>,577	) -> DispatchResult {578		<PalletCommon<T>>::set_collection_properties(collection, sender, properties)579	}580581	pub fn delete_collection_properties(582		collection: &CollectionHandle<T>,583		sender: &T::CrossAccountId,584		property_keys: Vec<PropertyKey>,585	) -> DispatchResult {586		<PalletCommon<T>>::delete_collection_properties(collection, sender, property_keys)587	}588589	pub fn set_property_permissions(590		collection: &CollectionHandle<T>,591		sender: &T::CrossAccountId,592		property_permissions: Vec<PropertyKeyPermission>,593	) -> DispatchResult {594		<PalletCommon<T>>::set_property_permissions(collection, sender, property_permissions)595	}596597	pub fn set_property_permission(598		collection: &CollectionHandle<T>,599		sender: &T::CrossAccountId,600		permission: PropertyKeyPermission,601	) -> DispatchResult {602		<PalletCommon<T>>::set_property_permission(collection, sender, permission)603	}604605	pub fn transfer(606		collection: &NonfungibleHandle<T>,607		from: &T::CrossAccountId,608		to: &T::CrossAccountId,609		token: TokenId,610		nesting_budget: &dyn Budget,611	) -> DispatchResult {612		ensure!(613			collection.limits.transfers_enabled(),614			<CommonError<T>>::TransferNotAllowed615		);616617		let token_data =618			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;619		// TODO: require sender to be token, owner, require admins to go through transfer_from620		ensure!(621			&token_data.owner == from622				|| (collection.limits.owner_can_transfer() && collection.is_owner_or_admin(from)),623			<CommonError<T>>::NoPermission624		);625626		if collection.permissions.access() == AccessMode::AllowList {627			collection.check_allowlist(from)?;628			collection.check_allowlist(to)?;629		}630		<PalletCommon<T>>::ensure_correct_receiver(to)?;631632		let balance_from = <AccountBalance<T>>::get((collection.id, from))633			.checked_sub(1)634			.ok_or(<CommonError<T>>::TokenValueTooLow)?;635		let balance_to = if from != to {636			let balance_to = <AccountBalance<T>>::get((collection.id, to))637				.checked_add(1)638				.ok_or(ArithmeticError::Overflow)?;639640			ensure!(641				balance_to < collection.limits.account_token_ownership_limit(),642				<CommonError<T>>::AccountTokenLimitExceeded,643			);644645			Some(balance_to)646		} else {647			None648		};649650		<PalletStructure<T>>::nest_if_sent_to_token(651			from.clone(),652			to,653			collection.id,654			token,655			nesting_budget,656		)?;657658		// =========659660		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);661662		<TokenData<T>>::insert(663			(collection.id, token),664			ItemData {665				owner: to.clone(),666				..token_data667			},668		);669670		if let Some(balance_to) = balance_to {671			// from != to672			if balance_from == 0 {673				<AccountBalance<T>>::remove((collection.id, from));674			} else {675				<AccountBalance<T>>::insert((collection.id, from), balance_from);676			}677			<AccountBalance<T>>::insert((collection.id, to), balance_to);678			<Owned<T>>::remove((collection.id, from, token));679			<Owned<T>>::insert((collection.id, to, token), true);680		}681		Self::set_allowance_unchecked(collection, from, token, None, true);682683		<PalletEvm<T>>::deposit_log(684			ERC721Events::Transfer {685				from: *from.as_eth(),686				to: *to.as_eth(),687				token_id: token.into(),688			}689			.to_log(collection_id_to_address(collection.id)),690		);691		<PalletCommon<T>>::deposit_event(CommonEvent::Transfer(692			collection.id,693			token,694			from.clone(),695			to.clone(),696			1,697		));698		Ok(())699	}700701	pub fn create_multiple_items(702		collection: &NonfungibleHandle<T>,703		sender: &T::CrossAccountId,704		data: Vec<CreateItemData<T>>,705		nesting_budget: &dyn Budget,706	) -> DispatchResult {707		if !collection.is_owner_or_admin(sender) {708			ensure!(709				collection.permissions.mint_mode(),710				<CommonError<T>>::PublicMintingNotAllowed711			);712			collection.check_allowlist(sender)?;713714			for item in data.iter() {715				collection.check_allowlist(&item.owner)?;716			}717		}718719		for data in data.iter() {720			<PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;721		}722723		let first_token = <TokensMinted<T>>::get(collection.id);724		let tokens_minted = first_token725			.checked_add(data.len() as u32)726			.ok_or(ArithmeticError::Overflow)?;727		ensure!(728			tokens_minted <= collection.limits.token_limit(),729			<CommonError<T>>::CollectionTokenLimitExceeded730		);731732		let mut balances = BTreeMap::new();733		for data in &data {734			let balance = balances735				.entry(&data.owner)736				.or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));737			*balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;738739			ensure!(740				*balance <= collection.limits.account_token_ownership_limit(),741				<CommonError<T>>::AccountTokenLimitExceeded,742			);743		}744745		for (i, data) in data.iter().enumerate() {746			let token = TokenId(first_token + i as u32 + 1);747748			<PalletStructure<T>>::check_nesting(749				sender.clone(),750				&data.owner,751				collection.id,752				token,753				nesting_budget,754			)?;755		}756757		// =========758759		with_transaction(|| {760			for (i, data) in data.iter().enumerate() {761				let token = first_token + i as u32 + 1;762763				<TokenData<T>>::insert(764					(collection.id, token),765					ItemData {766						// const_data: data.const_data.clone(),767						owner: data.owner.clone(),768					},769				);770771				<PalletStructure<T>>::nest_if_sent_to_token_unchecked(772					&data.owner,773					collection.id,774					TokenId(token),775				);776777				if let Err(e) = Self::set_token_properties(778					collection,779					sender,780					TokenId(token),781					data.properties.clone().into_inner(),782					true,783				) {784					return TransactionOutcome::Rollback(Err(e));785				}786			}787			TransactionOutcome::Commit(Ok(()))788		})?;789790		<TokensMinted<T>>::insert(collection.id, tokens_minted);791		for (account, balance) in balances {792			<AccountBalance<T>>::insert((collection.id, account), balance);793		}794		for (i, data) in data.into_iter().enumerate() {795			let token = first_token + i as u32 + 1;796			<Owned<T>>::insert((collection.id, &data.owner, token), true);797798			<PalletEvm<T>>::deposit_log(799				ERC721Events::Transfer {800					from: H160::default(),801					to: *data.owner.as_eth(),802					token_id: token.into(),803				}804				.to_log(collection_id_to_address(collection.id)),805			);806			<PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(807				collection.id,808				TokenId(token),809				data.owner.clone(),810				1,811			));812		}813		Ok(())814	}815816	pub fn set_allowance_unchecked(817		collection: &NonfungibleHandle<T>,818		sender: &T::CrossAccountId,819		token: TokenId,820		spender: Option<&T::CrossAccountId>,821		assume_implicit_eth: bool,822	) {823		if let Some(spender) = spender {824			let old_spender = <Allowance<T>>::get((collection.id, token));825			<Allowance<T>>::insert((collection.id, token), spender);826			// In ERC721 there is only one possible approved user of token, so we set827			// approved user to spender828			<PalletEvm<T>>::deposit_log(829				ERC721Events::Approval {830					owner: *sender.as_eth(),831					approved: *spender.as_eth(),832					token_id: token.into(),833				}834				.to_log(collection_id_to_address(collection.id)),835			);836			// In Unique chain, any token can have any amount of approved users, so we need to837			// set allowance of old owner to 0, and allowance of new owner to 1838			if old_spender.as_ref() != Some(spender) {839				if let Some(old_owner) = old_spender {840					<PalletCommon<T>>::deposit_event(CommonEvent::Approved(841						collection.id,842						token,843						sender.clone(),844						old_owner,845						0,846					));847				}848				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(849					collection.id,850					token,851					sender.clone(),852					spender.clone(),853					1,854				));855			}856		} else {857			let old_spender = <Allowance<T>>::take((collection.id, token));858			if !assume_implicit_eth {859				// In ERC721 there is only one possible approved user of token, so we set860				// approved user to zero address861				<PalletEvm<T>>::deposit_log(862					ERC721Events::Approval {863						owner: *sender.as_eth(),864						approved: H160::default(),865						token_id: token.into(),866					}867					.to_log(collection_id_to_address(collection.id)),868				);869			}870			// In Unique chain, any token can have any amount of approved users, so we need to871			// set allowance of old owner to 0872			if let Some(old_spender) = old_spender {873				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(874					collection.id,875					token,876					sender.clone(),877					old_spender,878					0,879				));880			}881		}882	}883884	pub fn set_allowance(885		collection: &NonfungibleHandle<T>,886		sender: &T::CrossAccountId,887		token: TokenId,888		spender: Option<&T::CrossAccountId>,889	) -> DispatchResult {890		if collection.permissions.access() == AccessMode::AllowList {891			collection.check_allowlist(sender)?;892			if let Some(spender) = spender {893				collection.check_allowlist(spender)?;894			}895		}896897		if let Some(spender) = spender {898			<PalletCommon<T>>::ensure_correct_receiver(spender)?;899		}900901		let token_data =902			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;903		if &token_data.owner != sender {904			ensure!(905				collection.ignores_owned_amount(sender),906				<CommonError<T>>::CantApproveMoreThanOwned907			);908		}909910		// =========911912		Self::set_allowance_unchecked(collection, sender, token, spender, false);913		Ok(())914	}915916	fn check_allowed(917		collection: &NonfungibleHandle<T>,918		spender: &T::CrossAccountId,919		from: &T::CrossAccountId,920		token: TokenId,921		nesting_budget: &dyn Budget,922	) -> DispatchResult {923		if spender.conv_eq(from) {924			return Ok(());925		}926		if collection.permissions.access() == AccessMode::AllowList {927			// `from`, `to` checked in [`transfer`]928			collection.check_allowlist(spender)?;929		}930		if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {931			// TODO: should collection owner be allowed to perform this transfer?932			ensure!(933				<PalletStructure<T>>::check_indirectly_owned(934					spender.clone(),935					source.0,936					source.1,937					None,938					nesting_budget939				)?,940				<CommonError<T>>::ApprovedValueTooLow,941			);942			return Ok(());943		}944		if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {945			return Ok(());946		}947		ensure!(948			collection.ignores_allowance(spender),949			<CommonError<T>>::ApprovedValueTooLow950		);951		Ok(())952	}953954	pub fn transfer_from(955		collection: &NonfungibleHandle<T>,956		spender: &T::CrossAccountId,957		from: &T::CrossAccountId,958		to: &T::CrossAccountId,959		token: TokenId,960		nesting_budget: &dyn Budget,961	) -> DispatchResult {962		Self::check_allowed(collection, spender, from, token, nesting_budget)?;963964		// =========965966		// Allowance is reset in [`transfer`]967		Self::transfer(collection, from, to, token, nesting_budget)968	}969970	pub fn burn_from(971		collection: &NonfungibleHandle<T>,972		spender: &T::CrossAccountId,973		from: &T::CrossAccountId,974		token: TokenId,975		nesting_budget: &dyn Budget,976	) -> DispatchResult {977		Self::check_allowed(collection, spender, from, token, nesting_budget)?;978979		// =========980981		Self::burn(collection, from, token)982	}983984	pub fn check_nesting(985		handle: &NonfungibleHandle<T>,986		sender: T::CrossAccountId,987		from: (CollectionId, TokenId),988		under: TokenId,989		nesting_budget: &dyn Budget,990	) -> DispatchResult {991		fn ensure_sender_allowed<T: Config>(992			collection: CollectionId,993			token: TokenId,994			for_nest: (CollectionId, TokenId),995			sender: T::CrossAccountId,996			budget: &dyn Budget,997		) -> DispatchResult {998			ensure!(999				<PalletStructure<T>>::check_indirectly_owned(1000					sender,1001					collection,1002					token,1003					Some(for_nest),1004					budget1005				)?,1006				<CommonError<T>>::OnlyOwnerAllowedToNest,1007			);1008			Ok(())1009		}1010		match handle.permissions.nesting() {1011			NestingRule::Disabled => fail!(<CommonError<T>>::NestingIsDisabled),1012			NestingRule::Owner => {1013				ensure_sender_allowed::<T>(handle.id, under, from, sender, nesting_budget)?1014			}1015			NestingRule::OwnerRestricted(whitelist) => {1016				ensure!(1017					whitelist.contains(&from.0),1018					<CommonError<T>>::SourceCollectionIsNotAllowedToNest1019				);1020				ensure_sender_allowed::<T>(handle.id, under, from, sender, nesting_budget)?1021			}1022			NestingRule::Permissive => {}1023		}1024		Ok(())1025	}10261027	fn nest(under: (CollectionId, TokenId), to_nest: (CollectionId, TokenId)) {1028		<TokenChildren<T>>::insert((under.0, under.1, (to_nest.0, to_nest.1)), true);1029	}10301031	fn unnest(under: (CollectionId, TokenId), to_unnest: (CollectionId, TokenId)) {1032		<TokenChildren<T>>::remove((under.0, under.1, to_unnest));1033	}10341035	fn collection_has_tokens(collection_id: CollectionId) -> bool {1036		<TokenData<T>>::iter_prefix((collection_id,))1037			.next()1038			.is_some()1039	}10401041	fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {1042		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1043			.next()1044			.is_some()1045	}10461047	pub fn token_children_ids(collection_id: CollectionId, token_id: TokenId) -> Vec<TokenChild> {1048		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1049			.map(|((child_collection_id, child_id), _)| TokenChild {1050				collection: child_collection_id,1051				token: child_id,1052			})1053			.collect()1054	}10551056	/// Delegated to `create_multiple_items`1057	pub fn create_item(1058		collection: &NonfungibleHandle<T>,1059		sender: &T::CrossAccountId,1060		data: CreateItemData<T>,1061		nesting_budget: &dyn Budget,1062	) -> DispatchResult {1063		Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1064	}1065}
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#![cfg_attr(not(feature = "std"), no_std)]1819use erc::ERC721Events;20use evm_coder::ToLog;21use frame_support::{22	BoundedVec, ensure, fail, transactional,23	storage::with_transaction,24	pallet_prelude::DispatchResultWithPostInfo,25	pallet_prelude::Weight,26	weights::{PostDispatchInfo, Pays},27};28use up_data_structs::{29	AccessMode, CollectionId, CustomDataLimit, TokenId, CreateCollectionData, CreateNftExData,30	mapping::TokenAddressMapping, NestingRule, budget::Budget, Property, PropertyPermission,31	PropertyKey, PropertyKeyPermission, Properties, PropertyScope, TrySetProperty, TokenChild,32};33use pallet_evm::{account::CrossAccountId, Pallet as PalletEvm};34use pallet_common::{35	Error as CommonError, Pallet as PalletCommon, Event as CommonEvent, CollectionHandle,36	eth::collection_id_to_address,37};38use pallet_structure::{Pallet as PalletStructure, Error as StructureError};39use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};40use sp_core::H160;41use sp_runtime::{ArithmeticError, DispatchError, DispatchResult, TransactionOutcome};42use sp_std::{vec::Vec, vec, collections::btree_map::BTreeMap, collections::btree_set::BTreeSet};43use core::ops::Deref;44use codec::{Encode, Decode, MaxEncodedLen};45use scale_info::TypeInfo;4647pub use pallet::*;48use weights::WeightInfo;49#[cfg(feature = "runtime-benchmarks")]50pub mod benchmarking;51pub mod common;52pub mod erc;53pub mod weights;5455pub type CreateItemData<T> = CreateNftExData<<T as pallet_evm::account::Config>::CrossAccountId>;56pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;5758#[struct_versioning::versioned(version = 2, upper)]59#[derive(Encode, Decode, TypeInfo, MaxEncodedLen)]60pub struct ItemData<CrossAccountId> {61	#[version(..2)]62	pub const_data: BoundedVec<u8, CustomDataLimit>,6364	#[version(..2)]65	pub variable_data: BoundedVec<u8, CustomDataLimit>,6667	pub owner: CrossAccountId,68}6970#[frame_support::pallet]71pub mod pallet {72	use super::*;73	use frame_support::{74		Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key, traits::StorageVersion,75	};76	use frame_system::pallet_prelude::*;77	use up_data_structs::{CollectionId, TokenId};78	use super::weights::WeightInfo;7980	#[pallet::error]81	pub enum Error<T> {82		/// Not Nonfungible item data used to mint in Nonfungible collection.83		NotNonfungibleDataUsedToMintFungibleCollectionToken,84		/// Used amount > 1 with NFT85		NonfungibleItemsHaveNoAmount,86		/// Unable to burn NFT with children87		CantBurnNftWithChildren,88	}8990	#[pallet::config]91	pub trait Config:92		frame_system::Config + pallet_common::Config + pallet_structure::Config + pallet_evm::Config93	{94		type WeightInfo: WeightInfo;95	}9697	const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);9899	#[pallet::pallet]100	#[pallet::storage_version(STORAGE_VERSION)]101	#[pallet::generate_store(pub(super) trait Store)]102	pub struct Pallet<T>(_);103104	#[pallet::storage]105	pub type TokensMinted<T: Config> =106		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;107	#[pallet::storage]108	pub type TokensBurnt<T: Config> =109		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;110111	#[pallet::storage]112	pub type TokenData<T: Config> = StorageNMap<113		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),114		Value = ItemData<T::CrossAccountId>,115		QueryKind = OptionQuery,116	>;117118	#[pallet::storage]119	#[pallet::getter(fn token_properties)]120	pub type TokenProperties<T: Config> = StorageNMap<121		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),122		Value = Properties,123		QueryKind = ValueQuery,124		OnEmpty = up_data_structs::TokenProperties,125	>;126127	/// Used to enumerate tokens owned by account128	#[pallet::storage]129	pub type Owned<T: Config> = StorageNMap<130		Key = (131			Key<Twox64Concat, CollectionId>,132			Key<Blake2_128Concat, T::CrossAccountId>,133			Key<Twox64Concat, TokenId>,134		),135		Value = bool,136		QueryKind = ValueQuery,137	>;138139	/// Used to enumerate token's children140	#[pallet::storage]141	#[pallet::getter(fn token_children)]142	pub type TokenChildren<T: Config> = StorageNMap<143		Key = (144			Key<Twox64Concat, CollectionId>,145			Key<Twox64Concat, TokenId>,146			Key<Twox64Concat, (CollectionId, TokenId)>,147		),148		Value = bool,149		QueryKind = ValueQuery,150	>;151152	#[pallet::storage]153	pub type AccountBalance<T: Config> = StorageNMap<154		Key = (155			Key<Twox64Concat, CollectionId>,156			Key<Blake2_128Concat, T::CrossAccountId>,157		),158		Value = u32,159		QueryKind = ValueQuery,160	>;161162	#[pallet::storage]163	pub type Allowance<T: Config> = StorageNMap<164		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),165		Value = T::CrossAccountId,166		QueryKind = OptionQuery,167	>;168169	#[pallet::hooks]170	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {171		fn on_runtime_upgrade() -> Weight {172			if StorageVersion::get::<Pallet<T>>() < StorageVersion::new(1) {173				let mut had_consts = BTreeSet::new();174				<TokenData<T>>::translate::<ItemDataVersion1<T::CrossAccountId>, _>(175					|(collection, token), v| {176						let mut props = vec![];177						if !v.const_data.is_empty() {178							props.push(Property {179								key: b"_old_constData".to_vec().try_into().unwrap(),180								value: v181									.const_data182									.clone()183									.into_inner()184									.try_into()185									.expect("const too long"),186							});187							had_consts.insert(collection);188						}189						if !v.variable_data.is_empty() {190							props.push(Property {191								key: b"_old_variableData".to_vec().try_into().unwrap(),192								value: v193									.variable_data194									.clone()195									.into_inner()196									.try_into()197									.expect("variable too long"),198							})199						}200						if !props.is_empty() {201							Self::set_scoped_token_properties(202								collection,203								token,204								PropertyScope::None,205								props.into_iter(),206							)207							.expect("existing token data exceeds property storage");208						}209						Some(<ItemDataVersion2<T::CrossAccountId>>::from(v))210					},211				);212				for collection in had_consts {213					<PalletCommon<T>>::set_property_permission_unchecked(214						collection,215						PropertyKeyPermission {216							key: b"_old_constData".to_vec().try_into().unwrap(),217							permission: PropertyPermission {218								mutable: false,219								collection_admin: true,220								token_owner: false,221							},222						},223					)224					.expect("failed to configure permission");225				}226			}227228			0229		}230	}231}232233pub struct NonfungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);234impl<T: Config> NonfungibleHandle<T> {235	pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {236		Self(inner)237	}238	pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {239		self.0240	}241	pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {242		&mut self.0243	}244}245impl<T: Config> WithRecorder<T> for NonfungibleHandle<T> {246	fn recorder(&self) -> &SubstrateRecorder<T> {247		self.0.recorder()248	}249	fn into_recorder(self) -> SubstrateRecorder<T> {250		self.0.into_recorder()251	}252}253impl<T: Config> Deref for NonfungibleHandle<T> {254	type Target = pallet_common::CollectionHandle<T>;255256	fn deref(&self) -> &Self::Target {257		&self.0258	}259}260261impl<T: Config> Pallet<T> {262	pub fn total_supply(collection: &NonfungibleHandle<T>) -> u32 {263		<TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)264	}265	pub fn token_exists(collection: &NonfungibleHandle<T>, token: TokenId) -> bool {266		<TokenData<T>>::contains_key((collection.id, token))267	}268269	pub fn set_scoped_token_property(270		collection_id: CollectionId,271		token_id: TokenId,272		scope: PropertyScope,273		property: Property,274	) -> DispatchResult {275		TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {276			properties.try_scoped_set(scope, property.key, property.value)277		})278		.map_err(<CommonError<T>>::from)?;279280		Ok(())281	}282283	pub fn set_scoped_token_properties(284		collection_id: CollectionId,285		token_id: TokenId,286		scope: PropertyScope,287		properties: impl Iterator<Item = Property>,288	) -> DispatchResult {289		TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {290			stored_properties.try_scoped_set_from_iter(scope, properties)291		})292		.map_err(<CommonError<T>>::from)?;293294		Ok(())295	}296297	pub fn current_token_id(collection_id: CollectionId) -> TokenId {298		TokenId(<TokensMinted<T>>::get(collection_id))299	}300}301302// unchecked calls skips any permission checks303impl<T: Config> Pallet<T> {304	pub fn init_collection(305		owner: T::CrossAccountId,306		data: CreateCollectionData<T::AccountId>,307	) -> Result<CollectionId, DispatchError> {308		<PalletCommon<T>>::init_collection(owner, data)309	}310	pub fn destroy_collection(311		collection: NonfungibleHandle<T>,312		sender: &T::CrossAccountId,313	) -> DispatchResult {314		let id = collection.id;315316		if Self::collection_has_tokens(id) {317			return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());318		}319320		// =========321322		PalletCommon::destroy_collection(collection.0, sender)?;323324		<TokenData<T>>::remove_prefix((id,), None);325		<TokenChildren<T>>::remove_prefix((id,), None);326		<Owned<T>>::remove_prefix((id,), None);327		<TokensMinted<T>>::remove(id);328		<TokensBurnt<T>>::remove(id);329		<Allowance<T>>::remove_prefix((id,), None);330		<AccountBalance<T>>::remove_prefix((id,), None);331		Ok(())332	}333334	pub fn burn(335		collection: &NonfungibleHandle<T>,336		sender: &T::CrossAccountId,337		token: TokenId,338	) -> DispatchResult {339		let token_data =340			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;341		ensure!(342			&token_data.owner == sender343				|| (collection.limits.owner_can_transfer() && collection.is_owner_or_admin(sender)),344			<CommonError<T>>::NoPermission345		);346347		if collection.permissions.access() == AccessMode::AllowList {348			collection.check_allowlist(sender)?;349		}350351		if Self::token_has_children(collection.id, token) {352			return Err(<Error<T>>::CantBurnNftWithChildren.into());353		}354355		let burnt = <TokensBurnt<T>>::get(collection.id)356			.checked_add(1)357			.ok_or(ArithmeticError::Overflow)?;358359		let balance = <AccountBalance<T>>::get((collection.id, token_data.owner.clone()))360			.checked_sub(1)361			.ok_or(ArithmeticError::Overflow)?;362363		// =========364365		if balance == 0 {366			<AccountBalance<T>>::remove((collection.id, token_data.owner.clone()));367		} else {368			<AccountBalance<T>>::insert((collection.id, token_data.owner.clone()), balance);369		}370371		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);372373		<Owned<T>>::remove((collection.id, &token_data.owner, token));374		<TokensBurnt<T>>::insert(collection.id, burnt);375		<TokenData<T>>::remove((collection.id, token));376		<TokenProperties<T>>::remove((collection.id, token));377		let old_spender = <Allowance<T>>::take((collection.id, token));378379		if let Some(old_spender) = old_spender {380			<PalletCommon<T>>::deposit_event(CommonEvent::Approved(381				collection.id,382				token,383				token_data.owner.clone(),384				old_spender,385				0,386			));387		}388389		<PalletEvm<T>>::deposit_log(390			ERC721Events::Transfer {391				from: *token_data.owner.as_eth(),392				to: H160::default(),393				token_id: token.into(),394			}395			.to_log(collection_id_to_address(collection.id)),396		);397		<PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(398			collection.id,399			token,400			token_data.owner,401			1,402		));403		Ok(())404	}405406	#[transactional]407	pub fn burn_recursively(408		collection: &NonfungibleHandle<T>,409		sender: &T::CrossAccountId,410		token: TokenId,411		self_budget: &dyn Budget,412		breadth_budget: &dyn Budget,413	) -> DispatchResultWithPostInfo {414		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);415416		let current_token_account =417			T::CrossTokenAddressMapping::token_to_address(collection.id, token);418419		let mut weight = 0 as Weight;420421		// This method is transactional, if user in fact doesn't have permissions to remove token -422		// tokens removed here will be restored after rejected transaction423		for ((collection, token), _) in <TokenChildren<T>>::iter_prefix((collection.id, token)) {424			ensure!(breadth_budget.consume(), <StructureError<T>>::BreadthLimit,);425			let PostDispatchInfo { actual_weight, .. } =426				<PalletStructure<T>>::burn_item_recursively(427					current_token_account.clone(),428					collection,429					token,430					self_budget,431					breadth_budget,432				)?;433			if let Some(actual_weight) = actual_weight {434				weight = weight.saturating_add(actual_weight);435			}436		}437438		Self::burn(collection, sender, token)?;439		DispatchResultWithPostInfo::Ok(PostDispatchInfo {440			actual_weight: Some(weight + <SelfWeightOf<T>>::burn_item()),441			pays_fee: Pays::Yes,442		})443	}444445	pub fn set_token_property(446		collection: &NonfungibleHandle<T>,447		sender: &T::CrossAccountId,448		token_id: TokenId,449		property: Property,450		is_token_create: bool,451	) -> DispatchResult {452		Self::check_token_change_permission(453			collection,454			sender,455			token_id,456			&property.key,457			is_token_create,458		)?;459460		<TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {461			let property = property.clone();462			properties.try_set(property.key, property.value)463		})464		.map_err(<CommonError<T>>::from)?;465466		<PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertySet(467			collection.id,468			token_id,469			property.key,470		));471472		Ok(())473	}474475	#[transactional]476	pub fn set_token_properties(477		collection: &NonfungibleHandle<T>,478		sender: &T::CrossAccountId,479		token_id: TokenId,480		properties: Vec<Property>,481		is_token_create: bool,482	) -> DispatchResult {483		for property in properties {484			Self::set_token_property(collection, sender, token_id, property, is_token_create)?;485		}486487		Ok(())488	}489490	pub fn delete_token_property(491		collection: &NonfungibleHandle<T>,492		sender: &T::CrossAccountId,493		token_id: TokenId,494		property_key: PropertyKey,495	) -> DispatchResult {496		Self::check_token_change_permission(collection, sender, token_id, &property_key, false)?;497498		<TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {499			properties.remove(&property_key)500		})501		.map_err(<CommonError<T>>::from)?;502503		<PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertyDeleted(504			collection.id,505			token_id,506			property_key,507		));508509		Ok(())510	}511512	fn check_token_change_permission(513		collection: &NonfungibleHandle<T>,514		sender: &T::CrossAccountId,515		token_id: TokenId,516		property_key: &PropertyKey,517		is_token_create: bool,518	) -> DispatchResult {519		let permission = <PalletCommon<T>>::property_permissions(collection.id)520			.get(property_key)521			.cloned()522			.unwrap_or_else(PropertyPermission::none);523524		let token_data = <TokenData<T>>::get((collection.id, token_id))525			.ok_or(<CommonError<T>>::TokenNotFound)?;526527		let check_token_owner = || -> DispatchResult {528			ensure!(&token_data.owner == sender, <CommonError<T>>::NoPermission);529			Ok(())530		};531532		let is_property_exists = TokenProperties::<T>::get((collection.id, token_id))533			.get(property_key)534			.is_some();535536		match permission {537			PropertyPermission { mutable: false, .. } if is_property_exists => {538				Err(<CommonError<T>>::NoPermission.into())539			}540541			PropertyPermission {542				collection_admin,543				token_owner,544				..545			} => {546				if is_token_create && (collection_admin || token_owner) {547					return Ok(());548				}549550				let mut check_result = Err(<CommonError<T>>::NoPermission.into());551552				if collection_admin {553					check_result = collection.check_is_owner_or_admin(sender);554				}555556				if token_owner {557					check_result.or_else(|_| check_token_owner())558				} else {559					check_result560				}561			}562		}563	}564565	#[transactional]566	pub fn delete_token_properties(567		collection: &NonfungibleHandle<T>,568		sender: &T::CrossAccountId,569		token_id: TokenId,570		property_keys: Vec<PropertyKey>,571	) -> DispatchResult {572		for key in property_keys {573			Self::delete_token_property(collection, sender, token_id, key)?;574		}575576		Ok(())577	}578579	pub fn set_collection_properties(580		collection: &NonfungibleHandle<T>,581		sender: &T::CrossAccountId,582		properties: Vec<Property>,583	) -> DispatchResult {584		<PalletCommon<T>>::set_collection_properties(collection, sender, properties)585	}586587	pub fn delete_collection_properties(588		collection: &CollectionHandle<T>,589		sender: &T::CrossAccountId,590		property_keys: Vec<PropertyKey>,591	) -> DispatchResult {592		<PalletCommon<T>>::delete_collection_properties(collection, sender, property_keys)593	}594595	pub fn set_property_permissions(596		collection: &CollectionHandle<T>,597		sender: &T::CrossAccountId,598		property_permissions: Vec<PropertyKeyPermission>,599	) -> DispatchResult {600		<PalletCommon<T>>::set_property_permissions(collection, sender, property_permissions)601	}602603	pub fn set_property_permission(604		collection: &CollectionHandle<T>,605		sender: &T::CrossAccountId,606		permission: PropertyKeyPermission,607	) -> DispatchResult {608		<PalletCommon<T>>::set_property_permission(collection, sender, permission)609	}610611	pub fn transfer(612		collection: &NonfungibleHandle<T>,613		from: &T::CrossAccountId,614		to: &T::CrossAccountId,615		token: TokenId,616		nesting_budget: &dyn Budget,617	) -> DispatchResult {618		ensure!(619			collection.limits.transfers_enabled(),620			<CommonError<T>>::TransferNotAllowed621		);622623		let token_data =624			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;625		// TODO: require sender to be token, owner, require admins to go through transfer_from626		ensure!(627			&token_data.owner == from628				|| (collection.limits.owner_can_transfer() && collection.is_owner_or_admin(from)),629			<CommonError<T>>::NoPermission630		);631632		if collection.permissions.access() == AccessMode::AllowList {633			collection.check_allowlist(from)?;634			collection.check_allowlist(to)?;635		}636		<PalletCommon<T>>::ensure_correct_receiver(to)?;637638		let balance_from = <AccountBalance<T>>::get((collection.id, from))639			.checked_sub(1)640			.ok_or(<CommonError<T>>::TokenValueTooLow)?;641		let balance_to = if from != to {642			let balance_to = <AccountBalance<T>>::get((collection.id, to))643				.checked_add(1)644				.ok_or(ArithmeticError::Overflow)?;645646			ensure!(647				balance_to < collection.limits.account_token_ownership_limit(),648				<CommonError<T>>::AccountTokenLimitExceeded,649			);650651			Some(balance_to)652		} else {653			None654		};655656		<PalletStructure<T>>::nest_if_sent_to_token(657			from.clone(),658			to,659			collection.id,660			token,661			nesting_budget,662		)?;663664		// =========665666		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);667668		<TokenData<T>>::insert(669			(collection.id, token),670			ItemData {671				owner: to.clone(),672				..token_data673			},674		);675676		if let Some(balance_to) = balance_to {677			// from != to678			if balance_from == 0 {679				<AccountBalance<T>>::remove((collection.id, from));680			} else {681				<AccountBalance<T>>::insert((collection.id, from), balance_from);682			}683			<AccountBalance<T>>::insert((collection.id, to), balance_to);684			<Owned<T>>::remove((collection.id, from, token));685			<Owned<T>>::insert((collection.id, to, token), true);686		}687		Self::set_allowance_unchecked(collection, from, token, None, true);688689		<PalletEvm<T>>::deposit_log(690			ERC721Events::Transfer {691				from: *from.as_eth(),692				to: *to.as_eth(),693				token_id: token.into(),694			}695			.to_log(collection_id_to_address(collection.id)),696		);697		<PalletCommon<T>>::deposit_event(CommonEvent::Transfer(698			collection.id,699			token,700			from.clone(),701			to.clone(),702			1,703		));704		Ok(())705	}706707	pub fn create_multiple_items(708		collection: &NonfungibleHandle<T>,709		sender: &T::CrossAccountId,710		data: Vec<CreateItemData<T>>,711		nesting_budget: &dyn Budget,712	) -> DispatchResult {713		if !collection.is_owner_or_admin(sender) {714			ensure!(715				collection.permissions.mint_mode(),716				<CommonError<T>>::PublicMintingNotAllowed717			);718			collection.check_allowlist(sender)?;719720			for item in data.iter() {721				collection.check_allowlist(&item.owner)?;722			}723		}724725		for data in data.iter() {726			<PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;727		}728729		let first_token = <TokensMinted<T>>::get(collection.id);730		let tokens_minted = first_token731			.checked_add(data.len() as u32)732			.ok_or(ArithmeticError::Overflow)?;733		ensure!(734			tokens_minted <= collection.limits.token_limit(),735			<CommonError<T>>::CollectionTokenLimitExceeded736		);737738		let mut balances = BTreeMap::new();739		for data in &data {740			let balance = balances741				.entry(&data.owner)742				.or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));743			*balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;744745			ensure!(746				*balance <= collection.limits.account_token_ownership_limit(),747				<CommonError<T>>::AccountTokenLimitExceeded,748			);749		}750751		for (i, data) in data.iter().enumerate() {752			let token = TokenId(first_token + i as u32 + 1);753754			<PalletStructure<T>>::check_nesting(755				sender.clone(),756				&data.owner,757				collection.id,758				token,759				nesting_budget,760			)?;761		}762763		// =========764765		with_transaction(|| {766			for (i, data) in data.iter().enumerate() {767				let token = first_token + i as u32 + 1;768769				<TokenData<T>>::insert(770					(collection.id, token),771					ItemData {772						// const_data: data.const_data.clone(),773						owner: data.owner.clone(),774					},775				);776777				<PalletStructure<T>>::nest_if_sent_to_token_unchecked(778					&data.owner,779					collection.id,780					TokenId(token),781				);782783				if let Err(e) = Self::set_token_properties(784					collection,785					sender,786					TokenId(token),787					data.properties.clone().into_inner(),788					true,789				) {790					return TransactionOutcome::Rollback(Err(e));791				}792			}793			TransactionOutcome::Commit(Ok(()))794		})?;795796		<TokensMinted<T>>::insert(collection.id, tokens_minted);797		for (account, balance) in balances {798			<AccountBalance<T>>::insert((collection.id, account), balance);799		}800		for (i, data) in data.into_iter().enumerate() {801			let token = first_token + i as u32 + 1;802			<Owned<T>>::insert((collection.id, &data.owner, token), true);803804			<PalletEvm<T>>::deposit_log(805				ERC721Events::Transfer {806					from: H160::default(),807					to: *data.owner.as_eth(),808					token_id: token.into(),809				}810				.to_log(collection_id_to_address(collection.id)),811			);812			<PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(813				collection.id,814				TokenId(token),815				data.owner.clone(),816				1,817			));818		}819		Ok(())820	}821822	pub fn set_allowance_unchecked(823		collection: &NonfungibleHandle<T>,824		sender: &T::CrossAccountId,825		token: TokenId,826		spender: Option<&T::CrossAccountId>,827		assume_implicit_eth: bool,828	) {829		if let Some(spender) = spender {830			let old_spender = <Allowance<T>>::get((collection.id, token));831			<Allowance<T>>::insert((collection.id, token), spender);832			// In ERC721 there is only one possible approved user of token, so we set833			// approved user to spender834			<PalletEvm<T>>::deposit_log(835				ERC721Events::Approval {836					owner: *sender.as_eth(),837					approved: *spender.as_eth(),838					token_id: token.into(),839				}840				.to_log(collection_id_to_address(collection.id)),841			);842			// In Unique chain, any token can have any amount of approved users, so we need to843			// set allowance of old owner to 0, and allowance of new owner to 1844			if old_spender.as_ref() != Some(spender) {845				if let Some(old_owner) = old_spender {846					<PalletCommon<T>>::deposit_event(CommonEvent::Approved(847						collection.id,848						token,849						sender.clone(),850						old_owner,851						0,852					));853				}854				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(855					collection.id,856					token,857					sender.clone(),858					spender.clone(),859					1,860				));861			}862		} else {863			let old_spender = <Allowance<T>>::take((collection.id, token));864			if !assume_implicit_eth {865				// In ERC721 there is only one possible approved user of token, so we set866				// approved user to zero address867				<PalletEvm<T>>::deposit_log(868					ERC721Events::Approval {869						owner: *sender.as_eth(),870						approved: H160::default(),871						token_id: token.into(),872					}873					.to_log(collection_id_to_address(collection.id)),874				);875			}876			// In Unique chain, any token can have any amount of approved users, so we need to877			// set allowance of old owner to 0878			if let Some(old_spender) = old_spender {879				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(880					collection.id,881					token,882					sender.clone(),883					old_spender,884					0,885				));886			}887		}888	}889890	pub fn set_allowance(891		collection: &NonfungibleHandle<T>,892		sender: &T::CrossAccountId,893		token: TokenId,894		spender: Option<&T::CrossAccountId>,895	) -> DispatchResult {896		if collection.permissions.access() == AccessMode::AllowList {897			collection.check_allowlist(sender)?;898			if let Some(spender) = spender {899				collection.check_allowlist(spender)?;900			}901		}902903		if let Some(spender) = spender {904			<PalletCommon<T>>::ensure_correct_receiver(spender)?;905		}906907		let token_data =908			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;909		if &token_data.owner != sender {910			ensure!(911				collection.ignores_owned_amount(sender),912				<CommonError<T>>::CantApproveMoreThanOwned913			);914		}915916		// =========917918		Self::set_allowance_unchecked(collection, sender, token, spender, false);919		Ok(())920	}921922	fn check_allowed(923		collection: &NonfungibleHandle<T>,924		spender: &T::CrossAccountId,925		from: &T::CrossAccountId,926		token: TokenId,927		nesting_budget: &dyn Budget,928	) -> DispatchResult {929		if spender.conv_eq(from) {930			return Ok(());931		}932		if collection.permissions.access() == AccessMode::AllowList {933			// `from`, `to` checked in [`transfer`]934			collection.check_allowlist(spender)?;935		}936		if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {937			// TODO: should collection owner be allowed to perform this transfer?938			ensure!(939				<PalletStructure<T>>::check_indirectly_owned(940					spender.clone(),941					source.0,942					source.1,943					None,944					nesting_budget945				)?,946				<CommonError<T>>::ApprovedValueTooLow,947			);948			return Ok(());949		}950		if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {951			return Ok(());952		}953		ensure!(954			collection.ignores_allowance(spender),955			<CommonError<T>>::ApprovedValueTooLow956		);957		Ok(())958	}959960	pub fn transfer_from(961		collection: &NonfungibleHandle<T>,962		spender: &T::CrossAccountId,963		from: &T::CrossAccountId,964		to: &T::CrossAccountId,965		token: TokenId,966		nesting_budget: &dyn Budget,967	) -> DispatchResult {968		Self::check_allowed(collection, spender, from, token, nesting_budget)?;969970		// =========971972		// Allowance is reset in [`transfer`]973		Self::transfer(collection, from, to, token, nesting_budget)974	}975976	pub fn burn_from(977		collection: &NonfungibleHandle<T>,978		spender: &T::CrossAccountId,979		from: &T::CrossAccountId,980		token: TokenId,981		nesting_budget: &dyn Budget,982	) -> DispatchResult {983		Self::check_allowed(collection, spender, from, token, nesting_budget)?;984985		// =========986987		Self::burn(collection, from, token)988	}989990	pub fn check_nesting(991		handle: &NonfungibleHandle<T>,992		sender: T::CrossAccountId,993		from: (CollectionId, TokenId),994		under: TokenId,995		nesting_budget: &dyn Budget,996	) -> DispatchResult {997		fn ensure_sender_allowed<T: Config>(998			collection: CollectionId,999			token: TokenId,1000			for_nest: (CollectionId, TokenId),1001			sender: T::CrossAccountId,1002			budget: &dyn Budget,1003		) -> DispatchResult {1004			ensure!(1005				<PalletStructure<T>>::check_indirectly_owned(1006					sender,1007					collection,1008					token,1009					Some(for_nest),1010					budget1011				)?,1012				<CommonError<T>>::OnlyOwnerAllowedToNest,1013			);1014			Ok(())1015		}1016		match handle.permissions.nesting() {1017			NestingRule::Disabled => fail!(<CommonError<T>>::NestingIsDisabled),1018			NestingRule::Owner => {1019				ensure_sender_allowed::<T>(handle.id, under, from, sender, nesting_budget)?1020			}1021			NestingRule::OwnerRestricted(whitelist) => {1022				ensure!(1023					whitelist.contains(&from.0),1024					<CommonError<T>>::SourceCollectionIsNotAllowedToNest1025				);1026				ensure_sender_allowed::<T>(handle.id, under, from, sender, nesting_budget)?1027			}1028			NestingRule::Permissive => {}1029		}1030		Ok(())1031	}10321033	fn nest(under: (CollectionId, TokenId), to_nest: (CollectionId, TokenId)) {1034		<TokenChildren<T>>::insert((under.0, under.1, (to_nest.0, to_nest.1)), true);1035	}10361037	fn unnest(under: (CollectionId, TokenId), to_unnest: (CollectionId, TokenId)) {1038		<TokenChildren<T>>::remove((under.0, under.1, to_unnest));1039	}10401041	fn collection_has_tokens(collection_id: CollectionId) -> bool {1042		<TokenData<T>>::iter_prefix((collection_id,))1043			.next()1044			.is_some()1045	}10461047	fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {1048		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1049			.next()1050			.is_some()1051	}10521053	pub fn token_children_ids(collection_id: CollectionId, token_id: TokenId) -> Vec<TokenChild> {1054		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1055			.map(|((child_collection_id, child_id), _)| TokenChild {1056				collection: child_collection_id,1057				token: child_id,1058			})1059			.collect()1060	}10611062	/// Delegated to `create_multiple_items`1063	pub fn create_item(1064		collection: &NonfungibleHandle<T>,1065		sender: &T::CrossAccountId,1066		data: CreateItemData<T>,1067		nesting_budget: &dyn Budget,1068	) -> DispatchResult {1069		Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1070	}1071}