12345678910111213141516171819202122extern crate alloc;2324use core::{25 char::{REPLACEMENT_CHARACTER, decode_utf16},26 convert::TryInto,27};28use evm_coder::{abi::AbiType, ToLog, generate_stubgen, solidity_interface, types::*};29use frame_support::{BoundedBTreeMap, BoundedVec};30use pallet_common::{31 CollectionHandle, CollectionPropertyPermissions, CommonCollectionOperations,32 Error as CommonError,33 erc::{CommonEvmHandler, CollectionCall, static_property::key},34 eth::{self, TokenUri},35};36use pallet_evm::{account::CrossAccountId, PrecompileHandle};37use pallet_evm_coder_substrate::{38 call, dispatch_to_evm,39 execution::{PreDispatch, Result, Error},40 frontier_contract,41};42use pallet_structure::{SelfWeightOf as StructureWeight, weights::WeightInfo as _};43use sp_core::{H160, U256, Get};44use sp_std::{collections::btree_map::BTreeMap, vec::Vec, vec};45use up_data_structs::{46 CollectionId, CollectionPropertiesVec, mapping::TokenAddressMapping, Property, PropertyKey,47 PropertyKeyPermission, PropertyPermission, TokenId, TokenOwnerError,48};4950use crate::{51 AccountBalance, Balance, Config, CreateItemData, Pallet, RefungibleHandle, TokenProperties,52 TokensMinted, TotalSupply, SelfWeightOf, weights::WeightInfo,53};5455frontier_contract! {56 macro_rules! RefungibleHandle_result {...}57 impl<T: Config> Contract for RefungibleHandle<T> {...}58}5960pub const ADDRESS_FOR_PARTIALLY_OWNED_TOKENS: H160 = H160::repeat_byte(0xff);616263#[derive(ToLog)]64pub enum ERC721TokenEvent {65 66 TokenChanged {67 68 #[indexed]69 token_id: U256,70 },71}727374#[solidity_interface(name = TokenProperties, events(ERC721TokenEvent), enum(derive(PreDispatch)), enum_attr(weight))]75impl<T: Config> RefungibleHandle<T> {76 77 78 79 80 81 82 #[solidity(hide)]83 #[weight(<SelfWeightOf<T>>::set_token_property_permissions(1))]84 fn set_token_property_permission(85 &mut self,86 caller: Caller,87 key: String,88 is_mutable: bool,89 collection_admin: bool,90 token_owner: bool,91 ) -> Result<()> {92 let caller = T::CrossAccountId::from_eth(caller);93 <Pallet<T>>::set_token_property_permissions(94 self,95 &caller,96 vec![PropertyKeyPermission {97 key: <Vec<u8>>::from(key)98 .try_into()99 .map_err(|_| "too long key")?,100 permission: PropertyPermission {101 mutable: is_mutable,102 collection_admin,103 token_owner,104 },105 }],106 )107 .map_err(dispatch_to_evm::<T>)108 }109110 111 112 113 #[weight(<SelfWeightOf<T>>::set_token_property_permissions(permissions.len() as u32))]114 fn set_token_property_permissions(115 &mut self,116 caller: Caller,117 permissions: Vec<eth::TokenPropertyPermission>,118 ) -> Result<()> {119 let caller = T::CrossAccountId::from_eth(caller);120 let perms = eth::TokenPropertyPermission::into_property_key_permissions(permissions)?;121122 <Pallet<T>>::set_token_property_permissions(self, &caller, perms)123 .map_err(dispatch_to_evm::<T>)124 }125126 127 fn token_property_permissions(&self) -> Result<Vec<eth::TokenPropertyPermission>> {128 let perms = <Pallet<T>>::token_property_permission(self.id);129 Ok(perms130 .into_iter()131 .map(eth::TokenPropertyPermission::from)132 .collect())133 }134135 136 137 138 139 140 #[solidity(hide)]141 #[weight(<SelfWeightOf<T>>::set_token_properties(1))]142 fn set_property(143 &mut self,144 caller: Caller,145 token_id: U256,146 key: String,147 value: Bytes,148 ) -> Result<()> {149 let caller = T::CrossAccountId::from_eth(caller);150 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;151 let key = <Vec<u8>>::from(key)152 .try_into()153 .map_err(|_| "key too long")?;154 let value = value.0.try_into().map_err(|_| "value too long")?;155156 let nesting_budget = self157 .recorder158 .weight_calls_budget(<StructureWeight<T>>::find_parent());159160 <Pallet<T>>::set_token_property(161 self,162 &caller,163 TokenId(token_id),164 Property { key, value },165 &nesting_budget,166 )167 .map_err(dispatch_to_evm::<T>)168 }169170 171 172 173 174 #[weight(<SelfWeightOf<T>>::set_token_properties(properties.len() as u32))]175 fn set_properties(176 &mut self,177 caller: Caller,178 token_id: U256,179 properties: Vec<eth::Property>,180 ) -> Result<()> {181 let caller = T::CrossAccountId::from_eth(caller);182 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;183184 let nesting_budget = self185 .recorder186 .weight_calls_budget(<StructureWeight<T>>::find_parent());187188 let properties = properties189 .into_iter()190 .map(eth::Property::try_into)191 .collect::<Result<Vec<_>>>()?;192193 <Pallet<T>>::set_token_properties(194 self,195 &caller,196 TokenId(token_id),197 properties.into_iter(),198 false,199 &nesting_budget,200 )201 .map_err(dispatch_to_evm::<T>)202 }203204 205 206 207 208 #[solidity(hide)]209 #[weight(<SelfWeightOf<T>>::delete_token_properties(1))]210 fn delete_property(&mut self, token_id: U256, caller: Caller, key: String) -> Result<()> {211 let caller = T::CrossAccountId::from_eth(caller);212 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;213 let key = <Vec<u8>>::from(key)214 .try_into()215 .map_err(|_| "key too long")?;216217 let nesting_budget = self218 .recorder219 .weight_calls_budget(<StructureWeight<T>>::find_parent());220221 <Pallet<T>>::delete_token_property(self, &caller, TokenId(token_id), key, &nesting_budget)222 .map_err(dispatch_to_evm::<T>)223 }224225 226 227 228 229 #[weight(<SelfWeightOf<T>>::delete_token_properties(keys.len() as u32))]230 fn delete_properties(231 &mut self,232 token_id: U256,233 caller: Caller,234 keys: Vec<String>,235 ) -> Result<()> {236 let caller = T::CrossAccountId::from_eth(caller);237 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;238 let keys = keys239 .into_iter()240 .map(|k| Ok(<Vec<u8>>::from(k).try_into().map_err(|_| "key too long")?))241 .collect::<Result<Vec<_>>>()?;242243 let nesting_budget = self244 .recorder245 .weight_calls_budget(<StructureWeight<T>>::find_parent());246247 <Pallet<T>>::delete_token_properties(248 self,249 &caller,250 TokenId(token_id),251 keys.into_iter(),252 &nesting_budget,253 )254 .map_err(dispatch_to_evm::<T>)255 }256257 258 259 260 261 262 fn property(&self, token_id: U256, key: String) -> Result<Bytes> {263 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;264 let key = <Vec<u8>>::from(key)265 .try_into()266 .map_err(|_| "key too long")?;267268 let props = <TokenProperties<T>>::get((self.id, token_id));269 let prop = props.get(&key).ok_or("key not found")?;270271 Ok(prop.to_vec().into())272 }273}274275#[derive(ToLog)]276pub enum ERC721Events {277 278 279 280 Transfer {281 #[indexed]282 from: Address,283 #[indexed]284 to: Address,285 #[indexed]286 token_id: U256,287 },288 289 Approval {290 #[indexed]291 owner: Address,292 #[indexed]293 approved: Address,294 #[indexed]295 token_id: U256,296 },297 298 #[allow(dead_code)]299 ApprovalForAll {300 #[indexed]301 owner: Address,302 #[indexed]303 operator: Address,304 approved: bool,305 },306}307308309310#[solidity_interface(name = ERC721Metadata, enum(derive(PreDispatch)), expect_selector = 0x5b5e139f)]311impl<T: Config> RefungibleHandle<T>312where313 T::AccountId: From<[u8; 32]> + AsRef<[u8; 32]>,314{315 316 317 #[solidity(hide, rename_selector = "name")]318 fn name_proxy(&self) -> Result<String> {319 self.name()320 }321322 323 324 #[solidity(hide, rename_selector = "symbol")]325 fn symbol_proxy(&self) -> Result<String> {326 self.symbol()327 }328329 330 331 332 333 334 335 336 337 338 #[solidity(rename_selector = "tokenURI")]339 fn token_uri(&self, token_id: U256) -> Result<String> {340 let token_id_u32: u32 = token_id.try_into().map_err(|_| "token id overflow")?;341342 match get_token_property(self, token_id_u32, &key::url()).as_deref() {343 Err(_) | Ok("") => (),344 Ok(url) => {345 return Ok(url.into());346 }347 };348349 let base_uri =350 pallet_common::Pallet::<T>::get_collection_property(self.id, &key::base_uri())351 .map(BoundedVec::into_inner)352 .map(String::from_utf8)353 .transpose()354 .map_err(|e| {355 Error::Revert(alloc::format!(356 "Can not convert value \"baseURI\" to string with error \"{}\"",357 e358 ))359 })?;360361 let base_uri = match base_uri.as_deref() {362 None | Some("") => {363 return Ok("".into());364 }365 Some(base_uri) => base_uri.into(),366 };367368 Ok(369 match get_token_property(self, token_id_u32, &key::suffix()).as_deref() {370 Err(_) | Ok("") => base_uri,371 Ok(suffix) => base_uri + suffix,372 },373 )374 }375}376377378379#[solidity_interface(name = ERC721Enumerable, enum(derive(PreDispatch)), expect_selector = 0x780e9d63)]380impl<T: Config> RefungibleHandle<T> {381 382 383 384 385 fn token_by_index(&self, index: U256) -> U256 {386 index387 }388389 390 fn token_of_owner_by_index(&self, _owner: Address, _index: U256) -> Result<U256> {391 392 Err("not implemented".into())393 }394395 396 397 398 fn total_supply(&self) -> Result<U256> {399 self.consume_store_reads(1)?;400 Ok(<Pallet<T>>::total_supply(self).into())401 }402}403404405406#[solidity_interface(name = ERC721, events(ERC721Events), enum(derive(PreDispatch)), enum_attr(weight), expect_selector = 0x80ac58cd)]407impl<T: Config> RefungibleHandle<T> {408 409 410 411 412 413 fn balance_of(&self, owner: Address) -> Result<U256> {414 self.consume_store_reads(1)?;415 let owner = T::CrossAccountId::from_eth(owner);416 let balance = <AccountBalance<T>>::get((self.id, owner));417 Ok(balance.into())418 }419420 421 422 423 424 425 426 427 fn owner_of(&self, token_id: U256) -> Result<Address> {428 self.consume_store_reads(2)?;429 let token = token_id.try_into()?;430 let owner = <Pallet<T>>::token_owner(self.id, token);431 owner432 .map(|address| *address.as_eth())433 .or_else(|err| match err {434 TokenOwnerError::NotFound => Err(Error::Revert("token not found".into())),435 TokenOwnerError::MultipleOwners => Ok(ADDRESS_FOR_PARTIALLY_OWNED_TOKENS),436 })437 }438439 440 #[solidity(rename_selector = "safeTransferFrom")]441 fn safe_transfer_from_with_data(442 &mut self,443 _from: Address,444 _to: Address,445 _token_id: U256,446 _data: Bytes,447 ) -> Result<()> {448 449 Err("not implemented".into())450 }451452 453 #[solidity(rename_selector = "safeTransferFrom")]454 fn safe_transfer_from(&mut self, _from: Address, _to: Address, _token_id: U256) -> Result<()> {455 456 Err("not implemented".into())457 }458459 460 461 462 463 464 465 466 467 468 469 #[weight(<SelfWeightOf<T>>::transfer_from_creating_removing())]470 fn transfer_from(471 &mut self,472 caller: Caller,473 from: Address,474 to: Address,475 token_id: U256,476 ) -> Result<()> {477 let caller = T::CrossAccountId::from_eth(caller);478 let from = T::CrossAccountId::from_eth(from);479 let to = T::CrossAccountId::from_eth(to);480 let token = token_id.try_into()?;481 let budget = self482 .recorder483 .weight_calls_budget(<StructureWeight<T>>::find_parent());484485 let balance = balance(&self, token, &from)?;486 ensure_single_owner(&self, token, balance)?;487488 <Pallet<T>>::transfer_from(self, &caller, &from, &to, token, balance, &budget)489 .map_err(dispatch_to_evm::<T>)?;490491 Ok(())492 }493494 495 fn approve(&mut self, _caller: Caller, _approved: Address, _token_id: U256) -> Result<()> {496 Err("not implemented".into())497 }498499 500 501 502 503 #[weight(<SelfWeightOf<T>>::set_allowance_for_all())]504 fn set_approval_for_all(505 &mut self,506 caller: Caller,507 operator: Address,508 approved: bool,509 ) -> Result<()> {510 let caller = T::CrossAccountId::from_eth(caller);511 let operator = T::CrossAccountId::from_eth(operator);512513 <Pallet<T>>::set_allowance_for_all(self, &caller, &operator, approved)514 .map_err(dispatch_to_evm::<T>)?;515 Ok(())516 }517518 519 fn get_approved(&self, _token_id: U256) -> Result<Address> {520 521 Err("not implemented".into())522 }523524 525 #[weight(<SelfWeightOf<T>>::allowance_for_all())]526 fn is_approved_for_all(&self, owner: Address, operator: Address) -> Result<bool> {527 let owner = T::CrossAccountId::from_eth(owner);528 let operator = T::CrossAccountId::from_eth(operator);529530 Ok(<Pallet<T>>::allowance_for_all(self, &owner, &operator))531 }532}533534535pub fn balance<T: Config>(536 collection: &RefungibleHandle<T>,537 token: TokenId,538 owner: &T::CrossAccountId,539) -> Result<u128> {540 collection.consume_store_reads(1)?;541 let balance = <Balance<T>>::get((collection.id, token, &owner));542 Ok(balance)543}544545546pub fn ensure_single_owner<T: Config>(547 collection: &RefungibleHandle<T>,548 token: TokenId,549 owner_balance: u128,550) -> Result<()> {551 collection.consume_store_reads(1)?;552 let total_supply = <TotalSupply<T>>::get((collection.id, token));553554 if owner_balance == 0 {555 return Err(dispatch_to_evm::<T>(556 <CommonError<T>>::MustBeTokenOwner.into(),557 ));558 }559560 if total_supply != owner_balance {561 return Err("token has multiple owners".into());562 }563 Ok(())564}565566567#[solidity_interface(name = ERC721Burnable, enum(derive(PreDispatch)), enum_attr(weight))]568impl<T: Config> RefungibleHandle<T> {569 570 571 572 573 #[weight(<SelfWeightOf<T>>::burn_item_fully())]574 fn burn(&mut self, caller: Caller, token_id: U256) -> Result<()> {575 let caller = T::CrossAccountId::from_eth(caller);576 let token = token_id.try_into()?;577578 let balance = balance(&self, token, &caller)?;579 ensure_single_owner(&self, token, balance)?;580581 <Pallet<T>>::burn(self, &caller, token, balance).map_err(dispatch_to_evm::<T>)?;582 Ok(())583 }584}585586587#[solidity_interface(name = ERC721UniqueMintable, enum(derive(PreDispatch)), enum_attr(weight))]588impl<T: Config> RefungibleHandle<T> {589 590 591 592 #[weight(<SelfWeightOf<T>>::create_item())]593 fn mint(&mut self, caller: Caller, to: Address) -> Result<U256> {594 let token_id: U256 = <TokensMinted<T>>::get(self.id)595 .checked_add(1)596 .ok_or("item id overflow")?597 .into();598 self.mint_check_id(caller, to, token_id)?;599 Ok(token_id)600 }601602 603 604 605 606 607 #[solidity(hide, rename_selector = "mint")]608 #[weight(<SelfWeightOf<T>>::create_item())]609 fn mint_check_id(&mut self, caller: Caller, to: Address, token_id: U256) -> Result<bool> {610 let caller = T::CrossAccountId::from_eth(caller);611 let to = T::CrossAccountId::from_eth(to);612 let token_id: u32 = token_id.try_into()?;613 let budget = self614 .recorder615 .weight_calls_budget(<StructureWeight<T>>::find_parent());616617 if <TokensMinted<T>>::get(self.id)618 .checked_add(1)619 .ok_or("item id overflow")?620 != token_id621 {622 return Err("item id should be next".into());623 }624625 let users = [(to.clone(), 1)]626 .into_iter()627 .collect::<BTreeMap<_, _>>()628 .try_into()629 .unwrap();630 <Pallet<T>>::create_item(631 self,632 &caller,633 CreateItemData::<T> {634 users,635 properties: CollectionPropertiesVec::default(),636 },637 &budget,638 )639 .map_err(dispatch_to_evm::<T>)?;640641 Ok(true)642 }643644 645 646 647 648 #[solidity(rename_selector = "mintWithTokenURI")]649 #[weight(<SelfWeightOf<T>>::create_item() + <SelfWeightOf<T>>::set_token_properties(1))]650 fn mint_with_token_uri(651 &mut self,652 caller: Caller,653 to: Address,654 token_uri: String,655 ) -> Result<U256> {656 let token_id: U256 = <TokensMinted<T>>::get(self.id)657 .checked_add(1)658 .ok_or("item id overflow")?659 .into();660 self.mint_with_token_uri_check_id(caller, to, token_id, token_uri)?;661 Ok(token_id)662 }663664 665 666 667 668 669 670 #[solidity(hide, rename_selector = "mintWithTokenURI")]671 #[weight(<SelfWeightOf<T>>::create_item() + <SelfWeightOf<T>>::set_token_properties(1))]672 fn mint_with_token_uri_check_id(673 &mut self,674 caller: Caller,675 to: Address,676 token_id: U256,677 token_uri: String,678 ) -> Result<bool> {679 let key = key::url();680 let permission = get_token_permission::<T>(self.id, &key)?;681 if !permission.collection_admin {682 return Err("Operation is not allowed".into());683 }684685 let caller = T::CrossAccountId::from_eth(caller);686 let to = T::CrossAccountId::from_eth(to);687 let token_id: u32 = token_id.try_into().map_err(|_| "amount overflow")?;688 let budget = self689 .recorder690 .weight_calls_budget(<StructureWeight<T>>::find_parent());691692 if <TokensMinted<T>>::get(self.id)693 .checked_add(1)694 .ok_or("item id overflow")?695 != token_id696 {697 return Err("item id should be next".into());698 }699700 let mut properties = CollectionPropertiesVec::default();701 properties702 .try_push(Property {703 key,704 value: token_uri705 .into_bytes()706 .try_into()707 .map_err(|_| "token uri is too long")?,708 })709 .map_err(|e| Error::Revert(alloc::format!("Can't add property: {:?}", e)))?;710711 let users = [(to.clone(), 1)]712 .into_iter()713 .collect::<BTreeMap<_, _>>()714 .try_into()715 .unwrap();716 <Pallet<T>>::create_item(717 self,718 &caller,719 CreateItemData::<T> { users, properties },720 &budget,721 )722 .map_err(dispatch_to_evm::<T>)?;723 Ok(true)724 }725}726727fn get_token_property<T: Config>(728 collection: &CollectionHandle<T>,729 token_id: u32,730 key: &up_data_structs::PropertyKey,731) -> Result<String> {732 collection.consume_store_reads(1)?;733 let properties = <TokenProperties<T>>::try_get((collection.id, token_id))734 .map_err(|_| Error::Revert("Token properties not found".into()))?;735 if let Some(property) = properties.get(key) {736 return Ok(String::from_utf8_lossy(property).into());737 }738739 Err("Property tokenURI not found".into())740}741742fn get_token_permission<T: Config>(743 collection_id: CollectionId,744 key: &PropertyKey,745) -> Result<PropertyPermission> {746 let token_property_permissions = CollectionPropertyPermissions::<T>::try_get(collection_id)747 .map_err(|_| Error::Revert("No permissions for collection".into()))?;748 let a = token_property_permissions749 .get(key)750 .map(Clone::clone)751 .ok_or_else(|| {752 let key = String::from_utf8(key.clone().into_inner()).unwrap_or_default();753 Error::Revert(alloc::format!("No permission for key {}", key))754 })?;755 Ok(a)756}757758759#[solidity_interface(name = ERC721UniqueExtensions, enum(derive(PreDispatch)), enum_attr(weight))]760impl<T: Config> RefungibleHandle<T>761where762 T::AccountId: From<[u8; 32]> + AsRef<[u8; 32]>,763{764 765 fn name(&self) -> Result<String> {766 Ok(decode_utf16(self.name.iter().copied())767 .map(|r| r.unwrap_or(REPLACEMENT_CHARACTER))768 .collect::<String>())769 }770771 772 fn symbol(&self) -> Result<String> {773 Ok(String::from_utf8_lossy(&self.token_prefix).into())774 }775776 777 fn description(&self) -> Result<String> {778 Ok(decode_utf16(self.description.iter().copied())779 .map(|r| r.unwrap_or(REPLACEMENT_CHARACTER))780 .collect::<String>())781 }782783 784 785 786 fn cross_owner_of(&self, token_id: U256) -> Result<eth::CrossAddress> {787 Self::token_owner(&self, token_id.try_into()?)788 .map(|o| eth::CrossAddress::from_sub_cross_account::<T>(&o))789 .or_else(|err| match err {790 TokenOwnerError::NotFound => Err(Error::Revert("token not found".into())),791 TokenOwnerError::MultipleOwners => Ok(eth::CrossAddress::from_eth(792 ADDRESS_FOR_PARTIALLY_OWNED_TOKENS,793 )),794 })795 }796797 798 799 800 801 802 fn properties(&self, token_id: U256, keys: Vec<String>) -> Result<Vec<eth::Property>> {803 let keys = keys804 .into_iter()805 .map(|key| {806 <Vec<u8>>::from(key)807 .try_into()808 .map_err(|_| Error::Revert("key too large".into()))809 })810 .collect::<Result<Vec<_>>>()?;811812 <Self as CommonCollectionOperations<T>>::token_properties(813 &self,814 token_id.try_into()?,815 if keys.is_empty() { None } else { Some(keys) },816 )817 .into_iter()818 .map(eth::Property::try_from)819 .collect::<Result<Vec<_>>>()820 }821 822 823 824 825 826 827 #[weight(<SelfWeightOf<T>>::transfer_creating_removing())]828 fn transfer(&mut self, caller: Caller, to: Address, token_id: U256) -> Result<()> {829 let caller = T::CrossAccountId::from_eth(caller);830 let to = T::CrossAccountId::from_eth(to);831 let token = token_id.try_into()?;832 let budget = self833 .recorder834 .weight_calls_budget(<StructureWeight<T>>::find_parent());835836 let balance = balance(self, token, &caller)?;837 ensure_single_owner(self, token, balance)?;838839 <Pallet<T>>::transfer(self, &caller, &to, token, balance, &budget)840 .map_err(dispatch_to_evm::<T>)?;841 Ok(())842 }843844 845 846 847 848 849 850 #[weight(<SelfWeightOf<T>>::transfer_creating_removing())]851 fn transfer_cross(852 &mut self,853 caller: Caller,854 to: eth::CrossAddress,855 token_id: U256,856 ) -> Result<()> {857 let caller = T::CrossAccountId::from_eth(caller);858 let to = to.into_sub_cross_account::<T>()?;859 let token = token_id.try_into()?;860 let budget = self861 .recorder862 .weight_calls_budget(<StructureWeight<T>>::find_parent());863864 let balance = balance(self, token, &caller)?;865 ensure_single_owner(self, token, balance)?;866867 <Pallet<T>>::transfer(self, &caller, &to, token, balance, &budget)868 .map_err(dispatch_to_evm::<T>)?;869 Ok(())870 }871872 873 874 875 876 877 878 #[weight(<SelfWeightOf<T>>::transfer_creating_removing())]879 fn transfer_from_cross(880 &mut self,881 caller: Caller,882 from: eth::CrossAddress,883 to: eth::CrossAddress,884 token_id: U256,885 ) -> Result<()> {886 let caller = T::CrossAccountId::from_eth(caller);887 let from = from.into_sub_cross_account::<T>()?;888 let to = to.into_sub_cross_account::<T>()?;889 let token_id = token_id.try_into()?;890 let budget = self891 .recorder892 .weight_calls_budget(<StructureWeight<T>>::find_parent());893894 let balance = balance(self, token_id, &from)?;895 ensure_single_owner(self, token_id, balance)?;896897 Pallet::<T>::transfer_from(self, &caller, &from, &to, token_id, balance, &budget)898 .map_err(dispatch_to_evm::<T>)?;899 Ok(())900 }901902 903 904 905 906 907 908 909 #[solidity(hide)]910 #[weight(<SelfWeightOf<T>>::burn_from())]911 fn burn_from(&mut self, caller: Caller, from: Address, token_id: U256) -> Result<()> {912 let caller = T::CrossAccountId::from_eth(caller);913 let from = T::CrossAccountId::from_eth(from);914 let token = token_id.try_into()?;915 let budget = self916 .recorder917 .weight_calls_budget(<StructureWeight<T>>::find_parent());918919 let balance = balance(self, token, &from)?;920 ensure_single_owner(self, token, balance)?;921922 <Pallet<T>>::burn_from(self, &caller, &from, token, balance, &budget)923 .map_err(dispatch_to_evm::<T>)?;924 Ok(())925 }926927 928 929 930 931 932 933 934 #[weight(<SelfWeightOf<T>>::burn_from())]935 fn burn_from_cross(936 &mut self,937 caller: Caller,938 from: eth::CrossAddress,939 token_id: U256,940 ) -> Result<()> {941 let caller = T::CrossAccountId::from_eth(caller);942 let from = from.into_sub_cross_account::<T>()?;943 let token = token_id.try_into()?;944 let budget = self945 .recorder946 .weight_calls_budget(<StructureWeight<T>>::find_parent());947948 let balance = balance(self, token, &from)?;949 ensure_single_owner(self, token, balance)?;950951 <Pallet<T>>::burn_from(self, &caller, &from, token, balance, &budget)952 .map_err(dispatch_to_evm::<T>)?;953 Ok(())954 }955956 957 fn next_token_id(&self) -> Result<U256> {958 self.consume_store_reads(1)?;959 Ok(<TokensMinted<T>>::get(self.id)960 .checked_add(1)961 .ok_or("item id overflow")?962 .into())963 }964965 966 967 968 969 970 #[solidity(hide)]971 #[weight(<SelfWeightOf<T>>::create_multiple_items(token_ids.len() as u32))]972 fn mint_bulk(&mut self, caller: Caller, to: Address, token_ids: Vec<U256>) -> Result<bool> {973 let caller = T::CrossAccountId::from_eth(caller);974 let to = T::CrossAccountId::from_eth(to);975 let mut expected_index = <TokensMinted<T>>::get(self.id)976 .checked_add(1)977 .ok_or("item id overflow")?;978 let budget = self979 .recorder980 .weight_calls_budget(<StructureWeight<T>>::find_parent());981982 let total_tokens = token_ids.len();983 for id in token_ids.into_iter() {984 let id: u32 = id.try_into().map_err(|_| "token id overflow")?;985 if id != expected_index {986 return Err("item id should be next".into());987 }988 expected_index = expected_index.checked_add(1).ok_or("item id overflow")?;989 }990 let users = [(to.clone(), 1)]991 .into_iter()992 .collect::<BTreeMap<_, _>>()993 .try_into()994 .unwrap();995 let create_item_data = CreateItemData::<T> {996 users,997 properties: CollectionPropertiesVec::default(),998 };999 let data = (0..total_tokens)1000 .map(|_| create_item_data.clone())1001 .collect();10021003 <Pallet<T>>::create_multiple_items(self, &caller, data, &budget)1004 .map_err(dispatch_to_evm::<T>)?;1005 Ok(true)1006 }10071008 1009 1010 1011 1012 1013 #[solidity(hide, rename_selector = "mintBulkWithTokenURI")]1014 #[weight(<SelfWeightOf<T>>::create_multiple_items(tokens.len() as u32) + <SelfWeightOf<T>>::set_token_properties(tokens.len() as u32))]1015 fn mint_bulk_with_token_uri(1016 &mut self,1017 caller: Caller,1018 to: Address,1019 tokens: Vec<TokenUri>,1020 ) -> Result<bool> {1021 let key = key::url();1022 let caller = T::CrossAccountId::from_eth(caller);1023 let to = T::CrossAccountId::from_eth(to);1024 let mut expected_index = <TokensMinted<T>>::get(self.id)1025 .checked_add(1)1026 .ok_or("item id overflow")?;1027 let budget = self1028 .recorder1029 .weight_calls_budget(<StructureWeight<T>>::find_parent());10301031 let mut data = Vec::with_capacity(tokens.len());1032 let users: BoundedBTreeMap<_, _, _> = [(to.clone(), 1)]1033 .into_iter()1034 .collect::<BTreeMap<_, _>>()1035 .try_into()1036 .unwrap();1037 for TokenUri { id, uri } in tokens {1038 let id: u32 = id.try_into().map_err(|_| "token id overflow")?;1039 if id != expected_index {1040 return Err("item id should be next".into());1041 }1042 expected_index = expected_index.checked_add(1).ok_or("item id overflow")?;10431044 let mut properties = CollectionPropertiesVec::default();1045 properties1046 .try_push(Property {1047 key: key.clone(),1048 value: uri1049 .into_bytes()1050 .try_into()1051 .map_err(|_| "token uri is too long")?,1052 })1053 .map_err(|e| Error::Revert(alloc::format!("Can't add property: {:?}", e)))?;10541055 let create_item_data = CreateItemData::<T> {1056 users: users.clone(),1057 properties,1058 };1059 data.push(create_item_data);1060 }10611062 <Pallet<T>>::create_multiple_items(self, &caller, data, &budget)1063 .map_err(dispatch_to_evm::<T>)?;1064 Ok(true)1065 }10661067 1068 1069 1070 1071 #[weight(<SelfWeightOf<T>>::create_item() + <SelfWeightOf<T>>::set_token_properties(properties.len() as u32))]1072 fn mint_cross(1073 &mut self,1074 caller: Caller,1075 to: eth::CrossAddress,1076 properties: Vec<eth::Property>,1077 ) -> Result<U256> {1078 let token_id = <TokensMinted<T>>::get(self.id)1079 .checked_add(1)1080 .ok_or("item id overflow")?;10811082 let to = to.into_sub_cross_account::<T>()?;10831084 let properties = properties1085 .into_iter()1086 .map(eth::Property::try_into)1087 .collect::<Result<Vec<_>>>()?1088 .try_into()1089 .map_err(|_| Error::Revert(alloc::format!("too many properties")))?;10901091 let caller = T::CrossAccountId::from_eth(caller);10921093 let budget = self1094 .recorder1095 .weight_calls_budget(<StructureWeight<T>>::find_parent());10961097 let users = [(to, 1)]1098 .into_iter()1099 .collect::<BTreeMap<_, _>>()1100 .try_into()1101 .unwrap();1102 <Pallet<T>>::create_item(1103 self,1104 &caller,1105 CreateItemData::<T> { users, properties },1106 &budget,1107 )1108 .map_err(dispatch_to_evm::<T>)?;11091110 Ok(token_id.into())1111 }11121113 1114 1115 1116 fn token_contract_address(&self, token: U256) -> Result<Address> {1117 Ok(T::EvmTokenAddressMapping::token_to_address(1118 self.id,1119 token.try_into().map_err(|_| "token id overflow")?,1120 ))1121 }11221123 1124 fn collection_helper_address(&self) -> Result<Address> {1125 Ok(T::ContractAddress::get())1126 }1127}11281129#[solidity_interface(1130 name = UniqueRefungible,1131 is(1132 ERC721,1133 ERC721Enumerable,1134 ERC721UniqueExtensions,1135 ERC721UniqueMintable,1136 ERC721Burnable,1137 ERC721Metadata(if(this.flags.erc721metadata)),1138 Collection(via(common_mut returns CollectionHandle<T>)),1139 TokenProperties,1140 ),1141 enum(derive(PreDispatch)),1142)]1143impl<T: Config> RefungibleHandle<T> where T::AccountId: From<[u8; 32]> + AsRef<[u8; 32]> {}114411451146generate_stubgen!(gen_impl, UniqueRefungibleCall<()>, true);1147generate_stubgen!(gen_iface, UniqueRefungibleCall<()>, false);11481149impl<T: Config> CommonEvmHandler for RefungibleHandle<T>1150where1151 T::AccountId: From<[u8; 32]> + AsRef<[u8; 32]>,1152{1153 const CODE: &'static [u8] = include_bytes!("./stubs/UniqueRefungible.raw");1154 fn call(1155 self,1156 handle: &mut impl PrecompileHandle,1157 ) -> Option<pallet_common::erc::PrecompileResult> {1158 call::<T, UniqueRefungibleCall<T>, _, _>(handle, self)1159 }1160}