difftreelog
fix destroy only not empty collections
in: master
4 files changed
pallets/common/src/lib.rsdiffbeforeafterboth--- a/pallets/common/src/lib.rs
+++ b/pallets/common/src/lib.rs
@@ -331,6 +331,8 @@
MustBeTokenOwner,
/// No permission to perform action
NoPermission,
+ /// Destroying only empty collections is allowed
+ CantDestroyNotEmptyCollection,
/// Collection is not in mint mode.
PublicMintingNotAllowed,
/// Address is not in allow list.
pallets/fungible/src/lib.rsdiffbeforeafterboth--- a/pallets/fungible/src/lib.rs
+++ b/pallets/fungible/src/lib.rs
@@ -145,6 +145,10 @@
) -> DispatchResult {
let id = collection.id;
+ if Self::collection_has_tokens(id) {
+ return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());
+ }
+
// =========
PalletCommon::destroy_collection(collection.0, sender)?;
@@ -155,6 +159,10 @@
Ok(())
}
+ fn collection_has_tokens(collection_id: CollectionId) -> bool {
+ <TotalSupply<T>>::get(collection_id) != 0
+ }
+
pub fn burn(
collection: &FungibleHandle<T>,
owner: &T::CrossAccountId,
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::{BoundedVec, ensure, fail, transactional, storage::with_transaction};22use up_data_structs::{23 AccessMode, CollectionId, CustomDataLimit, TokenId, CreateCollectionData, CreateNftExData,24 mapping::TokenAddressMapping, NestingRule, budget::Budget, Property, PropertyPermission,25 PropertyKey, PropertyKeyPermission, Properties, PropertyScope, TrySetProperty,26};27use pallet_evm::{account::CrossAccountId, Pallet as PalletEvm};28use pallet_common::{29 Error as CommonError, Pallet as PalletCommon, Event as CommonEvent, CollectionHandle,30 eth::collection_id_to_address,31};32use pallet_structure::Pallet as PalletStructure;33use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};34use sp_core::H160;35use sp_runtime::{ArithmeticError, DispatchError, DispatchResult, TransactionOutcome};36use sp_std::{vec::Vec, vec, collections::btree_set::BTreeSet};37use core::ops::Deref;38use sp_std::collections::btree_map::BTreeMap;39use codec::{Encode, Decode, MaxEncodedLen};40use scale_info::TypeInfo;4142pub use pallet::*;43#[cfg(feature = "runtime-benchmarks")]44pub mod benchmarking;45pub mod common;46pub mod erc;47pub mod weights;4849pub type CreateItemData<T> = CreateNftExData<<T as pallet_evm::account::Config>::CrossAccountId>;50pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;5152#[struct_versioning::versioned(version = 2, upper)]53#[derive(Encode, Decode, TypeInfo, MaxEncodedLen)]54pub struct ItemData<CrossAccountId> {55 #[version(..2)]56 pub const_data: BoundedVec<u8, CustomDataLimit>,5758 #[version(..2)]59 pub variable_data: BoundedVec<u8, CustomDataLimit>,6061 pub owner: CrossAccountId,62}6364#[frame_support::pallet]65pub mod pallet {66 use super::*;67 use frame_support::{68 Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key, traits::StorageVersion,69 };70 use frame_system::pallet_prelude::*;71 use up_data_structs::{CollectionId, TokenId};72 use super::weights::WeightInfo;7374 #[pallet::error]75 pub enum Error<T> {76 /// Not Nonfungible item data used to mint in Nonfungible collection.77 NotNonfungibleDataUsedToMintFungibleCollectionToken,78 /// Used amount > 1 with NFT79 NonfungibleItemsHaveNoAmount,80 /// Unable to burn NFT with children81 CantBurnNftWithChildren,82 /// Unable to burn a collection containing NFTs that have children83 CantBurnCollectionWithNestedTokens84 }8586 #[pallet::config]87 pub trait Config:88 frame_system::Config + pallet_common::Config + pallet_structure::Config + pallet_evm::Config89 {90 type WeightInfo: WeightInfo;91 }9293 const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);9495 #[pallet::pallet]96 #[pallet::storage_version(STORAGE_VERSION)]97 #[pallet::generate_store(pub(super) trait Store)]98 pub struct Pallet<T>(_);99100 #[pallet::storage]101 pub type TokensMinted<T: Config> =102 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;103 #[pallet::storage]104 pub type TokensBurnt<T: Config> =105 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;106107 #[pallet::storage]108 pub type TokenData<T: Config> = StorageNMap<109 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),110 Value = ItemData<T::CrossAccountId>,111 QueryKind = OptionQuery,112 >;113114 #[pallet::storage]115 #[pallet::getter(fn token_properties)]116 pub type TokenProperties<T: Config> = StorageNMap<117 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),118 Value = Properties,119 QueryKind = ValueQuery,120 OnEmpty = up_data_structs::TokenProperties,121 >;122123 /// Used to enumerate tokens owned by account124 #[pallet::storage]125 pub type Owned<T: Config> = StorageNMap<126 Key = (127 Key<Twox64Concat, CollectionId>,128 Key<Blake2_128Concat, T::CrossAccountId>,129 Key<Twox64Concat, TokenId>,130 ),131 Value = bool,132 QueryKind = ValueQuery,133 >;134135 /// Used to enumerate token's children136 #[pallet::storage]137 #[pallet::getter(fn token_children)]138 pub type TokenChildren<T: Config> = StorageNMap<139 Key = (140 Key<Twox64Concat, CollectionId>,141 Key<Twox64Concat, TokenId>,142 Key<Twox64Concat, (CollectionId, TokenId)>,143 ),144 Value = bool,145 QueryKind = ValueQuery,146 >;147148 #[pallet::storage]149 pub type AccountBalance<T: Config> = StorageNMap<150 Key = (151 Key<Twox64Concat, CollectionId>,152 Key<Blake2_128Concat, T::CrossAccountId>,153 ),154 Value = u32,155 QueryKind = ValueQuery,156 >;157158 #[pallet::storage]159 pub type Allowance<T: Config> = StorageNMap<160 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),161 Value = T::CrossAccountId,162 QueryKind = OptionQuery,163 >;164165 #[pallet::hooks]166 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {167 fn on_runtime_upgrade() -> Weight {168 if StorageVersion::get::<Pallet<T>>() < StorageVersion::new(1) {169 let mut had_consts = BTreeSet::new();170 <TokenData<T>>::translate::<ItemDataVersion1<T::CrossAccountId>, _>(|(collection, token), v| {171 let mut props = vec![];172 if !v.const_data.is_empty() {173 props.push(Property {174 key: b"_old_constData".to_vec().try_into().unwrap(),175 value: v.const_data.clone().into_inner().try_into().expect("const too long"),176 });177 had_consts.insert(collection);178 }179 if !v.variable_data.is_empty() {180 props.push(Property {181 key: b"_old_variableData".to_vec().try_into().unwrap(),182 value: v.variable_data.clone().into_inner().try_into().expect("variable too long"),183 })184 }185 if !props.is_empty() {186 Self::set_scoped_token_properties(187 collection,188 token,189 PropertyScope::None,190 props.into_iter(),191 ).expect("existing token data exceeds property storage");192 }193 Some(<ItemDataVersion2<T::CrossAccountId>>::from(v))194 });195 for collection in had_consts {196 <PalletCommon<T>>::set_property_permission_unchecked(197 collection,198 PropertyKeyPermission {199 key: b"_old_constData".to_vec().try_into().unwrap(),200 permission: PropertyPermission {201 mutable: false,202 collection_admin: true,203 token_owner: false,204 },205 }206 ).expect("failed to configure permission");207 }208 }209210 0211 }212 }213}214215pub struct NonfungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);216impl<T: Config> NonfungibleHandle<T> {217 pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {218 Self(inner)219 }220 pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {221 self.0222 }223 pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {224 &mut self.0225 }226}227impl<T: Config> WithRecorder<T> for NonfungibleHandle<T> {228 fn recorder(&self) -> &SubstrateRecorder<T> {229 self.0.recorder()230 }231 fn into_recorder(self) -> SubstrateRecorder<T> {232 self.0.into_recorder()233 }234}235impl<T: Config> Deref for NonfungibleHandle<T> {236 type Target = pallet_common::CollectionHandle<T>;237238 fn deref(&self) -> &Self::Target {239 &self.0240 }241}242243impl<T: Config> Pallet<T> {244 pub fn total_supply(collection: &NonfungibleHandle<T>) -> u32 {245 <TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)246 }247 pub fn token_exists(collection: &NonfungibleHandle<T>, token: TokenId) -> bool {248 <TokenData<T>>::contains_key((collection.id, token))249 }250251 pub fn set_scoped_token_property(252 collection_id: CollectionId,253 token_id: TokenId,254 scope: PropertyScope,255 property: Property,256 ) -> DispatchResult {257 TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {258 properties.try_scoped_set(scope, property.key, property.value)259 })260 .map_err(<CommonError<T>>::from)?;261262 Ok(())263 }264265 pub fn set_scoped_token_properties(266 collection_id: CollectionId,267 token_id: TokenId,268 scope: PropertyScope,269 properties: impl Iterator<Item=Property>,270 ) -> DispatchResult {271 TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {272 stored_properties.try_scoped_set_from_iter(scope, properties)273 })274 .map_err(<CommonError<T>>::from)?;275276 Ok(())277 }278279 pub fn current_token_id(collection_id: CollectionId) -> TokenId {280 TokenId(<TokensMinted<T>>::get(collection_id))281 }282}283284// unchecked calls skips any permission checks285impl<T: Config> Pallet<T> {286 pub fn init_collection(287 owner: T::AccountId,288 data: CreateCollectionData<T::AccountId>,289 ) -> Result<CollectionId, DispatchError> {290 <PalletCommon<T>>::init_collection(owner, data)291 }292 pub fn destroy_collection(293 collection: NonfungibleHandle<T>,294 sender: &T::CrossAccountId,295 ) -> DispatchResult {296 let id = collection.id;297298 if Self::collection_has_nested_tokens(id) {299 return Err(<Error<T>>::CantBurnCollectionWithNestedTokens.into());300 }301302 // =========303304 PalletCommon::destroy_collection(collection.0, sender)?;305306 <TokenData<T>>::remove_prefix((id,), None);307 <TokenChildren<T>>::remove_prefix((id,), None);308 <Owned<T>>::remove_prefix((id,), None);309 <TokensMinted<T>>::remove(id);310 <TokensBurnt<T>>::remove(id);311 <Allowance<T>>::remove_prefix((id,), None);312 <AccountBalance<T>>::remove_prefix((id,), None);313 Ok(())314 }315316 pub fn burn(317 collection: &NonfungibleHandle<T>,318 sender: &T::CrossAccountId,319 token: TokenId,320 ) -> DispatchResult {321 let token_data =322 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;323 ensure!(324 &token_data.owner == sender325 || (collection.limits.owner_can_transfer() && collection.is_owner_or_admin(sender)),326 <CommonError<T>>::NoPermission327 );328329 if collection.permissions.access() == AccessMode::AllowList {330 collection.check_allowlist(sender)?;331 }332333 if Self::token_has_children(collection.id, token) {334 return Err(<Error<T>>::CantBurnNftWithChildren.into());335 }336337 let burnt = <TokensBurnt<T>>::get(collection.id)338 .checked_add(1)339 .ok_or(ArithmeticError::Overflow)?;340341 let balance = <AccountBalance<T>>::get((collection.id, token_data.owner.clone()))342 .checked_sub(1)343 .ok_or(ArithmeticError::Overflow)?;344345 if balance == 0 {346 <AccountBalance<T>>::remove((collection.id, token_data.owner.clone()));347 } else {348 <AccountBalance<T>>::insert((collection.id, token_data.owner.clone()), balance);349 }350351 if let Some(owner) = T::CrossTokenAddressMapping::address_to_token(&token_data.owner) {352 Self::unnest(owner, (collection.id, token));353 }354355 // =========356357 <Owned<T>>::remove((collection.id, &token_data.owner, token));358 <TokensBurnt<T>>::insert(collection.id, burnt);359 <TokenData<T>>::remove((collection.id, token));360 <TokenProperties<T>>::remove((collection.id, token));361 let old_spender = <Allowance<T>>::take((collection.id, token));362363 if let Some(old_spender) = old_spender {364 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(365 collection.id,366 token,367 sender.clone(),368 old_spender,369 0,370 ));371 }372373 <PalletEvm<T>>::deposit_log(374 ERC721Events::Transfer {375 from: *token_data.owner.as_eth(),376 to: H160::default(),377 token_id: token.into(),378 }379 .to_log(collection_id_to_address(collection.id)),380 );381 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(382 collection.id,383 token,384 token_data.owner,385 1,386 ));387 Ok(())388 }389390 pub fn set_token_property(391 collection: &NonfungibleHandle<T>,392 sender: &T::CrossAccountId,393 token_id: TokenId,394 property: Property,395 ) -> DispatchResult {396 Self::check_token_change_permission(collection, sender, token_id, &property.key)?;397398 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {399 let property = property.clone();400 properties.try_set(property.key, property.value)401 })402 .map_err(<CommonError<T>>::from)?;403404 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertySet(405 collection.id,406 token_id,407 property.key,408 ));409410 Ok(())411 }412413 #[transactional]414 pub fn set_token_properties(415 collection: &NonfungibleHandle<T>,416 sender: &T::CrossAccountId,417 token_id: TokenId,418 properties: Vec<Property>,419 ) -> DispatchResult {420 for property in properties {421 Self::set_token_property(collection, sender, token_id, property)?;422 }423424 Ok(())425 }426427 pub fn delete_token_property(428 collection: &NonfungibleHandle<T>,429 sender: &T::CrossAccountId,430 token_id: TokenId,431 property_key: PropertyKey,432 ) -> DispatchResult {433 Self::check_token_change_permission(collection, sender, token_id, &property_key)?;434435 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {436 properties.remove(&property_key)437 })438 .map_err(<CommonError<T>>::from)?;439440 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertyDeleted(441 collection.id,442 token_id,443 property_key,444 ));445446 Ok(())447 }448449 fn check_token_change_permission(450 collection: &NonfungibleHandle<T>,451 sender: &T::CrossAccountId,452 token_id: TokenId,453 property_key: &PropertyKey,454 ) -> DispatchResult {455 let permission = <PalletCommon<T>>::property_permissions(collection.id)456 .get(property_key)457 .cloned()458 .unwrap_or_else(PropertyPermission::none);459460 let token_data = <TokenData<T>>::get((collection.id, token_id))461 .ok_or(<CommonError<T>>::TokenNotFound)?;462463 let check_token_owner = || -> DispatchResult {464 ensure!(&token_data.owner == sender, <CommonError<T>>::NoPermission);465 Ok(())466 };467468 let is_property_exists = TokenProperties::<T>::get((collection.id, token_id))469 .get(property_key)470 .is_some();471472 match permission {473 PropertyPermission { mutable: false, .. } if is_property_exists => {474 Err(<CommonError<T>>::NoPermission.into())475 }476477 PropertyPermission {478 collection_admin,479 token_owner,480 ..481 } => {482 let mut check_result = Err(<CommonError<T>>::NoPermission.into());483484 if collection_admin {485 check_result = collection.check_is_owner_or_admin(sender);486 }487488 if token_owner {489 check_result.or_else(|_| check_token_owner())490 } else {491 check_result492 }493 }494 }495 }496497 #[transactional]498 pub fn delete_token_properties(499 collection: &NonfungibleHandle<T>,500 sender: &T::CrossAccountId,501 token_id: TokenId,502 property_keys: Vec<PropertyKey>,503 ) -> DispatchResult {504 for key in property_keys {505 Self::delete_token_property(collection, sender, token_id, key)?;506 }507508 Ok(())509 }510511 pub fn set_collection_properties(512 collection: &NonfungibleHandle<T>,513 sender: &T::CrossAccountId,514 properties: Vec<Property>,515 ) -> DispatchResult {516 <PalletCommon<T>>::set_collection_properties(collection, sender, properties)517 }518519 pub fn delete_collection_properties(520 collection: &CollectionHandle<T>,521 sender: &T::CrossAccountId,522 property_keys: Vec<PropertyKey>,523 ) -> DispatchResult {524 <PalletCommon<T>>::delete_collection_properties(collection, sender, property_keys)525 }526527 pub fn set_property_permissions(528 collection: &CollectionHandle<T>,529 sender: &T::CrossAccountId,530 property_permissions: Vec<PropertyKeyPermission>,531 ) -> DispatchResult {532 <PalletCommon<T>>::set_property_permissions(collection, sender, property_permissions)533 }534535 pub fn set_property_permission(536 collection: &CollectionHandle<T>,537 sender: &T::CrossAccountId,538 permission: PropertyKeyPermission,539 ) -> DispatchResult {540 <PalletCommon<T>>::set_property_permission(collection, sender, permission)541 }542543 pub fn transfer(544 collection: &NonfungibleHandle<T>,545 from: &T::CrossAccountId,546 to: &T::CrossAccountId,547 token: TokenId,548 nesting_budget: &dyn Budget,549 ) -> DispatchResult {550 ensure!(551 collection.limits.transfers_enabled(),552 <CommonError<T>>::TransferNotAllowed553 );554555 let token_data =556 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;557 // TODO: require sender to be token, owner, require admins to go through transfer_from558 ensure!(559 &token_data.owner == from560 || (collection.limits.owner_can_transfer() && collection.is_owner_or_admin(from)),561 <CommonError<T>>::NoPermission562 );563564 if collection.permissions.access() == AccessMode::AllowList {565 collection.check_allowlist(from)?;566 collection.check_allowlist(to)?;567 }568 <PalletCommon<T>>::ensure_correct_receiver(to)?;569570 let balance_from = <AccountBalance<T>>::get((collection.id, from))571 .checked_sub(1)572 .ok_or(<CommonError<T>>::TokenValueTooLow)?;573 let balance_to = if from != to {574 let balance_to = <AccountBalance<T>>::get((collection.id, to))575 .checked_add(1)576 .ok_or(ArithmeticError::Overflow)?;577578 ensure!(579 balance_to < collection.limits.account_token_ownership_limit(),580 <CommonError<T>>::AccountTokenLimitExceeded,581 );582583 Some(balance_to)584 } else {585 None586 };587588 <PalletStructure<T>>::nest_if_sent_to_token(589 from.clone(),590 to,591 collection.id,592 token,593 nesting_budget594 )?;595596 // =========597598 <TokenData<T>>::insert(599 (collection.id, token),600 ItemData {601 owner: to.clone(),602 ..token_data603 },604 );605606 if let Some(balance_to) = balance_to {607 // from != to608 if balance_from == 0 {609 <AccountBalance<T>>::remove((collection.id, from));610 } else {611 <AccountBalance<T>>::insert((collection.id, from), balance_from);612 }613 <AccountBalance<T>>::insert((collection.id, to), balance_to);614 <Owned<T>>::remove((collection.id, from, token));615 <Owned<T>>::insert((collection.id, to, token), true);616 }617 Self::set_allowance_unchecked(collection, from, token, None, true);618619 <PalletEvm<T>>::deposit_log(620 ERC721Events::Transfer {621 from: *from.as_eth(),622 to: *to.as_eth(),623 token_id: token.into(),624 }625 .to_log(collection_id_to_address(collection.id)),626 );627 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(628 collection.id,629 token,630 from.clone(),631 to.clone(),632 1,633 ));634 Ok(())635 }636637 pub fn create_multiple_items(638 collection: &NonfungibleHandle<T>,639 sender: &T::CrossAccountId,640 data: Vec<CreateItemData<T>>,641 nesting_budget: &dyn Budget,642 ) -> DispatchResult {643 if !collection.is_owner_or_admin(sender) {644 ensure!(645 collection.permissions.mint_mode(),646 <CommonError<T>>::PublicMintingNotAllowed647 );648 collection.check_allowlist(sender)?;649650 for item in data.iter() {651 collection.check_allowlist(&item.owner)?;652 }653 }654655 for data in data.iter() {656 <PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;657 }658659 let first_token = <TokensMinted<T>>::get(collection.id);660 let tokens_minted = first_token661 .checked_add(data.len() as u32)662 .ok_or(ArithmeticError::Overflow)?;663 ensure!(664 tokens_minted <= collection.limits.token_limit(),665 <CommonError<T>>::CollectionTokenLimitExceeded666 );667668 let mut balances = BTreeMap::new();669 for data in &data {670 let balance = balances671 .entry(&data.owner)672 .or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));673 *balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;674675 ensure!(676 *balance <= collection.limits.account_token_ownership_limit(),677 <CommonError<T>>::AccountTokenLimitExceeded,678 );679 }680681 for (i, data) in data.iter().enumerate() {682 let token = TokenId(first_token + i as u32 + 1);683684 <PalletStructure<T>>::check_nesting(685 sender.clone(),686 &data.owner,687 collection.id,688 token,689 nesting_budget,690 )?;691 }692693 // =========694695 with_transaction(|| {696 for (i, data) in data.iter().enumerate() {697 let token = first_token + i as u32 + 1;698699 <TokenData<T>>::insert(700 (collection.id, token),701 ItemData {702 // const_data: data.const_data.clone(),703 owner: data.owner.clone(),704 },705 );706707 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(&data.owner, collection.id, TokenId(token));708709 if let Err(e) = Self::set_token_properties(710 collection,711 sender,712 TokenId(token),713 data.properties.clone().into_inner(),714 ) {715 return TransactionOutcome::Rollback(Err(e));716 }717 }718 TransactionOutcome::Commit(Ok(()))719 })?;720721 <TokensMinted<T>>::insert(collection.id, tokens_minted);722 for (account, balance) in balances {723 <AccountBalance<T>>::insert((collection.id, account), balance);724 }725 for (i, data) in data.into_iter().enumerate() {726 let token = first_token + i as u32 + 1;727 <Owned<T>>::insert((collection.id, &data.owner, token), true);728729 <PalletEvm<T>>::deposit_log(730 ERC721Events::Transfer {731 from: H160::default(),732 to: *data.owner.as_eth(),733 token_id: token.into(),734 }735 .to_log(collection_id_to_address(collection.id)),736 );737 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(738 collection.id,739 TokenId(token),740 data.owner.clone(),741 1,742 ));743 }744 Ok(())745 }746747 pub fn set_allowance_unchecked(748 collection: &NonfungibleHandle<T>,749 sender: &T::CrossAccountId,750 token: TokenId,751 spender: Option<&T::CrossAccountId>,752 assume_implicit_eth: bool,753 ) {754 if let Some(spender) = spender {755 let old_spender = <Allowance<T>>::get((collection.id, token));756 <Allowance<T>>::insert((collection.id, token), spender);757 // In ERC721 there is only one possible approved user of token, so we set758 // approved user to spender759 <PalletEvm<T>>::deposit_log(760 ERC721Events::Approval {761 owner: *sender.as_eth(),762 approved: *spender.as_eth(),763 token_id: token.into(),764 }765 .to_log(collection_id_to_address(collection.id)),766 );767 // In Unique chain, any token can have any amount of approved users, so we need to768 // set allowance of old owner to 0, and allowance of new owner to 1769 if old_spender.as_ref() != Some(spender) {770 if let Some(old_owner) = old_spender {771 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(772 collection.id,773 token,774 sender.clone(),775 old_owner,776 0,777 ));778 }779 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(780 collection.id,781 token,782 sender.clone(),783 spender.clone(),784 1,785 ));786 }787 } else {788 let old_spender = <Allowance<T>>::take((collection.id, token));789 if !assume_implicit_eth {790 // In ERC721 there is only one possible approved user of token, so we set791 // approved user to zero address792 <PalletEvm<T>>::deposit_log(793 ERC721Events::Approval {794 owner: *sender.as_eth(),795 approved: H160::default(),796 token_id: token.into(),797 }798 .to_log(collection_id_to_address(collection.id)),799 );800 }801 // In Unique chain, any token can have any amount of approved users, so we need to802 // set allowance of old owner to 0803 if let Some(old_spender) = old_spender {804 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(805 collection.id,806 token,807 sender.clone(),808 old_spender,809 0,810 ));811 }812 }813 }814815 pub fn set_allowance(816 collection: &NonfungibleHandle<T>,817 sender: &T::CrossAccountId,818 token: TokenId,819 spender: Option<&T::CrossAccountId>,820 ) -> DispatchResult {821 if collection.permissions.access() == AccessMode::AllowList {822 collection.check_allowlist(sender)?;823 if let Some(spender) = spender {824 collection.check_allowlist(spender)?;825 }826 }827828 if let Some(spender) = spender {829 <PalletCommon<T>>::ensure_correct_receiver(spender)?;830 }831 let token_data =832 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;833 if &token_data.owner != sender {834 ensure!(835 collection.ignores_owned_amount(sender),836 <CommonError<T>>::CantApproveMoreThanOwned837 );838 }839840 // =========841842 Self::set_allowance_unchecked(collection, sender, token, spender, false);843 Ok(())844 }845846 fn check_allowed(847 collection: &NonfungibleHandle<T>,848 spender: &T::CrossAccountId,849 from: &T::CrossAccountId,850 token: TokenId,851 nesting_budget: &dyn Budget,852 ) -> DispatchResult {853 if spender.conv_eq(from) {854 return Ok(());855 }856 if collection.permissions.access() == AccessMode::AllowList {857 // `from`, `to` checked in [`transfer`]858 collection.check_allowlist(spender)?;859 }860 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {861 // TODO: should collection owner be allowed to perform this transfer?862 ensure!(863 <PalletStructure<T>>::check_indirectly_owned(864 spender.clone(),865 source.0,866 source.1,867 None,868 nesting_budget869 )?,870 <CommonError<T>>::ApprovedValueTooLow,871 );872 return Ok(());873 }874 if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {875 return Ok(());876 }877 ensure!(878 collection.ignores_allowance(spender),879 <CommonError<T>>::ApprovedValueTooLow880 );881 Ok(())882 }883884 pub fn transfer_from(885 collection: &NonfungibleHandle<T>,886 spender: &T::CrossAccountId,887 from: &T::CrossAccountId,888 to: &T::CrossAccountId,889 token: TokenId,890 nesting_budget: &dyn Budget,891 ) -> DispatchResult {892 Self::check_allowed(collection, spender, from, token, nesting_budget)?;893894 // =========895896 // Allowance is reset in [`transfer`]897 Self::transfer(collection, from, to, token, nesting_budget)898 }899900 pub fn burn_from(901 collection: &NonfungibleHandle<T>,902 spender: &T::CrossAccountId,903 from: &T::CrossAccountId,904 token: TokenId,905 nesting_budget: &dyn Budget,906 ) -> DispatchResult {907 Self::check_allowed(collection, spender, from, token, nesting_budget)?;908909 // =========910911 Self::burn(collection, from, token)912 }913914 pub fn check_nesting(915 handle: &NonfungibleHandle<T>,916 sender: T::CrossAccountId,917 from: (CollectionId, TokenId),918 under: TokenId,919 nesting_budget: &dyn Budget,920 ) -> DispatchResult {921 fn ensure_sender_allowed<T: Config>(922 collection: CollectionId,923 token: TokenId,924 for_nest: (CollectionId, TokenId),925 sender: T::CrossAccountId,926 budget: &dyn Budget,927 ) -> DispatchResult {928 ensure!(929 <PalletStructure<T>>::check_indirectly_owned(930 sender,931 collection,932 token,933 Some(for_nest),934 budget935 )?,936 <CommonError<T>>::OnlyOwnerAllowedToNest,937 );938 Ok(())939 }940 match handle.permissions.nesting() {941 NestingRule::Disabled => fail!(<CommonError<T>>::NestingIsDisabled),942 NestingRule::Owner => {943 ensure_sender_allowed::<T>(handle.id, under, from, sender, nesting_budget)?944 }945 NestingRule::OwnerRestricted(whitelist) => {946 ensure!(947 whitelist.contains(&from.0),948 <CommonError<T>>::SourceCollectionIsNotAllowedToNest949 );950 ensure_sender_allowed::<T>(handle.id, under, from, sender, nesting_budget)?951 }952 }953 Ok(())954 }955956 fn nest(957 under: (CollectionId, TokenId),958 to_nest: (CollectionId, TokenId),959 ) {960 <TokenChildren<T>>::insert(961 (under.0, under.1, (to_nest.0, to_nest.1)),962 true963 );964 }965966 fn unnest(967 under: (CollectionId, TokenId),968 to_unnest: (CollectionId, TokenId),969 ) {970 <TokenChildren<T>>::remove(971 (under.0, under.1, to_unnest)972 );973 }974975 fn collection_has_nested_tokens(collection_id: CollectionId) -> bool {976 <TokenChildren<T>>::iter_prefix((collection_id,)).next().is_some()977 }978979 fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {980 <TokenChildren<T>>::iter_prefix((collection_id, token_id)).next().is_some()981 }982983 /// Delegated to `create_multiple_items`984 pub fn create_item(985 collection: &NonfungibleHandle<T>,986 sender: &T::CrossAccountId,987 data: CreateItemData<T>,988 nesting_budget: &dyn Budget,989 ) -> DispatchResult {990 Self::create_multiple_items(collection, sender, vec![data], nesting_budget)991 }992}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::{BoundedVec, ensure, fail, transactional, storage::with_transaction};22use up_data_structs::{23 AccessMode, CollectionId, CustomDataLimit, TokenId, CreateCollectionData, CreateNftExData,24 mapping::TokenAddressMapping, NestingRule, budget::Budget, Property, PropertyPermission,25 PropertyKey, PropertyKeyPermission, Properties, PropertyScope, TrySetProperty,26};27use pallet_evm::{account::CrossAccountId, Pallet as PalletEvm};28use pallet_common::{29 Error as CommonError, Pallet as PalletCommon, Event as CommonEvent, CollectionHandle,30 eth::collection_id_to_address,31};32use pallet_structure::Pallet as PalletStructure;33use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};34use sp_core::H160;35use sp_runtime::{ArithmeticError, DispatchError, DispatchResult, TransactionOutcome};36use sp_std::{vec::Vec, vec, collections::btree_set::BTreeSet};37use core::ops::Deref;38use sp_std::collections::btree_map::BTreeMap;39use codec::{Encode, Decode, MaxEncodedLen};40use scale_info::TypeInfo;4142pub use pallet::*;43#[cfg(feature = "runtime-benchmarks")]44pub mod benchmarking;45pub mod common;46pub mod erc;47pub mod weights;4849pub type CreateItemData<T> = CreateNftExData<<T as pallet_evm::account::Config>::CrossAccountId>;50pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;5152#[struct_versioning::versioned(version = 2, upper)]53#[derive(Encode, Decode, TypeInfo, MaxEncodedLen)]54pub struct ItemData<CrossAccountId> {55 #[version(..2)]56 pub const_data: BoundedVec<u8, CustomDataLimit>,5758 #[version(..2)]59 pub variable_data: BoundedVec<u8, CustomDataLimit>,6061 pub owner: CrossAccountId,62}6364#[frame_support::pallet]65pub mod pallet {66 use super::*;67 use frame_support::{68 Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key, traits::StorageVersion,69 };70 use frame_system::pallet_prelude::*;71 use up_data_structs::{CollectionId, TokenId};72 use super::weights::WeightInfo;7374 #[pallet::error]75 pub enum Error<T> {76 /// Not Nonfungible item data used to mint in Nonfungible collection.77 NotNonfungibleDataUsedToMintFungibleCollectionToken,78 /// Used amount > 1 with NFT79 NonfungibleItemsHaveNoAmount,80 /// Unable to burn NFT with children81 CantBurnNftWithChildren,82 }8384 #[pallet::config]85 pub trait Config:86 frame_system::Config + pallet_common::Config + pallet_structure::Config + pallet_evm::Config87 {88 type WeightInfo: WeightInfo;89 }9091 const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);9293 #[pallet::pallet]94 #[pallet::storage_version(STORAGE_VERSION)]95 #[pallet::generate_store(pub(super) trait Store)]96 pub struct Pallet<T>(_);9798 #[pallet::storage]99 pub type TokensMinted<T: Config> =100 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;101 #[pallet::storage]102 pub type TokensBurnt<T: Config> =103 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;104105 #[pallet::storage]106 pub type TokenData<T: Config> = StorageNMap<107 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),108 Value = ItemData<T::CrossAccountId>,109 QueryKind = OptionQuery,110 >;111112 #[pallet::storage]113 #[pallet::getter(fn token_properties)]114 pub type TokenProperties<T: Config> = StorageNMap<115 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),116 Value = Properties,117 QueryKind = ValueQuery,118 OnEmpty = up_data_structs::TokenProperties,119 >;120121 /// Used to enumerate tokens owned by account122 #[pallet::storage]123 pub type Owned<T: Config> = StorageNMap<124 Key = (125 Key<Twox64Concat, CollectionId>,126 Key<Blake2_128Concat, T::CrossAccountId>,127 Key<Twox64Concat, TokenId>,128 ),129 Value = bool,130 QueryKind = ValueQuery,131 >;132133 /// Used to enumerate token's children134 #[pallet::storage]135 #[pallet::getter(fn token_children)]136 pub type TokenChildren<T: Config> = StorageNMap<137 Key = (138 Key<Twox64Concat, CollectionId>,139 Key<Twox64Concat, TokenId>,140 Key<Twox64Concat, (CollectionId, TokenId)>,141 ),142 Value = bool,143 QueryKind = ValueQuery,144 >;145146 #[pallet::storage]147 pub type AccountBalance<T: Config> = StorageNMap<148 Key = (149 Key<Twox64Concat, CollectionId>,150 Key<Blake2_128Concat, T::CrossAccountId>,151 ),152 Value = u32,153 QueryKind = ValueQuery,154 >;155156 #[pallet::storage]157 pub type Allowance<T: Config> = StorageNMap<158 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),159 Value = T::CrossAccountId,160 QueryKind = OptionQuery,161 >;162163 #[pallet::hooks]164 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {165 fn on_runtime_upgrade() -> Weight {166 if StorageVersion::get::<Pallet<T>>() < StorageVersion::new(1) {167 let mut had_consts = BTreeSet::new();168 <TokenData<T>>::translate::<ItemDataVersion1<T::CrossAccountId>, _>(|(collection, token), v| {169 let mut props = vec![];170 if !v.const_data.is_empty() {171 props.push(Property {172 key: b"_old_constData".to_vec().try_into().unwrap(),173 value: v.const_data.clone().into_inner().try_into().expect("const too long"),174 });175 had_consts.insert(collection);176 }177 if !v.variable_data.is_empty() {178 props.push(Property {179 key: b"_old_variableData".to_vec().try_into().unwrap(),180 value: v.variable_data.clone().into_inner().try_into().expect("variable too long"),181 })182 }183 if !props.is_empty() {184 Self::set_scoped_token_properties(185 collection,186 token,187 PropertyScope::None,188 props.into_iter(),189 ).expect("existing token data exceeds property storage");190 }191 Some(<ItemDataVersion2<T::CrossAccountId>>::from(v))192 });193 for collection in had_consts {194 <PalletCommon<T>>::set_property_permission_unchecked(195 collection,196 PropertyKeyPermission {197 key: b"_old_constData".to_vec().try_into().unwrap(),198 permission: PropertyPermission {199 mutable: false,200 collection_admin: true,201 token_owner: false,202 },203 }204 ).expect("failed to configure permission");205 }206 }207208 0209 }210 }211}212213pub struct NonfungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);214impl<T: Config> NonfungibleHandle<T> {215 pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {216 Self(inner)217 }218 pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {219 self.0220 }221 pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {222 &mut self.0223 }224}225impl<T: Config> WithRecorder<T> for NonfungibleHandle<T> {226 fn recorder(&self) -> &SubstrateRecorder<T> {227 self.0.recorder()228 }229 fn into_recorder(self) -> SubstrateRecorder<T> {230 self.0.into_recorder()231 }232}233impl<T: Config> Deref for NonfungibleHandle<T> {234 type Target = pallet_common::CollectionHandle<T>;235236 fn deref(&self) -> &Self::Target {237 &self.0238 }239}240241impl<T: Config> Pallet<T> {242 pub fn total_supply(collection: &NonfungibleHandle<T>) -> u32 {243 <TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)244 }245 pub fn token_exists(collection: &NonfungibleHandle<T>, token: TokenId) -> bool {246 <TokenData<T>>::contains_key((collection.id, token))247 }248249 pub fn set_scoped_token_property(250 collection_id: CollectionId,251 token_id: TokenId,252 scope: PropertyScope,253 property: Property,254 ) -> DispatchResult {255 TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {256 properties.try_scoped_set(scope, property.key, property.value)257 })258 .map_err(<CommonError<T>>::from)?;259260 Ok(())261 }262263 pub fn set_scoped_token_properties(264 collection_id: CollectionId,265 token_id: TokenId,266 scope: PropertyScope,267 properties: impl Iterator<Item=Property>,268 ) -> DispatchResult {269 TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {270 stored_properties.try_scoped_set_from_iter(scope, properties)271 })272 .map_err(<CommonError<T>>::from)?;273274 Ok(())275 }276277 pub fn current_token_id(collection_id: CollectionId) -> TokenId {278 TokenId(<TokensMinted<T>>::get(collection_id))279 }280}281282// unchecked calls skips any permission checks283impl<T: Config> Pallet<T> {284 pub fn init_collection(285 owner: T::AccountId,286 data: CreateCollectionData<T::AccountId>,287 ) -> Result<CollectionId, DispatchError> {288 <PalletCommon<T>>::init_collection(owner, data)289 }290 pub fn destroy_collection(291 collection: NonfungibleHandle<T>,292 sender: &T::CrossAccountId,293 ) -> DispatchResult {294 let id = collection.id;295296 if Self::collection_has_tokens(id) {297 return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());298 }299300 // =========301302 PalletCommon::destroy_collection(collection.0, sender)?;303304 <TokenData<T>>::remove_prefix((id,), None);305 <TokenChildren<T>>::remove_prefix((id,), None);306 <Owned<T>>::remove_prefix((id,), None);307 <TokensMinted<T>>::remove(id);308 <TokensBurnt<T>>::remove(id);309 <Allowance<T>>::remove_prefix((id,), None);310 <AccountBalance<T>>::remove_prefix((id,), None);311 Ok(())312 }313314 pub fn burn(315 collection: &NonfungibleHandle<T>,316 sender: &T::CrossAccountId,317 token: TokenId,318 ) -> DispatchResult {319 let token_data =320 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;321 ensure!(322 &token_data.owner == sender323 || (collection.limits.owner_can_transfer() && collection.is_owner_or_admin(sender)),324 <CommonError<T>>::NoPermission325 );326327 if collection.permissions.access() == AccessMode::AllowList {328 collection.check_allowlist(sender)?;329 }330331 if Self::token_has_children(collection.id, token) {332 return Err(<Error<T>>::CantBurnNftWithChildren.into());333 }334335 let burnt = <TokensBurnt<T>>::get(collection.id)336 .checked_add(1)337 .ok_or(ArithmeticError::Overflow)?;338339 let balance = <AccountBalance<T>>::get((collection.id, token_data.owner.clone()))340 .checked_sub(1)341 .ok_or(ArithmeticError::Overflow)?;342343 if balance == 0 {344 <AccountBalance<T>>::remove((collection.id, token_data.owner.clone()));345 } else {346 <AccountBalance<T>>::insert((collection.id, token_data.owner.clone()), balance);347 }348349 if let Some(owner) = T::CrossTokenAddressMapping::address_to_token(&token_data.owner) {350 Self::unnest(owner, (collection.id, token));351 }352353 // =========354355 <Owned<T>>::remove((collection.id, &token_data.owner, token));356 <TokensBurnt<T>>::insert(collection.id, burnt);357 <TokenData<T>>::remove((collection.id, token));358 <TokenProperties<T>>::remove((collection.id, token));359 let old_spender = <Allowance<T>>::take((collection.id, token));360361 if let Some(old_spender) = old_spender {362 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(363 collection.id,364 token,365 sender.clone(),366 old_spender,367 0,368 ));369 }370371 <PalletEvm<T>>::deposit_log(372 ERC721Events::Transfer {373 from: *token_data.owner.as_eth(),374 to: H160::default(),375 token_id: token.into(),376 }377 .to_log(collection_id_to_address(collection.id)),378 );379 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(380 collection.id,381 token,382 token_data.owner,383 1,384 ));385 Ok(())386 }387388 pub fn set_token_property(389 collection: &NonfungibleHandle<T>,390 sender: &T::CrossAccountId,391 token_id: TokenId,392 property: Property,393 ) -> DispatchResult {394 Self::check_token_change_permission(collection, sender, token_id, &property.key)?;395396 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {397 let property = property.clone();398 properties.try_set(property.key, property.value)399 })400 .map_err(<CommonError<T>>::from)?;401402 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertySet(403 collection.id,404 token_id,405 property.key,406 ));407408 Ok(())409 }410411 #[transactional]412 pub fn set_token_properties(413 collection: &NonfungibleHandle<T>,414 sender: &T::CrossAccountId,415 token_id: TokenId,416 properties: Vec<Property>,417 ) -> DispatchResult {418 for property in properties {419 Self::set_token_property(collection, sender, token_id, property)?;420 }421422 Ok(())423 }424425 pub fn delete_token_property(426 collection: &NonfungibleHandle<T>,427 sender: &T::CrossAccountId,428 token_id: TokenId,429 property_key: PropertyKey,430 ) -> DispatchResult {431 Self::check_token_change_permission(collection, sender, token_id, &property_key)?;432433 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {434 properties.remove(&property_key)435 })436 .map_err(<CommonError<T>>::from)?;437438 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertyDeleted(439 collection.id,440 token_id,441 property_key,442 ));443444 Ok(())445 }446447 fn check_token_change_permission(448 collection: &NonfungibleHandle<T>,449 sender: &T::CrossAccountId,450 token_id: TokenId,451 property_key: &PropertyKey,452 ) -> DispatchResult {453 let permission = <PalletCommon<T>>::property_permissions(collection.id)454 .get(property_key)455 .cloned()456 .unwrap_or_else(PropertyPermission::none);457458 let token_data = <TokenData<T>>::get((collection.id, token_id))459 .ok_or(<CommonError<T>>::TokenNotFound)?;460461 let check_token_owner = || -> DispatchResult {462 ensure!(&token_data.owner == sender, <CommonError<T>>::NoPermission);463 Ok(())464 };465466 let is_property_exists = TokenProperties::<T>::get((collection.id, token_id))467 .get(property_key)468 .is_some();469470 match permission {471 PropertyPermission { mutable: false, .. } if is_property_exists => {472 Err(<CommonError<T>>::NoPermission.into())473 }474475 PropertyPermission {476 collection_admin,477 token_owner,478 ..479 } => {480 let mut check_result = Err(<CommonError<T>>::NoPermission.into());481482 if collection_admin {483 check_result = collection.check_is_owner_or_admin(sender);484 }485486 if token_owner {487 check_result.or_else(|_| check_token_owner())488 } else {489 check_result490 }491 }492 }493 }494495 #[transactional]496 pub fn delete_token_properties(497 collection: &NonfungibleHandle<T>,498 sender: &T::CrossAccountId,499 token_id: TokenId,500 property_keys: Vec<PropertyKey>,501 ) -> DispatchResult {502 for key in property_keys {503 Self::delete_token_property(collection, sender, token_id, key)?;504 }505506 Ok(())507 }508509 pub fn set_collection_properties(510 collection: &NonfungibleHandle<T>,511 sender: &T::CrossAccountId,512 properties: Vec<Property>,513 ) -> DispatchResult {514 <PalletCommon<T>>::set_collection_properties(collection, sender, properties)515 }516517 pub fn delete_collection_properties(518 collection: &CollectionHandle<T>,519 sender: &T::CrossAccountId,520 property_keys: Vec<PropertyKey>,521 ) -> DispatchResult {522 <PalletCommon<T>>::delete_collection_properties(collection, sender, property_keys)523 }524525 pub fn set_property_permissions(526 collection: &CollectionHandle<T>,527 sender: &T::CrossAccountId,528 property_permissions: Vec<PropertyKeyPermission>,529 ) -> DispatchResult {530 <PalletCommon<T>>::set_property_permissions(collection, sender, property_permissions)531 }532533 pub fn set_property_permission(534 collection: &CollectionHandle<T>,535 sender: &T::CrossAccountId,536 permission: PropertyKeyPermission,537 ) -> DispatchResult {538 <PalletCommon<T>>::set_property_permission(collection, sender, permission)539 }540541 pub fn transfer(542 collection: &NonfungibleHandle<T>,543 from: &T::CrossAccountId,544 to: &T::CrossAccountId,545 token: TokenId,546 nesting_budget: &dyn Budget,547 ) -> DispatchResult {548 ensure!(549 collection.limits.transfers_enabled(),550 <CommonError<T>>::TransferNotAllowed551 );552553 let token_data =554 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;555 // TODO: require sender to be token, owner, require admins to go through transfer_from556 ensure!(557 &token_data.owner == from558 || (collection.limits.owner_can_transfer() && collection.is_owner_or_admin(from)),559 <CommonError<T>>::NoPermission560 );561562 if collection.permissions.access() == AccessMode::AllowList {563 collection.check_allowlist(from)?;564 collection.check_allowlist(to)?;565 }566 <PalletCommon<T>>::ensure_correct_receiver(to)?;567568 let balance_from = <AccountBalance<T>>::get((collection.id, from))569 .checked_sub(1)570 .ok_or(<CommonError<T>>::TokenValueTooLow)?;571 let balance_to = if from != to {572 let balance_to = <AccountBalance<T>>::get((collection.id, to))573 .checked_add(1)574 .ok_or(ArithmeticError::Overflow)?;575576 ensure!(577 balance_to < collection.limits.account_token_ownership_limit(),578 <CommonError<T>>::AccountTokenLimitExceeded,579 );580581 Some(balance_to)582 } else {583 None584 };585586 <PalletStructure<T>>::nest_if_sent_to_token(587 from.clone(),588 to,589 collection.id,590 token,591 nesting_budget592 )?;593594 // =========595596 <TokenData<T>>::insert(597 (collection.id, token),598 ItemData {599 owner: to.clone(),600 ..token_data601 },602 );603604 if let Some(balance_to) = balance_to {605 // from != to606 if balance_from == 0 {607 <AccountBalance<T>>::remove((collection.id, from));608 } else {609 <AccountBalance<T>>::insert((collection.id, from), balance_from);610 }611 <AccountBalance<T>>::insert((collection.id, to), balance_to);612 <Owned<T>>::remove((collection.id, from, token));613 <Owned<T>>::insert((collection.id, to, token), true);614 }615 Self::set_allowance_unchecked(collection, from, token, None, true);616617 <PalletEvm<T>>::deposit_log(618 ERC721Events::Transfer {619 from: *from.as_eth(),620 to: *to.as_eth(),621 token_id: token.into(),622 }623 .to_log(collection_id_to_address(collection.id)),624 );625 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(626 collection.id,627 token,628 from.clone(),629 to.clone(),630 1,631 ));632 Ok(())633 }634635 pub fn create_multiple_items(636 collection: &NonfungibleHandle<T>,637 sender: &T::CrossAccountId,638 data: Vec<CreateItemData<T>>,639 nesting_budget: &dyn Budget,640 ) -> DispatchResult {641 if !collection.is_owner_or_admin(sender) {642 ensure!(643 collection.permissions.mint_mode(),644 <CommonError<T>>::PublicMintingNotAllowed645 );646 collection.check_allowlist(sender)?;647648 for item in data.iter() {649 collection.check_allowlist(&item.owner)?;650 }651 }652653 for data in data.iter() {654 <PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;655 }656657 let first_token = <TokensMinted<T>>::get(collection.id);658 let tokens_minted = first_token659 .checked_add(data.len() as u32)660 .ok_or(ArithmeticError::Overflow)?;661 ensure!(662 tokens_minted <= collection.limits.token_limit(),663 <CommonError<T>>::CollectionTokenLimitExceeded664 );665666 let mut balances = BTreeMap::new();667 for data in &data {668 let balance = balances669 .entry(&data.owner)670 .or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));671 *balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;672673 ensure!(674 *balance <= collection.limits.account_token_ownership_limit(),675 <CommonError<T>>::AccountTokenLimitExceeded,676 );677 }678679 for (i, data) in data.iter().enumerate() {680 let token = TokenId(first_token + i as u32 + 1);681682 <PalletStructure<T>>::check_nesting(683 sender.clone(),684 &data.owner,685 collection.id,686 token,687 nesting_budget,688 )?;689 }690691 // =========692693 with_transaction(|| {694 for (i, data) in data.iter().enumerate() {695 let token = first_token + i as u32 + 1;696697 <TokenData<T>>::insert(698 (collection.id, token),699 ItemData {700 // const_data: data.const_data.clone(),701 owner: data.owner.clone(),702 },703 );704705 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(&data.owner, collection.id, TokenId(token));706707 if let Err(e) = Self::set_token_properties(708 collection,709 sender,710 TokenId(token),711 data.properties.clone().into_inner(),712 ) {713 return TransactionOutcome::Rollback(Err(e));714 }715 }716 TransactionOutcome::Commit(Ok(()))717 })?;718719 <TokensMinted<T>>::insert(collection.id, tokens_minted);720 for (account, balance) in balances {721 <AccountBalance<T>>::insert((collection.id, account), balance);722 }723 for (i, data) in data.into_iter().enumerate() {724 let token = first_token + i as u32 + 1;725 <Owned<T>>::insert((collection.id, &data.owner, token), true);726727 <PalletEvm<T>>::deposit_log(728 ERC721Events::Transfer {729 from: H160::default(),730 to: *data.owner.as_eth(),731 token_id: token.into(),732 }733 .to_log(collection_id_to_address(collection.id)),734 );735 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(736 collection.id,737 TokenId(token),738 data.owner.clone(),739 1,740 ));741 }742 Ok(())743 }744745 pub fn set_allowance_unchecked(746 collection: &NonfungibleHandle<T>,747 sender: &T::CrossAccountId,748 token: TokenId,749 spender: Option<&T::CrossAccountId>,750 assume_implicit_eth: bool,751 ) {752 if let Some(spender) = spender {753 let old_spender = <Allowance<T>>::get((collection.id, token));754 <Allowance<T>>::insert((collection.id, token), spender);755 // In ERC721 there is only one possible approved user of token, so we set756 // approved user to spender757 <PalletEvm<T>>::deposit_log(758 ERC721Events::Approval {759 owner: *sender.as_eth(),760 approved: *spender.as_eth(),761 token_id: token.into(),762 }763 .to_log(collection_id_to_address(collection.id)),764 );765 // In Unique chain, any token can have any amount of approved users, so we need to766 // set allowance of old owner to 0, and allowance of new owner to 1767 if old_spender.as_ref() != Some(spender) {768 if let Some(old_owner) = old_spender {769 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(770 collection.id,771 token,772 sender.clone(),773 old_owner,774 0,775 ));776 }777 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(778 collection.id,779 token,780 sender.clone(),781 spender.clone(),782 1,783 ));784 }785 } else {786 let old_spender = <Allowance<T>>::take((collection.id, token));787 if !assume_implicit_eth {788 // In ERC721 there is only one possible approved user of token, so we set789 // approved user to zero address790 <PalletEvm<T>>::deposit_log(791 ERC721Events::Approval {792 owner: *sender.as_eth(),793 approved: H160::default(),794 token_id: token.into(),795 }796 .to_log(collection_id_to_address(collection.id)),797 );798 }799 // In Unique chain, any token can have any amount of approved users, so we need to800 // set allowance of old owner to 0801 if let Some(old_spender) = old_spender {802 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(803 collection.id,804 token,805 sender.clone(),806 old_spender,807 0,808 ));809 }810 }811 }812813 pub fn set_allowance(814 collection: &NonfungibleHandle<T>,815 sender: &T::CrossAccountId,816 token: TokenId,817 spender: Option<&T::CrossAccountId>,818 ) -> DispatchResult {819 if collection.permissions.access() == AccessMode::AllowList {820 collection.check_allowlist(sender)?;821 if let Some(spender) = spender {822 collection.check_allowlist(spender)?;823 }824 }825826 if let Some(spender) = spender {827 <PalletCommon<T>>::ensure_correct_receiver(spender)?;828 }829 let token_data =830 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;831 if &token_data.owner != sender {832 ensure!(833 collection.ignores_owned_amount(sender),834 <CommonError<T>>::CantApproveMoreThanOwned835 );836 }837838 // =========839840 Self::set_allowance_unchecked(collection, sender, token, spender, false);841 Ok(())842 }843844 fn check_allowed(845 collection: &NonfungibleHandle<T>,846 spender: &T::CrossAccountId,847 from: &T::CrossAccountId,848 token: TokenId,849 nesting_budget: &dyn Budget,850 ) -> DispatchResult {851 if spender.conv_eq(from) {852 return Ok(());853 }854 if collection.permissions.access() == AccessMode::AllowList {855 // `from`, `to` checked in [`transfer`]856 collection.check_allowlist(spender)?;857 }858 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {859 // TODO: should collection owner be allowed to perform this transfer?860 ensure!(861 <PalletStructure<T>>::check_indirectly_owned(862 spender.clone(),863 source.0,864 source.1,865 None,866 nesting_budget867 )?,868 <CommonError<T>>::ApprovedValueTooLow,869 );870 return Ok(());871 }872 if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {873 return Ok(());874 }875 ensure!(876 collection.ignores_allowance(spender),877 <CommonError<T>>::ApprovedValueTooLow878 );879 Ok(())880 }881882 pub fn transfer_from(883 collection: &NonfungibleHandle<T>,884 spender: &T::CrossAccountId,885 from: &T::CrossAccountId,886 to: &T::CrossAccountId,887 token: TokenId,888 nesting_budget: &dyn Budget,889 ) -> DispatchResult {890 Self::check_allowed(collection, spender, from, token, nesting_budget)?;891892 // =========893894 // Allowance is reset in [`transfer`]895 Self::transfer(collection, from, to, token, nesting_budget)896 }897898 pub fn burn_from(899 collection: &NonfungibleHandle<T>,900 spender: &T::CrossAccountId,901 from: &T::CrossAccountId,902 token: TokenId,903 nesting_budget: &dyn Budget,904 ) -> DispatchResult {905 Self::check_allowed(collection, spender, from, token, nesting_budget)?;906907 // =========908909 Self::burn(collection, from, token)910 }911912 pub fn check_nesting(913 handle: &NonfungibleHandle<T>,914 sender: T::CrossAccountId,915 from: (CollectionId, TokenId),916 under: TokenId,917 nesting_budget: &dyn Budget,918 ) -> DispatchResult {919 fn ensure_sender_allowed<T: Config>(920 collection: CollectionId,921 token: TokenId,922 for_nest: (CollectionId, TokenId),923 sender: T::CrossAccountId,924 budget: &dyn Budget,925 ) -> DispatchResult {926 ensure!(927 <PalletStructure<T>>::check_indirectly_owned(928 sender,929 collection,930 token,931 Some(for_nest),932 budget933 )?,934 <CommonError<T>>::OnlyOwnerAllowedToNest,935 );936 Ok(())937 }938 match handle.permissions.nesting() {939 NestingRule::Disabled => fail!(<CommonError<T>>::NestingIsDisabled),940 NestingRule::Owner => {941 ensure_sender_allowed::<T>(handle.id, under, from, sender, nesting_budget)?942 }943 NestingRule::OwnerRestricted(whitelist) => {944 ensure!(945 whitelist.contains(&from.0),946 <CommonError<T>>::SourceCollectionIsNotAllowedToNest947 );948 ensure_sender_allowed::<T>(handle.id, under, from, sender, nesting_budget)?949 }950 }951 Ok(())952 }953954 fn nest(955 under: (CollectionId, TokenId),956 to_nest: (CollectionId, TokenId),957 ) {958 <TokenChildren<T>>::insert(959 (under.0, under.1, (to_nest.0, to_nest.1)),960 true961 );962 }963964 fn unnest(965 under: (CollectionId, TokenId),966 to_unnest: (CollectionId, TokenId),967 ) {968 <TokenChildren<T>>::remove(969 (under.0, under.1, to_unnest)970 );971 }972973 fn collection_has_tokens(collection_id: CollectionId) -> bool {974 <TokenData<T>>::iter_prefix((collection_id,)).next().is_some()975 }976977 fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {978 <TokenChildren<T>>::iter_prefix((collection_id, token_id)).next().is_some()979 }980981 /// Delegated to `create_multiple_items`982 pub fn create_item(983 collection: &NonfungibleHandle<T>,984 sender: &T::CrossAccountId,985 data: CreateItemData<T>,986 nesting_budget: &dyn Budget,987 ) -> DispatchResult {988 Self::create_multiple_items(collection, sender, vec![data], nesting_budget)989 }990}pallets/refungible/src/lib.rsdiffbeforeafterboth--- a/pallets/refungible/src/lib.rs
+++ b/pallets/refungible/src/lib.rs
@@ -210,6 +210,10 @@
) -> DispatchResult {
let id = collection.id;
+ if Self::collection_has_tokens(id) {
+ return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());
+ }
+
// =========
PalletCommon::destroy_collection(collection.0, sender)?;
@@ -225,6 +229,10 @@
Ok(())
}
+ fn collection_has_tokens(collection_id: CollectionId) -> bool {
+ <TokenData<T>>::iter_prefix((collection_id,)).next().is_some()
+ }
+
pub fn burn_token(collection: &RefungibleHandle<T>, token_id: TokenId) -> DispatchResult {
let burnt = <TokensBurnt<T>>::get(collection.id)
.checked_add(1)