1234567891011121314151617use core::marker::PhantomData;18use evm_coder::{abi::AbiWriter, execution::*, generate_stubgen, solidity_interface, types::*, ToLog};19use ethereum as _;20use pallet_common::CollectionById;21use pallet_common::{CollectionHandle};22use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};23use pallet_evm::{24 ExitRevert, OnCreate, OnMethodCall, PrecompileResult, PrecompileFailure,25 account::CrossAccountId, Pallet as PalletEvm,26};27use sp_core::H160;28use up_data_structs::{29 CreateCollectionData, MAX_COLLECTION_DESCRIPTION_LENGTH, MAX_TOKEN_PREFIX_LENGTH,30 MAX_COLLECTION_NAME_LENGTH, OFFCHAIN_SCHEMA_LIMIT, VARIABLE_ON_CHAIN_SCHEMA_LIMIT,31 CONST_ON_CHAIN_SCHEMA_LIMIT,32};33use crate::{Config, Pallet};34use frame_support::traits::Get;3536use sp_std::{vec::Vec, rc::Rc};37use alloc::format;3839struct EvmCollection<T: Config>(SubstrateRecorder<T>);40impl<T: Config> WithRecorder<T> for EvmCollection<T> {41 fn recorder(&self) -> &SubstrateRecorder<T> {42 &self.043 }4445 fn into_recorder(self) -> SubstrateRecorder<T> {46 self.047 }48}4950#[derive(ToLog)]51pub enum CollectionEvent {52 CollectionCreated {53 #[indexed]54 owner: address,55 #[indexed]56 collection_id: address,57 },58}5960#[solidity_interface(name = "Collection")]61impl<T: Config> EvmCollection<T> {62 fn create_721_collection(63 &self,64 caller: caller,65 name: string,66 description: string,67 token_prefix: string,68 ) -> Result<address> {69 let caller = T::CrossAccountId::from_eth(caller);70 let name = name71 .encode_utf16()72 .collect::<Vec<u16>>()73 .try_into()74 .map_err(|_| error_feild_too_long("name", MAX_COLLECTION_NAME_LENGTH))?;75 let description = description76 .encode_utf16()77 .collect::<Vec<u16>>()78 .try_into()79 .map_err(|_| error_feild_too_long("description", MAX_COLLECTION_DESCRIPTION_LENGTH))?;80 let token_prefix = token_prefix81 .into_bytes()82 .try_into()83 .map_err(|_| error_feild_too_long("token_prefix", MAX_TOKEN_PREFIX_LENGTH))?;8485 let data = CreateCollectionData {86 name,87 description,88 token_prefix,89 ..Default::default()90 };9192 let collection_id =93 <pallet_nonfungible::Pallet<T>>::init_collection(caller.as_sub().clone(), data)94 .map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;9596 let address = pallet_common::eth::collection_id_to_address(collection_id);97 <PalletEvm<T>>::deposit_log(98 CollectionEvent::CollectionCreated {99 owner: *caller.as_eth(),100 collection_id: address,101 }102 .to_log(address),103 );104 Ok(address)105 }106107 fn set_sponsor(108 &self,109 caller: caller,110 collection_address: address,111 sponsor: address,112 ) -> Result<void> {113 let mut collection = collection_from_address(collection_address, &self.0)?;114 check_is_owner(caller, &collection)?;115116 let sponsor = T::CrossAccountId::from_eth(sponsor);117 collection.set_sponsor(sponsor.as_sub().clone());118 collection.save().map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;119 Ok(()).map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;120 Ok(())121 }122123 fn confirm_sponsorship(&self, caller: caller, collection_address: address) -> Result<void> {124 let mut collection = collection_from_address(collection_address, &self.0)?;125 let caller = T::CrossAccountId::from_eth(caller);126 if !collection.confirm_sponsorship(caller.as_sub()) {127 return Err(Error::Revert("Caller is not set as sponsor".into()));128 }129 collection.save().map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;130 Ok(())131 }132133 fn set_offchain_schema(134 &self,135 caller: caller,136 collection_address: address,137 schema: string,138 ) -> Result<void> {139 let mut collection = collection_from_address(collection_address, &self.0)?;140 check_is_owner(caller, &collection)?;141142 let schema = schema143 .into_bytes()144 .try_into()145 .map_err(|_| error_feild_too_long(stringify!(shema), OFFCHAIN_SCHEMA_LIMIT))?;146 147 collection.save().map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;148 Ok(())149 }150151 fn set_variable_on_chain_schema(152 &self,153 caller: caller,154 collection_address: address,155 variable: string,156 ) -> Result<void> {157 let mut collection = collection_from_address(collection_address, &self.0)?;158 check_is_owner(caller, &collection)?;159160 let variable = variable.into_bytes().try_into().map_err(|_| {161 error_feild_too_long(stringify!(variable), VARIABLE_ON_CHAIN_SCHEMA_LIMIT)162 })?;163 164 collection.save().map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;165 Ok(())166 }167168 fn set_const_on_chain_schema(169 &self,170 caller: caller,171 collection_address: address,172 const_on_chain: string,173 ) -> Result<void> {174 let mut collection = collection_from_address(collection_address, &self.0)?;175 check_is_owner(caller, &collection)?;176177 let const_on_chain = const_on_chain.into_bytes().try_into().map_err(|_| {178 error_feild_too_long(stringify!(const_on_chain), CONST_ON_CHAIN_SCHEMA_LIMIT)179 })?;180 181 collection.save().map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;182 Ok(())183 }184185 fn set_limits(186 &self,187 caller: caller,188 collection_address: address,189 limits_json: string,190 ) -> Result<void> {191 let mut collection = collection_from_address(collection_address, &self.0)?;192 check_is_owner(caller, &collection)?;193194 let limits = serde_json_core::from_str(limits_json.as_ref())195 .map_err(|e| Error::Revert(format!("Parse JSON error: {}", e)))?;196 collection.limits = limits.0;197 collection.save().map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;198 Ok(())199 }200}201202fn error_feild_too_long(feild: &str, bound: u32) -> Error {203 Error::Revert(format!("{} is too long. Max length is {}.", feild, bound))204}205206fn collection_from_address<T: Config>(207 collection_address: address,208 recorder: &SubstrateRecorder<T>,209) -> Result<CollectionHandle<T>> {210 let collection_id = pallet_common::eth::map_eth_to_id(&collection_address)211 .ok_or(Error::Revert("Contract is not an unique collection".into()))?;212 let collection =213 pallet_common::CollectionHandle::new_with_gas_limit(collection_id, recorder.gas_left())214 .ok_or(Error::Revert("Create collection handle error".into()))?;215 Ok(collection)216}217218fn check_is_owner<T: Config>(caller: caller, collection: &CollectionHandle<T>) -> Result<()> {219 let caller = T::CrossAccountId::from_eth(caller);220 collection221 .check_is_owner(&caller)222 .map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;223 Ok(())224}225226pub struct CollectionOnMethodCall<T: Config>(PhantomData<*const T>);227impl<T: Config> OnMethodCall<T> for CollectionOnMethodCall<T> {228 fn is_reserved(contract: &sp_core::H160) -> bool {229 contract == &T::ContractAddress::get()230 }231232 fn is_used(contract: &sp_core::H160) -> bool {233 contract == &T::ContractAddress::get()234 }235236 fn call(237 source: &sp_core::H160,238 target: &sp_core::H160,239 gas_left: u64,240 input: &[u8],241 value: sp_core::U256,242 ) -> Option<PrecompileResult> {243 if target != &T::ContractAddress::get() {244 return None;245 }246247 let helpers = EvmCollection::<T>(SubstrateRecorder::new(gas_left));248 pallet_evm_coder_substrate::call(*source, helpers, value, input)249 }250251 fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {252 (contract == &T::ContractAddress::get())253 .then(|| include_bytes!("./stubs/Collection.raw").to_vec())254 }255}256257generate_stubgen!(collection_impl, CollectionCall<()>, true);258generate_stubgen!(collection_iface, CollectionCall<()>, false);