git.delta.rocks / unique-network / refs/commits / 504df77ef385

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_std::vec::Vec;25use up_data_structs::{Property, SponsoringRateLimit, OwnerRestrictedSet, AccessMode};26use alloc::format;2728use crate::{Pallet, CollectionHandle, Config, CollectionProperties};2930#[derive(ToLog)]31pub enum CollectionHelpersEvents {32	CollectionCreated {33		#[indexed]34		owner: address,35		#[indexed]36		collection_id: address,37	},38}3940/// Does not always represent a full collection, for RFT it is either41/// collection (Implementing ERC721), or specific collection token (Implementing ERC20)42pub trait CommonEvmHandler {43	const CODE: &'static [u8];4445	fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult>;46}4748#[solidity_interface(name = "Collection")]49impl<T: Config> CollectionHandle<T>50// where51// 	T::AccountId: From<H256>52{53	fn set_collection_property(&mut self, caller: caller, key: string, value: bytes) -> Result<()> {54		let caller = T::CrossAccountId::from_eth(caller);55		let key = <Vec<u8>>::from(key)56			.try_into()57			.map_err(|_| "key too large")?;58		let value = value.try_into().map_err(|_| "value too large")?;5960		<Pallet<T>>::set_collection_property(self, &caller, Property { key, value })61			.map_err(dispatch_to_evm::<T>)62	}6364	fn delete_collection_property(&mut self, caller: caller, key: string) -> Result<()> {65		let caller = T::CrossAccountId::from_eth(caller);66		let key = <Vec<u8>>::from(key)67			.try_into()68			.map_err(|_| "key too large")?;6970		<Pallet<T>>::delete_collection_property(self, &caller, key).map_err(dispatch_to_evm::<T>)71	}7273	/// Throws error if key not found74	fn collection_property(&self, key: string) -> Result<bytes> {75		let key = <Vec<u8>>::from(key)76			.try_into()77			.map_err(|_| "key too large")?;7879		let props = <CollectionProperties<T>>::get(self.id);80		let prop = props.get(&key).ok_or("key not found")?;8182		Ok(prop.to_vec())83	}8485	fn set_collection_sponsor(&mut self, caller: caller, sponsor: address) -> Result<void> {86		check_is_owner(caller, self)?;8788		let sponsor = T::CrossAccountId::from_eth(sponsor);89		self.set_sponsor(sponsor.as_sub().clone())90			.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 !self97			.confirm_sponsorship(caller.as_sub())98			.map_err(dispatch_to_evm::<T>)?99		{100			return Err(Error::Revert("Caller is not set as sponsor".into()));101		}102		save(self)103	}104105	#[solidity(rename_selector = "setCollectionLimit")]106	fn set_int_limit(&mut self, caller: caller, limit: string, value: uint32) -> Result<void> {107		check_is_owner(caller, self)?;108		let mut limits = self.limits.clone();109110		match limit.as_str() {111			"accountTokenOwnershipLimit" => {112				limits.account_token_ownership_limit = Some(value);113			}114			"sponsoredDataSize" => {115				limits.sponsored_data_size = Some(value);116			}117			"sponsoredDataRateLimit" => {118				limits.sponsored_data_rate_limit = Some(SponsoringRateLimit::Blocks(value));119			}120			"tokenLimit" => {121				limits.token_limit = Some(value);122			}123			"sponsorTransferTimeout" => {124				limits.sponsor_transfer_timeout = Some(value);125			}126			"sponsorApproveTimeout" => {127				limits.sponsor_approve_timeout = Some(value);128			}129			_ => {130				return Err(Error::Revert(format!(131					"Unknown integer limit \"{}\"",132					limit133				)))134			}135		}136		self.limits = <Pallet<T>>::clamp_limits(self.mode.clone(), &self.limits, limits)137			.map_err(dispatch_to_evm::<T>)?;138		save(self)139	}140141	#[solidity(rename_selector = "setCollectionLimit")]142	fn set_bool_limit(&mut self, caller: caller, limit: string, value: bool) -> Result<void> {143		check_is_owner(caller, self)?;144		let mut limits = self.limits.clone();145146		match limit.as_str() {147			"ownerCanTransfer" => {148				limits.owner_can_transfer = Some(value);149			}150			"ownerCanDestroy" => {151				limits.owner_can_destroy = Some(value);152			}153			"transfersEnabled" => {154				limits.transfers_enabled = Some(value);155			}156			_ => {157				return Err(Error::Revert(format!(158					"Unknown boolean limit \"{}\"",159					limit160				)))161			}162		}163		self.limits = <Pallet<T>>::clamp_limits(self.mode.clone(), &self.limits, limits)164			.map_err(dispatch_to_evm::<T>)?;165		save(self)166	}167168	fn contract_address(&self, _caller: caller) -> Result<address> {169		Ok(crate::eth::collection_id_to_address(self.id))170	}171172	// fn add_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {173	// 	let mut new_admin_h256 = H256::default();174	// 	new_admin.to_little_endian(&mut new_admin_h256.0);175	// 	let account_id = T::AccountId::from(new_admin_h256);176	// 	let caller = T::CrossAccountId::from_eth(caller);177	// 	let new_admin = T::CrossAccountId::from_sub(account_id);178	// 	<Pallet<T>>::toggle_admin(&self, &caller, &new_admin, true)179	// 		.map_err(dispatch_to_evm::<T>)?;180	// 	Ok(())181	// }182183	// fn remove_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {184	// 	let mut new_admin_h256 = H256::default();185	// 	new_admin.to_little_endian(&mut new_admin_h256.0);186	// 	let account_id = T::AccountId::from(new_admin_h256);187	// 	let caller = T::CrossAccountId::from_eth(caller);188	// 	let new_admin = T::CrossAccountId::from_sub(account_id);189	// 	<Pallet<T>>::toggle_admin(&self, &caller, &new_admin, false)190	// 		.map_err(dispatch_to_evm::<T>)?;191	// 	Ok(())192	// }193194	fn add_collection_admin(&self, caller: caller, new_admin: address) -> Result<void> {195		let caller = T::CrossAccountId::from_eth(caller);196		self.check_is_owner_or_admin(&caller)197			.map_err(dispatch_to_evm::<T>)?;198		let new_admin = T::CrossAccountId::from_eth(new_admin);199		<Pallet<T>>::toggle_admin(&self, &caller, &new_admin, true)200			.map_err(dispatch_to_evm::<T>)?;201		Ok(())202	}203204	fn remove_collection_admin(&self, caller: caller, admin: address) -> Result<void> {205		let caller = T::CrossAccountId::from_eth(caller);206		self.check_is_owner_or_admin(&caller)207			.map_err(dispatch_to_evm::<T>)?;208		let admin = T::CrossAccountId::from_eth(admin);209		<Pallet<T>>::toggle_admin(&self, &caller, &admin, false).map_err(dispatch_to_evm::<T>)?;210		Ok(())211	}212213	#[solidity(rename_selector = "setCollectionNesting")]214	fn set_nesting_bool(&mut self, caller: caller, enable: bool) -> Result<void> {215		let caller = T::CrossAccountId::from_eth(caller);216		self.check_is_owner_or_admin(&caller)217			.map_err(dispatch_to_evm::<T>)?;218219		let mut permissions = self.collection.permissions.clone();220		let mut nesting = permissions.nesting().clone();221		nesting.token_owner = enable;222		nesting.restricted = None;223		permissions.nesting = Some(nesting);224225		self.collection.permissions = <Pallet<T>>::clamp_permissions(226			self.collection.mode.clone(),227			&self.collection.permissions,228			permissions,229		)230		.map_err(dispatch_to_evm::<T>)?;231232		save(self)233	}234235	#[solidity(rename_selector = "setCollectionNesting")]236	fn set_nesting(237		&mut self,238		caller: caller,239		enable: bool,240		collections: Vec<address>,241	) -> Result<void> {242		if collections.is_empty() {243			return Err("No addresses provided".into());244		}245		let caller = T::CrossAccountId::from_eth(caller);246		self.check_is_owner_or_admin(&caller)247			.map_err(dispatch_to_evm::<T>)?;248249		let mut permissions = self.collection.permissions.clone();250		match enable {251			false => {252				let mut nesting = permissions.nesting().clone();253				nesting.token_owner = false;254				nesting.restricted = None;255				permissions.nesting = Some(nesting);256			}257			true => {258				let mut bv = OwnerRestrictedSet::new();259				for i in collections {260					bv.try_insert(crate::eth::map_eth_to_id(&i).ok_or(Error::Revert(261						"Can't convert address into collection id".into(),262					))?)263					.map_err(|_| "too many collections")?;264				}265				let mut nesting = permissions.nesting().clone();266				nesting.token_owner = true;267				nesting.restricted = Some(bv);268				permissions.nesting = Some(nesting);269			}270		};271272		self.collection.permissions = <Pallet<T>>::clamp_permissions(273			self.collection.mode.clone(),274			&self.collection.permissions,275			permissions,276		)277		.map_err(dispatch_to_evm::<T>)?;278279		save(self)280	}281282	fn set_collection_access(&mut self, caller: caller, mode: uint8) -> Result<void> {283		let caller = T::CrossAccountId::from_eth(caller);284		self.check_is_owner_or_admin(&caller)285			.map_err(dispatch_to_evm::<T>)?;286		self.collection.permissions.access = Some(match mode {287			0 => AccessMode::Normal,288			1 => AccessMode::AllowList,289			_ => return Err("Not supported access mode".into()),290		});291		save(self)?;292		Ok(())293	}294295	fn add_to_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {296		let caller = check_is_owner_or_admin(caller, self)?;297		let user = T::CrossAccountId::from_eth(user);298		<Pallet<T>>::toggle_allowlist(self, &caller, &user, true).map_err(dispatch_to_evm::<T>)?;299		Ok(())300	}301302	fn remove_from_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {303		let caller = check_is_owner_or_admin(caller, self)?;304		let user = T::CrossAccountId::from_eth(user);305		<Pallet<T>>::toggle_allowlist(self, &caller, &user, false).map_err(dispatch_to_evm::<T>)?;306		Ok(())307	}308309	fn set_collection_mint_mode(&mut self, caller: caller, mode: bool) -> Result<void> {310		check_is_owner_or_admin(caller, self)?;311		self.collection.permissions.mint_mode = Some(mode);312		save(self)?;313		Ok(())314	}315}316317fn check_is_owner<T: Config>(caller: caller, collection: &CollectionHandle<T>) -> Result<void> {318	let caller = T::CrossAccountId::from_eth(caller);319	collection320		.check_is_owner(&caller)321		.map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;322	Ok(())323}324325fn check_is_owner_or_admin<T: Config>(326	caller: caller,327	collection: &CollectionHandle<T>,328) -> Result<T::CrossAccountId> {329	let caller = T::CrossAccountId::from_eth(caller);330	collection331		.check_is_owner_or_admin(&caller)332		.map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;333	Ok(caller)334}335336fn save<T: Config>(collection: &CollectionHandle<T>) -> Result<void> {337	// TODO possibly delete for the lack of transaction338	collection339		.check_is_internal()340		.map_err(dispatch_to_evm::<T>)?;341	<crate::CollectionById<T>>::insert(collection.id, collection.collection.clone());342	Ok(())343}344345pub fn token_uri_key() -> up_data_structs::PropertyKey {346	b"tokenURI"347		.to_vec()348		.try_into()349		.expect("length < limit; qed")350}