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};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 >;170}171172pub struct FungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);173impl<T: Config> FungibleHandle<T> {174 175 pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {176 Self(inner)177 }178179 180 pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {181 self.0182 }183 184 pub fn common_mut(&mut self) -> &mut pallet_common::CollectionHandle<T> {185 &mut self.0186 }187}188impl<T: Config> WithRecorder<T> for FungibleHandle<T> {189 fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {190 self.0.recorder()191 }192 fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {193 self.0.into_recorder()194 }195}196impl<T: Config> Deref for FungibleHandle<T> {197 type Target = pallet_common::CollectionHandle<T>;198199 fn deref(&self) -> &Self::Target {200 &self.0201 }202}203204205impl<T: Config> Pallet<T> {206 207 pub fn init_collection(208 owner: T::CrossAccountId,209 data: CreateCollectionData<T::AccountId>,210 ) -> Result<CollectionId, DispatchError> {211 <PalletCommon<T>>::init_collection(owner, data, false)212 }213214 215 pub fn destroy_collection(216 collection: FungibleHandle<T>,217 sender: &T::CrossAccountId,218 ) -> DispatchResult {219 let id = collection.id;220221 if Self::collection_has_tokens(id) {222 return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());223 }224225 226227 PalletCommon::destroy_collection(collection.0, sender)?;228229 <TotalSupply<T>>::remove(id);230 <Balance<T>>::remove_prefix((id,), None);231 <Allowance<T>>::remove_prefix((id,), None);232 Ok(())233 }234235 236 fn collection_has_tokens(collection_id: CollectionId) -> bool {237 <TotalSupply<T>>::get(collection_id) != 0238 }239240 241 242 243 pub fn burn(244 collection: &FungibleHandle<T>,245 owner: &T::CrossAccountId,246 amount: u128,247 ) -> DispatchResult {248 let total_supply = <TotalSupply<T>>::get(collection.id)249 .checked_sub(amount)250 .ok_or(<CommonError<T>>::TokenValueTooLow)?;251252 let balance = <Balance<T>>::get((collection.id, owner))253 .checked_sub(amount)254 .ok_or(<CommonError<T>>::TokenValueTooLow)?;255256 if collection.permissions.access() == AccessMode::AllowList {257 collection.check_allowlist(owner)?;258 }259260 261262 if balance == 0 {263 <Balance<T>>::remove((collection.id, owner));264 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, TokenId::default());265 } else {266 <Balance<T>>::insert((collection.id, owner), balance);267 }268 <TotalSupply<T>>::insert(collection.id, total_supply);269270 <PalletEvm<T>>::deposit_log(271 ERC20Events::Transfer {272 from: *owner.as_eth(),273 to: H160::default(),274 value: amount.into(),275 }276 .to_log(collection_id_to_address(collection.id)),277 );278 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(279 collection.id,280 TokenId::default(),281 owner.clone(),282 amount,283 ));284 Ok(())285 }286287 288 289 pub fn transfer(290 collection: &FungibleHandle<T>,291 from: &T::CrossAccountId,292 to: &T::CrossAccountId,293 amount: u128,294 nesting_budget: &dyn Budget,295 ) -> DispatchResult {296 ensure!(297 collection.limits.transfers_enabled(),298 <CommonError<T>>::TransferNotAllowed,299 );300301 if collection.permissions.access() == AccessMode::AllowList {302 collection.check_allowlist(from)?;303 collection.check_allowlist(to)?;304 }305 <PalletCommon<T>>::ensure_correct_receiver(to)?;306307 let balance_from = <Balance<T>>::get((collection.id, from))308 .checked_sub(amount)309 .ok_or(<CommonError<T>>::TokenValueTooLow)?;310 let balance_to = if from != to {311 Some(312 <Balance<T>>::get((collection.id, to))313 .checked_add(amount)314 .ok_or(ArithmeticError::Overflow)?,315 )316 } else {317 None318 };319320 321322 <PalletStructure<T>>::nest_if_sent_to_token(323 from.clone(),324 to,325 collection.id,326 TokenId::default(),327 nesting_budget,328 )?;329330 if let Some(balance_to) = balance_to {331 332 if balance_from == 0 {333 <Balance<T>>::remove((collection.id, from));334 <PalletStructure<T>>::unnest_if_nested(from, collection.id, TokenId::default());335 } else {336 <Balance<T>>::insert((collection.id, from), balance_from);337 }338 <Balance<T>>::insert((collection.id, to), balance_to);339 }340341 <PalletEvm<T>>::deposit_log(342 ERC20Events::Transfer {343 from: *from.as_eth(),344 to: *to.as_eth(),345 value: amount.into(),346 }347 .to_log(collection_id_to_address(collection.id)),348 );349 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(350 collection.id,351 TokenId::default(),352 from.clone(),353 to.clone(),354 amount,355 ));356 Ok(())357 }358359 360 pub fn create_multiple_items(361 collection: &FungibleHandle<T>,362 sender: &T::CrossAccountId,363 data: BTreeMap<T::CrossAccountId, u128>,364 nesting_budget: &dyn Budget,365 ) -> DispatchResult {366 if !collection.is_owner_or_admin(sender) {367 ensure!(368 collection.permissions.mint_mode(),369 <CommonError<T>>::PublicMintingNotAllowed370 );371 collection.check_allowlist(sender)?;372373 for (owner, _) in data.iter() {374 collection.check_allowlist(owner)?;375 }376 }377378 let total_supply = data379 .iter()380 .map(|(_, v)| *v)381 .try_fold(<TotalSupply<T>>::get(collection.id), |acc, v| {382 acc.checked_add(v)383 })384 .ok_or(ArithmeticError::Overflow)?;385386 let mut balances = data;387 for (k, v) in balances.iter_mut() {388 *v = <Balance<T>>::get((collection.id, &k))389 .checked_add(*v)390 .ok_or(ArithmeticError::Overflow)?;391 }392393 for (to, _) in balances.iter() {394 <PalletStructure<T>>::check_nesting(395 sender.clone(),396 to,397 collection.id,398 TokenId::default(),399 nesting_budget,400 )?;401 }402403 404405 <TotalSupply<T>>::insert(collection.id, total_supply);406 for (user, amount) in balances {407 <Balance<T>>::insert((collection.id, &user), amount);408 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(409 &user,410 collection.id,411 TokenId::default(),412 );413 <PalletEvm<T>>::deposit_log(414 ERC20Events::Transfer {415 from: H160::default(),416 to: *user.as_eth(),417 value: amount.into(),418 }419 .to_log(collection_id_to_address(collection.id)),420 );421 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(422 collection.id,423 TokenId::default(),424 user.clone(),425 amount,426 ));427 }428429 Ok(())430 }431432 fn set_allowance_unchecked(433 collection: &FungibleHandle<T>,434 owner: &T::CrossAccountId,435 spender: &T::CrossAccountId,436 amount: u128,437 ) {438 if amount == 0 {439 <Allowance<T>>::remove((collection.id, owner, spender));440 } else {441 <Allowance<T>>::insert((collection.id, owner, spender), amount);442 }443444 <PalletEvm<T>>::deposit_log(445 ERC20Events::Approval {446 owner: *owner.as_eth(),447 spender: *spender.as_eth(),448 value: amount.into(),449 }450 .to_log(collection_id_to_address(collection.id)),451 );452 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(453 collection.id,454 TokenId(0),455 owner.clone(),456 spender.clone(),457 amount,458 ));459 }460461 462 pub fn set_allowance(463 collection: &FungibleHandle<T>,464 owner: &T::CrossAccountId,465 spender: &T::CrossAccountId,466 amount: u128,467 ) -> DispatchResult {468 if collection.permissions.access() == AccessMode::AllowList {469 collection.check_allowlist(owner)?;470 collection.check_allowlist(spender)?;471 }472473 if <Balance<T>>::get((collection.id, owner)) < amount {474 ensure!(475 collection.ignores_owned_amount(owner),476 <CommonError<T>>::CantApproveMoreThanOwned477 );478 }479480 481482 Self::set_allowance_unchecked(collection, owner, spender, amount);483 Ok(())484 }485486 fn check_allowed(487 collection: &FungibleHandle<T>,488 spender: &T::CrossAccountId,489 from: &T::CrossAccountId,490 amount: u128,491 nesting_budget: &dyn Budget,492 ) -> Result<Option<u128>, DispatchError> {493 if spender.conv_eq(from) {494 return Ok(None);495 }496 if collection.permissions.access() == AccessMode::AllowList {497 498 collection.check_allowlist(spender)?;499 }500 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {501 502 ensure!(503 <PalletStructure<T>>::check_indirectly_owned(504 spender.clone(),505 source.0,506 source.1,507 None,508 nesting_budget509 )?,510 <CommonError<T>>::ApprovedValueTooLow,511 );512 return Ok(None);513 }514 let allowance = <Allowance<T>>::get((collection.id, from, spender)).checked_sub(amount);515 if allowance.is_none() {516 ensure!(517 collection.ignores_allowance(spender),518 <CommonError<T>>::ApprovedValueTooLow519 );520 }521522 Ok(allowance)523 }524525 526 pub fn transfer_from(527 collection: &FungibleHandle<T>,528 spender: &T::CrossAccountId,529 from: &T::CrossAccountId,530 to: &T::CrossAccountId,531 amount: u128,532 nesting_budget: &dyn Budget,533 ) -> DispatchResult {534 let allowance = Self::check_allowed(collection, spender, from, amount, nesting_budget)?;535536 537538 Self::transfer(collection, from, to, amount, nesting_budget)?;539 if let Some(allowance) = allowance {540 Self::set_allowance_unchecked(collection, from, spender, allowance);541 }542 Ok(())543 }544545 546 pub fn burn_from(547 collection: &FungibleHandle<T>,548 spender: &T::CrossAccountId,549 from: &T::CrossAccountId,550 amount: u128,551 nesting_budget: &dyn Budget,552 ) -> DispatchResult {553 let allowance = Self::check_allowed(collection, spender, from, amount, nesting_budget)?;554555 556557 Self::burn(collection, from, amount)?;558 if let Some(allowance) = allowance {559 Self::set_allowance_unchecked(collection, from, spender, allowance);560 }561 Ok(())562 }563564 565 pub fn create_item(566 collection: &FungibleHandle<T>,567 sender: &T::CrossAccountId,568 data: CreateItemData<T>,569 nesting_budget: &dyn Budget,570 ) -> DispatchResult {571 Self::create_multiple_items(572 collection,573 sender,574 [(data.0, data.1)].into_iter().collect(),575 nesting_budget,576 )577 }578}