git.delta.rocks / unique-network / refs/commits / 469576010118

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 evm_coder::{18	solidity_interface, solidity, ToLog,19	types::*,20	execution::{Result, Error},21};22pub use pallet_evm::{PrecompileOutput, PrecompileResult, PrecompileHandle, account::CrossAccountId};23use pallet_evm_coder_substrate::dispatch_to_evm;24use sp_core::{H160, U256, H256};25use sp_std::vec::Vec;26use up_data_structs::{Property, SponsoringRateLimit, NestingRule, OwnerRestrictedSet, AccessMode};27use alloc::format;2829use crate::{Pallet, CollectionHandle, Config, CollectionProperties};3031#[derive(ToLog)]32pub enum CollectionHelpersEvents {33	CollectionCreated {34		#[indexed]35		owner: address,36		#[indexed]37		collection_id: address,38	},39}4041/// Does not always represent a full collection, for RFT it is either42/// collection (Implementing ERC721), or specific collection token (Implementing ERC20)43pub trait CommonEvmHandler {44	const CODE: &'static [u8];4546	fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult>;47}4849#[solidity_interface(name = "Collection")]50impl<T: Config> CollectionHandle<T>51// where52// 	T::AccountId: From<H256>53{54	fn set_collection_property(&mut self, caller: caller, key: string, value: bytes) -> Result<()> {55		let caller = T::CrossAccountId::from_eth(caller);56		let key = <Vec<u8>>::from(key)57			.try_into()58			.map_err(|_| "key too large")?;59		let value = value.try_into().map_err(|_| "value too large")?;6061		<Pallet<T>>::set_collection_property(self, &caller, Property { key, value })62			.map_err(dispatch_to_evm::<T>)63	}6465	fn delete_collection_property(&mut self, caller: caller, key: string) -> Result<()> {66		let caller = T::CrossAccountId::from_eth(caller);67		let key = <Vec<u8>>::from(key)68			.try_into()69			.map_err(|_| "key too large")?;7071		<Pallet<T>>::delete_collection_property(self, &caller, key).map_err(dispatch_to_evm::<T>)72	}7374	/// Throws error if key not found75	fn collection_property(&self, key: string) -> Result<bytes> {76		let key = <Vec<u8>>::from(key)77			.try_into()78			.map_err(|_| "key too large")?;7980		let props = <CollectionProperties<T>>::get(self.id);81		let prop = props.get(&key).ok_or("key not found")?;8283		Ok(prop.to_vec())84	}8586	fn set_collection_sponsor(&mut self, caller: caller, sponsor: address) -> Result<void> {87		check_is_owner(caller, self)?;8889		let sponsor = T::CrossAccountId::from_eth(sponsor);90		self.set_sponsor(sponsor.as_sub().clone()).map_err(dispatch_to_evm::<T>)?;91		save(self)92	}9394	fn confirm_collection_sponsorship(&mut self, caller: caller) -> Result<void> {95		let caller = T::CrossAccountId::from_eth(caller);96		if !self.confirm_sponsorship(caller.as_sub()).map_err(dispatch_to_evm::<T>)? {97			return Err(Error::Revert("Caller is not set as sponsor".into()));98		}99		save(self)100	}101102	#[solidity(rename_selector = "setCollectionLimit")]103	fn set_int_limit(&mut self, caller: caller, limit: string, value: uint32) -> Result<void> {104		check_is_owner(caller, self)?;105		let mut limits = self.limits.clone();106107		match limit.as_str() {108			"accountTokenOwnershipLimit" => {109				limits.account_token_ownership_limit = Some(value);110			}111			"sponsoredDataSize" => {112				limits.sponsored_data_size = Some(value);113			}114			"sponsoredDataRateLimit" => {115				limits.sponsored_data_rate_limit = Some(SponsoringRateLimit::Blocks(value));116			}117			"tokenLimit" => {118				limits.token_limit = Some(value);119			}120			"sponsorTransferTimeout" => {121				limits.sponsor_transfer_timeout = Some(value);122			}123			"sponsorApproveTimeout" => {124				limits.sponsor_approve_timeout = Some(value);125			}126			_ => {127				return Err(Error::Revert(format!(128					"Unknown integer limit \"{}\"",129					limit130				)))131			}132		}133		self.limits = <Pallet<T>>::clamp_limits(self.mode.clone(), &self.limits, limits)134			.map_err(dispatch_to_evm::<T>)?;135		save(self)136	}137138	#[solidity(rename_selector = "setCollectionLimit")]139	fn set_bool_limit(&mut self, caller: caller, limit: string, value: bool) -> Result<void> {140		check_is_owner(caller, self)?;141		let mut limits = self.limits.clone();142143		match limit.as_str() {144			"ownerCanTransfer" => {145				limits.owner_can_transfer = Some(value);146			}147			"ownerCanDestroy" => {148				limits.owner_can_destroy = Some(value);149			}150			"transfersEnabled" => {151				limits.transfers_enabled = Some(value);152			}153			_ => {154				return Err(Error::Revert(format!(155					"Unknown boolean limit \"{}\"",156					limit157				)))158			}159		}160		self.limits = <Pallet<T>>::clamp_limits(self.mode.clone(), &self.limits, limits)161			.map_err(dispatch_to_evm::<T>)?;162		save(self)163	}164165	fn contract_address(&self, _caller: caller) -> Result<address> {166		Ok(crate::eth::collection_id_to_address(self.id))167	}168169	// fn add_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {170	// 	let mut new_admin_h256 = H256::default();171	// 	new_admin.to_little_endian(&mut new_admin_h256.0);172	// 	let account_id = T::AccountId::from(new_admin_h256);173	// 	let caller = T::CrossAccountId::from_eth(caller);174	// 	let new_admin = T::CrossAccountId::from_sub(account_id);175	// 	<Pallet<T>>::toggle_admin(&self, &caller, &new_admin, true)176	// 		.map_err(dispatch_to_evm::<T>)?;177	// 	Ok(())178	// }179180	// fn remove_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {181	// 	let mut new_admin_h256 = H256::default();182	// 	new_admin.to_little_endian(&mut new_admin_h256.0);183	// 	let account_id = T::AccountId::from(new_admin_h256);184	// 	let caller = T::CrossAccountId::from_eth(caller);185	// 	let new_admin = T::CrossAccountId::from_sub(account_id);186	// 	<Pallet<T>>::toggle_admin(&self, &caller, &new_admin, false)187	// 		.map_err(dispatch_to_evm::<T>)?;188	// 	Ok(())189	// }190191	fn add_collection_admin(&self, caller: caller, new_admin: address) -> Result<void> {192		let caller = T::CrossAccountId::from_eth(caller);193		self.check_is_owner_or_admin(&caller)194			.map_err(dispatch_to_evm::<T>)?;195		let new_admin = T::CrossAccountId::from_eth(new_admin);196		<Pallet<T>>::toggle_admin(&self, &caller, &new_admin, true)197			.map_err(dispatch_to_evm::<T>)?;198		Ok(())199	}200201	fn remove_collection_admin(&self, caller: caller, admin: address) -> Result<void> {202		let caller = T::CrossAccountId::from_eth(caller);203		self.check_is_owner_or_admin(&caller)204			.map_err(dispatch_to_evm::<T>)?;205		let admin = T::CrossAccountId::from_eth(admin);206		<Pallet<T>>::toggle_admin(&self, &caller, &admin, false).map_err(dispatch_to_evm::<T>)?;207		Ok(())208	}209210	#[solidity(rename_selector = "setCollectionNesting")]211	fn set_nesting_bool(&mut self, caller: caller, enable: bool) -> Result<void> {212		let caller = T::CrossAccountId::from_eth(caller);213		self.check_is_owner_or_admin(&caller)214			.map_err(dispatch_to_evm::<T>)?;215		self.collection.permissions.nesting = Some(match enable {216			false => NestingRule::Disabled,217			true => NestingRule::Owner,218		});219		save(self);220		Ok(())221	}222223	#[solidity(rename_selector = "setCollectionNesting")]224	fn set_nesting(225		&mut self,226		caller: caller,227		enable: bool,228		collections: Vec<address>,229	) -> Result<void> {230		if collections.is_empty() {231			return Err("No addresses provided".into());232		}233		if collections.len() >= OwnerRestrictedSet::bound() {234			return Err(Error::Revert(format!(235				"Out of bound: {} >= {}",236				collections.len(),237				OwnerRestrictedSet::bound()238			)));239		}240		let caller = T::CrossAccountId::from_eth(caller);241		self.check_is_owner_or_admin(&caller)242			.map_err(dispatch_to_evm::<T>)?;243		self.collection.permissions.nesting = Some(match enable {244			false => NestingRule::Disabled,245			true => {246				let mut bv = OwnerRestrictedSet::new();247				for i in collections {248					bv.try_insert(crate::eth::map_eth_to_id(&i).ok_or(Error::Revert(249						"Can't convert address into collection id".into(),250					))?)251					.map_err(|e| Error::Revert(format!("{:?}", e)))?;252				}253				NestingRule::OwnerRestricted(bv)254			}255		});256		save(self);257		Ok(())258	}259260	fn set_collection_access(&mut self, caller: caller, mode: uint8) -> Result<void> {261		let caller = T::CrossAccountId::from_eth(caller);262		self.check_is_owner_or_admin(&caller)263			.map_err(dispatch_to_evm::<T>)?;264		self.collection.permissions.access = Some(match mode {265			0 => AccessMode::Normal,266			1 => AccessMode::AllowList,267			_ => return Err("Not supported access mode".into()),268		});269		save(self);270		Ok(())271	}272273	fn add_to_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {274		let caller = check_is_owner_or_admin(caller, self)?;275		let user = T::CrossAccountId::from_eth(user);276		<Pallet<T>>::toggle_allowlist(self, &caller, &user, true).map_err(dispatch_to_evm::<T>)?;277		Ok(())278	}279280	fn remove_from_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {281		let caller = check_is_owner_or_admin(caller, self)?;282		let user = T::CrossAccountId::from_eth(user);283		<Pallet<T>>::toggle_allowlist(self, &caller, &user, false).map_err(dispatch_to_evm::<T>)?;284		Ok(())285	}286287	fn set_collection_mint_mode(&mut self, caller: caller, mode: bool) -> Result<void> {288		check_is_owner_or_admin(caller, self)?;289		self.collection.permissions.mint_mode = Some(mode);290		save(self);291		Ok(())292	}293}294295fn check_is_owner<T: Config>(caller: caller, collection: &CollectionHandle<T>) -> Result<void> {296	let caller = T::CrossAccountId::from_eth(caller);297	collection298		.check_is_owner(&caller)299		.map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;300	Ok(())301}302303fn check_is_owner_or_admin<T: Config>(304	caller: caller,305	collection: &CollectionHandle<T>,306) -> Result<T::CrossAccountId> {307	let caller = T::CrossAccountId::from_eth(caller);308	collection309		.check_is_owner_or_admin(&caller)310		.map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;311	Ok(caller)312}313314fn save<T: Config>(collection: &CollectionHandle<T>) -> Result<void> {315	collection.check_is_read_only().map_err(dispatch_to_evm::<T>)?;316	<crate::CollectionById<T>>::insert(collection.id, collection.collection.clone());317	Ok(())318}319320pub fn token_uri_key() -> up_data_structs::PropertyKey {321	b"tokenURI"322		.to_vec()323		.try_into()324		.expect("length < limit; qed")325}