git.delta.rocks / unique-network / refs/commits / 9322aa344eaa

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, BoundedVec};20use up_data_structs::{21	TokenId, CustomDataLimit, CreateItemExData, CollectionId, budget::Budget, Property,22};23use pallet_common::{CommonCollectionOperations, CommonWeightInfo, with_weight};24use sp_runtime::DispatchError;25use sp_std::vec::Vec;2627use crate::{28	AccountBalance, Allowance, Config, CreateItemData, Error, NonfungibleHandle, Owned, Pallet,29	SelfWeightOf, TokenData, weights::WeightInfo, TokensMinted,30};3132pub struct CommonWeights<T: Config>(PhantomData<T>);33impl<T: Config> CommonWeightInfo<T::CrossAccountId> for CommonWeights<T> {34	fn create_item() -> Weight {35		<SelfWeightOf<T>>::create_item()36	}3738	fn create_multiple_items_ex(data: &CreateItemExData<T::CrossAccountId>) -> Weight {39		match data {40			CreateItemExData::NFT(t) => <SelfWeightOf<T>>::create_multiple_items_ex(t.len() as u32),41			_ => 0,42		}43	}4445	fn create_multiple_items(amount: u32) -> Weight {46		<SelfWeightOf<T>>::create_multiple_items(amount)47	}4849	fn burn_item() -> Weight {50		<SelfWeightOf<T>>::burn_item()51	}5253	fn change_collection_properties(amount: u32) -> Weight {54		<SelfWeightOf<T>>::change_collection_properties(amount)55	}5657	fn change_token_properties(amount: u32) -> Weight {58		<SelfWeightOf<T>>::change_token_properties(amount)59	}6061	fn transfer() -> Weight {62		<SelfWeightOf<T>>::transfer()63	}6465	fn approve() -> Weight {66		<SelfWeightOf<T>>::approve()67	}6869	fn transfer_from() -> Weight {70		<SelfWeightOf<T>>::transfer_from()71	}7273	fn burn_from() -> Weight {74		<SelfWeightOf<T>>::burn_from()75	}7677	fn set_variable_metadata(bytes: u32) -> Weight {78		<SelfWeightOf<T>>::set_variable_metadata(bytes)79	}80}8182fn map_create_data<T: Config>(83	data: up_data_structs::CreateItemData,84	to: &T::CrossAccountId,85) -> Result<CreateItemData<T>, DispatchError> {86	match data {87		up_data_structs::CreateItemData::NFT(data) => Ok(CreateItemData::<T> {88			const_data: data.const_data,89			variable_data: data.variable_data,90			owner: to.clone(),91		}),92		_ => fail!(<Error<T>>::NotNonfungibleDataUsedToMintFungibleCollectionToken),93	}94}9596impl<T: Config> CommonCollectionOperations<T> for NonfungibleHandle<T> {97	fn create_item(98		&self,99		sender: T::CrossAccountId,100		to: T::CrossAccountId,101		data: up_data_structs::CreateItemData,102		nesting_budget: &dyn Budget,103	) -> DispatchResultWithPostInfo {104		with_weight(105			<Pallet<T>>::create_item(106				self,107				&sender,108				map_create_data::<T>(data, &to)?,109				nesting_budget,110			),111			<CommonWeights<T>>::create_item(),112		)113	}114115	fn create_multiple_items(116		&self,117		sender: T::CrossAccountId,118		to: T::CrossAccountId,119		data: Vec<up_data_structs::CreateItemData>,120		nesting_budget: &dyn Budget,121	) -> DispatchResultWithPostInfo {122		let data = data123			.into_iter()124			.map(|d| map_create_data::<T>(d, &to))125			.collect::<Result<Vec<_>, DispatchError>>()?;126127		let amount = data.len();128		with_weight(129			<Pallet<T>>::create_multiple_items(self, &sender, data, nesting_budget),130			<CommonWeights<T>>::create_multiple_items(amount as u32),131		)132	}133134	fn create_multiple_items_ex(135		&self,136		sender: <T>::CrossAccountId,137		data: up_data_structs::CreateItemExData<<T>::CrossAccountId>,138		nesting_budget: &dyn Budget,139	) -> DispatchResultWithPostInfo {140		let weight = <CommonWeights<T>>::create_multiple_items_ex(&data);141		let data = match data {142			up_data_structs::CreateItemExData::NFT(nft) => nft,143			_ => fail!(Error::<T>::NotNonfungibleDataUsedToMintFungibleCollectionToken),144		};145146		with_weight(147			<Pallet<T>>::create_multiple_items(self, &sender, data.into_inner(), nesting_budget),148			weight,149		)150	}151152	fn change_collection_properties(153		&self,154		sender: T::CrossAccountId,155		properties: Vec<Property>,156	) -> DispatchResultWithPostInfo {157		let weight = <CommonWeights<T>>::change_collection_properties(properties.len() as u32);158159		with_weight(160			<Pallet<T>>::change_collection_properties(self, &sender, properties),161			weight162		)163	}164165	fn change_token_properties(166		&self,167		sender: T::CrossAccountId,168		token_id: TokenId,169		properties: Vec<Property>,170	) -> DispatchResultWithPostInfo {171		let weight = <CommonWeights<T>>::change_token_properties(properties.len() as u32);172173		with_weight(174			<Pallet<T>>::change_token_properties(self, &sender, token_id, properties),175			weight176		)177	}178179	fn burn_item(180		&self,181		sender: T::CrossAccountId,182		token: TokenId,183		amount: u128,184	) -> DispatchResultWithPostInfo {185		ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);186		if amount == 1 {187			with_weight(188				<Pallet<T>>::burn(self, &sender, token),189				<CommonWeights<T>>::burn_item(),190			)191		} else {192			Ok(().into())193		}194	}195196	fn transfer(197		&self,198		from: T::CrossAccountId,199		to: T::CrossAccountId,200		token: TokenId,201		amount: u128,202		nesting_budget: &dyn Budget,203	) -> DispatchResultWithPostInfo {204		ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);205		if amount == 1 {206			with_weight(207				<Pallet<T>>::transfer(self, &from, &to, token, nesting_budget),208				<CommonWeights<T>>::transfer(),209			)210		} else {211			Ok(().into())212		}213	}214215	fn approve(216		&self,217		sender: T::CrossAccountId,218		spender: T::CrossAccountId,219		token: TokenId,220		amount: u128,221	) -> DispatchResultWithPostInfo {222		ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);223224		with_weight(225			if amount == 1 {226				<Pallet<T>>::set_allowance(self, &sender, token, Some(&spender))227			} else {228				<Pallet<T>>::set_allowance(self, &sender, token, None)229			},230			<CommonWeights<T>>::approve(),231		)232	}233234	fn transfer_from(235		&self,236		sender: T::CrossAccountId,237		from: T::CrossAccountId,238		to: T::CrossAccountId,239		token: TokenId,240		amount: u128,241		nesting_budget: &dyn Budget,242	) -> DispatchResultWithPostInfo {243		ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);244245		if amount == 1 {246			with_weight(247				<Pallet<T>>::transfer_from(self, &sender, &from, &to, token, nesting_budget),248				<CommonWeights<T>>::transfer_from(),249			)250		} else {251			Ok(().into())252		}253	}254255	fn burn_from(256		&self,257		sender: T::CrossAccountId,258		from: T::CrossAccountId,259		token: TokenId,260		amount: u128,261		nesting_budget: &dyn Budget,262	) -> DispatchResultWithPostInfo {263		ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);264265		if amount == 1 {266			with_weight(267				<Pallet<T>>::burn_from(self, &sender, &from, token, nesting_budget),268				<CommonWeights<T>>::burn_from(),269			)270		} else {271			Ok(().into())272		}273	}274275	fn set_variable_metadata(276		&self,277		sender: T::CrossAccountId,278		token: TokenId,279		data: BoundedVec<u8, CustomDataLimit>,280	) -> DispatchResultWithPostInfo {281		let len = data.len();282		with_weight(283			<Pallet<T>>::set_variable_metadata(self, &sender, token, data),284			<CommonWeights<T>>::set_variable_metadata(len as u32),285		)286	}287288	fn check_nesting(289		&self,290		sender: T::CrossAccountId,291		from: (CollectionId, TokenId),292		under: TokenId,293		budget: &dyn Budget,294	) -> sp_runtime::DispatchResult {295		<Pallet<T>>::check_nesting(self, sender, from, under, budget)296	}297298	fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId> {299		<Owned<T>>::iter_prefix((self.id, account))300			.map(|(id, _)| id)301			.collect()302	}303304	fn collection_tokens(&self) -> Vec<TokenId> {305		<TokenData<T>>::iter_prefix((self.id,))306			.map(|(id, _)| id)307			.collect()308	}309310	fn token_exists(&self, token: TokenId) -> bool {311		<Pallet<T>>::token_exists(self, token)312	}313314	fn last_token_id(&self) -> TokenId {315		TokenId(<TokensMinted<T>>::get(self.id))316	}317318	fn token_owner(&self, token: TokenId) -> Option<T::CrossAccountId> {319		<TokenData<T>>::get((self.id, token)).map(|t| t.owner)320	}321	fn const_metadata(&self, token: TokenId) -> Vec<u8> {322		<TokenData<T>>::get((self.id, token))323			.map(|t| t.const_data)324			.unwrap_or_default()325			.into_inner()326	}327	fn variable_metadata(&self, token: TokenId) -> Vec<u8> {328		<TokenData<T>>::get((self.id, token))329			.map(|t| t.variable_data)330			.unwrap_or_default()331			.into_inner()332	}333334	fn total_supply(&self) -> u32 {335		<Pallet<T>>::total_supply(self)336	}337338	fn account_balance(&self, account: T::CrossAccountId) -> u32 {339		<AccountBalance<T>>::get((self.id, account))340	}341342	fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128 {343		if <TokenData<T>>::get((self.id, token))344			.map(|a| a.owner == account)345			.unwrap_or(false)346		{347			1348		} else {349			0350		}351	}352353	fn allowance(354		&self,355		sender: T::CrossAccountId,356		spender: T::CrossAccountId,357		token: TokenId,358	) -> u128 {359		if <TokenData<T>>::get((self.id, token))360			.map(|a| a.owner != sender)361			.unwrap_or(true)362		{363			0364		} else if <Allowance<T>>::get((self.id, token)) == Some(spender) {365			1366		} else {367			0368		}369	}370}