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 }286 287 288 289 #[weight(<SelfWeightOf<T>>::repartition_item())]290 fn repartition(&mut self, caller: Caller, amount: U256) -> Result<bool> {291 let caller = T::CrossAccountId::from_eth(caller);292 let amount = amount.try_into().map_err(|_| "amount overflow")?;293294 <Pallet<T>>::repartition(self, &caller, self.1, amount).map_err(dispatch_to_evm::<T>)?;295 Ok(true)296 }297298 299 300 301 #[weight(<CommonWeights<T>>::transfer())]302 fn transfer_cross(&mut self, caller: Caller, to: CrossAddress, amount: U256) -> Result<bool> {303 let caller = T::CrossAccountId::from_eth(caller);304 let to = to.into_sub_cross_account::<T>()?;305 let amount = amount.try_into().map_err(|_| "amount overflow")?;306 let budget = self307 .recorder308 .weight_calls_budget(<StructureWeight<T>>::find_parent());309310 <Pallet<T>>::transfer(self, &caller, &to, self.1, amount, &budget)311 .map_err(dispatch_to_evm::<T>)?;312 Ok(true)313 }314315 316 317 318 319 #[weight(<CommonWeights<T>>::transfer_from())]320 fn transfer_from_cross(321 &mut self,322 caller: Caller,323 from: CrossAddress,324 to: CrossAddress,325 amount: U256,326 ) -> Result<bool> {327 let caller = T::CrossAccountId::from_eth(caller);328 let from = from.into_sub_cross_account::<T>()?;329 let to = to.into_sub_cross_account::<T>()?;330 let amount = amount.try_into().map_err(|_| "amount overflow")?;331 let budget = self332 .recorder333 .weight_calls_budget(<StructureWeight<T>>::find_parent());334335 <Pallet<T>>::transfer_from(self, &caller, &from, &to, self.1, amount, &budget)336 .map_err(dispatch_to_evm::<T>)?;337 Ok(true)338 }339}340341impl<T: Config> RefungibleTokenHandle<T> {342 pub fn into_inner(self) -> RefungibleHandle<T> {343 self.0344 }345 pub fn common_mut(&mut self) -> &mut RefungibleHandle<T> {346 &mut self.0347 }348}349350impl<T: Config> WithRecorder<T> for RefungibleTokenHandle<T> {351 fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {352 self.0.recorder()353 }354 fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {355 self.0.into_recorder()356 }357}358359impl<T: Config> Deref for RefungibleTokenHandle<T> {360 type Target = RefungibleHandle<T>;361362 fn deref(&self) -> &Self::Target {363 &self.0364 }365}366367#[solidity_interface(368 name = UniqueRefungibleToken,369 is(ERC20, ERC20UniqueExtensions, ERC1633),370 enum(derive(PreDispatch)),371)]372impl<T: Config> RefungibleTokenHandle<T> where T::AccountId: From<[u8; 32]> {}373374generate_stubgen!(gen_impl, UniqueRefungibleTokenCall<()>, true);375generate_stubgen!(gen_iface, UniqueRefungibleTokenCall<()>, false);376377impl<T: Config> CommonEvmHandler for RefungibleTokenHandle<T>378where379 T::AccountId: From<[u8; 32]>,380{381 const CODE: &'static [u8] = include_bytes!("./stubs/UniqueRefungibleToken.raw");382383 fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult> {384 call::<T, UniqueRefungibleTokenCall<T>, _, _>(handle, self)385 }386}