difftreelog
Breaking: migration to `Fungible` traits in app-promotion pallet
in: master
Migration from `Currency` traits to `Fungible` ones in constraints for `Config::Currency` associated type.
6 files changed
Cargo.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",
pallets/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
pallets/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']
pallets/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() {
pallets/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;
pallets/nonfungible/src/lib.rsdiffbeforeafterboth1// 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}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}