git.delta.rocks / unique-network / refs/commits / ca9fba1ceb64

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		}),144		_ => fail!(<Error<T>>::NotRefungibleDataUsedToMintFungibleCollectionToken),145	}146}147148impl<T: Config> CommonCollectionOperations<T> for RefungibleHandle<T> {149	fn create_item(150		&self,151		sender: T::CrossAccountId,152		to: T::CrossAccountId,153		data: up_data_structs::CreateItemData,154		nesting_budget: &dyn Budget,155	) -> DispatchResultWithPostInfo {156		with_weight(157			<Pallet<T>>::create_item(158				self,159				&sender,160				map_create_data::<T>(data, &to)?,161				nesting_budget,162			),163			<CommonWeights<T>>::create_item(),164		)165	}166167	fn create_multiple_items(168		&self,169		sender: T::CrossAccountId,170		to: T::CrossAccountId,171		data: Vec<up_data_structs::CreateItemData>,172		nesting_budget: &dyn Budget,173	) -> DispatchResultWithPostInfo {174		let weight = <CommonWeights<T>>::create_multiple_items(&data);175		let data = data176			.into_iter()177			.map(|d| map_create_data::<T>(d, &to))178			.collect::<Result<Vec<_>, DispatchError>>()?;179180		with_weight(181			<Pallet<T>>::create_multiple_items(self, &sender, data, nesting_budget),182			weight,183		)184	}185186	fn create_multiple_items_ex(187		&self,188		sender: <T>::CrossAccountId,189		data: CreateItemExData<T::CrossAccountId>,190		nesting_budget: &dyn Budget,191	) -> DispatchResultWithPostInfo {192		let weight = <CommonWeights<T>>::create_multiple_items_ex(&data);193		let data = match data {194			CreateItemExData::RefungibleMultipleOwners(r) => vec![r],195			CreateItemExData::RefungibleMultipleItems(r)196				if r.iter().all(|i| i.users.len() == 1) =>197			{198				r.into_inner()199			}200			_ => fail!(<Error<T>>::NotRefungibleDataUsedToMintFungibleCollectionToken),201		};202203		with_weight(204			<Pallet<T>>::create_multiple_items(self, &sender, data, nesting_budget),205			weight,206		)207	}208209	fn burn_item(210		&self,211		sender: T::CrossAccountId,212		token: TokenId,213		amount: u128,214	) -> DispatchResultWithPostInfo {215		with_weight(216			<Pallet<T>>::burn(self, &sender, token, amount),217			<CommonWeights<T>>::burn_item(),218		)219	}220221	fn burn_item_recursively(222		&self,223		sender: T::CrossAccountId,224		token: TokenId,225		self_budget: &dyn Budget,226		_breadth_budget: &dyn Budget,227	) -> DispatchResultWithPostInfo {228		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);229		with_weight(230			<Pallet<T>>::burn(231				self,232				&sender,233				token,234				<Balance<T>>::get((self.id, token, &sender)),235			),236			<CommonWeights<T>>::burn_recursively_self_raw(),237		)238	}239240	fn transfer(241		&self,242		from: T::CrossAccountId,243		to: T::CrossAccountId,244		token: TokenId,245		amount: u128,246		nesting_budget: &dyn Budget,247	) -> DispatchResultWithPostInfo {248		with_weight(249			<Pallet<T>>::transfer(self, &from, &to, token, amount, nesting_budget),250			<CommonWeights<T>>::transfer(),251		)252	}253254	fn approve(255		&self,256		sender: T::CrossAccountId,257		spender: T::CrossAccountId,258		token: TokenId,259		amount: u128,260	) -> DispatchResultWithPostInfo {261		with_weight(262			<Pallet<T>>::set_allowance(self, &sender, &spender, token, amount),263			<CommonWeights<T>>::approve(),264		)265	}266267	fn transfer_from(268		&self,269		sender: T::CrossAccountId,270		from: T::CrossAccountId,271		to: T::CrossAccountId,272		token: TokenId,273		amount: u128,274		nesting_budget: &dyn Budget,275	) -> DispatchResultWithPostInfo {276		with_weight(277			<Pallet<T>>::transfer_from(self, &sender, &from, &to, token, amount, nesting_budget),278			<CommonWeights<T>>::transfer_from(),279		)280	}281282	fn burn_from(283		&self,284		sender: T::CrossAccountId,285		from: T::CrossAccountId,286		token: TokenId,287		amount: u128,288		nesting_budget: &dyn Budget,289	) -> DispatchResultWithPostInfo {290		with_weight(291			<Pallet<T>>::burn_from(self, &sender, &from, token, amount, nesting_budget),292			<CommonWeights<T>>::burn_from(),293		)294	}295296	fn set_collection_properties(297		&self,298		_sender: T::CrossAccountId,299		_property: Vec<Property>,300	) -> DispatchResultWithPostInfo {301		fail!(<Error<T>>::SettingPropertiesNotAllowed)302	}303304	fn delete_collection_properties(305		&self,306		_sender: &T::CrossAccountId,307		_property_keys: Vec<PropertyKey>,308	) -> DispatchResultWithPostInfo {309		fail!(<Error<T>>::SettingPropertiesNotAllowed)310	}311312	fn set_token_properties(313		&self,314		_sender: T::CrossAccountId,315		_token_id: TokenId,316		_property: Vec<Property>,317		_nesting_budget: &dyn Budget,318	) -> DispatchResultWithPostInfo {319		fail!(<Error<T>>::SettingPropertiesNotAllowed)320	}321322	fn set_token_property_permissions(323		&self,324		_sender: &T::CrossAccountId,325		_property_permissions: Vec<PropertyKeyPermission>,326	) -> DispatchResultWithPostInfo {327		fail!(<Error<T>>::SettingPropertiesNotAllowed)328	}329330	fn delete_token_properties(331		&self,332		_sender: T::CrossAccountId,333		_token_id: TokenId,334		_property_keys: Vec<PropertyKey>,335		_nesting_budget: &dyn Budget,336	) -> DispatchResultWithPostInfo {337		fail!(<Error<T>>::SettingPropertiesNotAllowed)338	}339340	fn check_nesting(341		&self,342		_sender: <T>::CrossAccountId,343		_from: (CollectionId, TokenId),344		_under: TokenId,345		_nesting_budget: &dyn Budget,346	) -> sp_runtime::DispatchResult {347		fail!(<Error<T>>::RefungibleDisallowsNesting)348	}349350	fn nest(&self, _under: TokenId, _to_nest: (CollectionId, TokenId)) {}351352	fn unnest(&self, _under: TokenId, _to_nest: (CollectionId, TokenId)) {}353354	fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId> {355		<Owned<T>>::iter_prefix((self.id, account))356			.map(|(id, _)| id)357			.collect()358	}359360	fn collection_tokens(&self) -> Vec<TokenId> {361		<TokenData<T>>::iter_prefix((self.id,))362			.map(|(id, _)| id)363			.collect()364	}365366	fn token_exists(&self, token: TokenId) -> bool {367		<Pallet<T>>::token_exists(self, token)368	}369370	fn last_token_id(&self) -> TokenId {371		TokenId(<TokensMinted<T>>::get(self.id))372	}373374	fn token_owner(&self, _token: TokenId) -> Option<T::CrossAccountId> {375		None376	}377378	fn token_property(&self, _token_id: TokenId, _key: &PropertyKey) -> Option<PropertyValue> {379		None380	}381382	fn token_properties(383		&self,384		_token_id: TokenId,385		_keys: Option<Vec<PropertyKey>>,386	) -> Vec<Property> {387		Vec::new()388	}389390	fn total_supply(&self) -> u32 {391		<Pallet<T>>::total_supply(self)392	}393394	fn account_balance(&self, account: T::CrossAccountId) -> u32 {395		<AccountBalance<T>>::get((self.id, account))396	}397398	fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128 {399		<Balance<T>>::get((self.id, token, account))400	}401402	fn allowance(403		&self,404		sender: T::CrossAccountId,405		spender: T::CrossAccountId,406		token: TokenId,407	) -> u128 {408		<Allowance<T>>::get((self.id, token, sender, spender))409	}410411	fn refungible_extensions(&self) -> Option<&dyn RefungibleExtensions<T>> {412		Some(self)413	}414415	fn total_pieces(&self, token: TokenId) -> Option<u128> {416		<Pallet<T>>::total_pieces(self.id, token)417	}418}419420impl<T: Config> RefungibleExtensions<T> for RefungibleHandle<T> {421	fn repartition(422		&self,423		owner: &T::CrossAccountId,424		token: TokenId,425		amount: u128,426	) -> DispatchResultWithPostInfo {427		with_weight(428			<Pallet<T>>::repartition(self, owner, token, amount),429			<SelfWeightOf<T>>::repartition_item(),430		)431	}432}