git.delta.rocks / unique-network / refs/commits / 77688fbffa89

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, fail, weights::Weight};21use up_data_structs::{22	CollectionId, TokenId, CreateItemExData, CreateRefungibleExData, budget::Budget, Property,23	PropertyKey, PropertyValue, PropertyKeyPermission, CreateItemData,24};25use pallet_common::{CommonCollectionOperations, CommonWeightInfo, with_weight};26use sp_runtime::DispatchError;27use sp_std::{vec::Vec, vec};2829use crate::{30	AccountBalance, Allowance, Balance, Config, Error, Owned, Pallet, RefungibleHandle,31	SelfWeightOf, TokenData, weights::WeightInfo, TokensMinted,32};3334macro_rules! max_weight_of {35	($($method:ident ($($args:tt)*)),*) => {36		037		$(38			.max(<SelfWeightOf<T>>::$method($($args)*))39		)*40	};41}4243pub struct CommonWeights<T: Config>(PhantomData<T>);44impl<T: Config> CommonWeightInfo<T::CrossAccountId> for CommonWeights<T> {45	fn create_item() -> Weight {46		<SelfWeightOf<T>>::create_item()47	}4849	fn create_multiple_items(data: &[CreateItemData]) -> Weight {50		<SelfWeightOf<T>>::create_multiple_items(data.len() as u32)51	}5253	fn create_multiple_items_ex(call: &CreateItemExData<T::CrossAccountId>) -> Weight {54		match call {55			CreateItemExData::RefungibleMultipleOwners(i) => {56				<SelfWeightOf<T>>::create_multiple_items_ex_multiple_owners(i.users.len() as u32)57			}58			CreateItemExData::RefungibleMultipleItems(i) => {59				<SelfWeightOf<T>>::create_multiple_items_ex_multiple_items(i.len() as u32)60			}61			_ => 0,62		}63	}6465	fn burn_item() -> Weight {66		max_weight_of!(burn_item_partial(), burn_item_fully())67	}6869	fn set_collection_properties(_amount: u32) -> Weight {70		// Error71		072	}7374	fn delete_collection_properties(_amount: u32) -> Weight {75		// Error76		077	}7879	fn set_token_properties(amount: u32) -> Weight {80		<SelfWeightOf<T>>::set_token_properties(amount)81	}8283	fn delete_token_properties(amount: u32) -> Weight {84		<SelfWeightOf<T>>::delete_token_properties(amount)85	}8687	fn set_property_permissions(amount: u32) -> Weight {88		<SelfWeightOf<T>>::set_property_permissions(amount)89	}9091	fn transfer() -> Weight {92		max_weight_of!(93			transfer_normal(),94			transfer_creating(),95			transfer_removing(),96			transfer_creating_removing()97		)98	}99100	fn approve() -> Weight {101		<SelfWeightOf<T>>::approve()102	}103104	fn transfer_from() -> Weight {105		max_weight_of!(106			transfer_from_normal(),107			transfer_from_creating(),108			transfer_from_removing(),109			transfer_from_creating_removing()110		)111	}112113	fn burn_from() -> Weight {114		<SelfWeightOf<T>>::burn_from()115	}116}117118fn map_create_data<T: Config>(119	data: up_data_structs::CreateItemData,120	to: &T::CrossAccountId,121) -> Result<CreateRefungibleExData<T::CrossAccountId>, DispatchError> {122	match data {123		up_data_structs::CreateItemData::ReFungible(data) => Ok(CreateRefungibleExData {124			const_data: data.const_data,125			users: {126				let mut out = BTreeMap::new();127				out.insert(to.clone(), data.pieces);128				out.try_into().expect("limit > 0")129			},130		}),131		_ => fail!(<Error<T>>::NotRefungibleDataUsedToMintFungibleCollectionToken),132	}133}134135impl<T: Config> CommonCollectionOperations<T> for RefungibleHandle<T> {136	fn create_item(137		&self,138		sender: T::CrossAccountId,139		to: T::CrossAccountId,140		data: up_data_structs::CreateItemData,141		nesting_budget: &dyn Budget,142	) -> DispatchResultWithPostInfo {143		with_weight(144			<Pallet<T>>::create_item(145				self,146				&sender,147				map_create_data::<T>(data, &to)?,148				nesting_budget,149			),150			<CommonWeights<T>>::create_item(),151		)152	}153154	fn create_multiple_items(155		&self,156		sender: T::CrossAccountId,157		to: T::CrossAccountId,158		data: Vec<up_data_structs::CreateItemData>,159		nesting_budget: &dyn Budget,160	) -> DispatchResultWithPostInfo {161		let weight = <CommonWeights<T>>::create_multiple_items(&data);162		let data = data163			.into_iter()164			.map(|d| map_create_data::<T>(d, &to))165			.collect::<Result<Vec<_>, DispatchError>>()?;166167		with_weight(168			<Pallet<T>>::create_multiple_items(self, &sender, data, nesting_budget),169			weight,170		)171	}172173	fn create_multiple_items_ex(174		&self,175		sender: <T>::CrossAccountId,176		data: CreateItemExData<T::CrossAccountId>,177		nesting_budget: &dyn Budget,178	) -> DispatchResultWithPostInfo {179		let weight = <CommonWeights<T>>::create_multiple_items_ex(&data);180		let data = match data {181			CreateItemExData::RefungibleMultipleOwners(r) => vec![r],182			CreateItemExData::RefungibleMultipleItems(r)183				if r.iter().all(|i| i.users.len() == 1) =>184			{185				r.into_inner()186			}187			_ => fail!(<Error<T>>::NotRefungibleDataUsedToMintFungibleCollectionToken),188		};189190		with_weight(191			<Pallet<T>>::create_multiple_items(self, &sender, data, nesting_budget),192			weight,193		)194	}195196	fn burn_item(197		&self,198		sender: T::CrossAccountId,199		token: TokenId,200		amount: u128,201	) -> DispatchResultWithPostInfo {202		with_weight(203			<Pallet<T>>::burn(self, &sender, token, amount),204			<CommonWeights<T>>::burn_item(),205		)206	}207208	fn burn_item_unchecked(209		&self,210		owner: &T::CrossAccountId,211		token: TokenId,212		amount: u128,213	) -> sp_runtime::DispatchResult {214		<Pallet<T>>::burn_item_unchecked(self, owner, token, amount)215	}216217	fn transfer(218		&self,219		from: T::CrossAccountId,220		to: T::CrossAccountId,221		token: TokenId,222		amount: u128,223		nesting_budget: &dyn Budget,224	) -> DispatchResultWithPostInfo {225		with_weight(226			<Pallet<T>>::transfer(self, &from, &to, token, amount, nesting_budget),227			<CommonWeights<T>>::transfer(),228		)229	}230231	fn approve(232		&self,233		sender: T::CrossAccountId,234		spender: T::CrossAccountId,235		token: TokenId,236		amount: u128,237	) -> DispatchResultWithPostInfo {238		with_weight(239			<Pallet<T>>::set_allowance(self, &sender, &spender, token, amount),240			<CommonWeights<T>>::approve(),241		)242	}243244	fn transfer_from(245		&self,246		sender: T::CrossAccountId,247		from: T::CrossAccountId,248		to: T::CrossAccountId,249		token: TokenId,250		amount: u128,251		nesting_budget: &dyn Budget,252	) -> DispatchResultWithPostInfo {253		with_weight(254			<Pallet<T>>::transfer_from(self, &sender, &from, &to, token, amount, nesting_budget),255			<CommonWeights<T>>::transfer_from(),256		)257	}258259	fn burn_from(260		&self,261		sender: T::CrossAccountId,262		from: T::CrossAccountId,263		token: TokenId,264		amount: u128,265		nesting_budget: &dyn Budget,266	) -> DispatchResultWithPostInfo {267		with_weight(268			<Pallet<T>>::burn_from(self, &sender, &from, token, amount, nesting_budget),269			<CommonWeights<T>>::burn_from(),270		)271	}272273	fn set_collection_properties(274		&self,275		_sender: T::CrossAccountId,276		_property: Vec<Property>,277	) -> DispatchResultWithPostInfo {278		fail!(<Error<T>>::SettingPropertiesNotAllowed)279	}280281	fn delete_collection_properties(282		&self,283		_sender: &T::CrossAccountId,284		_property_keys: Vec<PropertyKey>,285	) -> DispatchResultWithPostInfo {286		fail!(<Error<T>>::SettingPropertiesNotAllowed)287	}288289	fn set_token_properties(290		&self,291		_sender: T::CrossAccountId,292		_token_id: TokenId,293		_property: Vec<Property>,294	) -> DispatchResultWithPostInfo {295		fail!(<Error<T>>::SettingPropertiesNotAllowed)296	}297298	fn set_property_permissions(299		&self,300		_sender: &T::CrossAccountId,301		_property_permissions: Vec<PropertyKeyPermission>,302	) -> DispatchResultWithPostInfo {303		fail!(<Error<T>>::SettingPropertiesNotAllowed)304	}305306	fn delete_token_properties(307		&self,308		_sender: T::CrossAccountId,309		_token_id: TokenId,310		_property_keys: Vec<PropertyKey>,311	) -> DispatchResultWithPostInfo {312		fail!(<Error<T>>::SettingPropertiesNotAllowed)313	}314315	fn check_nesting(316		&self,317		_sender: <T>::CrossAccountId,318		_from: (CollectionId, TokenId),319		_under: TokenId,320		_budget: &dyn Budget,321	) -> sp_runtime::DispatchResult {322		fail!(<Error<T>>::RefungibleDisallowsNesting)323	}324325	fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId> {326		<Owned<T>>::iter_prefix((self.id, account))327			.map(|(id, _)| id)328			.collect()329	}330331	fn collection_tokens(&self) -> Vec<TokenId> {332		<TokenData<T>>::iter_prefix((self.id,))333			.map(|(id, _)| id)334			.collect()335	}336337	fn token_exists(&self, token: TokenId) -> bool {338		<Pallet<T>>::token_exists(self, token)339	}340341	fn last_token_id(&self) -> TokenId {342		TokenId(<TokensMinted<T>>::get(self.id))343	}344345	fn token_owner(&self, _token: TokenId) -> Option<T::CrossAccountId> {346		None347	}348349	fn token_property(&self, _token_id: TokenId, _key: &PropertyKey) -> Option<PropertyValue> {350		None351	}352353	fn token_properties(354		&self,355		_token_id: TokenId,356		_keys: Option<Vec<PropertyKey>>,357	) -> Vec<Property> {358		Vec::new()359	}360361	fn total_supply(&self) -> u32 {362		<Pallet<T>>::total_supply(self)363	}364365	fn account_balance(&self, account: T::CrossAccountId) -> u32 {366		<AccountBalance<T>>::get((self.id, account))367	}368369	fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128 {370		<Balance<T>>::get((self.id, token, account))371	}372373	fn allowance(374		&self,375		sender: T::CrossAccountId,376		spender: T::CrossAccountId,377		token: TokenId,378	) -> u128 {379		<Allowance<T>>::get((self.id, token, sender, spender))380	}381}