git.delta.rocks / unique-network / refs/commits / 2f9c8d434e2f

1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617#![cfg_attr(not(feature = "std"), no_std)]1819use core::ops::Deref;20use frame_support::{ensure};21use up_data_structs::{AccessMode, CollectionId, TokenId, CreateCollectionData};22use pallet_common::{Error as CommonError, Event as CommonEvent, Pallet as PalletCommon};23use pallet_evm::account::CrossAccountId;24use pallet_evm_coder_substrate::WithRecorder;25use sp_core::H160;26use sp_runtime::{ArithmeticError, DispatchError, DispatchResult};27use sp_std::collections::btree_map::BTreeMap;2829pub use pallet::*;3031use crate::erc::ERC20Events;32#[cfg(feature = "runtime-benchmarks")]33pub mod benchmarking;34pub mod common;35pub mod erc;36pub mod weights;3738pub type CreateItemData<T> = (<T as pallet_evm::account::Config>::CrossAccountId, u128);39pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;4041#[frame_support::pallet]42pub mod pallet {43	use frame_support::{Blake2_128, Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key};44	use up_data_structs::CollectionId;45	use super::weights::WeightInfo;4647	#[pallet::error]48	pub enum Error<T> {49		/// Not Fungible item data used to mint in Fungible collection.50		NotFungibleDataUsedToMintFungibleCollectionToken,51		/// Not default id passed as TokenId argument52		FungibleItemsHaveNoId,53		/// Tried to set data for fungible item54		FungibleItemsDontHaveData,55	}5657	#[pallet::config]58	pub trait Config: frame_system::Config + pallet_common::Config {59		type WeightInfo: WeightInfo;60	}6162	#[pallet::pallet]63	#[pallet::generate_store(pub(super) trait Store)]64	pub struct Pallet<T>(_);6566	#[pallet::storage]67	pub type TotalSupply<T: Config> =68		StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u128, QueryKind = ValueQuery>;6970	#[pallet::storage]71	pub type Balance<T: Config> = StorageNMap<72		Key = (73			Key<Twox64Concat, CollectionId>,74			Key<Blake2_128Concat, T::CrossAccountId>,75		),76		Value = u128,77		QueryKind = ValueQuery,78	>;7980	#[pallet::storage]81	pub type Allowance<T: Config> = StorageNMap<82		Key = (83			Key<Twox64Concat, CollectionId>,84			Key<Blake2_128, T::CrossAccountId>,85			Key<Blake2_128Concat, T::CrossAccountId>,86		),87		Value = u128,88		QueryKind = ValueQuery,89	>;90}9192pub struct FungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);93impl<T: Config> FungibleHandle<T> {94	pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {95		Self(inner)96	}97	pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {98		self.099	}100}101impl<T: Config> WithRecorder<T> for FungibleHandle<T> {102	fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {103		self.0.recorder()104	}105	fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {106		self.0.into_recorder()107	}108}109impl<T: Config> Deref for FungibleHandle<T> {110	type Target = pallet_common::CollectionHandle<T>;111112	fn deref(&self) -> &Self::Target {113		&self.0114	}115}116117impl<T: Config> Pallet<T> {118	pub fn init_collection(119		owner: T::AccountId,120		data: CreateCollectionData<T::AccountId>,121	) -> Result<CollectionId, DispatchError> {122		<PalletCommon<T>>::init_collection(owner, data)123	}124	pub fn destroy_collection(125		collection: FungibleHandle<T>,126		sender: &T::CrossAccountId,127	) -> DispatchResult {128		let id = collection.id;129130		// =========131132		PalletCommon::destroy_collection(collection.0, sender)?;133134		<TotalSupply<T>>::remove(id);135		<Balance<T>>::remove_prefix((id,), None);136		<Allowance<T>>::remove_prefix((id,), None);137		Ok(())138	}139140	pub fn burn(141		collection: &FungibleHandle<T>,142		owner: &T::CrossAccountId,143		amount: u128,144	) -> DispatchResult {145		let total_supply = <TotalSupply<T>>::get(collection.id)146			.checked_sub(amount)147			.ok_or(<CommonError<T>>::TokenValueTooLow)?;148149		let balance = <Balance<T>>::get((collection.id, owner))150			.checked_sub(amount)151			.ok_or(<CommonError<T>>::TokenValueTooLow)?;152153		if collection.access == AccessMode::AllowList {154			collection.check_allowlist(owner)?;155		}156157		// =========158159		if balance == 0 {160			<Balance<T>>::remove((collection.id, owner));161		} else {162			<Balance<T>>::insert((collection.id, owner), balance);163		}164		<TotalSupply<T>>::insert(collection.id, total_supply);165166		collection.log_mirrored(ERC20Events::Transfer {167			from: *owner.as_eth(),168			to: H160::default(),169			value: amount.into(),170		});171		<PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(172			collection.id,173			TokenId::default(),174			owner.clone(),175			amount,176		));177		Ok(())178	}179180	pub fn transfer(181		collection: &FungibleHandle<T>,182		from: &T::CrossAccountId,183		to: &T::CrossAccountId,184		amount: u128,185	) -> DispatchResult {186		ensure!(187			collection.limits.transfers_enabled(),188			<CommonError<T>>::TransferNotAllowed,189		);190191		if collection.access == AccessMode::AllowList {192			collection.check_allowlist(from)?;193			collection.check_allowlist(to)?;194		}195		<PalletCommon<T>>::ensure_correct_receiver(to)?;196197		let balance_from = <Balance<T>>::get((collection.id, from))198			.checked_sub(amount)199			.ok_or(<CommonError<T>>::TokenValueTooLow)?;200		let balance_to = if from != to {201			Some(202				<Balance<T>>::get((collection.id, to))203					.checked_add(amount)204					.ok_or(ArithmeticError::Overflow)?,205			)206		} else {207			None208		};209210		// =========211212		if let Some(balance_to) = balance_to {213			// from != to214			if balance_from == 0 {215				<Balance<T>>::remove((collection.id, from));216			} else {217				<Balance<T>>::insert((collection.id, from), balance_from);218			}219			<Balance<T>>::insert((collection.id, to), balance_to);220		}221222		collection.log_mirrored(ERC20Events::Transfer {223			from: *from.as_eth(),224			to: *to.as_eth(),225			value: amount.into(),226		});227		<PalletCommon<T>>::deposit_event(CommonEvent::Transfer(228			collection.id,229			TokenId::default(),230			from.clone(),231			to.clone(),232			amount,233		));234		Ok(())235	}236237	pub fn create_multiple_items(238		collection: &FungibleHandle<T>,239		sender: &T::CrossAccountId,240		data: BTreeMap<T::CrossAccountId, u128>,241	) -> DispatchResult {242		if !collection.is_owner_or_admin(sender) {243			ensure!(244				collection.mint_mode,245				<CommonError<T>>::PublicMintingNotAllowed246			);247			collection.check_allowlist(sender)?;248249			for (owner, _) in data.iter() {250				collection.check_allowlist(owner)?;251			}252		}253254		let total_supply = data255			.iter()256			.map(|(_, v)| *v)257			.try_fold(<TotalSupply<T>>::get(collection.id), |acc, v| {258				acc.checked_add(v)259			})260			.ok_or(ArithmeticError::Overflow)?;261262		let mut balances = data;263		for (k, v) in balances.iter_mut() {264			*v = <Balance<T>>::get((collection.id, &k))265				.checked_add(*v)266				.ok_or(ArithmeticError::Overflow)?;267		}268269		// =========270271		<TotalSupply<T>>::insert(collection.id, total_supply);272		for (user, amount) in balances {273			<Balance<T>>::insert((collection.id, &user), amount);274275			collection.log_mirrored(ERC20Events::Transfer {276				from: H160::default(),277				to: *user.as_eth(),278				value: amount.into(),279			});280			<PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(281				collection.id,282				TokenId::default(),283				user.clone(),284				amount,285			));286		}287288		Ok(())289	}290291	fn set_allowance_unchecked(292		collection: &FungibleHandle<T>,293		owner: &T::CrossAccountId,294		spender: &T::CrossAccountId,295		amount: u128,296	) {297		if amount == 0 {298			<Allowance<T>>::remove((collection.id, owner, spender));299		} else {300			<Allowance<T>>::insert((collection.id, owner, spender), amount);301		}302303		collection.log_mirrored(ERC20Events::Approval {304			owner: *owner.as_eth(),305			spender: *spender.as_eth(),306			value: amount.into(),307		});308		<PalletCommon<T>>::deposit_event(CommonEvent::Approved(309			collection.id,310			TokenId(0),311			owner.clone(),312			spender.clone(),313			amount,314		));315	}316317	pub fn set_allowance(318		collection: &FungibleHandle<T>,319		owner: &T::CrossAccountId,320		spender: &T::CrossAccountId,321		amount: u128,322	) -> DispatchResult {323		if collection.access == AccessMode::AllowList {324			collection.check_allowlist(owner)?;325			collection.check_allowlist(spender)?;326		}327328		if <Balance<T>>::get((collection.id, owner)) < amount {329			ensure!(330				collection.ignores_owned_amount(owner),331				<CommonError<T>>::CantApproveMoreThanOwned332			);333		}334335		// =========336337		Self::set_allowance_unchecked(collection, owner, spender, amount);338		Ok(())339	}340341	pub fn transfer_from(342		collection: &FungibleHandle<T>,343		spender: &T::CrossAccountId,344		from: &T::CrossAccountId,345		to: &T::CrossAccountId,346		amount: u128,347	) -> DispatchResult {348		if spender.conv_eq(from) {349			return Self::transfer(collection, from, to, amount);350		}351		if collection.access == AccessMode::AllowList {352			// `from`, `to` checked in [`transfer`]353			collection.check_allowlist(spender)?;354		}355356		let allowance = <Allowance<T>>::get((collection.id, from, spender)).checked_sub(amount);357		if allowance.is_none() {358			ensure!(359				collection.ignores_allowance(spender),360				<CommonError<T>>::ApprovedValueTooLow361			);362		}363364		// =========365366		Self::transfer(collection, from, to, amount)?;367		if let Some(allowance) = allowance {368			Self::set_allowance_unchecked(collection, from, spender, allowance);369		}370		Ok(())371	}372373	pub fn burn_from(374		collection: &FungibleHandle<T>,375		spender: &T::CrossAccountId,376		from: &T::CrossAccountId,377		amount: u128,378	) -> DispatchResult {379		if spender.conv_eq(from) {380			return Self::burn(collection, from, amount);381		}382		if collection.access == AccessMode::AllowList {383			// `from` checked in [`burn`]384			collection.check_allowlist(spender)?;385		}386387		let allowance = <Allowance<T>>::get((collection.id, from, spender)).checked_sub(amount);388		if allowance.is_none() {389			ensure!(390				collection.ignores_allowance(spender),391				<CommonError<T>>::ApprovedValueTooLow392			);393		}394395		// =========396397		Self::burn(collection, from, amount)?;398		if let Some(allowance) = allowance {399			Self::set_allowance_unchecked(collection, from, spender, allowance);400		}401		Ok(())402	}403404	/// Delegated to `create_multiple_items`405	pub fn create_item(406		collection: &FungibleHandle<T>,407		sender: &T::CrossAccountId,408		data: CreateItemData<T>,409	) -> DispatchResult {410		Self::create_multiple_items(collection, sender, [(data.0, data.1)].into_iter().collect())411	}412}