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