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::{CommonEvmHandler, PrecompileResult, CollectionCall, token_uri_key},37 CollectionHandle, CollectionPropertyPermissions,38};39use pallet_evm::{account::CrossAccountId, PrecompileHandle};40use pallet_evm_coder_substrate::call;41use pallet_structure::{SelfWeightOf as StructureWeight, weights::WeightInfo as _};4243use crate::{44 AccountBalance, Config, CreateItemData, NonfungibleHandle, Pallet, TokenData, TokensMinted,45 SelfWeightOf, weights::WeightInfo, TokenProperties,46};474849#[solidity_interface(name = "TokenProperties")]50impl<T: Config> NonfungibleHandle<T> {51 52 53 54 55 56 57 fn set_token_property_permission(58 &mut self,59 caller: caller,60 key: string,61 is_mutable: bool,62 collection_admin: bool,63 token_owner: bool,64 ) -> Result<()> {65 let caller = T::CrossAccountId::from_eth(caller);66 <Pallet<T>>::set_property_permission(67 self,68 &caller,69 PropertyKeyPermission {70 key: <Vec<u8>>::from(key)71 .try_into()72 .map_err(|_| "too long key")?,73 permission: PropertyPermission {74 mutable: is_mutable,75 collection_admin,76 token_owner,77 },78 },79 )80 .map_err(dispatch_to_evm::<T>)81 }8283 84 85 86 87 88 fn set_property(89 &mut self,90 caller: caller,91 token_id: uint256,92 key: string,93 value: bytes,94 ) -> Result<()> {95 let caller = T::CrossAccountId::from_eth(caller);96 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;97 let key = <Vec<u8>>::from(key)98 .try_into()99 .map_err(|_| "key too long")?;100 let value = value.try_into().map_err(|_| "value too long")?;101102 let nesting_budget = self103 .recorder104 .weight_calls_budget(<StructureWeight<T>>::find_parent());105106 <Pallet<T>>::set_token_property(107 self,108 &caller,109 TokenId(token_id),110 Property { key, value },111 &nesting_budget,112 )113 .map_err(dispatch_to_evm::<T>)114 }115116 117 118 119 120 fn delete_property(&mut self, token_id: uint256, caller: caller, key: string) -> Result<()> {121 let caller = T::CrossAccountId::from_eth(caller);122 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;123 let key = <Vec<u8>>::from(key)124 .try_into()125 .map_err(|_| "key too long")?;126127 let nesting_budget = self128 .recorder129 .weight_calls_budget(<StructureWeight<T>>::find_parent());130131 <Pallet<T>>::delete_token_property(self, &caller, TokenId(token_id), key, &nesting_budget)132 .map_err(dispatch_to_evm::<T>)133 }134135 136 137 138 139 140 fn property(&self, token_id: uint256, key: string) -> Result<bytes> {141 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;142 let key = <Vec<u8>>::from(key)143 .try_into()144 .map_err(|_| "key too long")?;145146 let props = <TokenProperties<T>>::get((self.id, token_id));147 let prop = props.get(&key).ok_or("key not found")?;148149 Ok(prop.to_vec())150 }151}152153#[derive(ToLog)]154pub enum ERC721Events {155 156 157 158 159 160 Transfer {161 #[indexed]162 from: address,163 #[indexed]164 to: address,165 #[indexed]166 token_id: uint256,167 },168 169 170 171 172 Approval {173 #[indexed]174 owner: address,175 #[indexed]176 approved: address,177 #[indexed]178 token_id: uint256,179 },180 181 182 #[allow(dead_code)]183 ApprovalForAll {184 #[indexed]185 owner: address,186 #[indexed]187 operator: address,188 approved: bool,189 },190}191192#[derive(ToLog)]193pub enum ERC721MintableEvents {194 #[allow(dead_code)]195 MintingFinished {},196}197198199200#[solidity_interface(name = "ERC721Metadata")]201impl<T: Config> NonfungibleHandle<T> {202 203 fn name(&self) -> Result<string> {204 Ok(decode_utf16(self.name.iter().copied())205 .map(|r| r.unwrap_or(REPLACEMENT_CHARACTER))206 .collect::<string>())207 }208209 210 fn symbol(&self) -> Result<string> {211 Ok(string::from_utf8_lossy(&self.token_prefix).into())212 }213214 215 216 217 218 219 #[solidity(rename_selector = "tokenURI")]220 fn token_uri(&self, token_id: uint256) -> Result<string> {221 let key = token_uri_key();222 if !has_token_permission::<T>(self.id, &key) {223 return Err("No tokenURI permission".into());224 }225226 self.consume_store_reads(1)?;227 let token_id: u32 = token_id.try_into().map_err(|_| "token id overflow")?;228229 let properties = <TokenProperties<T>>::try_get((self.id, token_id))230 .map_err(|_| Error::Revert("Token properties not found".into()))?;231 if let Some(property) = properties.get(&key) {232 return Ok(string::from_utf8_lossy(property).into());233 }234235 Err("Property tokenURI not found".into())236 }237}238239240241#[solidity_interface(name = "ERC721Enumerable")]242impl<T: Config> NonfungibleHandle<T> {243 244 245 246 247 fn token_by_index(&self, index: uint256) -> Result<uint256> {248 Ok(index)249 }250251 252 fn token_of_owner_by_index(&self, _owner: address, _index: uint256) -> Result<uint256> {253 254 Err("not implemented".into())255 }256257 258 259 260 fn total_supply(&self) -> Result<uint256> {261 self.consume_store_reads(1)?;262 Ok(<Pallet<T>>::total_supply(self).into())263 }264}265266267268#[solidity_interface(name = "ERC721", events(ERC721Events))]269impl<T: Config> NonfungibleHandle<T> {270 271 272 273 274 275 fn balance_of(&self, owner: address) -> Result<uint256> {276 self.consume_store_reads(1)?;277 let owner = T::CrossAccountId::from_eth(owner);278 let balance = <AccountBalance<T>>::get((self.id, owner));279 Ok(balance.into())280 }281 282 283 284 285 286 fn owner_of(&self, token_id: uint256) -> Result<address> {287 self.consume_store_reads(1)?;288 let token: TokenId = token_id.try_into()?;289 Ok(*<TokenData<T>>::get((self.id, token))290 .ok_or("token not found")?291 .owner292 .as_eth())293 }294 295 fn safe_transfer_from_with_data(296 &mut self,297 _from: address,298 _to: address,299 _token_id: uint256,300 _data: bytes,301 _value: value,302 ) -> Result<void> {303 304 Err("not implemented".into())305 }306 307 fn safe_transfer_from(308 &mut self,309 _from: address,310 _to: address,311 _token_id: uint256,312 _value: value,313 ) -> Result<void> {314 315 Err("not implemented".into())316 }317318 319 320 321 322 323 324 325 326 327 328 #[weight(<SelfWeightOf<T>>::transfer_from())]329 fn transfer_from(330 &mut self,331 caller: caller,332 from: address,333 to: address,334 token_id: uint256,335 _value: value,336 ) -> Result<void> {337 let caller = T::CrossAccountId::from_eth(caller);338 let from = T::CrossAccountId::from_eth(from);339 let to = T::CrossAccountId::from_eth(to);340 let token = token_id.try_into()?;341 let budget = self342 .recorder343 .weight_calls_budget(<StructureWeight<T>>::find_parent());344345 <Pallet<T>>::transfer_from(self, &caller, &from, &to, token, &budget)346 .map_err(dispatch_to_evm::<T>)?;347 Ok(())348 }349350 351 352 353 354 355 356 #[weight(<SelfWeightOf<T>>::approve())]357 fn approve(358 &mut self,359 caller: caller,360 approved: address,361 token_id: uint256,362 _value: value,363 ) -> Result<void> {364 let caller = T::CrossAccountId::from_eth(caller);365 let approved = T::CrossAccountId::from_eth(approved);366 let token = token_id.try_into()?;367368 <Pallet<T>>::set_allowance(self, &caller, token, Some(&approved))369 .map_err(dispatch_to_evm::<T>)?;370 Ok(())371 }372373 374 fn set_approval_for_all(375 &mut self,376 _caller: caller,377 _operator: address,378 _approved: bool,379 ) -> Result<void> {380 381 Err("not implemented".into())382 }383384 385 fn get_approved(&self, _token_id: uint256) -> Result<address> {386 387 Err("not implemented".into())388 }389390 391 fn is_approved_for_all(&self, _owner: address, _operator: address) -> Result<address> {392 393 Err("not implemented".into())394 }395}396397398#[solidity_interface(name = "ERC721Burnable")]399impl<T: Config> NonfungibleHandle<T> {400 401 402 403 404 #[weight(<SelfWeightOf<T>>::burn_item())]405 fn burn(&mut self, caller: caller, token_id: uint256) -> Result<void> {406 let caller = T::CrossAccountId::from_eth(caller);407 let token = token_id.try_into()?;408409 <Pallet<T>>::burn(self, &caller, token).map_err(dispatch_to_evm::<T>)?;410 Ok(())411 }412}413414415#[solidity_interface(name = "ERC721Mintable", events(ERC721MintableEvents))]416impl<T: Config> NonfungibleHandle<T> {417 fn minting_finished(&self) -> Result<bool> {418 Ok(false)419 }420421 422 423 424 425 426 #[weight(<SelfWeightOf<T>>::create_item())]427 fn mint(&mut self, caller: caller, to: address, token_id: uint256) -> Result<bool> {428 let caller = T::CrossAccountId::from_eth(caller);429 let to = T::CrossAccountId::from_eth(to);430 let token_id: u32 = token_id.try_into()?;431 let budget = self432 .recorder433 .weight_calls_budget(<StructureWeight<T>>::find_parent());434435 if <TokensMinted<T>>::get(self.id)436 .checked_add(1)437 .ok_or("item id overflow")?438 != token_id439 {440 return Err("item id should be next".into());441 }442443 <Pallet<T>>::create_item(444 self,445 &caller,446 CreateItemData::<T> {447 properties: BoundedVec::default(),448 owner: to,449 },450 &budget,451 )452 .map_err(dispatch_to_evm::<T>)?;453454 Ok(true)455 }456457 458 459 460 461 462 463 #[solidity(rename_selector = "mintWithTokenURI")]464 #[weight(<SelfWeightOf<T>>::create_item())]465 fn mint_with_token_uri(466 &mut self,467 caller: caller,468 to: address,469 token_id: uint256,470 token_uri: string,471 ) -> Result<bool> {472 let key = token_uri_key();473 let permission = get_token_permission::<T>(self.id, &key)?;474 if !permission.collection_admin {475 return Err("Operation is not allowed".into());476 }477478 let caller = T::CrossAccountId::from_eth(caller);479 let to = T::CrossAccountId::from_eth(to);480 let token_id: u32 = token_id.try_into().map_err(|_| "amount overflow")?;481 let budget = self482 .recorder483 .weight_calls_budget(<StructureWeight<T>>::find_parent());484485 if <TokensMinted<T>>::get(self.id)486 .checked_add(1)487 .ok_or("item id overflow")?488 != token_id489 {490 return Err("item id should be next".into());491 }492493 let mut properties = CollectionPropertiesVec::default();494 properties495 .try_push(Property {496 key,497 value: token_uri498 .into_bytes()499 .try_into()500 .map_err(|_| "token uri is too long")?,501 })502 .map_err(|e| Error::Revert(alloc::format!("Can't add property: {:?}", e)))?;503504 <Pallet<T>>::create_item(505 self,506 &caller,507 CreateItemData::<T> {508 properties,509 owner: to,510 },511 &budget,512 )513 .map_err(dispatch_to_evm::<T>)?;514 Ok(true)515 }516517 518 fn finish_minting(&mut self, _caller: caller) -> Result<bool> {519 Err("not implementable".into())520 }521}522523fn get_token_permission<T: Config>(524 collection_id: CollectionId,525 key: &PropertyKey,526) -> Result<PropertyPermission> {527 let token_property_permissions = CollectionPropertyPermissions::<T>::try_get(collection_id)528 .map_err(|_| Error::Revert("No permissions for collection".into()))?;529 let a = token_property_permissions530 .get(key)531 .map(|p| p.clone())532 .ok_or_else(|| Error::Revert("No permission".into()))?;533 Ok(a)534}535536fn has_token_permission<T: Config>(collection_id: CollectionId, key: &PropertyKey) -> bool {537 if let Ok(token_property_permissions) =538 CollectionPropertyPermissions::<T>::try_get(collection_id)539 {540 return token_property_permissions.contains_key(key);541 }542543 false544}545546547#[solidity_interface(name = "ERC721UniqueExtensions")]548impl<T: Config> NonfungibleHandle<T> {549 550 551 552 553 554 555 #[weight(<SelfWeightOf<T>>::transfer())]556 fn transfer(557 &mut self,558 caller: caller,559 to: address,560 token_id: uint256,561 _value: value,562 ) -> Result<void> {563 let caller = T::CrossAccountId::from_eth(caller);564 let to = T::CrossAccountId::from_eth(to);565 let token = token_id.try_into()?;566 let budget = self567 .recorder568 .weight_calls_budget(<StructureWeight<T>>::find_parent());569570 <Pallet<T>>::transfer(self, &caller, &to, token, &budget).map_err(dispatch_to_evm::<T>)?;571 Ok(())572 }573574 575 576 577 578 579 580 581 #[weight(<SelfWeightOf<T>>::burn_from())]582 fn burn_from(583 &mut self,584 caller: caller,585 from: address,586 token_id: uint256,587 _value: value,588 ) -> Result<void> {589 let caller = T::CrossAccountId::from_eth(caller);590 let from = T::CrossAccountId::from_eth(from);591 let token = token_id.try_into()?;592 let budget = self593 .recorder594 .weight_calls_budget(<StructureWeight<T>>::find_parent());595596 <Pallet<T>>::burn_from(self, &caller, &from, token, &budget)597 .map_err(dispatch_to_evm::<T>)?;598 Ok(())599 }600601 602 fn next_token_id(&self) -> Result<uint256> {603 self.consume_store_reads(1)?;604 Ok(<TokensMinted<T>>::get(self.id)605 .checked_add(1)606 .ok_or("item id overflow")?607 .into())608 }609610 611 612 613 614 615 #[weight(<SelfWeightOf<T>>::create_multiple_items(token_ids.len() as u32))]616 fn mint_bulk(&mut self, caller: caller, to: address, token_ids: Vec<uint256>) -> Result<bool> {617 let caller = T::CrossAccountId::from_eth(caller);618 let to = T::CrossAccountId::from_eth(to);619 let mut expected_index = <TokensMinted<T>>::get(self.id)620 .checked_add(1)621 .ok_or("item id overflow")?;622 let budget = self623 .recorder624 .weight_calls_budget(<StructureWeight<T>>::find_parent());625626 let total_tokens = token_ids.len();627 for id in token_ids.into_iter() {628 let id: u32 = id.try_into().map_err(|_| "token id overflow")?;629 if id != expected_index {630 return Err("item id should be next".into());631 }632 expected_index = expected_index.checked_add(1).ok_or("item id overflow")?;633 }634 let data = (0..total_tokens)635 .map(|_| CreateItemData::<T> {636 properties: BoundedVec::default(),637 owner: to.clone(),638 })639 .collect();640641 <Pallet<T>>::create_multiple_items(self, &caller, data, &budget)642 .map_err(dispatch_to_evm::<T>)?;643 Ok(true)644 }645646 647 648 649 650 651 #[solidity(rename_selector = "mintBulkWithTokenURI")]652 #[weight(<SelfWeightOf<T>>::create_multiple_items(tokens.len() as u32))]653 fn mint_bulk_with_token_uri(654 &mut self,655 caller: caller,656 to: address,657 tokens: Vec<(uint256, string)>,658 ) -> Result<bool> {659 let key = token_uri_key();660 let caller = T::CrossAccountId::from_eth(caller);661 let to = T::CrossAccountId::from_eth(to);662 let mut expected_index = <TokensMinted<T>>::get(self.id)663 .checked_add(1)664 .ok_or("item id overflow")?;665 let budget = self666 .recorder667 .weight_calls_budget(<StructureWeight<T>>::find_parent());668669 let mut data = Vec::with_capacity(tokens.len());670 for (id, token_uri) in tokens {671 let id: u32 = id.try_into().map_err(|_| "token id overflow")?;672 if id != expected_index {673 return Err("item id should be next".into());674 }675 expected_index = expected_index.checked_add(1).ok_or("item id overflow")?;676677 let mut properties = CollectionPropertiesVec::default();678 properties679 .try_push(Property {680 key: key.clone(),681 value: token_uri682 .into_bytes()683 .try_into()684 .map_err(|_| "token uri is too long")?,685 })686 .map_err(|e| Error::Revert(alloc::format!("Can't add property: {:?}", e)))?;687688 data.push(CreateItemData::<T> {689 properties,690 owner: to.clone(),691 });692 }693694 <Pallet<T>>::create_multiple_items(self, &caller, data, &budget)695 .map_err(dispatch_to_evm::<T>)?;696 Ok(true)697 }698}699700#[solidity_interface(701 name = "UniqueNFT",702 is(703 ERC721,704 ERC721Metadata,705 ERC721Enumerable,706 ERC721UniqueExtensions,707 ERC721Mintable,708 ERC721Burnable,709 via("CollectionHandle<T>", common_mut, Collection),710 TokenProperties,711 )712)]713impl<T: Config> NonfungibleHandle<T> where T::AccountId: From<[u8; 32]> {}714715716generate_stubgen!(gen_impl, UniqueNFTCall<()>, true);717generate_stubgen!(gen_iface, UniqueNFTCall<()>, false);718719impl<T: Config> CommonEvmHandler for NonfungibleHandle<T>720where721 T::AccountId: From<[u8; 32]>,722{723 const CODE: &'static [u8] = include_bytes!("./stubs/UniqueNFT.raw");724725 fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult> {726 call::<T, UniqueNFTCall<T>, _, _>(handle, self)727 }728}