1use core::char::{REPLACEMENT_CHARACTER, decode_utf16};2use core::convert::TryInto;3use evm_coder::{ToLog, execution::*, generate_stubgen, solidity_interface, types::*};4use nft_data_structs::CollectionMode;5use pallet_common::erc::CommonEvmHandler;6use sp_core::{H160, U256};7use sp_std::vec::Vec;8use pallet_common::account::CrossAccountId;9use pallet_common::erc::PrecompileOutput;10use pallet_evm_coder_substrate::{call_internal, dispatch_to_evm};1112use crate::{Allowance, Balance, Config, FungibleHandle, Pallet, TotalSupply};1314#[derive(ToLog)]15pub enum ERC20Events {16 Transfer {17 #[indexed]18 from: address,19 #[indexed]20 to: address,21 value: uint256,22 },23 Approval {24 #[indexed]25 owner: address,26 #[indexed]27 spender: address,28 value: uint256,29 },30}3132#[solidity_interface(name = "ERC20", events(ERC20Events))]33impl<T: Config> FungibleHandle<T> {34 fn name(&self) -> Result<string> {35 Ok(decode_utf16(self.name.iter().copied())36 .map(|r| r.unwrap_or(REPLACEMENT_CHARACTER))37 .collect::<string>())38 }39 fn symbol(&self) -> Result<string> {40 Ok(string::from_utf8_lossy(&self.token_prefix).into())41 }42 fn total_supply(&self) -> Result<uint256> {43 Ok(<TotalSupply<T>>::get(self.id).into())44 }4546 fn decimals(&self) -> Result<uint8> {47 Ok(if let CollectionMode::Fungible(decimals) = &self.mode {48 *decimals49 } else {50 unreachable!()51 })52 }53 fn balance_of(&self, owner: address) -> Result<uint256> {54 let owner = T::CrossAccountId::from_eth(owner);55 let balance = <Balance<T>>::get((self.id, owner));56 Ok(balance.into())57 }58 fn transfer(&mut self, caller: caller, to: address, amount: uint256) -> Result<bool> {59 let caller = T::CrossAccountId::from_eth(caller);60 let to = T::CrossAccountId::from_eth(to);61 let amount = amount.try_into().map_err(|_| "amount overflow")?;6263 <Pallet<T>>::transfer(self, &caller, &to, amount).map_err(|_| "transfer error")?;64 Ok(true)65 }66 fn transfer_from(67 &mut self,68 caller: caller,69 from: address,70 to: address,71 amount: uint256,72 ) -> Result<bool> {73 let caller = T::CrossAccountId::from_eth(caller);74 let from = T::CrossAccountId::from_eth(from);75 let to = T::CrossAccountId::from_eth(to);76 let amount = amount.try_into().map_err(|_| "amount overflow")?;7778 <Pallet<T>>::transfer_from(self, &caller, &from, &to, amount)79 .map_err(dispatch_to_evm::<T>)?;80 Ok(true)81 }82 fn approve(&mut self, caller: caller, spender: address, amount: uint256) -> Result<bool> {83 let caller = T::CrossAccountId::from_eth(caller);84 let spender = T::CrossAccountId::from_eth(spender);85 let amount = amount.try_into().map_err(|_| "amount overflow")?;8687 <Pallet<T>>::set_allowance(self, &caller, &spender, amount)88 .map_err(dispatch_to_evm::<T>)?;89 Ok(true)90 }91 fn allowance(&self, owner: address, spender: address) -> Result<uint256> {92 let owner = T::CrossAccountId::from_eth(owner);93 let spender = T::CrossAccountId::from_eth(spender);9495 Ok(<Allowance<T>>::get((self.id, owner, spender)).into())96 }97}9899#[solidity_interface(name = "ERC20UniqueExtensions")]100impl<T: Config> FungibleHandle<T> {101 fn burn_from(&mut self, caller: caller, from: address, amount: uint256) -> Result<bool> {102 let caller = T::CrossAccountId::from_eth(caller);103 let from = T::CrossAccountId::from_eth(from);104 let amount = amount.try_into().map_err(|_| "amount overflow")?;105106 <Pallet<T>>::burn_from(self, &caller, &from, amount).map_err(dispatch_to_evm::<T>)?;107 Ok(true)108 }109}110111#[solidity_interface(name = "UniqueFungible", is(ERC20))]112impl<T: Config> FungibleHandle<T> {}113114generate_stubgen!(gen_impl, UniqueFungibleCall, true);115generate_stubgen!(gen_iface, UniqueFungibleCall, false);116117impl<T: Config> CommonEvmHandler for FungibleHandle<T> {118 const CODE: &'static [u8] = include_bytes!("./stubs/UniqueFungible.raw");119120 fn call(mut self, source: &H160, input: &[u8], value: U256) -> Option<PrecompileOutput> {121 let result = call_internal::<UniqueFungibleCall, _>(*source, &mut self, value, input);122 self.0.recorder.evm_to_precompile_output(result)123 }124}