git.delta.rocks / unique-network / refs/commits / 45f4f7d0be2d

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 sp_std::collections::btree_map::BTreeMap;20use frame_support::{dispatch::DispatchResultWithPostInfo, ensure, fail, weights::Weight, traits::Get};21use up_data_structs::{22	CollectionId, TokenId, CreateItemExData, CreateRefungibleExData, budget::Budget, Property,23	PropertyKey, PropertyValue, PropertyKeyPermission, CreateItemData,24};25use pallet_common::{CommonCollectionOperations, CommonWeightInfo, RefungibleExtensions, with_weight, weights::WeightInfo as _};26use pallet_structure::Error as StructureError;27use sp_runtime::{DispatchError};28use sp_std::{vec::Vec, vec};2930use crate::{31	AccountBalance, Allowance, Balance, Config, Error, Owned, Pallet, RefungibleHandle,32	SelfWeightOf, TokenData, weights::WeightInfo, TokensMinted,33};3435macro_rules! max_weight_of {36	($($method:ident ($($args:tt)*)),*) => {37		038		$(39			.max(<SelfWeightOf<T>>::$method($($args)*))40		)*41	};42}4344pub struct CommonWeights<T: Config>(PhantomData<T>);45impl<T: Config> CommonWeightInfo<T::CrossAccountId> for CommonWeights<T> {46	fn create_item() -> Weight {47		<SelfWeightOf<T>>::create_item()48	}4950	fn create_multiple_items(data: &[CreateItemData]) -> Weight {51		<SelfWeightOf<T>>::create_multiple_items(data.len() as u32)52	}5354	fn create_multiple_items_ex(call: &CreateItemExData<T::CrossAccountId>) -> Weight {55		match call {56			CreateItemExData::RefungibleMultipleOwners(i) => {57				<SelfWeightOf<T>>::create_multiple_items_ex_multiple_owners(i.users.len() as u32)58			}59			CreateItemExData::RefungibleMultipleItems(i) => {60				<SelfWeightOf<T>>::create_multiple_items_ex_multiple_items(i.len() as u32)61			}62			_ => 0,63		}64	}6566	fn burn_item() -> Weight {67		max_weight_of!(burn_item_partial(), burn_item_fully())68	}6970	fn set_collection_properties(amount: u32) -> Weight {71		<pallet_common::SelfWeightOf<T>>::set_collection_properties(amount)72	}7374	fn delete_collection_properties(amount: u32) -> Weight {75		<pallet_common::SelfWeightOf<T>>::delete_collection_properties(amount)76	}7778	fn set_token_properties(amount: u32) -> Weight {79		<SelfWeightOf<T>>::set_token_properties(amount)80	}8182	fn delete_token_properties(amount: u32) -> Weight {83		<SelfWeightOf<T>>::delete_token_properties(amount)84	}8586	fn set_token_property_permissions(amount: u32) -> Weight {87		<SelfWeightOf<T>>::set_token_property_permissions(amount)88	}8990	fn transfer() -> Weight {91		max_weight_of!(92			transfer_normal(),93			transfer_creating(),94			transfer_removing(),95			transfer_creating_removing()96		)97	}9899	fn approve() -> Weight {100		<SelfWeightOf<T>>::approve()101	}102103	fn transfer_from() -> Weight {104		max_weight_of!(105			transfer_from_normal(),106			transfer_from_creating(),107			transfer_from_removing(),108			transfer_from_creating_removing()109		)110	}111112	fn burn_from() -> Weight {113		<SelfWeightOf<T>>::burn_from()114	}115116	fn burn_recursively_self_raw() -> Weight {117		// Read to get total balance118		Self::burn_item() + T::DbWeight::get().reads(1)119	}120	fn burn_recursively_breadth_raw(_amount: u32) -> Weight {121		// Refungible token can't have children122		0123	}124}125126fn map_create_data<T: Config>(127	data: up_data_structs::CreateItemData,128	to: &T::CrossAccountId,129) -> Result<CreateRefungibleExData<T::CrossAccountId>, DispatchError> {130	match data {131		up_data_structs::CreateItemData::ReFungible(data) => Ok(CreateRefungibleExData {132			const_data: data.const_data,133			users: {134				let mut out = BTreeMap::new();135				out.insert(to.clone(), data.pieces);136				out.try_into().expect("limit > 0")137			},138			properties: data.properties,139		}),140		_ => fail!(<Error<T>>::NotRefungibleDataUsedToMintFungibleCollectionToken),141	}142}143144/// Implementation of `CommonCollectionOperations` for `RefungibleHandle`. It wraps Refungible Pallete145/// methods and adds weight info.146impl<T: Config> CommonCollectionOperations<T> for RefungibleHandle<T> {147	fn create_item(148		&self,149		sender: T::CrossAccountId,150		to: T::CrossAccountId,151		data: up_data_structs::CreateItemData,152		nesting_budget: &dyn Budget,153	) -> DispatchResultWithPostInfo {154		with_weight(155			<Pallet<T>>::create_item(156				self,157				&sender,158				map_create_data::<T>(data, &to)?,159				nesting_budget,160			),161			<CommonWeights<T>>::create_item(),162		)163	}164165	fn create_multiple_items(166		&self,167		sender: T::CrossAccountId,168		to: T::CrossAccountId,169		data: Vec<up_data_structs::CreateItemData>,170		nesting_budget: &dyn Budget,171	) -> DispatchResultWithPostInfo {172		let weight = <CommonWeights<T>>::create_multiple_items(&data);173		let data = data174			.into_iter()175			.map(|d| map_create_data::<T>(d, &to))176			.collect::<Result<Vec<_>, DispatchError>>()?;177178		with_weight(179			<Pallet<T>>::create_multiple_items(self, &sender, data, nesting_budget),180			weight,181		)182	}183184	fn create_multiple_items_ex(185		&self,186		sender: <T>::CrossAccountId,187		data: CreateItemExData<T::CrossAccountId>,188		nesting_budget: &dyn Budget,189	) -> DispatchResultWithPostInfo {190		let weight = <CommonWeights<T>>::create_multiple_items_ex(&data);191		let data = match data {192			CreateItemExData::RefungibleMultipleOwners(r) => vec![r],193			CreateItemExData::RefungibleMultipleItems(r)194				if r.iter().all(|i| i.users.len() == 1) =>195			{196				r.into_inner()197			}198			_ => fail!(<Error<T>>::NotRefungibleDataUsedToMintFungibleCollectionToken),199		};200201		with_weight(202			<Pallet<T>>::create_multiple_items(self, &sender, data, nesting_budget),203			weight,204		)205	}206207	fn burn_item(208		&self,209		sender: T::CrossAccountId,210		token: TokenId,211		amount: u128,212	) -> DispatchResultWithPostInfo {213		with_weight(214			<Pallet<T>>::burn(self, &sender, token, amount),215			<CommonWeights<T>>::burn_item(),216		)217	}218219	fn burn_item_recursively(220		&self,221		sender: T::CrossAccountId,222		token: TokenId,223		self_budget: &dyn Budget,224		_breadth_budget: &dyn Budget,225	) -> DispatchResultWithPostInfo {226		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);227		with_weight(228			<Pallet<T>>::burn(229				self,230				&sender,231				token,232				<Balance<T>>::get((self.id, token, &sender)),233			),234			<CommonWeights<T>>::burn_recursively_self_raw(),235		)236	}237238	fn transfer(239		&self,240		from: T::CrossAccountId,241		to: T::CrossAccountId,242		token: TokenId,243		amount: u128,244		nesting_budget: &dyn Budget,245	) -> DispatchResultWithPostInfo {246		with_weight(247			<Pallet<T>>::transfer(self, &from, &to, token, amount, nesting_budget),248			<CommonWeights<T>>::transfer(),249		)250	}251252	fn approve(253		&self,254		sender: T::CrossAccountId,255		spender: T::CrossAccountId,256		token: TokenId,257		amount: u128,258	) -> DispatchResultWithPostInfo {259		with_weight(260			<Pallet<T>>::set_allowance(self, &sender, &spender, token, amount),261			<CommonWeights<T>>::approve(),262		)263	}264265	fn transfer_from(266		&self,267		sender: T::CrossAccountId,268		from: T::CrossAccountId,269		to: T::CrossAccountId,270		token: TokenId,271		amount: u128,272		nesting_budget: &dyn Budget,273	) -> DispatchResultWithPostInfo {274		with_weight(275			<Pallet<T>>::transfer_from(self, &sender, &from, &to, token, amount, nesting_budget),276			<CommonWeights<T>>::transfer_from(),277		)278	}279280	fn burn_from(281		&self,282		sender: T::CrossAccountId,283		from: T::CrossAccountId,284		token: TokenId,285		amount: u128,286		nesting_budget: &dyn Budget,287	) -> DispatchResultWithPostInfo {288		with_weight(289			<Pallet<T>>::burn_from(self, &sender, &from, token, amount, nesting_budget),290			<CommonWeights<T>>::burn_from(),291		)292	}293294	fn set_collection_properties(295		&self,296		sender: T::CrossAccountId,297		properties: Vec<Property>,298	) -> DispatchResultWithPostInfo {299		let weight = <CommonWeights<T>>::set_collection_properties(properties.len() as u32);300301		with_weight(302			<Pallet<T>>::set_collection_properties(self, &sender, properties),303			weight,304		)305	}306307	fn delete_collection_properties(308		&self,309		sender: &T::CrossAccountId,310		property_keys: Vec<PropertyKey>,311	) -> DispatchResultWithPostInfo {312		let weight = <CommonWeights<T>>::delete_collection_properties(property_keys.len() as u32);313314		with_weight(315			<Pallet<T>>::delete_collection_properties(self, sender, property_keys),316			weight,317		)318	}319320	fn set_token_properties(321		&self,322		sender: T::CrossAccountId,323		token_id: TokenId,324		properties: Vec<Property>,325		nesting_budget: &dyn Budget,326	) -> DispatchResultWithPostInfo {327		let weight = <CommonWeights<T>>::set_token_properties(properties.len() as u32);328329		with_weight(330			<Pallet<T>>::set_token_properties(331				self,332				&sender,333				token_id,334				properties.into_iter(),335				false,336				nesting_budget,337			),338			weight,339		)340	}341342	fn set_token_property_permissions(343		&self,344		sender: &T::CrossAccountId,345		property_permissions: Vec<PropertyKeyPermission>,346	) -> DispatchResultWithPostInfo {347		let weight =348			<CommonWeights<T>>::set_token_property_permissions(property_permissions.len() as u32);349350		with_weight(351			<Pallet<T>>::set_token_property_permissions(self, sender, property_permissions),352			weight,353		)354	}355356	fn delete_token_properties(357		&self,358		sender: T::CrossAccountId,359		token_id: TokenId,360		property_keys: Vec<PropertyKey>,361		nesting_budget: &dyn Budget,362	) -> DispatchResultWithPostInfo {363		let weight = <CommonWeights<T>>::delete_token_properties(property_keys.len() as u32);364365		with_weight(366			<Pallet<T>>::delete_token_properties(367				self,368				&sender,369				token_id,370				property_keys.into_iter(),371				nesting_budget,372			),373			weight,374		)375	}376377	fn check_nesting(378		&self,379		_sender: <T>::CrossAccountId,380		_from: (CollectionId, TokenId),381		_under: TokenId,382		_nesting_budget: &dyn Budget,383	) -> sp_runtime::DispatchResult {384		fail!(<Error<T>>::RefungibleDisallowsNesting)385	}386387	fn nest(&self, _under: TokenId, _to_nest: (CollectionId, TokenId)) {}388389	fn unnest(&self, _under: TokenId, _to_nest: (CollectionId, TokenId)) {}390391	fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId> {392		<Owned<T>>::iter_prefix((self.id, account))393			.map(|(id, _)| id)394			.collect()395	}396397	fn collection_tokens(&self) -> Vec<TokenId> {398		<TokenData<T>>::iter_prefix((self.id,))399			.map(|(id, _)| id)400			.collect()401	}402403	fn token_exists(&self, token: TokenId) -> bool {404		<Pallet<T>>::token_exists(self, token)405	}406407	fn last_token_id(&self) -> TokenId {408		TokenId(<TokensMinted<T>>::get(self.id))409	}410411	fn token_owner(&self, token: TokenId) -> Option<T::CrossAccountId> {412		<Pallet<T>>::token_owner(self.id, token)413	}414415	fn token_property(&self, _token_id: TokenId, _key: &PropertyKey) -> Option<PropertyValue> {416		None417	}418419	fn token_properties(420		&self,421		_token_id: TokenId,422		_keys: Option<Vec<PropertyKey>>,423	) -> Vec<Property> {424		Vec::new()425	}426427	fn total_supply(&self) -> u32 {428		<Pallet<T>>::total_supply(self)429	}430431	fn account_balance(&self, account: T::CrossAccountId) -> u32 {432		<AccountBalance<T>>::get((self.id, account))433	}434435	fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128 {436		<Balance<T>>::get((self.id, token, account))437	}438439	fn allowance(440		&self,441		sender: T::CrossAccountId,442		spender: T::CrossAccountId,443		token: TokenId,444	) -> u128 {445		<Allowance<T>>::get((self.id, token, sender, spender))446	}447448	fn refungible_extensions(&self) -> Option<&dyn RefungibleExtensions<T>> {449		Some(self)450	}451452	fn total_pieces(&self, token: TokenId) -> Option<u128> {453		<Pallet<T>>::total_pieces(self.id, token)454	}455}456457impl<T: Config> RefungibleExtensions<T> for RefungibleHandle<T> {458	fn repartition(459		&self,460		owner: &T::CrossAccountId,461		token: TokenId,462		amount: u128,463	) -> DispatchResultWithPostInfo {464		with_weight(465			<Pallet<T>>::repartition(self, owner, token, amount),466			<SelfWeightOf<T>>::repartition_item(),467		)468	}469}