12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091#![cfg_attr(not(feature = "std"), no_std)]9293use core::ops::Deref;9495use erc::ERC721Events;96use evm_coder::ToLog;97use frame_support::{98 dispatch::{Pays, PostDispatchInfo},99 ensure, fail,100 pallet_prelude::*,101 storage::with_transaction,102 transactional, BoundedVec,103};104pub use pallet::*;105use pallet_common::{106 eth::collection_id_to_address, helpers::add_weight_to_post_info,107 weights::WeightInfo as CommonWeightInfo, CollectionHandle, Error as CommonError,108 Event as CommonEvent, Pallet as PalletCommon, SelfWeightOf as PalletCommonWeightOf,109};110use pallet_evm::{account::CrossAccountId, Pallet as PalletEvm};111use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};112use pallet_structure::Pallet as PalletStructure;113use parity_scale_codec::{Decode, Encode, MaxEncodedLen};114use scale_info::TypeInfo;115use sp_core::{Get, H160};116use sp_runtime::{ArithmeticError, DispatchError, DispatchResult, TransactionOutcome};117use sp_std::{collections::btree_map::BTreeMap, vec, vec::Vec};118use up_data_structs::{119 budget::Budget, mapping::TokenAddressMapping, AccessMode, AuxPropertyValue, CollectionId,120 CreateCollectionData, CreateNftExData, CustomDataLimit, PropertiesPermissionMap, Property,121 PropertyKey, PropertyKeyPermission, PropertyScope, PropertyValue, TokenChild, TokenId,122 TokenProperties as TokenPropertiesT,123};124use weights::WeightInfo;125#[cfg(feature = "runtime-benchmarks")]126pub mod benchmarking;127pub mod common;128pub mod erc;129pub mod weights;130131pub type CreateItemData<T> = CreateNftExData<<T as pallet_evm::Config>::CrossAccountId>;132pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;133134135136#[struct_versioning::versioned(version = 2, upper)]137#[derive(Encode, Decode, TypeInfo, MaxEncodedLen)]138pub struct ItemData<CrossAccountId> {139 #[version(..2)]140 pub const_data: BoundedVec<u8, CustomDataLimit>,141142 #[version(..2)]143 pub variable_data: BoundedVec<u8, CustomDataLimit>,144145 pub owner: CrossAccountId,146}147148#[frame_support::pallet]149pub mod pallet {150 use frame_support::{151 pallet_prelude::*, storage::Key, traits::StorageVersion, Blake2_128Concat, Twox64Concat,152 };153 use up_data_structs::{CollectionId, TokenId};154155 use super::{weights::WeightInfo, *};156157 #[pallet::error]158 pub enum Error<T> {159 160 NotNonfungibleDataUsedToMintFungibleCollectionToken,161 162 NonfungibleItemsHaveNoAmount,163 164 CantBurnNftWithChildren,165 }166167 #[pallet::config]168 pub trait Config:169 frame_system::Config + pallet_common::Config + pallet_structure::Config + pallet_evm::Config170 {171 type WeightInfo: WeightInfo;172 }173174 const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);175176 #[pallet::pallet]177 #[pallet::storage_version(STORAGE_VERSION)]178 pub struct Pallet<T>(_);179180 181 #[pallet::storage]182 pub type TokensMinted<T: Config> =183 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;184185 186 #[pallet::storage]187 pub type TokensBurnt<T: Config> =188 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;189190 191 #[pallet::storage]192 pub type TokenData<T: Config> = StorageNMap<193 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),194 Value = ItemData<T::CrossAccountId>,195 QueryKind = OptionQuery,196 >;197198 199 #[pallet::storage]200 #[pallet::getter(fn token_properties)]201 pub type TokenProperties<T: Config> = StorageNMap<202 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),203 Value = TokenPropertiesT,204 QueryKind = OptionQuery,205 >;206207 208 209 210 211 212 213 214 215 216 #[pallet::storage]217 #[pallet::getter(fn token_aux_property)]218 pub type TokenAuxProperties<T: Config> = StorageNMap<219 Key = (220 Key<Twox64Concat, CollectionId>,221 Key<Twox64Concat, TokenId>,222 Key<Twox64Concat, PropertyScope>,223 Key<Twox64Concat, PropertyKey>,224 ),225 Value = AuxPropertyValue,226 QueryKind = OptionQuery,227 >;228229 230 #[pallet::storage]231 pub type Owned<T: Config> = StorageNMap<232 Key = (233 Key<Twox64Concat, CollectionId>,234 Key<Blake2_128Concat, T::CrossAccountId>,235 Key<Twox64Concat, TokenId>,236 ),237 Value = bool,238 QueryKind = ValueQuery,239 >;240241 242 #[pallet::storage]243 #[pallet::getter(fn token_children)]244 pub type TokenChildren<T: Config> = StorageNMap<245 Key = (246 Key<Twox64Concat, CollectionId>,247 Key<Twox64Concat, TokenId>,248 Key<Twox64Concat, (CollectionId, TokenId)>,249 ),250 Value = bool,251 QueryKind = ValueQuery,252 >;253254 255 #[pallet::storage]256 pub type AccountBalance<T: Config> = StorageNMap<257 Key = (258 Key<Twox64Concat, CollectionId>,259 Key<Blake2_128Concat, T::CrossAccountId>,260 ),261 Value = u32,262 QueryKind = ValueQuery,263 >;264265 266 #[pallet::storage]267 pub type Allowance<T: Config> = StorageNMap<268 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),269 Value = T::CrossAccountId,270 QueryKind = OptionQuery,271 >;272273 274 #[pallet::storage]275 pub type CollectionAllowance<T: Config> = StorageNMap<276 Key = (277 Key<Twox64Concat, CollectionId>,278 Key<Blake2_128Concat, T::CrossAccountId>,279 Key<Blake2_128Concat, T::CrossAccountId>,280 ),281 Value = bool,282 QueryKind = ValueQuery,283 >;284285 #[pallet::genesis_config]286 pub struct GenesisConfig<T>(PhantomData<T>);287288 impl<T: Config> Default for GenesisConfig<T> {289 fn default() -> Self {290 Self(Default::default())291 }292 }293294 #[pallet::genesis_build]295 impl<T: Config> BuildGenesisConfig for GenesisConfig<T> {296 fn build(&self) {297 StorageVersion::new(1).put::<Pallet<T>>();298 }299 }300}301302pub struct NonfungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);303impl<T: Config> NonfungibleHandle<T> {304 pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {305 Self(inner)306 }307 pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {308 self.0309 }310 pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {311 &mut self.0312 }313}314315impl<T: Config> WithRecorder<T> for NonfungibleHandle<T> {316 fn recorder(&self) -> &SubstrateRecorder<T> {317 self.0.recorder()318 }319 fn into_recorder(self) -> SubstrateRecorder<T> {320 self.0.into_recorder()321 }322}323impl<T: Config> Deref for NonfungibleHandle<T> {324 type Target = pallet_common::CollectionHandle<T>;325326 fn deref(&self) -> &Self::Target {327 &self.0328 }329}330331impl<T: Config> Pallet<T> {332 333 pub fn total_supply(collection: &NonfungibleHandle<T>) -> u32 {334 <TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)335 }336337 338 339 340 pub fn token_exists(collection: &NonfungibleHandle<T>, token: TokenId) -> bool {341 <TokenData<T>>::contains_key((collection.id, token))342 }343344 345 346 347 pub fn try_mutate_token_aux_property<R, E>(348 collection_id: CollectionId,349 token_id: TokenId,350 scope: PropertyScope,351 key: PropertyKey,352 f: impl FnOnce(&mut Option<AuxPropertyValue>) -> Result<R, E>,353 ) -> Result<R, E> {354 <TokenAuxProperties<T>>::try_mutate((collection_id, token_id, scope, key), f)355 }356357 358 pub fn remove_token_aux_property(359 collection_id: CollectionId,360 token_id: TokenId,361 scope: PropertyScope,362 key: PropertyKey,363 ) {364 <TokenAuxProperties<T>>::remove((collection_id, token_id, scope, key));365 }366367 368 369 370 pub fn iterate_token_aux_properties(371 collection_id: CollectionId,372 token_id: TokenId,373 scope: PropertyScope,374 ) -> impl Iterator<Item = (PropertyKey, AuxPropertyValue)> {375 <TokenAuxProperties<T>>::iter_prefix((collection_id, token_id, scope))376 }377378 379 pub fn current_token_id(collection_id: CollectionId) -> TokenId {380 TokenId(<TokensMinted<T>>::get(collection_id))381 }382}383384385impl<T: Config> Pallet<T> {386 387 388 389 390 391 pub fn init_collection(392 owner: T::CrossAccountId,393 payer: T::CrossAccountId,394 data: CreateCollectionData<T::CrossAccountId>,395 ) -> Result<CollectionId, DispatchError> {396 <PalletCommon<T>>::init_collection(owner, payer, data)397 }398399 400 401 402 403 pub fn destroy_collection(404 collection: NonfungibleHandle<T>,405 sender: &T::CrossAccountId,406 ) -> DispatchResult {407 let id = collection.id;408409 if Self::collection_has_tokens(id) {410 return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());411 }412413 414415 PalletCommon::destroy_collection(collection.0, sender)?;416417 let _ = <TokenData<T>>::clear_prefix((id,), u32::MAX, None);418 let _ = <TokenChildren<T>>::clear_prefix((id,), u32::MAX, None);419 let _ = <Owned<T>>::clear_prefix((id,), u32::MAX, None);420 <TokensMinted<T>>::remove(id);421 <TokensBurnt<T>>::remove(id);422 let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);423 let _ = <AccountBalance<T>>::clear_prefix((id,), u32::MAX, None);424 let _ = <CollectionAllowance<T>>::clear_prefix((id,), u32::MAX, None);425 Ok(())426 }427428 429 430 431 432 433 434 435 436 437 pub fn burn(438 collection: &NonfungibleHandle<T>,439 sender: &T::CrossAccountId,440 token: TokenId,441 ) -> DispatchResult {442 let token_data =443 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;444 ensure!(&token_data.owner == sender, <CommonError<T>>::NoPermission);445446 if collection.permissions.access() == AccessMode::AllowList {447 collection.check_allowlist(sender)?;448 }449450 if Self::token_has_children(collection.id, token) {451 return Err(<Error<T>>::CantBurnNftWithChildren.into());452 }453454 let burnt = <TokensBurnt<T>>::get(collection.id)455 .checked_add(1)456 .ok_or(ArithmeticError::Overflow)?;457458 let balance = <AccountBalance<T>>::get((collection.id, token_data.owner.clone()))459 .checked_sub(1)460 .ok_or(ArithmeticError::Overflow)?;461462 463464 if balance == 0 {465 <AccountBalance<T>>::remove((collection.id, token_data.owner.clone()));466 } else {467 <AccountBalance<T>>::insert((collection.id, token_data.owner.clone()), balance);468 }469470 <PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);471472 <Owned<T>>::remove((collection.id, &token_data.owner, token));473 <TokensBurnt<T>>::insert(collection.id, burnt);474 <TokenData<T>>::remove((collection.id, token));475 <TokenProperties<T>>::remove((collection.id, token));476 let _ = <TokenAuxProperties<T>>::clear_prefix((collection.id, token), u32::MAX, None);477 let old_spender = <Allowance<T>>::take((collection.id, token));478479 if let Some(old_spender) = old_spender {480 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(481 collection.id,482 token,483 token_data.owner.clone(),484 old_spender,485 0,486 ));487 }488489 <PalletEvm<T>>::deposit_log(490 ERC721Events::Transfer {491 from: *token_data.owner.as_eth(),492 to: H160::default(),493 token_id: token.into(),494 }495 .to_log(collection_id_to_address(collection.id)),496 );497 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(498 collection.id,499 token,500 token_data.owner,501 1,502 ));503 Ok(())504 }505506 507 508 509 510 511 512 513 514 515 516 517 #[transactional]518 fn modify_token_properties(519 collection: &NonfungibleHandle<T>,520 sender: &T::CrossAccountId,521 token_id: TokenId,522 properties_updates: impl Iterator<Item = (PropertyKey, Option<PropertyValue>)>,523 nesting_budget: &dyn Budget,524 ) -> DispatchResult {525 let mut property_writer =526 pallet_common::ExistingTokenPropertyWriter::new(collection, sender);527528 property_writer.write_token_properties(529 sender,530 token_id,531 properties_updates,532 nesting_budget,533 erc::ERC721TokenEvent::TokenChanged {534 token_id: token_id.into(),535 }536 .to_log(T::ContractAddress::get()),537 )538 }539540 pub fn next_token_id(collection: &NonfungibleHandle<T>) -> Result<TokenId, DispatchError> {541 let next_token_id = <TokensMinted<T>>::get(collection.id)542 .checked_add(1)543 .ok_or(<CommonError<T>>::CollectionTokenLimitExceeded)?;544545 ensure!(546 collection.limits.token_limit() >= next_token_id,547 <CommonError<T>>::CollectionTokenLimitExceeded548 );549550 Ok(TokenId(next_token_id))551 }552553 554 555 556 557 558 pub fn set_token_properties(559 collection: &NonfungibleHandle<T>,560 sender: &T::CrossAccountId,561 token_id: TokenId,562 properties: impl Iterator<Item = Property>,563 nesting_budget: &dyn Budget,564 ) -> DispatchResult {565 Self::modify_token_properties(566 collection,567 sender,568 token_id,569 properties.map(|p| (p.key, Some(p.value))),570 nesting_budget,571 )572 }573574 575 576 577 578 579 pub fn set_token_property(580 collection: &NonfungibleHandle<T>,581 sender: &T::CrossAccountId,582 token_id: TokenId,583 property: Property,584 nesting_budget: &dyn Budget,585 ) -> DispatchResult {586 Self::set_token_properties(587 collection,588 sender,589 token_id,590 [property].into_iter(),591 nesting_budget,592 )593 }594595 596 597 598 599 600 pub fn delete_token_properties(601 collection: &NonfungibleHandle<T>,602 sender: &T::CrossAccountId,603 token_id: TokenId,604 property_keys: impl Iterator<Item = PropertyKey>,605 nesting_budget: &dyn Budget,606 ) -> DispatchResult {607 Self::modify_token_properties(608 collection,609 sender,610 token_id,611 property_keys.into_iter().map(|key| (key, None)),612 nesting_budget,613 )614 }615616 617 618 619 620 621 pub fn delete_token_property(622 collection: &NonfungibleHandle<T>,623 sender: &T::CrossAccountId,624 token_id: TokenId,625 property_key: PropertyKey,626 nesting_budget: &dyn Budget,627 ) -> DispatchResult {628 Self::delete_token_properties(629 collection,630 sender,631 token_id,632 [property_key].into_iter(),633 nesting_budget,634 )635 }636637 638 pub fn set_collection_properties(639 collection: &NonfungibleHandle<T>,640 sender: &T::CrossAccountId,641 properties: Vec<Property>,642 ) -> DispatchResult {643 <PalletCommon<T>>::set_collection_properties(collection, sender, properties.into_iter())644 }645646 647 pub fn delete_collection_properties(648 collection: &CollectionHandle<T>,649 sender: &T::CrossAccountId,650 property_keys: Vec<PropertyKey>,651 ) -> DispatchResult {652 <PalletCommon<T>>::delete_collection_properties(653 collection,654 sender,655 property_keys.into_iter(),656 )657 }658659 660 661 662 pub fn set_token_property_permissions(663 collection: &CollectionHandle<T>,664 sender: &T::CrossAccountId,665 property_permissions: Vec<PropertyKeyPermission>,666 ) -> DispatchResult {667 <PalletCommon<T>>::set_token_property_permissions(collection, sender, property_permissions)668 }669670 671 672 673 pub fn set_scoped_token_property_permissions(674 collection: &CollectionHandle<T>,675 sender: &T::CrossAccountId,676 scope: PropertyScope,677 property_permissions: Vec<PropertyKeyPermission>,678 ) -> DispatchResult {679 <PalletCommon<T>>::set_scoped_token_property_permissions(680 collection,681 sender,682 scope,683 property_permissions,684 )685 }686687 pub fn token_property_permission(collection_id: CollectionId) -> PropertiesPermissionMap {688 <PalletCommon<T>>::property_permissions(collection_id)689 }690691 pub fn check_token_immediate_ownership(692 collection: &NonfungibleHandle<T>,693 token: TokenId,694 possible_owner: &T::CrossAccountId,695 ) -> DispatchResult {696 let token_data =697 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;698 ensure!(699 &token_data.owner == possible_owner,700 <CommonError<T>>::NoPermission701 );702 Ok(())703 }704705 706 707 708 709 710 711 712 713 714 pub fn transfer(715 collection: &NonfungibleHandle<T>,716 from: &T::CrossAccountId,717 to: &T::CrossAccountId,718 token: TokenId,719 nesting_budget: &dyn Budget,720 ) -> DispatchResultWithPostInfo {721 ensure!(722 collection.limits.transfers_enabled(),723 <CommonError<T>>::TransferNotAllowed724 );725726 let mut actual_weight = <SelfWeightOf<T>>::transfer_raw();727 let token_data =728 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;729 ensure!(&token_data.owner == from, <CommonError<T>>::NoPermission);730731 if collection.permissions.access() == AccessMode::AllowList {732 collection.check_allowlist(from)?;733 collection.check_allowlist(to)?;734 actual_weight += <PalletCommonWeightOf<T>>::check_accesslist() * 2;735 }736 <PalletCommon<T>>::ensure_correct_receiver(to)?;737738 let balance_from = <AccountBalance<T>>::get((collection.id, from))739 .checked_sub(1)740 .ok_or(<CommonError<T>>::TokenValueTooLow)?;741 let balance_to = if from != to {742 let balance_to = <AccountBalance<T>>::get((collection.id, to))743 .checked_add(1)744 .ok_or(ArithmeticError::Overflow)?;745746 ensure!(747 balance_to < collection.limits.account_token_ownership_limit(),748 <CommonError<T>>::AccountTokenLimitExceeded,749 );750751 Some(balance_to)752 } else {753 None754 };755756 <PalletStructure<T>>::nest_if_sent_to_token(757 from.clone(),758 to,759 collection.id,760 token,761 nesting_budget,762 )?;763764 765766 <PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);767768 <TokenData<T>>::insert((collection.id, token), ItemData { owner: to.clone() });769770 if let Some(balance_to) = balance_to {771 772 if balance_from == 0 {773 <AccountBalance<T>>::remove((collection.id, from));774 } else {775 <AccountBalance<T>>::insert((collection.id, from), balance_from);776 }777 <AccountBalance<T>>::insert((collection.id, to), balance_to);778 <Owned<T>>::remove((collection.id, from, token));779 <Owned<T>>::insert((collection.id, to, token), true);780 }781 Self::set_allowance_unchecked(collection, from, token, None, true);782783 <PalletEvm<T>>::deposit_log(784 ERC721Events::Transfer {785 from: *from.as_eth(),786 to: *to.as_eth(),787 token_id: token.into(),788 }789 .to_log(collection_id_to_address(collection.id)),790 );791 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(792 collection.id,793 token,794 from.clone(),795 to.clone(),796 1,797 ));798799 Ok(PostDispatchInfo {800 actual_weight: Some(actual_weight),801 pays_fee: Pays::Yes,802 })803 }804805 806 807 808 809 810 811 812 813 814 815 pub fn create_multiple_items(816 collection: &NonfungibleHandle<T>,817 sender: &T::CrossAccountId,818 data: Vec<CreateItemData<T>>,819 nesting_budget: &dyn Budget,820 ) -> DispatchResult {821 if !collection.is_owner_or_admin(sender) {822 ensure!(823 collection.permissions.mint_mode(),824 <CommonError<T>>::PublicMintingNotAllowed825 );826 collection.check_allowlist(sender)?;827828 for item in data.iter() {829 collection.check_allowlist(&item.owner)?;830 }831 }832833 for data in data.iter() {834 <PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;835 }836837 let first_token = <TokensMinted<T>>::get(collection.id);838 let tokens_minted = first_token839 .checked_add(data.len() as u32)840 .ok_or(ArithmeticError::Overflow)?;841 ensure!(842 tokens_minted <= collection.limits.token_limit(),843 <CommonError<T>>::CollectionTokenLimitExceeded844 );845846 let mut balances = BTreeMap::new();847 for data in &data {848 let balance = balances849 .entry(&data.owner)850 .or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));851 *balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;852853 ensure!(854 *balance <= collection.limits.account_token_ownership_limit(),855 <CommonError<T>>::AccountTokenLimitExceeded,856 );857 }858859 for (i, data) in data.iter().enumerate() {860 let token = TokenId(first_token + i as u32 + 1);861862 <PalletStructure<T>>::check_nesting(863 sender.clone(),864 &data.owner,865 collection.id,866 token,867 nesting_budget,868 )?;869 }870871 872873 let mut property_writer = pallet_common::NewTokenPropertyWriter::new(collection, sender);874875 with_transaction(|| {876 for (i, data) in data.iter().enumerate() {877 let token = first_token + i as u32 + 1;878879 <TokenData<T>>::insert(880 (collection.id, token),881 ItemData {882 883 owner: data.owner.clone(),884 },885 );886887 let token = TokenId(token);888889 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(890 &data.owner,891 collection.id,892 token,893 );894895 if let Err(e) = property_writer.write_token_properties(896 sender.conv_eq(&data.owner),897 token,898 data.properties.clone().into_iter(),899 erc::ERC721TokenEvent::TokenChanged {900 token_id: token.into(),901 }902 .to_log(T::ContractAddress::get()),903 ) {904 return TransactionOutcome::Rollback(Err(e));905 }906 }907 TransactionOutcome::Commit(Ok(()))908 })?;909910 <TokensMinted<T>>::insert(collection.id, tokens_minted);911 for (account, balance) in balances {912 <AccountBalance<T>>::insert((collection.id, account), balance);913 }914 for (i, data) in data.into_iter().enumerate() {915 let token = first_token + i as u32 + 1;916 <Owned<T>>::insert((collection.id, &data.owner, token), true);917918 <PalletEvm<T>>::deposit_log(919 ERC721Events::Transfer {920 from: H160::default(),921 to: *data.owner.as_eth(),922 token_id: token.into(),923 }924 .to_log(collection_id_to_address(collection.id)),925 );926 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(927 collection.id,928 TokenId(token),929 data.owner.clone(),930 1,931 ));932 }933 Ok(())934 }935936 pub fn set_allowance_unchecked(937 collection: &NonfungibleHandle<T>,938 sender: &T::CrossAccountId,939 token: TokenId,940 spender: Option<&T::CrossAccountId>,941 assume_implicit_eth: bool,942 ) {943 if let Some(spender) = spender {944 let old_spender = <Allowance<T>>::get((collection.id, token));945 <Allowance<T>>::insert((collection.id, token), spender);946 947 948 <PalletEvm<T>>::deposit_log(949 ERC721Events::Approval {950 owner: *sender.as_eth(),951 approved: *spender.as_eth(),952 token_id: token.into(),953 }954 .to_log(collection_id_to_address(collection.id)),955 );956 957 958 if old_spender.as_ref() != Some(spender) {959 if let Some(old_owner) = old_spender {960 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(961 collection.id,962 token,963 sender.clone(),964 old_owner,965 0,966 ));967 }968 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(969 collection.id,970 token,971 sender.clone(),972 spender.clone(),973 1,974 ));975 }976 } else {977 let old_spender = <Allowance<T>>::take((collection.id, token));978 if !assume_implicit_eth {979 980 981 <PalletEvm<T>>::deposit_log(982 ERC721Events::Approval {983 owner: *sender.as_eth(),984 approved: H160::default(),985 token_id: token.into(),986 }987 .to_log(collection_id_to_address(collection.id)),988 );989 }990 991 992 if let Some(old_spender) = old_spender {993 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(994 collection.id,995 token,996 sender.clone(),997 old_spender,998 0,999 ));1000 }1001 }1002 }10031004 pub fn get_allowance(1005 collection: &NonfungibleHandle<T>,1006 token_id: TokenId,1007 ) -> Result<Option<T::CrossAccountId>, DispatchError> {1008 ensure!(1009 <TokenData<T>>::get((collection.id, token_id)).is_some(),1010 <CommonError<T>>::TokenNotFound1011 );1012 Ok(<Allowance<T>>::get((collection.id, token_id)))1013 }10141015 1016 1017 1018 pub fn set_allowance(1019 collection: &NonfungibleHandle<T>,1020 sender: &T::CrossAccountId,1021 token: TokenId,1022 spender: Option<&T::CrossAccountId>,1023 ) -> DispatchResult {1024 if collection.permissions.access() == AccessMode::AllowList {1025 collection.check_allowlist(sender)?;1026 if let Some(spender) = spender {1027 collection.check_allowlist(spender)?;1028 }1029 }10301031 if let Some(spender) = spender {1032 <PalletCommon<T>>::ensure_correct_receiver(spender)?;1033 }10341035 let token_data =1036 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;1037 if &token_data.owner != sender {1038 ensure!(1039 collection.ignores_owned_amount(sender),1040 <CommonError<T>>::CantApproveMoreThanOwned1041 );1042 }10431044 10451046 Self::set_allowance_unchecked(collection, sender, token, spender, false);1047 Ok(())1048 }10491050 1051 1052 1053 1054 1055 pub fn set_allowance_from(1056 collection: &NonfungibleHandle<T>,1057 sender: &T::CrossAccountId,1058 from: &T::CrossAccountId,1059 token: TokenId,1060 to: Option<&T::CrossAccountId>,1061 ) -> DispatchResult {1062 if collection.permissions.access() == AccessMode::AllowList {1063 collection.check_allowlist(sender)?;1064 collection.check_allowlist(from)?;1065 if let Some(to) = to {1066 collection.check_allowlist(to)?;1067 }1068 }10691070 if let Some(to) = to {1071 <PalletCommon<T>>::ensure_correct_receiver(to)?;1072 }10731074 ensure!(1075 sender.conv_eq(from),1076 <CommonError<T>>::AddressIsNotEthMirror1077 );10781079 let token_data =1080 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;1081 if token_data.owner != *from {1082 ensure!(1083 collection.limits.owner_can_transfer()1084 && (collection.is_owner_or_admin(sender) || collection.is_owner_or_admin(from)),1085 <CommonError<T>>::CantApproveMoreThanOwned1086 );1087 }10881089 10901091 Self::set_allowance_unchecked(collection, from, token, to, false);1092 Ok(())1093 }10941095 1096 fn check_allowed(1097 collection: &NonfungibleHandle<T>,1098 spender: &T::CrossAccountId,1099 from: &T::CrossAccountId,1100 token: TokenId,1101 nesting_budget: &dyn Budget,1102 ) -> DispatchResult {1103 if spender.conv_eq(from) {1104 return Ok(());1105 }1106 if collection.permissions.access() == AccessMode::AllowList {1107 1108 collection.check_allowlist(spender)?;1109 }11101111 if collection.ignores_token_restrictions(spender) {1112 return Ok(());1113 }11141115 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {1116 ensure!(1117 <PalletStructure<T>>::check_indirectly_owned(1118 spender.clone(),1119 source.0,1120 source.1,1121 None,1122 nesting_budget1123 )?,1124 <CommonError<T>>::ApprovedValueTooLow,1125 );1126 return Ok(());1127 }1128 if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {1129 return Ok(());1130 }1131 if <CollectionAllowance<T>>::get((collection.id, from, spender)) {1132 return Ok(());1133 }11341135 Err(<CommonError<T>>::ApprovedValueTooLow.into())1136 }11371138 1139 1140 1141 1142 1143 1144 pub fn transfer_from(1145 collection: &NonfungibleHandle<T>,1146 spender: &T::CrossAccountId,1147 from: &T::CrossAccountId,1148 to: &T::CrossAccountId,1149 token: TokenId,1150 nesting_budget: &dyn Budget,1151 ) -> DispatchResultWithPostInfo {1152 Self::check_allowed(collection, spender, from, token, nesting_budget)?;11531154 11551156 1157 let mut result = Self::transfer(collection, from, to, token, nesting_budget);1158 add_weight_to_post_info(&mut result, <SelfWeightOf<T>>::check_allowed_raw());1159 result1160 }11611162 1163 1164 1165 1166 1167 1168 pub fn burn_from(1169 collection: &NonfungibleHandle<T>,1170 spender: &T::CrossAccountId,1171 from: &T::CrossAccountId,1172 token: TokenId,1173 nesting_budget: &dyn Budget,1174 ) -> DispatchResult {1175 Self::check_allowed(collection, spender, from, token, nesting_budget)?;11761177 11781179 Self::burn(collection, from, token)1180 }11811182 1183 1184 pub fn check_nesting(1185 handle: &NonfungibleHandle<T>,1186 sender: T::CrossAccountId,1187 from: (CollectionId, TokenId),1188 under: TokenId,1189 nesting_budget: &dyn Budget,1190 ) -> DispatchResult {1191 let nesting = handle.permissions.nesting();11921193 #[cfg(not(feature = "runtime-benchmarks"))]1194 let permissive = false;1195 #[cfg(feature = "runtime-benchmarks")]1196 let permissive = nesting.permissive;11971198 if permissive {1199 ensure!(1200 <TokenData<T>>::contains_key((handle.id, under)),1201 <CommonError<T>>::TokenNotFound1202 );1203 } else if nesting.token_owner1204 && <PalletStructure<T>>::check_indirectly_owned(1205 sender.clone(),1206 handle.id,1207 under,1208 Some(from),1209 nesting_budget,1210 )? {1211 1212 } else if nesting.collection_admin && handle.is_owner_or_admin(&sender) {1213 1214 let _ = <PalletStructure<T>>::get_checked_topmost_owner(1215 handle.id,1216 under,1217 Some(from),1218 nesting_budget,1219 )?1220 .ok_or(<CommonError<T>>::TokenNotFound)?;1221 } else {1222 fail!(<CommonError<T>>::UserIsNotAllowedToNest);1223 }12241225 if let Some(whitelist) = &nesting.restricted {1226 ensure!(1227 whitelist.contains(&from.0),1228 <CommonError<T>>::SourceCollectionIsNotAllowedToNest1229 );1230 }1231 Ok(())1232 }12331234 fn nest(under: (CollectionId, TokenId), to_nest: (CollectionId, TokenId)) {1235 if to_nest.0 != pallet_common::NATIVE_FUNGIBLE_COLLECTION_ID {1236 <TokenChildren<T>>::insert((under.0, under.1, to_nest), true);1237 }1238 }12391240 fn unnest(under: (CollectionId, TokenId), to_unnest: (CollectionId, TokenId)) {1241 if to_unnest.0 != pallet_common::NATIVE_FUNGIBLE_COLLECTION_ID {1242 <TokenChildren<T>>::remove((under.0, under.1, to_unnest));1243 }1244 }12451246 fn collection_has_tokens(collection_id: CollectionId) -> bool {1247 <TokenData<T>>::iter_prefix((collection_id,))1248 .next()1249 .is_some()1250 }12511252 fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {1253 <TokenChildren<T>>::iter_prefix((collection_id, token_id))1254 .next()1255 .is_some()1256 }12571258 pub fn token_children_ids(collection_id: CollectionId, token_id: TokenId) -> Vec<TokenChild> {1259 <TokenChildren<T>>::iter_prefix((collection_id, token_id))1260 .map(|((child_collection_id, child_id), _)| TokenChild {1261 collection: child_collection_id,1262 token: child_id,1263 })1264 .collect()1265 }12661267 1268 1269 1270 1271 1272 pub fn create_item(1273 collection: &NonfungibleHandle<T>,1274 sender: &T::CrossAccountId,1275 data: CreateItemData<T>,1276 nesting_budget: &dyn Budget,1277 ) -> DispatchResult {1278 Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1279 }12801281 1282 1283 1284 1285 1286 1287 pub fn set_allowance_for_all(1288 collection: &NonfungibleHandle<T>,1289 owner: &T::CrossAccountId,1290 operator: &T::CrossAccountId,1291 approve: bool,1292 ) -> DispatchResult {1293 <PalletCommon<T>>::set_allowance_for_all(1294 collection,1295 owner,1296 operator,1297 approve,1298 || <CollectionAllowance<T>>::insert((collection.id, owner, operator), approve),1299 ERC721Events::ApprovalForAll {1300 owner: *owner.as_eth(),1301 operator: *operator.as_eth(),1302 approved: approve,1303 }1304 .to_log(collection_id_to_address(collection.id)),1305 )1306 }13071308 1309 pub fn allowance_for_all(1310 collection: &NonfungibleHandle<T>,1311 owner: &T::CrossAccountId,1312 operator: &T::CrossAccountId,1313 ) -> bool {1314 <CollectionAllowance<T>>::get((collection.id, owner, operator))1315 }13161317 pub fn repair_item(collection: &NonfungibleHandle<T>, token: TokenId) -> DispatchResult {1318 <TokenProperties<T>>::mutate((collection.id, token), |properties| {1319 if let Some(properties) = properties {1320 properties.recompute_consumed_space();1321 }1322 });13231324 Ok(())1325 }1326}