git.delta.rocks / unique-network / refs/commits / 73d9a3eafd0e

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};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		// Error72		073	}7475	fn delete_collection_properties(_amount: u32) -> Weight {76		// Error77		078	}7980	fn set_token_properties(_amount: u32) -> Weight {81		// Error82		083	}8485	fn delete_token_properties(_amount: u32) -> Weight {86		// Error87		088	}8990	fn set_token_property_permissions(_amount: u32) -> Weight {91		// Error92		093	}9495	fn transfer() -> Weight {96		max_weight_of!(97			transfer_normal(),98			transfer_creating(),99			transfer_removing(),100			transfer_creating_removing()101		)102	}103104	fn approve() -> Weight {105		<SelfWeightOf<T>>::approve()106	}107108	fn transfer_from() -> Weight {109		max_weight_of!(110			transfer_from_normal(),111			transfer_from_creating(),112			transfer_from_removing(),113			transfer_from_creating_removing()114		)115	}116117	fn burn_from() -> Weight {118		<SelfWeightOf<T>>::burn_from()119	}120121	fn burn_recursively_self_raw() -> Weight {122		// Read to get total balance123		Self::burn_item() + T::DbWeight::get().reads(1)124	}125	fn burn_recursively_breadth_raw(_amount: u32) -> Weight {126		// Refungible token can't have children127		0128	}129}130131fn map_create_data<T: Config>(132	data: up_data_structs::CreateItemData,133	to: &T::CrossAccountId,134) -> Result<CreateRefungibleExData<T::CrossAccountId>, DispatchError> {135	match data {136		up_data_structs::CreateItemData::ReFungible(data) => Ok(CreateRefungibleExData {137			const_data: data.const_data,138			users: {139				let mut out = BTreeMap::new();140				out.insert(to.clone(), data.pieces);141				out.try_into().expect("limit > 0")142			},143			properties: data.properties,144		}),145		_ => fail!(<Error<T>>::NotRefungibleDataUsedToMintFungibleCollectionToken),146	}147}148149/// Implementation of `CommonCollectionOperations` for `RefungibleHandle`. It wraps Refungible Pallete150/// methods and adds weight info.151impl<T: Config> CommonCollectionOperations<T> for RefungibleHandle<T> {152	fn create_item(153		&self,154		sender: T::CrossAccountId,155		to: T::CrossAccountId,156		data: up_data_structs::CreateItemData,157		nesting_budget: &dyn Budget,158	) -> DispatchResultWithPostInfo {159		with_weight(160			<Pallet<T>>::create_item(161				self,162				&sender,163				map_create_data::<T>(data, &to)?,164				nesting_budget,165			),166			<CommonWeights<T>>::create_item(),167		)168	}169170	fn create_multiple_items(171		&self,172		sender: T::CrossAccountId,173		to: T::CrossAccountId,174		data: Vec<up_data_structs::CreateItemData>,175		nesting_budget: &dyn Budget,176	) -> DispatchResultWithPostInfo {177		let weight = <CommonWeights<T>>::create_multiple_items(&data);178		let data = data179			.into_iter()180			.map(|d| map_create_data::<T>(d, &to))181			.collect::<Result<Vec<_>, DispatchError>>()?;182183		with_weight(184			<Pallet<T>>::create_multiple_items(self, &sender, data, nesting_budget),185			weight,186		)187	}188189	fn create_multiple_items_ex(190		&self,191		sender: <T>::CrossAccountId,192		data: CreateItemExData<T::CrossAccountId>,193		nesting_budget: &dyn Budget,194	) -> DispatchResultWithPostInfo {195		let weight = <CommonWeights<T>>::create_multiple_items_ex(&data);196		let data = match data {197			CreateItemExData::RefungibleMultipleOwners(r) => vec![r],198			CreateItemExData::RefungibleMultipleItems(r)199				if r.iter().all(|i| i.users.len() == 1) =>200			{201				r.into_inner()202			}203			_ => fail!(<Error<T>>::NotRefungibleDataUsedToMintFungibleCollectionToken),204		};205206		with_weight(207			<Pallet<T>>::create_multiple_items(self, &sender, data, nesting_budget),208			weight,209		)210	}211212	fn burn_item(213		&self,214		sender: T::CrossAccountId,215		token: TokenId,216		amount: u128,217	) -> DispatchResultWithPostInfo {218		with_weight(219			<Pallet<T>>::burn(self, &sender, token, amount),220			<CommonWeights<T>>::burn_item(),221		)222	}223224	fn burn_item_recursively(225		&self,226		sender: T::CrossAccountId,227		token: TokenId,228		self_budget: &dyn Budget,229		_breadth_budget: &dyn Budget,230	) -> DispatchResultWithPostInfo {231		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);232		with_weight(233			<Pallet<T>>::burn(234				self,235				&sender,236				token,237				<Balance<T>>::get((self.id, token, &sender)),238			),239			<CommonWeights<T>>::burn_recursively_self_raw(),240		)241	}242243	fn transfer(244		&self,245		from: T::CrossAccountId,246		to: T::CrossAccountId,247		token: TokenId,248		amount: u128,249		nesting_budget: &dyn Budget,250	) -> DispatchResultWithPostInfo {251		with_weight(252			<Pallet<T>>::transfer(self, &from, &to, token, amount, nesting_budget),253			<CommonWeights<T>>::transfer(),254		)255	}256257	fn approve(258		&self,259		sender: T::CrossAccountId,260		spender: T::CrossAccountId,261		token: TokenId,262		amount: u128,263	) -> DispatchResultWithPostInfo {264		with_weight(265			<Pallet<T>>::set_allowance(self, &sender, &spender, token, amount),266			<CommonWeights<T>>::approve(),267		)268	}269270	fn transfer_from(271		&self,272		sender: T::CrossAccountId,273		from: T::CrossAccountId,274		to: T::CrossAccountId,275		token: TokenId,276		amount: u128,277		nesting_budget: &dyn Budget,278	) -> DispatchResultWithPostInfo {279		with_weight(280			<Pallet<T>>::transfer_from(self, &sender, &from, &to, token, amount, nesting_budget),281			<CommonWeights<T>>::transfer_from(),282		)283	}284285	fn burn_from(286		&self,287		sender: T::CrossAccountId,288		from: T::CrossAccountId,289		token: TokenId,290		amount: u128,291		nesting_budget: &dyn Budget,292	) -> DispatchResultWithPostInfo {293		with_weight(294			<Pallet<T>>::burn_from(self, &sender, &from, token, amount, nesting_budget),295			<CommonWeights<T>>::burn_from(),296		)297	}298299	fn set_collection_properties(300		&self,301		sender: T::CrossAccountId,302		properties: Vec<Property>,303	) -> DispatchResultWithPostInfo {304		let weight = <CommonWeights<T>>::set_collection_properties(properties.len() as u32);305306		with_weight(307			<Pallet<T>>::set_collection_properties(self, &sender, properties),308			weight,309		)310	}311312	fn delete_collection_properties(313		&self,314		sender: &T::CrossAccountId,315		property_keys: Vec<PropertyKey>,316	) -> DispatchResultWithPostInfo {317		let weight = <CommonWeights<T>>::delete_collection_properties(property_keys.len() as u32);318319		with_weight(320			<Pallet<T>>::delete_collection_properties(self, sender, property_keys),321			weight,322		)323	}324325	fn set_token_properties(326		&self,327		sender: T::CrossAccountId,328		token_id: TokenId,329		properties: Vec<Property>,330		_nesting_budget: &dyn Budget,331	) -> DispatchResultWithPostInfo {332		let weight = <CommonWeights<T>>::set_token_properties(properties.len() as u32);333334		with_weight(335			<Pallet<T>>::set_token_properties(self, &sender, token_id, properties, false),336			weight,337		)338	}339340	fn set_token_property_permissions(341		&self,342		sender: &T::CrossAccountId,343		property_permissions: Vec<PropertyKeyPermission>,344	) -> DispatchResultWithPostInfo {345		let weight =346			<CommonWeights<T>>::set_token_property_permissions(property_permissions.len() as u32);347348		with_weight(349			<Pallet<T>>::set_token_property_permissions(self, sender, property_permissions),350			weight,351		)352	}353354	fn delete_token_properties(355		&self,356		sender: T::CrossAccountId,357		token_id: TokenId,358		property_keys: Vec<PropertyKey>,359		_nesting_budget: &dyn Budget,360	) -> DispatchResultWithPostInfo {361		let weight = <CommonWeights<T>>::delete_token_properties(property_keys.len() as u32);362363		with_weight(364			<Pallet<T>>::delete_token_properties(self, &sender, token_id, property_keys),365			weight,366		)367	}368369	fn check_nesting(370		&self,371		_sender: <T>::CrossAccountId,372		_from: (CollectionId, TokenId),373		_under: TokenId,374		_nesting_budget: &dyn Budget,375	) -> sp_runtime::DispatchResult {376		fail!(<Error<T>>::RefungibleDisallowsNesting)377	}378379	fn nest(&self, _under: TokenId, _to_nest: (CollectionId, TokenId)) {}380381	fn unnest(&self, _under: TokenId, _to_nest: (CollectionId, TokenId)) {}382383	fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId> {384		<Owned<T>>::iter_prefix((self.id, account))385			.map(|(id, _)| id)386			.collect()387	}388389	fn collection_tokens(&self) -> Vec<TokenId> {390		<TokenData<T>>::iter_prefix((self.id,))391			.map(|(id, _)| id)392			.collect()393	}394395	fn token_exists(&self, token: TokenId) -> bool {396		<Pallet<T>>::token_exists(self, token)397	}398399	fn last_token_id(&self) -> TokenId {400		TokenId(<TokensMinted<T>>::get(self.id))401	}402403	fn token_owner(&self, _token: TokenId) -> Option<T::CrossAccountId> {404		None405	}406407	fn token_property(&self, _token_id: TokenId, _key: &PropertyKey) -> Option<PropertyValue> {408		None409	}410411	fn token_properties(412		&self,413		_token_id: TokenId,414		_keys: Option<Vec<PropertyKey>>,415	) -> Vec<Property> {416		Vec::new()417	}418419	fn total_supply(&self) -> u32 {420		<Pallet<T>>::total_supply(self)421	}422423	fn account_balance(&self, account: T::CrossAccountId) -> u32 {424		<AccountBalance<T>>::get((self.id, account))425	}426427	fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128 {428		<Balance<T>>::get((self.id, token, account))429	}430431	fn allowance(432		&self,433		sender: T::CrossAccountId,434		spender: T::CrossAccountId,435		token: TokenId,436	) -> u128 {437		<Allowance<T>>::get((self.id, token, sender, spender))438	}439440	fn refungible_extensions(&self) -> Option<&dyn RefungibleExtensions<T>> {441		Some(self)442	}443444	fn total_pieces(&self, token: TokenId) -> Option<u128> {445		<Pallet<T>>::total_pieces(self.id, token)446	}447}448449impl<T: Config> RefungibleExtensions<T> for RefungibleHandle<T> {450	fn repartition(451		&self,452		owner: &T::CrossAccountId,453		token: TokenId,454		amount: u128,455	) -> DispatchResultWithPostInfo {456		with_weight(457			<Pallet<T>>::repartition(self, owner, token, amount),458			<SelfWeightOf<T>>::repartition_item(),459		)460	}461}