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 let depositor = from;722 Self::transfer_internal(collection, depositor, from, to, token, nesting_budget)723 }724725 726 727 728 pub fn transfer_internal(729 collection: &NonfungibleHandle<T>,730 depositor: &T::CrossAccountId,731 from: &T::CrossAccountId,732 to: &T::CrossAccountId,733 token: TokenId,734 nesting_budget: &dyn Budget,735 ) -> DispatchResultWithPostInfo {736 ensure!(737 collection.limits.transfers_enabled(),738 <CommonError<T>>::TransferNotAllowed739 );740741 let mut actual_weight = <SelfWeightOf<T>>::transfer_raw();742 let token_data =743 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;744 ensure!(&token_data.owner == from, <CommonError<T>>::NoPermission);745746 if collection.permissions.access() == AccessMode::AllowList {747 collection.check_allowlist(from)?;748 collection.check_allowlist(to)?;749 actual_weight += <PalletCommonWeightOf<T>>::check_accesslist() * 2;750 }751 <PalletCommon<T>>::ensure_correct_receiver(to)?;752753 let balance_from = <AccountBalance<T>>::get((collection.id, from))754 .checked_sub(1)755 .ok_or(<CommonError<T>>::TokenValueTooLow)?;756 let balance_to = if from != to {757 let balance_to = <AccountBalance<T>>::get((collection.id, to))758 .checked_add(1)759 .ok_or(ArithmeticError::Overflow)?;760761 ensure!(762 balance_to < collection.limits.account_token_ownership_limit(),763 <CommonError<T>>::AccountTokenLimitExceeded,764 );765766 Some(balance_to)767 } else {768 None769 };770771 <PalletStructure<T>>::nest_if_sent_to_token(772 depositor,773 to,774 collection.id,775 token,776 nesting_budget,777 )?;778779 780781 <PalletStructure<T>>::unnest_if_nested(&token_data.owner, collection.id, token);782783 <TokenData<T>>::insert((collection.id, token), ItemData { owner: to.clone() });784785 if let Some(balance_to) = balance_to {786 787 if balance_from == 0 {788 <AccountBalance<T>>::remove((collection.id, from));789 } else {790 <AccountBalance<T>>::insert((collection.id, from), balance_from);791 }792 <AccountBalance<T>>::insert((collection.id, to), balance_to);793 <Owned<T>>::remove((collection.id, from, token));794 <Owned<T>>::insert((collection.id, to, token), true);795 }796 Self::set_allowance_unchecked(collection, from, token, None, true);797798 <PalletEvm<T>>::deposit_log(799 ERC721Events::Transfer {800 from: *from.as_eth(),801 to: *to.as_eth(),802 token_id: token.into(),803 }804 .to_log(collection_id_to_address(collection.id)),805 );806 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(807 collection.id,808 token,809 from.clone(),810 to.clone(),811 1,812 ));813814 Ok(PostDispatchInfo {815 actual_weight: Some(actual_weight),816 pays_fee: Pays::Yes,817 })818 }819820 821 822 823 824 825 826 827 828 829 830 pub fn create_multiple_items(831 collection: &NonfungibleHandle<T>,832 sender: &T::CrossAccountId,833 data: Vec<CreateItemData<T>>,834 nesting_budget: &dyn Budget,835 ) -> DispatchResult {836 if !collection.is_owner_or_admin(sender) {837 ensure!(838 collection.permissions.mint_mode(),839 <CommonError<T>>::PublicMintingNotAllowed840 );841 collection.check_allowlist(sender)?;842843 for item in data.iter() {844 collection.check_allowlist(&item.owner)?;845 }846 }847848 for data in data.iter() {849 <PalletCommon<T>>::ensure_correct_receiver(&data.owner)?;850 }851852 let first_token = <TokensMinted<T>>::get(collection.id);853 let tokens_minted = first_token854 .checked_add(data.len() as u32)855 .ok_or(ArithmeticError::Overflow)?;856 ensure!(857 tokens_minted <= collection.limits.token_limit(),858 <CommonError<T>>::CollectionTokenLimitExceeded859 );860861 let mut balances = BTreeMap::new();862 for data in &data {863 let balance = balances864 .entry(&data.owner)865 .or_insert_with(|| <AccountBalance<T>>::get((collection.id, &data.owner)));866 *balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;867868 ensure!(869 *balance <= collection.limits.account_token_ownership_limit(),870 <CommonError<T>>::AccountTokenLimitExceeded,871 );872 }873874 for (i, data) in data.iter().enumerate() {875 let token = TokenId(first_token + i as u32 + 1);876877 <PalletStructure<T>>::check_nesting(878 sender,879 &data.owner,880 collection.id,881 token,882 nesting_budget,883 )?;884 }885886 887888 let mut property_writer = pallet_common::NewTokenPropertyWriter::new(collection, sender);889890 with_transaction(|| {891 for (i, data) in data.iter().enumerate() {892 let token = first_token + i as u32 + 1;893894 <TokenData<T>>::insert(895 (collection.id, token),896 ItemData {897 898 owner: data.owner.clone(),899 },900 );901902 let token = TokenId(token);903904 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(905 &data.owner,906 collection.id,907 token,908 );909910 if let Err(e) = property_writer.write_token_properties(911 sender.conv_eq(&data.owner),912 token,913 data.properties.clone().into_iter(),914 erc::ERC721TokenEvent::TokenChanged {915 token_id: token.into(),916 }917 .to_log(T::ContractAddress::get()),918 ) {919 return TransactionOutcome::Rollback(Err(e));920 }921 }922 TransactionOutcome::Commit(Ok(()))923 })?;924925 <TokensMinted<T>>::insert(collection.id, tokens_minted);926 for (account, balance) in balances {927 <AccountBalance<T>>::insert((collection.id, account), balance);928 }929 for (i, data) in data.into_iter().enumerate() {930 let token = first_token + i as u32 + 1;931 <Owned<T>>::insert((collection.id, &data.owner, token), true);932933 <PalletEvm<T>>::deposit_log(934 ERC721Events::Transfer {935 from: H160::default(),936 to: *data.owner.as_eth(),937 token_id: token.into(),938 }939 .to_log(collection_id_to_address(collection.id)),940 );941 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(942 collection.id,943 TokenId(token),944 data.owner.clone(),945 1,946 ));947 }948 Ok(())949 }950951 pub fn set_allowance_unchecked(952 collection: &NonfungibleHandle<T>,953 sender: &T::CrossAccountId,954 token: TokenId,955 spender: Option<&T::CrossAccountId>,956 assume_implicit_eth: bool,957 ) {958 if let Some(spender) = spender {959 let old_spender = <Allowance<T>>::get((collection.id, token));960 <Allowance<T>>::insert((collection.id, token), spender);961 962 963 <PalletEvm<T>>::deposit_log(964 ERC721Events::Approval {965 owner: *sender.as_eth(),966 approved: *spender.as_eth(),967 token_id: token.into(),968 }969 .to_log(collection_id_to_address(collection.id)),970 );971 972 973 if old_spender.as_ref() != Some(spender) {974 if let Some(old_owner) = old_spender {975 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(976 collection.id,977 token,978 sender.clone(),979 old_owner,980 0,981 ));982 }983 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(984 collection.id,985 token,986 sender.clone(),987 spender.clone(),988 1,989 ));990 }991 } else {992 let old_spender = <Allowance<T>>::take((collection.id, token));993 if !assume_implicit_eth {994 995 996 <PalletEvm<T>>::deposit_log(997 ERC721Events::Approval {998 owner: *sender.as_eth(),999 approved: H160::default(),1000 token_id: token.into(),1001 }1002 .to_log(collection_id_to_address(collection.id)),1003 );1004 }1005 1006 1007 if let Some(old_spender) = old_spender {1008 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(1009 collection.id,1010 token,1011 sender.clone(),1012 old_spender,1013 0,1014 ));1015 }1016 }1017 }10181019 pub fn get_allowance(1020 collection: &NonfungibleHandle<T>,1021 token_id: TokenId,1022 ) -> Result<Option<T::CrossAccountId>, DispatchError> {1023 ensure!(1024 <TokenData<T>>::get((collection.id, token_id)).is_some(),1025 <CommonError<T>>::TokenNotFound1026 );1027 Ok(<Allowance<T>>::get((collection.id, token_id)))1028 }10291030 1031 1032 1033 pub fn set_allowance(1034 collection: &NonfungibleHandle<T>,1035 sender: &T::CrossAccountId,1036 token: TokenId,1037 spender: Option<&T::CrossAccountId>,1038 ) -> DispatchResult {1039 if collection.permissions.access() == AccessMode::AllowList {1040 collection.check_allowlist(sender)?;1041 if let Some(spender) = spender {1042 collection.check_allowlist(spender)?;1043 }1044 }10451046 if let Some(spender) = spender {1047 <PalletCommon<T>>::ensure_correct_receiver(spender)?;1048 }10491050 let token_data =1051 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;1052 if &token_data.owner != sender {1053 ensure!(1054 collection.ignores_owned_amount(sender),1055 <CommonError<T>>::CantApproveMoreThanOwned1056 );1057 }10581059 10601061 Self::set_allowance_unchecked(collection, sender, token, spender, false);1062 Ok(())1063 }10641065 1066 1067 1068 1069 1070 pub fn set_allowance_from(1071 collection: &NonfungibleHandle<T>,1072 sender: &T::CrossAccountId,1073 from: &T::CrossAccountId,1074 token: TokenId,1075 to: Option<&T::CrossAccountId>,1076 ) -> DispatchResult {1077 if collection.permissions.access() == AccessMode::AllowList {1078 collection.check_allowlist(sender)?;1079 collection.check_allowlist(from)?;1080 if let Some(to) = to {1081 collection.check_allowlist(to)?;1082 }1083 }10841085 if let Some(to) = to {1086 <PalletCommon<T>>::ensure_correct_receiver(to)?;1087 }10881089 ensure!(1090 sender.conv_eq(from),1091 <CommonError<T>>::AddressIsNotEthMirror1092 );10931094 let token_data =1095 <TokenData<T>>::get((collection.id, token)).ok_or(<CommonError<T>>::TokenNotFound)?;1096 if token_data.owner != *from {1097 ensure!(1098 collection.limits.owner_can_transfer()1099 && (collection.is_owner_or_admin(sender) || collection.is_owner_or_admin(from)),1100 <CommonError<T>>::CantApproveMoreThanOwned1101 );1102 }11031104 11051106 Self::set_allowance_unchecked(collection, from, token, to, false);1107 Ok(())1108 }11091110 1111 fn check_allowed(1112 collection: &NonfungibleHandle<T>,1113 spender: &T::CrossAccountId,1114 from: &T::CrossAccountId,1115 token: TokenId,1116 nesting_budget: &dyn Budget,1117 ) -> DispatchResult {1118 if spender.conv_eq(from) {1119 return Ok(());1120 }1121 if collection.permissions.access() == AccessMode::AllowList {1122 1123 collection.check_allowlist(spender)?;1124 }11251126 if collection.ignores_token_restrictions(spender) {1127 return Ok(());1128 }11291130 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {1131 ensure!(1132 <PalletStructure<T>>::check_indirectly_owned(1133 spender.clone(),1134 source.0,1135 source.1,1136 None,1137 nesting_budget1138 )?,1139 <CommonError<T>>::ApprovedValueTooLow,1140 );1141 return Ok(());1142 }1143 if <Allowance<T>>::get((collection.id, token)).as_ref() == Some(spender) {1144 return Ok(());1145 }1146 if <CollectionAllowance<T>>::get((collection.id, from, spender)) {1147 return Ok(());1148 }11491150 Err(<CommonError<T>>::ApprovedValueTooLow.into())1151 }11521153 1154 1155 1156 1157 1158 1159 pub fn transfer_from(1160 collection: &NonfungibleHandle<T>,1161 spender: &T::CrossAccountId,1162 from: &T::CrossAccountId,1163 to: &T::CrossAccountId,1164 token: TokenId,1165 nesting_budget: &dyn Budget,1166 ) -> DispatchResultWithPostInfo {1167 Self::check_allowed(collection, spender, from, token, nesting_budget)?;11681169 11701171 1172 let mut result =1173 Self::transfer_internal(collection, spender, from, to, token, nesting_budget);1174 add_weight_to_post_info(&mut result, <SelfWeightOf<T>>::check_allowed_raw());1175 result1176 }11771178 1179 1180 1181 1182 1183 1184 pub fn burn_from(1185 collection: &NonfungibleHandle<T>,1186 spender: &T::CrossAccountId,1187 from: &T::CrossAccountId,1188 token: TokenId,1189 nesting_budget: &dyn Budget,1190 ) -> DispatchResult {1191 Self::check_allowed(collection, spender, from, token, nesting_budget)?;11921193 11941195 Self::burn(collection, from, token)1196 }11971198 1199 1200 pub fn check_nesting(1201 handle: &NonfungibleHandle<T>,1202 sender: &T::CrossAccountId,1203 from: (CollectionId, TokenId),1204 under: TokenId,1205 nesting_budget: &dyn Budget,1206 ) -> DispatchResult {1207 let nesting = handle.permissions.nesting();12081209 #[cfg(not(feature = "runtime-benchmarks"))]1210 let permissive = false;1211 #[cfg(feature = "runtime-benchmarks")]1212 let permissive = nesting.permissive;12131214 if permissive {1215 ensure!(1216 <TokenData<T>>::contains_key((handle.id, under)),1217 <CommonError<T>>::TokenNotFound1218 );1219 } else if nesting.token_owner1220 && <PalletStructure<T>>::check_indirectly_owned(1221 sender.clone(),1222 handle.id,1223 under,1224 Some(from),1225 nesting_budget,1226 )? {1227 1228 } else if nesting.collection_admin && handle.is_owner_or_admin(sender) {1229 1230 let _ = <PalletStructure<T>>::get_checked_topmost_owner(1231 handle.id,1232 under,1233 Some(from),1234 nesting_budget,1235 )?1236 .ok_or(<CommonError<T>>::TokenNotFound)?;1237 } else {1238 fail!(<CommonError<T>>::UserIsNotAllowedToNest);1239 }12401241 if let Some(whitelist) = &nesting.restricted {1242 ensure!(1243 whitelist.contains(&from.0),1244 <CommonError<T>>::SourceCollectionIsNotAllowedToNest1245 );1246 }1247 Ok(())1248 }12491250 fn nest(under: (CollectionId, TokenId), to_nest: (CollectionId, TokenId)) {1251 if to_nest.0 != pallet_common::NATIVE_FUNGIBLE_COLLECTION_ID {1252 <TokenChildren<T>>::insert((under.0, under.1, to_nest), true);1253 }1254 }12551256 fn unnest(under: (CollectionId, TokenId), to_unnest: (CollectionId, TokenId)) {1257 if to_unnest.0 != pallet_common::NATIVE_FUNGIBLE_COLLECTION_ID {1258 <TokenChildren<T>>::remove((under.0, under.1, to_unnest));1259 }1260 }12611262 fn collection_has_tokens(collection_id: CollectionId) -> bool {1263 <TokenData<T>>::iter_prefix((collection_id,))1264 .next()1265 .is_some()1266 }12671268 fn token_has_children(collection_id: CollectionId, token_id: TokenId) -> bool {1269 <TokenChildren<T>>::iter_prefix((collection_id, token_id))1270 .next()1271 .is_some()1272 }12731274 pub fn token_children_ids(collection_id: CollectionId, token_id: TokenId) -> Vec<TokenChild> {1275 <TokenChildren<T>>::iter_prefix((collection_id, token_id))1276 .map(|((child_collection_id, child_id), _)| TokenChild {1277 collection: child_collection_id,1278 token: child_id,1279 })1280 .collect()1281 }12821283 1284 1285 1286 1287 1288 pub fn create_item(1289 collection: &NonfungibleHandle<T>,1290 sender: &T::CrossAccountId,1291 data: CreateItemData<T>,1292 nesting_budget: &dyn Budget,1293 ) -> DispatchResult {1294 Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1295 }12961297 1298 1299 1300 1301 1302 1303 pub fn set_allowance_for_all(1304 collection: &NonfungibleHandle<T>,1305 owner: &T::CrossAccountId,1306 operator: &T::CrossAccountId,1307 approve: bool,1308 ) -> DispatchResult {1309 <PalletCommon<T>>::set_allowance_for_all(1310 collection,1311 owner,1312 operator,1313 approve,1314 || <CollectionAllowance<T>>::insert((collection.id, owner, operator), approve),1315 ERC721Events::ApprovalForAll {1316 owner: *owner.as_eth(),1317 operator: *operator.as_eth(),1318 approved: approve,1319 }1320 .to_log(collection_id_to_address(collection.id)),1321 )1322 }13231324 1325 pub fn allowance_for_all(1326 collection: &NonfungibleHandle<T>,1327 owner: &T::CrossAccountId,1328 operator: &T::CrossAccountId,1329 ) -> bool {1330 <CollectionAllowance<T>>::get((collection.id, owner, operator))1331 }13321333 pub fn repair_item(collection: &NonfungibleHandle<T>, token: TokenId) -> DispatchResult {1334 <TokenProperties<T>>::mutate((collection.id, token), |properties| {1335 if let Some(properties) = properties {1336 properties.recompute_consumed_space();1337 }1338 });13391340 Ok(())1341 }1342}