12345678910111213141516171819202122use core::{23 char::{REPLACEMENT_CHARACTER, decode_utf16},24 convert::TryInto,25 ops::Deref,26};27use evm_coder::{abi::AbiType, ToLog, generate_stubgen, solidity_interface, types::*};28use pallet_common::{29 erc::{CommonEvmHandler, PrecompileResult},30 eth::{collection_id_to_address, CrossAddress},31 CommonWeightInfo,32};33use pallet_evm::{account::CrossAccountId, PrecompileHandle};34use pallet_evm_coder_substrate::{35 call, dispatch_to_evm, WithRecorder, frontier_contract,36 execution::{Result, PreDispatch},37};38use pallet_structure::{SelfWeightOf as StructureWeight, weights::WeightInfo as _};39use sp_std::vec::Vec;40use sp_core::U256;41use up_data_structs::TokenId;4243use crate::{44 Allowance, Balance, Config, Pallet, RefungibleHandle, TotalSupply, common::CommonWeights,45 SelfWeightOf, weights::WeightInfo,46};4748495051pub struct RefungibleTokenHandle<T: Config>(pub RefungibleHandle<T>, pub TokenId);5253frontier_contract! {54 macro_rules! RefungibleTokenHandle_result {...}55 impl<T: Config> Contract for RefungibleTokenHandle<T> {...}56}5758#[solidity_interface(name = ERC1633, enum(derive(PreDispatch)), enum_attr(weight))]59impl<T: Config> RefungibleTokenHandle<T> {60 fn parent_token(&self) -> Address {61 collection_id_to_address(self.id)62 }6364 fn parent_token_id(&self) -> U256 {65 self.1.into()66 }67}6869#[derive(ToLog)]70pub enum ERC20Events {71 72 73 74 75 Transfer {76 #[indexed]77 from: Address,78 #[indexed]79 to: Address,80 value: U256,81 },82 83 84 Approval {85 #[indexed]86 owner: Address,87 #[indexed]88 spender: Address,89 value: U256,90 },91}929394959697#[solidity_interface(name = ERC20, events(ERC20Events), enum(derive(PreDispatch)), enum_attr(weight))]98impl<T: Config> RefungibleTokenHandle<T> {99 100 fn name(&self) -> String {101 decode_utf16(self.name.iter().copied())102 .map(|r| r.unwrap_or(REPLACEMENT_CHARACTER))103 .collect::<String>()104 }105106 107 fn symbol(&self) -> String {108 String::from_utf8_lossy(&self.token_prefix).into()109 }110111 112 fn total_supply(&self) -> Result<U256> {113 self.consume_store_reads(1)?;114 Ok(<TotalSupply<T>>::get((self.id, self.1)).into())115 }116117 118 fn decimals(&self) -> Result<u8> {119 120 Ok(0)121 }122123 124 125 126 fn balance_of(&self, owner: Address) -> Result<U256> {127 self.consume_store_reads(1)?;128 let owner = T::CrossAccountId::from_eth(owner);129 let balance = <Balance<T>>::get((self.id, self.1, owner));130 Ok(balance.into())131 }132133 134 135 136 #[weight(<CommonWeights<T>>::transfer())]137 fn transfer(&mut self, caller: Caller, to: Address, amount: U256) -> Result<bool> {138 let caller = T::CrossAccountId::from_eth(caller);139 let to = T::CrossAccountId::from_eth(to);140 let amount = amount.try_into().map_err(|_| "amount overflow")?;141 let budget = self142 .recorder143 .weight_calls_budget(<StructureWeight<T>>::find_parent());144145 <Pallet<T>>::transfer(self, &caller, &to, self.1, amount, &budget)146 .map_err(dispatch_to_evm::<T>)?;147 Ok(true)148 }149150 151 152 153 154 #[weight(<CommonWeights<T>>::transfer_from())]155 fn transfer_from(156 &mut self,157 caller: Caller,158 from: Address,159 to: Address,160 amount: U256,161 ) -> Result<bool> {162 let caller = T::CrossAccountId::from_eth(caller);163 let from = T::CrossAccountId::from_eth(from);164 let to = T::CrossAccountId::from_eth(to);165 let amount = amount.try_into().map_err(|_| "amount overflow")?;166 let budget = self167 .recorder168 .weight_calls_budget(<StructureWeight<T>>::find_parent());169170 <Pallet<T>>::transfer_from(self, &caller, &from, &to, self.1, amount, &budget)171 .map_err(dispatch_to_evm::<T>)?;172 Ok(true)173 }174175 176 177 178 179 180 181 182 #[weight(<SelfWeightOf<T>>::approve())]183 fn approve(&mut self, caller: Caller, spender: Address, amount: U256) -> Result<bool> {184 let caller = T::CrossAccountId::from_eth(caller);185 let spender = T::CrossAccountId::from_eth(spender);186 let amount = amount.try_into().map_err(|_| "amount overflow")?;187188 <Pallet<T>>::set_allowance(self, &caller, &spender, self.1, amount)189 .map_err(dispatch_to_evm::<T>)?;190 Ok(true)191 }192193 194 195 196 197 fn allowance(&self, owner: Address, spender: Address) -> Result<U256> {198 self.consume_store_reads(1)?;199 let owner = T::CrossAccountId::from_eth(owner);200 let spender = T::CrossAccountId::from_eth(spender);201202 Ok(<Allowance<T>>::get((self.id, self.1, owner, spender)).into())203 }204}205206#[solidity_interface(name = ERC20UniqueExtensions, enum(derive(PreDispatch)), enum_attr(weight))]207impl<T: Config> RefungibleTokenHandle<T>208where209 T::AccountId: From<[u8; 32]>,210{211 212 213 214 215 fn allowance_cross(&self, owner: CrossAddress, spender: CrossAddress) -> Result<U256> {216 let owner = owner.into_sub_cross_account::<T>()?;217 let spender = spender.into_sub_cross_account::<T>()?;218219 Ok(<Allowance<T>>::get((self.id, self.1, owner, spender)).into())220 }221222 223 224 225 226 #[weight(<SelfWeightOf<T>>::burn_from())]227 #[solidity(hide)]228 fn burn_from(&mut self, caller: Caller, from: Address, amount: U256) -> Result<bool> {229 let caller = T::CrossAccountId::from_eth(caller);230 let from = T::CrossAccountId::from_eth(from);231 let amount = amount.try_into().map_err(|_| "amount overflow")?;232 let budget = self233 .recorder234 .weight_calls_budget(<StructureWeight<T>>::find_parent());235236 <Pallet<T>>::burn_from(self, &caller, &from, self.1, amount, &budget)237 .map_err(dispatch_to_evm::<T>)?;238 Ok(true)239 }240241 242 243 244 245 #[weight(<SelfWeightOf<T>>::burn_from())]246 fn burn_from_cross(247 &mut self,248 caller: Caller,249 from: CrossAddress,250 amount: U256,251 ) -> Result<bool> {252 let caller = T::CrossAccountId::from_eth(caller);253 let from = from.into_sub_cross_account::<T>()?;254 let amount = amount.try_into().map_err(|_| "amount overflow")?;255 let budget = self256 .recorder257 .weight_calls_budget(<StructureWeight<T>>::find_parent());258259 <Pallet<T>>::burn_from(self, &caller, &from, self.1, amount, &budget)260 .map_err(dispatch_to_evm::<T>)?;261 Ok(true)262 }263264 265 266 267 268 269 270 271 #[weight(<SelfWeightOf<T>>::approve())]272 fn approve_cross(273 &mut self,274 caller: Caller,275 spender: CrossAddress,276 amount: U256,277 ) -> Result<bool> {278 let caller = T::CrossAccountId::from_eth(caller);279 let spender = spender.into_sub_cross_account::<T>()?;280 let amount = amount.try_into().map_err(|_| "amount overflow")?;281282 <Pallet<T>>::set_allowance(self, &caller, &spender, self.1, amount)283 .map_err(dispatch_to_evm::<T>)?;284 Ok(true)285 }286287 288 289 290 fn balance_of_cross(&self, owner: CrossAddress) -> Result<U256> {291 self.consume_store_reads(1)?;292 let balance = <Balance<T>>::get((self.id, self.1, owner.into_sub_cross_account::<T>()?));293 Ok(balance.into())294 }295296 297 298 299 #[weight(<SelfWeightOf<T>>::repartition_item())]300 fn repartition(&mut self, caller: Caller, amount: U256) -> Result<bool> {301 let caller = T::CrossAccountId::from_eth(caller);302 let amount = amount.try_into().map_err(|_| "amount overflow")?;303304 <Pallet<T>>::repartition(self, &caller, self.1, amount).map_err(dispatch_to_evm::<T>)?;305 Ok(true)306 }307308 309 310 311 #[weight(<CommonWeights<T>>::transfer())]312 fn transfer_cross(&mut self, caller: Caller, to: CrossAddress, amount: U256) -> Result<bool> {313 let caller = T::CrossAccountId::from_eth(caller);314 let to = to.into_sub_cross_account::<T>()?;315 let amount = amount.try_into().map_err(|_| "amount overflow")?;316 let budget = self317 .recorder318 .weight_calls_budget(<StructureWeight<T>>::find_parent());319320 <Pallet<T>>::transfer(self, &caller, &to, self.1, amount, &budget)321 .map_err(dispatch_to_evm::<T>)?;322 Ok(true)323 }324325 326 327 328 329 #[weight(<CommonWeights<T>>::transfer_from())]330 fn transfer_from_cross(331 &mut self,332 caller: Caller,333 from: CrossAddress,334 to: CrossAddress,335 amount: U256,336 ) -> Result<bool> {337 let caller = T::CrossAccountId::from_eth(caller);338 let from = from.into_sub_cross_account::<T>()?;339 let to = to.into_sub_cross_account::<T>()?;340 let amount = amount.try_into().map_err(|_| "amount overflow")?;341 let budget = self342 .recorder343 .weight_calls_budget(<StructureWeight<T>>::find_parent());344345 <Pallet<T>>::transfer_from(self, &caller, &from, &to, self.1, amount, &budget)346 .map_err(dispatch_to_evm::<T>)?;347 Ok(true)348 }349}350351impl<T: Config> RefungibleTokenHandle<T> {352 pub fn into_inner(self) -> RefungibleHandle<T> {353 self.0354 }355 pub fn common_mut(&mut self) -> &mut RefungibleHandle<T> {356 &mut self.0357 }358}359360impl<T: Config> WithRecorder<T> for RefungibleTokenHandle<T> {361 fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {362 self.0.recorder()363 }364 fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {365 self.0.into_recorder()366 }367}368369impl<T: Config> Deref for RefungibleTokenHandle<T> {370 type Target = RefungibleHandle<T>;371372 fn deref(&self) -> &Self::Target {373 &self.0374 }375}376377#[solidity_interface(378 name = UniqueRefungibleToken,379 is(ERC20, ERC20UniqueExtensions, ERC1633),380 enum(derive(PreDispatch)),381)]382impl<T: Config> RefungibleTokenHandle<T> where T::AccountId: From<[u8; 32]> {}383384generate_stubgen!(gen_impl, UniqueRefungibleTokenCall<()>, true);385generate_stubgen!(gen_iface, UniqueRefungibleTokenCall<()>, false);386387impl<T: Config> CommonEvmHandler for RefungibleTokenHandle<T>388where389 T::AccountId: From<[u8; 32]>,390{391 const CODE: &'static [u8] = include_bytes!("./stubs/UniqueRefungibleToken.raw");392393 fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult> {394 call::<T, UniqueRefungibleTokenCall<T>, _, _>(handle, self)395 }396}