git.delta.rocks / unique-network / refs/commits / 008833a7ebae

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/>.1617use core::marker::PhantomData;1819use frame_support::{dispatch::DispatchResultWithPostInfo, ensure, fail, weights::Weight, traits::Get};20use up_data_structs::{21	TokenId, CollectionId, CreateItemExData, budget::Budget, CreateItemData, TokenOwnerError,22};23use pallet_common::{24	CommonCollectionOperations, CommonWeightInfo, RefungibleExtensions, with_weight,25	weights::WeightInfo as _, SelfWeightOf as PalletCommonWeightOf,26};27use pallet_structure::Error as StructureError;28use sp_runtime::ArithmeticError;29use sp_std::{vec::Vec, vec};30use up_data_structs::{Property, PropertyKey, PropertyValue, PropertyKeyPermission};3132use crate::{33	Allowance, TotalSupply, Balance, Config, Error, FungibleHandle, Pallet, SelfWeightOf,34	weights::WeightInfo,35};3637pub struct CommonWeights<T: Config>(PhantomData<T>);38impl<T: Config> CommonWeightInfo<T::CrossAccountId> for CommonWeights<T> {39	fn create_multiple_items(_data: &[CreateItemData]) -> Weight {40		// All items minted for the same user, so it works same as create_item41		<SelfWeightOf<T>>::create_item()42	}4344	fn create_multiple_items_ex(data: &CreateItemExData<T::CrossAccountId>) -> Weight {45		match data {46			CreateItemExData::Fungible(f) => {47				<SelfWeightOf<T>>::create_multiple_items_ex(f.len() as u32)48			}49			_ => Weight::zero(),50		}51	}5253	fn burn_item() -> Weight {54		<SelfWeightOf<T>>::burn_item()55	}5657	fn set_collection_properties(amount: u32) -> Weight {58		<pallet_common::SelfWeightOf<T>>::set_collection_properties(amount)59	}6061	fn delete_collection_properties(amount: u32) -> Weight {62		<pallet_common::SelfWeightOf<T>>::delete_collection_properties(amount)63	}6465	fn set_token_properties(_amount: u32) -> Weight {66		// Error67		Weight::zero()68	}6970	fn delete_token_properties(_amount: u32) -> Weight {71		// Error72		Weight::zero()73	}7475	fn set_token_property_permissions(_amount: u32) -> Weight {76		// Error77		Weight::zero()78	}7980	fn transfer() -> Weight {81		<SelfWeightOf<T>>::transfer() + <PalletCommonWeightOf<T>>::check_accesslist() * 282	}8384	fn approve() -> Weight {85		<SelfWeightOf<T>>::approve()86	}8788	fn approve_from() -> Weight {89		<SelfWeightOf<T>>::approve_from()90	}9192	fn transfer_from() -> Weight {93		Self::transfer()94			+ <SelfWeightOf<T>>::check_allowed_raw()95			+ <SelfWeightOf<T>>::set_allowance_unchecked_raw()96	}9798	fn burn_from() -> Weight {99		<SelfWeightOf<T>>::burn_from()100	}101102	fn burn_recursively_self_raw() -> Weight {103		// Read to get total balance104		Self::burn_item() + T::DbWeight::get().reads(1)105	}106107	fn burn_recursively_breadth_raw(_amount: u32) -> Weight {108		// Fungible tokens can't have children109		Weight::zero()110	}111112	fn token_owner() -> Weight {113		Weight::zero()114	}115116	fn set_allowance_for_all() -> Weight {117		Weight::zero()118	}119120	fn force_repair_item() -> Weight {121		Weight::zero()122	}123}124125/// Implementation of `CommonCollectionOperations` for `FungibleHandle`. It wraps FungibleHandle Pallete126/// methods and adds weight info.127impl<T: Config> CommonCollectionOperations<T> for FungibleHandle<T> {128	fn create_item(129		&self,130		sender: T::CrossAccountId,131		to: T::CrossAccountId,132		data: up_data_structs::CreateItemData,133		nesting_budget: &dyn Budget,134	) -> DispatchResultWithPostInfo {135		match &data {136			up_data_structs::CreateItemData::Fungible(fungible_data) => with_weight(137				<Pallet<T>>::create_item(self, &sender, (to, fungible_data.value), nesting_budget),138				<CommonWeights<T>>::create_item(&data),139			),140			_ => fail!(<Error<T>>::NotFungibleDataUsedToMintFungibleCollectionToken),141		}142	}143144	fn create_multiple_items(145		&self,146		sender: T::CrossAccountId,147		to: T::CrossAccountId,148		data: Vec<up_data_structs::CreateItemData>,149		nesting_budget: &dyn Budget,150	) -> DispatchResultWithPostInfo {151		let mut sum: u128 = 0;152		for data in &data {153			match &data {154				up_data_structs::CreateItemData::Fungible(data) => {155					sum = sum156						.checked_add(data.value)157						.ok_or(ArithmeticError::Overflow)?;158				}159				_ => fail!(<Error<T>>::NotFungibleDataUsedToMintFungibleCollectionToken),160			}161		}162163		with_weight(164			<Pallet<T>>::create_item(self, &sender, (to, sum), nesting_budget),165			<CommonWeights<T>>::create_multiple_items(&data),166		)167	}168169	fn create_multiple_items_ex(170		&self,171		sender: <T>::CrossAccountId,172		data: up_data_structs::CreateItemExData<<T>::CrossAccountId>,173		nesting_budget: &dyn Budget,174	) -> DispatchResultWithPostInfo {175		let weight = <CommonWeights<T>>::create_multiple_items_ex(&data);176		let data = match data {177			up_data_structs::CreateItemExData::Fungible(f) => f,178			_ => fail!(<Error<T>>::NotFungibleDataUsedToMintFungibleCollectionToken),179		};180181		with_weight(182			<Pallet<T>>::create_multiple_items(self, &sender, data.into_inner(), nesting_budget),183			weight,184		)185	}186187	fn burn_item(188		&self,189		sender: T::CrossAccountId,190		token: TokenId,191		amount: u128,192	) -> DispatchResultWithPostInfo {193		ensure!(194			token == TokenId::default(),195			<Error<T>>::FungibleItemsHaveNoId196		);197198		with_weight(199			<Pallet<T>>::burn(self, &sender, amount),200			<CommonWeights<T>>::burn_item(),201		)202	}203204	fn burn_item_recursively(205		&self,206		sender: T::CrossAccountId,207		token: TokenId,208		self_budget: &dyn Budget,209		_breadth_budget: &dyn Budget,210	) -> DispatchResultWithPostInfo {211		// Should not happen?212		ensure!(213			token == TokenId::default(),214			<Error<T>>::FungibleItemsHaveNoId215		);216		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);217218		with_weight(219			<Pallet<T>>::burn(self, &sender, <Balance<T>>::get((self.id, &sender))),220			<CommonWeights<T>>::burn_recursively_self_raw(),221		)222	}223224	fn transfer(225		&self,226		from: T::CrossAccountId,227		to: T::CrossAccountId,228		token: TokenId,229		amount: u128,230		nesting_budget: &dyn Budget,231	) -> DispatchResultWithPostInfo {232		ensure!(233			token == TokenId::default(),234			<Error<T>>::FungibleItemsHaveNoId235		);236237		<Pallet<T>>::transfer(self, &from, &to, amount, nesting_budget)238	}239240	fn approve(241		&self,242		sender: T::CrossAccountId,243		spender: T::CrossAccountId,244		token: TokenId,245		amount: u128,246	) -> DispatchResultWithPostInfo {247		ensure!(248			token == TokenId::default(),249			<Error<T>>::FungibleItemsHaveNoId250		);251252		with_weight(253			<Pallet<T>>::set_allowance(self, &sender, &spender, amount),254			<CommonWeights<T>>::approve(),255		)256	}257258	fn approve_from(259		&self,260		sender: T::CrossAccountId,261		from: T::CrossAccountId,262		to: T::CrossAccountId,263		token: TokenId,264		amount: u128,265	) -> DispatchResultWithPostInfo {266		ensure!(267			token == TokenId::default(),268			<Error<T>>::FungibleItemsHaveNoId269		);270271		with_weight(272			<Pallet<T>>::set_allowance_from(self, &sender, &from, &to, amount),273			<CommonWeights<T>>::approve_from(),274		)275	}276277	fn transfer_from(278		&self,279		sender: T::CrossAccountId,280		from: T::CrossAccountId,281		to: T::CrossAccountId,282		token: TokenId,283		amount: u128,284		nesting_budget: &dyn Budget,285	) -> DispatchResultWithPostInfo {286		ensure!(287			token == TokenId::default(),288			<Error<T>>::FungibleItemsHaveNoId289		);290291		<Pallet<T>>::transfer_from(self, &sender, &from, &to, amount, nesting_budget)292	}293294	fn burn_from(295		&self,296		sender: T::CrossAccountId,297		from: T::CrossAccountId,298		token: TokenId,299		amount: u128,300		nesting_budget: &dyn Budget,301	) -> DispatchResultWithPostInfo {302		ensure!(303			token == TokenId::default(),304			<Error<T>>::FungibleItemsHaveNoId305		);306307		with_weight(308			<Pallet<T>>::burn_from(self, &sender, &from, amount, nesting_budget),309			<CommonWeights<T>>::burn_from(),310		)311	}312313	fn set_collection_properties(314		&self,315		sender: T::CrossAccountId,316		properties: Vec<Property>,317	) -> DispatchResultWithPostInfo {318		let weight = <CommonWeights<T>>::set_collection_properties(properties.len() as u32);319320		with_weight(321			<Pallet<T>>::set_collection_properties(self, &sender, properties),322			weight,323		)324	}325326	fn delete_collection_properties(327		&self,328		sender: &T::CrossAccountId,329		property_keys: Vec<PropertyKey>,330	) -> DispatchResultWithPostInfo {331		let weight = <CommonWeights<T>>::delete_collection_properties(property_keys.len() as u32);332333		with_weight(334			<Pallet<T>>::delete_collection_properties(self, sender, property_keys),335			weight,336		)337	}338339	fn set_token_properties(340		&self,341		_sender: T::CrossAccountId,342		_token_id: TokenId,343		_property: Vec<Property>,344		_nesting_budget: &dyn Budget,345	) -> DispatchResultWithPostInfo {346		fail!(<Error<T>>::SettingPropertiesNotAllowed)347	}348349	fn set_token_property_permissions(350		&self,351		_sender: &T::CrossAccountId,352		_property_permissions: Vec<PropertyKeyPermission>,353	) -> DispatchResultWithPostInfo {354		fail!(<Error<T>>::SettingPropertiesNotAllowed)355	}356357	fn delete_token_properties(358		&self,359		_sender: T::CrossAccountId,360		_token_id: TokenId,361		_property_keys: Vec<PropertyKey>,362		_nesting_budget: &dyn Budget,363	) -> DispatchResultWithPostInfo {364		fail!(<Error<T>>::SettingPropertiesNotAllowed)365	}366367	fn check_nesting(368		&self,369		_sender: <T>::CrossAccountId,370		_from: (CollectionId, TokenId),371		_under: TokenId,372		_nesting_budget: &dyn Budget,373	) -> sp_runtime::DispatchResult {374		fail!(<Error<T>>::FungibleDisallowsNesting)375	}376377	fn nest(&self, _under: TokenId, _to_nest: (CollectionId, TokenId)) {}378379	fn unnest(&self, _under: TokenId, _to_nest: (CollectionId, TokenId)) {}380381	fn collection_tokens(&self) -> Vec<TokenId> {382		vec![TokenId::default()]383	}384385	fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId> {386		if <Balance<T>>::get((self.id, account)) != 0 {387			vec![TokenId::default()]388		} else {389			vec![]390		}391	}392393	fn token_exists(&self, token: TokenId) -> bool {394		token == TokenId::default()395	}396397	fn last_token_id(&self) -> TokenId {398		TokenId::default()399	}400401	fn token_owner(&self, _token: TokenId) -> Result<T::CrossAccountId, TokenOwnerError> {402		Err(TokenOwnerError::MultipleOwners)403	}404405	/// Returns 10 tokens owners in no particular order.406	fn token_owners(&self, token: TokenId) -> Vec<T::CrossAccountId> {407		<Pallet<T>>::token_owners(self.id, token).unwrap_or_default()408	}409410	fn token_property(&self, _token_id: TokenId, _key: &PropertyKey) -> Option<PropertyValue> {411		None412	}413414	fn token_properties(415		&self,416		_token_id: TokenId,417		_keys: Option<Vec<PropertyKey>>,418	) -> Vec<Property> {419		Vec::new()420	}421422	fn total_supply(&self) -> u32 {423		1424	}425426	fn account_balance(&self, account: T::CrossAccountId) -> u32 {427		if <Balance<T>>::get((self.id, account)) != 0 {428			1429		} else {430			0431		}432	}433434	fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128 {435		if token != TokenId::default() {436			return 0;437		}438		<Balance<T>>::get((self.id, account))439	}440441	fn allowance(442		&self,443		sender: T::CrossAccountId,444		spender: T::CrossAccountId,445		token: TokenId,446	) -> u128 {447		if token != TokenId::default() {448			return 0;449		}450		<Allowance<T>>::get((self.id, sender, spender))451	}452453	fn refungible_extensions(&self) -> Option<&dyn RefungibleExtensions<T>> {454		None455	}456457	fn total_pieces(&self, token: TokenId) -> Option<u128> {458		if token != TokenId::default() {459			return None;460		}461		<TotalSupply<T>>::try_get(self.id).ok()462	}463464	fn set_allowance_for_all(465		&self,466		_owner: T::CrossAccountId,467		_operator: T::CrossAccountId,468		_approve: bool,469	) -> DispatchResultWithPostInfo {470		fail!(<Error<T>>::SettingAllowanceForAllNotAllowed)471	}472473	fn allowance_for_all(&self, _owner: T::CrossAccountId, _operator: T::CrossAccountId) -> bool {474		false475	}476477	/// Repairs a possibly broken item.478	fn repair_item(&self, _token: TokenId) -> DispatchResultWithPostInfo {479		fail!(<Error<T>>::FungibleTokensAreAlwaysValid)480	}481}