git.delta.rocks / unique-network / refs/commits / 011ebf22fd46

1use core::marker::PhantomData;23use sp_core::H160;45use crate::{CollectionId, TokenId};6use pallet_evm::account::CrossAccountId;78pub trait TokenAddressMapping<Address> {9	fn token_to_address(collection: CollectionId, token: TokenId) -> Address;10	fn address_to_token(address: &Address) -> Option<(CollectionId, TokenId)>;11	fn is_token_address(address: &Address) -> bool;12}1314pub struct EvmTokenAddressMapping;1516/// 0xf8238ccfff8ed887463fd5e00000000100000002  - collection 1, token 217const ETH_COLLECTION_TOKEN_PREFIX: [u8; 12] = [18	0xf8, 0x23, 0x8c, 0xcf, 0xff, 0x8e, 0xd8, 0x87, 0x46, 0x3f, 0xd5, 0xe0,19];2021impl TokenAddressMapping<H160> for EvmTokenAddressMapping {22	fn token_to_address(collection: CollectionId, token: TokenId) -> H160 {23		let mut out = [0; 20];24		out[0..12].copy_from_slice(&ETH_COLLECTION_TOKEN_PREFIX);25		out[12..16].copy_from_slice(&u32::to_be_bytes(collection.0));26		out[16..20].copy_from_slice(&u32::to_be_bytes(token.0));27		H160(out)28	}2930	fn address_to_token(eth: &H160) -> Option<(CollectionId, TokenId)> {31		if eth[0..12] != ETH_COLLECTION_TOKEN_PREFIX {32			return None;33		}34		let mut id_bytes = [0; 4];35		let mut token_id_bytes = [0; 4];36		id_bytes.copy_from_slice(&eth[12..16]);37		token_id_bytes.copy_from_slice(&eth[16..20]);38		Some((39			CollectionId(u32::from_be_bytes(id_bytes)),40			TokenId(u32::from_be_bytes(token_id_bytes)),41		))42	}4344	fn is_token_address(address: &H160) -> bool {45		address[0..12] == ETH_COLLECTION_TOKEN_PREFIX46	}47}4849pub struct CrossTokenAddressMapping<A>(PhantomData<A>);5051impl<A, C: CrossAccountId<A>> TokenAddressMapping<C> for CrossTokenAddressMapping<A> {52	fn token_to_address(collection: CollectionId, token: TokenId) -> C {53		C::from_eth(EvmTokenAddressMapping::token_to_address(collection, token))54	}5556	fn address_to_token(address: &C) -> Option<(CollectionId, TokenId)> {57		EvmTokenAddressMapping::address_to_token(address.as_eth())58	}5960	fn is_token_address(address: &C) -> bool {61		EvmTokenAddressMapping::is_token_address(address.as_eth())62	}63}