git.delta.rocks / unique-network / refs/commits / 31962dd005a3

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;18use evm_coder::{abi::AbiWriter, execution::*, generate_stubgen, solidity_interface, types::*, ToLog};19use ethereum as _;20use pallet_common::CollectionById;21use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};22use pallet_evm::{23	ExitRevert, OnCreate, OnMethodCall, PrecompileResult, PrecompileFailure,24	account::CrossAccountId, Pallet as PalletEvm,25};26use sp_core::H160;27use up_data_structs::{28	CreateCollectionData, MAX_COLLECTION_DESCRIPTION_LENGTH, MAX_TOKEN_PREFIX_LENGTH,29	MAX_COLLECTION_NAME_LENGTH,30};31use crate::{Config, Pallet};32use frame_support::traits::Get;3334use sp_std::{vec::Vec, rc::Rc};35use alloc::format;3637struct EvmCollection<T: Config>(SubstrateRecorder<T>);38impl<T: Config> WithRecorder<T> for EvmCollection<T> {39	fn recorder(&self) -> &SubstrateRecorder<T> {40		&self.041	}4243	fn into_recorder(self) -> SubstrateRecorder<T> {44		self.045	}46}4748#[derive(ToLog)]49pub enum CollectionEvent {50	CollectionCreated {51		#[indexed]52		owner: address,53		#[indexed]54		collection_id: address,55	},56}5758#[solidity_interface(name = "Collection")]59impl<T: Config> EvmCollection<T> {60	fn create_721_collection(61		&self,62		caller: caller,63		name: string,64		description: string,65		token_prefix: string,66	) -> Result<address> {67		let caller = T::CrossAccountId::from_eth(caller);68		let name = name69			.encode_utf16()70			.collect::<Vec<u16>>()71			.try_into()72			.map_err(|_| error_feild_too_long("name", MAX_COLLECTION_NAME_LENGTH))?;73		let description = description74			.encode_utf16()75			.collect::<Vec<u16>>()76			.try_into()77			.map_err(|_| error_feild_too_long("description", MAX_COLLECTION_DESCRIPTION_LENGTH))?;78		let token_prefix = token_prefix79			.into_bytes()80			.try_into()81			.map_err(|_| error_feild_too_long("token_prefix", MAX_TOKEN_PREFIX_LENGTH))?;8283		let data = CreateCollectionData {84			name,85			description,86			token_prefix,87			..Default::default()88		};8990		let collection_id =91			<pallet_nonfungible::Pallet<T>>::init_collection(caller.as_sub().clone(), data)92				.map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;9394		let address = pallet_common::eth::collection_id_to_address(collection_id);95		<PalletEvm<T>>::deposit_log(96			CollectionEvent::CollectionCreated {97				owner: *caller.as_eth(),98				collection_id: address,99			}100			.to_log(address),101		);102		Ok(address)103	}104105	// fn set_sponsor(collection_id: address, sponsor: address) -> Result<void> {106	// 	let collection_id =107	// 		pallet_common::eth::map_eth_to_id(&collection_id).ok_or(Error::Revert("".into()))?;108	// 	let mut collection = <CollectionById<T>>::get(collection_id).ok_or(Error::Revert("".into()))?;109	// 	let sponsor = T::CrossAccountId::from_eth(sponsor);110	// 	collection.sponsorship = SponsorshipState::Unconfirmed(sponsor.as_sub().clone());111	// 	<CollectionById<T>>::insert(collection_id, collection);112	// 	Ok(())113	// }114115	fn set_offchain_shema(shema: string) -> Result<void> {116		let shema = shema117			.into_bytes()118			.try_into()119			.map_err(|_| error_feild_too_long(stringify!(shema), OFFCHAIN_SCHEMA_LIMIT))?;120		collection.offchain_schema = shema;121		save(collection)122	}123124	fn confirm_sponsorship(&self, caller: caller, collection_address: address) -> Result<void> {125		let (_, mut collection) = collection_from_address(collection_address, &self.0)?;126		let caller = T::CrossAccountId::from_eth(caller);127		if !collection.confirm_sponsorship(caller.as_sub()) {128			return Err(Error::Revert("Caller is not set as sponsor".into()));129		}130		save(collection)131	}132133	// fn set_variable_on_chain_schema(shema: string) -> Result<void> {134	// 	Ok(())135	// }136137	fn set_variable_on_chain_schema(138		&self,139		caller: caller,140		collection_address: address,141		variable: string,142	) -> Result<void> {143		let (_, mut collection) = collection_from_address(collection_address, &self.0)?;144		check_is_owner(caller, &collection)?;145146		let variable = variable.into_bytes().try_into().map_err(|_| {147			error_feild_too_long(stringify!(variable), VARIABLE_ON_CHAIN_SCHEMA_LIMIT)148		})?;149		collection.variable_on_chain_schema = variable;150		save(collection)151	}152153	fn set_const_on_chain_schema(154		&self,155		caller: caller,156		collection_address: address,157		const_on_chain: string,158	) -> Result<void> {159		let (_, mut collection) = collection_from_address(collection_address, &self.0)?;160		check_is_owner(caller, &collection)?;161162		let const_on_chain = const_on_chain.into_bytes().try_into().map_err(|_| {163			error_feild_too_long(stringify!(const_on_chain), CONST_ON_CHAIN_SCHEMA_LIMIT)164		})?;165		collection.const_on_chain_schema = const_on_chain;166		save(collection)167	}168169	fn set_limits(170		&self,171		caller: caller,172		collection_address: address,173		limits_json: string,174	) -> Result<void> {175		let (_, mut collection) = collection_from_address(collection_address, &self.0)?;176		check_is_owner(caller, &collection)?;177178		let limits = serde_json::from_str(limits_json.as_ref())179			.map_err(|e| Error::Revert(format!("Parse JSON error: {}", e)))?;180		collection.limits = limits;181		save(collection)182	}183}184185fn error_feild_too_long(feild: &str, bound: u32) -> Error {186	Error::Revert(format!("{} is too long. Max length is {}.", feild, bound))187}188189pub struct CollectionOnMethodCall<T: Config>(PhantomData<*const T>);190impl<T: Config> OnMethodCall<T> for CollectionOnMethodCall<T> {191	fn is_reserved(contract: &sp_core::H160) -> bool {192		contract == &T::ContractAddress::get()193	}194195	fn is_used(contract: &sp_core::H160) -> bool {196		contract == &T::ContractAddress::get()197	}198199	fn call(200		source: &sp_core::H160,201		target: &sp_core::H160,202		gas_left: u64,203		input: &[u8],204		value: sp_core::U256,205	) -> Option<PrecompileResult> {206		// TODO: Extract to another OnMethodCall handler207		if target != &T::ContractAddress::get() {208			return None;209		}210211		let helpers = EvmCollection::<T>(SubstrateRecorder::new(gas_left));212		pallet_evm_coder_substrate::call(*source, helpers, value, input)213	}214215	fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {216		(contract == &T::ContractAddress::get())217			.then(|| include_bytes!("./stubs/Collection.raw").to_vec())218	}219}220221generate_stubgen!(collection_impl, CollectionCall<()>, true);222generate_stubgen!(collection_iface, CollectionCall<()>, false);