12345678910111213141516171819pub use pallet_evm::{PrecompileOutput, PrecompileResult, PrecompileHandle, account::CrossAccountId};20use evm_coder::{21 abi::AbiType,22 solidity_interface, solidity, ToLog,23 types::*,24 execution::{Result, Error},25 weight,26};27use pallet_evm_coder_substrate::dispatch_to_evm;28use sp_std::{vec, vec::Vec};29use sp_core::U256;30use up_data_structs::{31 AccessMode, CollectionMode, CollectionPermissions, OwnerRestrictedSet, Property,32 SponsoringRateLimit, SponsorshipState,33};3435use crate::{36 Pallet, CollectionHandle, Config, CollectionProperties, SelfWeightOf, eth, weights::WeightInfo,37};383940#[derive(ToLog)]41pub enum CollectionHelpersEvents {42 43 CollectionCreated {44 45 #[indexed]46 owner: address,4748 49 #[indexed]50 collection_id: address,51 },52 53 CollectionDestroyed {54 55 #[indexed]56 collection_id: address,57 },58 59 CollectionChanged {60 61 #[indexed]62 collection_id: address,63 },6465 66 TokenChanged {67 68 #[indexed]69 collection_id: address,70 71 token_id: U256,72 },73}74757677pub trait CommonEvmHandler {78 79 const CODE: &'static [u8];8081 82 fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult>;83}848586#[solidity_interface(name = Collection)]87impl<T: Config> CollectionHandle<T>88where89 T::AccountId: From<[u8; 32]> + AsRef<[u8; 32]>,90{91 92 93 94 95 #[solidity(hide)]96 #[weight(<SelfWeightOf<T>>::set_collection_properties(1))]97 fn set_collection_property(&mut self, caller: caller, key: string, value: bytes) -> Result<()> {98 let caller = T::CrossAccountId::from_eth(caller);99 let key = <Vec<u8>>::from(key)100 .try_into()101 .map_err(|_| "key too large")?;102 let value = value.0.try_into().map_err(|_| "value too large")?;103104 <Pallet<T>>::set_collection_property(self, &caller, Property { key, value })105 .map_err(dispatch_to_evm::<T>)106 }107108 109 110 111 #[weight(<SelfWeightOf<T>>::set_collection_properties(properties.len() as u32))]112 fn set_collection_properties(113 &mut self,114 caller: caller,115 properties: Vec<eth::Property>,116 ) -> Result<()> {117 let caller = T::CrossAccountId::from_eth(caller);118119 let properties = properties120 .into_iter()121 .map(eth::Property::try_into)122 .collect::<Result<Vec<_>>>()?;123124 <Pallet<T>>::set_collection_properties(self, &caller, properties.into_iter())125 .map_err(dispatch_to_evm::<T>)126 }127128 129 130 131 #[solidity(hide)]132 #[weight(<SelfWeightOf<T>>::delete_collection_properties(1))]133 fn delete_collection_property(&mut self, caller: caller, key: string) -> Result<()> {134 let caller = T::CrossAccountId::from_eth(caller);135 let key = <Vec<u8>>::from(key)136 .try_into()137 .map_err(|_| "key too large")?;138139 <Pallet<T>>::delete_collection_property(self, &caller, key).map_err(dispatch_to_evm::<T>)140 }141142 143 144 145 #[weight(<SelfWeightOf<T>>::delete_collection_properties(keys.len() as u32))]146 fn delete_collection_properties(&mut self, caller: caller, keys: Vec<string>) -> Result<()> {147 let caller = T::CrossAccountId::from_eth(caller);148 let keys = keys149 .into_iter()150 .map(|key| {151 <Vec<u8>>::from(key)152 .try_into()153 .map_err(|_| Error::Revert("key too large".into()))154 })155 .collect::<Result<Vec<_>>>()?;156157 <Pallet<T>>::delete_collection_properties(self, &caller, keys.into_iter())158 .map_err(dispatch_to_evm::<T>)159 }160161 162 163 164 165 166 167 fn collection_property(&self, key: string) -> Result<bytes> {168 let key = <Vec<u8>>::from(key)169 .try_into()170 .map_err(|_| "key too large")?;171172 let props = CollectionProperties::<T>::get(self.id);173 let prop = props.get(&key).ok_or("key not found")?;174175 Ok(bytes(prop.to_vec()))176 }177178 179 180 181 182 fn collection_properties(&self, keys: Vec<string>) -> Result<Vec<eth::Property>> {183 let keys = keys184 .into_iter()185 .map(|key| {186 <Vec<u8>>::from(key)187 .try_into()188 .map_err(|_| Error::Revert("key too large".into()))189 })190 .collect::<Result<Vec<_>>>()?;191192 let properties = Pallet::<T>::filter_collection_properties(193 self.id,194 if keys.is_empty() { None } else { Some(keys) },195 )196 .map_err(dispatch_to_evm::<T>)?;197198 let properties = properties199 .into_iter()200 .map(Property::try_into)201 .collect::<Result<Vec<_>>>()?;202 Ok(properties)203 }204205 206 207 208 209 210 #[solidity(hide)]211 fn set_collection_sponsor(&mut self, caller: caller, sponsor: address) -> Result<()> {212 self.consume_store_reads_and_writes(1, 1)?;213214 let caller = T::CrossAccountId::from_eth(caller);215216 let sponsor = T::CrossAccountId::from_eth(sponsor);217 self.set_sponsor(&caller, sponsor.as_sub().clone())218 .map_err(dispatch_to_evm::<T>)219 }220221 222 223 224 225 226 fn set_collection_sponsor_cross(227 &mut self,228 caller: caller,229 sponsor: eth::CrossAddress,230 ) -> Result<()> {231 self.consume_store_reads_and_writes(1, 1)?;232233 let caller = T::CrossAccountId::from_eth(caller);234235 let sponsor = sponsor.into_sub_cross_account::<T>()?;236 self.set_sponsor(&caller, sponsor.as_sub().clone())237 .map_err(dispatch_to_evm::<T>)238 }239240 241 fn has_collection_pending_sponsor(&self) -> Result<bool> {242 Ok(matches!(243 self.collection.sponsorship,244 SponsorshipState::Unconfirmed(_)245 ))246 }247248 249 250 251 fn confirm_collection_sponsorship(&mut self, caller: caller) -> Result<()> {252 self.consume_store_writes(1)?;253254 let caller = T::CrossAccountId::from_eth(caller);255 self.confirm_sponsorship(caller.as_sub())256 .map_err(dispatch_to_evm::<T>)257 }258259 260 fn remove_collection_sponsor(&mut self, caller: caller) -> Result<()> {261 self.consume_store_reads_and_writes(1, 1)?;262 let caller = T::CrossAccountId::from_eth(caller);263 self.remove_sponsor(&caller).map_err(dispatch_to_evm::<T>)264 }265266 267 268 269 fn collection_sponsor(&self) -> Result<eth::CrossAddress> {270 let sponsor = match self.collection.sponsorship.sponsor() {271 Some(sponsor) => sponsor,272 None => return Ok(Default::default()),273 };274275 Ok(eth::CrossAddress::from_sub::<T>(&sponsor))276 }277278 279 280 281 fn collection_limits(&self) -> Result<Vec<eth::CollectionLimit>> {282 let limits = &self.collection.limits;283284 let convert_value_from_bool = |ob: Option<bool>| match ob {285 Some(b) => Some(b as u32),286 None => None,287 };288289 Ok(vec![290 eth::CollectionLimit::new(291 eth::CollectionLimitField::AccountTokenOwnership,292 limits.account_token_ownership_limit,293 ),294 eth::CollectionLimit::new(295 eth::CollectionLimitField::SponsoredDataSize,296 limits.sponsored_data_size,297 ),298 limits299 .sponsored_data_rate_limit300 .and_then(|limit| {301 if let SponsoringRateLimit::Blocks(blocks) = limit {302 Some(eth::CollectionLimit::new(303 eth::CollectionLimitField::SponsoredDataRateLimit,304 Some(blocks),305 ))306 } else {307 None308 }309 })310 .unwrap_or(eth::CollectionLimit::new(311 eth::CollectionLimitField::SponsoredDataRateLimit,312 Default::default(),313 )),314 eth::CollectionLimit::new(eth::CollectionLimitField::TokenLimit, limits.token_limit),315 eth::CollectionLimit::new(316 eth::CollectionLimitField::SponsorTransferTimeout,317 limits.sponsor_transfer_timeout,318 ),319 eth::CollectionLimit::new(320 eth::CollectionLimitField::SponsorApproveTimeout,321 limits.sponsor_approve_timeout,322 ),323 eth::CollectionLimit::new(324 eth::CollectionLimitField::OwnerCanTransfer,325 convert_value_from_bool(limits.owner_can_transfer),326 ),327 eth::CollectionLimit::new(328 eth::CollectionLimitField::OwnerCanDestroy,329 convert_value_from_bool(limits.owner_can_destroy),330 ),331 eth::CollectionLimit::new(332 eth::CollectionLimitField::TransferEnabled,333 convert_value_from_bool(limits.transfers_enabled),334 ),335 ])336 }337338 339 340 341 #[solidity(rename_selector = "setCollectionLimit")]342 fn set_collection_limit(&mut self, caller: caller, limit: eth::CollectionLimit) -> Result<()> {343 self.consume_store_reads_and_writes(1, 1)?;344345 if !limit.has_value() {346 return Err(Error::Revert("user can't disable limits".into()));347 }348349 let caller = T::CrossAccountId::from_eth(caller);350 <Pallet<T>>::update_limits(&caller, self, limit.try_into()?).map_err(dispatch_to_evm::<T>)351 }352353 354 fn contract_address(&self) -> Result<address> {355 Ok(crate::eth::collection_id_to_address(self.id))356 }357358 359 360 fn add_collection_admin_cross(361 &mut self,362 caller: caller,363 new_admin: eth::CrossAddress,364 ) -> Result<()> {365 self.consume_store_reads_and_writes(2, 2)?;366367 let caller = T::CrossAccountId::from_eth(caller);368 let new_admin = new_admin.into_sub_cross_account::<T>()?;369 <Pallet<T>>::toggle_admin(self, &caller, &new_admin, true).map_err(dispatch_to_evm::<T>)?;370 Ok(())371 }372373 374 375 fn remove_collection_admin_cross(376 &mut self,377 caller: caller,378 admin: eth::CrossAddress,379 ) -> Result<()> {380 self.consume_store_reads_and_writes(2, 2)?;381382 let caller = T::CrossAccountId::from_eth(caller);383 let admin = admin.into_sub_cross_account::<T>()?;384 <Pallet<T>>::toggle_admin(self, &caller, &admin, false).map_err(dispatch_to_evm::<T>)?;385 Ok(())386 }387388 389 390 #[solidity(hide)]391 fn add_collection_admin(&mut self, caller: caller, new_admin: address) -> Result<()> {392 self.consume_store_reads_and_writes(2, 2)?;393394 let caller = T::CrossAccountId::from_eth(caller);395 let new_admin = T::CrossAccountId::from_eth(new_admin);396 <Pallet<T>>::toggle_admin(self, &caller, &new_admin, true).map_err(dispatch_to_evm::<T>)?;397 Ok(())398 }399400 401 402 403 #[solidity(hide)]404 fn remove_collection_admin(&mut self, caller: caller, admin: address) -> Result<()> {405 self.consume_store_reads_and_writes(2, 2)?;406407 let caller = T::CrossAccountId::from_eth(caller);408 let admin = T::CrossAccountId::from_eth(admin);409 <Pallet<T>>::toggle_admin(self, &caller, &admin, false).map_err(dispatch_to_evm::<T>)?;410 Ok(())411 }412413 414 415 416 #[solidity(rename_selector = "setCollectionNesting")]417 fn set_nesting_bool(&mut self, caller: caller, enable: bool) -> Result<()> {418 self.consume_store_reads_and_writes(1, 1)?;419420 let caller = T::CrossAccountId::from_eth(caller);421422 let mut permissions = self.collection.permissions.clone();423 let mut nesting = permissions.nesting().clone();424 nesting.token_owner = enable;425 nesting.restricted = None;426 permissions.nesting = Some(nesting);427428 <Pallet<T>>::update_permissions(&caller, self, permissions).map_err(dispatch_to_evm::<T>)429 }430431 432 433 434 435 #[solidity(rename_selector = "setCollectionNesting")]436 fn set_nesting(437 &mut self,438 caller: caller,439 enable: bool,440 collections: Vec<address>,441 ) -> Result<()> {442 self.consume_store_reads_and_writes(1, 1)?;443444 if collections.is_empty() {445 return Err("no addresses provided".into());446 }447 let caller = T::CrossAccountId::from_eth(caller);448449 let mut permissions = self.collection.permissions.clone();450 match enable {451 false => {452 let mut nesting = permissions.nesting().clone();453 nesting.token_owner = false;454 nesting.restricted = None;455 permissions.nesting = Some(nesting);456 }457 true => {458 let mut bv = OwnerRestrictedSet::new();459 for i in collections {460 bv.try_insert(crate::eth::map_eth_to_id(&i).ok_or_else(|| {461 Error::Revert("Can't convert address into collection id".into())462 })?)463 .map_err(|_| "too many collections")?;464 }465 let mut nesting = permissions.nesting().clone();466 nesting.token_owner = true;467 nesting.restricted = Some(bv);468 permissions.nesting = Some(nesting);469 }470 };471472 <Pallet<T>>::update_permissions(&caller, self, permissions).map_err(dispatch_to_evm::<T>)473 }474475 476 #[solidity(rename_selector = "collectionNestingRestrictedCollectionIds")]477 fn collection_nesting_restricted_ids(&self) -> Result<eth::CollectionNesting> {478 let nesting = self.collection.permissions.nesting();479480 Ok(eth::CollectionNesting::new(481 nesting.token_owner,482 nesting483 .restricted484 .clone()485 .map(|b| b.0.into_inner().iter().map(|id| id.0.into()).collect())486 .unwrap_or_default(),487 ))488 }489490 491 fn collection_nesting_permissions(&self) -> Result<Vec<eth::CollectionNestingPermission>> {492 let nesting = self.collection.permissions.nesting();493 Ok(vec![494 eth::CollectionNestingPermission::new(495 eth::CollectionPermissionField::CollectionAdmin,496 nesting.collection_admin,497 ),498 eth::CollectionNestingPermission::new(499 eth::CollectionPermissionField::TokenOwner,500 nesting.token_owner,501 ),502 ])503 }504 505 506 507 508 fn set_collection_access(&mut self, caller: caller, mode: u8) -> Result<()> {509 self.consume_store_reads_and_writes(1, 1)?;510511 let caller = T::CrossAccountId::from_eth(caller);512 let permissions = CollectionPermissions {513 access: Some(match mode {514 0 => AccessMode::Normal,515 1 => AccessMode::AllowList,516 _ => return Err("not supported access mode".into()),517 }),518 ..Default::default()519 };520 <Pallet<T>>::update_permissions(&caller, self, permissions).map_err(dispatch_to_evm::<T>)521 }522523 524 525 526 fn allowlisted_cross(&self, user: eth::CrossAddress) -> Result<bool> {527 let user = user.into_sub_cross_account::<T>()?;528 Ok(Pallet::<T>::allowed(self.id, user))529 }530531 532 533 534 #[solidity(hide)]535 fn add_to_collection_allow_list(&mut self, caller: caller, user: address) -> Result<()> {536 self.consume_store_writes(1)?;537538 let caller = T::CrossAccountId::from_eth(caller);539 let user = T::CrossAccountId::from_eth(user);540 <Pallet<T>>::toggle_allowlist(self, &caller, &user, true).map_err(dispatch_to_evm::<T>)?;541 Ok(())542 }543544 545 546 547 fn add_to_collection_allow_list_cross(548 &mut self,549 caller: caller,550 user: eth::CrossAddress,551 ) -> Result<()> {552 self.consume_store_writes(1)?;553554 let caller = T::CrossAccountId::from_eth(caller);555 let user = user.into_sub_cross_account::<T>()?;556 Pallet::<T>::toggle_allowlist(self, &caller, &user, true).map_err(dispatch_to_evm::<T>)?;557 Ok(())558 }559560 561 562 563 #[solidity(hide)]564 fn remove_from_collection_allow_list(&mut self, caller: caller, user: address) -> Result<()> {565 self.consume_store_writes(1)?;566567 let caller = T::CrossAccountId::from_eth(caller);568 let user = T::CrossAccountId::from_eth(user);569 <Pallet<T>>::toggle_allowlist(self, &caller, &user, false).map_err(dispatch_to_evm::<T>)?;570 Ok(())571 }572573 574 575 576 fn remove_from_collection_allow_list_cross(577 &mut self,578 caller: caller,579 user: eth::CrossAddress,580 ) -> Result<()> {581 self.consume_store_writes(1)?;582583 let caller = T::CrossAccountId::from_eth(caller);584 let user = user.into_sub_cross_account::<T>()?;585 Pallet::<T>::toggle_allowlist(self, &caller, &user, false).map_err(dispatch_to_evm::<T>)?;586 Ok(())587 }588589 590 591 592 fn set_collection_mint_mode(&mut self, caller: caller, mode: bool) -> Result<()> {593 self.consume_store_reads_and_writes(1, 1)?;594595 let caller = T::CrossAccountId::from_eth(caller);596 let permissions = CollectionPermissions {597 mint_mode: Some(mode),598 ..Default::default()599 };600 <Pallet<T>>::update_permissions(&caller, self, permissions).map_err(dispatch_to_evm::<T>)601 }602603 604 605 606 607 #[solidity(hide, rename_selector = "isOwnerOrAdmin")]608 fn is_owner_or_admin_eth(&self, user: address) -> Result<bool> {609 let user = T::CrossAccountId::from_eth(user);610 Ok(self.is_owner_or_admin(&user))611 }612613 614 615 616 617 fn is_owner_or_admin_cross(&self, user: eth::CrossAddress) -> Result<bool> {618 let user = user.into_sub_cross_account::<T>()?;619 Ok(self.is_owner_or_admin(&user))620 }621622 623 624 625 fn unique_collection_type(&self) -> Result<string> {626 let mode = match self.collection.mode {627 CollectionMode::Fungible(_) => "Fungible",628 CollectionMode::NFT => "NFT",629 CollectionMode::ReFungible => "ReFungible",630 };631 Ok(mode.into())632 }633634 635 636 637 638 fn collection_owner(&self) -> Result<eth::CrossAddress> {639 Ok(eth::CrossAddress::from_sub_cross_account::<T>(640 &T::CrossAccountId::from_sub(self.owner.clone()),641 ))642 }643644 645 646 647 648 #[solidity(hide, rename_selector = "changeCollectionOwner")]649 fn set_owner(&mut self, caller: caller, new_owner: address) -> Result<()> {650 self.consume_store_writes(1)?;651652 let caller = T::CrossAccountId::from_eth(caller);653 let new_owner = T::CrossAccountId::from_eth(new_owner);654 self.change_owner(caller, new_owner)655 .map_err(dispatch_to_evm::<T>)656 }657658 659 660 661 662 fn collection_admins(&self) -> Result<Vec<eth::CrossAddress>> {663 let result = crate::IsAdmin::<T>::iter_prefix((self.id,))664 .map(|(admin, _)| eth::CrossAddress::from_sub_cross_account::<T>(&admin))665 .collect();666 Ok(result)667 }668669 670 671 672 673 fn change_collection_owner_cross(674 &mut self,675 caller: caller,676 new_owner: eth::CrossAddress,677 ) -> Result<()> {678 self.consume_store_writes(1)?;679680 let caller = T::CrossAccountId::from_eth(caller);681 let new_owner = new_owner.into_sub_cross_account::<T>()?;682 self.change_owner(caller, new_owner)683 .map_err(dispatch_to_evm::<T>)684 }685}686687688pub mod static_property {689 use evm_coder::{690 execution::{Result, Error},691 };692 use alloc::format;693694 const EXPECT_CONVERT_ERROR: &str = "length < limit";695696 697 pub mod key {698 use super::*;699700 701 pub fn base_uri() -> up_data_structs::PropertyKey {702 property_key_from_bytes(b"baseURI").expect(EXPECT_CONVERT_ERROR)703 }704705 706 pub fn url() -> up_data_structs::PropertyKey {707 property_key_from_bytes(b"URI").expect(EXPECT_CONVERT_ERROR)708 }709710 711 pub fn suffix() -> up_data_structs::PropertyKey {712 property_key_from_bytes(b"URISuffix").expect(EXPECT_CONVERT_ERROR)713 }714715 716 pub fn parent_nft() -> up_data_structs::PropertyKey {717 property_key_from_bytes(b"parentNft").expect(EXPECT_CONVERT_ERROR)718 }719 }720721 722 pub fn property_key_from_bytes(bytes: &[u8]) -> Result<up_data_structs::PropertyKey> {723 bytes.to_vec().try_into().map_err(|_| {724 Error::Revert(format!(725 "Property key is too long. Max length is {}.",726 up_data_structs::PropertyKey::bound()727 ))728 })729 }730731 732 pub fn property_value_from_bytes(bytes: &[u8]) -> Result<up_data_structs::PropertyValue> {733 bytes.to_vec().try_into().map_err(|_| {734 Error::Revert(format!(735 "Property key is too long. Max length is {}.",736 up_data_structs::PropertyKey::bound()737 ))738 })739 }740}