12345678910111213141516171819202122extern crate alloc;2324#[cfg(not(feature = "std"))]25use alloc::format;2627use core::{28 char::{REPLACEMENT_CHARACTER, decode_utf16},29 convert::TryInto,30 ops::Deref,31};32use evm_coder::{ToLog, execution::*, generate_stubgen, solidity_interface, types::*, weight};33use pallet_common::{34 CommonWeightInfo,35 erc::{CommonEvmHandler, PrecompileResult},36 eth::collection_id_to_address,37};38use pallet_evm::{account::CrossAccountId, PrecompileHandle};39use pallet_evm_coder_substrate::{call, dispatch_to_evm, WithRecorder};40use pallet_structure::{SelfWeightOf as StructureWeight, weights::WeightInfo as _};41use sp_std::vec::Vec;42use up_data_structs::TokenId;4344use crate::{45 Allowance, Balance, common::CommonWeights, Config, Pallet, RefungibleHandle, SelfWeightOf,46 TotalSupply, weights::WeightInfo,47};4849pub struct RefungibleTokenHandle<T: Config>(pub RefungibleHandle<T>, pub TokenId);5051#[solidity_interface(name = ERC1633)]52impl<T: Config> RefungibleTokenHandle<T> {53 fn parent_token(&self) -> Result<address> {54 Ok(collection_id_to_address(self.id))55 }5657 fn parent_token_id(&self) -> Result<uint256> {58 Ok(self.1.into())59 }60}6162#[derive(ToLog)]63pub enum ERC20Events {64 65 66 67 68 Transfer {69 #[indexed]70 from: address,71 #[indexed]72 to: address,73 value: uint256,74 },75 76 77 Approval {78 #[indexed]79 owner: address,80 #[indexed]81 spender: address,82 value: uint256,83 },84}858687888990#[solidity_interface(name = ERC20, events(ERC20Events))]91impl<T: Config> RefungibleTokenHandle<T> {92 93 fn name(&self) -> Result<string> {94 Ok(decode_utf16(self.name.iter().copied())95 .map(|r| r.unwrap_or(REPLACEMENT_CHARACTER))96 .collect::<string>())97 }9899 100 fn symbol(&self) -> Result<string> {101 Ok(string::from_utf8_lossy(&self.token_prefix).into())102 }103104 105 fn total_supply(&self) -> Result<uint256> {106 self.consume_store_reads(1)?;107 Ok(<TotalSupply<T>>::get((self.id, self.1)).into())108 }109110 111 fn decimals(&self) -> Result<uint8> {112 113 Ok(0)114 }115116 117 118 119 fn balance_of(&self, owner: address) -> Result<uint256> {120 self.consume_store_reads(1)?;121 let owner = T::CrossAccountId::from_eth(owner);122 let balance = <Balance<T>>::get((self.id, self.1, owner));123 Ok(balance.into())124 }125126 127 128 129 #[weight(<CommonWeights<T>>::transfer())]130 fn transfer(&mut self, caller: caller, to: address, amount: uint256) -> Result<bool> {131 let caller = T::CrossAccountId::from_eth(caller);132 let to = T::CrossAccountId::from_eth(to);133 let amount = amount.try_into().map_err(|_| "amount overflow")?;134 let budget = self135 .recorder136 .weight_calls_budget(<StructureWeight<T>>::find_parent());137138 <Pallet<T>>::transfer(self, &caller, &to, self.1, amount, &budget)139 .map_err(dispatch_to_evm::<T>)?;140 Ok(true)141 }142143 144 145 146 147 #[weight(<CommonWeights<T>>::transfer_from())]148 fn transfer_from(149 &mut self,150 caller: caller,151 from: address,152 to: address,153 amount: uint256,154 ) -> Result<bool> {155 let caller = T::CrossAccountId::from_eth(caller);156 let from = T::CrossAccountId::from_eth(from);157 let to = T::CrossAccountId::from_eth(to);158 let amount = amount.try_into().map_err(|_| "amount overflow")?;159 let budget = self160 .recorder161 .weight_calls_budget(<StructureWeight<T>>::find_parent());162163 <Pallet<T>>::transfer_from(self, &caller, &from, &to, self.1, amount, &budget)164 .map_err(dispatch_to_evm::<T>)?;165 Ok(true)166 }167168 169 170 171 172 173 174 175 #[weight(<SelfWeightOf<T>>::approve())]176 fn approve(&mut self, caller: caller, spender: address, amount: uint256) -> Result<bool> {177 let caller = T::CrossAccountId::from_eth(caller);178 let spender = T::CrossAccountId::from_eth(spender);179 let amount = amount.try_into().map_err(|_| "amount overflow")?;180181 <Pallet<T>>::set_allowance(self, &caller, &spender, self.1, amount)182 .map_err(dispatch_to_evm::<T>)?;183 Ok(true)184 }185186 187 188 189 190 fn allowance(&self, owner: address, spender: address) -> Result<uint256> {191 self.consume_store_reads(1)?;192 let owner = T::CrossAccountId::from_eth(owner);193 let spender = T::CrossAccountId::from_eth(spender);194195 Ok(<Allowance<T>>::get((self.id, self.1, owner, spender)).into())196 }197}198199#[solidity_interface(name = ERC20UniqueExtensions)]200impl<T: Config> RefungibleTokenHandle<T> {201 202 203 204 205 #[weight(<SelfWeightOf<T>>::burn_from())]206 fn burn_from(&mut self, caller: caller, from: address, amount: uint256) -> Result<bool> {207 let caller = T::CrossAccountId::from_eth(caller);208 let from = T::CrossAccountId::from_eth(from);209 let amount = amount.try_into().map_err(|_| "amount overflow")?;210 let budget = self211 .recorder212 .weight_calls_budget(<StructureWeight<T>>::find_parent());213214 <Pallet<T>>::burn_from(self, &caller, &from, self.1, amount, &budget)215 .map_err(dispatch_to_evm::<T>)?;216 Ok(true)217 }218219 220 221 222 #[weight(<SelfWeightOf<T>>::repartition_item())]223 fn repartition(&mut self, caller: caller, amount: uint256) -> Result<bool> {224 let caller = T::CrossAccountId::from_eth(caller);225 let amount = amount.try_into().map_err(|_| "amount overflow")?;226227 <Pallet<T>>::repartition(self, &caller, self.1, amount).map_err(dispatch_to_evm::<T>)?;228 Ok(true)229 }230}231232impl<T: Config> RefungibleTokenHandle<T> {233 pub fn into_inner(self) -> RefungibleHandle<T> {234 self.0235 }236 pub fn common_mut(&mut self) -> &mut RefungibleHandle<T> {237 &mut self.0238 }239}240241impl<T: Config> WithRecorder<T> for RefungibleTokenHandle<T> {242 fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {243 self.0.recorder()244 }245 fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {246 self.0.into_recorder()247 }248}249250impl<T: Config> Deref for RefungibleTokenHandle<T> {251 type Target = RefungibleHandle<T>;252253 fn deref(&self) -> &Self::Target {254 &self.0255 }256}257258#[solidity_interface(259 name = UniqueRefungibleToken,260 is(ERC20, ERC20UniqueExtensions, ERC1633)261)]262impl<T: Config> RefungibleTokenHandle<T> where T::AccountId: From<[u8; 32]> {}263264generate_stubgen!(gen_impl, UniqueRefungibleTokenCall<()>, true);265generate_stubgen!(gen_iface, UniqueRefungibleTokenCall<()>, false);266267impl<T: Config> CommonEvmHandler for RefungibleTokenHandle<T>268where269 T::AccountId: From<[u8; 32]>,270{271 const CODE: &'static [u8] = include_bytes!("./stubs/UniqueRefungibleToken.raw");272273 fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult> {274 call::<T, UniqueRefungibleTokenCall<T>, _, _>(handle, self)275 }276}