git.delta.rocks / unique-network / refs/commits / 7cf62e7cc906

difftreelog

source

pallets/common/src/lib.rs35.7 KiBsourcehistory
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)]1819extern crate alloc;2021use core::ops::{Deref, DerefMut};22use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};23use sp_std::vec::Vec;24use pallet_evm::account::CrossAccountId;25use frame_support::{26	dispatch::{DispatchErrorWithPostInfo, DispatchResultWithPostInfo, Weight, PostDispatchInfo},27	ensure,28	traits::{Imbalance, Get, Currency, WithdrawReasons, ExistenceRequirement},29	BoundedVec,30	weights::Pays,31	transactional,32};33use pallet_evm::GasWeightMapping;34use up_data_structs::{35	COLLECTION_NUMBER_LIMIT,36	Collection,37	RpcCollection,38	CollectionId,39	CreateItemData,40	MAX_TOKEN_PREFIX_LENGTH,41	COLLECTION_ADMINS_LIMIT,42	TokenId,43	CollectionStats,44	MAX_TOKEN_OWNERSHIP,45	CollectionMode,46	NFT_SPONSOR_TRANSFER_TIMEOUT,47	FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,48	REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,49	MAX_SPONSOR_TIMEOUT,50	CUSTOM_DATA_LIMIT,51	CollectionLimits,52	CreateCollectionData,53	SponsorshipState,54	CreateItemExData,55	SponsoringRateLimit,56	budget::Budget,57	COLLECTION_FIELD_LIMIT,58	PhantomType,59	Property,60	Properties,61	PropertiesPermissionMap,62	PropertyKey,63	PropertyValue,64	PropertyPermission,65	PropertiesError,66	PropertyKeyPermission,67	TokenData,68	TrySetProperty,69	PropertyScope,70	// RMRK71	RmrkCollectionInfo,72	RmrkInstanceInfo,73	RmrkResourceInfo,74	RmrkPropertyInfo,75	RmrkBaseInfo,76	RmrkPartType,77	RmrkTheme,78	RmrkNftChild,79	CollectionPermissions,80	SchemaVersion,81};8283pub use pallet::*;84use sp_core::H160;85use sp_runtime::{ArithmeticError, DispatchError, DispatchResult};86#[cfg(feature = "runtime-benchmarks")]87pub mod benchmarking;88pub mod dispatch;89pub mod erc;90pub mod eth;91pub mod weights;9293pub type SelfWeightOf<T> = <T as Config>::WeightInfo;9495#[must_use = "Should call submit_logs or save, otherwise some data will be lost for evm side"]96pub struct CollectionHandle<T: Config> {97	pub id: CollectionId,98	collection: Collection<T::AccountId>,99	pub recorder: SubstrateRecorder<T>,100}101impl<T: Config> WithRecorder<T> for CollectionHandle<T> {102	fn recorder(&self) -> &SubstrateRecorder<T> {103		&self.recorder104	}105	fn into_recorder(self) -> SubstrateRecorder<T> {106		self.recorder107	}108}109impl<T: Config> CollectionHandle<T> {110	pub fn new_with_gas_limit(id: CollectionId, gas_limit: u64) -> Option<Self> {111		<CollectionById<T>>::get(id).map(|collection| Self {112			id,113			collection,114			recorder: SubstrateRecorder::new(gas_limit),115		})116	}117118	pub fn new_with_recorder(id: CollectionId, recorder: SubstrateRecorder<T>) -> Option<Self> {119		<CollectionById<T>>::get(id).map(|collection| Self {120			id,121			collection,122			recorder,123		})124	}125126	pub fn new(id: CollectionId) -> Option<Self> {127		Self::new_with_gas_limit(id, u64::MAX)128	}129	pub fn try_get(id: CollectionId) -> Result<Self, DispatchError> {130		Ok(Self::new(id).ok_or(<Error<T>>::CollectionNotFound)?)131	}132	pub fn consume_store_reads(&self, reads: u64) -> evm_coder::execution::Result<()> {133		self.recorder134			.consume_gas(T::GasWeightMapping::weight_to_gas(135				<T as frame_system::Config>::DbWeight::get()136					.read137					.saturating_mul(reads),138			))139	}140	pub fn consume_store_writes(&self, writes: u64) -> evm_coder::execution::Result<()> {141		self.recorder142			.consume_gas(T::GasWeightMapping::weight_to_gas(143				<T as frame_system::Config>::DbWeight::get()144					.write145					.saturating_mul(writes),146			))147	}148	pub fn save(self) -> DispatchResult {149		<CollectionById<T>>::insert(self.id, self.collection);150		Ok(())151	}152153	pub fn set_sponsor(&mut self, sponsor: T::AccountId) {154		self.collection.sponsorship = SponsorshipState::Unconfirmed(sponsor);155	}156157	pub fn confirm_sponsorship(&mut self, sender: &T::AccountId) -> bool {158		if self.collection.sponsorship.pending_sponsor() != Some(sender) {159			return false;160		};161162		self.collection.sponsorship = SponsorshipState::Confirmed(sender.clone());163		true164	}165}166impl<T: Config> Deref for CollectionHandle<T> {167	type Target = Collection<T::AccountId>;168169	fn deref(&self) -> &Self::Target {170		&self.collection171	}172}173174impl<T: Config> DerefMut for CollectionHandle<T> {175	fn deref_mut(&mut self) -> &mut Self::Target {176		&mut self.collection177	}178}179180impl<T: Config> CollectionHandle<T> {181	pub fn check_is_owner(&self, subject: &T::CrossAccountId) -> DispatchResult {182		ensure!(*subject.as_sub() == self.owner, <Error<T>>::NoPermission);183		Ok(())184	}185	pub fn is_owner_or_admin(&self, subject: &T::CrossAccountId) -> bool {186		*subject.as_sub() == self.owner || <IsAdmin<T>>::get((self.id, subject))187	}188	pub fn check_is_owner_or_admin(&self, subject: &T::CrossAccountId) -> DispatchResult {189		ensure!(self.is_owner_or_admin(subject), <Error<T>>::NoPermission);190		Ok(())191	}192	pub fn ignores_allowance(&self, user: &T::CrossAccountId) -> bool {193		self.limits.owner_can_transfer() && self.is_owner_or_admin(user)194	}195	pub fn ignores_owned_amount(&self, user: &T::CrossAccountId) -> bool {196		self.limits.owner_can_transfer() && self.is_owner_or_admin(user)197	}198	pub fn check_allowlist(&self, user: &T::CrossAccountId) -> DispatchResult {199		ensure!(200			<Allowlist<T>>::get((self.id, user)),201			<Error<T>>::AddressNotInAllowlist202		);203		Ok(())204	}205}206207#[frame_support::pallet]208pub mod pallet {209	use super::*;210	use pallet_evm::account;211	use dispatch::CollectionDispatch;212	use frame_support::{Blake2_128Concat, pallet_prelude::*, storage::Key, traits::StorageVersion};213	use frame_system::pallet_prelude::*;214	use frame_support::traits::Currency;215	use up_data_structs::{TokenId, mapping::TokenAddressMapping};216	use scale_info::TypeInfo;217	use weights::WeightInfo;218219	#[pallet::config]220	pub trait Config:221		frame_system::Config + pallet_evm_coder_substrate::Config + TypeInfo + account::Config222	{223		type WeightInfo: WeightInfo;224		type Event: IsType<<Self as frame_system::Config>::Event> + From<Event<Self>>;225226		type Currency: Currency<Self::AccountId>;227228		#[pallet::constant]229		type CollectionCreationPrice: Get<230			<<Self as Config>::Currency as Currency<Self::AccountId>>::Balance,231		>;232		type CollectionDispatch: CollectionDispatch<Self>;233234		type TreasuryAccountId: Get<Self::AccountId>;235236		type EvmTokenAddressMapping: TokenAddressMapping<H160>;237		type CrossTokenAddressMapping: TokenAddressMapping<Self::CrossAccountId>;238	}239240	const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);241242	#[pallet::pallet]243	#[pallet::storage_version(STORAGE_VERSION)]244	#[pallet::generate_store(pub(super) trait Store)]245	pub struct Pallet<T>(_);246247	#[pallet::extra_constants]248	impl<T: Config> Pallet<T> {249		pub fn collection_admins_limit() -> u32 {250			COLLECTION_ADMINS_LIMIT251		}252	}253254	#[pallet::event]255	#[pallet::generate_deposit(pub fn deposit_event)]256	pub enum Event<T: Config> {257		/// New collection was created258		///259		/// # Arguments260		///261		/// * collection_id: Globally unique identifier of newly created collection.262		///263		/// * mode: [CollectionMode] converted into u8.264		///265		/// * account_id: Collection owner.266		CollectionCreated(CollectionId, u8, T::AccountId),267268		/// New collection was destroyed269		///270		/// # Arguments271		///272		/// * collection_id: Globally unique identifier of collection.273		CollectionDestroyed(CollectionId),274275		/// New item was created.276		///277		/// # Arguments278		///279		/// * collection_id: Id of the collection where item was created.280		///281		/// * item_id: Id of an item. Unique within the collection.282		///283		/// * recipient: Owner of newly created item284		///285		/// * amount: Always 1 for NFT286		ItemCreated(CollectionId, TokenId, T::CrossAccountId, u128),287288		/// Collection item was burned.289		///290		/// # Arguments291		///292		/// * collection_id.293		///294		/// * item_id: Identifier of burned NFT.295		///296		/// * owner: which user has destroyed its tokens297		///298		/// * amount: Always 1 for NFT299		ItemDestroyed(CollectionId, TokenId, T::CrossAccountId, u128),300301		/// Item was transferred302		///303		/// * collection_id: Id of collection to which item is belong304		///305		/// * item_id: Id of an item306		///307		/// * sender: Original owner of item308		///309		/// * recipient: New owner of item310		///311		/// * amount: Always 1 for NFT312		Transfer(313			CollectionId,314			TokenId,315			T::CrossAccountId,316			T::CrossAccountId,317			u128,318		),319320		/// * collection_id321		///322		/// * item_id323		///324		/// * sender325		///326		/// * spender327		///328		/// * amount329		Approved(330			CollectionId,331			TokenId,332			T::CrossAccountId,333			T::CrossAccountId,334			u128,335		),336337		CollectionPropertySet(CollectionId, PropertyKey),338339		CollectionPropertyDeleted(CollectionId, PropertyKey),340341		TokenPropertySet(CollectionId, TokenId, PropertyKey),342343		TokenPropertyDeleted(CollectionId, TokenId, PropertyKey),344345		PropertyPermissionSet(CollectionId, PropertyKey),346	}347348	#[pallet::error]349	pub enum Error<T> {350		/// This collection does not exist.351		CollectionNotFound,352		/// Sender parameter and item owner must be equal.353		MustBeTokenOwner,354		/// No permission to perform action355		NoPermission,356		/// Destroying only empty collections is allowed357		CantDestroyNotEmptyCollection,358		/// Collection is not in mint mode.359		PublicMintingNotAllowed,360		/// Address is not in allow list.361		AddressNotInAllowlist,362363		/// Collection name can not be longer than 63 char.364		CollectionNameLimitExceeded,365		/// Collection description can not be longer than 255 char.366		CollectionDescriptionLimitExceeded,367		/// Token prefix can not be longer than 15 char.368		CollectionTokenPrefixLimitExceeded,369		/// Total collections bound exceeded.370		TotalCollectionsLimitExceeded,371		/// Exceeded max admin count372		CollectionAdminCountExceeded,373		/// Collection limit bounds per collection exceeded374		CollectionLimitBoundsExceeded,375		/// Tried to enable permissions which are only permitted to be disabled376		OwnerPermissionsCantBeReverted,377		/// Collection settings not allowing items transferring378		TransferNotAllowed,379		/// Account token limit exceeded per collection380		AccountTokenLimitExceeded,381		/// Collection token limit exceeded382		CollectionTokenLimitExceeded,383		/// Metadata flag frozen384		MetadataFlagFrozen,385386		/// Item not exists.387		TokenNotFound,388		/// Item balance not enough.389		TokenValueTooLow,390		/// Requested value more than approved.391		ApprovedValueTooLow,392		/// Tried to approve more than owned393		CantApproveMoreThanOwned,394395		/// Can't transfer tokens to ethereum zero address396		AddressIsZero,397		/// Target collection doesn't supports this operation398		UnsupportedOperation,399400		/// Not sufficient founds to perform action401		NotSufficientFounds,402403		/// Collection has nesting disabled404		NestingIsDisabled,405		/// Only owner may nest tokens under this collection406		OnlyOwnerAllowedToNest,407		/// Only tokens from specific collections may nest tokens under this408		SourceCollectionIsNotAllowedToNest,409410		/// Tried to store more data than allowed in collection field411		CollectionFieldSizeExceeded,412413		/// Tried to store more property data than allowed414		NoSpaceForProperty,415416		/// Tried to store more property keys than allowed417		PropertyLimitReached,418419		/// Property key is too long420		PropertyKeyIsTooLong,421422		/// Only ASCII letters, digits, and '_', '-' are allowed423		InvalidCharacterInPropertyKey,424425		/// Empty property keys are forbidden426		EmptyPropertyKey,427	}428429	#[pallet::storage]430	pub type CreatedCollectionCount<T> = StorageValue<Value = CollectionId, QueryKind = ValueQuery>;431	#[pallet::storage]432	pub type DestroyedCollectionCount<T> =433		StorageValue<Value = CollectionId, QueryKind = ValueQuery>;434435	/// Collection info436	#[pallet::storage]437	pub type CollectionById<T> = StorageMap<438		Hasher = Blake2_128Concat,439		Key = CollectionId,440		Value = Collection<<T as frame_system::Config>::AccountId>,441		QueryKind = OptionQuery,442	>;443444	/// Collection properties445	#[pallet::storage]446	#[pallet::getter(fn collection_properties)]447	pub type CollectionProperties<T> = StorageMap<448		Hasher = Blake2_128Concat,449		Key = CollectionId,450		Value = Properties,451		QueryKind = ValueQuery,452		OnEmpty = up_data_structs::CollectionProperties,453	>;454455	#[pallet::storage]456	#[pallet::getter(fn property_permissions)]457	pub type CollectionPropertyPermissions<T> = StorageMap<458		Hasher = Blake2_128Concat,459		Key = CollectionId,460		Value = PropertiesPermissionMap,461		QueryKind = ValueQuery,462	>;463464	#[pallet::storage]465	pub type AdminAmount<T> = StorageMap<466		Hasher = Blake2_128Concat,467		Key = CollectionId,468		Value = u32,469		QueryKind = ValueQuery,470	>;471472	/// List of collection admins473	#[pallet::storage]474	pub type IsAdmin<T: Config> = StorageNMap<475		Key = (476			Key<Blake2_128Concat, CollectionId>,477			Key<Blake2_128Concat, T::CrossAccountId>,478		),479		Value = bool,480		QueryKind = ValueQuery,481	>;482483	/// Allowlisted collection users484	#[pallet::storage]485	pub type Allowlist<T: Config> = StorageNMap<486		Key = (487			Key<Blake2_128Concat, CollectionId>,488			Key<Blake2_128Concat, T::CrossAccountId>,489		),490		Value = bool,491		QueryKind = ValueQuery,492	>;493494	/// Not used by code, exists only to provide some types to metadata495	#[pallet::storage]496	pub type DummyStorageValue<T: Config> = StorageValue<497		Value = (498			CollectionStats,499			CollectionId,500			TokenId,501			PhantomType<(502				TokenData<T::CrossAccountId>,503				RpcCollection<T::AccountId>,504505				// RMRK506				RmrkCollectionInfo<T::AccountId>,507				RmrkInstanceInfo<T::AccountId>,508				RmrkResourceInfo,509				RmrkPropertyInfo,510				RmrkBaseInfo<T::AccountId>,511				RmrkPartType,512				RmrkTheme,513				RmrkNftChild,514			)>,515		),516		QueryKind = OptionQuery,517	>;518519	#[pallet::hooks]520	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {521		fn on_runtime_upgrade() -> Weight {522			if StorageVersion::get::<Pallet<T>>() < StorageVersion::new(1) {523				use up_data_structs::{CollectionVersion1, CollectionVersion2};524				<CollectionById<T>>::translate::<CollectionVersion1<T::AccountId>, _>(|id, v| {525					let mut props = Vec::new();526					if !v.offchain_schema.is_empty() {527						props.push(Property {528							key: b"_old_offchainSchema".to_vec().try_into().unwrap(),529							value: v.offchain_schema.clone().into_inner().try_into().expect("offchain schema too big"),530						});531					}532					if !v.variable_on_chain_schema.is_empty() {533						props.push(Property {534							key: b"_old_variableOnChainSchema".to_vec().try_into().unwrap(),535							value: v.variable_on_chain_schema.clone().into_inner().try_into().expect("offchain schema too big"),536						});537					}538					if !v.const_on_chain_schema.is_empty() {539						props.push(Property {540							key: b"_old_constOnChainSchema".to_vec().try_into().unwrap(),541							value: v.const_on_chain_schema.clone().into_inner().try_into().expect("offchain schema too big"),542						});543					}544					props.push(Property {545						key: b"_old_schemaVersion".to_vec().try_into().unwrap(),546						value: match v.schema_version {547							SchemaVersion::ImageURL => b"ImageUrl".as_slice(),548							SchemaVersion::Unique => b"Unique".as_slice(),549						}.to_vec().try_into().unwrap(),550					});551					Self::set_scoped_collection_properties(552						id,553						PropertyScope::None,554						props.into_iter(),555					).expect("existing data larger than properties");556					let mut new = CollectionVersion2::from(v.clone());557					new.permissions.access = Some(v.access);558					new.permissions.mint_mode = Some(v.mint_mode);559					Some(new)560				});561			}562563			0564		}565	}566}567568impl<T: Config> Pallet<T> {569	/// Ethereum receiver 0x0000000000000000000000000000000000000000 is reserved, and shouldn't own tokens570	pub fn ensure_correct_receiver(receiver: &T::CrossAccountId) -> DispatchResult {571		ensure!(572			&T::CrossAccountId::from_eth(H160([0; 20])) != receiver,573			<Error<T>>::AddressIsZero574		);575		Ok(())576	}577	pub fn adminlist(collection: CollectionId) -> Vec<T::CrossAccountId> {578		<IsAdmin<T>>::iter_prefix((collection,))579			.map(|(a, _)| a)580			.collect()581	}582	pub fn allowlist(collection: CollectionId) -> Vec<T::CrossAccountId> {583		<Allowlist<T>>::iter_prefix((collection,))584			.map(|(a, _)| a)585			.collect()586	}587	pub fn allowed(collection: CollectionId, user: T::CrossAccountId) -> bool {588		<Allowlist<T>>::get((collection, user))589	}590	pub fn collection_stats() -> CollectionStats {591		let created = <CreatedCollectionCount<T>>::get();592		let destroyed = <DestroyedCollectionCount<T>>::get();593		CollectionStats {594			created: created.0,595			destroyed: destroyed.0,596			alive: created.0 - destroyed.0,597		}598	}599600	pub fn effective_collection_limits(collection: CollectionId) -> Option<CollectionLimits> {601		let collection = <CollectionById<T>>::get(collection);602		if collection.is_none() {603			return None;604		}605606		let collection = collection.unwrap();607		let limits = collection.limits;608		let effective_limits = CollectionLimits {609			account_token_ownership_limit: Some(limits.account_token_ownership_limit()),610			sponsored_data_size: Some(limits.sponsored_data_size()),611			sponsored_data_rate_limit: Some(612				limits613					.sponsored_data_rate_limit614					.unwrap_or(SponsoringRateLimit::SponsoringDisabled),615			),616			token_limit: Some(limits.token_limit()),617			sponsor_transfer_timeout: Some(limits.sponsor_transfer_timeout(618				match collection.mode {619					CollectionMode::NFT => NFT_SPONSOR_TRANSFER_TIMEOUT,620					CollectionMode::Fungible(_) => FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,621					CollectionMode::ReFungible => REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,622				},623			)),624			sponsor_approve_timeout: Some(limits.sponsor_approve_timeout()),625			owner_can_transfer: Some(limits.owner_can_transfer()),626			owner_can_destroy: Some(limits.owner_can_destroy()),627			transfers_enabled: Some(limits.transfers_enabled()),628		};629630		Some(effective_limits)631	}632633	pub fn rpc_collection(collection: CollectionId) -> Option<RpcCollection<T::AccountId>> {634		let Collection {635			name,636			description,637			owner,638			mode,639			token_prefix,640			sponsorship,641			limits,642			permissions,643		} = <CollectionById<T>>::get(collection)?;644645		let token_property_permissions = <CollectionPropertyPermissions<T>>::get(collection)646			.into_iter()647			.map(|(key, permission)| PropertyKeyPermission {648				key,649				permission,650			})651			.collect();652653		let properties = <CollectionProperties<T>>::get(collection)654			.into_iter()655			.map(|(key, value)| Property {656				key,657				value,658			})659			.collect();660661		let permissions = CollectionPermissions {662			access: Some(permissions.access()),663			mint_mode: Some(permissions.mint_mode()),664			nesting: Some(permissions.nesting().clone()),665		};666667		Some(RpcCollection {668			name: name.into_inner(),669			description: description.into_inner(),670			owner,671			mode,672			token_prefix: token_prefix.into_inner(),673			sponsorship,674			limits,675			permissions,676			token_property_permissions,677			properties,678		})679	}680}681682macro_rules! limit_default {683	($old:ident, $new:ident, $($field:ident $(($arg:expr))? => $check:expr),* $(,)?) => {{684		$(685			if let Some($new) = $new.$field {686				let $old = $old.$field($($arg)?);687				let _ = $new;688				let _ = $old;689				$check690			} else {691				$new.$field = $old.$field692			}693		)*694	}};695}696macro_rules! limit_default_clone {697	($old:ident, $new:ident, $($field:ident $(($arg:expr))? => $check:expr),* $(,)?) => {{698		$(699			if let Some($new) = $new.$field.clone() {700				let $old = $old.$field($($arg)?);701				let _ = $new;702				let _ = $old;703				$check704			} else {705				$new.$field = $old.$field.clone()706			}707		)*708	}};709}710711impl<T: Config> Pallet<T> {712	pub fn init_collection(713		owner: T::AccountId,714		data: CreateCollectionData<T::AccountId>,715	) -> Result<CollectionId, DispatchError> {716		{717			ensure!(718				data.token_prefix.len() <= MAX_TOKEN_PREFIX_LENGTH as usize,719				Error::<T>::CollectionTokenPrefixLimitExceeded720			);721		}722723		let created_count = <CreatedCollectionCount<T>>::get()724			.0725			.checked_add(1)726			.ok_or(ArithmeticError::Overflow)?;727		let destroyed_count = <DestroyedCollectionCount<T>>::get().0;728		let id = CollectionId(created_count);729730		// bound Total number of collections731		ensure!(732			created_count - destroyed_count <= COLLECTION_NUMBER_LIMIT,733			<Error<T>>::TotalCollectionsLimitExceeded734		);735736		// =========737738		let collection = Collection {739			owner: owner.clone(),740			name: data.name,741			mode: data.mode.clone(),742			description: data.description,743			token_prefix: data.token_prefix,744			sponsorship: data745				.pending_sponsor746				.map(SponsorshipState::Unconfirmed)747				.unwrap_or_default(),748			limits: data749				.limits750				.map(|limits| Self::clamp_limits(data.mode.clone(), &Default::default(), limits))751				.unwrap_or_else(|| Ok(CollectionLimits::default()))?,752			permissions: data753				.permissions754				.map(|permissions| Self::clamp_permissions(data.mode.clone(), &Default::default(), permissions))755				.unwrap_or_else(|| Ok(CollectionPermissions::default()))?,756		};757758		let mut collection_properties = up_data_structs::CollectionProperties::get();759		collection_properties760			.try_set_from_iter(data.properties.into_iter())761			.map_err(<Error<T>>::from)?;762763		CollectionProperties::<T>::insert(id, collection_properties);764765		let mut token_props_permissions = PropertiesPermissionMap::new();766		token_props_permissions767			.try_set_from_iter(data.token_property_permissions.into_iter())768			.map_err(<Error<T>>::from)?;769770		CollectionPropertyPermissions::<T>::insert(id, token_props_permissions);771772		// Take a (non-refundable) deposit of collection creation773		{774			let mut imbalance =775				<<<T as Config>::Currency as Currency<T::AccountId>>::PositiveImbalance>::zero();776			imbalance.subsume(777				<<T as Config>::Currency as Currency<T::AccountId>>::deposit_creating(778					&T::TreasuryAccountId::get(),779					T::CollectionCreationPrice::get(),780				),781			);782			<T as Config>::Currency::settle(783				&owner,784				imbalance,785				WithdrawReasons::TRANSFER,786				ExistenceRequirement::KeepAlive,787			)788			.map_err(|_| Error::<T>::NotSufficientFounds)?;789		}790791		<CreatedCollectionCount<T>>::put(created_count);792		<Pallet<T>>::deposit_event(Event::CollectionCreated(id, data.mode.id(), owner.clone()));793		<CollectionById<T>>::insert(id, collection);794		Ok(id)795	}796797	pub fn destroy_collection(798		collection: CollectionHandle<T>,799		sender: &T::CrossAccountId,800	) -> DispatchResult {801		ensure!(802			collection.limits.owner_can_destroy(),803			<Error<T>>::NoPermission,804		);805		collection.check_is_owner(sender)?;806807		let destroyed_collections = <DestroyedCollectionCount<T>>::get()808			.0809			.checked_add(1)810			.ok_or(ArithmeticError::Overflow)?;811812		// =========813814		<DestroyedCollectionCount<T>>::put(destroyed_collections);815		<CollectionById<T>>::remove(collection.id);816		<AdminAmount<T>>::remove(collection.id);817		<IsAdmin<T>>::remove_prefix((collection.id,), None);818		<Allowlist<T>>::remove_prefix((collection.id,), None);819		<CollectionProperties<T>>::remove(collection.id);820821		<Pallet<T>>::deposit_event(Event::CollectionDestroyed(collection.id));822		Ok(())823	}824825	pub fn set_collection_property(826		collection: &CollectionHandle<T>,827		sender: &T::CrossAccountId,828		property: Property,829	) -> DispatchResult {830		collection.check_is_owner_or_admin(sender)?;831832		CollectionProperties::<T>::try_mutate(collection.id, |properties| {833			let property = property.clone();834			properties.try_set(property.key, property.value)835		})836		.map_err(<Error<T>>::from)?;837838		Self::deposit_event(Event::CollectionPropertySet(collection.id, property.key));839840		Ok(())841	}842843	pub fn set_scoped_collection_property(844		collection_id: CollectionId,845		scope: PropertyScope,846		property: Property,847	) -> DispatchResult {848		CollectionProperties::<T>::try_mutate(collection_id, |properties| {849			properties.try_scoped_set(scope, property.key, property.value)850		})851		.map_err(<Error<T>>::from)?;852853		Ok(())854	}855856	pub fn set_scoped_collection_properties(857		collection_id: CollectionId,858		scope: PropertyScope,859		properties: impl Iterator<Item = Property>,860	) -> DispatchResult {861		CollectionProperties::<T>::try_mutate(collection_id, |stored_properties| {862			stored_properties.try_scoped_set_from_iter(scope, properties)863		})864		.map_err(<Error<T>>::from)?;865866		Ok(())867	}868869	#[transactional]870	pub fn set_collection_properties(871		collection: &CollectionHandle<T>,872		sender: &T::CrossAccountId,873		properties: Vec<Property>,874	) -> DispatchResult {875		for property in properties {876			Self::set_collection_property(collection, sender, property)?;877		}878879		Ok(())880	}881882	pub fn delete_collection_property(883		collection: &CollectionHandle<T>,884		sender: &T::CrossAccountId,885		property_key: PropertyKey,886	) -> DispatchResult {887		collection.check_is_owner_or_admin(sender)?;888889		CollectionProperties::<T>::try_mutate(collection.id, |properties| {890			properties.remove(&property_key)891		})892		.map_err(<Error<T>>::from)?;893894		Self::deposit_event(Event::CollectionPropertyDeleted(895			collection.id,896			property_key,897		));898899		Ok(())900	}901902	#[transactional]903	pub fn delete_collection_properties(904		collection: &CollectionHandle<T>,905		sender: &T::CrossAccountId,906		property_keys: Vec<PropertyKey>,907	) -> DispatchResult {908		for key in property_keys {909			Self::delete_collection_property(collection, sender, key)?;910		}911912		Ok(())913	}914915	// For migrations916	pub fn set_property_permission_unchecked(917		collection: CollectionId,918		property_permission: PropertyKeyPermission,919	) -> DispatchResult {920		<CollectionPropertyPermissions<T>>::try_mutate(collection, |permissions| {921			permissions.try_set(property_permission.key, property_permission.permission)922		})923		.map_err(<Error<T>>::from)?;924		Ok(())925	}926927	pub fn set_property_permission(928		collection: &CollectionHandle<T>,929		sender: &T::CrossAccountId,930		property_permission: PropertyKeyPermission,931	) -> DispatchResult {932		collection.check_is_owner_or_admin(sender)?;933934		let all_permissions = CollectionPropertyPermissions::<T>::get(collection.id);935		let current_permission = all_permissions.get(&property_permission.key);936		if matches![937			current_permission,938			Some(PropertyPermission { mutable: false, .. })939		] {940			return Err(<Error<T>>::NoPermission.into());941		}942943		CollectionPropertyPermissions::<T>::try_mutate(collection.id, |permissions| {944			let property_permission = property_permission.clone();945			permissions.try_set(property_permission.key, property_permission.permission)946		})947		.map_err(<Error<T>>::from)?;948949		Self::deposit_event(Event::PropertyPermissionSet(950			collection.id,951			property_permission.key,952		));953954		Ok(())955	}956957	#[transactional]958	pub fn set_property_permissions(959		collection: &CollectionHandle<T>,960		sender: &T::CrossAccountId,961		property_permissions: Vec<PropertyKeyPermission>,962	) -> DispatchResult {963		for prop_pemission in property_permissions {964			Self::set_property_permission(collection, sender, prop_pemission)?;965		}966967		Ok(())968	}969970	pub fn get_collection_property(971		collection_id: CollectionId,972		key: &PropertyKey,973	) -> Option<PropertyValue> {974		Self::collection_properties(collection_id).get(key).cloned()975	}976977	pub fn bytes_keys_to_property_keys(978		keys: Vec<Vec<u8>>,979	) -> Result<Vec<PropertyKey>, DispatchError> {980		keys.into_iter()981			.map(|key| -> Result<PropertyKey, DispatchError> {982				key.try_into()983					.map_err(|_| <Error<T>>::PropertyKeyIsTooLong.into())984			})985			.collect::<Result<Vec<PropertyKey>, DispatchError>>()986	}987988	pub fn filter_collection_properties(989		collection_id: CollectionId,990		keys: Option<Vec<PropertyKey>>,991	) -> Result<Vec<Property>, DispatchError> {992		let properties = Self::collection_properties(collection_id);993994		let properties = keys995			.map(|keys| {996				keys.into_iter()997					.filter_map(|key| {998						properties.get(&key).map(|value| Property {999							key,1000							value: value.clone(),1001						})1002					})1003					.collect()1004			})1005			.unwrap_or_else(|| {1006				properties1007					.into_iter()1008					.map(|(key, value)| Property {1009						key,1010						value,1011					})1012					.collect()1013			});10141015		Ok(properties)1016	}10171018	pub fn filter_property_permissions(1019		collection_id: CollectionId,1020		keys: Option<Vec<PropertyKey>>,1021	) -> Result<Vec<PropertyKeyPermission>, DispatchError> {1022		let permissions = Self::property_permissions(collection_id);10231024		let key_permissions = keys1025			.map(|keys| {1026				keys.into_iter()1027					.filter_map(|key| {1028						permissions1029							.get(&key)1030							.map(|permission| PropertyKeyPermission {1031								key,1032								permission: permission.clone(),1033							})1034					})1035					.collect()1036			})1037			.unwrap_or_else(|| {1038				permissions1039					.into_iter()1040					.map(|(key, permission)| PropertyKeyPermission {1041						key,1042						permission,1043					})1044					.collect()1045			});10461047		Ok(key_permissions)1048	}10491050	pub fn toggle_allowlist(1051		collection: &CollectionHandle<T>,1052		sender: &T::CrossAccountId,1053		user: &T::CrossAccountId,1054		allowed: bool,1055	) -> DispatchResult {1056		collection.check_is_owner_or_admin(sender)?;10571058		// =========10591060		if allowed {1061			<Allowlist<T>>::insert((collection.id, user), true);1062		} else {1063			<Allowlist<T>>::remove((collection.id, user));1064		}10651066		Ok(())1067	}10681069	pub fn toggle_admin(1070		collection: &CollectionHandle<T>,1071		sender: &T::CrossAccountId,1072		user: &T::CrossAccountId,1073		admin: bool,1074	) -> DispatchResult {1075		collection.check_is_owner_or_admin(sender)?;10761077		let was_admin = <IsAdmin<T>>::get((collection.id, user));1078		if was_admin == admin {1079			return Ok(());1080		}1081		let amount = <AdminAmount<T>>::get(collection.id);10821083		if admin {1084			let amount = amount1085				.checked_add(1)1086				.ok_or(<Error<T>>::CollectionAdminCountExceeded)?;1087			ensure!(1088				amount <= Self::collection_admins_limit(),1089				<Error<T>>::CollectionAdminCountExceeded,1090			);10911092			// =========10931094			<AdminAmount<T>>::insert(collection.id, amount);1095			<IsAdmin<T>>::insert((collection.id, user), true);1096		} else {1097			<AdminAmount<T>>::insert(collection.id, amount.saturating_sub(1));1098			<IsAdmin<T>>::remove((collection.id, user));1099		}11001101		Ok(())1102	}11031104	pub fn clamp_limits(1105		mode: CollectionMode,1106		old_limit: &CollectionLimits,1107		mut new_limit: CollectionLimits,1108	) -> Result<CollectionLimits, DispatchError> {1109		limit_default!(old_limit, new_limit,1110			account_token_ownership_limit => ensure!(1111				new_limit <= MAX_TOKEN_OWNERSHIP,1112				<Error<T>>::CollectionLimitBoundsExceeded,1113			),1114			sponsored_data_size => ensure!(1115				new_limit <= CUSTOM_DATA_LIMIT,1116				<Error<T>>::CollectionLimitBoundsExceeded,1117			),11181119			sponsored_data_rate_limit => {},1120			token_limit => ensure!(1121				old_limit >= new_limit && new_limit > 0,1122				<Error<T>>::CollectionTokenLimitExceeded1123			),11241125			sponsor_transfer_timeout(match mode {1126				CollectionMode::NFT => NFT_SPONSOR_TRANSFER_TIMEOUT,1127				CollectionMode::Fungible(_) => FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,1128				CollectionMode::ReFungible => REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,1129			}) => ensure!(1130				new_limit <= MAX_SPONSOR_TIMEOUT,1131				<Error<T>>::CollectionLimitBoundsExceeded,1132			),1133			sponsor_approve_timeout => {},1134			owner_can_transfer => ensure!(1135				old_limit || !new_limit,1136				<Error<T>>::OwnerPermissionsCantBeReverted,1137			),1138			owner_can_destroy => ensure!(1139				old_limit || !new_limit,1140				<Error<T>>::OwnerPermissionsCantBeReverted,1141			),1142			transfers_enabled => {},1143		);1144		Ok(new_limit)1145	}1146	pub fn clamp_permissions(1147		mode: CollectionMode,1148		old_limit: &CollectionPermissions,1149		mut new_limit: CollectionPermissions,1150	) -> Result<CollectionPermissions, DispatchError> {1151		limit_default_clone!(old_limit, new_limit,1152			access => {},1153			mint_mode => {},1154			nesting => {},1155		);1156		Ok(new_limit)1157	}1158}11591160#[macro_export]1161macro_rules! unsupported {1162	() => {1163		Err(<Error<T>>::UnsupportedOperation.into())1164	};1165}11661167/// Worst cases1168pub trait CommonWeightInfo<CrossAccountId> {1169	fn create_item() -> Weight;1170	fn create_multiple_items(amount: &[CreateItemData]) -> Weight;1171	fn create_multiple_items_ex(cost: &CreateItemExData<CrossAccountId>) -> Weight;1172	fn burn_item() -> Weight;1173	fn set_collection_properties(amount: u32) -> Weight;1174	fn delete_collection_properties(amount: u32) -> Weight;1175	fn set_token_properties(amount: u32) -> Weight;1176	fn delete_token_properties(amount: u32) -> Weight;1177	fn set_property_permissions(amount: u32) -> Weight;1178	fn transfer() -> Weight;1179	fn approve() -> Weight;1180	fn transfer_from() -> Weight;1181	fn burn_from() -> Weight;1182}11831184pub trait CommonCollectionOperations<T: Config> {1185	fn create_item(1186		&self,1187		sender: T::CrossAccountId,1188		to: T::CrossAccountId,1189		data: CreateItemData,1190		nesting_budget: &dyn Budget,1191	) -> DispatchResultWithPostInfo;1192	fn create_multiple_items(1193		&self,1194		sender: T::CrossAccountId,1195		to: T::CrossAccountId,1196		data: Vec<CreateItemData>,1197		nesting_budget: &dyn Budget,1198	) -> DispatchResultWithPostInfo;1199	fn create_multiple_items_ex(1200		&self,1201		sender: T::CrossAccountId,1202		data: CreateItemExData<T::CrossAccountId>,1203		nesting_budget: &dyn Budget,1204	) -> DispatchResultWithPostInfo;1205	fn burn_item(1206		&self,1207		sender: T::CrossAccountId,1208		token: TokenId,1209		amount: u128,1210	) -> DispatchResultWithPostInfo;1211	fn set_collection_properties(1212		&self,1213		sender: T::CrossAccountId,1214		properties: Vec<Property>,1215	) -> DispatchResultWithPostInfo;1216	fn delete_collection_properties(1217		&self,1218		sender: &T::CrossAccountId,1219		property_keys: Vec<PropertyKey>,1220	) -> DispatchResultWithPostInfo;1221	fn set_token_properties(1222		&self,1223		sender: T::CrossAccountId,1224		token_id: TokenId,1225		property: Vec<Property>,1226	) -> DispatchResultWithPostInfo;1227	fn delete_token_properties(1228		&self,1229		sender: T::CrossAccountId,1230		token_id: TokenId,1231		property_keys: Vec<PropertyKey>,1232	) -> DispatchResultWithPostInfo;1233	fn set_property_permissions(1234		&self,1235		sender: &T::CrossAccountId,1236		property_permissions: Vec<PropertyKeyPermission>,1237	) -> DispatchResultWithPostInfo;1238	fn transfer(1239		&self,1240		sender: T::CrossAccountId,1241		to: T::CrossAccountId,1242		token: TokenId,1243		amount: u128,1244		nesting_budget: &dyn Budget,1245	) -> DispatchResultWithPostInfo;1246	fn approve(1247		&self,1248		sender: T::CrossAccountId,1249		spender: T::CrossAccountId,1250		token: TokenId,1251		amount: u128,1252	) -> DispatchResultWithPostInfo;1253	fn transfer_from(1254		&self,1255		sender: T::CrossAccountId,1256		from: T::CrossAccountId,1257		to: T::CrossAccountId,1258		token: TokenId,1259		amount: u128,1260		nesting_budget: &dyn Budget,1261	) -> DispatchResultWithPostInfo;1262	fn burn_from(1263		&self,1264		sender: T::CrossAccountId,1265		from: T::CrossAccountId,1266		token: TokenId,1267		amount: u128,1268		nesting_budget: &dyn Budget,1269	) -> DispatchResultWithPostInfo;12701271	fn check_nesting(1272		&self,1273		sender: T::CrossAccountId,1274		from: (CollectionId, TokenId),1275		under: TokenId,1276		budget: &dyn Budget,1277	) -> DispatchResult;12781279	fn nest(1280		&self,1281		under: TokenId,1282		to_nest: (CollectionId, TokenId)1283	);12841285	fn unnest(1286		&self,1287		under: TokenId,1288		to_nest: (CollectionId, TokenId)1289	);12901291	fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId>;1292	fn collection_tokens(&self) -> Vec<TokenId>;1293	fn token_exists(&self, token: TokenId) -> bool;1294	fn last_token_id(&self) -> TokenId;12951296	fn token_owner(&self, token: TokenId) -> Option<T::CrossAccountId>;1297	fn token_property(&self, token_id: TokenId, key: &PropertyKey) -> Option<PropertyValue>;1298	fn token_properties(&self, token_id: TokenId, keys: Option<Vec<PropertyKey>>) -> Vec<Property>;1299	/// Amount of unique collection tokens1300	fn total_supply(&self) -> u32;1301	/// Amount of different tokens account has (Applicable to nonfungible/refungible)1302	fn account_balance(&self, account: T::CrossAccountId) -> u32;1303	/// Amount of specific token account have (Applicable to fungible/refungible)1304	fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128;1305	fn allowance(1306		&self,1307		sender: T::CrossAccountId,1308		spender: T::CrossAccountId,1309		token: TokenId,1310	) -> u128;1311}13121313// Flexible enough for implementing CommonCollectionOperations1314pub fn with_weight(res: DispatchResult, weight: Weight) -> DispatchResultWithPostInfo {1315	let post_info = PostDispatchInfo {1316		actual_weight: Some(weight),1317		pays_fee: Pays::Yes,1318	};1319	match res {1320		Ok(()) => Ok(post_info),1321		Err(error) => Err(DispatchErrorWithPostInfo { post_info, error }),1322	}1323}13241325impl<T: Config> From<PropertiesError> for Error<T> {1326	fn from(error: PropertiesError) -> Self {1327		match error {1328			PropertiesError::NoSpaceForProperty => Self::NoSpaceForProperty,1329			PropertiesError::PropertyLimitReached => Self::PropertyLimitReached,1330			PropertiesError::InvalidCharacterInPropertyKey => Self::InvalidCharacterInPropertyKey,1331			PropertiesError::PropertyKeyIsTooLong => Self::PropertyKeyIsTooLong,1332			PropertiesError::EmptyPropertyKey => Self::EmptyPropertyKey,1333		}1334	}1335}