12345678910111213141516171819202122extern crate alloc;23use core::{24 char::{REPLACEMENT_CHARACTER, decode_utf16},25 convert::TryInto,26};27use evm_coder::{ToLog, execution::*, generate_stubgen, solidity, solidity_interface, types::*, weight};28use frame_support::BoundedVec;29use up_data_structs::{30 TokenId, PropertyPermission, PropertyKeyPermission, Property, CollectionId, PropertyKey,31 CollectionPropertiesVec,32};33use pallet_evm_coder_substrate::dispatch_to_evm;34use sp_std::vec::Vec;35use pallet_common::{36 erc::{37 CommonEvmHandler, PrecompileResult, CollectionCall, static_property::key,38 static_property::value,39 },40 CollectionHandle, CollectionPropertyPermissions,41};42use pallet_evm::{account::CrossAccountId, PrecompileHandle};43use pallet_evm_coder_substrate::call;44use pallet_structure::{SelfWeightOf as StructureWeight, weights::WeightInfo as _};4546use crate::{47 AccountBalance, Config, CreateItemData, NonfungibleHandle, Pallet, TokenData, TokensMinted,48 SelfWeightOf, weights::WeightInfo, TokenProperties,49};505152#[solidity_interface(name = TokenProperties)]53impl<T: Config> NonfungibleHandle<T> {54 55 56 57 58 59 60 fn set_token_property_permission(61 &mut self,62 caller: caller,63 key: string,64 is_mutable: bool,65 collection_admin: bool,66 token_owner: bool,67 ) -> Result<()> {68 let caller = T::CrossAccountId::from_eth(caller);69 <Pallet<T>>::set_property_permission(70 self,71 &caller,72 PropertyKeyPermission {73 key: <Vec<u8>>::from(key)74 .try_into()75 .map_err(|_| "too long key")?,76 permission: PropertyPermission {77 mutable: is_mutable,78 collection_admin,79 token_owner,80 },81 },82 )83 .map_err(dispatch_to_evm::<T>)84 }8586 87 88 89 90 91 fn set_property(92 &mut self,93 caller: caller,94 token_id: uint256,95 key: string,96 value: bytes,97 ) -> Result<()> {98 let caller = T::CrossAccountId::from_eth(caller);99 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;100 let key = <Vec<u8>>::from(key)101 .try_into()102 .map_err(|_| "key too long")?;103 let value = value.try_into().map_err(|_| "value too long")?;104105 let nesting_budget = self106 .recorder107 .weight_calls_budget(<StructureWeight<T>>::find_parent());108109 <Pallet<T>>::set_token_property(110 self,111 &caller,112 TokenId(token_id),113 Property { key, value },114 &nesting_budget,115 )116 .map_err(dispatch_to_evm::<T>)117 }118119 120 121 122 123 fn delete_property(&mut self, token_id: uint256, caller: caller, key: string) -> Result<()> {124 let caller = T::CrossAccountId::from_eth(caller);125 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;126 let key = <Vec<u8>>::from(key)127 .try_into()128 .map_err(|_| "key too long")?;129130 let nesting_budget = self131 .recorder132 .weight_calls_budget(<StructureWeight<T>>::find_parent());133134 <Pallet<T>>::delete_token_property(self, &caller, TokenId(token_id), key, &nesting_budget)135 .map_err(dispatch_to_evm::<T>)136 }137138 139 140 141 142 143 fn property(&self, token_id: uint256, key: string) -> Result<bytes> {144 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;145 let key = <Vec<u8>>::from(key)146 .try_into()147 .map_err(|_| "key too long")?;148149 let props = <TokenProperties<T>>::get((self.id, token_id));150 let prop = props.get(&key).ok_or("key not found")?;151152 Ok(prop.to_vec())153 }154}155156#[derive(ToLog)]157pub enum ERC721Events {158 159 160 161 162 163 Transfer {164 #[indexed]165 from: address,166 #[indexed]167 to: address,168 #[indexed]169 token_id: uint256,170 },171 172 173 174 175 Approval {176 #[indexed]177 owner: address,178 #[indexed]179 approved: address,180 #[indexed]181 token_id: uint256,182 },183 184 185 #[allow(dead_code)]186 ApprovalForAll {187 #[indexed]188 owner: address,189 #[indexed]190 operator: address,191 approved: bool,192 },193}194195#[derive(ToLog)]196pub enum ERC721MintableEvents {197 #[allow(dead_code)]198 MintingFinished {},199}200201202203#[solidity_interface(name = ERC721Metadata, expect_selector = 0x5b5e139f)]204impl<T: Config> NonfungibleHandle<T> {205 206 207 #[solidity(hide, rename_selector = "name")]208 fn name_proxy(&self) -> Result<string> {209 self.name()210 }211212 213 214 #[solidity(hide, rename_selector = "symbol")]215 fn symbol_proxy(&self) -> Result<string> {216 self.symbol()217 }218219 220 221 222 223 224 225 226 227 228 #[solidity(rename_selector = "tokenURI")]229 fn token_uri(&self, token_id: uint256) -> Result<string> {230 let token_id_u32: u32 = token_id.try_into().map_err(|_| "token id overflow")?;231232 match get_token_property(self, token_id_u32, &key::url()).as_deref() {233 Err(_) | Ok("") => (),234 Ok(url) => {235 return Ok(url.into());236 }237 };238239 let base_uri =240 pallet_common::Pallet::<T>::get_collection_property(self.id, &key::base_uri())241 .map(BoundedVec::into_inner)242 .map(string::from_utf8)243 .transpose()244 .map_err(|e| {245 Error::Revert(alloc::format!(246 "Can not convert value \"baseURI\" to string with error \"{}\"",247 e248 ))249 })?;250251 let base_uri = match base_uri.as_deref() {252 None | Some("") => {253 return Ok("".into());254 }255 Some(base_uri) => base_uri.into(),256 };257258 Ok(259 match get_token_property(self, token_id_u32, &key::suffix()).as_deref() {260 Err(_) | Ok("") => base_uri,261 Ok(suffix) => base_uri + suffix,262 },263 )264 }265}266267268269#[solidity_interface(name = ERC721Enumerable, expect_selector = 0x780e9d63)]270impl<T: Config> NonfungibleHandle<T> {271 272 273 274 275 fn token_by_index(&self, index: uint256) -> Result<uint256> {276 Ok(index)277 }278279 280 fn token_of_owner_by_index(&self, _owner: address, _index: uint256) -> Result<uint256> {281 282 Err("not implemented".into())283 }284285 286 287 288 fn total_supply(&self) -> Result<uint256> {289 self.consume_store_reads(1)?;290 Ok(<Pallet<T>>::total_supply(self).into())291 }292}293294295296#[solidity_interface(name = ERC721, events(ERC721Events), expect_selector = 0x80ac58cd)]297impl<T: Config> NonfungibleHandle<T> {298 299 300 301 302 303 fn balance_of(&self, owner: address) -> Result<uint256> {304 self.consume_store_reads(1)?;305 let owner = T::CrossAccountId::from_eth(owner);306 let balance = <AccountBalance<T>>::get((self.id, owner));307 Ok(balance.into())308 }309 310 311 312 313 314 fn owner_of(&self, token_id: uint256) -> Result<address> {315 self.consume_store_reads(1)?;316 let token: TokenId = token_id.try_into()?;317 Ok(*<TokenData<T>>::get((self.id, token))318 .ok_or("token not found")?319 .owner320 .as_eth())321 }322 323 #[solidity(rename_selector = "safeTransferFrom")]324 fn safe_transfer_from_with_data(325 &mut self,326 _from: address,327 _to: address,328 _token_id: uint256,329 _data: bytes,330 ) -> Result<void> {331 332 Err("not implemented".into())333 }334 335 fn safe_transfer_from(336 &mut self,337 _from: address,338 _to: address,339 _token_id: uint256,340 ) -> Result<void> {341 342 Err("not implemented".into())343 }344345 346 347 348 349 350 351 352 353 354 #[weight(<SelfWeightOf<T>>::transfer_from())]355 fn transfer_from(356 &mut self,357 caller: caller,358 from: address,359 to: address,360 token_id: uint256,361 ) -> Result<void> {362 let caller = T::CrossAccountId::from_eth(caller);363 let from = T::CrossAccountId::from_eth(from);364 let to = T::CrossAccountId::from_eth(to);365 let token = token_id.try_into()?;366 let budget = self367 .recorder368 .weight_calls_budget(<StructureWeight<T>>::find_parent());369370 <Pallet<T>>::transfer_from(self, &caller, &from, &to, token, &budget)371 .map_err(dispatch_to_evm::<T>)?;372 Ok(())373 }374375 376 377 378 379 380 381 #[weight(<SelfWeightOf<T>>::approve())]382 fn approve(&mut self, caller: caller, approved: address, token_id: uint256) -> Result<void> {383 let caller = T::CrossAccountId::from_eth(caller);384 let approved = T::CrossAccountId::from_eth(approved);385 let token = token_id.try_into()?;386387 <Pallet<T>>::set_allowance(self, &caller, token, Some(&approved))388 .map_err(dispatch_to_evm::<T>)?;389 Ok(())390 }391392 393 fn set_approval_for_all(394 &mut self,395 _caller: caller,396 _operator: address,397 _approved: bool,398 ) -> Result<void> {399 400 Err("not implemented".into())401 }402403 404 fn get_approved(&self, _token_id: uint256) -> Result<address> {405 406 Err("not implemented".into())407 }408409 410 fn is_approved_for_all(&self, _owner: address, _operator: address) -> Result<address> {411 412 Err("not implemented".into())413 }414}415416417#[solidity_interface(name = ERC721Burnable)]418impl<T: Config> NonfungibleHandle<T> {419 420 421 422 423 #[weight(<SelfWeightOf<T>>::burn_item())]424 fn burn(&mut self, caller: caller, token_id: uint256) -> Result<void> {425 let caller = T::CrossAccountId::from_eth(caller);426 let token = token_id.try_into()?;427428 <Pallet<T>>::burn(self, &caller, token).map_err(dispatch_to_evm::<T>)?;429 Ok(())430 }431}432433434#[solidity_interface(name = ERC721Mintable, events(ERC721MintableEvents))]435impl<T: Config> NonfungibleHandle<T> {436 fn minting_finished(&self) -> Result<bool> {437 Ok(false)438 }439440 441 442 443 444 445 #[weight(<SelfWeightOf<T>>::create_item())]446 fn mint(&mut self, caller: caller, to: address, token_id: uint256) -> Result<bool> {447 let caller = T::CrossAccountId::from_eth(caller);448 let to = T::CrossAccountId::from_eth(to);449 let token_id: u32 = token_id.try_into()?;450 let budget = self451 .recorder452 .weight_calls_budget(<StructureWeight<T>>::find_parent());453454 if <TokensMinted<T>>::get(self.id)455 .checked_add(1)456 .ok_or("item id overflow")?457 != token_id458 {459 return Err("item id should be next".into());460 }461462 <Pallet<T>>::create_item(463 self,464 &caller,465 CreateItemData::<T> {466 properties: BoundedVec::default(),467 owner: to,468 },469 &budget,470 )471 .map_err(dispatch_to_evm::<T>)?;472473 Ok(true)474 }475476 477 478 479 480 481 482 #[solidity(rename_selector = "mintWithTokenURI")]483 #[weight(<SelfWeightOf<T>>::create_item())]484 fn mint_with_token_uri(485 &mut self,486 caller: caller,487 to: address,488 token_id: uint256,489 token_uri: string,490 ) -> Result<bool> {491 let key = key::url();492 let permission = get_token_permission::<T>(self.id, &key)?;493 if !permission.collection_admin {494 return Err("Operation is not allowed".into());495 }496497 let caller = T::CrossAccountId::from_eth(caller);498 let to = T::CrossAccountId::from_eth(to);499 let token_id: u32 = token_id.try_into().map_err(|_| "amount overflow")?;500 let budget = self501 .recorder502 .weight_calls_budget(<StructureWeight<T>>::find_parent());503504 if <TokensMinted<T>>::get(self.id)505 .checked_add(1)506 .ok_or("item id overflow")?507 != token_id508 {509 return Err("item id should be next".into());510 }511512 let mut properties = CollectionPropertiesVec::default();513 properties514 .try_push(Property {515 key,516 value: token_uri517 .into_bytes()518 .try_into()519 .map_err(|_| "token uri is too long")?,520 })521 .map_err(|e| Error::Revert(alloc::format!("Can't add property: {:?}", e)))?;522523 <Pallet<T>>::create_item(524 self,525 &caller,526 CreateItemData::<T> {527 properties,528 owner: to,529 },530 &budget,531 )532 .map_err(dispatch_to_evm::<T>)?;533 Ok(true)534 }535536 537 fn finish_minting(&mut self, _caller: caller) -> Result<bool> {538 Err("not implementable".into())539 }540}541542fn get_token_property<T: Config>(543 collection: &CollectionHandle<T>,544 token_id: u32,545 key: &up_data_structs::PropertyKey,546) -> Result<string> {547 collection.consume_store_reads(1)?;548 let properties = <TokenProperties<T>>::try_get((collection.id, token_id))549 .map_err(|_| Error::Revert("Token properties not found".into()))?;550 if let Some(property) = properties.get(key) {551 return Ok(string::from_utf8_lossy(property).into());552 }553554 Err("Property tokenURI not found".into())555}556557fn get_token_permission<T: Config>(558 collection_id: CollectionId,559 key: &PropertyKey,560) -> Result<PropertyPermission> {561 let token_property_permissions = CollectionPropertyPermissions::<T>::try_get(collection_id)562 .map_err(|_| Error::Revert("No permissions for collection".into()))?;563 let a = token_property_permissions564 .get(key)565 .map(Clone::clone)566 .ok_or_else(|| {567 let key = string::from_utf8(key.clone().into_inner()).unwrap_or_default();568 Error::Revert(alloc::format!("No permission for key {}", key))569 })?;570 Ok(a)571}572573574#[solidity_interface(name = ERC721UniqueExtensions)]575impl<T: Config> NonfungibleHandle<T> {576 577 fn name(&self) -> Result<string> {578 Ok(decode_utf16(self.name.iter().copied())579 .map(|r| r.unwrap_or(REPLACEMENT_CHARACTER))580 .collect::<string>())581 }582583 584 fn symbol(&self) -> Result<string> {585 Ok(string::from_utf8_lossy(&self.token_prefix).into())586 }587588 589 590 591 592 593 #[weight(<SelfWeightOf<T>>::transfer())]594 fn transfer(&mut self, caller: caller, to: address, token_id: uint256) -> Result<void> {595 let caller = T::CrossAccountId::from_eth(caller);596 let to = T::CrossAccountId::from_eth(to);597 let token = token_id.try_into()?;598 let budget = self599 .recorder600 .weight_calls_budget(<StructureWeight<T>>::find_parent());601602 <Pallet<T>>::transfer(self, &caller, &to, token, &budget).map_err(dispatch_to_evm::<T>)?;603 Ok(())604 }605606 607 608 609 610 611 612 #[weight(<SelfWeightOf<T>>::burn_from())]613 fn burn_from(&mut self, caller: caller, from: address, token_id: uint256) -> Result<void> {614 let caller = T::CrossAccountId::from_eth(caller);615 let from = T::CrossAccountId::from_eth(from);616 let token = token_id.try_into()?;617 let budget = self618 .recorder619 .weight_calls_budget(<StructureWeight<T>>::find_parent());620621 <Pallet<T>>::burn_from(self, &caller, &from, token, &budget)622 .map_err(dispatch_to_evm::<T>)?;623 Ok(())624 }625626 627 fn next_token_id(&self) -> Result<uint256> {628 self.consume_store_reads(1)?;629 Ok(<TokensMinted<T>>::get(self.id)630 .checked_add(1)631 .ok_or("item id overflow")?632 .into())633 }634635 636 637 638 639 640 #[weight(<SelfWeightOf<T>>::create_multiple_items(token_ids.len() as u32))]641 fn mint_bulk(&mut self, caller: caller, to: address, token_ids: Vec<uint256>) -> Result<bool> {642 let caller = T::CrossAccountId::from_eth(caller);643 let to = T::CrossAccountId::from_eth(to);644 let mut expected_index = <TokensMinted<T>>::get(self.id)645 .checked_add(1)646 .ok_or("item id overflow")?;647 let budget = self648 .recorder649 .weight_calls_budget(<StructureWeight<T>>::find_parent());650651 let total_tokens = token_ids.len();652 for id in token_ids.into_iter() {653 let id: u32 = id.try_into().map_err(|_| "token id overflow")?;654 if id != expected_index {655 return Err("item id should be next".into());656 }657 expected_index = expected_index.checked_add(1).ok_or("item id overflow")?;658 }659 let data = (0..total_tokens)660 .map(|_| CreateItemData::<T> {661 properties: BoundedVec::default(),662 owner: to.clone(),663 })664 .collect();665666 <Pallet<T>>::create_multiple_items(self, &caller, data, &budget)667 .map_err(dispatch_to_evm::<T>)?;668 Ok(true)669 }670671 672 673 674 675 676 #[solidity(rename_selector = "mintBulkWithTokenURI")]677 #[weight(<SelfWeightOf<T>>::create_multiple_items(tokens.len() as u32))]678 fn mint_bulk_with_token_uri(679 &mut self,680 caller: caller,681 to: address,682 tokens: Vec<(uint256, string)>,683 ) -> Result<bool> {684 let key = key::url();685 let caller = T::CrossAccountId::from_eth(caller);686 let to = T::CrossAccountId::from_eth(to);687 let mut expected_index = <TokensMinted<T>>::get(self.id)688 .checked_add(1)689 .ok_or("item id overflow")?;690 let budget = self691 .recorder692 .weight_calls_budget(<StructureWeight<T>>::find_parent());693694 let mut data = Vec::with_capacity(tokens.len());695 for (id, token_uri) in tokens {696 let id: u32 = id.try_into().map_err(|_| "token id overflow")?;697 if id != expected_index {698 return Err("item id should be next".into());699 }700 expected_index = expected_index.checked_add(1).ok_or("item id overflow")?;701702 let mut properties = CollectionPropertiesVec::default();703 properties704 .try_push(Property {705 key: key.clone(),706 value: token_uri707 .into_bytes()708 .try_into()709 .map_err(|_| "token uri is too long")?,710 })711 .map_err(|e| Error::Revert(alloc::format!("Can't add property: {:?}", e)))?;712713 data.push(CreateItemData::<T> {714 properties,715 owner: to.clone(),716 });717 }718719 <Pallet<T>>::create_multiple_items(self, &caller, data, &budget)720 .map_err(dispatch_to_evm::<T>)?;721 Ok(true)722 }723}724725#[solidity_interface(726 name = UniqueNFT,727 is(728 ERC721,729 ERC721Enumerable,730 ERC721UniqueExtensions,731 ERC721Mintable,732 ERC721Burnable,733 ERC721Metadata(if(this.flags.erc721metadata)),734 Collection(via(common_mut returns CollectionHandle<T>)),735 TokenProperties,736 )737)]738impl<T: Config> NonfungibleHandle<T> where T::AccountId: From<[u8; 32]> + AsRef<[u8; 32]> {}739740741generate_stubgen!(gen_impl, UniqueNFTCall<()>, true);742generate_stubgen!(gen_iface, UniqueNFTCall<()>, false);743744impl<T: Config> CommonEvmHandler for NonfungibleHandle<T>745where746 T::AccountId: From<[u8; 32]> + AsRef<[u8; 32]>,747{748 const CODE: &'static [u8] = include_bytes!("./stubs/UniqueNFT.raw");749750 fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult> {751 call::<T, UniqueNFTCall<T>, _, _>(handle, self)752 }753}