12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788#![cfg_attr(not(feature = "std"), no_std)]8990use crate::erc_token::ERC20Events;91use crate::erc::ERC721Events;9293use codec::{Encode, Decode, MaxEncodedLen};94use core::ops::Deref;95use derivative::Derivative;96use evm_coder::ToLog;97use frame_support::{98 BoundedBTreeMap, BoundedVec, ensure, fail, storage::with_transaction, transactional,99 pallet_prelude::ConstU32,100};101use pallet_evm::{account::CrossAccountId, Pallet as PalletEvm};102use pallet_evm_coder_substrate::WithRecorder;103use pallet_common::{104 CommonCollectionOperations, Error as CommonError, eth::collection_id_to_address,105 Event as CommonEvent, Pallet as PalletCommon,106};107use pallet_structure::Pallet as PalletStructure;108use scale_info::TypeInfo;109use sp_core::H160;110use sp_runtime::{ArithmeticError, DispatchError, DispatchResult, TransactionOutcome};111use sp_std::{vec::Vec, vec, collections::btree_map::BTreeMap};112use up_data_structs::{113 AccessMode, budget::Budget, CollectionId, CollectionFlags, CollectionPropertiesVec,114 CreateCollectionData, CustomDataLimit, mapping::TokenAddressMapping, MAX_ITEMS_PER_BATCH,115 MAX_REFUNGIBLE_PIECES, Property, PropertyKey, PropertyKeyPermission, PropertyPermission,116 PropertyScope, PropertyValue, TokenId, TrySetProperty,117};118119pub use pallet::*;120#[cfg(feature = "runtime-benchmarks")]121pub mod benchmarking;122pub mod common;123pub mod erc;124pub mod erc_token;125pub mod weights;126127#[derive(Derivative, Clone)]128pub struct CreateItemData<CrossAccountId> {129 #[derivative(Debug(format_with = "bounded::map_debug"))]130 pub users: BoundedBTreeMap<CrossAccountId, u128, ConstU32<MAX_ITEMS_PER_BATCH>>,131 #[derivative(Debug(format_with = "bounded::vec_debug"))]132 pub properties: CollectionPropertiesVec,133}134pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;135136137138#[struct_versioning::versioned(version = 2, upper)]139#[derive(Encode, Decode, Default, TypeInfo, MaxEncodedLen)]140pub struct ItemData {141 pub const_data: BoundedVec<u8, CustomDataLimit>,142143 #[version(..2)]144 pub variable_data: BoundedVec<u8, CustomDataLimit>,145}146147#[frame_support::pallet]148pub mod pallet {149 use super::*;150 use frame_support::{151 Blake2_128, Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key,152 traits::StorageVersion,153 };154 use frame_system::pallet_prelude::*;155 use up_data_structs::{CollectionId, TokenId};156 use super::weights::WeightInfo;157158 #[pallet::error]159 pub enum Error<T> {160 161 NotRefungibleDataUsedToMintFungibleCollectionToken,162 163 WrongRefungiblePieces,164 165 RepartitionWhileNotOwningAllPieces,166 167 RefungibleDisallowsNesting,168 169 SettingPropertiesNotAllowed,170 }171172 #[pallet::config]173 pub trait Config:174 frame_system::Config + pallet_common::Config + pallet_structure::Config175 {176 type WeightInfo: WeightInfo;177 }178179 const STORAGE_VERSION: StorageVersion = StorageVersion::new(2);180181 #[pallet::pallet]182 #[pallet::storage_version(STORAGE_VERSION)]183 #[pallet::generate_store(pub(super) trait Store)]184 pub struct Pallet<T>(_);185186 187 #[pallet::storage]188 pub type TokensMinted<T: Config> =189 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;190191 192 #[pallet::storage]193 pub type TokensBurnt<T: Config> =194 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;195196 197 198 #[pallet::storage]199 #[deprecated(since = "0.2.0", note = "ItemData is no more contains usefull data")]200 pub type TokenData<T: Config> = StorageNMap<201 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),202 Value = ItemData,203 QueryKind = ValueQuery,204 >;205206 207 #[pallet::storage]208 #[pallet::getter(fn token_properties)]209 pub type TokenProperties<T: Config> = StorageNMap<210 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),211 Value = up_data_structs::Properties,212 QueryKind = ValueQuery,213 OnEmpty = up_data_structs::TokenProperties,214 >;215216 217 #[pallet::storage]218 pub type TotalSupply<T: Config> = StorageNMap<219 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),220 Value = u128,221 QueryKind = ValueQuery,222 >;223224 225 #[pallet::storage]226 pub type Owned<T: Config> = StorageNMap<227 Key = (228 Key<Twox64Concat, CollectionId>,229 Key<Blake2_128Concat, T::CrossAccountId>,230 Key<Twox64Concat, TokenId>,231 ),232 Value = bool,233 QueryKind = ValueQuery,234 >;235236 237 #[pallet::storage]238 pub type AccountBalance<T: Config> = StorageNMap<239 Key = (240 Key<Twox64Concat, CollectionId>,241 242 Key<Blake2_128Concat, T::CrossAccountId>,243 ),244 Value = u32,245 QueryKind = ValueQuery,246 >;247248 249 #[pallet::storage]250 pub type Balance<T: Config> = StorageNMap<251 Key = (252 Key<Twox64Concat, CollectionId>,253 Key<Twox64Concat, TokenId>,254 255 Key<Blake2_128Concat, T::CrossAccountId>,256 ),257 Value = u128,258 QueryKind = ValueQuery,259 >;260261 262 #[pallet::storage]263 pub type Allowance<T: Config> = StorageNMap<264 Key = (265 Key<Twox64Concat, CollectionId>,266 Key<Twox64Concat, TokenId>,267 268 Key<Blake2_128, T::CrossAccountId>,269 270 Key<Blake2_128Concat, T::CrossAccountId>,271 ),272 Value = u128,273 QueryKind = ValueQuery,274 >;275276 #[pallet::hooks]277 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {278 fn on_runtime_upgrade() -> Weight {279 let storage_version = StorageVersion::get::<Pallet<T>>();280 if storage_version < StorageVersion::new(2) {281 #[allow(deprecated)]282 let _ = <TokenData<T>>::clear(u32::MAX, None);283 }284 StorageVersion::new(2).put::<Pallet<T>>();285286 Weight::zero()287 }288 }289}290291pub struct RefungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);292impl<T: Config> RefungibleHandle<T> {293 pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {294 Self(inner)295 }296 pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {297 self.0298 }299 pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {300 &mut self.0301 }302}303304impl<T: Config> Deref for RefungibleHandle<T> {305 type Target = pallet_common::CollectionHandle<T>;306307 fn deref(&self) -> &Self::Target {308 &self.0309 }310}311312impl<T: Config> WithRecorder<T> for RefungibleHandle<T> {313 fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {314 self.0.recorder()315 }316 fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {317 self.0.into_recorder()318 }319}320321impl<T: Config> Pallet<T> {322 323 pub fn total_supply(collection: &RefungibleHandle<T>) -> u32 {324 <TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)325 }326327 328 329 330 pub fn token_exists(collection: &RefungibleHandle<T>, token: TokenId) -> bool {331 <TotalSupply<T>>::contains_key((collection.id, token))332 }333334 pub fn set_scoped_token_property(335 collection_id: CollectionId,336 token_id: TokenId,337 scope: PropertyScope,338 property: Property,339 ) -> DispatchResult {340 TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {341 properties.try_scoped_set(scope, property.key, property.value)342 })343 .map_err(<CommonError<T>>::from)?;344345 Ok(())346 }347348 pub fn set_scoped_token_properties(349 collection_id: CollectionId,350 token_id: TokenId,351 scope: PropertyScope,352 properties: impl Iterator<Item = Property>,353 ) -> DispatchResult {354 TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {355 stored_properties.try_scoped_set_from_iter(scope, properties)356 })357 .map_err(<CommonError<T>>::from)?;358359 Ok(())360 }361}362363364impl<T: Config> Pallet<T> {365 366 367 368 369 370 pub fn init_collection(371 owner: T::CrossAccountId,372 payer: T::CrossAccountId,373 data: CreateCollectionData<T::AccountId>,374 flags: CollectionFlags,375 ) -> Result<CollectionId, DispatchError> {376 <PalletCommon<T>>::init_collection(owner, payer, data, flags)377 }378379 380 381 382 383 pub fn destroy_collection(384 collection: RefungibleHandle<T>,385 sender: &T::CrossAccountId,386 ) -> DispatchResult {387 let id = collection.id;388389 if Self::collection_has_tokens(id) {390 return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());391 }392393 394395 PalletCommon::destroy_collection(collection.0, sender)?;396397 <TokensMinted<T>>::remove(id);398 <TokensBurnt<T>>::remove(id);399 let _ = <TotalSupply<T>>::clear_prefix((id,), u32::MAX, None);400 let _ = <Balance<T>>::clear_prefix((id,), u32::MAX, None);401 let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);402 let _ = <Owned<T>>::clear_prefix((id,), u32::MAX, None);403 let _ = <AccountBalance<T>>::clear_prefix((id,), u32::MAX, None);404 Ok(())405 }406407 fn collection_has_tokens(collection_id: CollectionId) -> bool {408 <TotalSupply<T>>::iter_prefix((collection_id,))409 .next()410 .is_some()411 }412413 pub fn burn_token_unchecked(414 collection: &RefungibleHandle<T>,415 owner: &T::CrossAccountId,416 token_id: TokenId,417 ) -> DispatchResult {418 let burnt = <TokensBurnt<T>>::get(collection.id)419 .checked_add(1)420 .ok_or(ArithmeticError::Overflow)?;421422 <TokensBurnt<T>>::insert(collection.id, burnt);423 <TokenProperties<T>>::remove((collection.id, token_id));424 <TotalSupply<T>>::remove((collection.id, token_id));425 let _ = <Balance<T>>::clear_prefix((collection.id, token_id), u32::MAX, None);426 let _ = <Allowance<T>>::clear_prefix((collection.id, token_id), u32::MAX, None);427 <PalletEvm<T>>::deposit_log(428 ERC721Events::Transfer {429 from: *owner.as_eth(),430 to: H160::default(),431 token_id: token_id.into(),432 }433 .to_log(collection_id_to_address(collection.id)),434 );435 Ok(())436 }437438 439 440 441 442 443 444 445 446 447 448 449 pub fn burn(450 collection: &RefungibleHandle<T>,451 owner: &T::CrossAccountId,452 token: TokenId,453 amount: u128,454 ) -> DispatchResult {455 let total_supply = <TotalSupply<T>>::get((collection.id, token))456 .checked_sub(amount)457 .ok_or(<CommonError<T>>::TokenValueTooLow)?;458459 460 if total_supply == 0 {461 462 ensure!(463 <Balance<T>>::get((collection.id, token, owner)) == amount,464 <CommonError<T>>::TokenValueTooLow465 );466 let account_balance = <AccountBalance<T>>::get((collection.id, owner))467 .checked_sub(1)468 469 .ok_or(ArithmeticError::Underflow)?;470471 472473 <Owned<T>>::remove((collection.id, owner, token));474 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, token);475 <AccountBalance<T>>::insert((collection.id, owner), account_balance);476 Self::burn_token_unchecked(collection, owner, token)?;477 <PalletEvm<T>>::deposit_log(478 ERC20Events::Transfer {479 from: *owner.as_eth(),480 to: H160::default(),481 value: amount.into(),482 }483 .to_log(collection_id_to_address(collection.id)),484 );485 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(486 collection.id,487 token,488 owner.clone(),489 amount,490 ));491 return Ok(());492 }493494 let balance = <Balance<T>>::get((collection.id, token, owner))495 .checked_sub(amount)496 .ok_or(<CommonError<T>>::TokenValueTooLow)?;497 let account_balance = if balance == 0 {498 <AccountBalance<T>>::get((collection.id, owner))499 .checked_sub(1)500 501 .ok_or(ArithmeticError::Underflow)?502 } else {503 0504 };505506 507508 if balance == 0 {509 <Owned<T>>::remove((collection.id, owner, token));510 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, token);511 <Balance<T>>::remove((collection.id, token, owner));512 <AccountBalance<T>>::insert((collection.id, owner), account_balance);513514 if let Some(user) = Self::token_owner(collection.id, token) {515 <PalletEvm<T>>::deposit_log(516 ERC721Events::Transfer {517 from: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,518 to: *user.as_eth(),519 token_id: token.into(),520 }521 .to_log(collection_id_to_address(collection.id)),522 );523 }524 } else {525 <Balance<T>>::insert((collection.id, token, owner), balance);526 }527 <TotalSupply<T>>::insert((collection.id, token), total_supply);528529 <PalletEvm<T>>::deposit_log(530 ERC20Events::Transfer {531 from: *owner.as_eth(),532 to: H160::default(),533 value: amount.into(),534 }535 .to_log(T::EvmTokenAddressMapping::token_to_address(536 collection.id,537 token,538 )),539 );540 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(541 collection.id,542 token,543 owner.clone(),544 amount,545 ));546 Ok(())547 }548549 #[transactional]550 fn modify_token_properties(551 collection: &RefungibleHandle<T>,552 sender: &T::CrossAccountId,553 token_id: TokenId,554 properties: impl Iterator<Item = (PropertyKey, Option<PropertyValue>)>,555 is_token_create: bool,556 nesting_budget: &dyn Budget,557 ) -> DispatchResult {558 let is_collection_admin = || collection.is_owner_or_admin(sender);559 let is_token_owner = || -> Result<bool, DispatchError> {560 let balance = collection.balance(sender.clone(), token_id);561 let total_pieces: u128 =562 Self::total_pieces(collection.id, token_id).unwrap_or(u128::MAX);563 if balance != total_pieces {564 return Ok(false);565 }566567 let is_bundle_owner = <PalletStructure<T>>::check_indirectly_owned(568 sender.clone(),569 collection.id,570 token_id,571 None,572 nesting_budget,573 )?;574575 Ok(is_bundle_owner)576 };577578 for (key, value) in properties {579 let permission = <PalletCommon<T>>::property_permissions(collection.id)580 .get(&key)581 .cloned()582 .unwrap_or_else(PropertyPermission::none);583584 let is_property_exists = TokenProperties::<T>::get((collection.id, token_id))585 .get(&key)586 .is_some();587588 match permission {589 PropertyPermission { mutable: false, .. } if is_property_exists => {590 return Err(<CommonError<T>>::NoPermission.into());591 }592593 PropertyPermission {594 collection_admin,595 token_owner,596 ..597 } => {598 599 let is_token_create =600 is_token_create && (collection_admin || token_owner) && value.is_some();601 if !(is_token_create602 || (collection_admin && is_collection_admin())603 || (token_owner && is_token_owner()?))604 {605 fail!(<CommonError<T>>::NoPermission);606 }607 }608 }609610 match value {611 Some(value) => {612 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {613 properties.try_set(key.clone(), value)614 })615 .map_err(<CommonError<T>>::from)?;616617 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertySet(618 collection.id,619 token_id,620 key,621 ));622 }623 None => {624 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {625 properties.remove(&key)626 })627 .map_err(<CommonError<T>>::from)?;628629 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertyDeleted(630 collection.id,631 token_id,632 key,633 ));634 }635 }636 }637638 Ok(())639 }640641 pub fn set_token_properties(642 collection: &RefungibleHandle<T>,643 sender: &T::CrossAccountId,644 token_id: TokenId,645 properties: impl Iterator<Item = Property>,646 is_token_create: bool,647 nesting_budget: &dyn Budget,648 ) -> DispatchResult {649 Self::modify_token_properties(650 collection,651 sender,652 token_id,653 properties.map(|p| (p.key, Some(p.value))),654 is_token_create,655 nesting_budget,656 )657 }658659 pub fn set_token_property(660 collection: &RefungibleHandle<T>,661 sender: &T::CrossAccountId,662 token_id: TokenId,663 property: Property,664 nesting_budget: &dyn Budget,665 ) -> DispatchResult {666 let is_token_create = false;667668 Self::set_token_properties(669 collection,670 sender,671 token_id,672 [property].into_iter(),673 is_token_create,674 nesting_budget,675 )676 }677678 pub fn delete_token_properties(679 collection: &RefungibleHandle<T>,680 sender: &T::CrossAccountId,681 token_id: TokenId,682 property_keys: impl Iterator<Item = PropertyKey>,683 nesting_budget: &dyn Budget,684 ) -> DispatchResult {685 let is_token_create = false;686687 Self::modify_token_properties(688 collection,689 sender,690 token_id,691 property_keys.into_iter().map(|key| (key, None)),692 is_token_create,693 nesting_budget,694 )695 }696697 pub fn delete_token_property(698 collection: &RefungibleHandle<T>,699 sender: &T::CrossAccountId,700 token_id: TokenId,701 property_key: PropertyKey,702 nesting_budget: &dyn Budget,703 ) -> DispatchResult {704 Self::delete_token_properties(705 collection,706 sender,707 token_id,708 [property_key].into_iter(),709 nesting_budget,710 )711 }712713 714 715 716 717 718 719 720 721 722 pub fn transfer(723 collection: &RefungibleHandle<T>,724 from: &T::CrossAccountId,725 to: &T::CrossAccountId,726 token: TokenId,727 amount: u128,728 nesting_budget: &dyn Budget,729 ) -> DispatchResult {730 ensure!(amount > 0, <CommonError<T>>::ZeroTransferNotAllowed);731732 ensure!(733 collection.limits.transfers_enabled(),734 <CommonError<T>>::TransferNotAllowed735 );736737 if collection.permissions.access() == AccessMode::AllowList {738 collection.check_allowlist(from)?;739 collection.check_allowlist(to)?;740 }741 <PalletCommon<T>>::ensure_correct_receiver(to)?;742743 let initial_balance_from = <Balance<T>>::get((collection.id, token, from));744 let updated_balance_from = initial_balance_from745 .checked_sub(amount)746 .ok_or(<CommonError<T>>::TokenValueTooLow)?;747 let mut create_target = false;748 let from_to_differ = from != to;749 let updated_balance_to = if from != to {750 let old_balance = <Balance<T>>::get((collection.id, token, to));751 if old_balance == 0 {752 create_target = true;753 }754 Some(755 old_balance756 .checked_add(amount)757 .ok_or(ArithmeticError::Overflow)?,758 )759 } else {760 None761 };762763 let account_balance_from = if updated_balance_from == 0 {764 Some(765 <AccountBalance<T>>::get((collection.id, from))766 .checked_sub(1)767 768 .ok_or(ArithmeticError::Underflow)?,769 )770 } else {771 None772 };773 774 775 let account_balance_to = if create_target && from_to_differ {776 let account_balance_to = <AccountBalance<T>>::get((collection.id, to))777 .checked_add(1)778 .ok_or(ArithmeticError::Overflow)?;779 ensure!(780 account_balance_to < collection.limits.account_token_ownership_limit(),781 <CommonError<T>>::AccountTokenLimitExceeded,782 );783784 Some(account_balance_to)785 } else {786 None787 };788789 790791 <PalletStructure<T>>::nest_if_sent_to_token(792 from.clone(),793 to,794 collection.id,795 token,796 nesting_budget,797 )?;798799 if let Some(updated_balance_to) = updated_balance_to {800 801 if updated_balance_from == 0 {802 <Balance<T>>::remove((collection.id, token, from));803 <PalletStructure<T>>::unnest_if_nested(from, collection.id, token);804 } else {805 <Balance<T>>::insert((collection.id, token, from), updated_balance_from);806 }807 <Balance<T>>::insert((collection.id, token, to), updated_balance_to);808 if let Some(account_balance_from) = account_balance_from {809 <AccountBalance<T>>::insert((collection.id, from), account_balance_from);810 <Owned<T>>::remove((collection.id, from, token));811 }812 if let Some(account_balance_to) = account_balance_to {813 <AccountBalance<T>>::insert((collection.id, to), account_balance_to);814 <Owned<T>>::insert((collection.id, to, token), true);815 }816 }817818 <PalletEvm<T>>::deposit_log(819 ERC20Events::Transfer {820 from: *from.as_eth(),821 to: *to.as_eth(),822 value: amount.into(),823 }824 .to_log(T::EvmTokenAddressMapping::token_to_address(825 collection.id,826 token,827 )),828 );829830 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(831 collection.id,832 token,833 from.clone(),834 to.clone(),835 amount,836 ));837838 let total_supply = <TotalSupply<T>>::get((collection.id, token));839840 if amount == total_supply {841 842 <PalletEvm<T>>::deposit_log(843 ERC721Events::Transfer {844 from: *from.as_eth(),845 to: *to.as_eth(),846 token_id: token.into(),847 }848 .to_log(collection_id_to_address(collection.id)),849 );850 } else if let Some(updated_balance_to) = updated_balance_to {851 852 853 if initial_balance_from == total_supply {854 855 856 <PalletEvm<T>>::deposit_log(857 ERC721Events::Transfer {858 from: *from.as_eth(),859 to: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,860 token_id: token.into(),861 }862 .to_log(collection_id_to_address(collection.id)),863 );864 } else if updated_balance_to == total_supply {865 866 <PalletEvm<T>>::deposit_log(867 ERC721Events::Transfer {868 from: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,869 to: *to.as_eth(),870 token_id: token.into(),871 }872 .to_log(collection_id_to_address(collection.id)),873 );874 }875 }876877 Ok(())878 }879880 881 882 883 884 885 pub fn create_multiple_items(886 collection: &RefungibleHandle<T>,887 sender: &T::CrossAccountId,888 data: Vec<CreateItemData<T::CrossAccountId>>,889 nesting_budget: &dyn Budget,890 ) -> DispatchResult {891 if !collection.is_owner_or_admin(sender) {892 ensure!(893 collection.permissions.mint_mode(),894 <CommonError<T>>::PublicMintingNotAllowed895 );896 collection.check_allowlist(sender)?;897898 for item in data.iter() {899 for user in item.users.keys() {900 collection.check_allowlist(user)?;901 }902 }903 }904905 for item in data.iter() {906 for (owner, _) in item.users.iter() {907 <PalletCommon<T>>::ensure_correct_receiver(owner)?;908 }909 }910911 912 let totals = data913 .iter()914 .map(|data| {915 Ok(data916 .users917 .iter()918 .map(|u| u.1)919 .try_fold(0u128, |acc, v| acc.checked_add(*v))920 .ok_or(ArithmeticError::Overflow)?)921 })922 .collect::<Result<Vec<_>, DispatchError>>()?;923 for total in &totals {924 ensure!(925 *total <= MAX_REFUNGIBLE_PIECES,926 <Error<T>>::WrongRefungiblePieces927 );928 }929930 let first_token_id = <TokensMinted<T>>::get(collection.id);931 let tokens_minted = first_token_id932 .checked_add(data.len() as u32)933 .ok_or(ArithmeticError::Overflow)?;934 ensure!(935 tokens_minted < collection.limits.token_limit(),936 <CommonError<T>>::CollectionTokenLimitExceeded937 );938939 let mut balances = BTreeMap::new();940 for data in &data {941 for owner in data.users.keys() {942 let balance = balances943 .entry(owner)944 .or_insert_with(|| <AccountBalance<T>>::get((collection.id, owner)));945 *balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;946947 ensure!(948 *balance <= collection.limits.account_token_ownership_limit(),949 <CommonError<T>>::AccountTokenLimitExceeded,950 );951 }952 }953954 for (i, token) in data.iter().enumerate() {955 let token_id = TokenId(first_token_id + i as u32 + 1);956 for (to, _) in token.users.iter() {957 <PalletStructure<T>>::check_nesting(958 sender.clone(),959 to,960 collection.id,961 token_id,962 nesting_budget,963 )?;964 }965 }966967 968969 with_transaction(|| {970 for (i, data) in data.iter().enumerate() {971 let token_id = first_token_id + i as u32 + 1;972 <TotalSupply<T>>::insert((collection.id, token_id), totals[i]);973974 for (user, amount) in data.users.iter() {975 if *amount == 0 {976 continue;977 }978 <Balance<T>>::insert((collection.id, token_id, &user), amount);979 <Owned<T>>::insert((collection.id, &user, TokenId(token_id)), true);980 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(981 user,982 collection.id,983 TokenId(token_id),984 );985 }986987 if let Err(e) = Self::set_token_properties(988 collection,989 sender,990 TokenId(token_id),991 data.properties.clone().into_iter(),992 true,993 nesting_budget,994 ) {995 return TransactionOutcome::Rollback(Err(e));996 }997 }998 TransactionOutcome::Commit(Ok(()))999 })?;10001001 <TokensMinted<T>>::insert(collection.id, tokens_minted);10021003 for (account, balance) in balances {1004 <AccountBalance<T>>::insert((collection.id, account), balance);1005 }10061007 for (i, token) in data.into_iter().enumerate() {1008 let token_id = first_token_id + i as u32 + 1;10091010 let receivers = token1011 .users1012 .into_iter()1013 .filter(|(_, amount)| *amount > 0)1014 .collect::<Vec<_>>();10151016 if let [(user, _)] = receivers.as_slice() {1017 1018 <PalletEvm<T>>::deposit_log(1019 ERC721Events::Transfer {1020 from: H160::default(),1021 to: *user.as_eth(),1022 token_id: token_id.into(),1023 }1024 .to_log(collection_id_to_address(collection.id)),1025 );1026 } else if let [_, ..] = receivers.as_slice() {1027 1028 <PalletEvm<T>>::deposit_log(1029 ERC721Events::Transfer {1030 from: H160::default(),1031 to: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,1032 token_id: token_id.into(),1033 }1034 .to_log(collection_id_to_address(collection.id)),1035 );1036 }10371038 for (user, amount) in receivers.into_iter() {1039 <PalletEvm<T>>::deposit_log(1040 ERC20Events::Transfer {1041 from: H160::default(),1042 to: *user.as_eth(),1043 value: amount.into(),1044 }1045 .to_log(T::EvmTokenAddressMapping::token_to_address(1046 collection.id,1047 TokenId(token_id),1048 )),1049 );1050 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(1051 collection.id,1052 TokenId(token_id),1053 user,1054 amount,1055 ));1056 }1057 }1058 Ok(())1059 }10601061 pub fn set_allowance_unchecked(1062 collection: &RefungibleHandle<T>,1063 sender: &T::CrossAccountId,1064 spender: &T::CrossAccountId,1065 token: TokenId,1066 amount: u128,1067 ) {1068 if amount == 0 {1069 <Allowance<T>>::remove((collection.id, token, sender, spender));1070 } else {1071 <Allowance<T>>::insert((collection.id, token, sender, spender), amount);1072 }10731074 <PalletEvm<T>>::deposit_log(1075 ERC20Events::Approval {1076 owner: *sender.as_eth(),1077 spender: *spender.as_eth(),1078 value: amount.into(),1079 }1080 .to_log(T::EvmTokenAddressMapping::token_to_address(1081 collection.id,1082 token,1083 )),1084 );1085 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(1086 collection.id,1087 token,1088 sender.clone(),1089 spender.clone(),1090 amount,1091 ))1092 }10931094 1095 1096 1097 pub fn set_allowance(1098 collection: &RefungibleHandle<T>,1099 sender: &T::CrossAccountId,1100 spender: &T::CrossAccountId,1101 token: TokenId,1102 amount: u128,1103 ) -> DispatchResult {1104 if collection.permissions.access() == AccessMode::AllowList {1105 collection.check_allowlist(sender)?;1106 collection.check_allowlist(spender)?;1107 }11081109 <PalletCommon<T>>::ensure_correct_receiver(spender)?;11101111 if <Balance<T>>::get((collection.id, token, sender)) < amount {1112 ensure!(1113 collection.ignores_owned_amount(sender) && Self::token_exists(collection, token),1114 <CommonError<T>>::CantApproveMoreThanOwned1115 );1116 }11171118 11191120 Self::set_allowance_unchecked(collection, sender, spender, token, amount);1121 Ok(())1122 }11231124 1125 fn check_allowed(1126 collection: &RefungibleHandle<T>,1127 spender: &T::CrossAccountId,1128 from: &T::CrossAccountId,1129 token: TokenId,1130 amount: u128,1131 nesting_budget: &dyn Budget,1132 ) -> Result<Option<u128>, DispatchError> {1133 if spender.conv_eq(from) {1134 return Ok(None);1135 }1136 if collection.permissions.access() == AccessMode::AllowList {1137 1138 collection.check_allowlist(spender)?;1139 }1140 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {1141 1142 ensure!(1143 <PalletStructure<T>>::check_indirectly_owned(1144 spender.clone(),1145 source.0,1146 source.1,1147 None,1148 nesting_budget1149 )?,1150 <CommonError<T>>::ApprovedValueTooLow,1151 );1152 return Ok(None);1153 }1154 let allowance =1155 <Allowance<T>>::get((collection.id, token, from, &spender)).checked_sub(amount);1156 if allowance.is_none() {1157 ensure!(1158 collection.ignores_allowance(spender),1159 <CommonError<T>>::ApprovedValueTooLow1160 );1161 }1162 Ok(allowance)1163 }11641165 1166 1167 1168 1169 1170 1171 pub fn transfer_from(1172 collection: &RefungibleHandle<T>,1173 spender: &T::CrossAccountId,1174 from: &T::CrossAccountId,1175 to: &T::CrossAccountId,1176 token: TokenId,1177 amount: u128,1178 nesting_budget: &dyn Budget,1179 ) -> DispatchResult {1180 let allowance =1181 Self::check_allowed(collection, spender, from, token, amount, nesting_budget)?;11821183 11841185 Self::transfer(collection, from, to, token, amount, nesting_budget)?;1186 if let Some(allowance) = allowance {1187 Self::set_allowance_unchecked(collection, from, spender, token, allowance);1188 }1189 Ok(())1190 }11911192 1193 1194 1195 1196 1197 1198 pub fn burn_from(1199 collection: &RefungibleHandle<T>,1200 spender: &T::CrossAccountId,1201 from: &T::CrossAccountId,1202 token: TokenId,1203 amount: u128,1204 nesting_budget: &dyn Budget,1205 ) -> DispatchResult {1206 let allowance =1207 Self::check_allowed(collection, spender, from, token, amount, nesting_budget)?;12081209 12101211 Self::burn(collection, from, token, amount)?;1212 if let Some(allowance) = allowance {1213 Self::set_allowance_unchecked(collection, from, spender, token, allowance);1214 }1215 Ok(())1216 }12171218 1219 1220 1221 1222 1223 1224 1225 pub fn create_item(1226 collection: &RefungibleHandle<T>,1227 sender: &T::CrossAccountId,1228 data: CreateItemData<T::CrossAccountId>,1229 nesting_budget: &dyn Budget,1230 ) -> DispatchResult {1231 Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1232 }12331234 1235 1236 1237 1238 1239 1240 1241 pub fn repartition(1242 collection: &RefungibleHandle<T>,1243 owner: &T::CrossAccountId,1244 token: TokenId,1245 amount: u128,1246 ) -> DispatchResult {1247 ensure!(1248 amount <= MAX_REFUNGIBLE_PIECES,1249 <Error<T>>::WrongRefungiblePieces1250 );1251 ensure!(amount > 0, <CommonError<T>>::TokenValueTooLow);1252 1253 let total_pieces = Self::total_pieces(collection.id, token).unwrap_or(u128::MAX);1254 let balance = <Balance<T>>::get((collection.id, token, owner));1255 ensure!(1256 total_pieces == balance,1257 <Error<T>>::RepartitionWhileNotOwningAllPieces1258 );12591260 <Balance<T>>::insert((collection.id, token, owner), amount);1261 <TotalSupply<T>>::insert((collection.id, token), amount);12621263 if amount > total_pieces {1264 let mint_amount = amount - total_pieces;1265 <PalletEvm<T>>::deposit_log(1266 ERC20Events::Transfer {1267 from: H160::default(),1268 to: *owner.as_eth(),1269 value: mint_amount.into(),1270 }1271 .to_log(T::EvmTokenAddressMapping::token_to_address(1272 collection.id,1273 token,1274 )),1275 );1276 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(1277 collection.id,1278 token,1279 owner.clone(),1280 mint_amount,1281 ));1282 } else if total_pieces > amount {1283 let burn_amount = total_pieces - amount;1284 <PalletEvm<T>>::deposit_log(1285 ERC20Events::Transfer {1286 from: *owner.as_eth(),1287 to: H160::default(),1288 value: burn_amount.into(),1289 }1290 .to_log(T::EvmTokenAddressMapping::token_to_address(1291 collection.id,1292 token,1293 )),1294 );1295 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(1296 collection.id,1297 token,1298 owner.clone(),1299 burn_amount,1300 ));1301 }13021303 Ok(())1304 }13051306 fn token_owner(collection_id: CollectionId, token_id: TokenId) -> Option<T::CrossAccountId> {1307 let mut owner = None;1308 let mut count = 0;1309 for key in Balance::<T>::iter_key_prefix((collection_id, token_id)) {1310 count += 1;1311 if count > 1 {1312 return None;1313 }1314 owner = Some(key);1315 }1316 owner1317 }13181319 fn total_pieces(collection_id: CollectionId, token_id: TokenId) -> Option<u128> {1320 <TotalSupply<T>>::try_get((collection_id, token_id)).ok()1321 }13221323 pub fn set_collection_properties(1324 collection: &RefungibleHandle<T>,1325 sender: &T::CrossAccountId,1326 properties: Vec<Property>,1327 ) -> DispatchResult {1328 <PalletCommon<T>>::set_collection_properties(collection, sender, properties)1329 }13301331 pub fn delete_collection_properties(1332 collection: &RefungibleHandle<T>,1333 sender: &T::CrossAccountId,1334 property_keys: Vec<PropertyKey>,1335 ) -> DispatchResult {1336 <PalletCommon<T>>::delete_collection_properties(collection, sender, property_keys)1337 }13381339 pub fn set_token_property_permissions(1340 collection: &RefungibleHandle<T>,1341 sender: &T::CrossAccountId,1342 property_permissions: Vec<PropertyKeyPermission>,1343 ) -> DispatchResult {1344 <PalletCommon<T>>::set_token_property_permissions(collection, sender, property_permissions)1345 }13461347 pub fn set_scoped_token_property_permissions(1348 collection: &RefungibleHandle<T>,1349 sender: &T::CrossAccountId,1350 scope: PropertyScope,1351 property_permissions: Vec<PropertyKeyPermission>,1352 ) -> DispatchResult {1353 <PalletCommon<T>>::set_scoped_token_property_permissions(1354 collection,1355 sender,1356 scope,1357 property_permissions,1358 )1359 }13601361 1362 1363 1364 1365 1366 1367 pub fn token_owners(1368 collection_id: CollectionId,1369 token: TokenId,1370 ) -> Option<Vec<T::CrossAccountId>> {1371 let res: Vec<T::CrossAccountId> = <Balance<T>>::iter_prefix((collection_id, token))1372 .map(|(owner, _amount)| owner)1373 .take(10)1374 .collect();13751376 if res.is_empty() {1377 None1378 } else {1379 Some(res)1380 }1381 }1382}