git.delta.rocks / unique-network / refs/commits / 24f9c77907a0

difftreelog

Breaking: migration to `Fungible` traits in app-promotion pallet

PraetorP2023-05-19parent: #03bc12d.patch.diff
in: master
Migration from `Currency` traits to `Fungible` ones in constraints for `Config::Currency` associated type.

6 files changed

modifiedCargo.lockdiffbeforeafterboth
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -6125,7 +6125,7 @@
 
 [[package]]
 name = "pallet-app-promotion"
-version = "0.1.6"
+version = "0.2.0"
 dependencies = [
  "frame-benchmarking",
  "frame-support",
modifiedpallets/app-promotion/CHANGELOG.mddiffbeforeafterboth
--- a/pallets/app-promotion/CHANGELOG.md
+++ b/pallets/app-promotion/CHANGELOG.md
@@ -3,6 +3,13 @@
 All notable changes to this project will be documented in this file.
 
 <!-- bureaucrate goes here -->
+## [0.2.0] - 2023-05-19
+
+### Changed
+
+- **Breaking:** migration from `Currency` traits to `Fungible` ones in constraints
+  for `Config::Currency` associated type.
+
 ## [0.1.6] - 2023-04-19
 
 - ### Fixed
modifiedpallets/app-promotion/Cargo.tomldiffbeforeafterboth
--- a/pallets/app-promotion/Cargo.toml
+++ b/pallets/app-promotion/Cargo.toml
@@ -9,7 +9,7 @@
 license = 'GPLv3'
 name = 'pallet-app-promotion'
 repository = 'https://github.com/UniqueNetwork/unique-chain'
-version = '0.1.6'
+version = '0.2.0'
 
 [package.metadata.docs.rs]
 targets = ['x86_64-unknown-linux-gnu']
modifiedpallets/app-promotion/src/lib.rsdiffbeforeafterboth
--- a/pallets/app-promotion/src/lib.rs
+++ b/pallets/app-promotion/src/lib.rs
@@ -70,7 +70,9 @@
 use frame_support::{
 	dispatch::{DispatchResult},
 	traits::{
-		Currency, Get, LockableCurrency, WithdrawReasons, tokens::Balance, ExistenceRequirement,
+		Get, LockableCurrency,
+		tokens::Balance,
+		fungible::{Inspect, InspectFreeze, Mutate, MutateFreeze},
 	},
 	ensure, BoundedVec,
 };
@@ -90,24 +92,25 @@
 const PENDING_LIMIT_PER_BLOCK: u32 = 3;
 
 type BalanceOf<T> =
-	<<T as Config>::Currency as Currency<<T as frame_system::Config>::AccountId>>::Balance;
+	<<T as Config>::Currency as Inspect<<T as frame_system::Config>::AccountId>>::Balance;
 
 #[frame_support::pallet]
 pub mod pallet {
 	use super::*;
 	use frame_support::{
-		Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key, PalletId,
-		traits::ReservableCurrency, weights::Weight,
+		Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key, PalletId, weights::Weight,
 	};
 	use frame_system::pallet_prelude::*;
+use sp_runtime::DispatchError;
 
 	#[pallet::config]
 	pub trait Config:
 		frame_system::Config + pallet_evm::Config + pallet_configuration::Config
 	{
 		/// Type to interact with the native token
-		type Currency: ExtendedLockableCurrency<Self::AccountId>
-			+ ReservableCurrency<Self::AccountId>;
+		type Currency: MutateFreeze<Self::AccountId>
+			+ Mutate<Self::AccountId> 
+			+ ExtendedLockableCurrency<Self::AccountId, Balance = <<Self as Config>::Currency as Inspect<Self::AccountId>>::Balance>;
 
 		/// Type for interacting with collections
 		type CollectionHandler: CollectionHandler<
@@ -125,6 +128,10 @@
 		#[pallet::constant]
 		type PalletId: Get<PalletId>;
 
+		/// Freeze identifier used by the pallet
+		#[pallet::constant]
+		type FreezeIdentifier: Get<<<Self as Config>::Currency as InspectFreeze<Self::AccountId>>::Id>;
+
 		/// In relay blocks.
 		#[pallet::constant]
 		type RecalculationInterval: Get<Self::BlockNumber>;
@@ -205,10 +212,12 @@
 		PendingForBlockOverflow,
 		/// The error is due to the fact that the collection/contract must already be sponsored in order to perform the action.
 		SponsorNotSet,
-		/// Errors caused by incorrect actions with a locked balance.
+		/// 
 		IncorrectLockedBalanceOperation,
 		/// Errors caused by insufficient staked balance.
 		InsufficientStakedBalance,
+		/// Errors caused by incorrect state of a staker in context of the pallet.
+		InconsistencyState
 	}
 
 	/// Stores the total staked amount.
@@ -263,6 +272,10 @@
 	pub type PreviousCalculatedRecord<T: Config> =
 		StorageValue<Value = (T::AccountId, T::BlockNumber), QueryKind = OptionQuery>;
 
+	// #[pallet::storage]
+	// pub(crate) type UpgradedToFreezes<T: Config> =
+	// 	StorageValue<Value = bool, QueryKind = ValueQuery>;
+
 	#[pallet::hooks]
 	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {
 		/// Block overflow is impossible due to the fact that the unstake algorithm in on_initialize
@@ -277,22 +290,149 @@
 
 			if !block_pending.is_empty() {
 				block_pending.into_iter().for_each(|(staker, amount)| {
-					Self::get_locked_balance(&staker).map(|b| {
-						let new_state = b.amount.checked_sub(&amount).unwrap_or_default();
-						Self::set_lock_unchecked(&staker, new_state);
+					Self::get_freezed_balance(&staker).map(|b| {
+						let new_state = b.checked_sub(&amount).unwrap_or_default();
+						Self::set_freeze_unchecked(&staker, new_state);
 					});
 				});
 			}
 
 			<T as Config>::WeightInfo::on_initialize(counter)
 		}
+
+		// fn on_runtime_upgrade() -> Weight {
+		// 	use scale_info::prelude::collections::HashSet;
+		// 	let mut consumed_weight = Weight::zero();
+		// 	let mut add_weight = |reads, writes, weight| {
+		// 		consumed_weight += T::DbWeight::get().reads_writes(reads, writes);
+		// 		consumed_weight += weight;
+		// 	};
+
+		// 	let mut stakes_unstakes = vec![];
+
+		// 	if <UpgradedToFreezes<T>>::get() {
+		// 		add_weight(1, 0, Weight::zero());
+		// 		return consumed_weight;
+		// 	} else {
+		// 		add_weight(1, 1, Weight::zero());
+		// 		<UpgradedToFreezes<T>>::set(true);
+		// 	}
+		// 	<Staked<T>>::iter_keys().for_each(|(staker_id, _)| {
+		// 		add_weight(1, 0, Weight::zero());
+		// 		stakes_unstakes.push(staker_id);
+		// 	});
+
+		// 	<PendingUnstake<T>>::iter().for_each(|(_, v)| {
+		// 		add_weight(1, 0, Weight::zero());
+		// 		v.into_iter().for_each(|(staker, _)| {
+		// 			stakes_unstakes.push(staker);
+		// 		});
+		// 	});
+
+		// 	// filter duplicated id.
+		// 	stakes_unstakes = stakes_unstakes
+		// 		.into_iter()
+		// 		.map(|key| key)
+		// 		.collect::<HashSet<_>>()
+		// 		.into_iter()
+		// 		.collect();
+
+		// 	stakes_unstakes
+		// 		.map(|a| (a, <Pallet<T>>::get_locked_balance(&a).amount))
+		// 		.for_each(|(staker, amount)| {
+		// 			<<T as Config>::Currency as LockableCurrency<T::AccountId>>::remove_lock(
+		// 				LOCK_IDENTIFIER,
+		// 				&staker,
+		// 			);
+		// 			<<T as Config>::Currency as MutateFreeze<T::AccountId>>::set_freeze(
+		// 				&<T as Config>::FreezeIdentifier::get(),
+		// 				&staker,
+		// 				amount,
+		// 			);
+		// 			add_weight(1, 2, Weight::zero())
+		// 		});
+
+		// 	consumed_weight
+		// }
+
+		// #[cfg(feature = "try-runtime")]
+		// fn pre_upgrade() -> Result<Vec<u8>, &'static str> {
+		// 	use sp_std::collections::btree_map::BTreeMap;
+		// 	if <UpgradedToFreezes<T>>::get() {
+		// 		return Ok(Default::default());
+		// 	}
+		// 	// Staker -> (total (stakes and unstakes) locked by promotion);
+		// 	let mut pre_state: BTreeMap<T::AccountId, BalanceOf<T>> = BTreeMap::new();
+
+		// 	<Staked<T>>::iter().for_each(|((staker, _), (amount, _))| {
+		// 		if let Some(locked_balance) = pre_state.get_mut(&staker) {
+		// 			*locked_balance += amount;
+		// 		} else {
+		// 			pre_state.insert(staker, amount);
+		// 		}
+		// 	});
+
+		// 	<PendingUnstake<T>>::iter().for_each(|(_, v)| {
+		// 		v.into_iter().for_each(|(staker, amount)| {
+		// 			if let Some(locked_balance) = pre_state.get_mut(&staker) {
+		// 				*locked_balance += amount;
+		// 			} else {
+		// 				pre_state.insert(staker, amount);
+		// 			}
+		// 		})
+		// 	});
+
+		// 	Ok(pre_state.encode())
+		// }
+
+		// #[cfg(feature = "try-runtime")]
+		// fn post_upgrade(pre_state: Vec<u8>) -> Result<(), &'static str> {
+		// 	use sp_std::collections::btree_map::BTreeMap;
+
+		// 	if <UpgradedToFreezes<T>>::get() {
+		// 		return Ok(());
+		// 	}
+
+		// 	let mut is_ok = true;
+
+		// 	let pre_state: BTreeMap<T::AccountId, BalanceOf<T>> =
+		// 		Decode::decode(&mut &pre_state[..]).map_err(|_| "failed to decode pre_state")?;
+		// 	for (staker, freezed_by_promo) in pre_state.into_iter() {
+		// 		let storage_freeze_state = <<T as Config>::Currency as InspectFreeze<
+		// 			T::AccountId,
+		// 		>>::balance_frozen(
+		// 			&<T as Config>::FreezeIdentifier::get(), staker
+		// 		);
+		// 		if storage_freeze_state != freezed_by_promo {
+		// 			is_ok = false;
+		// 			log::error!(
+		// 						"Incorrect freezed balance for {:?}. New balance: {:?}. Before runtime upgrade: locked by promo - {:?}",
+		// 						staker, storage_freeze_state, freezed_by_promo
+		// 					);
+		// 		}
+
+		// 		if !<Pallet<T>>::get_locked_balance(&staker).amount.is_zero() {
+		// 			is_ok = false;
+		// 			log::error!(
+		// 				"Incorrect(non-zero) locked by app promo balance for {:?}",
+		// 				staker
+		// 			);
+		// 		}
+		// 	}
+
+		// 	if is_ok {
+		// 		Ok(())
+		// 	} else {
+		// 		Err("Incorrect balance for some of stakers... See logs")
+		// 	}
+		// }
 	}
 
 	#[pallet::call]
 	impl<T: Config> Pallet<T>
 	where
 		T::BlockNumber: From<u32> + Into<u32>,
-		<<T as Config>::Currency as Currency<T::AccountId>>::Balance: Sum + From<u128>,
+		<<T as Config>::Currency as Inspect<T::AccountId>>::Balance: Sum + From<u128>,
 	{
 		/// Sets an address as the the admin.
 		///
@@ -338,22 +478,21 @@
 			);
 			let config = <PalletConfiguration<T>>::get();
 
-			let balance =
-				<<T as Config>::Currency as Currency<T::AccountId>>::free_balance(&staker_id);
+			let balance = <<T as Config>::Currency as Inspect<T::AccountId>>::balance(&staker_id);
 
-			// checks that we can lock `amount` on the `staker` account.
+			// checks that we can freeze `amount` on the `staker` account.
 			ensure!(
 				amount
-					<= match Self::get_locked_balance(&staker_id) {
-						Some(lock) => balance
-							.checked_sub(&lock.amount)
+					<= match Self::get_freezed_balance(&staker_id) {
+						Some(freezed_by_pallet) => balance
+							.checked_sub(&freezed_by_pallet)
 							.ok_or(ArithmeticError::Underflow)?,
 						None => balance,
 					},
 				ArithmeticError::Underflow
 			);
 
-			Self::add_lock_balance(&staker_id, amount)?;
+			Self::add_freeze_balance(&staker_id, amount)?;
 
 			let block_number = T::RelayBlockNumberProvider::current_block_number();
 
@@ -599,14 +738,14 @@
 				let flush_stake = || -> DispatchResult {
 					if let Some(last_id) = &*last_id.borrow() {
 						if !income_acc.borrow().is_zero() {
-							<<T as Config>::Currency as Currency<T::AccountId>>::transfer(
+							<<T as Config>::Currency as Mutate<T::AccountId>>::transfer(
 								&T::TreasuryAccountId::get(),
 								last_id,
 								*income_acc.borrow(),
-								ExistenceRequirement::KeepAlive,
+								frame_support::traits::tokens::Preservation::Protect,
 							)?;
 
-							Self::add_lock_balance(last_id, *income_acc.borrow())?;
+							Self::add_freeze_balance(last_id, *income_acc.borrow())?;
 							<TotalStaked<T>>::try_mutate(|staked| -> DispatchResult {
 								*staked = staked
 									.checked_add(&*income_acc.borrow())
@@ -681,6 +820,42 @@
 
 			Ok(())
 		}
+
+		///  Migrates lock state into freeze one
+		///
+		///  # Arguments
+		///
+		/// * `origin`: Must be `Signed`.
+		/// * `stakers`: Accounts to be upgraded.
+		#[pallet::call_index(9)]
+		#[pallet::weight(T::DbWeight::get().reads_writes(2, 2) * stakers.len() as u64)]
+		pub fn upgrade_accounts(
+			origin: OriginFor<T>,
+			stakers: Vec<T::AccountId>,
+		) -> DispatchResult {
+			ensure_signed(origin)?;
+
+			stakers.into_iter().try_for_each(|s| -> Result<_, DispatchError> {
+				if let Some(lock) = Self::get_locked_balance(&s) {
+					
+					if let Some(_) = Self::get_freezed_balance(&s) {
+						return Err(Error::<T>::InconsistencyState.into())
+					}
+					
+					<<T as Config>::Currency as LockableCurrency<T::AccountId>>::remove_lock(
+						LOCK_IDENTIFIER,
+						&s,
+					);
+
+					Self::set_freeze_unchecked(&s, lock.amount);
+					Ok(())
+				} else {
+					Ok(())
+				}
+			})?;
+			
+			Ok(())
+		}
 	}
 }
 
@@ -791,11 +966,23 @@
 	///
 	/// - `staker`: staker account.
 	/// - `amount`: amount of added locked funds.
-	fn add_lock_balance(staker: &T::AccountId, amount: BalanceOf<T>) -> DispatchResult {
-		Self::get_locked_balance(staker)
-			.map_or(<BalanceOf<T>>::default(), |l| l.amount)
+	// fn add_lock_balance(staker: &T::AccountId, amount: BalanceOf<T>) -> DispatchResult {
+	// 	Self::get_locked_balance(staker)
+	// 		.map_or(<BalanceOf<T>>::default(), |l| l.amount)
+	// 		.checked_add(&amount)
+	// 		.map(|new_lock| Self::set_lock_unchecked(staker, new_lock))
+	// 		.ok_or(ArithmeticError::Overflow.into())
+	// }
+
+	/// Adds the balance to freezed by the pallet.
+	///
+	/// - `staker`: staker account.
+	/// - `amount`: amount of added freezed funds.
+	fn add_freeze_balance(staker: &T::AccountId, amount: BalanceOf<T>) -> DispatchResult {
+		Self::get_freezed_balance(staker)
+			.unwrap_or_default()
 			.checked_add(&amount)
-			.map(|new_lock| Self::set_lock_unchecked(staker, new_lock))
+			.map(|freeze| Self::set_freeze_unchecked(staker, freeze))
 			.ok_or(ArithmeticError::Overflow.into())
 	}
 
@@ -803,19 +990,38 @@
 	///
 	/// - `staker`: staker account.
 	/// - `amount`: amount of locked funds.
-	fn set_lock_unchecked(staker: &T::AccountId, amount: BalanceOf<T>) {
+	// fn set_lock_unchecked(staker: &T::AccountId, amount: BalanceOf<T>) {
+	// 	if amount.is_zero() {
+	// 		<<T as Config>::Currency as LockableCurrency<T::AccountId>>::remove_lock(
+	// 			LOCK_IDENTIFIER,
+	// 			&staker,
+	// 		);
+	// 	} else {
+	// 		<<T as Config>::Currency as LockableCurrency<T::AccountId>>::set_lock(
+	// 			LOCK_IDENTIFIER,
+	// 			staker,
+	// 			amount,
+	// 			WithdrawReasons::all(),
+	// 		)
+	// 	}
+	// }
+
+	/// Sets the new state of a balance freezed by the pallet.
+	///
+	/// - `staker`: staker account.
+	/// - `amount`: amount of freezed funds.
+	fn set_freeze_unchecked(staker: &T::AccountId, amount: BalanceOf<T>) {
 		if amount.is_zero() {
-			<<T as Config>::Currency as LockableCurrency<T::AccountId>>::remove_lock(
-				LOCK_IDENTIFIER,
+			<<T as Config>::Currency as MutateFreeze<T::AccountId>>::thaw(
+				&T::FreezeIdentifier::get(),
 				&staker,
 			);
 		} else {
-			<<T as Config>::Currency as LockableCurrency<T::AccountId>>::set_lock(
-				LOCK_IDENTIFIER,
+			<<T as Config>::Currency as MutateFreeze<T::AccountId>>::set_freeze(
+				&T::FreezeIdentifier::get(),
 				staker,
 				amount,
-				WithdrawReasons::all(),
-			)
+			);
 		}
 	}
 
@@ -830,6 +1036,22 @@
 			.find(|l| l.id == LOCK_IDENTIFIER)
 	}
 
+	/// Returns the balance freezed by the pallet for the staker.
+	///
+	/// - `staker`: staker account.
+	pub fn get_freezed_balance(staker: &T::AccountId) -> Option<BalanceOf<T>> {
+		let res = <<T as Config>::Currency as InspectFreeze<T::AccountId>>::balance_frozen(
+			&T::FreezeIdentifier::get(),
+			staker,
+		);
+
+		if res == Zero::zero() {
+			None
+		} else {
+			Some(res)
+		}
+	}
+
 	/// Returns the total staked balance for the staker.
 	///
 	/// - `staker`: staker account.
@@ -926,7 +1148,7 @@
 
 impl<T: Config> Pallet<T>
 where
-	<<T as Config>::Currency as Currency<T::AccountId>>::Balance: Sum,
+	<<T as Config>::Currency as Inspect<T::AccountId>>::Balance: Sum,
 {
 	/// Returns the amount reserved by the pending.
 	/// If `staker` is `None`, returns the total pending.
@@ -987,13 +1209,8 @@
 		// checks that we can do unstake in the block
 		ensure!(!pendings.is_full(), Error::<T>::PendingForBlockOverflow);
 
-		let mut total_stakes = 0u64;
-
 		let total_staked: BalanceOf<T> = Staked::<T>::drain_prefix((&staker_id,))
-			.map(|(_, (amount, _))| {
-				total_stakes += 1;
-				amount
-			})
+			.map(|(_, (amount, _))| amount)
 			.sum();
 
 		if total_staked.is_zero() {
modifiedpallets/common/src/lib.rsdiffbeforeafterboth
--- a/pallets/common/src/lib.rs
+++ b/pallets/common/src/lib.rs
@@ -420,7 +420,6 @@
 
 #[frame_support::pallet]
 pub mod pallet {
-	use core::marker::PhantomData;
 
 	use super::*;
 	use dispatch::CollectionDispatch;
modifiedpallets/nonfungible/src/lib.rsdiffbeforeafterboth
before · pallets/nonfungible/src/lib.rs
1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617//! # Nonfungible Pallet18//!19//! The Nonfungible pallet provides functionality for handling nonfungible collections and tokens.20//!21//! - [`Config`]22//! - [`NonfungibleHandle`]23//! - [`Pallet`]24//! - [`CommonWeights`](common::CommonWeights)25//!26//! ## Overview27//!28//! The Nonfungible pallet provides functions for:29//!30//! - NFT collection creation and removal31//! - Minting and burning of NFT tokens32//! - Retrieving account balances33//! - Transfering NFT tokens34//! - Setting and checking allowance for NFT tokens35//! - Setting properties and permissions for NFT collections and tokens36//! - Nesting and unnesting tokens37//!38//! ### Terminology39//!40//! - **NFT token:** Non fungible token.41//!42//! - **NFT Collection:** A collection of NFT tokens. All NFT tokens are part of a collection.43//!   Each collection can define it's own properties, properties for it's tokens and set of permissions.44//!45//! - **Balance:** Number of NFT tokens owned by an account46//!47//! - **Allowance:** NFT tokens owned by one account that another account is allowed to make operations on48//!49//! - **Burning:** The process of “deleting” a token from a collection and from50//!   an account balance of the owner.51//!52//! - **Nesting:** Setting up parent-child relationship between tokens. Nested tokens are inhereting53//!   owner from their parent. There could be multiple levels of nesting. Token couldn't be nested in54//!   it's child token i.e. parent-child relationship graph shouldn't have cycles.55//!56//! - **Properties:** Key-Values pairs. Token properties are attached to a token. Collection properties are57//!   attached to a collection. Set of permissions could be defined for each property.58//!59//! ### Implementations60//!61//! The Nonfungible pallet provides implementations for the following traits. If these traits provide62//! the functionality that you need, then you can avoid coupling with the Nonfungible pallet.63//!64//! - [`CommonWeightInfo`](pallet_common::CommonWeightInfo): Functions for retrieval of transaction weight65//! - [`CommonCollectionOperations`](pallet_common::CommonCollectionOperations): Functions for dealing66//!   with collections67//!68//! ## Interface69//!70//! ### Dispatchable Functions71//!72//! - `init_collection` - Create NFT collection. NFT collection can be configured to allow or deny access for73//!   some accounts.74//! - `destroy_collection` - Destroy exising NFT collection. There should be no tokens in the collection.75//! - `burn` - Burn NFT token owned by account.76//! - `transfer` - Transfer NFT token. Transfers should be enabled for NFT collection.77//!   Nests the NFT token if it is sent to another token.78//! - `create_item` - Mint NFT token in collection. Sender should have permission to mint tokens.79//! - `set_allowance` - Set allowance for another account.80//! - `set_token_property` - Set token property value.81//! - `delete_token_property` - Remove property from the token.82//! - `set_collection_properties` - Set collection properties.83//! - `delete_collection_properties` - Remove properties from the collection.84//! - `set_property_permission` - Set collection property permission.85//! - `set_token_property_permissions` - Set token property permissions.86//!87//! ## Assumptions88//!89//! * To perform operations on tokens sender should be in collection's allow list if collection access mode is `AllowList`.9091#![cfg_attr(not(feature = "std"), no_std)]9293use erc::ERC721Events;94use evm_coder::ToLog;95use frame_support::{96	BoundedVec, ensure, fail, transactional,97	storage::with_transaction,98	pallet_prelude::DispatchResultWithPostInfo,99	pallet_prelude::Weight,100	dispatch::{PostDispatchInfo, Pays},101};102use up_data_structs::{103	AccessMode, CollectionId, CollectionFlags, CustomDataLimit, TokenId, CreateCollectionData,104	CreateNftExData, mapping::TokenAddressMapping, budget::Budget, Property, PropertyKey,105	PropertyValue, PropertyKeyPermission, PropertyScope, TrySetProperty, TokenChild,106	AuxPropertyValue, PropertiesPermissionMap, TokenProperties as TokenPropertiesT,107};108use pallet_evm::{account::CrossAccountId, Pallet as PalletEvm};109use pallet_common::{110	Error as CommonError, Pallet as PalletCommon, Event as CommonEvent, CollectionHandle,111	eth::collection_id_to_address, SelfWeightOf as PalletCommonWeightOf,112	weights::WeightInfo as CommonWeightInfo, helpers::add_weight_to_post_info,113};114use pallet_structure::{Pallet as PalletStructure, Error as StructureError};115use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};116use sp_core::{Get, H160};117use sp_runtime::{ArithmeticError, DispatchError, DispatchResult, TransactionOutcome};118use sp_std::{vec::Vec, vec, collections::btree_map::BTreeMap};119use core::ops::Deref;120use codec::{Encode, Decode, MaxEncodedLen};121use scale_info::TypeInfo;122123pub use pallet::*;124use weights::WeightInfo;125#[cfg(feature = "runtime-benchmarks")]126pub mod benchmarking;127pub mod common;128pub mod erc;129pub mod weights;130131pub type CreateItemData<T> = CreateNftExData<<T as pallet_evm::Config>::CrossAccountId>;132pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;133134/// Token data, stored independently from other data used to describe it135/// for the convenience of database access. Notably contains the owner account address.136#[struct_versioning::versioned(version = 2, upper)]137#[derive(Encode, Decode, TypeInfo, MaxEncodedLen)]138pub struct ItemData<CrossAccountId> {139	#[version(..2)]140	pub const_data: BoundedVec<u8, CustomDataLimit>,141142	#[version(..2)]143	pub variable_data: BoundedVec<u8, CustomDataLimit>,144145	pub owner: CrossAccountId,146}147148#[frame_support::pallet]149pub mod pallet {150	use super::*;151	use frame_support::{152		Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key, traits::StorageVersion,153	};154	use frame_system::pallet_prelude::*;155	use up_data_structs::{CollectionId, TokenId};156	use super::weights::WeightInfo;157158	#[pallet::error]159	pub enum Error<T> {160		/// Not Nonfungible item data used to mint in Nonfungible collection.161		NotNonfungibleDataUsedToMintFungibleCollectionToken,162		/// Used amount > 1 with NFT163		NonfungibleItemsHaveNoAmount,164		/// Unable to burn NFT with children165		CantBurnNftWithChildren,166	}167168	#[pallet::config]169	pub trait Config:170		frame_system::Config + pallet_common::Config + pallet_structure::Config + pallet_evm::Config171	{172		type WeightInfo: WeightInfo;173	}174175	const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);176177	#[pallet::pallet]178	#[pallet::storage_version(STORAGE_VERSION)]179	pub struct Pallet<T>(_);180181	/// Total amount of minted tokens in a collection.182	#[pallet::storage]183	pub type TokensMinted<T: Config> =184		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;185186	/// Amount of burnt tokens in a collection.187	#[pallet::storage]188	pub type TokensBurnt<T: Config> =189		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;190191	/// Token data, used to partially describe a token.192	#[pallet::storage]193	pub type TokenData<T: Config> = StorageNMap<194		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),195		Value = ItemData<T::CrossAccountId>,196		QueryKind = OptionQuery,197	>;198199	/// Map of key-value pairs, describing the metadata of a token.200	#[pallet::storage]201	#[pallet::getter(fn token_properties)]202	pub type TokenProperties<T: Config> = StorageNMap<203		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),204		Value = TokenPropertiesT,205		QueryKind = ValueQuery,206	>;207208	/// Custom data of a token that is serialized to bytes,209	/// primarily reserved for on-chain operations,210	/// normally obscured from the external users.211	///212	/// Auxiliary properties are slightly different from213	/// usual [`TokenProperties`] due to an unlimited number214	/// and separately stored and written-to key-value pairs.215	///216	/// Currently unused.217	#[pallet::storage]218	#[pallet::getter(fn token_aux_property)]219	pub type TokenAuxProperties<T: Config> = StorageNMap<220		Key = (221			Key<Twox64Concat, CollectionId>,222			Key<Twox64Concat, TokenId>,223			Key<Twox64Concat, PropertyScope>,224			Key<Twox64Concat, PropertyKey>,225		),226		Value = AuxPropertyValue,227		QueryKind = OptionQuery,228	>;229230	/// Used to enumerate tokens owned by account.231	#[pallet::storage]232	pub type Owned<T: Config> = StorageNMap<233		Key = (234			Key<Twox64Concat, CollectionId>,235			Key<Blake2_128Concat, T::CrossAccountId>,236			Key<Twox64Concat, TokenId>,237		),238		Value = bool,239		QueryKind = ValueQuery,240	>;241242	/// Used to enumerate token's children.243	#[pallet::storage]244	#[pallet::getter(fn token_children)]245	pub type TokenChildren<T: Config> = StorageNMap<246		Key = (247			Key<Twox64Concat, CollectionId>,248			Key<Twox64Concat, TokenId>,249			Key<Twox64Concat, (CollectionId, TokenId)>,250		),251		Value = bool,252		QueryKind = ValueQuery,253	>;254255	/// Amount of tokens owned by an account in a collection.256	#[pallet::storage]257	pub type AccountBalance<T: Config> = StorageNMap<258		Key = (259			Key<Twox64Concat, CollectionId>,260			Key<Blake2_128Concat, T::CrossAccountId>,261		),262		Value = u32,263		QueryKind = ValueQuery,264	>;265266	/// Allowance set by a token owner for another user to perform one of certain transactions on a token.267	#[pallet::storage]268	pub type Allowance<T: Config> = StorageNMap<269		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),270		Value = T::CrossAccountId,271		QueryKind = OptionQuery,272	>;273274	/// Operator set by a wallet owner that could perform certain transactions on all tokens in the wallet.275	#[pallet::storage]276	pub type CollectionAllowance<T: Config> = StorageNMap<277		Key = (278			Key<Twox64Concat, CollectionId>,279			Key<Blake2_128Concat, T::CrossAccountId>,280			Key<Blake2_128Concat, T::CrossAccountId>,281		),282		Value = bool,283		QueryKind = ValueQuery,284	>;285286	#[pallet::genesis_config]287	pub struct GenesisConfig<T>(PhantomData<T>);288289	#[cfg(feature = "std")]290	impl<T: Config> Default for GenesisConfig<T> {291		fn default() -> Self {292			Self(Default::default())293		}294	}295296	#[pallet::genesis_build]297	impl<T: Config> GenesisBuild<T> for GenesisConfig<T> {298		fn build(&self) {299			StorageVersion::new(1).put::<Pallet<T>>();300		}301	}302}303304pub struct NonfungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);305impl<T: Config> NonfungibleHandle<T> {306	pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {307		Self(inner)308	}309	pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {310		self.0311	}312	pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {313		&mut self.0314	}315}316317impl<T: Config> WithRecorder<T> for NonfungibleHandle<T> {318	fn recorder(&self) -> &SubstrateRecorder<T> {319		self.0.recorder()320	}321	fn into_recorder(self) -> SubstrateRecorder<T> {322		self.0.into_recorder()323	}324}325impl<T: Config> Deref for NonfungibleHandle<T> {326	type Target = pallet_common::CollectionHandle<T>;327328	fn deref(&self) -> &Self::Target {329		&self.0330	}331}332333impl<T: Config> Pallet<T> {334	/// Get number of NFT tokens in collection.335	pub fn total_supply(collection: &NonfungibleHandle<T>) -> u32 {336		<TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)337	}338339	/// Check that NFT token exists.340	///341	/// - `token`: Token ID.342	pub fn token_exists(collection: &NonfungibleHandle<T>, token: TokenId) -> bool {343		<TokenData<T>>::contains_key((collection.id, token))344	}345346	/// Set the token property with the scope.347	///348	/// - `property`: Contains key-value pair.349	pub fn set_scoped_token_property(350		collection_id: CollectionId,351		token_id: TokenId,352		scope: PropertyScope,353		property: Property,354	) -> DispatchResult {355		TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {356			properties.try_scoped_set(scope, property.key, property.value)357		})358		.map_err(<CommonError<T>>::from)?;359360		Ok(())361	}362363	/// Batch operation to set multiple properties with the same scope.364	pub fn set_scoped_token_properties(365		collection_id: CollectionId,366		token_id: TokenId,367		scope: PropertyScope,368		properties: impl Iterator<Item = Property>,369	) -> DispatchResult {370		TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {371			stored_properties.try_scoped_set_from_iter(scope, properties)372		})373		.map_err(<CommonError<T>>::from)?;374375		Ok(())376	}377378	/// Add or edit auxiliary data for the property.379	///380	/// - `f`: function that adds or edits auxiliary data.381	pub fn try_mutate_token_aux_property<R, E>(382		collection_id: CollectionId,383		token_id: TokenId,384		scope: PropertyScope,385		key: PropertyKey,386		f: impl FnOnce(&mut Option<AuxPropertyValue>) -> Result<R, E>,387	) -> Result<R, E> {388		<TokenAuxProperties<T>>::try_mutate((collection_id, token_id, scope, key), f)389	}390391	/// Remove auxiliary data for the property.392	pub fn remove_token_aux_property(393		collection_id: CollectionId,394		token_id: TokenId,395		scope: PropertyScope,396		key: PropertyKey,397	) {398		<TokenAuxProperties<T>>::remove((collection_id, token_id, scope, key));399	}400401	/// Get all auxiliary data in a given scope.402	///403	/// Returns iterator over Property Key - Data pairs.404	pub fn iterate_token_aux_properties(405		collection_id: CollectionId,406		token_id: TokenId,407		scope: PropertyScope,408	) -> impl Iterator<Item = (PropertyKey, AuxPropertyValue)> {409		<TokenAuxProperties<T>>::iter_prefix((collection_id, token_id, scope))410	}411412	/// Get ID of the last minted token413	pub fn current_token_id(collection_id: CollectionId) -> TokenId {414		TokenId(<TokensMinted<T>>::get(collection_id))415	}416}417418// unchecked calls skips any permission checks419impl<T: Config> Pallet<T> {420	/// Create NFT collection421	///422	/// `init_collection` will take non-refundable deposit for collection creation.423	///424	/// - `data`: Contains settings for collection limits and permissions.425	pub fn init_collection(426		owner: T::CrossAccountId,427		payer: T::CrossAccountId,428		data: CreateCollectionData<T::AccountId>,429		flags: CollectionFlags,430	) -> Result<CollectionId, DispatchError> {431		<PalletCommon<T>>::init_collection(owner, payer, data, flags)432	}433434	/// Destroy NFT collection435	///436	/// `destroy_collection` will throw error if collection contains any tokens.437	/// Only owner can destroy collection.438	pub fn destroy_collection(439		collection: NonfungibleHandle<T>,440		sender: &T::CrossAccountId,441	) -> DispatchResult {442		let id = collection.id;443444		if Self::collection_has_tokens(id) {445			return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());446		}447448		// =========449450		PalletCommon::destroy_collection(collection.0, sender)?;451452		let _ = <TokenData<T>>::clear_prefix((id,), u32::MAX, None);453		let _ = <TokenChildren<T>>::clear_prefix((id,), u32::MAX, None);454		let _ = <Owned<T>>::clear_prefix((id,), u32::MAX, None);455		<TokensMinted<T>>::remove(id);456		<TokensBurnt<T>>::remove(id);457		let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);458		let _ = <AccountBalance<T>>::clear_prefix((id,), u32::MAX, None);459		let _ = <CollectionAllowance<T>>::clear_prefix((id,), u32::MAX, None);460		Ok(())461	}462463	/// Burn NFT token464	///465	/// `burn` removes `token` from the `collection`, from it's owner and from the parent token466	/// if the token is nested.467	/// Only the owner can `burn` the token. The `token` shouldn't have any nested tokens.468	/// Also removes all corresponding properties and auxiliary properties.469	///470	/// - `token`: Token that should be burned471	/// - `collection`: Collection that contains the token472	pub fn burn(473		collection: &NonfungibleHandle<T>,474		sender: &T::CrossAccountId,475		token: TokenId,476	) -> DispatchResult {477		let token_data =478			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;479		ensure!(&token_data.owner == sender, <CommonError<T>>::NoPermission);480481		if collection.permissions.access() == AccessMode::AllowList {482			collection.check_allowlist(sender)?;483		}484485		if Self::token_has_children(collection.id, token) {486			return Err(<Error<T>>::CantBurnNftWithChildren.into());487		}488489		let burnt = <TokensBurnt<T>>::get(collection.id)490			.checked_add(1)491			.ok_or(ArithmeticError::Overflow)?;492493		let balance = <AccountBalance<T>>::get((collection.id, token_data.owner.clone()))494			.checked_sub(1)495			.ok_or(ArithmeticError::Overflow)?;496497		// =========498499		if balance == 0 {500			<AccountBalance<T>>::remove((collection.id, token_data.owner.clone()));501		} else {502			<AccountBalance<T>>::insert((collection.id, token_data.owner.clone()), balance);503		}504505		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);506507		<Owned<T>>::remove((collection.id, &token_data.owner, token));508		<TokensBurnt<T>>::insert(collection.id, burnt);509		<TokenData<T>>::remove((collection.id, token));510		<TokenProperties<T>>::remove((collection.id, token));511		let _ = <TokenAuxProperties<T>>::clear_prefix((collection.id, token), u32::MAX, None);512		let old_spender = <Allowance<T>>::take((collection.id, token));513514		if let Some(old_spender) = old_spender {515			<PalletCommon<T>>::deposit_event(CommonEvent::Approved(516				collection.id,517				token,518				token_data.owner.clone(),519				old_spender,520				0,521			));522		}523524		<PalletEvm<T>>::deposit_log(525			ERC721Events::Transfer {526				from: *token_data.owner.as_eth(),527				to: H160::default(),528				token_id: token.into(),529			}530			.to_log(collection_id_to_address(collection.id)),531		);532		<PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(533			collection.id,534			token,535			token_data.owner,536			1,537		));538		Ok(())539	}540541	/// Same as [`burn`] but burns all the tokens that are nested in the token first542	///543	/// - `self_budget`: Limit for searching children in depth.544	/// - `breadth_budget`: Limit of breadth of searching children.545	///546	/// [`burn`]: struct.Pallet.html#method.burn547	#[transactional]548	pub fn burn_recursively(549		collection: &NonfungibleHandle<T>,550		sender: &T::CrossAccountId,551		token: TokenId,552		self_budget: &dyn Budget,553		breadth_budget: &dyn Budget,554	) -> DispatchResultWithPostInfo {555		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);556557		let current_token_account =558			T::CrossTokenAddressMapping::token_to_address(collection.id, token);559560		let mut weight = Weight::zero();561562		// This method is transactional, if user in fact doesn't have permissions to remove token -563		// tokens removed here will be restored after rejected transaction564		for ((collection, token), _) in <TokenChildren<T>>::iter_prefix((collection.id, token)) {565			ensure!(breadth_budget.consume(), <StructureError<T>>::BreadthLimit,);566			let PostDispatchInfo { actual_weight, .. } =567				<PalletStructure<T>>::burn_item_recursively(568					current_token_account.clone(),569					collection,570					token,571					self_budget,572					breadth_budget,573				)?;574			if let Some(actual_weight) = actual_weight {575				weight = weight.saturating_add(actual_weight);576			}577		}578579		Self::burn(collection, sender, token)?;580		DispatchResultWithPostInfo::Ok(PostDispatchInfo {581			actual_weight: Some(weight + <SelfWeightOf<T>>::burn_item()),582			pays_fee: Pays::Yes,583		})584	}585586	/// A batch operation to add, edit or remove properties for a token.587	///588	/// - `nesting_budget`: Limit for searching parents in-depth to check ownership.589	/// - `is_token_create`: Indicates that method is called during token initialization.590	///   Allows to bypass ownership check.591	///592	/// All affected properties should have `mutable` permission593	/// to be **deleted** or to be **set more than once**,594	/// and the sender should have permission to edit those properties.595	///596	/// This function fires an event for each property change.597	/// In case of an error, all the changes (including the events) will be reverted598	/// since the function is transactional.599	#[transactional]600	fn modify_token_properties(601		collection: &NonfungibleHandle<T>,602		sender: &T::CrossAccountId,603		token_id: TokenId,604		properties_updates: impl Iterator<Item = (PropertyKey, Option<PropertyValue>)>,605		is_token_create: bool,606		nesting_budget: &dyn Budget,607	) -> DispatchResult {608		let is_token_owner = || {609			let is_owned = <PalletStructure<T>>::check_indirectly_owned(610				sender.clone(),611				collection.id,612				token_id,613				None,614				nesting_budget,615			)?;616617			Ok(is_owned)618		};619620		let stored_properties = <TokenProperties<T>>::get((collection.id, token_id));621622		<PalletCommon<T>>::modify_token_properties(623			collection,624			sender,625			token_id,626			properties_updates,627			is_token_create,628			stored_properties,629			is_token_owner,630			|properties| <TokenProperties<T>>::set((collection.id, token_id), properties),631			erc::ERC721TokenEvent::TokenChanged {632				token_id: token_id.into(),633			}634			.to_log(T::ContractAddress::get()),635		)636	}637638	/// Batch operation to add or edit properties for the token639	///640	/// Same as [`modify_token_properties`] but doesn't allow to remove properties641	///642	/// [`modify_token_properties`]: struct.Pallet.html#method.modify_token_properties643	pub fn set_token_properties(644		collection: &NonfungibleHandle<T>,645		sender: &T::CrossAccountId,646		token_id: TokenId,647		properties: impl Iterator<Item = Property>,648		is_token_create: bool,649		nesting_budget: &dyn Budget,650	) -> DispatchResult {651		Self::modify_token_properties(652			collection,653			sender,654			token_id,655			properties.map(|p| (p.key, Some(p.value))),656			is_token_create,657			nesting_budget,658		)659	}660661	/// Add or edit single property for the token662	///663	/// Calls [`set_token_properties`] internally664	///665	/// [`set_token_properties`]: struct.Pallet.html#method.set_token_properties666	pub fn set_token_property(667		collection: &NonfungibleHandle<T>,668		sender: &T::CrossAccountId,669		token_id: TokenId,670		property: Property,671		nesting_budget: &dyn Budget,672	) -> DispatchResult {673		let is_token_create = false;674675		Self::set_token_properties(676			collection,677			sender,678			token_id,679			[property].into_iter(),680			is_token_create,681			nesting_budget,682		)683	}684685	/// Batch operation to remove properties from the token686	///687	/// Same as [`modify_token_properties`] but doesn't allow to add or edit properties688	///689	/// [`modify_token_properties`]: struct.Pallet.html#method.modify_token_properties690	pub fn delete_token_properties(691		collection: &NonfungibleHandle<T>,692		sender: &T::CrossAccountId,693		token_id: TokenId,694		property_keys: impl Iterator<Item = PropertyKey>,695		nesting_budget: &dyn Budget,696	) -> DispatchResult {697		let is_token_create = false;698699		Self::modify_token_properties(700			collection,701			sender,702			token_id,703			property_keys.into_iter().map(|key| (key, None)),704			is_token_create,705			nesting_budget,706		)707	}708709	/// Remove single property from the token710	///711	/// Calls [`delete_token_properties`] internally712	///713	/// [`delete_token_properties`]: struct.Pallet.html#method.delete_token_properties714	pub fn delete_token_property(715		collection: &NonfungibleHandle<T>,716		sender: &T::CrossAccountId,717		token_id: TokenId,718		property_key: PropertyKey,719		nesting_budget: &dyn Budget,720	) -> DispatchResult {721		Self::delete_token_properties(722			collection,723			sender,724			token_id,725			[property_key].into_iter(),726			nesting_budget,727		)728	}729730	/// Add or edit properties for the collection731	pub fn set_collection_properties(732		collection: &NonfungibleHandle<T>,733		sender: &T::CrossAccountId,734		properties: Vec<Property>,735	) -> DispatchResult {736		<PalletCommon<T>>::set_collection_properties(collection, sender, properties.into_iter())737	}738739	/// Remove properties from the collection740	pub fn delete_collection_properties(741		collection: &CollectionHandle<T>,742		sender: &T::CrossAccountId,743		property_keys: Vec<PropertyKey>,744	) -> DispatchResult {745		<PalletCommon<T>>::delete_collection_properties(746			collection,747			sender,748			property_keys.into_iter(),749		)750	}751752	/// Set property permissions for the token.753	///754	/// Sender should be the owner or admin of token's collection.755	pub fn set_token_property_permissions(756		collection: &CollectionHandle<T>,757		sender: &T::CrossAccountId,758		property_permissions: Vec<PropertyKeyPermission>,759	) -> DispatchResult {760		<PalletCommon<T>>::set_token_property_permissions(collection, sender, property_permissions)761	}762763	/// Set property permissions for the token with scope.764	///765	/// Sender should be the owner or admin of token's collection.766	pub fn set_scoped_token_property_permissions(767		collection: &CollectionHandle<T>,768		sender: &T::CrossAccountId,769		scope: PropertyScope,770		property_permissions: Vec<PropertyKeyPermission>,771	) -> DispatchResult {772		<PalletCommon<T>>::set_scoped_token_property_permissions(773			collection,774			sender,775			scope,776			property_permissions,777		)778	}779780	pub fn token_property_permission(collection_id: CollectionId) -> PropertiesPermissionMap {781		<PalletCommon<T>>::property_permissions(collection_id)782	}783784	pub fn check_token_immediate_ownership(785		collection: &NonfungibleHandle<T>,786		token: TokenId,787		possible_owner: &T::CrossAccountId,788	) -> DispatchResult {789		let token_data =790			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;791		ensure!(792			&token_data.owner == possible_owner,793			<CommonError<T>>::NoPermission794		);795		Ok(())796	}797798	/// Transfer NFT token from one account to another.799	///800	/// `from` account stops being the owner and `to` account becomes the owner of the token.801	/// If `to` is token than `to` becomes owner of the token and the token become nested.802	/// Unnests token from previous parent if it was nested before.803	/// Removes allowance for the token if there was any.804	/// Throws if transfers aren't allowed for collection or if receiver reached token ownership limit.805	///806	/// - `nesting_budget`: Limit for token nesting depth807	pub fn transfer(808		collection: &NonfungibleHandle<T>,809		from: &T::CrossAccountId,810		to: &T::CrossAccountId,811		token: TokenId,812		nesting_budget: &dyn Budget,813	) -> DispatchResultWithPostInfo {814		ensure!(815			collection.limits.transfers_enabled(),816			<CommonError<T>>::TransferNotAllowed817		);818819		let mut actual_weight = <SelfWeightOf<T>>::transfer_raw();820		let token_data =821			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;822		ensure!(&token_data.owner == from, <CommonError<T>>::NoPermission);823824		if collection.permissions.access() == AccessMode::AllowList {825			collection.check_allowlist(from)?;826			collection.check_allowlist(to)?;827			actual_weight += <PalletCommonWeightOf<T>>::check_accesslist() * 2;828		}829		<PalletCommon<T>>::ensure_correct_receiver(to)?;830831		let balance_from = <AccountBalance<T>>::get((collection.id, from))832			.checked_sub(1)833			.ok_or(<CommonError<T>>::TokenValueTooLow)?;834		let balance_to = if from != to {835			let balance_to = <AccountBalance<T>>::get((collection.id, to))836				.checked_add(1)837				.ok_or(ArithmeticError::Overflow)?;838839			ensure!(840				balance_to < collection.limits.account_token_ownership_limit(),841				<CommonError<T>>::AccountTokenLimitExceeded,842			);843844			Some(balance_to)845		} else {846			None847		};848849		<PalletStructure<T>>::nest_if_sent_to_token(850			from.clone(),851			to,852			collection.id,853			token,854			nesting_budget,855		)?;856857		// =========858859		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);860861		<TokenData<T>>::insert(862			(collection.id, token),863			ItemData {864				owner: to.clone(),865				..token_data866			},867		);868869		if let Some(balance_to) = balance_to {870			// from != to871			if balance_from == 0 {872				<AccountBalance<T>>::remove((collection.id, from));873			} else {874				<AccountBalance<T>>::insert((collection.id, from), balance_from);875			}876			<AccountBalance<T>>::insert((collection.id, to), balance_to);877			<Owned<T>>::remove((collection.id, from, token));878			<Owned<T>>::insert((collection.id, to, token), true);879		}880		Self::set_allowance_unchecked(collection, from, token, None, true);881882		<PalletEvm<T>>::deposit_log(883			ERC721Events::Transfer {884				from: *from.as_eth(),885				to: *to.as_eth(),886				token_id: token.into(),887			}888			.to_log(collection_id_to_address(collection.id)),889		);890		<PalletCommon<T>>::deposit_event(CommonEvent::Transfer(891			collection.id,892			token,893			from.clone(),894			to.clone(),895			1,896		));897898		Ok(PostDispatchInfo {899			actual_weight: Some(actual_weight),900			pays_fee: Pays::Yes,901		})902	}903904	/// Batch operation to mint multiple NFT tokens.905	///906	/// The sender should be the owner/admin of the collection or collection should be configured907	/// to allow public minting.908	/// Throws if amount of tokens reached it's limit for the collection or if caller reached909	/// token ownership limit.910	///911	/// - `data`: Contains list of token properties and users who will become the owners of the912	///   corresponging tokens.913	/// - `nesting_budget`: Limit for token nesting depth914	pub fn create_multiple_items(915		collection: &NonfungibleHandle<T>,916		sender: &T::CrossAccountId,917		data: Vec<CreateItemData<T>>,918		nesting_budget: &dyn Budget,919	) -> DispatchResult {920		if !collection.is_owner_or_admin(sender) {921			ensure!(922				collection.permissions.mint_mode(),923				<CommonError<T>>::PublicMintingNotAllowed924			);925			collection.check_allowlist(sender)?;926927			for item in data.iter() {928				collection.check_allowlist(&item.owner)?;929			}930		}931932		for data in data.iter() {933			<PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;934		}935936		let first_token = <TokensMinted<T>>::get(collection.id);937		let tokens_minted = first_token938			.checked_add(data.len() as u32)939			.ok_or(ArithmeticError::Overflow)?;940		ensure!(941			tokens_minted <= collection.limits.token_limit(),942			<CommonError<T>>::CollectionTokenLimitExceeded943		);944945		let mut balances = BTreeMap::new();946		for data in &data {947			let balance = balances948				.entry(&data.owner)949				.or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));950			*balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;951952			ensure!(953				*balance <= collection.limits.account_token_ownership_limit(),954				<CommonError<T>>::AccountTokenLimitExceeded,955			);956		}957958		for (i, data) in data.iter().enumerate() {959			let token = TokenId(first_token + i as u32 + 1);960961			<PalletStructure<T>>::check_nesting(962				sender.clone(),963				&data.owner,964				collection.id,965				token,966				nesting_budget,967			)?;968		}969970		// =========971972		with_transaction(|| {973			for (i, data) in data.iter().enumerate() {974				let token = first_token + i as u32 + 1;975976				<TokenData<T>>::insert(977					(collection.id, token),978					ItemData {979						// const_data: data.const_data.clone(),980						owner: data.owner.clone(),981					},982				);983984				<PalletStructure<T>>::nest_if_sent_to_token_unchecked(985					&data.owner,986					collection.id,987					TokenId(token),988				);989990				if let Err(e) = Self::set_token_properties(991					collection,992					sender,993					TokenId(token),994					data.properties.clone().into_iter(),995					true,996					nesting_budget,997				) {998					return TransactionOutcome::Rollback(Err(e));999				}1000			}1001			TransactionOutcome::Commit(Ok(()))1002		})?;10031004		<TokensMinted<T>>::insert(collection.id, tokens_minted);1005		for (account, balance) in balances {1006			<AccountBalance<T>>::insert((collection.id, account), balance);1007		}1008		for (i, data) in data.into_iter().enumerate() {1009			let token = first_token + i as u32 + 1;1010			<Owned<T>>::insert((collection.id, &data.owner, token), true);10111012			<PalletEvm<T>>::deposit_log(1013				ERC721Events::Transfer {1014					from: H160::default(),1015					to: *data.owner.as_eth(),1016					token_id: token.into(),1017				}1018				.to_log(collection_id_to_address(collection.id)),1019			);1020			<PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(1021				collection.id,1022				TokenId(token),1023				data.owner.clone(),1024				1,1025			));1026		}1027		Ok(())1028	}10291030	pub fn set_allowance_unchecked(1031		collection: &NonfungibleHandle<T>,1032		sender: &T::CrossAccountId,1033		token: TokenId,1034		spender: Option<&T::CrossAccountId>,1035		assume_implicit_eth: bool,1036	) {1037		if let Some(spender) = spender {1038			let old_spender = <Allowance<T>>::get((collection.id, token));1039			<Allowance<T>>::insert((collection.id, token), spender);1040			// In ERC721 there is only one possible approved user of token, so we set1041			// approved user to spender1042			<PalletEvm<T>>::deposit_log(1043				ERC721Events::Approval {1044					owner: *sender.as_eth(),1045					approved: *spender.as_eth(),1046					token_id: token.into(),1047				}1048				.to_log(collection_id_to_address(collection.id)),1049			);1050			// In Unique chain, any token can have any amount of approved users, so we need to1051			// set allowance of old owner to 0, and allowance of new owner to 11052			if old_spender.as_ref() != Some(spender) {1053				if let Some(old_owner) = old_spender {1054					<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1055						collection.id,1056						token,1057						sender.clone(),1058						old_owner,1059						0,1060					));1061				}1062				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1063					collection.id,1064					token,1065					sender.clone(),1066					spender.clone(),1067					1,1068				));1069			}1070		} else {1071			let old_spender = <Allowance<T>>::take((collection.id, token));1072			if !assume_implicit_eth {1073				// In ERC721 there is only one possible approved user of token, so we set1074				// approved user to zero address1075				<PalletEvm<T>>::deposit_log(1076					ERC721Events::Approval {1077						owner: *sender.as_eth(),1078						approved: H160::default(),1079						token_id: token.into(),1080					}1081					.to_log(collection_id_to_address(collection.id)),1082				);1083			}1084			// In Unique chain, any token can have any amount of approved users, so we need to1085			// set allowance of old owner to 01086			if let Some(old_spender) = old_spender {1087				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1088					collection.id,1089					token,1090					sender.clone(),1091					old_spender,1092					0,1093				));1094			}1095		}1096	}10971098	pub fn get_allowance(1099		collection: &NonfungibleHandle<T>,1100		token_id: TokenId,1101	) -> Result<Option<T::CrossAccountId>, DispatchError> {1102		ensure!(1103			<TokenData<T>>::get((collection.id, token_id)).is_some(),1104			<CommonError<T>>::TokenNotFound1105		);1106		Ok(<Allowance<T>>::get((collection.id, token_id)))1107	}11081109	/// Set allowance for the spender to `transfer` or `burn` sender's token.1110	///1111	/// - `token`: Token the spender is allowed to `transfer` or `burn`.1112	pub fn set_allowance(1113		collection: &NonfungibleHandle<T>,1114		sender: &T::CrossAccountId,1115		token: TokenId,1116		spender: Option<&T::CrossAccountId>,1117	) -> DispatchResult {1118		if collection.permissions.access() == AccessMode::AllowList {1119			collection.check_allowlist(sender)?;1120			if let Some(spender) = spender {1121				collection.check_allowlist(spender)?;1122			}1123		}11241125		if let Some(spender) = spender {1126			<PalletCommon<T>>::ensure_correct_receiver(spender)?;1127		}11281129		let token_data =1130			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;1131		if &token_data.owner != sender {1132			ensure!(1133				collection.ignores_owned_amount(sender),1134				<CommonError<T>>::CantApproveMoreThanOwned1135			);1136		}11371138		// =========11391140		Self::set_allowance_unchecked(collection, sender, token, spender, false);1141		Ok(())1142	}11431144	/// Set allowance for the spender to `transfer` or `burn` sender's token from eth mirror.1145	///1146	/// - `from`: Address of sender's eth mirror.1147	/// - `to`: Adress of spender.1148	/// - `token`: Token the spender is allowed to `transfer` or `burn`.1149	pub fn set_allowance_from(1150		collection: &NonfungibleHandle<T>,1151		sender: &T::CrossAccountId,1152		from: &T::CrossAccountId,1153		token: TokenId,1154		to: Option<&T::CrossAccountId>,1155	) -> DispatchResult {1156		if collection.permissions.access() == AccessMode::AllowList {1157			collection.check_allowlist(sender)?;1158			collection.check_allowlist(from)?;1159			if let Some(to) = to {1160				collection.check_allowlist(to)?;1161			}1162		}11631164		if let Some(to) = to {1165			<PalletCommon<T>>::ensure_correct_receiver(to)?;1166		}11671168		ensure!(1169			sender.conv_eq(from),1170			<CommonError<T>>::AddressIsNotEthMirror1171		);11721173		let token_data =1174			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;1175		if token_data.owner != *from {1176			ensure!(1177				collection.limits.owner_can_transfer()1178					&& (collection.is_owner_or_admin(sender) || collection.is_owner_or_admin(from)),1179				<CommonError<T>>::CantApproveMoreThanOwned1180			);1181		}11821183		// =========11841185		Self::set_allowance_unchecked(collection, from, token, to, false);1186		Ok(())1187	}11881189	/// Checks allowance for the spender to use the token.1190	fn check_allowed(1191		collection: &NonfungibleHandle<T>,1192		spender: &T::CrossAccountId,1193		from: &T::CrossAccountId,1194		token: TokenId,1195		nesting_budget: &dyn Budget,1196	) -> DispatchResult {1197		if spender.conv_eq(from) {1198			return Ok(());1199		}1200		if collection.permissions.access() == AccessMode::AllowList {1201			// `from`, `to` checked in [`transfer`]1202			collection.check_allowlist(spender)?;1203		}12041205		if collection.ignores_token_restrictions(spender) {1206			return Ok(());1207		}12081209		if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {1210			ensure!(1211				<PalletStructure<T>>::check_indirectly_owned(1212					spender.clone(),1213					source.0,1214					source.1,1215					None,1216					nesting_budget1217				)?,1218				<CommonError<T>>::ApprovedValueTooLow,1219			);1220			return Ok(());1221		}1222		if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {1223			return Ok(());1224		}1225		if <CollectionAllowance<T>>::get((collection.id, from, spender)) {1226			return Ok(());1227		}12281229		Err(<CommonError<T>>::ApprovedValueTooLow.into())1230	}12311232	/// Transfer NFT token from one account to another.1233	///1234	/// Same as the [`transfer`] but spender doesn't needs to be the owner of the token.1235	/// The owner should set allowance for the spender to transfer token.1236	///1237	/// [`transfer`]: struct.Pallet.html#method.transfer1238	pub fn transfer_from(1239		collection: &NonfungibleHandle<T>,1240		spender: &T::CrossAccountId,1241		from: &T::CrossAccountId,1242		to: &T::CrossAccountId,1243		token: TokenId,1244		nesting_budget: &dyn Budget,1245	) -> DispatchResultWithPostInfo {1246		Self::check_allowed(collection, spender, from, token, nesting_budget)?;12471248		// =========12491250		// Allowance is reset in [`transfer`]1251		let mut result = Self::transfer(collection, from, to, token, nesting_budget);1252		add_weight_to_post_info(&mut result, <SelfWeightOf<T>>::check_allowed_raw());1253		result1254	}12551256	/// Burn NFT token for `from` account.1257	///1258	/// Same as the [`burn`] but spender doesn't need to be an owner of the token. The owner should1259	/// set allowance for the spender to burn token.1260	///1261	/// [`burn`]: struct.Pallet.html#method.burn1262	pub fn burn_from(1263		collection: &NonfungibleHandle<T>,1264		spender: &T::CrossAccountId,1265		from: &T::CrossAccountId,1266		token: TokenId,1267		nesting_budget: &dyn Budget,1268	) -> DispatchResult {1269		Self::check_allowed(collection, spender, from, token, nesting_budget)?;12701271		// =========12721273		Self::burn(collection, from, token)1274	}12751276	/// Check that `from` token could be nested in `under` token.1277	///1278	pub fn check_nesting(1279		handle: &NonfungibleHandle<T>,1280		sender: T::CrossAccountId,1281		from: (CollectionId, TokenId),1282		under: TokenId,1283		nesting_budget: &dyn Budget,1284	) -> DispatchResult {1285		let nesting = handle.permissions.nesting();12861287		#[cfg(not(feature = "runtime-benchmarks"))]1288		let permissive = false;1289		#[cfg(feature = "runtime-benchmarks")]1290		let permissive = nesting.permissive;12911292		if permissive {1293			ensure!(1294				<TokenData<T>>::contains_key((handle.id, under)),1295				<CommonError<T>>::TokenNotFound1296			);1297		} else if nesting.token_owner1298			&& <PalletStructure<T>>::check_indirectly_owned(1299				sender.clone(),1300				handle.id,1301				under,1302				Some(from),1303				nesting_budget,1304			)? {1305			// Pass, token existence and ouroboros checks are done in `check_indirectly_owned`1306		} else if nesting.collection_admin && handle.is_owner_or_admin(&sender) {1307			// token existence and ouroboros checks are done in `get_checked_topmost_owner`1308			let _ = <PalletStructure<T>>::get_checked_topmost_owner(1309				handle.id,1310				under,1311				Some(from),1312				nesting_budget,1313			)?1314			.ok_or(<CommonError<T>>::TokenNotFound)?;1315		} else {1316			fail!(<CommonError<T>>::UserIsNotAllowedToNest);1317		}13181319		if let Some(whitelist) = &nesting.restricted {1320			ensure!(1321				whitelist.contains(&from.0),1322				<CommonError<T>>::SourceCollectionIsNotAllowedToNest1323			);1324		}1325		Ok(())1326	}13271328	fn nest(under: (CollectionId, TokenId), to_nest: (CollectionId, TokenId)) {1329		<TokenChildren<T>>::insert((under.0, under.1, (to_nest.0, to_nest.1)), true);1330	}13311332	fn unnest(under: (CollectionId, TokenId), to_unnest: (CollectionId, TokenId)) {1333		<TokenChildren<T>>::remove((under.0, under.1, to_unnest));1334	}13351336	fn collection_has_tokens(collection_id: CollectionId) -> bool {1337		<TokenData<T>>::iter_prefix((collection_id,))1338			.next()1339			.is_some()1340	}13411342	fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {1343		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1344			.next()1345			.is_some()1346	}13471348	pub fn token_children_ids(collection_id: CollectionId, token_id: TokenId) -> Vec<TokenChild> {1349		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1350			.map(|((child_collection_id, child_id), _)| TokenChild {1351				collection: child_collection_id,1352				token: child_id,1353			})1354			.collect()1355	}13561357	/// Mint single NFT token.1358	///1359	/// Delegated to [`create_multiple_items`]1360	///1361	/// [`create_multiple_items`]: struct.Pallet.html#method.create_multiple_items1362	pub fn create_item(1363		collection: &NonfungibleHandle<T>,1364		sender: &T::CrossAccountId,1365		data: CreateItemData<T>,1366		nesting_budget: &dyn Budget,1367	) -> DispatchResult {1368		Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1369	}13701371	/// Sets or unsets the approval of a given operator.1372	///1373	/// The `operator` is allowed to transfer all token pieces of the `owner` on their behalf.1374	/// - `owner`: Token owner1375	/// - `operator`: Operator1376	/// - `approve`: Should operator status be granted or revoked?1377	pub fn set_allowance_for_all(1378		collection: &NonfungibleHandle<T>,1379		owner: &T::CrossAccountId,1380		operator: &T::CrossAccountId,1381		approve: bool,1382	) -> DispatchResult {1383		<PalletCommon<T>>::set_allowance_for_all(1384			collection,1385			owner,1386			operator,1387			approve,1388			|| <CollectionAllowance<T>>::insert((collection.id, owner, operator), approve),1389			ERC721Events::ApprovalForAll {1390				owner: *owner.as_eth(),1391				operator: *operator.as_eth(),1392				approved: approve,1393			}1394			.to_log(collection_id_to_address(collection.id)),1395		)1396	}13971398	/// Tells whether the given `owner` approves the `operator`.1399	pub fn allowance_for_all(1400		collection: &NonfungibleHandle<T>,1401		owner: &T::CrossAccountId,1402		operator: &T::CrossAccountId,1403	) -> bool {1404		<CollectionAllowance<T>>::get((collection.id, owner, operator))1405	}14061407	pub fn repair_item(collection: &NonfungibleHandle<T>, token: TokenId) -> DispatchResult {1408		<TokenProperties<T>>::mutate((collection.id, token), |properties| {1409			properties.recompute_consumed_space();1410		});14111412		Ok(())1413	}1414}
after · pallets/nonfungible/src/lib.rs
1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617//! # Nonfungible Pallet18//!19//! The Nonfungible pallet provides functionality for handling nonfungible collections and tokens.20//!21//! - [`Config`]22//! - [`NonfungibleHandle`]23//! - [`Pallet`]24//! - [`CommonWeights`](common::CommonWeights)25//!26//! ## Overview27//!28//! The Nonfungible pallet provides functions for:29//!30//! - NFT collection creation and removal31//! - Minting and burning of NFT tokens32//! - Retrieving account balances33//! - Transfering NFT tokens34//! - Setting and checking allowance for NFT tokens35//! - Setting properties and permissions for NFT collections and tokens36//! - Nesting and unnesting tokens37//!38//! ### Terminology39//!40//! - **NFT token:** Non fungible token.41//!42//! - **NFT Collection:** A collection of NFT tokens. All NFT tokens are part of a collection.43//!   Each collection can define it's own properties, properties for it's tokens and set of permissions.44//!45//! - **Balance:** Number of NFT tokens owned by an account46//!47//! - **Allowance:** NFT tokens owned by one account that another account is allowed to make operations on48//!49//! - **Burning:** The process of “deleting” a token from a collection and from50//!   an account balance of the owner.51//!52//! - **Nesting:** Setting up parent-child relationship between tokens. Nested tokens are inhereting53//!   owner from their parent. There could be multiple levels of nesting. Token couldn't be nested in54//!   it's child token i.e. parent-child relationship graph shouldn't have cycles.55//!56//! - **Properties:** Key-Values pairs. Token properties are attached to a token. Collection properties are57//!   attached to a collection. Set of permissions could be defined for each property.58//!59//! ### Implementations60//!61//! The Nonfungible pallet provides implementations for the following traits. If these traits provide62//! the functionality that you need, then you can avoid coupling with the Nonfungible pallet.63//!64//! - [`CommonWeightInfo`](pallet_common::CommonWeightInfo): Functions for retrieval of transaction weight65//! - [`CommonCollectionOperations`](pallet_common::CommonCollectionOperations): Functions for dealing66//!   with collections67//!68//! ## Interface69//!70//! ### Dispatchable Functions71//!72//! - `init_collection` - Create NFT collection. NFT collection can be configured to allow or deny access for73//!   some accounts.74//! - `destroy_collection` - Destroy exising NFT collection. There should be no tokens in the collection.75//! - `burn` - Burn NFT token owned by account.76//! - `transfer` - Transfer NFT token. Transfers should be enabled for NFT collection.77//!   Nests the NFT token if it is sent to another token.78//! - `create_item` - Mint NFT token in collection. Sender should have permission to mint tokens.79//! - `set_allowance` - Set allowance for another account.80//! - `set_token_property` - Set token property value.81//! - `delete_token_property` - Remove property from the token.82//! - `set_collection_properties` - Set collection properties.83//! - `delete_collection_properties` - Remove properties from the collection.84//! - `set_property_permission` - Set collection property permission.85//! - `set_token_property_permissions` - Set token property permissions.86//!87//! ## Assumptions88//!89//! * To perform operations on tokens sender should be in collection's allow list if collection access mode is `AllowList`.9091#![cfg_attr(not(feature = "std"), no_std)]9293use erc::ERC721Events;94use evm_coder::ToLog;95use frame_support::{96	BoundedVec, ensure, fail, transactional,97	storage::with_transaction,98	pallet_prelude::DispatchResultWithPostInfo,99	pallet_prelude::Weight,100	dispatch::{PostDispatchInfo, Pays},101};102use up_data_structs::{103	AccessMode, CollectionId, CollectionFlags, CustomDataLimit, TokenId, CreateCollectionData,104	CreateNftExData, mapping::TokenAddressMapping, budget::Budget, Property, PropertyKey,105	PropertyValue, PropertyKeyPermission, PropertyScope, TrySetProperty, TokenChild,106	AuxPropertyValue, PropertiesPermissionMap, TokenProperties as TokenPropertiesT,107};108use pallet_evm::{account::CrossAccountId, Pallet as PalletEvm};109use pallet_common::{110	Error as CommonError, Pallet as PalletCommon, Event as CommonEvent, CollectionHandle,111	eth::collection_id_to_address, SelfWeightOf as PalletCommonWeightOf,112	weights::WeightInfo as CommonWeightInfo, helpers::add_weight_to_post_info,113};114use pallet_structure::{Pallet as PalletStructure, Error as StructureError};115use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};116use sp_core::{Get, H160};117use sp_runtime::{ArithmeticError, DispatchError, DispatchResult, TransactionOutcome};118use sp_std::{vec::Vec, vec, collections::btree_map::BTreeMap};119use core::ops::Deref;120use codec::{Encode, Decode, MaxEncodedLen};121use scale_info::TypeInfo;122123pub use pallet::*;124use weights::WeightInfo;125#[cfg(feature = "runtime-benchmarks")]126pub mod benchmarking;127pub mod common;128pub mod erc;129pub mod weights;130131pub type CreateItemData<T> = CreateNftExData<<T as pallet_evm::Config>::CrossAccountId>;132pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;133134/// Token data, stored independently from other data used to describe it135/// for the convenience of database access. Notably contains the owner account address.136#[struct_versioning::versioned(version = 2, upper)]137#[derive(Encode, Decode, TypeInfo, MaxEncodedLen)]138pub struct ItemData<CrossAccountId> {139	#[version(..2)]140	pub const_data: BoundedVec<u8, CustomDataLimit>,141142	#[version(..2)]143	pub variable_data: BoundedVec<u8, CustomDataLimit>,144145	pub owner: CrossAccountId,146}147148#[frame_support::pallet]149pub mod pallet {150	use super::*;151	use frame_support::{152		Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key, traits::StorageVersion,153	};154	use up_data_structs::{CollectionId, TokenId};155	use super::weights::WeightInfo;156157	#[pallet::error]158	pub enum Error<T> {159		/// Not Nonfungible item data used to mint in Nonfungible collection.160		NotNonfungibleDataUsedToMintFungibleCollectionToken,161		/// Used amount > 1 with NFT162		NonfungibleItemsHaveNoAmount,163		/// Unable to burn NFT with children164		CantBurnNftWithChildren,165	}166167	#[pallet::config]168	pub trait Config:169		frame_system::Config + pallet_common::Config + pallet_structure::Config + pallet_evm::Config170	{171		type WeightInfo: WeightInfo;172	}173174	const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);175176	#[pallet::pallet]177	#[pallet::storage_version(STORAGE_VERSION)]178	pub struct Pallet<T>(_);179180	/// Total amount of minted tokens in a collection.181	#[pallet::storage]182	pub type TokensMinted<T: Config> =183		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;184185	/// Amount of burnt tokens in a collection.186	#[pallet::storage]187	pub type TokensBurnt<T: Config> =188		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;189190	/// Token data, used to partially describe a token.191	#[pallet::storage]192	pub type TokenData<T: Config> = StorageNMap<193		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),194		Value = ItemData<T::CrossAccountId>,195		QueryKind = OptionQuery,196	>;197198	/// Map of key-value pairs, describing the metadata of a token.199	#[pallet::storage]200	#[pallet::getter(fn token_properties)]201	pub type TokenProperties<T: Config> = StorageNMap<202		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),203		Value = TokenPropertiesT,204		QueryKind = ValueQuery,205	>;206207	/// Custom data of a token that is serialized to bytes,208	/// primarily reserved for on-chain operations,209	/// normally obscured from the external users.210	///211	/// Auxiliary properties are slightly different from212	/// usual [`TokenProperties`] due to an unlimited number213	/// and separately stored and written-to key-value pairs.214	///215	/// Currently unused.216	#[pallet::storage]217	#[pallet::getter(fn token_aux_property)]218	pub type TokenAuxProperties<T: Config> = StorageNMap<219		Key = (220			Key<Twox64Concat, CollectionId>,221			Key<Twox64Concat, TokenId>,222			Key<Twox64Concat, PropertyScope>,223			Key<Twox64Concat, PropertyKey>,224		),225		Value = AuxPropertyValue,226		QueryKind = OptionQuery,227	>;228229	/// Used to enumerate tokens owned by account.230	#[pallet::storage]231	pub type Owned<T: Config> = StorageNMap<232		Key = (233			Key<Twox64Concat, CollectionId>,234			Key<Blake2_128Concat, T::CrossAccountId>,235			Key<Twox64Concat, TokenId>,236		),237		Value = bool,238		QueryKind = ValueQuery,239	>;240241	/// Used to enumerate token's children.242	#[pallet::storage]243	#[pallet::getter(fn token_children)]244	pub type TokenChildren<T: Config> = StorageNMap<245		Key = (246			Key<Twox64Concat, CollectionId>,247			Key<Twox64Concat, TokenId>,248			Key<Twox64Concat, (CollectionId, TokenId)>,249		),250		Value = bool,251		QueryKind = ValueQuery,252	>;253254	/// Amount of tokens owned by an account in a collection.255	#[pallet::storage]256	pub type AccountBalance<T: Config> = StorageNMap<257		Key = (258			Key<Twox64Concat, CollectionId>,259			Key<Blake2_128Concat, T::CrossAccountId>,260		),261		Value = u32,262		QueryKind = ValueQuery,263	>;264265	/// Allowance set by a token owner for another user to perform one of certain transactions on a token.266	#[pallet::storage]267	pub type Allowance<T: Config> = StorageNMap<268		Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),269		Value = T::CrossAccountId,270		QueryKind = OptionQuery,271	>;272273	/// Operator set by a wallet owner that could perform certain transactions on all tokens in the wallet.274	#[pallet::storage]275	pub type CollectionAllowance<T: Config> = StorageNMap<276		Key = (277			Key<Twox64Concat, CollectionId>,278			Key<Blake2_128Concat, T::CrossAccountId>,279			Key<Blake2_128Concat, T::CrossAccountId>,280		),281		Value = bool,282		QueryKind = ValueQuery,283	>;284285	#[pallet::genesis_config]286	pub struct GenesisConfig<T>(PhantomData<T>);287288	#[cfg(feature = "std")]289	impl<T: Config> Default for GenesisConfig<T> {290		fn default() -> Self {291			Self(Default::default())292		}293	}294295	#[pallet::genesis_build]296	impl<T: Config> GenesisBuild<T> for GenesisConfig<T> {297		fn build(&self) {298			StorageVersion::new(1).put::<Pallet<T>>();299		}300	}301}302303pub struct NonfungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);304impl<T: Config> NonfungibleHandle<T> {305	pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {306		Self(inner)307	}308	pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {309		self.0310	}311	pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {312		&mut self.0313	}314}315316impl<T: Config> WithRecorder<T> for NonfungibleHandle<T> {317	fn recorder(&self) -> &SubstrateRecorder<T> {318		self.0.recorder()319	}320	fn into_recorder(self) -> SubstrateRecorder<T> {321		self.0.into_recorder()322	}323}324impl<T: Config> Deref for NonfungibleHandle<T> {325	type Target = pallet_common::CollectionHandle<T>;326327	fn deref(&self) -> &Self::Target {328		&self.0329	}330}331332impl<T: Config> Pallet<T> {333	/// Get number of NFT tokens in collection.334	pub fn total_supply(collection: &NonfungibleHandle<T>) -> u32 {335		<TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)336	}337338	/// Check that NFT token exists.339	///340	/// - `token`: Token ID.341	pub fn token_exists(collection: &NonfungibleHandle<T>, token: TokenId) -> bool {342		<TokenData<T>>::contains_key((collection.id, token))343	}344345	/// Set the token property with the scope.346	///347	/// - `property`: Contains key-value pair.348	pub fn set_scoped_token_property(349		collection_id: CollectionId,350		token_id: TokenId,351		scope: PropertyScope,352		property: Property,353	) -> DispatchResult {354		TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {355			properties.try_scoped_set(scope, property.key, property.value)356		})357		.map_err(<CommonError<T>>::from)?;358359		Ok(())360	}361362	/// Batch operation to set multiple properties with the same scope.363	pub fn set_scoped_token_properties(364		collection_id: CollectionId,365		token_id: TokenId,366		scope: PropertyScope,367		properties: impl Iterator<Item = Property>,368	) -> DispatchResult {369		TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {370			stored_properties.try_scoped_set_from_iter(scope, properties)371		})372		.map_err(<CommonError<T>>::from)?;373374		Ok(())375	}376377	/// Add or edit auxiliary data for the property.378	///379	/// - `f`: function that adds or edits auxiliary data.380	pub fn try_mutate_token_aux_property<R, E>(381		collection_id: CollectionId,382		token_id: TokenId,383		scope: PropertyScope,384		key: PropertyKey,385		f: impl FnOnce(&mut Option<AuxPropertyValue>) -> Result<R, E>,386	) -> Result<R, E> {387		<TokenAuxProperties<T>>::try_mutate((collection_id, token_id, scope, key), f)388	}389390	/// Remove auxiliary data for the property.391	pub fn remove_token_aux_property(392		collection_id: CollectionId,393		token_id: TokenId,394		scope: PropertyScope,395		key: PropertyKey,396	) {397		<TokenAuxProperties<T>>::remove((collection_id, token_id, scope, key));398	}399400	/// Get all auxiliary data in a given scope.401	///402	/// Returns iterator over Property Key - Data pairs.403	pub fn iterate_token_aux_properties(404		collection_id: CollectionId,405		token_id: TokenId,406		scope: PropertyScope,407	) -> impl Iterator<Item = (PropertyKey, AuxPropertyValue)> {408		<TokenAuxProperties<T>>::iter_prefix((collection_id, token_id, scope))409	}410411	/// Get ID of the last minted token412	pub fn current_token_id(collection_id: CollectionId) -> TokenId {413		TokenId(<TokensMinted<T>>::get(collection_id))414	}415}416417// unchecked calls skips any permission checks418impl<T: Config> Pallet<T> {419	/// Create NFT collection420	///421	/// `init_collection` will take non-refundable deposit for collection creation.422	///423	/// - `data`: Contains settings for collection limits and permissions.424	pub fn init_collection(425		owner: T::CrossAccountId,426		payer: T::CrossAccountId,427		data: CreateCollectionData<T::AccountId>,428		flags: CollectionFlags,429	) -> Result<CollectionId, DispatchError> {430		<PalletCommon<T>>::init_collection(owner, payer, data, flags)431	}432433	/// Destroy NFT collection434	///435	/// `destroy_collection` will throw error if collection contains any tokens.436	/// Only owner can destroy collection.437	pub fn destroy_collection(438		collection: NonfungibleHandle<T>,439		sender: &T::CrossAccountId,440	) -> DispatchResult {441		let id = collection.id;442443		if Self::collection_has_tokens(id) {444			return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());445		}446447		// =========448449		PalletCommon::destroy_collection(collection.0, sender)?;450451		let _ = <TokenData<T>>::clear_prefix((id,), u32::MAX, None);452		let _ = <TokenChildren<T>>::clear_prefix((id,), u32::MAX, None);453		let _ = <Owned<T>>::clear_prefix((id,), u32::MAX, None);454		<TokensMinted<T>>::remove(id);455		<TokensBurnt<T>>::remove(id);456		let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);457		let _ = <AccountBalance<T>>::clear_prefix((id,), u32::MAX, None);458		let _ = <CollectionAllowance<T>>::clear_prefix((id,), u32::MAX, None);459		Ok(())460	}461462	/// Burn NFT token463	///464	/// `burn` removes `token` from the `collection`, from it's owner and from the parent token465	/// if the token is nested.466	/// Only the owner can `burn` the token. The `token` shouldn't have any nested tokens.467	/// Also removes all corresponding properties and auxiliary properties.468	///469	/// - `token`: Token that should be burned470	/// - `collection`: Collection that contains the token471	pub fn burn(472		collection: &NonfungibleHandle<T>,473		sender: &T::CrossAccountId,474		token: TokenId,475	) -> DispatchResult {476		let token_data =477			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;478		ensure!(&token_data.owner == sender, <CommonError<T>>::NoPermission);479480		if collection.permissions.access() == AccessMode::AllowList {481			collection.check_allowlist(sender)?;482		}483484		if Self::token_has_children(collection.id, token) {485			return Err(<Error<T>>::CantBurnNftWithChildren.into());486		}487488		let burnt = <TokensBurnt<T>>::get(collection.id)489			.checked_add(1)490			.ok_or(ArithmeticError::Overflow)?;491492		let balance = <AccountBalance<T>>::get((collection.id, token_data.owner.clone()))493			.checked_sub(1)494			.ok_or(ArithmeticError::Overflow)?;495496		// =========497498		if balance == 0 {499			<AccountBalance<T>>::remove((collection.id, token_data.owner.clone()));500		} else {501			<AccountBalance<T>>::insert((collection.id, token_data.owner.clone()), balance);502		}503504		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);505506		<Owned<T>>::remove((collection.id, &token_data.owner, token));507		<TokensBurnt<T>>::insert(collection.id, burnt);508		<TokenData<T>>::remove((collection.id, token));509		<TokenProperties<T>>::remove((collection.id, token));510		let _ = <TokenAuxProperties<T>>::clear_prefix((collection.id, token), u32::MAX, None);511		let old_spender = <Allowance<T>>::take((collection.id, token));512513		if let Some(old_spender) = old_spender {514			<PalletCommon<T>>::deposit_event(CommonEvent::Approved(515				collection.id,516				token,517				token_data.owner.clone(),518				old_spender,519				0,520			));521		}522523		<PalletEvm<T>>::deposit_log(524			ERC721Events::Transfer {525				from: *token_data.owner.as_eth(),526				to: H160::default(),527				token_id: token.into(),528			}529			.to_log(collection_id_to_address(collection.id)),530		);531		<PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(532			collection.id,533			token,534			token_data.owner,535			1,536		));537		Ok(())538	}539540	/// Same as [`burn`] but burns all the tokens that are nested in the token first541	///542	/// - `self_budget`: Limit for searching children in depth.543	/// - `breadth_budget`: Limit of breadth of searching children.544	///545	/// [`burn`]: struct.Pallet.html#method.burn546	#[transactional]547	pub fn burn_recursively(548		collection: &NonfungibleHandle<T>,549		sender: &T::CrossAccountId,550		token: TokenId,551		self_budget: &dyn Budget,552		breadth_budget: &dyn Budget,553	) -> DispatchResultWithPostInfo {554		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);555556		let current_token_account =557			T::CrossTokenAddressMapping::token_to_address(collection.id, token);558559		let mut weight = Weight::zero();560561		// This method is transactional, if user in fact doesn't have permissions to remove token -562		// tokens removed here will be restored after rejected transaction563		for ((collection, token), _) in <TokenChildren<T>>::iter_prefix((collection.id, token)) {564			ensure!(breadth_budget.consume(), <StructureError<T>>::BreadthLimit,);565			let PostDispatchInfo { actual_weight, .. } =566				<PalletStructure<T>>::burn_item_recursively(567					current_token_account.clone(),568					collection,569					token,570					self_budget,571					breadth_budget,572				)?;573			if let Some(actual_weight) = actual_weight {574				weight = weight.saturating_add(actual_weight);575			}576		}577578		Self::burn(collection, sender, token)?;579		DispatchResultWithPostInfo::Ok(PostDispatchInfo {580			actual_weight: Some(weight + <SelfWeightOf<T>>::burn_item()),581			pays_fee: Pays::Yes,582		})583	}584585	/// A batch operation to add, edit or remove properties for a token.586	///587	/// - `nesting_budget`: Limit for searching parents in-depth to check ownership.588	/// - `is_token_create`: Indicates that method is called during token initialization.589	///   Allows to bypass ownership check.590	///591	/// All affected properties should have `mutable` permission592	/// to be **deleted** or to be **set more than once**,593	/// and the sender should have permission to edit those properties.594	///595	/// This function fires an event for each property change.596	/// In case of an error, all the changes (including the events) will be reverted597	/// since the function is transactional.598	#[transactional]599	fn modify_token_properties(600		collection: &NonfungibleHandle<T>,601		sender: &T::CrossAccountId,602		token_id: TokenId,603		properties_updates: impl Iterator<Item = (PropertyKey, Option<PropertyValue>)>,604		is_token_create: bool,605		nesting_budget: &dyn Budget,606	) -> DispatchResult {607		let is_token_owner = || {608			let is_owned = <PalletStructure<T>>::check_indirectly_owned(609				sender.clone(),610				collection.id,611				token_id,612				None,613				nesting_budget,614			)?;615616			Ok(is_owned)617		};618619		let stored_properties = <TokenProperties<T>>::get((collection.id, token_id));620621		<PalletCommon<T>>::modify_token_properties(622			collection,623			sender,624			token_id,625			properties_updates,626			is_token_create,627			stored_properties,628			is_token_owner,629			|properties| <TokenProperties<T>>::set((collection.id, token_id), properties),630			erc::ERC721TokenEvent::TokenChanged {631				token_id: token_id.into(),632			}633			.to_log(T::ContractAddress::get()),634		)635	}636637	/// Batch operation to add or edit properties for the token638	///639	/// Same as [`modify_token_properties`] but doesn't allow to remove properties640	///641	/// [`modify_token_properties`]: struct.Pallet.html#method.modify_token_properties642	pub fn set_token_properties(643		collection: &NonfungibleHandle<T>,644		sender: &T::CrossAccountId,645		token_id: TokenId,646		properties: impl Iterator<Item = Property>,647		is_token_create: bool,648		nesting_budget: &dyn Budget,649	) -> DispatchResult {650		Self::modify_token_properties(651			collection,652			sender,653			token_id,654			properties.map(|p| (p.key, Some(p.value))),655			is_token_create,656			nesting_budget,657		)658	}659660	/// Add or edit single property for the token661	///662	/// Calls [`set_token_properties`] internally663	///664	/// [`set_token_properties`]: struct.Pallet.html#method.set_token_properties665	pub fn set_token_property(666		collection: &NonfungibleHandle<T>,667		sender: &T::CrossAccountId,668		token_id: TokenId,669		property: Property,670		nesting_budget: &dyn Budget,671	) -> DispatchResult {672		let is_token_create = false;673674		Self::set_token_properties(675			collection,676			sender,677			token_id,678			[property].into_iter(),679			is_token_create,680			nesting_budget,681		)682	}683684	/// Batch operation to remove properties from the token685	///686	/// Same as [`modify_token_properties`] but doesn't allow to add or edit properties687	///688	/// [`modify_token_properties`]: struct.Pallet.html#method.modify_token_properties689	pub fn delete_token_properties(690		collection: &NonfungibleHandle<T>,691		sender: &T::CrossAccountId,692		token_id: TokenId,693		property_keys: impl Iterator<Item = PropertyKey>,694		nesting_budget: &dyn Budget,695	) -> DispatchResult {696		let is_token_create = false;697698		Self::modify_token_properties(699			collection,700			sender,701			token_id,702			property_keys.into_iter().map(|key| (key, None)),703			is_token_create,704			nesting_budget,705		)706	}707708	/// Remove single property from the token709	///710	/// Calls [`delete_token_properties`] internally711	///712	/// [`delete_token_properties`]: struct.Pallet.html#method.delete_token_properties713	pub fn delete_token_property(714		collection: &NonfungibleHandle<T>,715		sender: &T::CrossAccountId,716		token_id: TokenId,717		property_key: PropertyKey,718		nesting_budget: &dyn Budget,719	) -> DispatchResult {720		Self::delete_token_properties(721			collection,722			sender,723			token_id,724			[property_key].into_iter(),725			nesting_budget,726		)727	}728729	/// Add or edit properties for the collection730	pub fn set_collection_properties(731		collection: &NonfungibleHandle<T>,732		sender: &T::CrossAccountId,733		properties: Vec<Property>,734	) -> DispatchResult {735		<PalletCommon<T>>::set_collection_properties(collection, sender, properties.into_iter())736	}737738	/// Remove properties from the collection739	pub fn delete_collection_properties(740		collection: &CollectionHandle<T>,741		sender: &T::CrossAccountId,742		property_keys: Vec<PropertyKey>,743	) -> DispatchResult {744		<PalletCommon<T>>::delete_collection_properties(745			collection,746			sender,747			property_keys.into_iter(),748		)749	}750751	/// Set property permissions for the token.752	///753	/// Sender should be the owner or admin of token's collection.754	pub fn set_token_property_permissions(755		collection: &CollectionHandle<T>,756		sender: &T::CrossAccountId,757		property_permissions: Vec<PropertyKeyPermission>,758	) -> DispatchResult {759		<PalletCommon<T>>::set_token_property_permissions(collection, sender, property_permissions)760	}761762	/// Set property permissions for the token with scope.763	///764	/// Sender should be the owner or admin of token's collection.765	pub fn set_scoped_token_property_permissions(766		collection: &CollectionHandle<T>,767		sender: &T::CrossAccountId,768		scope: PropertyScope,769		property_permissions: Vec<PropertyKeyPermission>,770	) -> DispatchResult {771		<PalletCommon<T>>::set_scoped_token_property_permissions(772			collection,773			sender,774			scope,775			property_permissions,776		)777	}778779	pub fn token_property_permission(collection_id: CollectionId) -> PropertiesPermissionMap {780		<PalletCommon<T>>::property_permissions(collection_id)781	}782783	pub fn check_token_immediate_ownership(784		collection: &NonfungibleHandle<T>,785		token: TokenId,786		possible_owner: &T::CrossAccountId,787	) -> DispatchResult {788		let token_data =789			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;790		ensure!(791			&token_data.owner == possible_owner,792			<CommonError<T>>::NoPermission793		);794		Ok(())795	}796797	/// Transfer NFT token from one account to another.798	///799	/// `from` account stops being the owner and `to` account becomes the owner of the token.800	/// If `to` is token than `to` becomes owner of the token and the token become nested.801	/// Unnests token from previous parent if it was nested before.802	/// Removes allowance for the token if there was any.803	/// Throws if transfers aren't allowed for collection or if receiver reached token ownership limit.804	///805	/// - `nesting_budget`: Limit for token nesting depth806	pub fn transfer(807		collection: &NonfungibleHandle<T>,808		from: &T::CrossAccountId,809		to: &T::CrossAccountId,810		token: TokenId,811		nesting_budget: &dyn Budget,812	) -> DispatchResultWithPostInfo {813		ensure!(814			collection.limits.transfers_enabled(),815			<CommonError<T>>::TransferNotAllowed816		);817818		let mut actual_weight = <SelfWeightOf<T>>::transfer_raw();819		let token_data =820			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;821		ensure!(&token_data.owner == from, <CommonError<T>>::NoPermission);822823		if collection.permissions.access() == AccessMode::AllowList {824			collection.check_allowlist(from)?;825			collection.check_allowlist(to)?;826			actual_weight += <PalletCommonWeightOf<T>>::check_accesslist() * 2;827		}828		<PalletCommon<T>>::ensure_correct_receiver(to)?;829830		let balance_from = <AccountBalance<T>>::get((collection.id, from))831			.checked_sub(1)832			.ok_or(<CommonError<T>>::TokenValueTooLow)?;833		let balance_to = if from != to {834			let balance_to = <AccountBalance<T>>::get((collection.id, to))835				.checked_add(1)836				.ok_or(ArithmeticError::Overflow)?;837838			ensure!(839				balance_to < collection.limits.account_token_ownership_limit(),840				<CommonError<T>>::AccountTokenLimitExceeded,841			);842843			Some(balance_to)844		} else {845			None846		};847848		<PalletStructure<T>>::nest_if_sent_to_token(849			from.clone(),850			to,851			collection.id,852			token,853			nesting_budget,854		)?;855856		// =========857858		<PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);859860		<TokenData<T>>::insert(861			(collection.id, token),862			ItemData {863				owner: to.clone(),864				..token_data865			},866		);867868		if let Some(balance_to) = balance_to {869			// from != to870			if balance_from == 0 {871				<AccountBalance<T>>::remove((collection.id, from));872			} else {873				<AccountBalance<T>>::insert((collection.id, from), balance_from);874			}875			<AccountBalance<T>>::insert((collection.id, to), balance_to);876			<Owned<T>>::remove((collection.id, from, token));877			<Owned<T>>::insert((collection.id, to, token), true);878		}879		Self::set_allowance_unchecked(collection, from, token, None, true);880881		<PalletEvm<T>>::deposit_log(882			ERC721Events::Transfer {883				from: *from.as_eth(),884				to: *to.as_eth(),885				token_id: token.into(),886			}887			.to_log(collection_id_to_address(collection.id)),888		);889		<PalletCommon<T>>::deposit_event(CommonEvent::Transfer(890			collection.id,891			token,892			from.clone(),893			to.clone(),894			1,895		));896897		Ok(PostDispatchInfo {898			actual_weight: Some(actual_weight),899			pays_fee: Pays::Yes,900		})901	}902903	/// Batch operation to mint multiple NFT tokens.904	///905	/// The sender should be the owner/admin of the collection or collection should be configured906	/// to allow public minting.907	/// Throws if amount of tokens reached it's limit for the collection or if caller reached908	/// token ownership limit.909	///910	/// - `data`: Contains list of token properties and users who will become the owners of the911	///   corresponging tokens.912	/// - `nesting_budget`: Limit for token nesting depth913	pub fn create_multiple_items(914		collection: &NonfungibleHandle<T>,915		sender: &T::CrossAccountId,916		data: Vec<CreateItemData<T>>,917		nesting_budget: &dyn Budget,918	) -> DispatchResult {919		if !collection.is_owner_or_admin(sender) {920			ensure!(921				collection.permissions.mint_mode(),922				<CommonError<T>>::PublicMintingNotAllowed923			);924			collection.check_allowlist(sender)?;925926			for item in data.iter() {927				collection.check_allowlist(&item.owner)?;928			}929		}930931		for data in data.iter() {932			<PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;933		}934935		let first_token = <TokensMinted<T>>::get(collection.id);936		let tokens_minted = first_token937			.checked_add(data.len() as u32)938			.ok_or(ArithmeticError::Overflow)?;939		ensure!(940			tokens_minted <= collection.limits.token_limit(),941			<CommonError<T>>::CollectionTokenLimitExceeded942		);943944		let mut balances = BTreeMap::new();945		for data in &data {946			let balance = balances947				.entry(&data.owner)948				.or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));949			*balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;950951			ensure!(952				*balance <= collection.limits.account_token_ownership_limit(),953				<CommonError<T>>::AccountTokenLimitExceeded,954			);955		}956957		for (i, data) in data.iter().enumerate() {958			let token = TokenId(first_token + i as u32 + 1);959960			<PalletStructure<T>>::check_nesting(961				sender.clone(),962				&data.owner,963				collection.id,964				token,965				nesting_budget,966			)?;967		}968969		// =========970971		with_transaction(|| {972			for (i, data) in data.iter().enumerate() {973				let token = first_token + i as u32 + 1;974975				<TokenData<T>>::insert(976					(collection.id, token),977					ItemData {978						// const_data: data.const_data.clone(),979						owner: data.owner.clone(),980					},981				);982983				<PalletStructure<T>>::nest_if_sent_to_token_unchecked(984					&data.owner,985					collection.id,986					TokenId(token),987				);988989				if let Err(e) = Self::set_token_properties(990					collection,991					sender,992					TokenId(token),993					data.properties.clone().into_iter(),994					true,995					nesting_budget,996				) {997					return TransactionOutcome::Rollback(Err(e));998				}999			}1000			TransactionOutcome::Commit(Ok(()))1001		})?;10021003		<TokensMinted<T>>::insert(collection.id, tokens_minted);1004		for (account, balance) in balances {1005			<AccountBalance<T>>::insert((collection.id, account), balance);1006		}1007		for (i, data) in data.into_iter().enumerate() {1008			let token = first_token + i as u32 + 1;1009			<Owned<T>>::insert((collection.id, &data.owner, token), true);10101011			<PalletEvm<T>>::deposit_log(1012				ERC721Events::Transfer {1013					from: H160::default(),1014					to: *data.owner.as_eth(),1015					token_id: token.into(),1016				}1017				.to_log(collection_id_to_address(collection.id)),1018			);1019			<PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(1020				collection.id,1021				TokenId(token),1022				data.owner.clone(),1023				1,1024			));1025		}1026		Ok(())1027	}10281029	pub fn set_allowance_unchecked(1030		collection: &NonfungibleHandle<T>,1031		sender: &T::CrossAccountId,1032		token: TokenId,1033		spender: Option<&T::CrossAccountId>,1034		assume_implicit_eth: bool,1035	) {1036		if let Some(spender) = spender {1037			let old_spender = <Allowance<T>>::get((collection.id, token));1038			<Allowance<T>>::insert((collection.id, token), spender);1039			// In ERC721 there is only one possible approved user of token, so we set1040			// approved user to spender1041			<PalletEvm<T>>::deposit_log(1042				ERC721Events::Approval {1043					owner: *sender.as_eth(),1044					approved: *spender.as_eth(),1045					token_id: token.into(),1046				}1047				.to_log(collection_id_to_address(collection.id)),1048			);1049			// In Unique chain, any token can have any amount of approved users, so we need to1050			// set allowance of old owner to 0, and allowance of new owner to 11051			if old_spender.as_ref() != Some(spender) {1052				if let Some(old_owner) = old_spender {1053					<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1054						collection.id,1055						token,1056						sender.clone(),1057						old_owner,1058						0,1059					));1060				}1061				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1062					collection.id,1063					token,1064					sender.clone(),1065					spender.clone(),1066					1,1067				));1068			}1069		} else {1070			let old_spender = <Allowance<T>>::take((collection.id, token));1071			if !assume_implicit_eth {1072				// In ERC721 there is only one possible approved user of token, so we set1073				// approved user to zero address1074				<PalletEvm<T>>::deposit_log(1075					ERC721Events::Approval {1076						owner: *sender.as_eth(),1077						approved: H160::default(),1078						token_id: token.into(),1079					}1080					.to_log(collection_id_to_address(collection.id)),1081				);1082			}1083			// In Unique chain, any token can have any amount of approved users, so we need to1084			// set allowance of old owner to 01085			if let Some(old_spender) = old_spender {1086				<PalletCommon<T>>::deposit_event(CommonEvent::Approved(1087					collection.id,1088					token,1089					sender.clone(),1090					old_spender,1091					0,1092				));1093			}1094		}1095	}10961097	pub fn get_allowance(1098		collection: &NonfungibleHandle<T>,1099		token_id: TokenId,1100	) -> Result<Option<T::CrossAccountId>, DispatchError> {1101		ensure!(1102			<TokenData<T>>::get((collection.id, token_id)).is_some(),1103			<CommonError<T>>::TokenNotFound1104		);1105		Ok(<Allowance<T>>::get((collection.id, token_id)))1106	}11071108	/// Set allowance for the spender to `transfer` or `burn` sender's token.1109	///1110	/// - `token`: Token the spender is allowed to `transfer` or `burn`.1111	pub fn set_allowance(1112		collection: &NonfungibleHandle<T>,1113		sender: &T::CrossAccountId,1114		token: TokenId,1115		spender: Option<&T::CrossAccountId>,1116	) -> DispatchResult {1117		if collection.permissions.access() == AccessMode::AllowList {1118			collection.check_allowlist(sender)?;1119			if let Some(spender) = spender {1120				collection.check_allowlist(spender)?;1121			}1122		}11231124		if let Some(spender) = spender {1125			<PalletCommon<T>>::ensure_correct_receiver(spender)?;1126		}11271128		let token_data =1129			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;1130		if &token_data.owner != sender {1131			ensure!(1132				collection.ignores_owned_amount(sender),1133				<CommonError<T>>::CantApproveMoreThanOwned1134			);1135		}11361137		// =========11381139		Self::set_allowance_unchecked(collection, sender, token, spender, false);1140		Ok(())1141	}11421143	/// Set allowance for the spender to `transfer` or `burn` sender's token from eth mirror.1144	///1145	/// - `from`: Address of sender's eth mirror.1146	/// - `to`: Adress of spender.1147	/// - `token`: Token the spender is allowed to `transfer` or `burn`.1148	pub fn set_allowance_from(1149		collection: &NonfungibleHandle<T>,1150		sender: &T::CrossAccountId,1151		from: &T::CrossAccountId,1152		token: TokenId,1153		to: Option<&T::CrossAccountId>,1154	) -> DispatchResult {1155		if collection.permissions.access() == AccessMode::AllowList {1156			collection.check_allowlist(sender)?;1157			collection.check_allowlist(from)?;1158			if let Some(to) = to {1159				collection.check_allowlist(to)?;1160			}1161		}11621163		if let Some(to) = to {1164			<PalletCommon<T>>::ensure_correct_receiver(to)?;1165		}11661167		ensure!(1168			sender.conv_eq(from),1169			<CommonError<T>>::AddressIsNotEthMirror1170		);11711172		let token_data =1173			<TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;1174		if token_data.owner != *from {1175			ensure!(1176				collection.limits.owner_can_transfer()1177					&& (collection.is_owner_or_admin(sender) || collection.is_owner_or_admin(from)),1178				<CommonError<T>>::CantApproveMoreThanOwned1179			);1180		}11811182		// =========11831184		Self::set_allowance_unchecked(collection, from, token, to, false);1185		Ok(())1186	}11871188	/// Checks allowance for the spender to use the token.1189	fn check_allowed(1190		collection: &NonfungibleHandle<T>,1191		spender: &T::CrossAccountId,1192		from: &T::CrossAccountId,1193		token: TokenId,1194		nesting_budget: &dyn Budget,1195	) -> DispatchResult {1196		if spender.conv_eq(from) {1197			return Ok(());1198		}1199		if collection.permissions.access() == AccessMode::AllowList {1200			// `from`, `to` checked in [`transfer`]1201			collection.check_allowlist(spender)?;1202		}12031204		if collection.ignores_token_restrictions(spender) {1205			return Ok(());1206		}12071208		if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {1209			ensure!(1210				<PalletStructure<T>>::check_indirectly_owned(1211					spender.clone(),1212					source.0,1213					source.1,1214					None,1215					nesting_budget1216				)?,1217				<CommonError<T>>::ApprovedValueTooLow,1218			);1219			return Ok(());1220		}1221		if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {1222			return Ok(());1223		}1224		if <CollectionAllowance<T>>::get((collection.id, from, spender)) {1225			return Ok(());1226		}12271228		Err(<CommonError<T>>::ApprovedValueTooLow.into())1229	}12301231	/// Transfer NFT token from one account to another.1232	///1233	/// Same as the [`transfer`] but spender doesn't needs to be the owner of the token.1234	/// The owner should set allowance for the spender to transfer token.1235	///1236	/// [`transfer`]: struct.Pallet.html#method.transfer1237	pub fn transfer_from(1238		collection: &NonfungibleHandle<T>,1239		spender: &T::CrossAccountId,1240		from: &T::CrossAccountId,1241		to: &T::CrossAccountId,1242		token: TokenId,1243		nesting_budget: &dyn Budget,1244	) -> DispatchResultWithPostInfo {1245		Self::check_allowed(collection, spender, from, token, nesting_budget)?;12461247		// =========12481249		// Allowance is reset in [`transfer`]1250		let mut result = Self::transfer(collection, from, to, token, nesting_budget);1251		add_weight_to_post_info(&mut result, <SelfWeightOf<T>>::check_allowed_raw());1252		result1253	}12541255	/// Burn NFT token for `from` account.1256	///1257	/// Same as the [`burn`] but spender doesn't need to be an owner of the token. The owner should1258	/// set allowance for the spender to burn token.1259	///1260	/// [`burn`]: struct.Pallet.html#method.burn1261	pub fn burn_from(1262		collection: &NonfungibleHandle<T>,1263		spender: &T::CrossAccountId,1264		from: &T::CrossAccountId,1265		token: TokenId,1266		nesting_budget: &dyn Budget,1267	) -> DispatchResult {1268		Self::check_allowed(collection, spender, from, token, nesting_budget)?;12691270		// =========12711272		Self::burn(collection, from, token)1273	}12741275	/// Check that `from` token could be nested in `under` token.1276	///1277	pub fn check_nesting(1278		handle: &NonfungibleHandle<T>,1279		sender: T::CrossAccountId,1280		from: (CollectionId, TokenId),1281		under: TokenId,1282		nesting_budget: &dyn Budget,1283	) -> DispatchResult {1284		let nesting = handle.permissions.nesting();12851286		#[cfg(not(feature = "runtime-benchmarks"))]1287		let permissive = false;1288		#[cfg(feature = "runtime-benchmarks")]1289		let permissive = nesting.permissive;12901291		if permissive {1292			ensure!(1293				<TokenData<T>>::contains_key((handle.id, under)),1294				<CommonError<T>>::TokenNotFound1295			);1296		} else if nesting.token_owner1297			&& <PalletStructure<T>>::check_indirectly_owned(1298				sender.clone(),1299				handle.id,1300				under,1301				Some(from),1302				nesting_budget,1303			)? {1304			// Pass, token existence and ouroboros checks are done in `check_indirectly_owned`1305		} else if nesting.collection_admin && handle.is_owner_or_admin(&sender) {1306			// token existence and ouroboros checks are done in `get_checked_topmost_owner`1307			let _ = <PalletStructure<T>>::get_checked_topmost_owner(1308				handle.id,1309				under,1310				Some(from),1311				nesting_budget,1312			)?1313			.ok_or(<CommonError<T>>::TokenNotFound)?;1314		} else {1315			fail!(<CommonError<T>>::UserIsNotAllowedToNest);1316		}13171318		if let Some(whitelist) = &nesting.restricted {1319			ensure!(1320				whitelist.contains(&from.0),1321				<CommonError<T>>::SourceCollectionIsNotAllowedToNest1322			);1323		}1324		Ok(())1325	}13261327	fn nest(under: (CollectionId, TokenId), to_nest: (CollectionId, TokenId)) {1328		<TokenChildren<T>>::insert((under.0, under.1, (to_nest.0, to_nest.1)), true);1329	}13301331	fn unnest(under: (CollectionId, TokenId), to_unnest: (CollectionId, TokenId)) {1332		<TokenChildren<T>>::remove((under.0, under.1, to_unnest));1333	}13341335	fn collection_has_tokens(collection_id: CollectionId) -> bool {1336		<TokenData<T>>::iter_prefix((collection_id,))1337			.next()1338			.is_some()1339	}13401341	fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {1342		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1343			.next()1344			.is_some()1345	}13461347	pub fn token_children_ids(collection_id: CollectionId, token_id: TokenId) -> Vec<TokenChild> {1348		<TokenChildren<T>>::iter_prefix((collection_id, token_id))1349			.map(|((child_collection_id, child_id), _)| TokenChild {1350				collection: child_collection_id,1351				token: child_id,1352			})1353			.collect()1354	}13551356	/// Mint single NFT token.1357	///1358	/// Delegated to [`create_multiple_items`]1359	///1360	/// [`create_multiple_items`]: struct.Pallet.html#method.create_multiple_items1361	pub fn create_item(1362		collection: &NonfungibleHandle<T>,1363		sender: &T::CrossAccountId,1364		data: CreateItemData<T>,1365		nesting_budget: &dyn Budget,1366	) -> DispatchResult {1367		Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1368	}13691370	/// Sets or unsets the approval of a given operator.1371	///1372	/// The `operator` is allowed to transfer all token pieces of the `owner` on their behalf.1373	/// - `owner`: Token owner1374	/// - `operator`: Operator1375	/// - `approve`: Should operator status be granted or revoked?1376	pub fn set_allowance_for_all(1377		collection: &NonfungibleHandle<T>,1378		owner: &T::CrossAccountId,1379		operator: &T::CrossAccountId,1380		approve: bool,1381	) -> DispatchResult {1382		<PalletCommon<T>>::set_allowance_for_all(1383			collection,1384			owner,1385			operator,1386			approve,1387			|| <CollectionAllowance<T>>::insert((collection.id, owner, operator), approve),1388			ERC721Events::ApprovalForAll {1389				owner: *owner.as_eth(),1390				operator: *operator.as_eth(),1391				approved: approve,1392			}1393			.to_log(collection_id_to_address(collection.id)),1394		)1395	}13961397	/// Tells whether the given `owner` approves the `operator`.1398	pub fn allowance_for_all(1399		collection: &NonfungibleHandle<T>,1400		owner: &T::CrossAccountId,1401		operator: &T::CrossAccountId,1402	) -> bool {1403		<CollectionAllowance<T>>::get((collection.id, owner, operator))1404	}14051406	pub fn repair_item(collection: &NonfungibleHandle<T>, token: TokenId) -> DispatchResult {1407		<TokenProperties<T>>::mutate((collection.id, token), |properties| {1408			properties.recompute_consumed_space();1409		});14101411		Ok(())1412	}1413}