difftreelog
fix check errors in erc
in: master
1 file changed
pallets/common/src/erc.rsdiffbeforeafterboth1// 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())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(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(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_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {174 // let mut new_admin_h256 = H256::default();175 // new_admin.to_little_endian(&mut new_admin_h256.0);176 // let account_id = T::AccountId::from(new_admin_h256);177 // let caller = T::CrossAccountId::from_eth(caller);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_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {185 // let mut new_admin_h256 = H256::default();186 // new_admin.to_little_endian(&mut new_admin_h256.0);187 // let account_id = T::AccountId::from(new_admin_h256);188 // let caller = T::CrossAccountId::from_eth(caller);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 self.check_is_owner_or_admin(&caller)198 .map_err(dispatch_to_evm::<T>)?;199 let new_admin = T::CrossAccountId::from_eth(new_admin);200 <Pallet<T>>::toggle_admin(&self, &caller, &new_admin, true)201 .map_err(dispatch_to_evm::<T>)?;202 Ok(())203 }204205 fn remove_collection_admin(&self, caller: caller, admin: address) -> Result<void> {206 let caller = T::CrossAccountId::from_eth(caller);207 self.check_is_owner_or_admin(&caller)208 .map_err(dispatch_to_evm::<T>)?;209 let admin = T::CrossAccountId::from_eth(admin);210 <Pallet<T>>::toggle_admin(&self, &caller, &admin, false).map_err(dispatch_to_evm::<T>)?;211 Ok(())212 }213214 #[solidity(rename_selector = "setCollectionNesting")]215 fn set_nesting_bool(&mut self, caller: caller, enable: bool) -> Result<void> {216 let caller = T::CrossAccountId::from_eth(caller);217 self.check_is_owner_or_admin(&caller)218 .map_err(dispatch_to_evm::<T>)?;219 self.collection.permissions.nesting = Some(match enable {220 false => NestingRule::Disabled,221 true => NestingRule::Owner,222 });223 save(self);224 Ok(())225 }226227 #[solidity(rename_selector = "setCollectionNesting")]228 fn set_nesting(229 &mut self,230 caller: caller,231 enable: bool,232 collections: Vec<address>,233 ) -> Result<void> {234 if collections.is_empty() {235 return Err("No addresses provided".into());236 }237 if collections.len() >= OwnerRestrictedSet::bound() {238 return Err(Error::Revert(format!(239 "Out of bound: {} >= {}",240 collections.len(),241 OwnerRestrictedSet::bound()242 )));243 }244 let caller = T::CrossAccountId::from_eth(caller);245 self.check_is_owner_or_admin(&caller)246 .map_err(dispatch_to_evm::<T>)?;247 self.collection.permissions.nesting = Some(match enable {248 false => NestingRule::Disabled,249 true => {250 let mut bv = OwnerRestrictedSet::new();251 for i in collections {252 bv.try_insert(crate::eth::map_eth_to_id(&i).ok_or(Error::Revert(253 "Can't convert address into collection id".into(),254 ))?)255 .map_err(|e| Error::Revert(format!("{:?}", e)))?;256 }257 NestingRule::OwnerRestricted(bv)258 }259 });260 save(self);261 Ok(())262 }263264 fn set_collection_access(&mut self, caller: caller, mode: uint8) -> Result<void> {265 let caller = T::CrossAccountId::from_eth(caller);266 self.check_is_owner_or_admin(&caller)267 .map_err(dispatch_to_evm::<T>)?;268 self.collection.permissions.access = Some(match mode {269 0 => AccessMode::Normal,270 1 => AccessMode::AllowList,271 _ => return Err("Not supported access mode".into()),272 });273 save(self);274 Ok(())275 }276277 fn add_to_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {278 let caller = check_is_owner_or_admin(caller, self)?;279 let user = T::CrossAccountId::from_eth(user);280 <Pallet<T>>::toggle_allowlist(self, &caller, &user, true).map_err(dispatch_to_evm::<T>)?;281 Ok(())282 }283284 fn remove_from_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {285 let caller = check_is_owner_or_admin(caller, self)?;286 let user = T::CrossAccountId::from_eth(user);287 <Pallet<T>>::toggle_allowlist(self, &caller, &user, false).map_err(dispatch_to_evm::<T>)?;288 Ok(())289 }290291 fn set_collection_mint_mode(&mut self, caller: caller, mode: bool) -> Result<void> {292 check_is_owner_or_admin(caller, self)?;293 self.collection.permissions.mint_mode = Some(mode);294 save(self);295 Ok(())296 }297}298299fn check_is_owner<T: Config>(caller: caller, collection: &CollectionHandle<T>) -> Result<void> {300 let caller = T::CrossAccountId::from_eth(caller);301 collection302 .check_is_owner(&caller)303 .map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;304 Ok(())305}306307fn check_is_owner_or_admin<T: Config>(308 caller: caller,309 collection: &CollectionHandle<T>,310) -> Result<T::CrossAccountId> {311 let caller = T::CrossAccountId::from_eth(caller);312 collection313 .check_is_owner_or_admin(&caller)314 .map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;315 Ok(caller)316}317318fn save<T: Config>(collection: &CollectionHandle<T>) -> Result<void> {319 // TODO possibly delete for the lack of transaction320 collection321 .check_is_internal()322 .map_err(dispatch_to_evm::<T>)?;323 <crate::CollectionById<T>>::insert(collection.id, collection.collection.clone());324 Ok(())325}326327pub fn token_uri_key() -> up_data_structs::PropertyKey {328 b"tokenURI"329 .to_vec()330 .try_into()331 .expect("length < limit; qed")332}