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)]140#[deprecated(since = "0.2.0", note = "ItemData is no more contains usefull data")]141pub struct ItemData {142 pub const_data: BoundedVec<u8, CustomDataLimit>,143144 #[version(..2)]145 pub variable_data: BoundedVec<u8, CustomDataLimit>,146}147148#[frame_support::pallet]149pub mod pallet {150 use super::*;151 use frame_support::{152 Blake2_128, Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key,153 traits::StorageVersion,154 };155 use frame_system::pallet_prelude::*;156 use up_data_structs::{CollectionId, TokenId};157 use super::weights::WeightInfo;158159 #[pallet::error]160 pub enum Error<T> {161 162 NotRefungibleDataUsedToMintFungibleCollectionToken,163 164 WrongRefungiblePieces,165 166 RepartitionWhileNotOwningAllPieces,167 168 RefungibleDisallowsNesting,169 170 SettingPropertiesNotAllowed,171 }172173 #[pallet::config]174 pub trait Config:175 frame_system::Config + pallet_common::Config + pallet_structure::Config176 {177 type WeightInfo: WeightInfo;178 }179180 const STORAGE_VERSION: StorageVersion = StorageVersion::new(2);181182 #[pallet::pallet]183 #[pallet::storage_version(STORAGE_VERSION)]184 #[pallet::generate_store(pub(super) trait Store)]185 pub struct Pallet<T>(_);186187 188 #[pallet::storage]189 pub type TokensMinted<T: Config> =190 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;191192 193 #[pallet::storage]194 pub type TokensBurnt<T: Config> =195 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;196197 198 199 #[pallet::storage]200 #[deprecated(since = "0.2.0", note = "ItemData is no more contains usefull data")]201 pub type TokenData<T: Config> = StorageNMap<202 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),203 Value = ItemData,204 QueryKind = ValueQuery,205 >;206207 208 #[pallet::storage]209 #[pallet::getter(fn token_properties)]210 pub type TokenProperties<T: Config> = StorageNMap<211 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),212 Value = up_data_structs::Properties,213 QueryKind = ValueQuery,214 OnEmpty = up_data_structs::TokenProperties,215 >;216217 218 #[pallet::storage]219 pub type TotalSupply<T: Config> = StorageNMap<220 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),221 Value = u128,222 QueryKind = ValueQuery,223 >;224225 226 #[pallet::storage]227 pub type Owned<T: Config> = StorageNMap<228 Key = (229 Key<Twox64Concat, CollectionId>,230 Key<Blake2_128Concat, T::CrossAccountId>,231 Key<Twox64Concat, TokenId>,232 ),233 Value = bool,234 QueryKind = ValueQuery,235 >;236237 238 #[pallet::storage]239 pub type AccountBalance<T: Config> = StorageNMap<240 Key = (241 Key<Twox64Concat, CollectionId>,242 243 Key<Blake2_128Concat, T::CrossAccountId>,244 ),245 Value = u32,246 QueryKind = ValueQuery,247 >;248249 250 #[pallet::storage]251 pub type Balance<T: Config> = StorageNMap<252 Key = (253 Key<Twox64Concat, CollectionId>,254 Key<Twox64Concat, TokenId>,255 256 Key<Blake2_128Concat, T::CrossAccountId>,257 ),258 Value = u128,259 QueryKind = ValueQuery,260 >;261262 263 #[pallet::storage]264 pub type Allowance<T: Config> = StorageNMap<265 Key = (266 Key<Twox64Concat, CollectionId>,267 Key<Twox64Concat, TokenId>,268 269 Key<Blake2_128, T::CrossAccountId>,270 271 Key<Blake2_128Concat, T::CrossAccountId>,272 ),273 Value = u128,274 QueryKind = ValueQuery,275 >;276277 #[pallet::hooks]278 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {279 fn on_runtime_upgrade() -> Weight {280 let storage_version = StorageVersion::get::<Pallet<T>>();281 if storage_version < StorageVersion::new(2) {282 #[allow(deprecated)]283 <TokenData<T>>::remove_all(None);284 }285 StorageVersion::new(2).put::<Pallet<T>>();286287 Weight::zero()288 }289 }290}291292pub struct RefungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);293impl<T: Config> RefungibleHandle<T> {294 pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {295 Self(inner)296 }297 pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {298 self.0299 }300 pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {301 &mut self.0302 }303}304305impl<T: Config> Deref for RefungibleHandle<T> {306 type Target = pallet_common::CollectionHandle<T>;307308 fn deref(&self) -> &Self::Target {309 &self.0310 }311}312313impl<T: Config> WithRecorder<T> for RefungibleHandle<T> {314 fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {315 self.0.recorder()316 }317 fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {318 self.0.into_recorder()319 }320}321322impl<T: Config> Pallet<T> {323 324 pub fn total_supply(collection: &RefungibleHandle<T>) -> u32 {325 <TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)326 }327328 329 330 331 pub fn token_exists(collection: &RefungibleHandle<T>, token: TokenId) -> bool {332 <TotalSupply<T>>::contains_key((collection.id, token))333 }334335 pub fn set_scoped_token_property(336 collection_id: CollectionId,337 token_id: TokenId,338 scope: PropertyScope,339 property: Property,340 ) -> DispatchResult {341 TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {342 properties.try_scoped_set(scope, property.key, property.value)343 })344 .map_err(<CommonError<T>>::from)?;345346 Ok(())347 }348349 pub fn set_scoped_token_properties(350 collection_id: CollectionId,351 token_id: TokenId,352 scope: PropertyScope,353 properties: impl Iterator<Item = Property>,354 ) -> DispatchResult {355 TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {356 stored_properties.try_scoped_set_from_iter(scope, properties)357 })358 .map_err(<CommonError<T>>::from)?;359360 Ok(())361 }362}363364365impl<T: Config> Pallet<T> {366 367 368 369 370 371 pub fn init_collection(372 owner: T::CrossAccountId,373 payer: T::CrossAccountId,374 data: CreateCollectionData<T::AccountId>,375 flags: CollectionFlags,376 ) -> Result<CollectionId, DispatchError> {377 <PalletCommon<T>>::init_collection(owner, payer, data, flags)378 }379380 381 382 383 384 pub fn destroy_collection(385 collection: RefungibleHandle<T>,386 sender: &T::CrossAccountId,387 ) -> DispatchResult {388 let id = collection.id;389390 if Self::collection_has_tokens(id) {391 return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());392 }393394 395396 PalletCommon::destroy_collection(collection.0, sender)?;397398 <TokensMinted<T>>::remove(id);399 <TokensBurnt<T>>::remove(id);400 let _ = <TotalSupply<T>>::clear_prefix((id,), u32::MAX, None);401 let _ = <Balance<T>>::clear_prefix((id,), u32::MAX, None);402 let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);403 let _ = <Owned<T>>::clear_prefix((id,), u32::MAX, None);404 let _ = <AccountBalance<T>>::clear_prefix((id,), u32::MAX, None);405 Ok(())406 }407408 fn collection_has_tokens(collection_id: CollectionId) -> bool {409 <TotalSupply<T>>::iter_prefix((collection_id,))410 .next()411 .is_some()412 }413414 pub fn burn_token_unchecked(415 collection: &RefungibleHandle<T>,416 owner: &T::CrossAccountId,417 token_id: TokenId,418 ) -> DispatchResult {419 let burnt = <TokensBurnt<T>>::get(collection.id)420 .checked_add(1)421 .ok_or(ArithmeticError::Overflow)?;422423 <TokensBurnt<T>>::insert(collection.id, burnt);424 <TokenProperties<T>>::remove((collection.id, token_id));425 <TotalSupply<T>>::remove((collection.id, token_id));426 let _ = <Balance<T>>::clear_prefix((collection.id, token_id), u32::MAX, None);427 let _ = <Allowance<T>>::clear_prefix((collection.id, token_id), u32::MAX, None);428 <PalletEvm<T>>::deposit_log(429 ERC721Events::Transfer {430 from: *owner.as_eth(),431 to: H160::default(),432 token_id: token_id.into(),433 }434 .to_log(collection_id_to_address(collection.id)),435 );436 Ok(())437 }438439 440 441 442 443 444 445 446 447 448 449 450 pub fn burn(451 collection: &RefungibleHandle<T>,452 owner: &T::CrossAccountId,453 token: TokenId,454 amount: u128,455 ) -> DispatchResult {456 let total_supply = <TotalSupply<T>>::get((collection.id, token))457 .checked_sub(amount)458 .ok_or(<CommonError<T>>::TokenValueTooLow)?;459460 461 if total_supply == 0 {462 463 ensure!(464 <Balance<T>>::get((collection.id, token, owner)) == amount,465 <CommonError<T>>::TokenValueTooLow466 );467 let account_balance = <AccountBalance<T>>::get((collection.id, owner))468 .checked_sub(1)469 470 .ok_or(ArithmeticError::Underflow)?;471472 473474 <Owned<T>>::remove((collection.id, owner, token));475 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, token);476 <AccountBalance<T>>::insert((collection.id, owner), account_balance);477 Self::burn_token_unchecked(collection, owner, token)?;478 <PalletEvm<T>>::deposit_log(479 ERC20Events::Transfer {480 from: *owner.as_eth(),481 to: H160::default(),482 value: amount.into(),483 }484 .to_log(collection_id_to_address(collection.id)),485 );486 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(487 collection.id,488 token,489 owner.clone(),490 amount,491 ));492 return Ok(());493 }494495 let balance = <Balance<T>>::get((collection.id, token, owner))496 .checked_sub(amount)497 .ok_or(<CommonError<T>>::TokenValueTooLow)?;498 let account_balance = if balance == 0 {499 <AccountBalance<T>>::get((collection.id, owner))500 .checked_sub(1)501 502 .ok_or(ArithmeticError::Underflow)?503 } else {504 0505 };506507 508509 if balance == 0 {510 <Owned<T>>::remove((collection.id, owner, token));511 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, token);512 <Balance<T>>::remove((collection.id, token, owner));513 <AccountBalance<T>>::insert((collection.id, owner), account_balance);514515 if let Some(user) = Self::token_owner(collection.id, token) {516 <PalletEvm<T>>::deposit_log(517 ERC721Events::Transfer {518 from: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,519 to: *user.as_eth(),520 token_id: token.into(),521 }522 .to_log(collection_id_to_address(collection.id)),523 );524 }525 } else {526 <Balance<T>>::insert((collection.id, token, owner), balance);527 }528 <TotalSupply<T>>::insert((collection.id, token), total_supply);529530 <PalletEvm<T>>::deposit_log(531 ERC20Events::Transfer {532 from: *owner.as_eth(),533 to: H160::default(),534 value: amount.into(),535 }536 .to_log(T::EvmTokenAddressMapping::token_to_address(537 collection.id,538 token,539 )),540 );541 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(542 collection.id,543 token,544 owner.clone(),545 amount,546 ));547 Ok(())548 }549550 #[transactional]551 fn modify_token_properties(552 collection: &RefungibleHandle<T>,553 sender: &T::CrossAccountId,554 token_id: TokenId,555 properties: impl Iterator<Item = (PropertyKey, Option<PropertyValue>)>,556 is_token_create: bool,557 nesting_budget: &dyn Budget,558 ) -> DispatchResult {559 let is_collection_admin = || collection.is_owner_or_admin(sender);560 let is_token_owner = || -> Result<bool, DispatchError> {561 let balance = collection.balance(sender.clone(), token_id);562 let total_pieces: u128 =563 Self::total_pieces(collection.id, token_id).unwrap_or(u128::MAX);564 if balance != total_pieces {565 return Ok(false);566 }567568 let is_bundle_owner = <PalletStructure<T>>::check_indirectly_owned(569 sender.clone(),570 collection.id,571 token_id,572 None,573 nesting_budget,574 )?;575576 Ok(is_bundle_owner)577 };578579 for (key, value) in properties {580 let permission = <PalletCommon<T>>::property_permissions(collection.id)581 .get(&key)582 .cloned()583 .unwrap_or_else(PropertyPermission::none);584585 let is_property_exists = TokenProperties::<T>::get((collection.id, token_id))586 .get(&key)587 .is_some();588589 match permission {590 PropertyPermission { mutable: false, .. } if is_property_exists => {591 return Err(<CommonError<T>>::NoPermission.into());592 }593594 PropertyPermission {595 collection_admin,596 token_owner,597 ..598 } => {599 600 let is_token_create =601 is_token_create && (collection_admin || token_owner) && value.is_some();602 if !(is_token_create603 || (collection_admin && is_collection_admin())604 || (token_owner && is_token_owner()?))605 {606 fail!(<CommonError<T>>::NoPermission);607 }608 }609 }610611 match value {612 Some(value) => {613 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {614 properties.try_set(key.clone(), value)615 })616 .map_err(<CommonError<T>>::from)?;617618 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertySet(619 collection.id,620 token_id,621 key,622 ));623 }624 None => {625 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {626 properties.remove(&key)627 })628 .map_err(<CommonError<T>>::from)?;629630 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertyDeleted(631 collection.id,632 token_id,633 key,634 ));635 }636 }637 }638639 Ok(())640 }641642 pub fn set_token_properties(643 collection: &RefungibleHandle<T>,644 sender: &T::CrossAccountId,645 token_id: TokenId,646 properties: impl Iterator<Item = Property>,647 is_token_create: bool,648 nesting_budget: &dyn Budget,649 ) -> DispatchResult {650 Self::modify_token_properties(651 collection,652 sender,653 token_id,654 properties.map(|p| (p.key, Some(p.value))),655 is_token_create,656 nesting_budget,657 )658 }659660 pub fn set_token_property(661 collection: &RefungibleHandle<T>,662 sender: &T::CrossAccountId,663 token_id: TokenId,664 property: Property,665 nesting_budget: &dyn Budget,666 ) -> DispatchResult {667 let is_token_create = false;668669 Self::set_token_properties(670 collection,671 sender,672 token_id,673 [property].into_iter(),674 is_token_create,675 nesting_budget,676 )677 }678679 pub fn delete_token_properties(680 collection: &RefungibleHandle<T>,681 sender: &T::CrossAccountId,682 token_id: TokenId,683 property_keys: impl Iterator<Item = PropertyKey>,684 nesting_budget: &dyn Budget,685 ) -> DispatchResult {686 let is_token_create = false;687688 Self::modify_token_properties(689 collection,690 sender,691 token_id,692 property_keys.into_iter().map(|key| (key, None)),693 is_token_create,694 nesting_budget,695 )696 }697698 pub fn delete_token_property(699 collection: &RefungibleHandle<T>,700 sender: &T::CrossAccountId,701 token_id: TokenId,702 property_key: PropertyKey,703 nesting_budget: &dyn Budget,704 ) -> DispatchResult {705 Self::delete_token_properties(706 collection,707 sender,708 token_id,709 [property_key].into_iter(),710 nesting_budget,711 )712 }713714 715 716 717 718 719 720 721 722 723 pub fn transfer(724 collection: &RefungibleHandle<T>,725 from: &T::CrossAccountId,726 to: &T::CrossAccountId,727 token: TokenId,728 amount: u128,729 nesting_budget: &dyn Budget,730 ) -> DispatchResult {731 ensure!(732 collection.limits.transfers_enabled(),733 <CommonError<T>>::TransferNotAllowed734 );735736 if collection.permissions.access() == AccessMode::AllowList {737 collection.check_allowlist(from)?;738 collection.check_allowlist(to)?;739 }740 <PalletCommon<T>>::ensure_correct_receiver(to)?;741742 let initial_balance_from = <Balance<T>>::get((collection.id, token, from));743 let updated_balance_from = initial_balance_from744 .checked_sub(amount)745 .ok_or(<CommonError<T>>::TokenValueTooLow)?;746 let mut create_target = false;747 let from_to_differ = from != to;748 let updated_balance_to = if from != to {749 let old_balance = <Balance<T>>::get((collection.id, token, to));750 if old_balance == 0 {751 create_target = true;752 }753 Some(754 old_balance755 .checked_add(amount)756 .ok_or(ArithmeticError::Overflow)?,757 )758 } else {759 None760 };761762 let account_balance_from = if updated_balance_from == 0 {763 Some(764 <AccountBalance<T>>::get((collection.id, from))765 .checked_sub(1)766 767 .ok_or(ArithmeticError::Underflow)?,768 )769 } else {770 None771 };772 773 774 let account_balance_to = if create_target && from_to_differ {775 let account_balance_to = <AccountBalance<T>>::get((collection.id, to))776 .checked_add(1)777 .ok_or(ArithmeticError::Overflow)?;778 ensure!(779 account_balance_to < collection.limits.account_token_ownership_limit(),780 <CommonError<T>>::AccountTokenLimitExceeded,781 );782783 Some(account_balance_to)784 } else {785 None786 };787788 789790 <PalletStructure<T>>::nest_if_sent_to_token(791 from.clone(),792 to,793 collection.id,794 token,795 nesting_budget,796 )?;797798 if let Some(updated_balance_to) = updated_balance_to {799 800 if updated_balance_from == 0 {801 <Balance<T>>::remove((collection.id, token, from));802 <PalletStructure<T>>::unnest_if_nested(from, collection.id, token);803 } else {804 <Balance<T>>::insert((collection.id, token, from), updated_balance_from);805 }806 <Balance<T>>::insert((collection.id, token, to), updated_balance_to);807 if let Some(account_balance_from) = account_balance_from {808 <AccountBalance<T>>::insert((collection.id, from), account_balance_from);809 <Owned<T>>::remove((collection.id, from, token));810 }811 if let Some(account_balance_to) = account_balance_to {812 <AccountBalance<T>>::insert((collection.id, to), account_balance_to);813 <Owned<T>>::insert((collection.id, to, token), true);814 }815 }816817 <PalletEvm<T>>::deposit_log(818 ERC20Events::Transfer {819 from: *from.as_eth(),820 to: *to.as_eth(),821 value: amount.into(),822 }823 .to_log(T::EvmTokenAddressMapping::token_to_address(824 collection.id,825 token,826 )),827 );828829 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(830 collection.id,831 token,832 from.clone(),833 to.clone(),834 amount,835 ));836837 let total_supply = <TotalSupply<T>>::get((collection.id, token));838839 if amount == total_supply {840 841 <PalletEvm<T>>::deposit_log(842 ERC721Events::Transfer {843 from: *from.as_eth(),844 to: *to.as_eth(),845 token_id: token.into(),846 }847 .to_log(collection_id_to_address(collection.id)),848 );849 } else if let Some(updated_balance_to) = updated_balance_to {850 851 852 if initial_balance_from == total_supply {853 854 855 <PalletEvm<T>>::deposit_log(856 ERC721Events::Transfer {857 from: *from.as_eth(),858 to: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,859 token_id: token.into(),860 }861 .to_log(collection_id_to_address(collection.id)),862 );863 } else if updated_balance_to == total_supply {864 865 <PalletEvm<T>>::deposit_log(866 ERC721Events::Transfer {867 from: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,868 to: *to.as_eth(),869 token_id: token.into(),870 }871 .to_log(collection_id_to_address(collection.id)),872 );873 }874 }875876 Ok(())877 }878879 880 881 882 883 884 pub fn create_multiple_items(885 collection: &RefungibleHandle<T>,886 sender: &T::CrossAccountId,887 data: Vec<CreateItemData<T::CrossAccountId>>,888 nesting_budget: &dyn Budget,889 ) -> DispatchResult {890 if !collection.is_owner_or_admin(sender) {891 ensure!(892 collection.permissions.mint_mode(),893 <CommonError<T>>::PublicMintingNotAllowed894 );895 collection.check_allowlist(sender)?;896897 for item in data.iter() {898 for user in item.users.keys() {899 collection.check_allowlist(user)?;900 }901 }902 }903904 for item in data.iter() {905 for (owner, _) in item.users.iter() {906 <PalletCommon<T>>::ensure_correct_receiver(owner)?;907 }908 }909910 911 let totals = data912 .iter()913 .map(|data| {914 Ok(data915 .users916 .iter()917 .map(|u| u.1)918 .try_fold(0u128, |acc, v| acc.checked_add(*v))919 .ok_or(ArithmeticError::Overflow)?)920 })921 .collect::<Result<Vec<_>, DispatchError>>()?;922 for total in &totals {923 ensure!(924 *total <= MAX_REFUNGIBLE_PIECES,925 <Error<T>>::WrongRefungiblePieces926 );927 }928929 let first_token_id = <TokensMinted<T>>::get(collection.id);930 let tokens_minted = first_token_id931 .checked_add(data.len() as u32)932 .ok_or(ArithmeticError::Overflow)?;933 ensure!(934 tokens_minted < collection.limits.token_limit(),935 <CommonError<T>>::CollectionTokenLimitExceeded936 );937938 let mut balances = BTreeMap::new();939 for data in &data {940 for owner in data.users.keys() {941 let balance = balances942 .entry(owner)943 .or_insert_with(|| <AccountBalance<T>>::get((collection.id, owner)));944 *balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;945946 ensure!(947 *balance <= collection.limits.account_token_ownership_limit(),948 <CommonError<T>>::AccountTokenLimitExceeded,949 );950 }951 }952953 for (i, token) in data.iter().enumerate() {954 let token_id = TokenId(first_token_id + i as u32 + 1);955 for (to, _) in token.users.iter() {956 <PalletStructure<T>>::check_nesting(957 sender.clone(),958 to,959 collection.id,960 token_id,961 nesting_budget,962 )?;963 }964 }965966 967968 with_transaction(|| {969 for (i, data) in data.iter().enumerate() {970 let token_id = first_token_id + i as u32 + 1;971 <TotalSupply<T>>::insert((collection.id, token_id), totals[i]);972973 for (user, amount) in data.users.iter() {974 if *amount == 0 {975 continue;976 }977 <Balance<T>>::insert((collection.id, token_id, &user), amount);978 <Owned<T>>::insert((collection.id, &user, TokenId(token_id)), true);979 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(980 user,981 collection.id,982 TokenId(token_id),983 );984 }985986 if let Err(e) = Self::set_token_properties(987 collection,988 sender,989 TokenId(token_id),990 data.properties.clone().into_iter(),991 true,992 nesting_budget,993 ) {994 return TransactionOutcome::Rollback(Err(e));995 }996 }997 TransactionOutcome::Commit(Ok(()))998 })?;9991000 <TokensMinted<T>>::insert(collection.id, tokens_minted);10011002 for (account, balance) in balances {1003 <AccountBalance<T>>::insert((collection.id, account), balance);1004 }10051006 for (i, token) in data.into_iter().enumerate() {1007 let token_id = first_token_id + i as u32 + 1;10081009 let receivers = token1010 .users1011 .into_iter()1012 .filter(|(_, amount)| *amount > 0)1013 .collect::<Vec<_>>();10141015 if let [(user, _)] = receivers.as_slice() {1016 1017 <PalletEvm<T>>::deposit_log(1018 ERC721Events::Transfer {1019 from: H160::default(),1020 to: *user.as_eth(),1021 token_id: token_id.into(),1022 }1023 .to_log(collection_id_to_address(collection.id)),1024 );1025 } else if let [_, ..] = receivers.as_slice() {1026 1027 <PalletEvm<T>>::deposit_log(1028 ERC721Events::Transfer {1029 from: H160::default(),1030 to: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,1031 token_id: token_id.into(),1032 }1033 .to_log(collection_id_to_address(collection.id)),1034 );1035 }10361037 for (user, amount) in receivers.into_iter() {1038 <PalletEvm<T>>::deposit_log(1039 ERC20Events::Transfer {1040 from: H160::default(),1041 to: *user.as_eth(),1042 value: amount.into(),1043 }1044 .to_log(T::EvmTokenAddressMapping::token_to_address(1045 collection.id,1046 TokenId(token_id),1047 )),1048 );1049 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(1050 collection.id,1051 TokenId(token_id),1052 user,1053 amount,1054 ));1055 }1056 }1057 Ok(())1058 }10591060 pub fn set_allowance_unchecked(1061 collection: &RefungibleHandle<T>,1062 sender: &T::CrossAccountId,1063 spender: &T::CrossAccountId,1064 token: TokenId,1065 amount: u128,1066 ) {1067 if amount == 0 {1068 <Allowance<T>>::remove((collection.id, token, sender, spender));1069 } else {1070 <Allowance<T>>::insert((collection.id, token, sender, spender), amount);1071 }10721073 <PalletEvm<T>>::deposit_log(1074 ERC20Events::Approval {1075 owner: *sender.as_eth(),1076 spender: *spender.as_eth(),1077 value: amount.into(),1078 }1079 .to_log(T::EvmTokenAddressMapping::token_to_address(1080 collection.id,1081 token,1082 )),1083 );1084 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(1085 collection.id,1086 token,1087 sender.clone(),1088 spender.clone(),1089 amount,1090 ))1091 }10921093 1094 1095 1096 pub fn set_allowance(1097 collection: &RefungibleHandle<T>,1098 sender: &T::CrossAccountId,1099 spender: &T::CrossAccountId,1100 token: TokenId,1101 amount: u128,1102 ) -> DispatchResult {1103 if collection.permissions.access() == AccessMode::AllowList {1104 collection.check_allowlist(sender)?;1105 collection.check_allowlist(spender)?;1106 }11071108 <PalletCommon<T>>::ensure_correct_receiver(spender)?;11091110 if <Balance<T>>::get((collection.id, token, sender)) < amount {1111 ensure!(1112 collection.ignores_owned_amount(sender) && Self::token_exists(collection, token),1113 <CommonError<T>>::CantApproveMoreThanOwned1114 );1115 }11161117 11181119 Self::set_allowance_unchecked(collection, sender, spender, token, amount);1120 Ok(())1121 }11221123 1124 fn check_allowed(1125 collection: &RefungibleHandle<T>,1126 spender: &T::CrossAccountId,1127 from: &T::CrossAccountId,1128 token: TokenId,1129 amount: u128,1130 nesting_budget: &dyn Budget,1131 ) -> Result<Option<u128>, DispatchError> {1132 if spender.conv_eq(from) {1133 return Ok(None);1134 }1135 if collection.permissions.access() == AccessMode::AllowList {1136 1137 collection.check_allowlist(spender)?;1138 }1139 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {1140 1141 ensure!(1142 <PalletStructure<T>>::check_indirectly_owned(1143 spender.clone(),1144 source.0,1145 source.1,1146 None,1147 nesting_budget1148 )?,1149 <CommonError<T>>::ApprovedValueTooLow,1150 );1151 return Ok(None);1152 }1153 let allowance =1154 <Allowance<T>>::get((collection.id, token, from, &spender)).checked_sub(amount);1155 if allowance.is_none() {1156 ensure!(1157 collection.ignores_allowance(spender),1158 <CommonError<T>>::ApprovedValueTooLow1159 );1160 }1161 Ok(allowance)1162 }11631164 1165 1166 1167 1168 1169 1170 pub fn transfer_from(1171 collection: &RefungibleHandle<T>,1172 spender: &T::CrossAccountId,1173 from: &T::CrossAccountId,1174 to: &T::CrossAccountId,1175 token: TokenId,1176 amount: u128,1177 nesting_budget: &dyn Budget,1178 ) -> DispatchResult {1179 let allowance =1180 Self::check_allowed(collection, spender, from, token, amount, nesting_budget)?;11811182 11831184 Self::transfer(collection, from, to, token, amount, nesting_budget)?;1185 if let Some(allowance) = allowance {1186 Self::set_allowance_unchecked(collection, from, spender, token, allowance);1187 }1188 Ok(())1189 }11901191 1192 1193 1194 1195 1196 1197 pub fn burn_from(1198 collection: &RefungibleHandle<T>,1199 spender: &T::CrossAccountId,1200 from: &T::CrossAccountId,1201 token: TokenId,1202 amount: u128,1203 nesting_budget: &dyn Budget,1204 ) -> DispatchResult {1205 let allowance =1206 Self::check_allowed(collection, spender, from, token, amount, nesting_budget)?;12071208 12091210 Self::burn(collection, from, token, amount)?;1211 if let Some(allowance) = allowance {1212 Self::set_allowance_unchecked(collection, from, spender, token, allowance);1213 }1214 Ok(())1215 }12161217 1218 1219 1220 1221 1222 1223 1224 pub fn create_item(1225 collection: &RefungibleHandle<T>,1226 sender: &T::CrossAccountId,1227 data: CreateItemData<T::CrossAccountId>,1228 nesting_budget: &dyn Budget,1229 ) -> DispatchResult {1230 Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1231 }12321233 1234 1235 1236 1237 1238 1239 1240 pub fn repartition(1241 collection: &RefungibleHandle<T>,1242 owner: &T::CrossAccountId,1243 token: TokenId,1244 amount: u128,1245 ) -> DispatchResult {1246 ensure!(1247 amount <= MAX_REFUNGIBLE_PIECES,1248 <Error<T>>::WrongRefungiblePieces1249 );1250 ensure!(amount > 0, <CommonError<T>>::TokenValueTooLow);1251 1252 let total_pieces = Self::total_pieces(collection.id, token).unwrap_or(u128::MAX);1253 let balance = <Balance<T>>::get((collection.id, token, owner));1254 ensure!(1255 total_pieces == balance,1256 <Error<T>>::RepartitionWhileNotOwningAllPieces1257 );12581259 <Balance<T>>::insert((collection.id, token, owner), amount);1260 <TotalSupply<T>>::insert((collection.id, token), amount);12611262 if amount > total_pieces {1263 let mint_amount = amount - total_pieces;1264 <PalletEvm<T>>::deposit_log(1265 ERC20Events::Transfer {1266 from: H160::default(),1267 to: *owner.as_eth(),1268 value: mint_amount.into(),1269 }1270 .to_log(T::EvmTokenAddressMapping::token_to_address(1271 collection.id,1272 token,1273 )),1274 );1275 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(1276 collection.id,1277 token,1278 owner.clone(),1279 mint_amount,1280 ));1281 } else if total_pieces > amount {1282 let burn_amount = total_pieces - amount;1283 <PalletEvm<T>>::deposit_log(1284 ERC20Events::Transfer {1285 from: *owner.as_eth(),1286 to: H160::default(),1287 value: burn_amount.into(),1288 }1289 .to_log(T::EvmTokenAddressMapping::token_to_address(1290 collection.id,1291 token,1292 )),1293 );1294 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(1295 collection.id,1296 token,1297 owner.clone(),1298 burn_amount,1299 ));1300 }13011302 Ok(())1303 }13041305 fn token_owner(collection_id: CollectionId, token_id: TokenId) -> Option<T::CrossAccountId> {1306 let mut owner = None;1307 let mut count = 0;1308 for key in Balance::<T>::iter_key_prefix((collection_id, token_id)) {1309 count += 1;1310 if count > 1 {1311 return None;1312 }1313 owner = Some(key);1314 }1315 owner1316 }13171318 fn total_pieces(collection_id: CollectionId, token_id: TokenId) -> Option<u128> {1319 <TotalSupply<T>>::try_get((collection_id, token_id)).ok()1320 }13211322 pub fn set_collection_properties(1323 collection: &RefungibleHandle<T>,1324 sender: &T::CrossAccountId,1325 properties: Vec<Property>,1326 ) -> DispatchResult {1327 <PalletCommon<T>>::set_collection_properties(collection, sender, properties)1328 }13291330 pub fn delete_collection_properties(1331 collection: &RefungibleHandle<T>,1332 sender: &T::CrossAccountId,1333 property_keys: Vec<PropertyKey>,1334 ) -> DispatchResult {1335 <PalletCommon<T>>::delete_collection_properties(collection, sender, property_keys)1336 }13371338 pub fn set_token_property_permissions(1339 collection: &RefungibleHandle<T>,1340 sender: &T::CrossAccountId,1341 property_permissions: Vec<PropertyKeyPermission>,1342 ) -> DispatchResult {1343 <PalletCommon<T>>::set_token_property_permissions(collection, sender, property_permissions)1344 }13451346 pub fn set_scoped_token_property_permissions(1347 collection: &RefungibleHandle<T>,1348 sender: &T::CrossAccountId,1349 scope: PropertyScope,1350 property_permissions: Vec<PropertyKeyPermission>,1351 ) -> DispatchResult {1352 <PalletCommon<T>>::set_scoped_token_property_permissions(1353 collection,1354 sender,1355 scope,1356 property_permissions,1357 )1358 }13591360 1361 1362 1363 1364 1365 1366 pub fn token_owners(1367 collection_id: CollectionId,1368 token: TokenId,1369 ) -> Option<Vec<T::CrossAccountId>> {1370 let res: Vec<T::CrossAccountId> = <Balance<T>>::iter_prefix((collection_id, token))1371 .map(|(owner, _amount)| owner)1372 .take(10)1373 .collect();13741375 if res.is_empty() {1376 None1377 } else {1378 Some(res)1379 }1380 }1381}