12345678910111213141516171819use evm_coder::{20 AbiCoder,21 types::{uint256, address},22};23pub use pallet_evm::{Config, account::CrossAccountId};24use sp_core::H160;25use up_data_structs::CollectionId;26272829const ETH_COLLECTION_PREFIX: [u8; 16] = [30 0x17, 0xc4, 0xe6, 0x45, 0x3c, 0xc4, 0x9a, 0xaa, 0xae, 0xac, 0xa8, 0x94, 0xe6, 0xd9, 0x68, 0x3e,31];323334pub fn map_eth_to_id(eth: &H160) -> Option<CollectionId> {35 if eth[0..16] != ETH_COLLECTION_PREFIX {36 return None;37 }38 let mut id_bytes = [0; 4];39 id_bytes.copy_from_slice(ð[16..20]);40 Some(CollectionId(u32::from_be_bytes(id_bytes)))41}424344pub fn collection_id_to_address(id: CollectionId) -> H160 {45 let mut out = [0; 20];46 out[0..16].copy_from_slice(Ð_COLLECTION_PREFIX);47 out[16..20].copy_from_slice(&u32::to_be_bytes(id.0));48 H160(out)49}505152pub fn is_collection(address: &H160) -> bool {53 address[0..16] == ETH_COLLECTION_PREFIX54}555657pub fn convert_uint256_to_cross_account<T: Config>(from: uint256) -> T::CrossAccountId58where59 T::AccountId: From<[u8; 32]>,60{61 let mut new_admin_arr = [0_u8; 32];62 from.to_big_endian(&mut new_admin_arr);63 let account_id = T::AccountId::from(new_admin_arr);64 T::CrossAccountId::from_sub(account_id)65}66676869707172pub fn convert_tuple_to_cross_account<T: Config>(73 eth_cross_account_id: (address, uint256),74) -> evm_coder::execution::Result<T::CrossAccountId>75where76 T::AccountId: From<[u8; 32]>,77{78 if eth_cross_account_id == Default::default() {79 Err("All fields of cross account is zeroed".into())80 } else if eth_cross_account_id.0 == Default::default() {81 Ok(convert_uint256_to_cross_account::<T>(82 eth_cross_account_id.1,83 ))84 } else if eth_cross_account_id.1 == Default::default() {85 Ok(T::CrossAccountId::from_eth(eth_cross_account_id.0))86 } else {87 Err("All fields of cross account is non zeroed".into())88 }89}909192#[derive(Debug, Default, AbiCoder)]93pub struct EthCrossAccount {94 pub(crate) eth: address,95 pub(crate) sub: uint256,96}9798impl EthCrossAccount {99 100 pub fn from_sub_cross_account<T>(cross_account_id: &T::CrossAccountId) -> Self101 where102 T: pallet_evm::Config,103 T::AccountId: AsRef<[u8; 32]>,104 {105 if cross_account_id.is_canonical_substrate() {106 Self::from_sub::<T>(cross_account_id.as_sub())107 } else {108 Self {109 eth: *cross_account_id.as_eth(),110 sub: Default::default(),111 }112 }113 }114 115 pub fn from_sub<T>(account_id: &T::AccountId) -> Self116 where117 T: pallet_evm::Config,118 T::AccountId: AsRef<[u8; 32]>,119 {120 Self {121 eth: Default::default(),122 sub: uint256::from_big_endian(account_id.as_ref()),123 }124 }125 126 pub fn into_sub_cross_account<T>(&self) -> evm_coder::execution::Result<T::CrossAccountId>127 where128 T: pallet_evm::Config,129 T::AccountId: From<[u8; 32]>,130 {131 if self.eth == Default::default() && self.sub == Default::default() {132 Err("All fields of cross account is zeroed".into())133 } else if self.eth == Default::default() {134 Ok(convert_uint256_to_cross_account::<T>(self.sub))135 } else if self.sub == Default::default() {136 Ok(T::CrossAccountId::from_eth(self.eth))137 } else {138 Err("All fields of cross account is non zeroed".into())139 }140 }141}142143144#[derive(Debug, Default, Clone, Copy, AbiCoder)]145#[repr(u8)]146pub enum CollectionLimits {147 148 #[default]149 AccountTokenOwnership,150151 152 SponsoredDataSize,153154 155 SponsoredDataRateLimit,156157 158 TokenLimit,159160 161 SponsorTransferTimeout,162163 164 SponsorApproveTimeout,165166 167 OwnerCanTransfer,168169 170 OwnerCanDestroy,171172 173 TransferEnabled,174}175176#[derive(Default, Debug, Clone, Copy, AbiCoder)]177#[repr(u8)]178pub enum CollectionPermissions {179 180 #[default]181 TokenOwner,182183 184 CollectionAdmin,185}186187188#[derive(AbiCoder, Copy, Clone, Default, Debug)]189#[repr(u8)]190pub enum EthTokenPermissions {191 192 #[default]193 Mutable,194195 196 TokenOwner,197198 199 CollectionAdmin,200}