12345678910111213141516171819202122use core::{23 char::{REPLACEMENT_CHARACTER, decode_utf16},24 convert::TryInto,25 ops::Deref,26};27use evm_coder::{28 abi::AbiType, ToLog, execution::*, generate_stubgen, solidity_interface, solidity, types::*,29 weight,30};31use pallet_common::{32 CommonWeightInfo,33 erc::{CommonEvmHandler, PrecompileResult},34 eth::collection_id_to_address,35};36use pallet_evm::{account::CrossAccountId, PrecompileHandle};37use pallet_evm_coder_substrate::{call, dispatch_to_evm, WithRecorder};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, common::CommonWeights, Config, Pallet, RefungibleHandle, SelfWeightOf,45 TotalSupply, weights::WeightInfo,46};4748495051pub struct RefungibleTokenHandle<T: Config>(pub RefungibleHandle<T>, pub TokenId);5253#[solidity_interface(name = ERC1633)]54impl<T: Config> RefungibleTokenHandle<T> {55 fn parent_token(&self) -> Result<Address> {56 Ok(collection_id_to_address(self.id))57 }5859 fn parent_token_id(&self) -> Result<U256> {60 Ok(self.1.into())61 }62}6364#[derive(ToLog)]65pub enum ERC20Events {66 67 68 69 70 Transfer {71 #[indexed]72 from: Address,73 #[indexed]74 to: Address,75 value: U256,76 },77 78 79 Approval {80 #[indexed]81 owner: Address,82 #[indexed]83 spender: Address,84 value: U256,85 },86}878889909192#[solidity_interface(name = ERC20, events(ERC20Events))]93impl<T: Config> RefungibleTokenHandle<T> {94 95 fn name(&self) -> Result<String> {96 Ok(decode_utf16(self.name.iter().copied())97 .map(|r| r.unwrap_or(REPLACEMENT_CHARACTER))98 .collect::<String>())99 }100101 102 fn symbol(&self) -> Result<String> {103 Ok(String::from_utf8_lossy(&self.token_prefix).into())104 }105106 107 fn total_supply(&self) -> Result<U256> {108 self.consume_store_reads(1)?;109 Ok(<TotalSupply<T>>::get((self.id, self.1)).into())110 }111112 113 fn decimals(&self) -> Result<u8> {114 115 Ok(0)116 }117118 119 120 121 fn balance_of(&self, owner: Address) -> Result<U256> {122 self.consume_store_reads(1)?;123 let owner = T::CrossAccountId::from_eth(owner);124 let balance = <Balance<T>>::get((self.id, self.1, owner));125 Ok(balance.into())126 }127128 129 130 131 #[weight(<CommonWeights<T>>::transfer())]132 fn transfer(&mut self, caller: Caller, to: Address, amount: U256) -> Result<bool> {133 let caller = T::CrossAccountId::from_eth(caller);134 let to = T::CrossAccountId::from_eth(to);135 let amount = amount.try_into().map_err(|_| "amount overflow")?;136 let budget = self137 .recorder138 .weight_calls_budget(<StructureWeight<T>>::find_parent());139140 <Pallet<T>>::transfer(self, &caller, &to, self.1, amount, &budget)141 .map_err(dispatch_to_evm::<T>)?;142 Ok(true)143 }144145 146 147 148 149 #[weight(<CommonWeights<T>>::transfer_from())]150 fn transfer_from(151 &mut self,152 caller: Caller,153 from: Address,154 to: Address,155 amount: U256,156 ) -> Result<bool> {157 let caller = T::CrossAccountId::from_eth(caller);158 let from = T::CrossAccountId::from_eth(from);159 let to = T::CrossAccountId::from_eth(to);160 let amount = amount.try_into().map_err(|_| "amount overflow")?;161 let budget = self162 .recorder163 .weight_calls_budget(<StructureWeight<T>>::find_parent());164165 <Pallet<T>>::transfer_from(self, &caller, &from, &to, self.1, amount, &budget)166 .map_err(dispatch_to_evm::<T>)?;167 Ok(true)168 }169170 171 172 173 174 175 176 177 #[weight(<SelfWeightOf<T>>::approve())]178 fn approve(&mut self, caller: Caller, spender: Address, amount: U256) -> Result<bool> {179 let caller = T::CrossAccountId::from_eth(caller);180 let spender = T::CrossAccountId::from_eth(spender);181 let amount = amount.try_into().map_err(|_| "amount overflow")?;182183 <Pallet<T>>::set_allowance(self, &caller, &spender, self.1, amount)184 .map_err(dispatch_to_evm::<T>)?;185 Ok(true)186 }187188 189 190 191 192 fn allowance(&self, owner: Address, spender: Address) -> Result<U256> {193 self.consume_store_reads(1)?;194 let owner = T::CrossAccountId::from_eth(owner);195 let spender = T::CrossAccountId::from_eth(spender);196197 Ok(<Allowance<T>>::get((self.id, self.1, owner, spender)).into())198 }199}200201#[solidity_interface(name = ERC20UniqueExtensions)]202impl<T: Config> RefungibleTokenHandle<T>203where204 T::AccountId: From<[u8; 32]>,205{206 207 208 209 210 fn allowance_cross(211 &self,212 owner: pallet_common::eth::CrossAddress,213 spender: pallet_common::eth::CrossAddress,214 ) -> Result<U256> {215 let owner = owner.into_sub_cross_account::<T>()?;216 let spender = spender.into_sub_cross_account::<T>()?;217218 Ok(<Allowance<T>>::get((self.id, self.1, owner, spender)).into())219 }220221 222 223 224 225 #[weight(<SelfWeightOf<T>>::burn_from())]226 #[solidity(hide)]227 fn burn_from(&mut self, caller: Caller, from: Address, amount: U256) -> Result<bool> {228 let caller = T::CrossAccountId::from_eth(caller);229 let from = T::CrossAccountId::from_eth(from);230 let amount = amount.try_into().map_err(|_| "amount overflow")?;231 let budget = self232 .recorder233 .weight_calls_budget(<StructureWeight<T>>::find_parent());234235 <Pallet<T>>::burn_from(self, &caller, &from, self.1, amount, &budget)236 .map_err(dispatch_to_evm::<T>)?;237 Ok(true)238 }239240 241 242 243 244 #[weight(<SelfWeightOf<T>>::burn_from())]245 fn burn_from_cross(246 &mut self,247 caller: Caller,248 from: pallet_common::eth::CrossAddress,249 amount: U256,250 ) -> Result<bool> {251 let caller = T::CrossAccountId::from_eth(caller);252 let from = from.into_sub_cross_account::<T>()?;253 let amount = amount.try_into().map_err(|_| "amount overflow")?;254 let budget = self255 .recorder256 .weight_calls_budget(<StructureWeight<T>>::find_parent());257258 <Pallet<T>>::burn_from(self, &caller, &from, self.1, amount, &budget)259 .map_err(dispatch_to_evm::<T>)?;260 Ok(true)261 }262263 264 265 266 267 268 269 270 #[weight(<SelfWeightOf<T>>::approve())]271 fn approve_cross(272 &mut self,273 caller: Caller,274 spender: pallet_common::eth::CrossAddress,275 amount: U256,276 ) -> Result<bool> {277 let caller = T::CrossAccountId::from_eth(caller);278 let spender = spender.into_sub_cross_account::<T>()?;279 let amount = amount.try_into().map_err(|_| "amount overflow")?;280281 <Pallet<T>>::set_allowance(self, &caller, &spender, self.1, amount)282 .map_err(dispatch_to_evm::<T>)?;283 Ok(true)284 }285 286 287 288 #[weight(<SelfWeightOf<T>>::repartition_item())]289 fn repartition(&mut self, caller: Caller, amount: U256) -> Result<bool> {290 let caller = T::CrossAccountId::from_eth(caller);291 let amount = amount.try_into().map_err(|_| "amount overflow")?;292293 <Pallet<T>>::repartition(self, &caller, self.1, amount).map_err(dispatch_to_evm::<T>)?;294 Ok(true)295 }296297 298 299 300 #[weight(<CommonWeights<T>>::transfer())]301 fn transfer_cross(302 &mut self,303 caller: Caller,304 to: pallet_common::eth::CrossAddress,305 amount: U256,306 ) -> Result<bool> {307 let caller = T::CrossAccountId::from_eth(caller);308 let to = to.into_sub_cross_account::<T>()?;309 let amount = amount.try_into().map_err(|_| "amount overflow")?;310 let budget = self311 .recorder312 .weight_calls_budget(<StructureWeight<T>>::find_parent());313314 <Pallet<T>>::transfer(self, &caller, &to, self.1, amount, &budget)315 .map_err(dispatch_to_evm::<T>)?;316 Ok(true)317 }318319 320 321 322 323 #[weight(<CommonWeights<T>>::transfer_from())]324 fn transfer_from_cross(325 &mut self,326 caller: Caller,327 from: pallet_common::eth::CrossAddress,328 to: pallet_common::eth::CrossAddress,329 amount: U256,330 ) -> Result<bool> {331 let caller = T::CrossAccountId::from_eth(caller);332 let from = from.into_sub_cross_account::<T>()?;333 let to = to.into_sub_cross_account::<T>()?;334 let amount = amount.try_into().map_err(|_| "amount overflow")?;335 let budget = self336 .recorder337 .weight_calls_budget(<StructureWeight<T>>::find_parent());338339 <Pallet<T>>::transfer_from(self, &caller, &from, &to, self.1, amount, &budget)340 .map_err(dispatch_to_evm::<T>)?;341 Ok(true)342 }343}344345impl<T: Config> RefungibleTokenHandle<T> {346 pub fn into_inner(self) -> RefungibleHandle<T> {347 self.0348 }349 pub fn common_mut(&mut self) -> &mut RefungibleHandle<T> {350 &mut self.0351 }352}353354impl<T: Config> WithRecorder<T> for RefungibleTokenHandle<T> {355 fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {356 self.0.recorder()357 }358 fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {359 self.0.into_recorder()360 }361}362363impl<T: Config> Deref for RefungibleTokenHandle<T> {364 type Target = RefungibleHandle<T>;365366 fn deref(&self) -> &Self::Target {367 &self.0368 }369}370371#[solidity_interface(372 name = UniqueRefungibleToken,373 is(ERC20, ERC20UniqueExtensions, ERC1633)374)]375impl<T: Config> RefungibleTokenHandle<T> where T::AccountId: From<[u8; 32]> {}376377generate_stubgen!(gen_impl, UniqueRefungibleTokenCall<()>, true);378generate_stubgen!(gen_iface, UniqueRefungibleTokenCall<()>, false);379380impl<T: Config> CommonEvmHandler for RefungibleTokenHandle<T>381where382 T::AccountId: From<[u8; 32]>,383{384 const CODE: &'static [u8] = include_bytes!("./stubs/UniqueRefungibleToken.raw");385386 fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult> {387 call::<T, UniqueRefungibleTokenCall<T>, _, _>(handle, self)388 }389}