12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879#![cfg_attr(not(feature = "std"), no_std)]8081use core::ops::Deref;82use evm_coder::ToLog;83use frame_support::{ensure};84use pallet_evm::account::CrossAccountId;85use up_data_structs::{86 AccessMode, CollectionId, TokenId, CreateCollectionData, mapping::TokenAddressMapping,87 budget::Budget,88};89use pallet_common::{90 Error as CommonError, Event as CommonEvent, Pallet as PalletCommon,91 eth::collection_id_to_address,92};93use pallet_evm::Pallet as PalletEvm;94use pallet_structure::Pallet as PalletStructure;95use pallet_evm_coder_substrate::WithRecorder;96use sp_core::H160;97use sp_runtime::{ArithmeticError, DispatchError, DispatchResult};98use sp_std::{collections::btree_map::BTreeMap, vec::Vec};99100pub use pallet::*;101102use crate::erc::ERC20Events;103#[cfg(feature = "runtime-benchmarks")]104pub mod benchmarking;105pub mod common;106pub mod erc;107pub mod weights;108109pub type CreateItemData<T> = (<T as pallet_evm::account::Config>::CrossAccountId, u128);110pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;111112#[frame_support::pallet]113pub mod pallet {114 use frame_support::{Blake2_128, Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key};115 use up_data_structs::CollectionId;116 use super::weights::WeightInfo;117118 #[pallet::error]119 pub enum Error<T> {120 121 NotFungibleDataUsedToMintFungibleCollectionToken,122 123 FungibleItemsHaveNoId,124 125 FungibleItemsDontHaveData,126 127 FungibleDisallowsNesting,128 129 SettingPropertiesNotAllowed,130 }131132 #[pallet::config]133 pub trait Config:134 frame_system::Config + pallet_common::Config + pallet_structure::Config + pallet_evm::Config135 {136 type WeightInfo: WeightInfo;137 }138139 #[pallet::pallet]140 #[pallet::generate_store(pub(super) trait Store)]141 pub struct Pallet<T>(_);142143 144 #[pallet::storage]145 pub type TotalSupply<T: Config> =146 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u128, QueryKind = ValueQuery>;147148 149 #[pallet::storage]150 pub type Balance<T: Config> = StorageNMap<151 Key = (152 Key<Twox64Concat, CollectionId>,153 Key<Blake2_128Concat, T::CrossAccountId>,154 ),155 Value = u128,156 QueryKind = ValueQuery,157 >;158159 160 #[pallet::storage]161 pub type Allowance<T: Config> = StorageNMap<162 Key = (163 Key<Twox64Concat, CollectionId>,164 Key<Blake2_128, T::CrossAccountId>,165 Key<Blake2_128Concat, T::CrossAccountId>,166 ),167 Value = u128,168 QueryKind = ValueQuery,169 >;170171 172 #[pallet::storage]173 pub type ForeignCollection<T: Config> =174 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = bool, QueryKind = ValueQuery>;175}176177178179180pub struct FungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);181182183impl<T: Config> FungibleHandle<T> {184 185 pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {186 Self(inner)187 }188189 190 pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {191 self.0192 }193 194 pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {195 &mut self.0196 }197}198impl<T: Config> WithRecorder<T> for FungibleHandle<T> {199 fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {200 self.0.recorder()201 }202 fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {203 self.0.into_recorder()204 }205}206impl<T: Config> Deref for FungibleHandle<T> {207 type Target = pallet_common::CollectionHandle<T>;208209 fn deref(&self) -> &Self::Target {210 &self.0211 }212}213214215impl<T: Config> Pallet<T> {216 217 pub fn init_collection(218 owner: T::CrossAccountId,219 data: CreateCollectionData<T::AccountId>,220 ) -> Result<CollectionId, DispatchError> {221 <PalletCommon<T>>::init_collection(owner, data, false)222 }223224 225 pub fn init_foreign_collection(226 owner: T::CrossAccountId,227 data: CreateCollectionData<T::AccountId>,228 ) -> Result<CollectionId, DispatchError> {229 let id = <PalletCommon<T>>::init_collection(owner, data, false)?;230 <ForeignCollection<T>>::insert(id, true);231 Ok(id)232 }233234 235 pub fn destroy_collection(236 collection: FungibleHandle<T>,237 sender: &T::CrossAccountId,238 ) -> DispatchResult {239 let id = collection.id;240241 if Self::collection_has_tokens(id) {242 return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());243 }244245 246247 PalletCommon::destroy_collection(collection.0, sender)?;248249 <ForeignCollection<T>>::remove(id);250 <TotalSupply<T>>::remove(id);251 let _ = <Balance<T>>::clear_prefix((id,), u32::MAX, None);252 let _ = <Allowance<T>>::clear_prefix((id,), u32::MAX, None);253 Ok(())254 }255256 257 fn collection_has_tokens(collection_id: CollectionId) -> bool {258 <TotalSupply<T>>::get(collection_id) != 0259 }260261 262 263 264 pub fn burn(265 collection: &FungibleHandle<T>,266 owner: &T::CrossAccountId,267 amount: u128,268 ) -> DispatchResult {269 let total_supply = <TotalSupply<T>>::get(collection.id)270 .checked_sub(amount)271 .ok_or(<CommonError<T>>::TokenValueTooLow)?;272273 let balance = <Balance<T>>::get((collection.id, owner))274 .checked_sub(amount)275 .ok_or(<CommonError<T>>::TokenValueTooLow)?;276277 278 ensure!(279 !<ForeignCollection<T>>::get(collection.id),280 <CommonError<T>>::NoPermission281 );282283 if collection.permissions.access() == AccessMode::AllowList {284 collection.check_allowlist(owner)?;285 }286287 288289 if balance == 0 {290 <Balance<T>>::remove((collection.id, owner));291 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, TokenId::default());292 } else {293 <Balance<T>>::insert((collection.id, owner), balance);294 }295 <TotalSupply<T>>::insert(collection.id, total_supply);296297 <PalletEvm<T>>::deposit_log(298 ERC20Events::Transfer {299 from: *owner.as_eth(),300 to: H160::default(),301 value: amount.into(),302 }303 .to_log(collection_id_to_address(collection.id)),304 );305 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(306 collection.id,307 TokenId::default(),308 owner.clone(),309 amount,310 ));311 Ok(())312 }313314 315 pub fn burn_foreign(316 collection: &FungibleHandle<T>,317 owner: &T::CrossAccountId,318 amount: u128,319 ) -> DispatchResult {320 let total_supply = <TotalSupply<T>>::get(collection.id)321 .checked_sub(amount)322 .ok_or(<CommonError<T>>::TokenValueTooLow)?;323324 let balance = <Balance<T>>::get((collection.id, owner))325 .checked_sub(amount)326 .ok_or(<CommonError<T>>::TokenValueTooLow)?;327 328329 if balance == 0 {330 <Balance<T>>::remove((collection.id, owner));331 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, TokenId::default());332 } else {333 <Balance<T>>::insert((collection.id, owner), balance);334 }335 <TotalSupply<T>>::insert(collection.id, total_supply);336337 <PalletEvm<T>>::deposit_log(338 ERC20Events::Transfer {339 from: *owner.as_eth(),340 to: H160::default(),341 value: amount.into(),342 }343 .to_log(collection_id_to_address(collection.id)),344 );345 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(346 collection.id,347 TokenId::default(),348 owner.clone(),349 amount,350 ));351 Ok(())352 }353354 355 356 357 358 359 360 361 pub fn transfer(362 collection: &FungibleHandle<T>,363 from: &T::CrossAccountId,364 to: &T::CrossAccountId,365 amount: u128,366 nesting_budget: &dyn Budget,367 ) -> DispatchResult {368 ensure!(369 collection.limits.transfers_enabled(),370 <CommonError<T>>::TransferNotAllowed,371 );372373 if collection.permissions.access() == AccessMode::AllowList {374 collection.check_allowlist(from)?;375 collection.check_allowlist(to)?;376 }377 <PalletCommon<T>>::ensure_correct_receiver(to)?;378379 let balance_from = <Balance<T>>::get((collection.id, from))380 .checked_sub(amount)381 .ok_or(<CommonError<T>>::TokenValueTooLow)?;382 let balance_to = if from != to {383 Some(384 <Balance<T>>::get((collection.id, to))385 .checked_add(amount)386 .ok_or(ArithmeticError::Overflow)?,387 )388 } else {389 None390 };391392 393394 <PalletStructure<T>>::nest_if_sent_to_token(395 from.clone(),396 to,397 collection.id,398 TokenId::default(),399 nesting_budget,400 )?;401402 if let Some(balance_to) = balance_to {403 404 if balance_from == 0 {405 <Balance<T>>::remove((collection.id, from));406 <PalletStructure<T>>::unnest_if_nested(from, collection.id, TokenId::default());407 } else {408 <Balance<T>>::insert((collection.id, from), balance_from);409 }410 <Balance<T>>::insert((collection.id, to), balance_to);411 }412413 <PalletEvm<T>>::deposit_log(414 ERC20Events::Transfer {415 from: *from.as_eth(),416 to: *to.as_eth(),417 value: amount.into(),418 }419 .to_log(collection_id_to_address(collection.id)),420 );421 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(422 collection.id,423 TokenId::default(),424 from.clone(),425 to.clone(),426 amount,427 ));428 Ok(())429 }430431 432 433 pub fn create_multiple_items(434 collection: &FungibleHandle<T>,435 sender: &T::CrossAccountId,436 data: BTreeMap<T::CrossAccountId, u128>,437 nesting_budget: &dyn Budget,438 ) -> DispatchResult {439 440 ensure!(441 !<ForeignCollection<T>>::get(collection.id),442 <CommonError<T>>::NoPermission443 );444445 if !collection.is_owner_or_admin(sender) {446 ensure!(447 collection.permissions.mint_mode(),448 <CommonError<T>>::PublicMintingNotAllowed449 );450 collection.check_allowlist(sender)?;451452 for (owner, _) in data.iter() {453 collection.check_allowlist(owner)?;454 }455 }456457 let total_supply = data458 .iter()459 .map(|(_, v)| *v)460 .try_fold(<TotalSupply<T>>::get(collection.id), |acc, v| {461 acc.checked_add(v)462 })463 .ok_or(ArithmeticError::Overflow)?;464465 for (to, _) in data.iter() {466 <PalletStructure<T>>::check_nesting(467 sender.clone(),468 to,469 collection.id,470 TokenId::default(),471 nesting_budget,472 )?;473 }474475 let updated_balances = data476 .into_iter()477 .map(|(user, amount)| {478 let updated_balance = <Balance<T>>::get((collection.id, &user))479 .checked_add(amount)480 .ok_or(ArithmeticError::Overflow)?;481 Ok((user, amount, updated_balance))482 })483 .collect::<Result<Vec<_>, DispatchError>>()?;484485 486487 <TotalSupply<T>>::insert(collection.id, total_supply);488 for (user, amount, updated_balance) in updated_balances {489 <Balance<T>>::insert((collection.id, &user), updated_balance);490 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(491 &user,492 collection.id,493 TokenId::default(),494 );495 <PalletEvm<T>>::deposit_log(496 ERC20Events::Transfer {497 from: H160::default(),498 to: *user.as_eth(),499 value: amount.into(),500 }501 .to_log(collection_id_to_address(collection.id)),502 );503 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(504 collection.id,505 TokenId::default(),506 user.clone(),507 amount,508 ));509 }510511 Ok(())512 }513514 515 516 pub fn create_multiple_items_foreign(517 collection: &FungibleHandle<T>,518 sender: &T::CrossAccountId,519 data: BTreeMap<T::CrossAccountId, u128>,520 nesting_budget: &dyn Budget,521 ) -> DispatchResult {522 let total_supply = data523 .iter()524 .map(|(_, v)| *v)525 .try_fold(<TotalSupply<T>>::get(collection.id), |acc, v| {526 acc.checked_add(v)527 })528 .ok_or(ArithmeticError::Overflow)?;529530 let mut balances = data;531 for (k, v) in balances.iter_mut() {532 *v = <Balance<T>>::get((collection.id, &k))533 .checked_add(*v)534 .ok_or(ArithmeticError::Overflow)?;535 }536537 for (to, _) in balances.iter() {538 <PalletStructure<T>>::check_nesting(539 sender.clone(),540 to,541 collection.id,542 TokenId::default(),543 nesting_budget,544 )?;545 }546547 548549 <TotalSupply<T>>::insert(collection.id, total_supply);550 for (user, amount) in balances {551 <Balance<T>>::insert((collection.id, &user), amount);552 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(553 &user,554 collection.id,555 TokenId::default(),556 );557 <PalletEvm<T>>::deposit_log(558 ERC20Events::Transfer {559 from: H160::default(),560 to: *user.as_eth(),561 value: amount.into(),562 }563 .to_log(collection_id_to_address(collection.id)),564 );565 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(566 collection.id,567 TokenId::default(),568 user.clone(),569 amount,570 ));571 }572573 Ok(())574 }575576 fn set_allowance_unchecked(577 collection: &FungibleHandle<T>,578 owner: &T::CrossAccountId,579 spender: &T::CrossAccountId,580 amount: u128,581 ) {582 if amount == 0 {583 <Allowance<T>>::remove((collection.id, owner, spender));584 } else {585 <Allowance<T>>::insert((collection.id, owner, spender), amount);586 }587588 <PalletEvm<T>>::deposit_log(589 ERC20Events::Approval {590 owner: *owner.as_eth(),591 spender: *spender.as_eth(),592 value: amount.into(),593 }594 .to_log(collection_id_to_address(collection.id)),595 );596 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(597 collection.id,598 TokenId(0),599 owner.clone(),600 spender.clone(),601 amount,602 ));603 }604605 606 607 608 609 610 611 pub fn set_allowance(612 collection: &FungibleHandle<T>,613 owner: &T::CrossAccountId,614 spender: &T::CrossAccountId,615 amount: u128,616 ) -> DispatchResult {617 if collection.permissions.access() == AccessMode::AllowList {618 collection.check_allowlist(owner)?;619 collection.check_allowlist(spender)?;620 }621622 if <Balance<T>>::get((collection.id, owner)) < amount {623 ensure!(624 collection.ignores_owned_amount(owner),625 <CommonError<T>>::CantApproveMoreThanOwned626 );627 }628629 630631 Self::set_allowance_unchecked(collection, owner, spender, amount);632 Ok(())633 }634635 636 637 638 639 640 641 642 fn check_allowed(643 collection: &FungibleHandle<T>,644 spender: &T::CrossAccountId,645 from: &T::CrossAccountId,646 amount: u128,647 nesting_budget: &dyn Budget,648 ) -> Result<Option<u128>, DispatchError> {649 if spender.conv_eq(from) {650 return Ok(None);651 }652 if collection.permissions.access() == AccessMode::AllowList {653 654 collection.check_allowlist(spender)?;655 }656 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {657 658 ensure!(659 <PalletStructure<T>>::check_indirectly_owned(660 spender.clone(),661 source.0,662 source.1,663 None,664 nesting_budget665 )?,666 <CommonError<T>>::ApprovedValueTooLow,667 );668 return Ok(None);669 }670 let allowance = <Allowance<T>>::get((collection.id, from, spender)).checked_sub(amount);671 if allowance.is_none() {672 ensure!(673 collection.ignores_allowance(spender),674 <CommonError<T>>::ApprovedValueTooLow675 );676 }677678 Ok(allowance)679 }680681 682 683 684 685686 pub fn transfer_from(687 collection: &FungibleHandle<T>,688 spender: &T::CrossAccountId,689 from: &T::CrossAccountId,690 to: &T::CrossAccountId,691 amount: u128,692 nesting_budget: &dyn Budget,693 ) -> DispatchResult {694 let allowance = Self::check_allowed(collection, spender, from, amount, nesting_budget)?;695696 697698 Self::transfer(collection, from, to, amount, nesting_budget)?;699 if let Some(allowance) = allowance {700 Self::set_allowance_unchecked(collection, from, spender, allowance);701 }702 Ok(())703 }704705 706 707 708 709 710 pub fn burn_from(711 collection: &FungibleHandle<T>,712 spender: &T::CrossAccountId,713 from: &T::CrossAccountId,714 amount: u128,715 nesting_budget: &dyn Budget,716 ) -> DispatchResult {717 let allowance = Self::check_allowed(collection, spender, from, amount, nesting_budget)?;718719 720721 Self::burn(collection, from, amount)?;722 if let Some(allowance) = allowance {723 Self::set_allowance_unchecked(collection, from, spender, allowance);724 }725 Ok(())726 }727728 729 730 731 732 733 734 735 pub fn create_item(736 collection: &FungibleHandle<T>,737 sender: &T::CrossAccountId,738 data: CreateItemData<T>,739 nesting_budget: &dyn Budget,740 ) -> DispatchResult {741 Self::create_multiple_items(742 collection,743 sender,744 [(data.0, data.1)].into_iter().collect(),745 nesting_budget,746 )747 }748749 750 751 752 753 pub fn create_item_foreign(754 collection: &FungibleHandle<T>,755 sender: &T::CrossAccountId,756 data: CreateItemData<T>,757 nesting_budget: &dyn Budget,758 ) -> DispatchResult {759 Self::create_multiple_items_foreign(760 collection,761 sender,762 [(data.0, data.1)].into_iter().collect(),763 nesting_budget,764 )765 }766767 768 769 770 771 772 773 pub fn token_owners(774 collection: CollectionId,775 _token: TokenId,776 ) -> Option<Vec<T::CrossAccountId>> {777 let res: Vec<T::CrossAccountId> = <Balance<T>>::iter_prefix((collection,))778 .map(|(owner, _amount)| owner)779 .take(10)780 .collect();781782 if res.is_empty() {783 None784 } else {785 Some(res)786 }787 }788}