git.delta.rocks / unique-network / refs/commits / 7563962a61ba

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, traits::Get};20use up_data_structs::{TokenId, CollectionId, CreateItemExData, budget::Budget, CreateItemData};21use pallet_common::{CommonCollectionOperations, CommonWeightInfo, RefungibleExtensions, with_weight};22use pallet_structure::Error as StructureError;23use sp_runtime::ArithmeticError;24use sp_std::{vec::Vec, vec};25use up_data_structs::{Property, PropertyKey, PropertyValue, PropertyKeyPermission};2627use crate::{28	Allowance, TotalSupply, Balance, Config, Error, FungibleHandle, Pallet, SelfWeightOf,29	weights::WeightInfo,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(_data: &[CreateItemData]) -> Weight {39		// All items minted for the same user, so it works same as create_item40		Self::create_item()41	}4243	fn create_multiple_items_ex(data: &CreateItemExData<T::CrossAccountId>) -> Weight {44		match data {45			CreateItemExData::Fungible(f) => {46				<SelfWeightOf<T>>::create_multiple_items_ex(f.len() as u32)47			}48			_ => 0,49		}50	}5152	fn burn_item() -> Weight {53		<SelfWeightOf<T>>::burn_item()54	}5556	fn set_collection_properties(_amount: u32) -> Weight {57		// Error58		059	}6061	fn delete_collection_properties(_amount: u32) -> Weight {62		// Error63		064	}6566	fn set_token_properties(_amount: u32) -> Weight {67		// Error68		069	}7071	fn delete_token_properties(_amount: u32) -> Weight {72		// Error73		074	}7576	fn set_token_property_permissions(_amount: u32) -> Weight {77		// Error78		079	}8081	fn transfer() -> Weight {82		<SelfWeightOf<T>>::transfer()83	}8485	fn approve() -> Weight {86		<SelfWeightOf<T>>::approve()87	}8889	fn transfer_from() -> Weight {90		<SelfWeightOf<T>>::transfer_from()91	}9293	fn burn_from() -> Weight {94		<SelfWeightOf<T>>::burn_from()95	}9697	fn burn_recursively_self_raw() -> Weight {98		// Read to get total balance99		Self::burn_item() + T::DbWeight::get().reads(1)100	}101102	fn burn_recursively_breadth_raw(_amount: u32) -> Weight {103		// Fungible tokens can't have children104		0105	}106}107108impl<T: Config> CommonCollectionOperations<T> for FungibleHandle<T> {109	fn create_item(110		&self,111		sender: T::CrossAccountId,112		to: T::CrossAccountId,113		data: up_data_structs::CreateItemData,114		nesting_budget: &dyn Budget,115	) -> DispatchResultWithPostInfo {116		match data {117			up_data_structs::CreateItemData::Fungible(data) => with_weight(118				<Pallet<T>>::create_item(self, &sender, (to, data.value), nesting_budget),119				<CommonWeights<T>>::create_item(),120			),121			_ => fail!(<Error<T>>::NotFungibleDataUsedToMintFungibleCollectionToken),122		}123	}124125	fn create_multiple_items(126		&self,127		sender: T::CrossAccountId,128		to: T::CrossAccountId,129		data: Vec<up_data_structs::CreateItemData>,130		nesting_budget: &dyn Budget,131	) -> DispatchResultWithPostInfo {132		let mut sum: u128 = 0;133		for data in data {134			match data {135				up_data_structs::CreateItemData::Fungible(data) => {136					sum = sum137						.checked_add(data.value)138						.ok_or(ArithmeticError::Overflow)?;139				}140				_ => fail!(<Error<T>>::NotFungibleDataUsedToMintFungibleCollectionToken),141			}142		}143144		with_weight(145			<Pallet<T>>::create_item(self, &sender, (to, sum), nesting_budget),146			<CommonWeights<T>>::create_item(),147		)148	}149150	fn create_multiple_items_ex(151		&self,152		sender: <T>::CrossAccountId,153		data: up_data_structs::CreateItemExData<<T>::CrossAccountId>,154		nesting_budget: &dyn Budget,155	) -> DispatchResultWithPostInfo {156		let weight = <CommonWeights<T>>::create_multiple_items_ex(&data);157		let data = match data {158			up_data_structs::CreateItemExData::Fungible(f) => f,159			_ => fail!(<Error<T>>::NotFungibleDataUsedToMintFungibleCollectionToken),160		};161162		with_weight(163			<Pallet<T>>::create_multiple_items(self, &sender, data.into_inner(), nesting_budget),164			weight,165		)166	}167168	fn burn_item(169		&self,170		sender: T::CrossAccountId,171		token: TokenId,172		amount: u128,173	) -> DispatchResultWithPostInfo {174		ensure!(175			token == TokenId::default(),176			<Error<T>>::FungibleItemsHaveNoId177		);178179		with_weight(180			<Pallet<T>>::burn(self, &sender, amount),181			<CommonWeights<T>>::burn_item(),182		)183	}184185	fn burn_item_recursively(186		&self,187		sender: T::CrossAccountId,188		token: TokenId,189		self_budget: &dyn Budget,190		_breadth_budget: &dyn Budget,191	) -> DispatchResultWithPostInfo {192		// Should not happen?193		ensure!(194			token == TokenId::default(),195			<Error<T>>::FungibleItemsHaveNoId196		);197		ensure!(self_budget.consume(), <StructureError<T>>::DepthLimit,);198199		with_weight(200			<Pallet<T>>::burn(self, &sender, <Balance<T>>::get((self.id, &sender))),201			<CommonWeights<T>>::burn_recursively_self_raw(),202		)203	}204205	fn transfer(206		&self,207		from: T::CrossAccountId,208		to: T::CrossAccountId,209		token: TokenId,210		amount: u128,211		nesting_budget: &dyn Budget,212	) -> DispatchResultWithPostInfo {213		ensure!(214			token == TokenId::default(),215			<Error<T>>::FungibleItemsHaveNoId216		);217218		with_weight(219			<Pallet<T>>::transfer(self, &from, &to, amount, nesting_budget),220			<CommonWeights<T>>::transfer(),221		)222	}223224	fn approve(225		&self,226		sender: T::CrossAccountId,227		spender: T::CrossAccountId,228		token: TokenId,229		amount: u128,230	) -> DispatchResultWithPostInfo {231		ensure!(232			token == TokenId::default(),233			<Error<T>>::FungibleItemsHaveNoId234		);235236		with_weight(237			<Pallet<T>>::set_allowance(self, &sender, &spender, amount),238			<CommonWeights<T>>::approve(),239		)240	}241242	fn transfer_from(243		&self,244		sender: T::CrossAccountId,245		from: T::CrossAccountId,246		to: T::CrossAccountId,247		token: TokenId,248		amount: u128,249		nesting_budget: &dyn Budget,250	) -> DispatchResultWithPostInfo {251		ensure!(252			token == TokenId::default(),253			<Error<T>>::FungibleItemsHaveNoId254		);255256		with_weight(257			<Pallet<T>>::transfer_from(self, &sender, &from, &to, amount, nesting_budget),258			<CommonWeights<T>>::transfer_from(),259		)260	}261262	fn burn_from(263		&self,264		sender: T::CrossAccountId,265		from: T::CrossAccountId,266		token: TokenId,267		amount: u128,268		nesting_budget: &dyn Budget,269	) -> DispatchResultWithPostInfo {270		ensure!(271			token == TokenId::default(),272			<Error<T>>::FungibleItemsHaveNoId273		);274275		with_weight(276			<Pallet<T>>::burn_from(self, &sender, &from, amount, nesting_budget),277			<CommonWeights<T>>::burn_from(),278		)279	}280281	fn set_collection_properties(282		&self,283		_sender: T::CrossAccountId,284		_property: Vec<Property>,285	) -> DispatchResultWithPostInfo {286		fail!(<Error<T>>::SettingPropertiesNotAllowed)287	}288289	fn delete_collection_properties(290		&self,291		_sender: &T::CrossAccountId,292		_property_keys: Vec<PropertyKey>,293	) -> DispatchResultWithPostInfo {294		fail!(<Error<T>>::SettingPropertiesNotAllowed)295	}296297	fn set_token_properties(298		&self,299		_sender: T::CrossAccountId,300		_token_id: TokenId,301		_property: Vec<Property>,302		_nesting_budget: &dyn Budget,303	) -> DispatchResultWithPostInfo {304		fail!(<Error<T>>::SettingPropertiesNotAllowed)305	}306307	fn set_token_property_permissions(308		&self,309		_sender: &T::CrossAccountId,310		_property_permissions: Vec<PropertyKeyPermission>,311	) -> DispatchResultWithPostInfo {312		fail!(<Error<T>>::SettingPropertiesNotAllowed)313	}314315	fn delete_token_properties(316		&self,317		_sender: T::CrossAccountId,318		_token_id: TokenId,319		_property_keys: Vec<PropertyKey>,320		_nesting_budget: &dyn Budget,321	) -> DispatchResultWithPostInfo {322		fail!(<Error<T>>::SettingPropertiesNotAllowed)323	}324325	fn check_nesting(326		&self,327		_sender: <T>::CrossAccountId,328		_from: (CollectionId, TokenId),329		_under: TokenId,330		_nesting_budget: &dyn Budget,331	) -> sp_runtime::DispatchResult {332		fail!(<Error<T>>::FungibleDisallowsNesting)333	}334335	fn nest(&self, _under: TokenId, _to_nest: (CollectionId, TokenId)) {}336337	fn unnest(&self, _under: TokenId, _to_nest: (CollectionId, TokenId)) {}338339	fn collection_tokens(&self) -> Vec<TokenId> {340		vec![TokenId::default()]341	}342343	fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId> {344		if <Balance<T>>::get((self.id, account)) != 0 {345			vec![TokenId::default()]346		} else {347			vec![]348		}349	}350351	fn token_exists(&self, token: TokenId) -> bool {352		token == TokenId::default()353	}354355	fn last_token_id(&self) -> TokenId {356		TokenId::default()357	}358359	fn token_owner(&self, _token: TokenId) -> Option<T::CrossAccountId> {360		None361	}362363	fn token_property(&self, _token_id: TokenId, _key: &PropertyKey) -> Option<PropertyValue> {364		None365	}366367	fn token_properties(368		&self,369		_token_id: TokenId,370		_keys: Option<Vec<PropertyKey>>,371	) -> Vec<Property> {372		Vec::new()373	}374375	fn total_supply(&self) -> u32 {376		1377	}378379	fn account_balance(&self, account: T::CrossAccountId) -> u32 {380		if <Balance<T>>::get((self.id, account)) != 0 {381			1382		} else {383			0384		}385	}386387	fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128 {388		if token != TokenId::default() {389			return 0;390		}391		<Balance<T>>::get((self.id, account))392	}393394	fn allowance(395		&self,396		sender: T::CrossAccountId,397		spender: T::CrossAccountId,398		token: TokenId,399	) -> u128 {400		if token != TokenId::default() {401			return 0;402		}403		<Allowance<T>>::get((self.id, sender, spender))404	}405406	fn refungible_extensions(&self) -> Option<&dyn RefungibleExtensions<T>> {407		None408	}409410	fn total_pieces(&self, token: TokenId) -> Option<u128> {411		if token != TokenId::default() {412			return None;413		}414		<TotalSupply<T>>::try_get(self.id).ok()415	}416}