git.delta.rocks / unique-network / refs/commits / 87ff08974106

difftreelog

CORE-386 Fix PR

Trubnikov Sergey2022-06-03parent: #9e201d9.patch.diff
in: master

7 files changed

modifiedpallets/common/src/erc.rsdiffbeforeafterboth
--- a/pallets/common/src/erc.rs
+++ b/pallets/common/src/erc.rs
@@ -261,13 +261,13 @@
 		Ok(())
 	}
 
-	fn set_collection_access(&mut self, caller: caller, mode: string) -> Result<void> {
+	fn set_collection_access(&mut self, caller: caller, mode: uint8) -> Result<void> {
 		let caller = T::CrossAccountId::from_eth(caller);
 		self.check_is_owner_or_admin(&caller)
 			.map_err(dispatch_to_evm::<T>)?;
-		self.collection.permissions.access = Some(match mode.as_str() {
-			"Normal" => AccessMode::Normal,
-			"AllowList" => AccessMode::AllowList,
+		self.collection.permissions.access = Some(match mode {
+			0 => AccessMode::Normal,
+			1 => AccessMode::AllowList,
 			_ => return Err("Not supported access mode".into()),
 		});
 		save(self);
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(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}
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				//TODO: исследовать угрозы при публичном минтинге.547				if is_token_create && (collection_admin || token_owner) {548					return Ok(());549				}550551				let mut check_result = Err(<CommonError<T>>::NoPermission.into());552553				if collection_admin {554					check_result = collection.check_is_owner_or_admin(sender);555				}556557				if token_owner {558					check_result.or_else(|_| check_token_owner())559				} else {560					check_result561				}562			}563		}564	}565566	#[transactional]567	pub fn delete_token_properties(568		collection: &NonfungibleHandle<T>,569		sender: &T::CrossAccountId,570		token_id: TokenId,571		property_keys: Vec<PropertyKey>,572	) -> DispatchResult {573		for key in property_keys {574			Self::delete_token_property(collection, sender, token_id, key)?;575		}576577		Ok(())578	}579580	pub fn set_collection_properties(581		collection: &NonfungibleHandle<T>,582		sender: &T::CrossAccountId,583		properties: Vec<Property>,584	) -> DispatchResult {585		<PalletCommon<T>>::set_collection_properties(collection, sender, properties)586	}587588	pub fn delete_collection_properties(589		collection: &CollectionHandle<T>,590		sender: &T::CrossAccountId,591		property_keys: Vec<PropertyKey>,592	) -> DispatchResult {593		<PalletCommon<T>>::delete_collection_properties(collection, sender, property_keys)594	}595596	pub fn set_property_permissions(597		collection: &CollectionHandle<T>,598		sender: &T::CrossAccountId,599		property_permissions: Vec<PropertyKeyPermission>,600	) -> DispatchResult {601		<PalletCommon<T>>::set_property_permissions(collection, sender, property_permissions)602	}603604	pub fn set_property_permission(605		collection: &CollectionHandle<T>,606		sender: &T::CrossAccountId,607		permission: PropertyKeyPermission,608	) -> DispatchResult {609		<PalletCommon<T>>::set_property_permission(collection, sender, permission)610	}611612	pub fn transfer(613		collection: &NonfungibleHandle<T>,614		from: &T::CrossAccountId,615		to: &T::CrossAccountId,616		token: TokenId,617		nesting_budget: &dyn Budget,618	) -> DispatchResult {619		ensure!(620			collection.limits.transfers_enabled(),621			<CommonError<T>>::TransferNotAllowed622		);623624		let token_data =625			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;626		// TODO: require sender to be token, owner, require admins to go through transfer_from627		ensure!(628			&token_data.owner == from629				|| (collection.limits.owner_can_transfer() && collection.is_owner_or_admin(from)),630			<CommonError<T>>::NoPermission631		);632633		if collection.permissions.access() == AccessMode::AllowList {634			collection.check_allowlist(from)?;635			collection.check_allowlist(to)?;636		}637		<PalletCommon<T>>::ensure_correct_receiver(to)?;638639		let balance_from = <AccountBalance<T>>::get((collection.id, from))640			.checked_sub(1)641			.ok_or(<CommonError<T>>::TokenValueTooLow)?;642		let balance_to = if from != to {643			let balance_to = <AccountBalance<T>>::get((collection.id, to))644				.checked_add(1)645				.ok_or(ArithmeticError::Overflow)?;646647			ensure!(648				balance_to < collection.limits.account_token_ownership_limit(),649				<CommonError<T>>::AccountTokenLimitExceeded,650			);651652			Some(balance_to)653		} else {654			None655		};656657		<PalletStructure<T>>::nest_if_sent_to_token(658			from.clone(),659			to,660			collection.id,661			token,662			nesting_budget,663		)?;664665		// =========666667		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);668669		<TokenData<T>>::insert(670			(collection.id, token),671			ItemData {672				owner: to.clone(),673				..token_data674			},675		);676677		if let Some(balance_to) = balance_to {678			// from != to679			if balance_from == 0 {680				<AccountBalance<T>>::remove((collection.id, from));681			} else {682				<AccountBalance<T>>::insert((collection.id, from), balance_from);683			}684			<AccountBalance<T>>::insert((collection.id, to), balance_to);685			<Owned<T>>::remove((collection.id, from, token));686			<Owned<T>>::insert((collection.id, to, token), true);687		}688		Self::set_allowance_unchecked(collection, from, token, None, true);689690		<PalletEvm<T>>::deposit_log(691			ERC721Events::Transfer {692				from: *from.as_eth(),693				to: *to.as_eth(),694				token_id: token.into(),695			}696			.to_log(collection_id_to_address(collection.id)),697		);698		<PalletCommon<T>>::deposit_event(CommonEvent::Transfer(699			collection.id,700			token,701			from.clone(),702			to.clone(),703			1,704		));705		Ok(())706	}707708	pub fn create_multiple_items(709		collection: &NonfungibleHandle<T>,710		sender: &T::CrossAccountId,711		data: Vec<CreateItemData<T>>,712		nesting_budget: &dyn Budget,713	) -> DispatchResult {714		if !collection.is_owner_or_admin(sender) {715			ensure!(716				collection.permissions.mint_mode(),717				<CommonError<T>>::PublicMintingNotAllowed718			);719			collection.check_allowlist(sender)?;720721			for item in data.iter() {722				collection.check_allowlist(&item.owner)?;723			}724		}725726		for data in data.iter() {727			<PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;728		}729730		let first_token = <TokensMinted<T>>::get(collection.id);731		let tokens_minted = first_token732			.checked_add(data.len() as u32)733			.ok_or(ArithmeticError::Overflow)?;734		ensure!(735			tokens_minted <= collection.limits.token_limit(),736			<CommonError<T>>::CollectionTokenLimitExceeded737		);738739		let mut balances = BTreeMap::new();740		for data in &data {741			let balance = balances742				.entry(&data.owner)743				.or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));744			*balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;745746			ensure!(747				*balance <= collection.limits.account_token_ownership_limit(),748				<CommonError<T>>::AccountTokenLimitExceeded,749			);750		}751752		for (i, data) in data.iter().enumerate() {753			let token = TokenId(first_token + i as u32 + 1);754755			<PalletStructure<T>>::check_nesting(756				sender.clone(),757				&data.owner,758				collection.id,759				token,760				nesting_budget,761			)?;762		}763764		// =========765766		with_transaction(|| {767			for (i, data) in data.iter().enumerate() {768				let token = first_token + i as u32 + 1;769770				<TokenData<T>>::insert(771					(collection.id, token),772					ItemData {773						// const_data: data.const_data.clone(),774						owner: data.owner.clone(),775					},776				);777778				<PalletStructure<T>>::nest_if_sent_to_token_unchecked(779					&data.owner,780					collection.id,781					TokenId(token),782				);783784				if let Err(e) = Self::set_token_properties(785					collection,786					sender,787					TokenId(token),788					data.properties.clone().into_inner(),789					true,790				) {791					return TransactionOutcome::Rollback(Err(e));792				}793			}794			TransactionOutcome::Commit(Ok(()))795		})?;796797		<TokensMinted<T>>::insert(collection.id, tokens_minted);798		for (account, balance) in balances {799			<AccountBalance<T>>::insert((collection.id, account), balance);800		}801		for (i, data) in data.into_iter().enumerate() {802			let token = first_token + i as u32 + 1;803			<Owned<T>>::insert((collection.id, &data.owner, token), true);804805			<PalletEvm<T>>::deposit_log(806				ERC721Events::Transfer {807					from: H160::default(),808					to: *data.owner.as_eth(),809					token_id: token.into(),810				}811				.to_log(collection_id_to_address(collection.id)),812			);813			<PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(814				collection.id,815				TokenId(token),816				data.owner.clone(),817				1,818			));819		}820		Ok(())821	}822823	pub fn set_allowance_unchecked(824		collection: &NonfungibleHandle<T>,825		sender: &T::CrossAccountId,826		token: TokenId,827		spender: Option<&T::CrossAccountId>,828		assume_implicit_eth: bool,829	) {830		if let Some(spender) = spender {831			let old_spender = <Allowance<T>>::get((collection.id, token));832			<Allowance<T>>::insert((collection.id, token), spender);833			// In ERC721 there is only one possible approved user of token, so we set834			// approved user to spender835			<PalletEvm<T>>::deposit_log(836				ERC721Events::Approval {837					owner: *sender.as_eth(),838					approved: *spender.as_eth(),839					token_id: token.into(),840				}841				.to_log(collection_id_to_address(collection.id)),842			);843			// In Unique chain, any token can have any amount of approved users, so we need to844			// set allowance of old owner to 0, and allowance of new owner to 1845			if old_spender.as_ref() != Some(spender) {846				if let Some(old_owner) = old_spender {847					<PalletCommon<T>>::deposit_event(CommonEvent::Approved(848						collection.id,849						token,850						sender.clone(),851						old_owner,852						0,853					));854				}855				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(856					collection.id,857					token,858					sender.clone(),859					spender.clone(),860					1,861				));862			}863		} else {864			let old_spender = <Allowance<T>>::take((collection.id, token));865			if !assume_implicit_eth {866				// In ERC721 there is only one possible approved user of token, so we set867				// approved user to zero address868				<PalletEvm<T>>::deposit_log(869					ERC721Events::Approval {870						owner: *sender.as_eth(),871						approved: H160::default(),872						token_id: token.into(),873					}874					.to_log(collection_id_to_address(collection.id)),875				);876			}877			// In Unique chain, any token can have any amount of approved users, so we need to878			// set allowance of old owner to 0879			if let Some(old_spender) = old_spender {880				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(881					collection.id,882					token,883					sender.clone(),884					old_spender,885					0,886				));887			}888		}889	}890891	pub fn set_allowance(892		collection: &NonfungibleHandle<T>,893		sender: &T::CrossAccountId,894		token: TokenId,895		spender: Option<&T::CrossAccountId>,896	) -> DispatchResult {897		if collection.permissions.access() == AccessMode::AllowList {898			collection.check_allowlist(sender)?;899			if let Some(spender) = spender {900				collection.check_allowlist(spender)?;901			}902		}903904		if let Some(spender) = spender {905			<PalletCommon<T>>::ensure_correct_receiver(spender)?;906		}907908		let token_data =909			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;910		if &token_data.owner != sender {911			ensure!(912				collection.ignores_owned_amount(sender),913				<CommonError<T>>::CantApproveMoreThanOwned914			);915		}916917		// =========918919		Self::set_allowance_unchecked(collection, sender, token, spender, false);920		Ok(())921	}922923	fn check_allowed(924		collection: &NonfungibleHandle<T>,925		spender: &T::CrossAccountId,926		from: &T::CrossAccountId,927		token: TokenId,928		nesting_budget: &dyn Budget,929	) -> DispatchResult {930		if spender.conv_eq(from) {931			return Ok(());932		}933		if collection.permissions.access() == AccessMode::AllowList {934			// `from`, `to` checked in [`transfer`]935			collection.check_allowlist(spender)?;936		}937		if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {938			// TODO: should collection owner be allowed to perform this transfer?939			ensure!(940				<PalletStructure<T>>::check_indirectly_owned(941					spender.clone(),942					source.0,943					source.1,944					None,945					nesting_budget946				)?,947				<CommonError<T>>::ApprovedValueTooLow,948			);949			return Ok(());950		}951		if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {952			return Ok(());953		}954		ensure!(955			collection.ignores_allowance(spender),956			<CommonError<T>>::ApprovedValueTooLow957		);958		Ok(())959	}960961	pub fn transfer_from(962		collection: &NonfungibleHandle<T>,963		spender: &T::CrossAccountId,964		from: &T::CrossAccountId,965		to: &T::CrossAccountId,966		token: TokenId,967		nesting_budget: &dyn Budget,968	) -> DispatchResult {969		Self::check_allowed(collection, spender, from, token, nesting_budget)?;970971		// =========972973		// Allowance is reset in [`transfer`]974		Self::transfer(collection, from, to, token, nesting_budget)975	}976977	pub fn burn_from(978		collection: &NonfungibleHandle<T>,979		spender: &T::CrossAccountId,980		from: &T::CrossAccountId,981		token: TokenId,982		nesting_budget: &dyn Budget,983	) -> DispatchResult {984		Self::check_allowed(collection, spender, from, token, nesting_budget)?;985986		// =========987988		Self::burn(collection, from, token)989	}990991	pub fn check_nesting(992		handle: &NonfungibleHandle<T>,993		sender: T::CrossAccountId,994		from: (CollectionId, TokenId),995		under: TokenId,996		nesting_budget: &dyn Budget,997	) -> DispatchResult {998		fn ensure_sender_allowed<T: Config>(999			collection: CollectionId,1000			token: TokenId,1001			for_nest: (CollectionId, TokenId),1002			sender: T::CrossAccountId,1003			budget: &dyn Budget,1004		) -> DispatchResult {1005			ensure!(1006				<PalletStructure<T>>::check_indirectly_owned(1007					sender,1008					collection,1009					token,1010					Some(for_nest),1011					budget1012				)?,1013				<CommonError<T>>::OnlyOwnerAllowedToNest,1014			);1015			Ok(())1016		}1017		match handle.permissions.nesting() {1018			NestingRule::Disabled => fail!(<CommonError<T>>::NestingIsDisabled),1019			NestingRule::Owner => {1020				ensure_sender_allowed::<T>(handle.id, under, from, sender, nesting_budget)?1021			}1022			NestingRule::OwnerRestricted(whitelist) => {1023				ensure!(1024					whitelist.contains(&from.0),1025					<CommonError<T>>::SourceCollectionIsNotAllowedToNest1026				);1027				ensure_sender_allowed::<T>(handle.id, under, from, sender, nesting_budget)?1028			}1029			NestingRule::Permissive => {}1030		}1031		Ok(())1032	}10331034	fn nest(under: (CollectionId, TokenId), to_nest: (CollectionId, TokenId)) {1035		<TokenChildren<T>>::insert((under.0, under.1, (to_nest.0, to_nest.1)), true);1036	}10371038	fn unnest(under: (CollectionId, TokenId), to_unnest: (CollectionId, TokenId)) {1039		<TokenChildren<T>>::remove((under.0, under.1, to_unnest));1040	}10411042	fn collection_has_tokens(collection_id: CollectionId) -> bool {1043		<TokenData<T>>::iter_prefix((collection_id,))1044			.next()1045			.is_some()1046	}10471048	fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {1049		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1050			.next()1051			.is_some()1052	}10531054	pub fn token_children_ids(collection_id: CollectionId, token_id: TokenId) -> Vec<TokenChild> {1055		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1056			.map(|((child_collection_id, child_id), _)| TokenChild {1057				collection: child_collection_id,1058				token: child_id,1059			})1060			.collect()1061	}10621063	/// Delegated to `create_multiple_items`1064	pub fn create_item(1065		collection: &NonfungibleHandle<T>,1066		sender: &T::CrossAccountId,1067		data: CreateItemData<T>,1068		nesting_budget: &dyn Budget,1069	) -> DispatchResult {1070		Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1071	}1072}
modifiedpallets/nonfungible/src/stubs/UniqueNFT.rawdiffbeforeafterboth

binary blob — no preview

modifiedpallets/nonfungible/src/stubs/UniqueNFT.soldiffbeforeafterboth
--- a/pallets/nonfungible/src/stubs/UniqueNFT.sol
+++ b/pallets/nonfungible/src/stubs/UniqueNFT.sol
@@ -51,135 +51,6 @@
 	event MintingFinished();
 }
 
-// Selector: 1248b7d1
-contract Collection is Dummy, ERC165 {
-	// Selector: setCollectionProperty(string,bytes) 2f073f66
-	function setCollectionProperty(string memory key, bytes memory value)
-		public
-	{
-		require(false, stub_error);
-		key;
-		value;
-		dummy = 0;
-	}
-
-	// Selector: deleteCollectionProperty(string) 7b7debce
-	function deleteCollectionProperty(string memory key) public {
-		require(false, stub_error);
-		key;
-		dummy = 0;
-	}
-
-	// Throws error if key not found
-	//
-	// Selector: collectionProperty(string) cf24fd6d
-	function collectionProperty(string memory key)
-		public
-		view
-		returns (bytes memory)
-	{
-		require(false, stub_error);
-		key;
-		dummy;
-		return hex"";
-	}
-
-	// Selector: setCollectionSponsor(address) 7623402e
-	function setCollectionSponsor(address sponsor) public {
-		require(false, stub_error);
-		sponsor;
-		dummy = 0;
-	}
-
-	// Selector: confirmCollectionSponsorship() 3c50e97a
-	function confirmCollectionSponsorship() public {
-		require(false, stub_error);
-		dummy = 0;
-	}
-
-	// Selector: setCollectionLimit(string,uint32) 6a3841db
-	function setCollectionLimit(string memory limit, uint32 value) public {
-		require(false, stub_error);
-		limit;
-		value;
-		dummy = 0;
-	}
-
-	// Selector: setCollectionLimit(string,bool) 993b7fba
-	function setCollectionLimit(string memory limit, bool value) public {
-		require(false, stub_error);
-		limit;
-		value;
-		dummy = 0;
-	}
-
-	// Selector: contractAddress() f6b4dfb4
-	function contractAddress() public view returns (address) {
-		require(false, stub_error);
-		dummy;
-		return 0x0000000000000000000000000000000000000000;
-	}
-
-	// Selector: addCollectionAdmin(address) 92e462c7
-	function addCollectionAdmin(address newAdmin) public view {
-		require(false, stub_error);
-		newAdmin;
-		dummy;
-	}
-
-	// Selector: removeCollectionAdmin(address) fafd7b42
-	function removeCollectionAdmin(address admin) public view {
-		require(false, stub_error);
-		admin;
-		dummy;
-	}
-
-	// Selector: setCollectionNesting(bool) 112d4586
-	function setCollectionNesting(bool enable) public {
-		require(false, stub_error);
-		enable;
-		dummy = 0;
-	}
-
-	// Selector: setCollectionNesting(bool,address[]) 64872396
-	function setCollectionNesting(bool enable, address[] memory collections)
-		public
-	{
-		require(false, stub_error);
-		enable;
-		collections;
-		dummy = 0;
-	}
-
-	// Selector: setCollectionAccess(string) 392172a9
-	function setCollectionAccess(string memory mode) public {
-		require(false, stub_error);
-		mode;
-		dummy = 0;
-	}
-
-	// Selector: addToCollectionAllowList(address) 67844fe6
-	function addToCollectionAllowList(address user) public view {
-		require(false, stub_error);
-		user;
-		dummy;
-	}
-
-	// Selector: removeFromCollectionAllowList(address) 85c51acb
-	function removeFromCollectionAllowList(address user) public view {
-		require(false, stub_error);
-		user;
-		dummy;
-	}
-
-	// Selector: setCollectionMintMode(bool) 00018e84
-	function setCollectionMintMode(bool mode) public {
-		require(false, stub_error);
-		mode;
-		dummy = 0;
-	}
-}
-
 // Selector: 41369377
 contract TokenProperties is Dummy, ERC165 {
 	// Selector: setTokenPropertyPermission(string,bool,bool,bool) 222d97fa
@@ -426,6 +297,135 @@
 	}
 }
 
+// Selector: 6aea9834
+contract Collection is Dummy, ERC165 {
+	// Selector: setCollectionProperty(string,bytes) 2f073f66
+	function setCollectionProperty(string memory key, bytes memory value)
+		public
+	{
+		require(false, stub_error);
+		key;
+		value;
+		dummy = 0;
+	}
+
+	// Selector: deleteCollectionProperty(string) 7b7debce
+	function deleteCollectionProperty(string memory key) public {
+		require(false, stub_error);
+		key;
+		dummy = 0;
+	}
+
+	// Throws error if key not found
+	//
+	// Selector: collectionProperty(string) cf24fd6d
+	function collectionProperty(string memory key)
+		public
+		view
+		returns (bytes memory)
+	{
+		require(false, stub_error);
+		key;
+		dummy;
+		return hex"";
+	}
+
+	// Selector: setCollectionSponsor(address) 7623402e
+	function setCollectionSponsor(address sponsor) public {
+		require(false, stub_error);
+		sponsor;
+		dummy = 0;
+	}
+
+	// Selector: confirmCollectionSponsorship() 3c50e97a
+	function confirmCollectionSponsorship() public {
+		require(false, stub_error);
+		dummy = 0;
+	}
+
+	// Selector: setCollectionLimit(string,uint32) 6a3841db
+	function setCollectionLimit(string memory limit, uint32 value) public {
+		require(false, stub_error);
+		limit;
+		value;
+		dummy = 0;
+	}
+
+	// Selector: setCollectionLimit(string,bool) 993b7fba
+	function setCollectionLimit(string memory limit, bool value) public {
+		require(false, stub_error);
+		limit;
+		value;
+		dummy = 0;
+	}
+
+	// Selector: contractAddress() f6b4dfb4
+	function contractAddress() public view returns (address) {
+		require(false, stub_error);
+		dummy;
+		return 0x0000000000000000000000000000000000000000;
+	}
+
+	// Selector: addCollectionAdmin(address) 92e462c7
+	function addCollectionAdmin(address newAdmin) public view {
+		require(false, stub_error);
+		newAdmin;
+		dummy;
+	}
+
+	// Selector: removeCollectionAdmin(address) fafd7b42
+	function removeCollectionAdmin(address admin) public view {
+		require(false, stub_error);
+		admin;
+		dummy;
+	}
+
+	// Selector: setCollectionNesting(bool) 112d4586
+	function setCollectionNesting(bool enable) public {
+		require(false, stub_error);
+		enable;
+		dummy = 0;
+	}
+
+	// Selector: setCollectionNesting(bool,address[]) 64872396
+	function setCollectionNesting(bool enable, address[] memory collections)
+		public
+	{
+		require(false, stub_error);
+		enable;
+		collections;
+		dummy = 0;
+	}
+
+	// Selector: setCollectionAccess(uint8) 41835d4c
+	function setCollectionAccess(uint8 mode) public {
+		require(false, stub_error);
+		mode;
+		dummy = 0;
+	}
+
+	// Selector: addToCollectionAllowList(address) 67844fe6
+	function addToCollectionAllowList(address user) public view {
+		require(false, stub_error);
+		user;
+		dummy;
+	}
+
+	// Selector: removeFromCollectionAllowList(address) 85c51acb
+	function removeFromCollectionAllowList(address user) public view {
+		require(false, stub_error);
+		user;
+		dummy;
+	}
+
+	// Selector: setCollectionMintMode(bool) 00018e84
+	function setCollectionMintMode(bool mode) public {
+		require(false, stub_error);
+		mode;
+		dummy = 0;
+	}
+}
+
 // Selector: 780e9d63
 contract ERC721Enumerable is Dummy, ERC165 {
 	// Selector: tokenByIndex(uint256) 4f6ccce7
modifiedtests/src/eth/api/UniqueNFT.soldiffbeforeafterboth
--- a/tests/src/eth/api/UniqueNFT.sol
+++ b/tests/src/eth/api/UniqueNFT.sol
@@ -42,64 +42,6 @@
 	event MintingFinished();
 }
 
-// Selector: 1248b7d1
-interface Collection is Dummy, ERC165 {
-	// Selector: setCollectionProperty(string,bytes) 2f073f66
-	function setCollectionProperty(string memory key, bytes memory value)
-		external;
-
-	// Selector: deleteCollectionProperty(string) 7b7debce
-	function deleteCollectionProperty(string memory key) external;
-
-	// Throws error if key not found
-	//
-	// Selector: collectionProperty(string) cf24fd6d
-	function collectionProperty(string memory key)
-		external
-		view
-		returns (bytes memory);
-
-	// Selector: setCollectionSponsor(address) 7623402e
-	function setCollectionSponsor(address sponsor) external;
-
-	// Selector: confirmCollectionSponsorship() 3c50e97a
-	function confirmCollectionSponsorship() external;
-
-	// Selector: setCollectionLimit(string,uint32) 6a3841db
-	function setCollectionLimit(string memory limit, uint32 value) external;
-
-	// Selector: setCollectionLimit(string,bool) 993b7fba
-	function setCollectionLimit(string memory limit, bool value) external;
-
-	// Selector: contractAddress() f6b4dfb4
-	function contractAddress() external view returns (address);
-
-	// Selector: addCollectionAdmin(address) 92e462c7
-	function addCollectionAdmin(address newAdmin) external view;
-
-	// Selector: removeCollectionAdmin(address) fafd7b42
-	function removeCollectionAdmin(address admin) external view;
-
-	// Selector: setCollectionNesting(bool) 112d4586
-	function setCollectionNesting(bool enable) external;
-
-	// Selector: setCollectionNesting(bool,address[]) 64872396
-	function setCollectionNesting(bool enable, address[] memory collections)
-		external;
-
-	// Selector: setCollectionAccess(string) 392172a9
-	function setCollectionAccess(string memory mode) external;
-
-	// Selector: addToCollectionAllowList(address) 67844fe6
-	function addToCollectionAllowList(address user) external view;
-
-	// Selector: removeFromCollectionAllowList(address) 85c51acb
-	function removeFromCollectionAllowList(address user) external view;
-
-	// Selector: setCollectionMintMode(bool) 00018e84
-	function setCollectionMintMode(bool mode) external;
-}
-
 // Selector: 41369377
 interface TokenProperties is Dummy, ERC165 {
 	// Selector: setTokenPropertyPermission(string,bool,bool,bool) 222d97fa
@@ -232,6 +174,64 @@
 	function finishMinting() external returns (bool);
 }
 
+// Selector: 6aea9834
+interface Collection is Dummy, ERC165 {
+	// Selector: setCollectionProperty(string,bytes) 2f073f66
+	function setCollectionProperty(string memory key, bytes memory value)
+		external;
+
+	// Selector: deleteCollectionProperty(string) 7b7debce
+	function deleteCollectionProperty(string memory key) external;
+
+	// Throws error if key not found
+	//
+	// Selector: collectionProperty(string) cf24fd6d
+	function collectionProperty(string memory key)
+		external
+		view
+		returns (bytes memory);
+
+	// Selector: setCollectionSponsor(address) 7623402e
+	function setCollectionSponsor(address sponsor) external;
+
+	// Selector: confirmCollectionSponsorship() 3c50e97a
+	function confirmCollectionSponsorship() external;
+
+	// Selector: setCollectionLimit(string,uint32) 6a3841db
+	function setCollectionLimit(string memory limit, uint32 value) external;
+
+	// Selector: setCollectionLimit(string,bool) 993b7fba
+	function setCollectionLimit(string memory limit, bool value) external;
+
+	// Selector: contractAddress() f6b4dfb4
+	function contractAddress() external view returns (address);
+
+	// Selector: addCollectionAdmin(address) 92e462c7
+	function addCollectionAdmin(address newAdmin) external view;
+
+	// Selector: removeCollectionAdmin(address) fafd7b42
+	function removeCollectionAdmin(address admin) external view;
+
+	// Selector: setCollectionNesting(bool) 112d4586
+	function setCollectionNesting(bool enable) external;
+
+	// Selector: setCollectionNesting(bool,address[]) 64872396
+	function setCollectionNesting(bool enable, address[] memory collections)
+		external;
+
+	// Selector: setCollectionAccess(uint8) 41835d4c
+	function setCollectionAccess(uint8 mode) external;
+
+	// Selector: addToCollectionAllowList(address) 67844fe6
+	function addToCollectionAllowList(address user) external view;
+
+	// Selector: removeFromCollectionAllowList(address) 85c51acb
+	function removeFromCollectionAllowList(address user) external view;
+
+	// Selector: setCollectionMintMode(bool) 00018e84
+	function setCollectionMintMode(bool mode) external;
+}
+
 // Selector: 780e9d63
 interface ERC721Enumerable is Dummy, ERC165 {
 	// Selector: tokenByIndex(uint256) 4f6ccce7
modifiedtests/src/eth/contractSponsoring.test.tsdiffbeforeafterboth
--- a/tests/src/eth/contractSponsoring.test.ts
+++ b/tests/src/eth/contractSponsoring.test.ts
@@ -247,7 +247,8 @@
     expect(oldPermissions.mintMode).to.be.false;
     expect(oldPermissions.access).to.be.equal('Normal');
 
-    await collectionEvm.methods.setCollectionAccess('AllowList').send({from: owner});
+    //TODO: change value, when enum generated
+    await collectionEvm.methods.setCollectionAccess(1 /*'AllowList'*/).send({from: owner});
     await collectionEvm.methods.addToCollectionAllowList(user).send({from: owner});
     await collectionEvm.methods.setCollectionMintMode(true).send({from: owner});
 
modifiedtests/src/eth/nonFungibleAbi.jsondiffbeforeafterboth
--- a/tests/src/eth/nonFungibleAbi.json
+++ b/tests/src/eth/nonFungibleAbi.json
@@ -341,7 +341,7 @@
     "type": "function"
   },
   {
-    "inputs": [{ "internalType": "string", "name": "mode", "type": "string" }],
+    "inputs": [{ "internalType": "uint8", "name": "mode", "type": "uint8" }],
     "name": "setCollectionAccess",
     "outputs": [],
     "stateMutability": "nonpayable",