git.delta.rocks / unique-network / refs/commits / 34309ac789c2

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/>.1617pub mod sponsoring;1819use fp_evm::PrecompileResult;20use pallet_common::{21	CollectionById,22	erc::CommonEvmHandler,23	eth::{map_eth_to_id, map_eth_to_token_id},24};25use pallet_fungible::FungibleHandle;26use pallet_nonfungible::NonfungibleHandle;27use pallet_refungible::{RefungibleHandle, erc::RefungibleTokenHandle};28use sp_std::borrow::ToOwned;29use sp_std::vec::Vec;30use sp_core::{H160, U256};31use crate::{CollectionMode, Config, dispatch::Dispatched};32use pallet_common::CollectionHandle;3334pub struct UniqueErcSupport<T: Config>(core::marker::PhantomData<T>);3536impl<T: Config> pallet_evm::OnMethodCall<T> for UniqueErcSupport<T> {37	fn is_reserved(target: &H160) -> bool {38		map_eth_to_id(target).is_some()39	}40	fn is_used(target: &H160) -> bool {41		map_eth_to_id(target)42			.map(<CollectionById<T>>::contains_key)43			.unwrap_or(false)44	}45	fn get_code(target: &H160) -> Option<Vec<u8>> {46		if let Some(collection_id) = map_eth_to_id(target) {47			let collection = <CollectionById<T>>::get(collection_id)?;48			Some(49				match collection.mode {50					CollectionMode::NFT => <NonfungibleHandle<T>>::CODE,51					CollectionMode::Fungible(_) => <FungibleHandle<T>>::CODE,52					CollectionMode::ReFungible => <RefungibleHandle<T>>::CODE,53				}54				.to_owned(),55			)56		} else if let Some((collection_id, _token_id)) = map_eth_to_token_id(target) {57			let collection = <CollectionById<T>>::get(collection_id)?;58			if collection.mode != CollectionMode::ReFungible {59				return None;60			}61			// TODO: check token existence62			Some(<RefungibleTokenHandle<T>>::CODE.to_owned())63		} else {64			None65		}66	}67	fn call(68		source: &H160,69		target: &H160,70		gas_limit: u64,71		input: &[u8],72		value: U256,73	) -> Option<PrecompileResult> {74		if let Some(collection_id) = map_eth_to_id(target) {75			let collection = <CollectionHandle<T>>::new_with_gas_limit(collection_id, gas_limit)?;76			let dispatched = Dispatched::dispatch(collection);7778			match dispatched {79				Dispatched::Fungible(h) => h.call(source, input, value),80				Dispatched::Nonfungible(h) => h.call(source, input, value),81				Dispatched::Refungible(h) => h.call(source, input, value),82			}83		} else if let Some((collection_id, token_id)) = map_eth_to_token_id(target) {84			let collection = <CollectionHandle<T>>::new_with_gas_limit(collection_id, gas_limit)?;85			if collection.mode != CollectionMode::ReFungible {86				return None;87			}8889			let handle = RefungibleHandle::cast(collection);90			// TODO: check token existence91			RefungibleTokenHandle(handle, token_id).call(source, input, value)92		} else {93			None94		}95	}96}9798pub mod evm_collection {99	use core::marker::PhantomData;100	use evm_coder::{execution::*, generate_stubgen, solidity_interface, types::*, ToLog};101	use ethereum as _;102	use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};103	use pallet_evm::{OnMethodCall, PrecompileResult, account::CrossAccountId};104	use up_data_structs::{105		CreateCollectionData, MAX_COLLECTION_DESCRIPTION_LENGTH, MAX_TOKEN_PREFIX_LENGTH,106		MAX_COLLECTION_NAME_LENGTH,107	};108	use frame_support::traits::Get;109	use sp_core::H160;110	use pallet_common::{CollectionHandle, save_eth, pallet::CollectionById};111	112	use sp_std::{vec::Vec, rc::Rc};113	use alloc::format;114	115	pub trait Config:116		frame_system::Config117		+ pallet_evm_coder_substrate::Config118		+ pallet_evm::account::Config119		+ pallet_nonfungible::Config120	{121		type ContractAddress: Get<H160>;122	}123124	struct EvmCollectionHelper<T: Config>(Rc<SubstrateRecorder<T>>);125	impl<T: Config> WithRecorder<T> for EvmCollectionHelper<T> {126		fn recorder(&self) -> &SubstrateRecorder<T> {127			&self.0128		}129	130		fn into_recorder(self) -> Rc<SubstrateRecorder<T>> {131			self.0132		}133	}134135	#[solidity_interface(name = "CollectionHelper")]136	impl<T: Config> EvmCollectionHelper<T> {137		fn create_721_collection(138			&self,139			caller: caller,140			name: string,141			description: string,142			token_prefix: string,143		) -> Result<address> {144			let caller = T::CrossAccountId::from_eth(caller);145			let name = name146				.encode_utf16()147				.collect::<Vec<u16>>()148				.try_into()149				.map_err(|_| error_feild_too_long(stringify!(name), MAX_COLLECTION_NAME_LENGTH))?;150			let description = description151				.encode_utf16()152				.collect::<Vec<u16>>()153				.try_into()154				.map_err(|_| {155					error_feild_too_long(stringify!(description), MAX_COLLECTION_DESCRIPTION_LENGTH)156				})?;157			let token_prefix = token_prefix158				.into_bytes()159				.try_into()160				.map_err(|_| error_feild_too_long(stringify!(token_prefix), MAX_TOKEN_PREFIX_LENGTH))?;161	162			let data = CreateCollectionData {163				name,164				description,165				token_prefix,166				..Default::default()167			};168	169			let collection_id =170				<pallet_nonfungible::Pallet<T>>::init_collection(caller.as_sub().clone(), data)171					.map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;172	173			let address = pallet_common::eth::collection_id_to_address(collection_id);174			self.0.log_mirrored(EthCollectionEvent::CollectionCreated {175				owner: *caller.as_eth(),176				collection_id: address,177			});178			Ok(address)179		}180181		fn is_collection_exist(&self, _caller: caller, collection_address: address) -> Result<bool> {182			if let Some(id) = pallet_common::eth::map_eth_to_id(&collection_address) {183				let collection_id = id;184				return Ok(<CollectionById<T>>::contains_key(collection_id));185			}186187			Ok(false)188		}189	}190	191	struct EvmCollection<T: Config>(Rc<SubstrateRecorder<T>>);192	impl<T: Config> WithRecorder<T> for EvmCollection<T> {193		fn recorder(&self) -> &SubstrateRecorder<T> {194			&self.0195		}196	197		fn into_recorder(self) -> Rc<SubstrateRecorder<T>> {198			self.0199		}200	}201	202	#[derive(ToLog)]203	pub enum EthCollectionEvent {204		CollectionCreated {205			#[indexed]206			owner: address,207			#[indexed]208			collection_id: address,209		},210	}211	212	#[solidity_interface(name = "Collection")]213	impl<T: Config> EvmCollection<T> {214		fn set_sponsor(215			&self,216			caller: caller,217			sponsor: address,218		) -> Result<void> {219			let mut collection = collection_from_address(self.contract_address(caller).unwrap(), &self.0)?;220			check_is_owner(caller, &collection)?;221	222			let sponsor = T::CrossAccountId::from_eth(sponsor);223			collection.set_sponsor(sponsor.as_sub().clone());224			save_eth(collection)225		}226	227		fn confirm_sponsorship(&self, caller: caller) -> Result<void> {228			let mut collection = collection_from_address(self.contract_address(caller).unwrap(), &self.0)?;229			let caller = T::CrossAccountId::from_eth(caller);230			if !collection.confirm_sponsorship(caller.as_sub()) {231				return Err(Error::Revert("Caller is not set as sponsor".into()));232			}233			save_eth(collection)234		}235	236		fn set_limits(237			&self,238			caller: caller,239			limits_json: string,240		) -> Result<void> {241			let mut collection = collection_from_address(self.contract_address(caller).unwrap(), &self.0)?;242			check_is_owner(caller, &collection)?;243	244			let limits = serde_json_core::from_str(limits_json.as_ref())245				.map_err(|e| Error::Revert(format!("Parse JSON error: {}", e)))?;246			collection.limits = limits.0;247			save_eth(collection)248		}249250		fn contract_address(&self, _caller: caller) -> Result<address> {251			Ok(self.0.contract())252		}253	}254	255	fn error_feild_too_long(feild: &str, bound: u32) -> Error {256		Error::Revert(format!("{} is too long. Max length is {}.", feild, bound))257	}258	259	fn collection_from_address<T: Config>(260		collection_address: address,261		recorder: &Rc<SubstrateRecorder<T>>,262	) -> Result<CollectionHandle<T>> {263		let collection_id = pallet_common::eth::map_eth_to_id(&collection_address)264			.ok_or(Error::Revert("Contract is not an unique collection".into()))?;265		let collection =266			pallet_common::CollectionHandle::new_with_recorder(collection_id, recorder.clone())267				.ok_or(Error::Revert("Create collection handle error".into()))?;268		Ok(collection)269	}270	271	fn check_is_owner<T: Config>(caller: caller, collection: &CollectionHandle<T>) -> Result<()> {272		let caller = T::CrossAccountId::from_eth(caller);273		collection274			.check_is_owner(&caller)275			.map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;276		Ok(())277	}278279	pub struct CollectionHelperOnMethodCall<T: Config>(PhantomData<*const T>);280	impl<T: Config> OnMethodCall<T> for CollectionHelperOnMethodCall<T> {281		fn is_reserved(contract: &sp_core::H160) -> bool {282			contract == &T::ContractAddress::get()283		}284	285		fn is_used(contract: &sp_core::H160) -> bool {286			contract == &T::ContractAddress::get()287		}288	289		fn call(290			source: &sp_core::H160,291			target: &sp_core::H160,292			gas_left: u64,293			input: &[u8],294			value: sp_core::U256,295		) -> Option<PrecompileResult> {296			if target != &T::ContractAddress::get() {297				return None;298			}299	300			let helpers = EvmCollectionHelper::<T>(Rc::new(SubstrateRecorder::<T>::new(*target, gas_left)));301			pallet_evm_coder_substrate::call(*source, helpers, value, input)302		}303	304		fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {305			(contract == &T::ContractAddress::get())306				.then(|| include_bytes!("./stubs/CollectionHelper.raw").to_vec())307		}308	}309	310	generate_stubgen!(collection_helper_impl, CollectionHelperCall<()>, true);311	generate_stubgen!(collection_helper_iface, CollectionHelperCall<()>, false);312	313	pub struct CollectionOnMethodCall<T: Config>(PhantomData<*const T>);314	impl<T: Config> OnMethodCall<T> for CollectionOnMethodCall<T> {315		fn is_reserved(contract: &sp_core::H160) -> bool {316			contract == &T::ContractAddress::get()317		}318	319		fn is_used(contract: &sp_core::H160) -> bool {320			contract == &T::ContractAddress::get()321		}322	323		fn call(324			source: &sp_core::H160,325			target: &sp_core::H160,326			gas_left: u64,327			input: &[u8],328			value: sp_core::U256,329		) -> Option<PrecompileResult> {330			if !pallet_common::eth::is_collection(target) {331				return None;332			}333334			let helpers = EvmCollection::<T>(Rc::new(SubstrateRecorder::<T>::new(*target, gas_left)));335			pallet_evm_coder_substrate::call(*source, helpers, value, input)336		}337	338		fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {339			(contract == &T::ContractAddress::get())340				.then(|| include_bytes!("./stubs/Collection.raw").to_vec())341		}342	}343	344	generate_stubgen!(collection_impl, CollectionCall<()>, true);345	generate_stubgen!(collection_iface, CollectionCall<()>, false);346}