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, erc::CollectionHelpersEvents,106};107use pallet_structure::Pallet as PalletStructure;108use scale_info::TypeInfo;109use sp_core::{Get, 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, PropertiesPermissionMap, CreateRefungibleExMultipleOwners,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;126127pub type CreateItemData<T> =128 CreateRefungibleExMultipleOwners<<T as pallet_evm::Config>::CrossAccountId>;129pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;130131132133#[struct_versioning::versioned(version = 2, upper)]134#[derive(Encode, Decode, Default, TypeInfo, MaxEncodedLen)]135pub struct ItemData {136 pub const_data: BoundedVec<u8, CustomDataLimit>,137138 #[version(..2)]139 pub variable_data: BoundedVec<u8, CustomDataLimit>,140}141142#[frame_support::pallet]143pub mod pallet {144 use super::*;145 use frame_support::{146 Blake2_128, Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key,147 traits::StorageVersion,148 };149 use frame_system::pallet_prelude::*;150 use up_data_structs::{CollectionId, TokenId};151 use super::weights::WeightInfo;152153 #[pallet::error]154 pub enum Error<T> {155 156 NotRefungibleDataUsedToMintFungibleCollectionToken,157 158 WrongRefungiblePieces,159 160 RepartitionWhileNotOwningAllPieces,161 162 RefungibleDisallowsNesting,163 164 SettingPropertiesNotAllowed,165 }166167 #[pallet::config]168 pub trait Config:169 frame_system::Config + pallet_common::Config + pallet_structure::Config170 {171 type WeightInfo: WeightInfo;172 }173174 const STORAGE_VERSION: StorageVersion = StorageVersion::new(2);175176 #[pallet::pallet]177 #[pallet::storage_version(STORAGE_VERSION)]178 #[pallet::generate_store(pub(super) trait Store)]179 pub struct Pallet<T>(_);180181 182 #[pallet::storage]183 pub type TokensMinted<T: Config> =184 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;185186 187 #[pallet::storage]188 pub type TokensBurnt<T: Config> =189 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;190191 192 193 #[pallet::storage]194 #[deprecated(since = "0.2.0", note = "ItemData is no more contains usefull data")]195 pub type TokenData<T: Config> = StorageNMap<196 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),197 Value = ItemData,198 QueryKind = ValueQuery,199 >;200201 202 #[pallet::storage]203 #[pallet::getter(fn token_properties)]204 pub type TokenProperties<T: Config> = StorageNMap<205 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),206 Value = up_data_structs::Properties,207 QueryKind = ValueQuery,208 OnEmpty = up_data_structs::TokenProperties,209 >;210211 212 #[pallet::storage]213 pub type TotalSupply<T: Config> = StorageNMap<214 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),215 Value = u128,216 QueryKind = ValueQuery,217 >;218219 220 #[pallet::storage]221 pub type Owned<T: Config> = StorageNMap<222 Key = (223 Key<Twox64Concat, CollectionId>,224 Key<Blake2_128Concat, T::CrossAccountId>,225 Key<Twox64Concat, TokenId>,226 ),227 Value = bool,228 QueryKind = ValueQuery,229 >;230231 232 #[pallet::storage]233 pub type AccountBalance<T: Config> = StorageNMap<234 Key = (235 Key<Twox64Concat, CollectionId>,236 237 Key<Blake2_128Concat, T::CrossAccountId>,238 ),239 Value = u32,240 QueryKind = ValueQuery,241 >;242243 244 #[pallet::storage]245 pub type Balance<T: Config> = StorageNMap<246 Key = (247 Key<Twox64Concat, CollectionId>,248 Key<Twox64Concat, TokenId>,249 250 Key<Blake2_128Concat, T::CrossAccountId>,251 ),252 Value = u128,253 QueryKind = ValueQuery,254 >;255256 257 #[pallet::storage]258 pub type Allowance<T: Config> = StorageNMap<259 Key = (260 Key<Twox64Concat, CollectionId>,261 Key<Twox64Concat, TokenId>,262 263 Key<Blake2_128, T::CrossAccountId>,264 265 Key<Blake2_128Concat, T::CrossAccountId>,266 ),267 Value = u128,268 QueryKind = ValueQuery,269 >;270271 272 #[pallet::storage]273 pub type CollectionAllowance<T: Config> = StorageNMap<274 Key = (275 Key<Twox64Concat, CollectionId>,276 Key<Blake2_128Concat, T::CrossAccountId>,277 Key<Blake2_128Concat, T::CrossAccountId>,278 ),279 Value = bool,280 QueryKind = ValueQuery,281 >;282283 #[pallet::hooks]284 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {285 fn on_runtime_upgrade() -> Weight {286 let storage_version = StorageVersion::get::<Pallet<T>>();287 if storage_version < StorageVersion::new(2) {288 #[allow(deprecated)]289 let _ = <TokenData<T>>::clear(u32::MAX, None);290 }291 StorageVersion::new(2).put::<Pallet<T>>();292293 Weight::zero()294 }295 }296}297298pub struct RefungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);299impl<T: Config> RefungibleHandle<T> {300 pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {301 Self(inner)302 }303 pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {304 self.0305 }306 pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {307 &mut self.0308 }309}310311impl<T: Config> Deref for RefungibleHandle<T> {312 type Target = pallet_common::CollectionHandle<T>;313314 fn deref(&self) -> &Self::Target {315 &self.0316 }317}318319impl<T: Config> WithRecorder<T> for RefungibleHandle<T> {320 fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {321 self.0.recorder()322 }323 fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {324 self.0.into_recorder()325 }326}327328impl<T: Config> Pallet<T> {329 330 pub fn total_supply(collection: &RefungibleHandle<T>) -> u32 {331 <TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)332 }333334 335 336 337 pub fn token_exists(collection: &RefungibleHandle<T>, token: TokenId) -> bool {338 <TotalSupply<T>>::contains_key((collection.id, token))339 }340341 pub fn set_scoped_token_property(342 collection_id: CollectionId,343 token_id: TokenId,344 scope: PropertyScope,345 property: Property,346 ) -> DispatchResult {347 TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {348 properties.try_scoped_set(scope, property.key, property.value)349 })350 .map_err(<CommonError<T>>::from)?;351352 Ok(())353 }354355 pub fn set_scoped_token_properties(356 collection_id: CollectionId,357 token_id: TokenId,358 scope: PropertyScope,359 properties: impl Iterator<Item = Property>,360 ) -> DispatchResult {361 TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {362 stored_properties.try_scoped_set_from_iter(scope, properties)363 })364 .map_err(<CommonError<T>>::from)?;365366 Ok(())367 }368}369370371impl<T: Config> Pallet<T> {372 373 374 375 376 377 pub fn init_collection(378 owner: T::CrossAccountId,379 payer: T::CrossAccountId,380 data: CreateCollectionData<T::AccountId>,381 flags: CollectionFlags,382 ) -> Result<CollectionId, DispatchError> {383 <PalletCommon<T>>::init_collection(owner, payer, data, flags)384 }385386 387 388 389 390 pub fn destroy_collection(391 collection: RefungibleHandle<T>,392 sender: &T::CrossAccountId,393 ) -> DispatchResult {394 let id = collection.id;395396 if Self::collection_has_tokens(id) {397 return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());398 }399400 401402 PalletCommon::destroy_collection(collection.0, sender)?;403404 <TokensMinted<T>>::remove(id);405 <TokensBurnt<T>>::remove(id);406 let _ = <TotalSupply<T>>::clear_prefix((id,), u32::MAX, None);407 let _ = <Balance<T>>::clear_prefix((id,), u32::MAX, None);408 let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);409 let _ = <Owned<T>>::clear_prefix((id,), u32::MAX, None);410 let _ = <AccountBalance<T>>::clear_prefix((id,), u32::MAX, None);411 Ok(())412 }413414 fn collection_has_tokens(collection_id: CollectionId) -> bool {415 <TotalSupply<T>>::iter_prefix((collection_id,))416 .next()417 .is_some()418 }419420 pub fn burn_token_unchecked(421 collection: &RefungibleHandle<T>,422 owner: &T::CrossAccountId,423 token_id: TokenId,424 ) -> DispatchResult {425 let burnt = <TokensBurnt<T>>::get(collection.id)426 .checked_add(1)427 .ok_or(ArithmeticError::Overflow)?;428429 <TokensBurnt<T>>::insert(collection.id, burnt);430 <TokenProperties<T>>::remove((collection.id, token_id));431 <TotalSupply<T>>::remove((collection.id, token_id));432 let _ = <Balance<T>>::clear_prefix((collection.id, token_id), u32::MAX, None);433 let _ = <Allowance<T>>::clear_prefix((collection.id, token_id), u32::MAX, None);434 <PalletEvm<T>>::deposit_log(435 ERC721Events::Transfer {436 from: *owner.as_eth(),437 to: H160::default(),438 token_id: token_id.into(),439 }440 .to_log(collection_id_to_address(collection.id)),441 );442 Ok(())443 }444445 446 447 448 449 450 451 452 453 454 455 456 pub fn burn(457 collection: &RefungibleHandle<T>,458 owner: &T::CrossAccountId,459 token: TokenId,460 amount: u128,461 ) -> DispatchResult {462 if <Balance<T>>::get((collection.id, token, owner)) == 0 {463 return Err(<CommonError<T>>::TokenValueTooLow.into());464 }465466 let total_supply = <TotalSupply<T>>::get((collection.id, token))467 .checked_sub(amount)468 .ok_or(<CommonError<T>>::TokenValueTooLow)?;469470 471 if total_supply == 0 {472 473 ensure!(474 <Balance<T>>::get((collection.id, token, owner)) == amount,475 <CommonError<T>>::TokenValueTooLow476 );477 let account_balance = <AccountBalance<T>>::get((collection.id, owner))478 .checked_sub(1)479 480 .ok_or(ArithmeticError::Underflow)?;481482 483484 <Owned<T>>::remove((collection.id, owner, token));485 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, token);486 <AccountBalance<T>>::insert((collection.id, owner), account_balance);487 Self::burn_token_unchecked(collection, owner, token)?;488 <PalletEvm<T>>::deposit_log(489 ERC20Events::Transfer {490 from: *owner.as_eth(),491 to: H160::default(),492 value: amount.into(),493 }494 .to_log(collection_id_to_address(collection.id)),495 );496 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(497 collection.id,498 token,499 owner.clone(),500 amount,501 ));502 return Ok(());503 }504505 let balance = <Balance<T>>::get((collection.id, token, owner))506 .checked_sub(amount)507 .ok_or(<CommonError<T>>::TokenValueTooLow)?;508 let account_balance = if balance == 0 {509 <AccountBalance<T>>::get((collection.id, owner))510 .checked_sub(1)511 512 .ok_or(ArithmeticError::Underflow)?513 } else {514 0515 };516517 518519 if balance == 0 {520 <Owned<T>>::remove((collection.id, owner, token));521 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, token);522 <Balance<T>>::remove((collection.id, token, owner));523 <AccountBalance<T>>::insert((collection.id, owner), account_balance);524525 if let Some(user) = Self::token_owner(collection.id, token) {526 <PalletEvm<T>>::deposit_log(527 ERC721Events::Transfer {528 from: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,529 to: *user.as_eth(),530 token_id: token.into(),531 }532 .to_log(collection_id_to_address(collection.id)),533 );534 }535 } else {536 <Balance<T>>::insert((collection.id, token, owner), balance);537 }538 <TotalSupply<T>>::insert((collection.id, token), total_supply);539540 <PalletEvm<T>>::deposit_log(541 ERC20Events::Transfer {542 from: *owner.as_eth(),543 to: H160::default(),544 value: amount.into(),545 }546 .to_log(T::EvmTokenAddressMapping::token_to_address(547 collection.id,548 token,549 )),550 );551 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(552 collection.id,553 token,554 owner.clone(),555 amount,556 ));557 Ok(())558 }559560 #[transactional]561 fn modify_token_properties(562 collection: &RefungibleHandle<T>,563 sender: &T::CrossAccountId,564 token_id: TokenId,565 properties: impl Iterator<Item = (PropertyKey, Option<PropertyValue>)>,566 is_token_create: bool,567 nesting_budget: &dyn Budget,568 ) -> DispatchResult {569 let is_collection_admin = || collection.is_owner_or_admin(sender);570 let is_token_owner = || -> Result<bool, DispatchError> {571 let balance = collection.balance(sender.clone(), token_id);572 let total_pieces: u128 =573 Self::total_pieces(collection.id, token_id).unwrap_or(u128::MAX);574 if balance != total_pieces {575 return Ok(false);576 }577578 let is_bundle_owner = <PalletStructure<T>>::check_indirectly_owned(579 sender.clone(),580 collection.id,581 token_id,582 None,583 nesting_budget,584 )?;585586 Ok(is_bundle_owner)587 };588589 for (key, value) in properties {590 let permission = <PalletCommon<T>>::property_permissions(collection.id)591 .get(&key)592 .cloned()593 .unwrap_or_else(PropertyPermission::none);594595 let is_property_exists = TokenProperties::<T>::get((collection.id, token_id))596 .get(&key)597 .is_some();598599 match permission {600 PropertyPermission { mutable: false, .. } if is_property_exists => {601 return Err(<CommonError<T>>::NoPermission.into());602 }603604 PropertyPermission {605 collection_admin,606 token_owner,607 ..608 } => {609 610 let is_token_create =611 is_token_create && (collection_admin || token_owner) && value.is_some();612 if !(is_token_create613 || (collection_admin && is_collection_admin())614 || (token_owner && is_token_owner()?))615 {616 fail!(<CommonError<T>>::NoPermission);617 }618 }619 }620621 match value {622 Some(value) => {623 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {624 properties.try_set(key.clone(), value)625 })626 .map_err(<CommonError<T>>::from)?;627628 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertySet(629 collection.id,630 token_id,631 key,632 ));633 }634 None => {635 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {636 properties.remove(&key)637 })638 .map_err(<CommonError<T>>::from)?;639640 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertyDeleted(641 collection.id,642 token_id,643 key,644 ));645 }646 }647648 <PalletEvm<T>>::deposit_log(649 CollectionHelpersEvents::TokenChanged {650 collection_id: collection_id_to_address(collection.id),651 token_id: token_id.into(),652 }653 .to_log(T::ContractAddress::get()),654 );655 }656657 Ok(())658 }659660 pub fn set_token_properties(661 collection: &RefungibleHandle<T>,662 sender: &T::CrossAccountId,663 token_id: TokenId,664 properties: impl Iterator<Item = Property>,665 is_token_create: bool,666 nesting_budget: &dyn Budget,667 ) -> DispatchResult {668 Self::modify_token_properties(669 collection,670 sender,671 token_id,672 properties.map(|p| (p.key, Some(p.value))),673 is_token_create,674 nesting_budget,675 )676 }677678 pub fn set_token_property(679 collection: &RefungibleHandle<T>,680 sender: &T::CrossAccountId,681 token_id: TokenId,682 property: Property,683 nesting_budget: &dyn Budget,684 ) -> DispatchResult {685 let is_token_create = false;686687 Self::set_token_properties(688 collection,689 sender,690 token_id,691 [property].into_iter(),692 is_token_create,693 nesting_budget,694 )695 }696697 pub fn delete_token_properties(698 collection: &RefungibleHandle<T>,699 sender: &T::CrossAccountId,700 token_id: TokenId,701 property_keys: impl Iterator<Item = PropertyKey>,702 nesting_budget: &dyn Budget,703 ) -> DispatchResult {704 let is_token_create = false;705706 Self::modify_token_properties(707 collection,708 sender,709 token_id,710 property_keys.into_iter().map(|key| (key, None)),711 is_token_create,712 nesting_budget,713 )714 }715716 pub fn delete_token_property(717 collection: &RefungibleHandle<T>,718 sender: &T::CrossAccountId,719 token_id: TokenId,720 property_key: PropertyKey,721 nesting_budget: &dyn Budget,722 ) -> DispatchResult {723 Self::delete_token_properties(724 collection,725 sender,726 token_id,727 [property_key].into_iter(),728 nesting_budget,729 )730 }731732 733 734 735 736 737 738 739 740 741 pub fn transfer(742 collection: &RefungibleHandle<T>,743 from: &T::CrossAccountId,744 to: &T::CrossAccountId,745 token: TokenId,746 amount: u128,747 nesting_budget: &dyn Budget,748 ) -> DispatchResult {749 ensure!(750 collection.limits.transfers_enabled(),751 <CommonError<T>>::TransferNotAllowed752 );753754 if collection.permissions.access() == AccessMode::AllowList {755 collection.check_allowlist(from)?;756 collection.check_allowlist(to)?;757 }758 <PalletCommon<T>>::ensure_correct_receiver(to)?;759760 let initial_balance_from = <Balance<T>>::get((collection.id, token, from));761762 if initial_balance_from == 0 {763 return Err(<CommonError<T>>::TokenValueTooLow.into());764 }765766 let updated_balance_from = initial_balance_from767 .checked_sub(amount)768 .ok_or(<CommonError<T>>::TokenValueTooLow)?;769 let mut create_target = false;770 let from_to_differ = from != to;771 let updated_balance_to = if from != to && amount != 0 {772 let old_balance = <Balance<T>>::get((collection.id, token, to));773 if old_balance == 0 {774 create_target = true;775 }776 Some(777 old_balance778 .checked_add(amount)779 .ok_or(ArithmeticError::Overflow)?,780 )781 } else {782 None783 };784785 let account_balance_from = if updated_balance_from == 0 {786 Some(787 <AccountBalance<T>>::get((collection.id, from))788 .checked_sub(1)789 790 .ok_or(ArithmeticError::Underflow)?,791 )792 } else {793 None794 };795 796 797 let account_balance_to = if create_target && from_to_differ {798 let account_balance_to = <AccountBalance<T>>::get((collection.id, to))799 .checked_add(1)800 .ok_or(ArithmeticError::Overflow)?;801 ensure!(802 account_balance_to < collection.limits.account_token_ownership_limit(),803 <CommonError<T>>::AccountTokenLimitExceeded,804 );805806 Some(account_balance_to)807 } else {808 None809 };810811 812813 if let Some(updated_balance_to) = updated_balance_to {814 815816 <PalletStructure<T>>::nest_if_sent_to_token(817 from.clone(),818 to,819 collection.id,820 token,821 nesting_budget,822 )?;823824 if updated_balance_from == 0 {825 <Balance<T>>::remove((collection.id, token, from));826 <PalletStructure<T>>::unnest_if_nested(from, collection.id, token);827 } else {828 <Balance<T>>::insert((collection.id, token, from), updated_balance_from);829 }830 <Balance<T>>::insert((collection.id, token, to), updated_balance_to);831 if let Some(account_balance_from) = account_balance_from {832 <AccountBalance<T>>::insert((collection.id, from), account_balance_from);833 <Owned<T>>::remove((collection.id, from, token));834 }835 if let Some(account_balance_to) = account_balance_to {836 <AccountBalance<T>>::insert((collection.id, to), account_balance_to);837 <Owned<T>>::insert((collection.id, to, token), true);838 }839 }840841 <PalletEvm<T>>::deposit_log(842 ERC20Events::Transfer {843 from: *from.as_eth(),844 to: *to.as_eth(),845 value: amount.into(),846 }847 .to_log(T::EvmTokenAddressMapping::token_to_address(848 collection.id,849 token,850 )),851 );852853 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(854 collection.id,855 token,856 from.clone(),857 to.clone(),858 amount,859 ));860861 let total_supply = <TotalSupply<T>>::get((collection.id, token));862863 if amount == total_supply {864 865 <PalletEvm<T>>::deposit_log(866 ERC721Events::Transfer {867 from: *from.as_eth(),868 to: *to.as_eth(),869 token_id: token.into(),870 }871 .to_log(collection_id_to_address(collection.id)),872 );873 } else if let Some(updated_balance_to) = updated_balance_to {874 875 876 if initial_balance_from == total_supply {877 878 879 <PalletEvm<T>>::deposit_log(880 ERC721Events::Transfer {881 from: *from.as_eth(),882 to: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,883 token_id: token.into(),884 }885 .to_log(collection_id_to_address(collection.id)),886 );887 } else if updated_balance_to == total_supply {888 889 <PalletEvm<T>>::deposit_log(890 ERC721Events::Transfer {891 from: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,892 to: *to.as_eth(),893 token_id: token.into(),894 }895 .to_log(collection_id_to_address(collection.id)),896 );897 }898 }899900 Ok(())901 }902903 904 905 906 907 908 pub fn create_multiple_items(909 collection: &RefungibleHandle<T>,910 sender: &T::CrossAccountId,911 data: Vec<CreateItemData<T>>,912 nesting_budget: &dyn Budget,913 ) -> DispatchResult {914 if !collection.is_owner_or_admin(sender) {915 ensure!(916 collection.permissions.mint_mode(),917 <CommonError<T>>::PublicMintingNotAllowed918 );919 collection.check_allowlist(sender)?;920921 for item in data.iter() {922 for user in item.users.keys() {923 collection.check_allowlist(user)?;924 }925 }926 }927928 for item in data.iter() {929 for (owner, _) in item.users.iter() {930 <PalletCommon<T>>::ensure_correct_receiver(owner)?;931 }932 }933934 935 let totals = data936 .iter()937 .map(|data| {938 Ok(data939 .users940 .iter()941 .map(|u| u.1)942 .try_fold(0u128, |acc, v| acc.checked_add(*v))943 .ok_or(ArithmeticError::Overflow)?)944 })945 .collect::<Result<Vec<_>, DispatchError>>()?;946 for total in &totals {947 ensure!(948 *total <= MAX_REFUNGIBLE_PIECES,949 <Error<T>>::WrongRefungiblePieces950 );951 }952953 let first_token_id = <TokensMinted<T>>::get(collection.id);954 let tokens_minted = first_token_id955 .checked_add(data.len() as u32)956 .ok_or(ArithmeticError::Overflow)?;957 ensure!(958 tokens_minted < collection.limits.token_limit(),959 <CommonError<T>>::CollectionTokenLimitExceeded960 );961962 let mut balances = BTreeMap::new();963 for data in &data {964 for owner in data.users.keys() {965 let balance = balances966 .entry(owner)967 .or_insert_with(|| <AccountBalance<T>>::get((collection.id, owner)));968 *balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;969970 ensure!(971 *balance <= collection.limits.account_token_ownership_limit(),972 <CommonError<T>>::AccountTokenLimitExceeded,973 );974 }975 }976977 for (i, token) in data.iter().enumerate() {978 let token_id = TokenId(first_token_id + i as u32 + 1);979 for (to, _) in token.users.iter() {980 <PalletStructure<T>>::check_nesting(981 sender.clone(),982 to,983 collection.id,984 token_id,985 nesting_budget,986 )?;987 }988 }989990 991992 with_transaction(|| {993 for (i, data) in data.iter().enumerate() {994 let token_id = first_token_id + i as u32 + 1;995 <TotalSupply<T>>::insert((collection.id, token_id), totals[i]);996997 for (user, amount) in data.users.iter() {998 if *amount == 0 {999 continue;1000 }1001 <Balance<T>>::insert((collection.id, token_id, &user), amount);1002 <Owned<T>>::insert((collection.id, &user, TokenId(token_id)), true);1003 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(1004 user,1005 collection.id,1006 TokenId(token_id),1007 );1008 }10091010 if let Err(e) = Self::set_token_properties(1011 collection,1012 sender,1013 TokenId(token_id),1014 data.properties.clone().into_iter(),1015 true,1016 nesting_budget,1017 ) {1018 return TransactionOutcome::Rollback(Err(e));1019 }1020 }1021 TransactionOutcome::Commit(Ok(()))1022 })?;10231024 <TokensMinted<T>>::insert(collection.id, tokens_minted);10251026 for (account, balance) in balances {1027 <AccountBalance<T>>::insert((collection.id, account), balance);1028 }10291030 for (i, token) in data.into_iter().enumerate() {1031 let token_id = first_token_id + i as u32 + 1;10321033 let receivers = token1034 .users1035 .into_iter()1036 .filter(|(_, amount)| *amount > 0)1037 .collect::<Vec<_>>();10381039 if let [(user, _)] = receivers.as_slice() {1040 1041 <PalletEvm<T>>::deposit_log(1042 ERC721Events::Transfer {1043 from: H160::default(),1044 to: *user.as_eth(),1045 token_id: token_id.into(),1046 }1047 .to_log(collection_id_to_address(collection.id)),1048 );1049 } else if let [_, ..] = receivers.as_slice() {1050 1051 <PalletEvm<T>>::deposit_log(1052 ERC721Events::Transfer {1053 from: H160::default(),1054 to: erc::ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,1055 token_id: token_id.into(),1056 }1057 .to_log(collection_id_to_address(collection.id)),1058 );1059 }10601061 for (user, amount) in receivers.into_iter() {1062 <PalletEvm<T>>::deposit_log(1063 ERC20Events::Transfer {1064 from: H160::default(),1065 to: *user.as_eth(),1066 value: amount.into(),1067 }1068 .to_log(T::EvmTokenAddressMapping::token_to_address(1069 collection.id,1070 TokenId(token_id),1071 )),1072 );1073 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(1074 collection.id,1075 TokenId(token_id),1076 user,1077 amount,1078 ));1079 }1080 }1081 Ok(())1082 }10831084 pub fn set_allowance_unchecked(1085 collection: &RefungibleHandle<T>,1086 sender: &T::CrossAccountId,1087 spender: &T::CrossAccountId,1088 token: TokenId,1089 amount: u128,1090 ) {1091 if amount == 0 {1092 <Allowance<T>>::remove((collection.id, token, sender, spender));1093 } else {1094 <Allowance<T>>::insert((collection.id, token, sender, spender), amount);1095 }10961097 <PalletEvm<T>>::deposit_log(1098 ERC20Events::Approval {1099 owner: *sender.as_eth(),1100 spender: *spender.as_eth(),1101 value: amount.into(),1102 }1103 .to_log(T::EvmTokenAddressMapping::token_to_address(1104 collection.id,1105 token,1106 )),1107 );1108 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(1109 collection.id,1110 token,1111 sender.clone(),1112 spender.clone(),1113 amount,1114 ))1115 }11161117 1118 1119 1120 pub fn set_allowance(1121 collection: &RefungibleHandle<T>,1122 sender: &T::CrossAccountId,1123 spender: &T::CrossAccountId,1124 token: TokenId,1125 amount: u128,1126 ) -> DispatchResult {1127 if collection.permissions.access() == AccessMode::AllowList {1128 collection.check_allowlist(sender)?;1129 collection.check_allowlist(spender)?;1130 }11311132 <PalletCommon<T>>::ensure_correct_receiver(spender)?;11331134 if <Balance<T>>::get((collection.id, token, sender)) < amount {1135 ensure!(1136 collection.ignores_owned_amount(sender) && Self::token_exists(collection, token),1137 <CommonError<T>>::CantApproveMoreThanOwned1138 );1139 }11401141 11421143 Self::set_allowance_unchecked(collection, sender, spender, token, amount);1144 Ok(())1145 }11461147 1148 fn check_allowed(1149 collection: &RefungibleHandle<T>,1150 spender: &T::CrossAccountId,1151 from: &T::CrossAccountId,1152 token: TokenId,1153 amount: u128,1154 nesting_budget: &dyn Budget,1155 ) -> Result<Option<u128>, DispatchError> {1156 if spender.conv_eq(from) {1157 return Ok(None);1158 }1159 if collection.permissions.access() == AccessMode::AllowList {1160 1161 collection.check_allowlist(spender)?;1162 }1163 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {1164 1165 ensure!(1166 <PalletStructure<T>>::check_indirectly_owned(1167 spender.clone(),1168 source.0,1169 source.1,1170 None,1171 nesting_budget1172 )?,1173 <CommonError<T>>::ApprovedValueTooLow,1174 );1175 return Ok(None);1176 }1177 let allowance =1178 <Allowance<T>>::get((collection.id, token, from, &spender)).checked_sub(amount);11791180 1181 if <CollectionAllowance<T>>::get((collection.id, from, spender)) {1182 return Ok(allowance);1183 }11841185 if allowance.is_none() {1186 ensure!(1187 collection.ignores_allowance(spender),1188 <CommonError<T>>::ApprovedValueTooLow1189 );1190 }1191 Ok(allowance)1192 }11931194 1195 1196 1197 1198 1199 1200 pub fn transfer_from(1201 collection: &RefungibleHandle<T>,1202 spender: &T::CrossAccountId,1203 from: &T::CrossAccountId,1204 to: &T::CrossAccountId,1205 token: TokenId,1206 amount: u128,1207 nesting_budget: &dyn Budget,1208 ) -> DispatchResult {1209 let allowance =1210 Self::check_allowed(collection, spender, from, token, amount, nesting_budget)?;12111212 12131214 Self::transfer(collection, from, to, token, amount, nesting_budget)?;1215 if let Some(allowance) = allowance {1216 Self::set_allowance_unchecked(collection, from, spender, token, allowance);1217 }1218 Ok(())1219 }12201221 1222 1223 1224 1225 1226 1227 pub fn burn_from(1228 collection: &RefungibleHandle<T>,1229 spender: &T::CrossAccountId,1230 from: &T::CrossAccountId,1231 token: TokenId,1232 amount: u128,1233 nesting_budget: &dyn Budget,1234 ) -> DispatchResult {1235 let allowance =1236 Self::check_allowed(collection, spender, from, token, amount, nesting_budget)?;12371238 12391240 Self::burn(collection, from, token, amount)?;1241 if let Some(allowance) = allowance {1242 Self::set_allowance_unchecked(collection, from, spender, token, allowance);1243 }1244 Ok(())1245 }12461247 1248 1249 1250 1251 1252 1253 1254 pub fn create_item(1255 collection: &RefungibleHandle<T>,1256 sender: &T::CrossAccountId,1257 data: CreateItemData<T>,1258 nesting_budget: &dyn Budget,1259 ) -> DispatchResult {1260 Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1261 }12621263 1264 1265 1266 1267 1268 1269 1270 pub fn repartition(1271 collection: &RefungibleHandle<T>,1272 owner: &T::CrossAccountId,1273 token: TokenId,1274 amount: u128,1275 ) -> DispatchResult {1276 ensure!(1277 amount <= MAX_REFUNGIBLE_PIECES,1278 <Error<T>>::WrongRefungiblePieces1279 );1280 ensure!(amount > 0, <CommonError<T>>::TokenValueTooLow);1281 1282 let total_pieces = Self::total_pieces(collection.id, token).unwrap_or(u128::MAX);1283 let balance = <Balance<T>>::get((collection.id, token, owner));1284 ensure!(1285 total_pieces == balance,1286 <Error<T>>::RepartitionWhileNotOwningAllPieces1287 );12881289 <Balance<T>>::insert((collection.id, token, owner), amount);1290 <TotalSupply<T>>::insert((collection.id, token), amount);12911292 if amount > total_pieces {1293 let mint_amount = amount - total_pieces;1294 <PalletEvm<T>>::deposit_log(1295 ERC20Events::Transfer {1296 from: H160::default(),1297 to: *owner.as_eth(),1298 value: mint_amount.into(),1299 }1300 .to_log(T::EvmTokenAddressMapping::token_to_address(1301 collection.id,1302 token,1303 )),1304 );1305 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(1306 collection.id,1307 token,1308 owner.clone(),1309 mint_amount,1310 ));1311 } else if total_pieces > amount {1312 let burn_amount = total_pieces - amount;1313 <PalletEvm<T>>::deposit_log(1314 ERC20Events::Transfer {1315 from: *owner.as_eth(),1316 to: H160::default(),1317 value: burn_amount.into(),1318 }1319 .to_log(T::EvmTokenAddressMapping::token_to_address(1320 collection.id,1321 token,1322 )),1323 );1324 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(1325 collection.id,1326 token,1327 owner.clone(),1328 burn_amount,1329 ));1330 }13311332 Ok(())1333 }13341335 fn token_owner(collection_id: CollectionId, token_id: TokenId) -> Option<T::CrossAccountId> {1336 let mut owner = None;1337 let mut count = 0;1338 for key in Balance::<T>::iter_key_prefix((collection_id, token_id)) {1339 count += 1;1340 if count > 1 {1341 return None;1342 }1343 owner = Some(key);1344 }1345 owner1346 }13471348 fn total_pieces(collection_id: CollectionId, token_id: TokenId) -> Option<u128> {1349 <TotalSupply<T>>::try_get((collection_id, token_id)).ok()1350 }13511352 pub fn set_collection_properties(1353 collection: &RefungibleHandle<T>,1354 sender: &T::CrossAccountId,1355 properties: Vec<Property>,1356 ) -> DispatchResult {1357 <PalletCommon<T>>::set_collection_properties(collection, sender, properties)1358 }13591360 pub fn delete_collection_properties(1361 collection: &RefungibleHandle<T>,1362 sender: &T::CrossAccountId,1363 property_keys: Vec<PropertyKey>,1364 ) -> DispatchResult {1365 <PalletCommon<T>>::delete_collection_properties(collection, sender, property_keys)1366 }13671368 pub fn set_token_property_permissions(1369 collection: &RefungibleHandle<T>,1370 sender: &T::CrossAccountId,1371 property_permissions: Vec<PropertyKeyPermission>,1372 ) -> DispatchResult {1373 <PalletCommon<T>>::set_token_property_permissions(collection, sender, property_permissions)1374 }13751376 pub fn token_property_permission(collection_id: CollectionId) -> PropertiesPermissionMap {1377 <PalletCommon<T>>::property_permissions(collection_id)1378 }13791380 pub fn set_scoped_token_property_permissions(1381 collection: &RefungibleHandle<T>,1382 sender: &T::CrossAccountId,1383 scope: PropertyScope,1384 property_permissions: Vec<PropertyKeyPermission>,1385 ) -> DispatchResult {1386 <PalletCommon<T>>::set_scoped_token_property_permissions(1387 collection,1388 sender,1389 scope,1390 property_permissions,1391 )1392 }13931394 1395 1396 1397 1398 1399 1400 pub fn token_owners(1401 collection_id: CollectionId,1402 token: TokenId,1403 ) -> Option<Vec<T::CrossAccountId>> {1404 let res: Vec<T::CrossAccountId> = <Balance<T>>::iter_prefix((collection_id, token))1405 .map(|(owner, _amount)| owner)1406 .take(10)1407 .collect();14081409 if res.is_empty() {1410 None1411 } else {1412 Some(res)1413 }1414 }14151416 1417 1418 1419 1420 1421 1422 pub fn set_allowance_for_all(1423 collection: &RefungibleHandle<T>,1424 owner: &T::CrossAccountId,1425 operator: &T::CrossAccountId,1426 approve: bool,1427 ) -> DispatchResult {1428 if collection.permissions.access() == AccessMode::AllowList {1429 collection.check_allowlist(owner)?;1430 collection.check_allowlist(operator)?;1431 }14321433 <PalletCommon<T>>::ensure_correct_receiver(operator)?;14341435 14361437 <CollectionAllowance<T>>::insert((collection.id, owner, operator), approve);1438 <PalletEvm<T>>::deposit_log(1439 ERC721Events::ApprovalForAll {1440 owner: *owner.as_eth(),1441 operator: *operator.as_eth(),1442 approved: approve,1443 }1444 .to_log(collection_id_to_address(collection.id)),1445 );1446 <PalletCommon<T>>::deposit_event(CommonEvent::ApprovedForAll(1447 collection.id,1448 owner.clone(),1449 operator.clone(),1450 approve,1451 ));1452 Ok(())1453 }14541455 1456 pub fn allowance_for_all(1457 collection: &RefungibleHandle<T>,1458 owner: &T::CrossAccountId,1459 operator: &T::CrossAccountId,1460 ) -> bool {1461 <CollectionAllowance<T>>::get((collection.id, owner, operator))1462 }14631464 pub fn repair_item(collection: &RefungibleHandle<T>, token: TokenId) -> DispatchResult {1465 <TokenProperties<T>>::mutate((collection.id, token), |properties| {1466 properties.recompute_consumed_space();1467 });14681469 Ok(())1470 }1471}