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 frame_support::{dispatch::DispatchResultWithPostInfo, ensure, fail, weights::Weight};20use up_data_structs::{21	TokenId, CreateItemExData, CollectionId, budget::Budget, Property, PropertyKey,22	PropertyKeyPermission, PropertyValue,23};24use pallet_common::{25	CommonCollectionOperations, CommonWeightInfo, with_weight, weights::WeightInfo as _,26};27use sp_runtime::DispatchError;28use sp_std::vec::Vec;2930use crate::{31	AccountBalance, Allowance, Config, CreateItemData, Error, NonfungibleHandle, Owned, Pallet,32	SelfWeightOf, TokenData, weights::WeightInfo, TokensMinted,33};3435pub struct CommonWeights<T: Config>(PhantomData<T>);36impl<T: Config> CommonWeightInfo<T::CrossAccountId> for CommonWeights<T> {37	fn create_item() -> Weight {38		<SelfWeightOf<T>>::create_item()39	}4041	fn create_multiple_items_ex(data: &CreateItemExData<T::CrossAccountId>) -> Weight {42		match data {43			CreateItemExData::NFT(t) => {44				<SelfWeightOf<T>>::create_multiple_items_ex(t.len() as u32)45					+ t.iter()46						.map(|t| {47							if t.properties.len() > 0 {48								Self::set_token_properties(t.properties.len() as u32)49							} else {50								051							}52						})53						.sum::<u64>()54			}55			_ => 0,56		}57	}5859	fn create_multiple_items(data: &[up_data_structs::CreateItemData]) -> Weight {60		<SelfWeightOf<T>>::create_multiple_items(data.len() as u32)61			+ data62				.iter()63				.filter_map(|t| match t {64					up_data_structs::CreateItemData::NFT(n) if n.properties.len() > 0 => {65						Some(Self::set_token_properties(n.properties.len() as u32))66					}67					_ => None,68				})69				.sum::<u64>()70	}7172	fn burn_item() -> Weight {73		<SelfWeightOf<T>>::burn_item()74	}7576	fn set_collection_properties(amount: u32) -> Weight {77		<pallet_common::SelfWeightOf<T>>::set_collection_properties(amount)78	}7980	fn delete_collection_properties(amount: u32) -> Weight {81		<pallet_common::SelfWeightOf<T>>::delete_collection_properties(amount)82	}8384	fn set_token_properties(amount: u32) -> Weight {85		<SelfWeightOf<T>>::set_token_properties(amount)86	}8788	fn delete_token_properties(amount: u32) -> Weight {89		<SelfWeightOf<T>>::delete_token_properties(amount)90	}9192	fn set_property_permissions(amount: u32) -> Weight {93		<SelfWeightOf<T>>::set_property_permissions(amount)94	}9596	fn transfer() -> Weight {97		<SelfWeightOf<T>>::transfer()98	}99100	fn approve() -> Weight {101		<SelfWeightOf<T>>::approve()102	}103104	fn transfer_from() -> Weight {105		<SelfWeightOf<T>>::transfer_from()106	}107108	fn burn_from() -> Weight {109		<SelfWeightOf<T>>::burn_from()110	}111}112113fn map_create_data<T: Config>(114	data: up_data_structs::CreateItemData,115	to: &T::CrossAccountId,116) -> Result<CreateItemData<T>, DispatchError> {117	match data {118		up_data_structs::CreateItemData::NFT(data) => Ok(CreateItemData::<T> {119			properties: data.properties,120			owner: to.clone(),121		}),122		_ => fail!(<Error<T>>::NotNonfungibleDataUsedToMintFungibleCollectionToken),123	}124}125126impl<T: Config> CommonCollectionOperations<T> for NonfungibleHandle<T> {127	fn create_item(128		&self,129		sender: T::CrossAccountId,130		to: T::CrossAccountId,131		data: up_data_structs::CreateItemData,132		nesting_budget: &dyn Budget,133	) -> DispatchResultWithPostInfo {134		with_weight(135			<Pallet<T>>::create_item(136				self,137				&sender,138				map_create_data::<T>(data, &to)?,139				nesting_budget,140			),141			<CommonWeights<T>>::create_item(),142		)143	}144145	fn create_multiple_items(146		&self,147		sender: T::CrossAccountId,148		to: T::CrossAccountId,149		data: Vec<up_data_structs::CreateItemData>,150		nesting_budget: &dyn Budget,151	) -> DispatchResultWithPostInfo {152		let weight = <CommonWeights<T>>::create_multiple_items(&data);153		let data = data154			.into_iter()155			.map(|d| map_create_data::<T>(d, &to))156			.collect::<Result<Vec<_>, DispatchError>>()?;157158		with_weight(159			<Pallet<T>>::create_multiple_items(self, &sender, data, nesting_budget),160			weight,161		)162	}163164	fn create_multiple_items_ex(165		&self,166		sender: <T>::CrossAccountId,167		data: up_data_structs::CreateItemExData<<T>::CrossAccountId>,168		nesting_budget: &dyn Budget,169	) -> DispatchResultWithPostInfo {170		let weight = <CommonWeights<T>>::create_multiple_items_ex(&data);171		let data = match data {172			up_data_structs::CreateItemExData::NFT(nft) => nft,173			_ => fail!(Error::<T>::NotNonfungibleDataUsedToMintFungibleCollectionToken),174		};175176		with_weight(177			<Pallet<T>>::create_multiple_items(self, &sender, data.into_inner(), nesting_budget),178			weight,179		)180	}181182	fn set_collection_properties(183		&self,184		sender: T::CrossAccountId,185		properties: Vec<Property>,186	) -> DispatchResultWithPostInfo {187		let weight = <CommonWeights<T>>::set_collection_properties(properties.len() as u32);188189		with_weight(190			<Pallet<T>>::set_collection_properties(self, &sender, properties),191			weight,192		)193	}194195	fn delete_collection_properties(196		&self,197		sender: &T::CrossAccountId,198		property_keys: Vec<PropertyKey>,199	) -> DispatchResultWithPostInfo {200		let weight = <CommonWeights<T>>::delete_collection_properties(property_keys.len() as u32);201202		with_weight(203			<Pallet<T>>::delete_collection_properties(self, sender, property_keys),204			weight,205		)206	}207208	fn set_token_properties(209		&self,210		sender: T::CrossAccountId,211		token_id: TokenId,212		properties: Vec<Property>,213	) -> DispatchResultWithPostInfo {214		let weight = <CommonWeights<T>>::set_token_properties(properties.len() as u32);215216		with_weight(217			<Pallet<T>>::set_token_properties(self, &sender, token_id, properties),218			weight,219		)220	}221222	fn delete_token_properties(223		&self,224		sender: T::CrossAccountId,225		token_id: TokenId,226		property_keys: Vec<PropertyKey>,227	) -> DispatchResultWithPostInfo {228		let weight = <CommonWeights<T>>::delete_token_properties(property_keys.len() as u32);229230		with_weight(231			<Pallet<T>>::delete_token_properties(self, &sender, token_id, property_keys),232			weight,233		)234	}235236	fn set_property_permissions(237		&self,238		sender: &T::CrossAccountId,239		property_permissions: Vec<PropertyKeyPermission>,240	) -> DispatchResultWithPostInfo {241		let weight =242			<CommonWeights<T>>::set_property_permissions(property_permissions.len() as u32);243244		with_weight(245			<Pallet<T>>::set_property_permissions(self, sender, property_permissions),246			weight,247		)248	}249250	fn burn_item(251		&self,252		sender: T::CrossAccountId,253		token: TokenId,254		amount: u128,255	) -> DispatchResultWithPostInfo {256		ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);257		if amount == 1 {258			with_weight(259				<Pallet<T>>::burn(self, &sender, token),260				<CommonWeights<T>>::burn_item(),261			)262		} else {263			Ok(().into())264		}265	}266267	fn burn_item_unchecked(268		&self,269		owner:& T::CrossAccountId,270		token: TokenId,271		amount: u128,272	) -> sp_runtime::DispatchResult {273		if amount == 1 {274			<Pallet<T>>::burn_item_unchecked(self, owner, token)275		} else {276			Ok(())277		}278	}279280	fn transfer(281		&self,282		from: T::CrossAccountId,283		to: T::CrossAccountId,284		token: TokenId,285		amount: u128,286		nesting_budget: &dyn Budget,287	) -> DispatchResultWithPostInfo {288		ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);289		if amount == 1 {290			with_weight(291				<Pallet<T>>::transfer(self, &from, &to, token, nesting_budget),292				<CommonWeights<T>>::transfer(),293			)294		} else {295			Ok(().into())296		}297	}298299	fn approve(300		&self,301		sender: T::CrossAccountId,302		spender: T::CrossAccountId,303		token: TokenId,304		amount: u128,305	) -> DispatchResultWithPostInfo {306		ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);307308		with_weight(309			if amount == 1 {310				<Pallet<T>>::set_allowance(self, &sender, token, Some(&spender))311			} else {312				<Pallet<T>>::set_allowance(self, &sender, token, None)313			},314			<CommonWeights<T>>::approve(),315		)316	}317318	fn transfer_from(319		&self,320		sender: T::CrossAccountId,321		from: T::CrossAccountId,322		to: T::CrossAccountId,323		token: TokenId,324		amount: u128,325		nesting_budget: &dyn Budget,326	) -> DispatchResultWithPostInfo {327		ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);328329		if amount == 1 {330			with_weight(331				<Pallet<T>>::transfer_from(self, &sender, &from, &to, token, nesting_budget),332				<CommonWeights<T>>::transfer_from(),333			)334		} else {335			Ok(().into())336		}337	}338339	fn burn_from(340		&self,341		sender: T::CrossAccountId,342		from: T::CrossAccountId,343		token: TokenId,344		amount: u128,345		nesting_budget: &dyn Budget,346	) -> DispatchResultWithPostInfo {347		ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);348349		if amount == 1 {350			with_weight(351				<Pallet<T>>::burn_from(self, &sender, &from, token, nesting_budget),352				<CommonWeights<T>>::burn_from(),353			)354		} else {355			Ok(().into())356		}357	}358359	fn check_nesting(360		&self,361		sender: T::CrossAccountId,362		from: (CollectionId, TokenId),363		under: TokenId,364		budget: &dyn Budget,365	) -> sp_runtime::DispatchResult {366		<Pallet<T>>::check_nesting(self, sender, from, under, budget)367	}368369	fn nest(370		&self,371		under: TokenId,372		to_nest: (CollectionId, TokenId)373	) {374		<Pallet<T>>::nest((self.id, under), to_nest);375	}376377	fn unnest(378		&self,379		under: TokenId,380		to_unnest: (CollectionId, TokenId)381	) {382		<Pallet<T>>::unnest((self.id, under), to_unnest);383	}384385	fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId> {386		<Owned<T>>::iter_prefix((self.id, account))387			.map(|(id, _)| id)388			.collect()389	}390391	fn collection_tokens(&self) -> Vec<TokenId> {392		<TokenData<T>>::iter_prefix((self.id,))393			.map(|(id, _)| id)394			.collect()395	}396397	fn token_exists(&self, token: TokenId) -> bool {398		<Pallet<T>>::token_exists(self, token)399	}400401	fn last_token_id(&self) -> TokenId {402		TokenId(<TokensMinted<T>>::get(self.id))403	}404405	fn token_owner(&self, token: TokenId) -> Option<T::CrossAccountId> {406		<TokenData<T>>::get((self.id, token)).map(|t| t.owner)407	}408409	fn token_property(&self, token_id: TokenId, key: &PropertyKey) -> Option<PropertyValue> {410		<Pallet<T>>::token_properties((self.id, token_id))411			.get(key)412			.cloned()413	}414415	fn token_properties(&self, token_id: TokenId, keys: Option<Vec<PropertyKey>>) -> Vec<Property> {416		let properties = <Pallet<T>>::token_properties((self.id, token_id));417418		keys.map(|keys| {419			keys.into_iter()420				.filter_map(|key| {421					properties.get(&key).map(|value| Property {422						key,423						value: value.clone(),424					})425				})426				.collect()427		})428		.unwrap_or_else(|| {429			properties430				.into_iter()431				.map(|(key, value)| Property {432					key: key.clone(),433					value: value.clone(),434				})435				.collect()436		})437	}438439	fn total_supply(&self) -> u32 {440		<Pallet<T>>::total_supply(self)441	}442443	fn account_balance(&self, account: T::CrossAccountId) -> u32 {444		<AccountBalance<T>>::get((self.id, account))445	}446447	fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128 {448		if <TokenData<T>>::get((self.id, token))449			.map(|a| a.owner == account)450			.unwrap_or(false)451		{452			1453		} else {454			0455		}456	}457458	fn allowance(459		&self,460		sender: T::CrossAccountId,461		spender: T::CrossAccountId,462		token: TokenId,463	) -> u128 {464		if <TokenData<T>>::get((self.id, token))465			.map(|a| a.owner != sender)466			.unwrap_or(true)467		{468			0469		} else if <Allowance<T>>::get((self.id, token)) == Some(spender) {470			1471		} else {472			0473		}474	}475}