12345678910111213141516171819use core::convert::TryInto;20use pallet_common::CollectionHandle;21use pallet_evm::account::CrossAccountId;22use pallet_fungible::Config as FungibleConfig;23use pallet_refungible::Config as RefungibleConfig;24use pallet_nonfungible::Config as NonfungibleConfig;25use pallet_unique::Config as UniqueConfig;26use up_data_structs::{CreateItemData, CreateNftData, TokenId};2728use super::common;29use crate::runtime_common::sponsoring::*;3031use pallet_refungible::{32 erc::{33 ERC721BurnableCall, ERC721Call, ERC721EnumerableCall, ERC721MetadataCall,34 ERC721UniqueExtensionsCall, ERC721UniqueMintableCall, TokenPropertiesCall,35 UniqueRefungibleCall,36 },37 erc_token::{38 ERC1633Call, ERC20Call, ERC20UniqueExtensionsCall, RefungibleTokenHandle,39 UniqueRefungibleTokenCall,40 },41};4243pub fn call_sponsor<T>(44 call: UniqueRefungibleCall<T>,45 collection: CollectionHandle<T>,46 who: &T::CrossAccountId,47) -> Option<()>48where49 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,50{51 use UniqueRefungibleCall::*;5253 match call {54 55 ERC165Call(_, _) => None,5657 ERC721Enumerable(call) => erc721::enumerable_call_sponsor(call, collection, who),58 ERC721Burnable(call) => erc721::burnable_call_sponsor(call, collection, who),59 ERC721Metadata(call) => erc721::metadata_call_sponsor(call, collection, who),60 Collection(call) => common::collection_call_sponsor(call, collection, who),61 ERC721(call) => erc721::call_sponsor(call, collection, who),62 ERC721UniqueExtensions(call) => {63 erc721::unique_extensions_call_sponsor(call, collection, who)64 }65 ERC721UniqueMintable(call) => erc721::unique_mintable_call_sponsor(call, collection, who),66 TokenProperties(call) => token_properties_call_sponsor(call, collection, who),67 }68}6970pub fn token_properties_call_sponsor<T>(71 call: TokenPropertiesCall<T>,72 collection: CollectionHandle<T>,73 who: &T::CrossAccountId,74) -> Option<()>75where76 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,77{78 use TokenPropertiesCall::*;7980 match call {81 82 ERC165Call(_, _) | Property { .. } | TokenPropertyPermissions => None,8384 85 SetTokenPropertyPermission { .. }86 | SetTokenPropertyPermissions { .. }87 | SetProperties { .. }88 | DeleteProperty { .. }89 | DeleteProperties { .. } => None,9091 SetProperty {92 token_id,93 key,94 value,95 ..96 } => {97 let token_id = TokenId::try_from(token_id).ok()?;98 withdraw_set_token_property::<T>(&collection, &who, &token_id, key.len() + value.len())99 }100 }101}102103pub fn token_call_sponsor<T>(104 call: UniqueRefungibleTokenCall<T>,105 token: RefungibleTokenHandle<T>,106 who: &T::CrossAccountId,107) -> Option<()>108where109 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,110{111 use UniqueRefungibleTokenCall::*;112113 match call {114 115 ERC165Call(_, _) => None,116117 ERC20(call) => erc20::call_sponsor(call, token, who),118 ERC20UniqueExtensions(call) => erc20::unique_extensions_call_sponsor(call, token, who),119 ERC1633(call) => erc1633::call_sponsor(call, token, who),120 }121}122123mod erc721 {124 use super::*;125126 pub fn call_sponsor<T>(127 call: ERC721Call<T>,128 collection: CollectionHandle<T>,129 _who: &T::CrossAccountId,130 ) -> Option<()>131 where132 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,133 {134 use ERC721Call::*;135136 match call {137 138 ERC165Call(_, _)139 | BalanceOf { .. }140 | OwnerOf { .. }141 | GetApproved { .. }142 | IsApprovedForAll { .. }143 | CollectionHelperAddress => None,144145 146 SafeTransferFromWithData { .. }147 | SafeTransferFrom { .. }148 | SetApprovalForAll { .. } => None,149150 TransferFrom { token_id, from, .. } => {151 let token_id = TokenId::try_from(token_id).ok()?;152 let from = T::CrossAccountId::from_eth(from);153 withdraw_transfer::<T>(&collection, &from, &token_id)154 }155156 157 Approve { .. } => None,158 }159 }160161 pub fn enumerable_call_sponsor<T>(162 call: ERC721EnumerableCall<T>,163 _collection: CollectionHandle<T>,164 _who: &T::CrossAccountId,165 ) -> Option<()>166 where167 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,168 {169 use ERC721EnumerableCall::*;170171 match call {172 173 ERC165Call(_, _) | TokenByIndex { .. } | TokenOfOwnerByIndex { .. } | TotalSupply => {174 None175 }176 }177 }178179 pub fn burnable_call_sponsor<T>(180 call: ERC721BurnableCall<T>,181 _collection: CollectionHandle<T>,182 _who: &T::CrossAccountId,183 ) -> Option<()>184 where185 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,186 {187 use ERC721BurnableCall::*;188189 match call {190 191 ERC165Call(_, _) => None,192193 194 Burn { .. } => None,195 }196 }197198 pub fn metadata_call_sponsor<T>(199 call: ERC721MetadataCall<T>,200 _collection: CollectionHandle<T>,201 _who: &T::CrossAccountId,202 ) -> Option<()>203 where204 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,205 {206 use ERC721MetadataCall::*;207208 match call {209 210 ERC165Call(_, _) | NameProxy | SymbolProxy | TokenUri { .. } => None,211 }212 }213214 pub fn unique_extensions_call_sponsor<T>(215 call: ERC721UniqueExtensionsCall<T>,216 collection: CollectionHandle<T>,217 who: &T::CrossAccountId,218 ) -> Option<()>219 where220 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,221 {222 use ERC721UniqueExtensionsCall::*;223224 match call {225 226 ERC165Call(_, _)227 | Name228 | Symbol229 | Description230 | CrossOwnerOf { .. }231 | Properties { .. }232 | NextTokenId233 | TokenContractAddress { .. } => None,234235 236 TransferCross { .. }237 | TransferFromCross { .. }238 | BurnFrom { .. }239 | BurnFromCross { .. }240 | MintCross { .. }241 | MintBulk { .. }242 | MintBulkWithTokenUri { .. } => None,243244 Transfer { token_id, .. } => {245 let token_id = TokenId::try_from(token_id).ok()?;246 withdraw_transfer::<T>(&collection, &who, &token_id)247 }248 }249 }250251 pub fn unique_mintable_call_sponsor<T>(252 call: ERC721UniqueMintableCall<T>,253 collection: CollectionHandle<T>,254 who: &T::CrossAccountId,255 ) -> Option<()>256 where257 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,258 {259 use ERC721UniqueMintableCall::*;260261 match call {262 263 ERC165Call(_, _) | MintingFinished => None,264265 266 FinishMinting => None,267268 Mint { .. }269 | MintCheckId { .. }270 | MintWithTokenUri { .. }271 | MintWithTokenUriCheckId { .. } => withdraw_create_item::<T>(272 &collection,273 &who,274 &CreateItemData::NFT(CreateNftData::default()),275 ),276 }277 }278}279280mod erc20 {281 use super::*;282283 pub fn call_sponsor<T>(284 call: ERC20Call<T>,285 token: RefungibleTokenHandle<T>,286 who: &T::CrossAccountId,287 ) -> Option<()>288 where289 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,290 {291 use ERC20Call::*;292293 match call {294 295 ERC165Call(_, _)296 | Name297 | Symbol298 | TotalSupply299 | Decimals300 | BalanceOf { .. }301 | Allowance { .. } => None,302303 Transfer { .. } => {304 let RefungibleTokenHandle(handle, token_id) = token;305 let token_id = token_id.try_into().ok()?;306 withdraw_transfer::<T>(&handle, &who, &token_id)307 }308 TransferFrom { from, .. } => {309 let RefungibleTokenHandle(handle, token_id) = token;310 let token_id = token_id.try_into().ok()?;311 let from = T::CrossAccountId::from_eth(from);312 withdraw_transfer::<T>(&handle, &from, &token_id)313 }314 Approve { .. } => {315 let RefungibleTokenHandle(handle, token_id) = token;316 let token_id = token_id.try_into().ok()?;317 withdraw_approve::<T>(&handle, who.as_sub(), &token_id)318 }319 }320 }321322 pub fn unique_extensions_call_sponsor<T>(323 call: ERC20UniqueExtensionsCall<T>,324 _token: RefungibleTokenHandle<T>,325 _who: &T::CrossAccountId,326 ) -> Option<()>327 where328 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,329 {330 use ERC20UniqueExtensionsCall::*;331332 match call {333 334 ERC165Call(_, _) => None,335336 337 BurnFrom { .. }338 | BurnFromCross { .. }339 | ApproveCross { .. }340 | TransferCross { .. }341 | TransferFromCross { .. }342 | Repartition { .. } => None,343 }344 }345}346347mod erc1633 {348 use super::*;349350 pub fn call_sponsor<T>(351 call: ERC1633Call<T>,352 _token: RefungibleTokenHandle<T>,353 _who: &T::CrossAccountId,354 ) -> Option<()>355 where356 T: UniqueConfig + FungibleConfig + NonfungibleConfig + RefungibleConfig,357 {358 use ERC1633Call::*;359360 match call {361 362 ERC165Call(_, _) | ParentToken | ParentTokenId => None,363 }364 }365}