`088ceb16caa8`

source

pallets/unique/src/eth/mod.rs11.0 KiBsource history
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/>.1617//! Implementation of CollectionHelpers contract.1819use core::marker::PhantomData;20use ethereum as _;21use evm_coder::{execution::*, generate_stubgen, solidity_interface, solidity, weight, types::*};22use frame_support::traits::Get;23use pallet_common::{24	CollectionById,25	dispatch::CollectionDispatch,26	erc::{27		CollectionHelpersEvents,28		static_property::{key},29	},30	Pallet as PalletCommon,31};32use pallet_evm_coder_substrate::{dispatch_to_evm, SubstrateRecorder, WithRecorder};33use pallet_evm::{account::CrossAccountId, OnMethodCall, PrecompileHandle, PrecompileResult};34use sp_std::vec;35use up_data_structs::{36	CollectionName, CollectionDescription, CollectionTokenPrefix, CreateCollectionData,37	CollectionMode, PropertyValue, CollectionFlags,38};3940use crate::{Config, SelfWeightOf, weights::WeightInfo};4142use sp_std::vec::Vec;43use alloc::format;4445/// See [`CollectionHelpersCall`]46pub struct EvmCollectionHelpers<T: Config>(SubstrateRecorder<T>);47impl<T: Config> WithRecorder<T> for EvmCollectionHelpers<T> {48	fn recorder(&self) -> &SubstrateRecorder<T> {49		&self.050	}5152	fn into_recorder(self) -> SubstrateRecorder<T> {53		self.054	}55}5657fn convert_data<T: Config>(58	caller: caller,59	name: string,60	description: string,61	token_prefix: string,62) -> Result<(63	T::CrossAccountId,64	CollectionName,65	CollectionDescription,66	CollectionTokenPrefix,67)> {68	let caller = T::CrossAccountId::from_eth(caller);69	let name = name70		.encode_utf16()71		.collect::<Vec<u16>>()72		.try_into()73		.map_err(|_| error_field_too_long(stringify!(name), CollectionName::bound()))?;74	let description = description75		.encode_utf16()76		.collect::<Vec<u16>>()77		.try_into()78		.map_err(|_| {79			error_field_too_long(stringify!(description), CollectionDescription::bound())80		})?;81	let token_prefix = token_prefix.into_bytes().try_into().map_err(|_| {82		error_field_too_long(stringify!(token_prefix), CollectionTokenPrefix::bound())83	})?;84	Ok((caller, name, description, token_prefix))85}8687fn create_refungible_collection_internal<88	T: Config + pallet_nonfungible::Config + pallet_refungible::Config,89>(90	caller: caller,91	value: value,92	name: string,93	description: string,94	token_prefix: string,95) -> Result<address> {96	let (caller, name, description, token_prefix) =97		convert_data::<T>(caller, name, description, token_prefix)?;98	let data = CreateCollectionData {99		name,100		mode: CollectionMode::ReFungible,101		description,102		token_prefix,103		..Default::default()104	};105	check_sent_amount_equals_collection_creation_price::<T>(value)?;106	let collection_helpers_address =107		T::CrossAccountId::from_eth(<T as pallet_common::Config>::ContractAddress::get());108109	let collection_id = T::CollectionDispatch::create(110		caller.clone(),111		collection_helpers_address,112		data,113		Default::default(),114	)115	.map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;116	let address = pallet_common::eth::collection_id_to_address(collection_id);117	Ok(address)118}119120fn check_sent_amount_equals_collection_creation_price<T: Config>(value: value) -> Result<()> {121	let value = value.as_u128();122	let creation_price: u128 = T::CollectionCreationPrice::get()123		.try_into()124		.map_err(|_| ()) // workaround for `expect` requiring `Debug` trait125		.expect("Collection creation price should be convertible to u128");126	if value != creation_price {127		return Err(format!(128			"Sent amount not equals to collection creation price ({0})",129			creation_price130		)131		.into());132	}133	Ok(())134}135136/// @title Contract, which allows users to operate with collections137#[solidity_interface(name = CollectionHelpers, events(CollectionHelpersEvents))]138impl<T> EvmCollectionHelpers<T>139where140	T: Config + pallet_common::Config + pallet_nonfungible::Config + pallet_refungible::Config,141{142	/// Create an NFT collection143	/// @param name Name of the collection144	/// @param description Informative description of the collection145	/// @param tokenPrefix Token prefix to represent the collection tokens in UI and user applications146	/// @return address Address of the newly created collection147	#[weight(<SelfWeightOf<T>>::create_collection())]148	#[solidity(rename_selector = "createNFTCollection")]149	fn create_nft_collection(150		&mut self,151		caller: caller,152		value: value,153		name: string,154		description: string,155		token_prefix: string,156	) -> Result<address> {157		let (caller, name, description, token_prefix) =158			convert_data::<T>(caller, name, description, token_prefix)?;159		let data = CreateCollectionData {160			name,161			mode: CollectionMode::NFT,162			description,163			token_prefix,164			..Default::default()165		};166		check_sent_amount_equals_collection_creation_price::<T>(value)?;167		let collection_helpers_address =168			T::CrossAccountId::from_eth(<T as pallet_common::Config>::ContractAddress::get());169		let collection_id = T::CollectionDispatch::create(170			caller,171			collection_helpers_address,172			data,173			Default::default(),174		)175		.map_err(dispatch_to_evm::<T>)?;176177		let address = pallet_common::eth::collection_id_to_address(collection_id);178		Ok(address)179	}180	/// Create an NFT collection181	/// @param name Name of the collection182	/// @param description Informative description of the collection183	/// @param tokenPrefix Token prefix to represent the collection tokens in UI and user applications184	/// @return address Address of the newly created collection185	#[weight(<SelfWeightOf<T>>::create_collection())]186	#[deprecated(note = "mathod was renamed to `create_nft_collection`, prefer it instead")]187	fn create_nonfungible_collection(188		&mut self,189		caller: caller,190		value: value,191		name: string,192		description: string,193		token_prefix: string,194	) -> Result<address> {195		self.create_nft_collection(caller, value, name, description, token_prefix)196	}197198	#[weight(<SelfWeightOf<T>>::create_collection())]199	#[solidity(rename_selector = "createRFTCollection")]200	fn create_rft_collection(201		&mut self,202		caller: caller,203		value: value,204		name: string,205		description: string,206		token_prefix: string,207	) -> Result<address> {208		create_refungible_collection_internal::<T>(caller, value, name, description, token_prefix)209	}210211	#[solidity(rename_selector = "makeCollectionERC721MetadataCompatible")]212	fn make_collection_metadata_compatible(213		&mut self,214		caller: caller,215		collection: address,216		base_uri: string,217	) -> Result<()> {218		let caller = T::CrossAccountId::from_eth(caller);219		let collection =220			pallet_common::eth::map_eth_to_id(&collection).ok_or("not a collection address")?;221		let mut collection =222			<crate::CollectionHandle<T>>::new(collection).ok_or("collection not found")?;223224		if !matches!(225			collection.mode,226			CollectionMode::NFT | CollectionMode::ReFungible227		) {228			return Err("target collection should be either NFT or Refungible".into());229		}230231		self.recorder().consume_sstore()?;232		collection233			.check_is_owner_or_admin(&caller)234			.map_err(dispatch_to_evm::<T>)?;235236		if collection.flags.erc721metadata {237			return Err("target collection is already Erc721Metadata compatible".into());238		}239		collection.flags.erc721metadata = true;240241		let all_permissions = <pallet_common::CollectionPropertyPermissions<T>>::get(collection.id);242		if all_permissions.get(&key::url()).is_none() {243			self.recorder().consume_sstore()?;244			<PalletCommon<T>>::set_property_permission(245				&collection,246				&caller,247				up_data_structs::PropertyKeyPermission {248					key: key::url(),249					permission: up_data_structs::PropertyPermission {250						mutable: true,251						collection_admin: true,252						token_owner: false,253					},254				},255			)256			.map_err(dispatch_to_evm::<T>)?;257		}258		if all_permissions.get(&key::suffix()).is_none() {259			self.recorder().consume_sstore()?;260			<PalletCommon<T>>::set_property_permission(261				&collection,262				&caller,263				up_data_structs::PropertyKeyPermission {264					key: key::suffix(),265					permission: up_data_structs::PropertyPermission {266						mutable: true,267						collection_admin: true,268						token_owner: false,269					},270				},271			)272			.map_err(dispatch_to_evm::<T>)?;273		}274275		let all_properties = <pallet_common::CollectionProperties<T>>::get(collection.id);276		if all_properties.get(&key::base_uri()).is_none() && !base_uri.is_empty() {277			self.recorder().consume_sstore()?;278			<PalletCommon<T>>::set_collection_properties(279				&collection,280				&caller,281				vec![up_data_structs::Property {282					key: key::base_uri(),283					value: base_uri284						.into_bytes()285						.try_into()286						.map_err(|_| "base uri is too large")?,287				}],288			)289			.map_err(dispatch_to_evm::<T>)?;290		}291292		self.recorder().consume_sstore()?;293		collection.save().map_err(dispatch_to_evm::<T>)?;294295		Ok(())296	}297298	/// Check if a collection exists299	/// @param collectionAddress Address of the collection in question300	/// @return bool Does the collection exist?301	fn is_collection_exist(&self, _caller: caller, collection_address: address) -> Result<bool> {302		if let Some(id) = pallet_common::eth::map_eth_to_id(&collection_address) {303			let collection_id = id;304			return Ok(<CollectionById<T>>::contains_key(collection_id));305		}306307		Ok(false)308	}309310	fn collection_creation_fee(&self) -> Result<value> {311		let price: u128 = T::CollectionCreationPrice::get()312			.try_into()313			.map_err(|_| ()) // workaround for `expect` requiring `Debug` trait314			.expect("Collection creation price should be convertible to u128");315		Ok(price.into())316	}317}318319/// Implements [`OnMethodCall`], which delegates call to [`EvmCollectionHelpers`]320pub struct CollectionHelpersOnMethodCall<T: Config>(PhantomData<*const T>);321impl<T: Config + pallet_nonfungible::Config + pallet_refungible::Config> OnMethodCall<T>322	for CollectionHelpersOnMethodCall<T>323{324	fn is_reserved(contract: &sp_core::H160) -> bool {325		contract == &T::ContractAddress::get()326	}327328	fn is_used(contract: &sp_core::H160) -> bool {329		contract == &T::ContractAddress::get()330	}331332	fn call(handle: &mut impl PrecompileHandle) -> Option<PrecompileResult> {333		if handle.code_address() != T::ContractAddress::get() {334			return None;335		}336337		let helpers =338			EvmCollectionHelpers::<T>(SubstrateRecorder::<T>::new(handle.remaining_gas()));339		pallet_evm_coder_substrate::call(handle, helpers)340	}341342	fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {343		(contract == &T::ContractAddress::get())344			.then(|| include_bytes!("./stubs/CollectionHelpers.raw").to_vec())345	}346}347348generate_stubgen!(collection_helper_impl, CollectionHelpersCall<()>, true);349generate_stubgen!(collection_helper_iface, CollectionHelpersCall<()>, false);350351fn error_field_too_long(feild: &str, bound: usize) -> Error {352	Error::Revert(format!("{} is too long. Max length is {}.", feild, bound))353}