1use core::char::{decode_utf16, REPLACEMENT_CHARACTER};2use evm_coder::{ToLog, execution::Result, solidity, solidity_interface, types::*};3use nft_data_structs::{CreateItemData, CreateNftData};4use core::convert::TryInto;5use crate::{6 Allowances, Module, Balance, CollectionHandle, CollectionMode, Config, NftItemList,7 ItemListIndex,8};9use frame_support::storage::{StorageMap, StorageDoubleMap};10use pallet_evm::AddressMapping;11use super::account::CrossAccountId;12use sp_std::{vec, vec::Vec};1314#[solidity_interface(name = "ERC165")]15impl<T: Config> CollectionHandle<T> {16 fn supports_interface(&self, interface_id: bytes4) -> Result<bool> {17 Ok(match self.mode {18 CollectionMode::Fungible(_) => UniqueFungibleCall::supports_interface(interface_id),19 CollectionMode::NFT => UniqueNFTCall::supports_interface(interface_id),20 _ => false,21 })22 }23}2425#[solidity_interface(name = "InlineNameSymbol")]26impl<T: Config> CollectionHandle<T> {27 fn name(&self) -> Result<string> {28 Ok(decode_utf16(self.name.iter().copied())29 .map(|r| r.unwrap_or(REPLACEMENT_CHARACTER))30 .collect::<string>())31 }3233 fn symbol(&self) -> Result<string> {34 Ok(string::from_utf8_lossy(&self.token_prefix).into())35 }36}3738#[solidity_interface(name = "InlineTotalSupply")]39impl<T: Config> CollectionHandle<T> {40 fn total_supply(&self) -> Result<uint256> {41 42 Ok(0.into())43 }44}4546#[solidity_interface(name = "ERC721Metadata", inline_is(InlineNameSymbol))]47impl<T: Config> CollectionHandle<T> {48 #[solidity(rename_selector = "tokenURI")]49 fn token_uri(&self, token_id: uint256) -> Result<string> {50 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;51 Ok(string::from_utf8_lossy(52 &<NftItemList<T>>::get(self.id, token_id)53 .ok_or("token not found")?54 .const_data,55 )56 .into())57 }58}5960#[solidity_interface(name = "ERC721Enumerable", inline_is(InlineTotalSupply))]61impl<T: Config> CollectionHandle<T> {62 fn token_by_index(&self, index: uint256) -> Result<uint256> {63 Ok(index)64 }6566 fn token_of_owner_by_index(&self, _owner: address, _index: uint256) -> Result<uint256> {67 68 Err("not implemented".into())69 }70}7172#[derive(ToLog)]73pub enum ERC721Events {74 Transfer {75 #[indexed]76 from: address,77 #[indexed]78 to: address,79 #[indexed]80 token_id: uint256,81 },82 Approval {83 #[indexed]84 owner: address,85 #[indexed]86 approved: address,87 #[indexed]88 token_id: uint256,89 },90 #[allow(dead_code)]91 ApprovalForAll {92 #[indexed]93 owner: address,94 #[indexed]95 operator: address,96 approved: bool,97 },98}99100#[solidity_interface(name = "ERC721", is(ERC165), events(ERC721Events))]101impl<T: Config> CollectionHandle<T> {102 #[solidity(rename_selector = "balanceOf")]103 fn balance_of_nft(&self, owner: address) -> Result<uint256> {104 let owner = T::EvmAddressMapping::into_account_id(owner);105 let balance = <Balance<T>>::get(self.id, owner);106 Ok(balance.into())107 }108 fn owner_of(&self, token_id: uint256) -> Result<address> {109 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;110 let token = <NftItemList<T>>::get(self.id, token_id).ok_or("unknown token")?;111 Ok(*token.owner.as_eth())112 }113 fn safe_transfer_from_with_data(114 &mut self,115 _from: address,116 _to: address,117 _token_id: uint256,118 _data: bytes,119 _value: value,120 ) -> Result<void> {121 122 Err("not implemented".into())123 }124 fn safe_transfer_from(125 &mut self,126 _from: address,127 _to: address,128 _token_id: uint256,129 _value: value,130 ) -> Result<void> {131 132 Err("not implemented".into())133 }134135 fn transfer_from(136 &mut self,137 caller: caller,138 from: address,139 to: address,140 token_id: uint256,141 _value: value,142 ) -> Result<void> {143 let caller = T::CrossAccountId::from_eth(caller);144 let from = T::CrossAccountId::from_eth(from);145 let to = T::CrossAccountId::from_eth(to);146 let token_id = token_id.try_into().map_err(|_| "token_id overflow")?;147148 <Module<T>>::transfer_from_internal(&caller, &from, &to, self, token_id, 1)149 .map_err(|_| "transferFrom error")?;150 Ok(())151 }152153 fn approve(154 &mut self,155 caller: caller,156 approved: address,157 token_id: uint256,158 _value: value,159 ) -> Result<void> {160 let caller = T::CrossAccountId::from_eth(caller);161 let approved = T::CrossAccountId::from_eth(approved);162 let token_id = token_id.try_into().map_err(|_| "token_id overflow")?;163164 <Module<T>>::approve_internal(&caller, &approved, self, token_id, 1)165 .map_err(|_| "approve internal")?;166 Ok(())167 }168169 fn set_approval_for_all(170 &mut self,171 _caller: caller,172 _operator: address,173 _approved: bool,174 ) -> Result<void> {175 176 Err("not implemented".into())177 }178179 fn get_approved(&self, _token_id: uint256) -> Result<address> {180 181 Err("not implemented".into())182 }183184 fn is_approved_for_all(&self, _owner: address, _operator: address) -> Result<address> {185 186 Err("not implemented".into())187 }188}189190#[solidity_interface(name = "ERC721Burnable")]191impl<T: Config> CollectionHandle<T> {192 fn burn(&mut self, caller: caller, token_id: uint256) -> Result<void> {193 let caller = T::CrossAccountId::from_eth(caller);194 let token_id = token_id.try_into().map_err(|_| "amount overflow")?;195196 <Module<T>>::burn_item_internal(&caller, &self, token_id, 1).map_err(|_| "burn error")?;197 Ok(())198 }199}200201#[derive(ToLog)]202pub enum ERC721MintableEvents {203 #[allow(dead_code)]204 MintingFinished {},205}206207#[solidity_interface(name = "ERC721Mintable", events(ERC721MintableEvents))]208impl<T: Config> CollectionHandle<T> {209 fn minting_finished(&self) -> Result<bool> {210 Ok(false)211 }212213 fn mint(&mut self, caller: caller, to: address, token_id: uint256) -> Result<bool> {214 let caller = T::CrossAccountId::from_eth(caller);215 let to = T::CrossAccountId::from_eth(to);216 let token_id: u32 = token_id.try_into().map_err(|_| "amount overflow")?;217 if <ItemListIndex>::get(self.id)218 .checked_add(1)219 .ok_or("item id overflow")?220 != token_id221 {222 return Err("item id should be next".into());223 }224225 <Module<T>>::create_item_internal(226 &caller,227 &self,228 &to,229 CreateItemData::NFT(CreateNftData {230 const_data: vec![],231 variable_data: vec![],232 }),233 )234 .map_err(|_| "mint error")?;235 Ok(true)236 }237238 #[solidity(rename_selector = "mintWithTokenURI")]239 fn mint_with_token_uri(240 &mut self,241 caller: caller,242 to: address,243 token_id: uint256,244 token_uri: string,245 ) -> Result<bool> {246 let caller = T::CrossAccountId::from_eth(caller);247 let to = T::CrossAccountId::from_eth(to);248 let token_id: u32 = token_id.try_into().map_err(|_| "amount overflow")?;249 if <ItemListIndex>::get(self.id)250 .checked_add(1)251 .ok_or("item id overflow")?252 != token_id253 {254 return Err("item id should be next".into());255 }256257 <Module<T>>::create_item_internal(258 &caller,259 &self,260 &to,261 CreateItemData::NFT(CreateNftData {262 const_data: token_uri.into(),263 variable_data: vec![],264 }),265 )266 .map_err(|_| "mint error")?;267 Ok(true)268 }269270 fn finish_minting(&mut self, _caller: caller) -> Result<bool> {271 Err("not implementable".into())272 }273}274275#[solidity_interface(name = "ERC721UniqueExtensions")]276impl<T: Config> CollectionHandle<T> {277 #[solidity(rename_selector = "transfer")]278 fn transfer_nft(279 &mut self,280 caller: caller,281 to: address,282 token_id: uint256,283 _value: value,284 ) -> Result<void> {285 let caller = T::CrossAccountId::from_eth(caller);286 let to = T::CrossAccountId::from_eth(to);287 let token_id = token_id.try_into().map_err(|_| "amount overflow")?;288289 <Module<T>>::transfer_internal(&caller, &to, self, token_id, 1)290 .map_err(|_| "transfer error")?;291 Ok(())292 }293294 fn next_token_id(&self) -> Result<uint256> {295 Ok(ItemListIndex::get(self.id)296 .checked_add(1)297 .ok_or("item id overflow")?298 .into())299 }300}301302#[solidity_interface(303 name = "UniqueNFT",304 is(305 ERC165,306 ERC721,307 ERC721Metadata,308 ERC721Enumerable,309 ERC721UniqueExtensions,310 ERC721Mintable,311 ERC721Burnable,312 )313)]314impl<T: Config> CollectionHandle<T> {}315316#[derive(ToLog)]317pub enum ERC20Events {318 Transfer {319 #[indexed]320 from: address,321 #[indexed]322 to: address,323 value: uint256,324 },325 Approval {326 #[indexed]327 owner: address,328 #[indexed]329 spender: address,330 value: uint256,331 },332}333334#[solidity_interface(335 name = "ERC20",336 inline_is(InlineNameSymbol, InlineTotalSupply),337 events(ERC20Events)338)]339impl<T: Config> CollectionHandle<T> {340 fn decimals(&self) -> Result<uint8> {341 Ok(if let CollectionMode::Fungible(decimals) = &self.mode {342 *decimals343 } else {344 unreachable!()345 })346 }347 fn balance_of(&self, owner: address) -> Result<uint256> {348 let owner = T::EvmAddressMapping::into_account_id(owner);349 let balance = <Balance<T>>::get(self.id, owner);350 Ok(balance.into())351 }352 fn transfer(&mut self, caller: caller, to: address, amount: uint256) -> Result<bool> {353 let caller = T::CrossAccountId::from_eth(caller);354 let to = T::CrossAccountId::from_eth(to);355 let amount = amount.try_into().map_err(|_| "amount overflow")?;356357 <Module<T>>::transfer_internal(&caller, &to, self, 1, amount)358 .map_err(|_| "transfer error")?;359 Ok(true)360 }361 #[solidity(rename_selector = "transferFrom")]362 fn transfer_from_fungible(363 &mut self,364 caller: caller,365 from: address,366 to: address,367 amount: uint256,368 ) -> Result<bool> {369 let caller = T::CrossAccountId::from_eth(caller);370 let from = T::CrossAccountId::from_eth(from);371 let to = T::CrossAccountId::from_eth(to);372 let amount = amount.try_into().map_err(|_| "amount overflow")?;373374 <Module<T>>::transfer_from_internal(&caller, &from, &to, self, 1, amount)375 .map_err(|_| "transferFrom error")?;376 Ok(true)377 }378 #[solidity(rename_selector = "approve")]379 fn approve_fungible(380 &mut self,381 caller: caller,382 spender: address,383 amount: uint256,384 ) -> Result<bool> {385 let caller = T::CrossAccountId::from_eth(caller);386 let spender = T::CrossAccountId::from_eth(spender);387 let amount = amount.try_into().map_err(|_| "amount overflow")?;388389 <Module<T>>::approve_internal(&caller, &spender, self, 1, amount)390 .map_err(|_| "approve internal")?;391 Ok(true)392 }393 fn allowance(&self, owner: address, spender: address) -> Result<uint256> {394 let owner = T::CrossAccountId::from_eth(owner);395 let spender = T::CrossAccountId::from_eth(spender);396397 Ok(<Allowances<T>>::get(self.id, (1, owner.as_sub(), spender.as_sub())).into())398 }399}400401#[solidity_interface(name = "UniqueFungible", is(ERC165, ERC20))]402impl<T: Config> CollectionHandle<T> {}403404macro_rules! generate_code {405 ($name:ident, $decl:ident, $is_impl:literal) => {406 #[test]407 #[ignore]408 fn $name() {409 use sp_std::collections::btree_set::BTreeSet;410 let mut out = BTreeSet::new();411 $decl::generate_solidity_interface(&mut out, $is_impl);412 println!("=== SNIP START ===");413 println!("// SPDX-License-Identifier: OTHER");414 println!("// This code is automatically generated with `cargo test --package pallet-nft -- eth::erc::{} --exact --nocapture --ignored`", stringify!(name));415 println!();416 println!("pragma solidity >=0.8.0 <0.9.0;");417 println!();418 for b in out {419 println!("{}", b);420 }421 println!("=== SNIP END ===");422 }423 };424}425426427generate_code!(nft_impl, UniqueNFTCall, true);428generate_code!(nft_iface, UniqueNFTCall, false);429430generate_code!(fungible_impl, UniqueFungibleCall, true);431generate_code!(fungible_iface, UniqueFungibleCall, false);