git.delta.rocks / unique-network / refs/commits / 3b810a7cb2bc

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};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>51where52	T::AccountId: From<[u8; 32]>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_or_admin(caller, self)?;8889		let sponsor = T::CrossAccountId::from_eth(sponsor);90		self.set_sponsor(sponsor.as_sub().clone())91			.map_err(dispatch_to_evm::<T>)?;92		save(self)93	}9495	fn confirm_collection_sponsorship(&mut self, caller: caller) -> Result<void> {96		let caller = T::CrossAccountId::from_eth(caller);97		if !self98			.confirm_sponsorship(caller.as_sub())99			.map_err(dispatch_to_evm::<T>)?100		{101			return Err(Error::Revert("Caller is not set as sponsor".into()));102		}103		save(self)104	}105106	#[solidity(rename_selector = "setCollectionLimit")]107	fn set_int_limit(&mut self, caller: caller, limit: string, value: uint32) -> Result<void> {108		check_is_owner_or_admin(caller, self)?;109		let mut limits = self.limits.clone();110111		match limit.as_str() {112			"accountTokenOwnershipLimit" => {113				limits.account_token_ownership_limit = Some(value);114			}115			"sponsoredDataSize" => {116				limits.sponsored_data_size = Some(value);117			}118			"sponsoredDataRateLimit" => {119				limits.sponsored_data_rate_limit = Some(SponsoringRateLimit::Blocks(value));120			}121			"tokenLimit" => {122				limits.token_limit = Some(value);123			}124			"sponsorTransferTimeout" => {125				limits.sponsor_transfer_timeout = Some(value);126			}127			"sponsorApproveTimeout" => {128				limits.sponsor_approve_timeout = Some(value);129			}130			_ => {131				return Err(Error::Revert(format!(132					"Unknown integer limit \"{}\"",133					limit134				)))135			}136		}137		self.limits = <Pallet<T>>::clamp_limits(self.mode.clone(), &self.limits, limits)138			.map_err(dispatch_to_evm::<T>)?;139		save(self)140	}141142	#[solidity(rename_selector = "setCollectionLimit")]143	fn set_bool_limit(&mut self, caller: caller, limit: string, value: bool) -> Result<void> {144		check_is_owner_or_admin(caller, self)?;145		let mut limits = self.limits.clone();146147		match limit.as_str() {148			"ownerCanTransfer" => {149				limits.owner_can_transfer = Some(value);150			}151			"ownerCanDestroy" => {152				limits.owner_can_destroy = Some(value);153			}154			"transfersEnabled" => {155				limits.transfers_enabled = Some(value);156			}157			_ => {158				return Err(Error::Revert(format!(159					"Unknown boolean limit \"{}\"",160					limit161				)))162			}163		}164		self.limits = <Pallet<T>>::clamp_limits(self.mode.clone(), &self.limits, limits)165			.map_err(dispatch_to_evm::<T>)?;166		save(self)167	}168169	fn contract_address(&self, _caller: caller) -> Result<address> {170		Ok(crate::eth::collection_id_to_address(self.id))171	}172173	fn add_collection_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {174		let caller = T::CrossAccountId::from_eth(caller);175		let mut new_admin_arr: [u8; 32] = Default::default();176		new_admin.to_big_endian(&mut new_admin_arr);177		let account_id = T::AccountId::from(new_admin_arr);178		let new_admin = T::CrossAccountId::from_sub(account_id);179		<Pallet<T>>::toggle_admin(self, &caller, &new_admin, true)180			.map_err(dispatch_to_evm::<T>)?;181		Ok(())182	}183184	fn remove_collection_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {185		let caller = T::CrossAccountId::from_eth(caller);186		let mut new_admin_arr: [u8; 32] = Default::default();187		new_admin.to_big_endian(&mut new_admin_arr);188		let account_id = T::AccountId::from(new_admin_arr);189		let new_admin = T::CrossAccountId::from_sub(account_id);190		<Pallet<T>>::toggle_admin(self, &caller, &new_admin, false)191			.map_err(dispatch_to_evm::<T>)?;192		Ok(())193	}194195	fn add_collection_admin(&self, caller: caller, new_admin: address) -> Result<void> {196		let caller = T::CrossAccountId::from_eth(caller);197		let new_admin = T::CrossAccountId::from_eth(new_admin);198		<Pallet<T>>::toggle_admin(self, &caller, &new_admin, true)199			.map_err(dispatch_to_evm::<T>)?;200		Ok(())201	}202203	fn remove_admin(&self, caller: caller, admin: address) -> Result<void> {204		let caller = T::CrossAccountId::from_eth(caller);205		let admin = T::CrossAccountId::from_eth(admin);206		<Pallet<T>>::toggle_admin(self, &caller, &admin, false)207			.map_err(dispatch_to_evm::<T>)?;208		Ok(())209	}210211	#[solidity(rename_selector = "setNesting")]212	fn set_nesting_bool(&mut self, caller: caller, enable: bool) -> Result<void> {213		check_is_owner_or_admin(caller, self)?;214		self.collection.permissions.nesting = Some(match enable {215			false => NestingRule::Disabled,216			true => NestingRule::Owner,217		});218		save(self)?;219		Ok(())220	}221222	#[solidity(rename_selector = "setNesting")]223	fn set_nesting(&mut self, caller: caller, enable: bool, collections: Vec<address>) -> Result<void> {224		if collections.is_empty() {225			return Err("No addresses provided".into());226		}227		if collections.len() >= OwnerRestrictedSet::bound() {228			return Err(Error::Revert(format!("Out of bound: {} >= {}", collections.len(), OwnerRestrictedSet::bound())));229		}230		check_is_owner_or_admin(caller, self)?;231		self.collection.permissions.nesting = Some(match enable {232			false => NestingRule::Disabled,233			true => {234				let mut bv = OwnerRestrictedSet::new();235				for i in collections {236					bv.try_insert(crate::eth::map_eth_to_id(&i).ok_or_else(|| Error::Revert(237						"Can't convert address into collection id".into(),238						))?)239					.map_err(|e| Error::Revert(format!("{:?}", e)))?;240				}241				NestingRule::OwnerRestricted (bv)242			}243		});244		save(self)?;245		Ok(())246	}247248	fn set_collection_access(&mut self, caller: caller, mode: uint8) -> Result<void> {249		check_is_owner_or_admin(caller, self)?;250		self.collection.permissions.access = Some(match mode {251			0 => AccessMode::Normal,252			1 => AccessMode::AllowList,253			_ => return Err("Not supported access mode".into()),254		});255		save(self)?;256		Ok(())257	}258259	fn add_to_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {260		let caller = T::CrossAccountId::from_eth(caller);261		let user = T::CrossAccountId::from_eth(user);262		<Pallet<T>>::toggle_allowlist(self, &caller, &user, true)263			.map_err(dispatch_to_evm::<T>)?;264		Ok(())265	}266267	fn remove_from_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {268		let caller = T::CrossAccountId::from_eth(caller);269		let user = T::CrossAccountId::from_eth(user);270		<Pallet<T>>::toggle_allowlist(self, &caller, &user, false)271			.map_err(dispatch_to_evm::<T>)?;272		Ok(())273	}274275	fn set_mint_mode(&mut self, caller: caller, mode: bool) -> Result<void> {276		check_is_owner_or_admin(caller, self)?;277		self.collection.permissions.mint_mode = Some(mode);278		save(self)?;279		Ok(())280	}281}282283fn check_is_owner_or_admin<T: Config>(284	caller: caller,285	collection: &CollectionHandle<T>,286) -> Result<T::CrossAccountId> {287	let caller = T::CrossAccountId::from_eth(caller);288	collection289		.check_is_owner_or_admin(&caller)290		.map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;291	Ok(caller)292}293294fn save<T: Config>(collection: &CollectionHandle<T>) {295	<crate::CollectionById<T>>::insert(collection.id, collection.collection.clone());296	Ok(())297}298299pub fn token_uri_key() -> up_data_structs::PropertyKey {300	b"tokenURI"301		.to_vec()302		.try_into()303		.expect("length < limit; qed")304}