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, types::*, weight,29};30use pallet_common::{31 CommonWeightInfo,32 erc::{CommonEvmHandler, PrecompileResult},33 eth::collection_id_to_address,34};35use pallet_evm::{account::CrossAccountId, PrecompileHandle};36use pallet_evm_coder_substrate::{call, dispatch_to_evm, WithRecorder};37use pallet_structure::{SelfWeightOf as StructureWeight, weights::WeightInfo as _};38use sp_std::vec::Vec;39use up_data_structs::TokenId;4041use crate::{42 Allowance, Balance, common::CommonWeights, Config, Pallet, RefungibleHandle, SelfWeightOf,43 TotalSupply, weights::WeightInfo,44};4546474849pub 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>201where202 T::AccountId: From<[u8; 32]>,203{204 205 206 207 208 #[weight(<SelfWeightOf<T>>::burn_from())]209 fn burn_from(&mut self, caller: caller, from: address, amount: uint256) -> Result<bool> {210 let caller = T::CrossAccountId::from_eth(caller);211 let from = T::CrossAccountId::from_eth(from);212 let amount = amount.try_into().map_err(|_| "amount overflow")?;213 let budget = self214 .recorder215 .weight_calls_budget(<StructureWeight<T>>::find_parent());216217 <Pallet<T>>::burn_from(self, &caller, &from, self.1, amount, &budget)218 .map_err(dispatch_to_evm::<T>)?;219 Ok(true)220 }221222 223 224 225 226 #[weight(<SelfWeightOf<T>>::burn_from())]227 fn burn_from_cross(228 &mut self,229 caller: caller,230 from: pallet_common::eth::CrossAddress,231 amount: uint256,232 ) -> Result<bool> {233 let caller = T::CrossAccountId::from_eth(caller);234 let from = from.into_sub_cross_account::<T>()?;235 let amount = amount.try_into().map_err(|_| "amount overflow")?;236 let budget = self237 .recorder238 .weight_calls_budget(<StructureWeight<T>>::find_parent());239240 <Pallet<T>>::burn_from(self, &caller, &from, self.1, amount, &budget)241 .map_err(dispatch_to_evm::<T>)?;242 Ok(true)243 }244245 246 247 248 249 250 251 252 #[weight(<SelfWeightOf<T>>::approve())]253 fn approve_cross(254 &mut self,255 caller: caller,256 spender: pallet_common::eth::CrossAddress,257 amount: uint256,258 ) -> Result<bool> {259 let caller = T::CrossAccountId::from_eth(caller);260 let spender = spender.into_sub_cross_account::<T>()?;261 let amount = amount.try_into().map_err(|_| "amount overflow")?;262263 <Pallet<T>>::set_allowance(self, &caller, &spender, self.1, amount)264 .map_err(dispatch_to_evm::<T>)?;265 Ok(true)266 }267 268 269 270 #[weight(<SelfWeightOf<T>>::repartition_item())]271 fn repartition(&mut self, caller: caller, amount: uint256) -> Result<bool> {272 let caller = T::CrossAccountId::from_eth(caller);273 let amount = amount.try_into().map_err(|_| "amount overflow")?;274275 <Pallet<T>>::repartition(self, &caller, self.1, amount).map_err(dispatch_to_evm::<T>)?;276 Ok(true)277 }278279 280 281 282 #[weight(<CommonWeights<T>>::transfer())]283 fn transfer_cross(284 &mut self,285 caller: caller,286 to: pallet_common::eth::CrossAddress,287 amount: uint256,288 ) -> Result<bool> {289 let caller = T::CrossAccountId::from_eth(caller);290 let to = to.into_sub_cross_account::<T>()?;291 let amount = amount.try_into().map_err(|_| "amount overflow")?;292 let budget = self293 .recorder294 .weight_calls_budget(<StructureWeight<T>>::find_parent());295296 <Pallet<T>>::transfer(self, &caller, &to, self.1, amount, &budget)297 .map_err(dispatch_to_evm::<T>)?;298 Ok(true)299 }300301 302 303 304 305 #[weight(<CommonWeights<T>>::transfer_from())]306 fn transfer_from_cross(307 &mut self,308 caller: caller,309 from: pallet_common::eth::CrossAddress,310 to: pallet_common::eth::CrossAddress,311 amount: uint256,312 ) -> Result<bool> {313 let caller = T::CrossAccountId::from_eth(caller);314 let from = from.into_sub_cross_account::<T>()?;315 let to = to.into_sub_cross_account::<T>()?;316 let amount = amount.try_into().map_err(|_| "amount overflow")?;317 let budget = self318 .recorder319 .weight_calls_budget(<StructureWeight<T>>::find_parent());320321 <Pallet<T>>::transfer_from(self, &caller, &from, &to, self.1, amount, &budget)322 .map_err(dispatch_to_evm::<T>)?;323 Ok(true)324 }325}326327impl<T: Config> RefungibleTokenHandle<T> {328 pub fn into_inner(self) -> RefungibleHandle<T> {329 self.0330 }331 pub fn common_mut(&mut self) -> &mut RefungibleHandle<T> {332 &mut self.0333 }334}335336impl<T: Config> WithRecorder<T> for RefungibleTokenHandle<T> {337 fn recorder(&self) -> &pallet_evm_coder_substrate::SubstrateRecorder<T> {338 self.0.recorder()339 }340 fn into_recorder(self) -> pallet_evm_coder_substrate::SubstrateRecorder<T> {341 self.0.into_recorder()342 }343}344345impl<T: Config> Deref for RefungibleTokenHandle<T> {346 type Target = RefungibleHandle<T>;347348 fn deref(&self) -> &Self::Target {349 &self.0350 }351}352353#[solidity_interface(354 name = UniqueRefungibleToken,355 is(ERC20, ERC20UniqueExtensions, ERC1633)356)]357impl<T: Config> RefungibleTokenHandle<T> where T::AccountId: From<[u8; 32]> {}358359generate_stubgen!(gen_impl, UniqueRefungibleTokenCall<()>, true);360generate_stubgen!(gen_iface, UniqueRefungibleTokenCall<()>, false);361362impl<T: Config> CommonEvmHandler for RefungibleTokenHandle<T>363where364 T::AccountId: From<[u8; 32]>,365{366 const CODE: &'static [u8] = include_bytes!("./stubs/UniqueRefungibleToken.raw");367368 fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult> {369 call::<T, UniqueRefungibleTokenCall<T>, _, _>(handle, self)370 }371}