123456789101112131415161718use std::sync::Arc;1920use codec::{Decode, Encode};21use jsonrpsee::{22 core::{RpcResult as Result},23 proc_macros::rpc,24};25use anyhow::anyhow;26use sp_runtime::traits::{AtLeast32BitUnsigned, Member};27use up_data_structs::{28 RpcCollection, CollectionId, CollectionStats, CollectionLimits, TokenId, Property,29 PropertyKeyPermission, TokenData, TokenChild,30};31use sp_api::{BlockId, BlockT, ProvideRuntimeApi, ApiExt};32use sp_blockchain::HeaderBackend;33use up_rpc::UniqueApi as UniqueRuntimeApi;34use app_promotion_rpc::AppPromotionApi as AppPromotionRuntimeApi;353637use rmrk_rpc::RmrkApi as RmrkRuntimeApi;38use up_data_structs::{39 RmrkCollectionId, RmrkNftId, RmrkBaseId, RmrkNftChild, RmrkThemeName, RmrkResourceId,40};4142pub use app_promotion_unique_rpc::AppPromotionApiServer;43pub use rmrk_unique_rpc::RmrkApiServer;4445#[rpc(server)]46#[async_trait]47pub trait UniqueApi<BlockHash, CrossAccountId, AccountId> {48 49 #[method(name = "unique_accountTokens")]50 fn account_tokens(51 &self,52 collection: CollectionId,53 account: CrossAccountId,54 at: Option<BlockHash>,55 ) -> Result<Vec<TokenId>>;5657 58 #[method(name = "unique_collectionTokens")]59 fn collection_tokens(60 &self,61 collection: CollectionId,62 at: Option<BlockHash>,63 ) -> Result<Vec<TokenId>>;6465 66 #[method(name = "unique_tokenExists")]67 fn token_exists(68 &self,69 collection: CollectionId,70 token: TokenId,71 at: Option<BlockHash>,72 ) -> Result<bool>;7374 75 #[method(name = "unique_tokenOwner")]76 fn token_owner(77 &self,78 collection: CollectionId,79 token: TokenId,80 at: Option<BlockHash>,81 ) -> Result<Option<CrossAccountId>>;8283 84 #[method(name = "unique_tokenOwners")]85 fn token_owners(86 &self,87 collection: CollectionId,88 token: TokenId,89 at: Option<BlockHash>,90 ) -> Result<Vec<CrossAccountId>>;9192 93 #[method(name = "unique_topmostTokenOwner")]94 fn topmost_token_owner(95 &self,96 collection: CollectionId,97 token: TokenId,98 at: Option<BlockHash>,99 ) -> Result<Option<CrossAccountId>>;100101 102 #[method(name = "unique_tokenChildren")]103 fn token_children(104 &self,105 collection: CollectionId,106 token: TokenId,107 at: Option<BlockHash>,108 ) -> Result<Vec<TokenChild>>;109110 111 #[method(name = "unique_collectionProperties")]112 fn collection_properties(113 &self,114 collection: CollectionId,115 keys: Option<Vec<String>>,116 at: Option<BlockHash>,117 ) -> Result<Vec<Property>>;118119 120 #[method(name = "unique_tokenProperties")]121 fn token_properties(122 &self,123 collection: CollectionId,124 token_id: TokenId,125 keys: Option<Vec<String>>,126 at: Option<BlockHash>,127 ) -> Result<Vec<Property>>;128129 130 #[method(name = "unique_propertyPermissions")]131 fn property_permissions(132 &self,133 collection: CollectionId,134 keys: Option<Vec<String>>,135 at: Option<BlockHash>,136 ) -> Result<Vec<PropertyKeyPermission>>;137138 139 #[method(name = "unique_tokenData")]140 fn token_data(141 &self,142 collection: CollectionId,143 token_id: TokenId,144 keys: Option<Vec<String>>,145 at: Option<BlockHash>,146 ) -> Result<TokenData<CrossAccountId>>;147148 149 #[method(name = "unique_totalSupply")]150 fn total_supply(&self, collection: CollectionId, at: Option<BlockHash>) -> Result<u32>;151152 153 #[method(name = "unique_accountBalance")]154 fn account_balance(155 &self,156 collection: CollectionId,157 account: CrossAccountId,158 at: Option<BlockHash>,159 ) -> Result<u32>;160161 162 #[method(name = "unique_balance")]163 fn balance(164 &self,165 collection: CollectionId,166 account: CrossAccountId,167 token: TokenId,168 at: Option<BlockHash>,169 ) -> Result<String>;170171 172 #[method(name = "unique_allowance")]173 fn allowance(174 &self,175 collection: CollectionId,176 sender: CrossAccountId,177 spender: CrossAccountId,178 token: TokenId,179 at: Option<BlockHash>,180 ) -> Result<String>;181182 183 #[method(name = "unique_adminlist")]184 fn adminlist(185 &self,186 collection: CollectionId,187 at: Option<BlockHash>,188 ) -> Result<Vec<CrossAccountId>>;189190 191 #[method(name = "unique_allowlist")]192 fn allowlist(193 &self,194 collection: CollectionId,195 at: Option<BlockHash>,196 ) -> Result<Vec<CrossAccountId>>;197198 199 #[method(name = "unique_allowed")]200 fn allowed(201 &self,202 collection: CollectionId,203 user: CrossAccountId,204 at: Option<BlockHash>,205 ) -> Result<bool>;206207 208 #[method(name = "unique_lastTokenId")]209 fn last_token_id(&self, collection: CollectionId, at: Option<BlockHash>) -> Result<TokenId>;210211 212 #[method(name = "unique_collectionById")]213 fn collection_by_id(214 &self,215 collection: CollectionId,216 at: Option<BlockHash>,217 ) -> Result<Option<RpcCollection<AccountId>>>;218219 220 #[method(name = "unique_collectionStats")]221 fn collection_stats(&self, at: Option<BlockHash>) -> Result<CollectionStats>;222223 224 #[method(name = "unique_nextSponsored")]225 fn next_sponsored(226 &self,227 collection: CollectionId,228 account: CrossAccountId,229 token: TokenId,230 at: Option<BlockHash>,231 ) -> Result<Option<u64>>;232233 234 #[method(name = "unique_effectiveCollectionLimits")]235 fn effective_collection_limits(236 &self,237 collection_id: CollectionId,238 at: Option<BlockHash>,239 ) -> Result<Option<CollectionLimits>>;240241 242 #[method(name = "unique_totalPieces")]243 fn total_pieces(244 &self,245 collection_id: CollectionId,246 token_id: TokenId,247 at: Option<BlockHash>,248 ) -> Result<Option<String>>;249250 251 #[method(name = "unique_allowanceForAll")]252 fn allowance_for_all(253 &self,254 collection: CollectionId,255 owner: CrossAccountId,256 operator: CrossAccountId,257 at: Option<BlockHash>,258 ) -> Result<bool>;259}260261mod app_promotion_unique_rpc {262 use super::*;263264 #[rpc(server)]265 #[async_trait]266 pub trait AppPromotionApi<BlockHash, BlockNumber, CrossAccountId, AccountId> {267 268 #[method(name = "appPromotion_totalStaked")]269 fn total_staked(270 &self,271 staker: Option<CrossAccountId>,272 at: Option<BlockHash>,273 ) -> Result<String>;274275 276 #[method(name = "appPromotion_totalStakedPerBlock")]277 fn total_staked_per_block(278 &self,279 staker: CrossAccountId,280 at: Option<BlockHash>,281 ) -> Result<Vec<(BlockNumber, String)>>;282283 284 #[method(name = "appPromotion_pendingUnstake")]285 fn pending_unstake(286 &self,287 staker: Option<CrossAccountId>,288 at: Option<BlockHash>,289 ) -> Result<String>;290291 292 #[method(name = "appPromotion_pendingUnstakePerBlock")]293 fn pending_unstake_per_block(294 &self,295 staker: CrossAccountId,296 at: Option<BlockHash>,297 ) -> Result<Vec<(BlockNumber, String)>>;298 }299}300301mod rmrk_unique_rpc {302 use super::*;303304 #[rpc(server)]305 #[async_trait]306 pub trait RmrkApi<307 BlockHash,308 AccountId,309 CollectionInfo,310 NftInfo,311 ResourceInfo,312 PropertyInfo,313 BaseInfo,314 PartType,315 Theme,316 >317 {318 319 #[method(name = "rmrk_lastCollectionIdx")]320 fn last_collection_idx(&self, at: Option<BlockHash>) -> Result<RmrkCollectionId>;321322 323 #[method(name = "rmrk_collectionById")]324 fn collection_by_id(325 &self,326 id: RmrkCollectionId,327 at: Option<BlockHash>,328 ) -> Result<Option<CollectionInfo>>;329330 331 #[method(name = "rmrk_nftById")]332 fn nft_by_id(333 &self,334 collection_id: RmrkCollectionId,335 nft_id: RmrkNftId,336 at: Option<BlockHash>,337 ) -> Result<Option<NftInfo>>;338339 340 #[method(name = "rmrk_accountTokens")]341 fn account_tokens(342 &self,343 account_id: AccountId,344 collection_id: RmrkCollectionId,345 at: Option<BlockHash>,346 ) -> Result<Vec<RmrkNftId>>;347348 349 #[method(name = "rmrk_nftChildren")]350 fn nft_children(351 &self,352 collection_id: RmrkCollectionId,353 nft_id: RmrkNftId,354 at: Option<BlockHash>,355 ) -> Result<Vec<RmrkNftChild>>;356357 358 #[method(name = "rmrk_collectionProperties")]359 fn collection_properties(360 &self,361 collection_id: RmrkCollectionId,362 filter_keys: Option<Vec<String>>,363 at: Option<BlockHash>,364 ) -> Result<Vec<PropertyInfo>>;365366 367 #[method(name = "rmrk_nftProperties")]368 fn nft_properties(369 &self,370 collection_id: RmrkCollectionId,371 nft_id: RmrkNftId,372 filter_keys: Option<Vec<String>>,373 at: Option<BlockHash>,374 ) -> Result<Vec<PropertyInfo>>;375376 377 #[method(name = "rmrk_nftResources")]378 fn nft_resources(379 &self,380 collection_id: RmrkCollectionId,381 nft_id: RmrkNftId,382 at: Option<BlockHash>,383 ) -> Result<Vec<ResourceInfo>>;384385 386 #[method(name = "rmrk_nftResourcePriority")]387 fn nft_resource_priority(388 &self,389 collection_id: RmrkCollectionId,390 nft_id: RmrkNftId,391 resource_id: RmrkResourceId,392 at: Option<BlockHash>,393 ) -> Result<Option<u32>>;394395 396 #[method(name = "rmrk_base")]397 fn base(&self, base_id: RmrkBaseId, at: Option<BlockHash>) -> Result<Option<BaseInfo>>;398399 400 #[method(name = "rmrk_baseParts")]401 fn base_parts(&self, base_id: RmrkBaseId, at: Option<BlockHash>) -> Result<Vec<PartType>>;402403 404 #[method(name = "rmrk_themeNames")]405 fn theme_names(406 &self,407 base_id: RmrkBaseId,408 at: Option<BlockHash>,409 ) -> Result<Vec<RmrkThemeName>>;410411 412 #[method(name = "rmrk_themes")]413 fn theme(414 &self,415 base_id: RmrkBaseId,416 theme_name: String,417 filter_keys: Option<Vec<String>>,418 at: Option<BlockHash>,419 ) -> Result<Option<Theme>>;420 }421}422423macro_rules! define_struct_for_server_api {424 ($name:ident) => {425 pub struct $name<C, P> {426 client: Arc<C>,427 _marker: std::marker::PhantomData<P>,428 }429430 impl<C, P> $name<C, P> {431 pub fn new(client: Arc<C>) -> Self {432 Self {433 client,434 _marker: Default::default(),435 }436 }437 }438 };439}440441define_struct_for_server_api!(Unique);442define_struct_for_server_api!(AppPromotion);443define_struct_for_server_api!(Rmrk);444445macro_rules! pass_method {446 (447 $method_name:ident(448 $($(#[map(|$map_arg:ident| $map:expr)])? $name:ident: $ty:ty),* $(,)?449 ) -> $result:ty $(=> $mapper:expr)?,450 451 $runtime_api_macro:ident452 $(; changed_in $ver:expr, $changed_method_name:ident ($($changed_name:expr), * $(,)?) => $fixer:expr)*453 ) => {454 fn $method_name(455 &self,456 $(457 $name: $ty,458 )*459 at: Option<<Block as BlockT>::Hash>,460 ) -> Result<$result> {461 let api = self.client.runtime_api();462 let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));463 let _api_version = if let Ok(Some(api_version)) =464 api.api_version::<$runtime_api_macro!()>(&at)465 {466 api_version467 } else {468 469 return Err(anyhow!("api is not available").into())470 };471472 let result = $(if _api_version < $ver {473 api.$changed_method_name(&at, $($changed_name),*).map(|r| r.map($fixer))474 } else)*475 { api.$method_name(&at, $($((|$map_arg: $ty| $map))? ($name)),*) };476477 Ok(result478 .map_err(|e| anyhow!("unable to query: {e}"))?479 .map_err(|e| anyhow!("runtime error: {e:?}"))$(.map($mapper))??)480 }481 };482}483484macro_rules! unique_api {485 () => {486 dyn UniqueRuntimeApi<Block, CrossAccountId, AccountId>487 };488}489490macro_rules! app_promotion_api {491 () => {492 dyn AppPromotionRuntimeApi<Block, BlockNumber, CrossAccountId, AccountId>493 };494}495496macro_rules! rmrk_api {497 () => {498 dyn RmrkRuntimeApi<Block, AccountId, CollectionInfo, NftInfo, ResourceInfo, PropertyInfo, BaseInfo, PartType, Theme>499 };500}501502#[allow(deprecated)]503impl<C, Block, CrossAccountId, AccountId>504 UniqueApiServer<<Block as BlockT>::Hash, CrossAccountId, AccountId> for Unique<C, Block>505where506 Block: BlockT,507 AccountId: Decode,508 C: 'static + ProvideRuntimeApi<Block> + HeaderBackend<Block>,509 C::Api: UniqueRuntimeApi<Block, CrossAccountId, AccountId>,510 CrossAccountId: pallet_evm::account::CrossAccountId<AccountId>,511{512 pass_method!(513 account_tokens(collection: CollectionId, account: CrossAccountId) -> Vec<TokenId>, unique_api514 );515 pass_method!(516 collection_tokens(collection: CollectionId) -> Vec<TokenId>, unique_api517 );518 pass_method!(519 token_exists(collection: CollectionId, token: TokenId) -> bool, unique_api520 );521 pass_method!(522 token_owner(collection: CollectionId, token: TokenId) -> Option<CrossAccountId>, unique_api523 );524 pass_method!(525 topmost_token_owner(collection: CollectionId, token: TokenId) -> Option<CrossAccountId>, unique_api526 );527 pass_method!(token_children(collection: CollectionId, token: TokenId) -> Vec<TokenChild>, unique_api);528 pass_method!(total_supply(collection: CollectionId) -> u32, unique_api);529 pass_method!(account_balance(collection: CollectionId, account: CrossAccountId) -> u32, unique_api);530 pass_method!(balance(collection: CollectionId, account: CrossAccountId, token: TokenId) -> String => |v| v.to_string(), unique_api);531 pass_method!(532 allowance(collection: CollectionId, sender: CrossAccountId, spender: CrossAccountId, token: TokenId) -> String => |v| v.to_string(),533 unique_api534 );535536 pass_method!(collection_properties(537 collection: CollectionId,538539 #[map(|keys| string_keys_to_bytes_keys(keys))]540 keys: Option<Vec<String>>541 ) -> Vec<Property>, unique_api);542543 pass_method!(token_properties(544 collection: CollectionId,545 token_id: TokenId,546547 #[map(|keys| string_keys_to_bytes_keys(keys))]548 keys: Option<Vec<String>>549 ) -> Vec<Property>, unique_api);550551 pass_method!(property_permissions(552 collection: CollectionId,553554 #[map(|keys| string_keys_to_bytes_keys(keys))]555 keys: Option<Vec<String>>556 ) -> Vec<PropertyKeyPermission>, unique_api);557558 pass_method!(559 token_data(560 collection: CollectionId,561 token_id: TokenId,562563 #[map(|keys| string_keys_to_bytes_keys(keys))]564 keys: Option<Vec<String>>,565 ) -> TokenData<CrossAccountId>, unique_api;566 changed_in 3, token_data_before_version_3(collection, token_id, string_keys_to_bytes_keys(keys)) => |value| value.into()567 );568569 pass_method!(adminlist(collection: CollectionId) -> Vec<CrossAccountId>, unique_api);570 pass_method!(allowlist(collection: CollectionId) -> Vec<CrossAccountId>, unique_api);571 pass_method!(allowed(collection: CollectionId, user: CrossAccountId) -> bool, unique_api);572 pass_method!(last_token_id(collection: CollectionId) -> TokenId, unique_api);573 pass_method!(574 collection_by_id(collection: CollectionId) -> Option<RpcCollection<AccountId>>, unique_api;575 changed_in 3, collection_by_id_before_version_3(collection) => |value| value.map(|coll| coll.into())576 );577 pass_method!(collection_stats() -> CollectionStats, unique_api);578 pass_method!(next_sponsored(collection: CollectionId, account: CrossAccountId, token: TokenId) -> Option<u64>, unique_api);579 pass_method!(effective_collection_limits(collection_id: CollectionId) -> Option<CollectionLimits>, unique_api);580 pass_method!(total_pieces(collection_id: CollectionId, token_id: TokenId) -> Option<String> => |o| o.map(|number| number.to_string()) , unique_api);581 pass_method!(token_owners(collection: CollectionId, token: TokenId) -> Vec<CrossAccountId>, unique_api);582 pass_method!(allowance_for_all(collection: CollectionId, owner: CrossAccountId, operator: CrossAccountId) -> bool, unique_api);583}584585impl<C, Block, BlockNumber, CrossAccountId, AccountId>586 app_promotion_unique_rpc::AppPromotionApiServer<587 <Block as BlockT>::Hash,588 BlockNumber,589 CrossAccountId,590 AccountId,591 > for AppPromotion<C, Block>592where593 Block: BlockT,594 BlockNumber: Decode + Member + AtLeast32BitUnsigned,595 AccountId: Decode,596 C: 'static + ProvideRuntimeApi<Block> + HeaderBackend<Block>,597 CrossAccountId: pallet_evm::account::CrossAccountId<AccountId>,598 C::Api: AppPromotionRuntimeApi<Block, BlockNumber, CrossAccountId, AccountId>,599{600 pass_method!(total_staked(staker: Option<CrossAccountId>) -> String => |v| v.to_string(), app_promotion_api);601 pass_method!(total_staked_per_block(staker: CrossAccountId) -> Vec<(BlockNumber, String)> =>602 |v| v603 .into_iter()604 .map(|(b, a)| (b, a.to_string()))605 .collect::<Vec<_>>(), app_promotion_api);606 pass_method!(pending_unstake(staker: Option<CrossAccountId>) -> String => |v| v.to_string(), app_promotion_api);607 pass_method!(pending_unstake_per_block(staker: CrossAccountId) -> Vec<(BlockNumber, String)> =>608 |v| v609 .into_iter()610 .map(|(b, a)| (b, a.to_string()))611 .collect::<Vec<_>>(), app_promotion_api);612}613614#[allow(deprecated)]615impl<616 C,617 Block,618 AccountId,619 CollectionInfo,620 NftInfo,621 ResourceInfo,622 PropertyInfo,623 BaseInfo,624 PartType,625 Theme,626 >627 rmrk_unique_rpc::RmrkApiServer<628 <Block as BlockT>::Hash,629 AccountId,630 CollectionInfo,631 NftInfo,632 ResourceInfo,633 PropertyInfo,634 BaseInfo,635 PartType,636 Theme,637 > for Rmrk<C, Block>638where639 C: Send + Sync + 'static + ProvideRuntimeApi<Block> + HeaderBackend<Block>,640 C::Api: RmrkRuntimeApi<641 Block,642 AccountId,643 CollectionInfo,644 NftInfo,645 ResourceInfo,646 PropertyInfo,647 BaseInfo,648 PartType,649 Theme,650 >,651 AccountId: Decode + Encode,652 CollectionInfo: Decode,653 NftInfo: Decode,654 ResourceInfo: Decode,655 PropertyInfo: Decode,656 BaseInfo: Decode,657 PartType: Decode,658 Theme: Decode,659 Block: BlockT,660{661 pass_method!(last_collection_idx() -> RmrkCollectionId, rmrk_api);662 pass_method!(collection_by_id(id: RmrkCollectionId) -> Option<CollectionInfo>, rmrk_api);663 pass_method!(nft_by_id(collection_id: RmrkCollectionId, nft_id: RmrkNftId) -> Option<NftInfo>, rmrk_api);664 pass_method!(account_tokens(account_id: AccountId, collection_id: RmrkCollectionId) -> Vec<RmrkNftId>, rmrk_api);665 pass_method!(nft_children(collection_id: RmrkCollectionId, nft_id: RmrkNftId) -> Vec<RmrkNftChild>, rmrk_api);666 pass_method!(667 collection_properties(668 collection_id: RmrkCollectionId,669670 #[map(|keys| string_keys_to_bytes_keys(keys))]671 filter_keys: Option<Vec<String>>672 ) -> Vec<PropertyInfo>,673 rmrk_api674 );675 pass_method!(676 nft_properties(677 collection_id: RmrkCollectionId,678 nft_id: RmrkNftId,679680 #[map(|keys| string_keys_to_bytes_keys(keys))]681 filter_keys: Option<Vec<String>>682 ) -> Vec<PropertyInfo>,683 rmrk_api684 );685 pass_method!(nft_resources(collection_id: RmrkCollectionId, nft_id: RmrkNftId) -> Vec<ResourceInfo>, rmrk_api);686 pass_method!(nft_resource_priority(collection_id: RmrkCollectionId, nft_id: RmrkNftId, resource_id: RmrkResourceId) -> Option<u32>, rmrk_api);687 pass_method!(base(base_id: RmrkBaseId) -> Option<BaseInfo>, rmrk_api);688 pass_method!(base_parts(base_id: RmrkBaseId) -> Vec<PartType>, rmrk_api);689 pass_method!(theme_names(base_id: RmrkBaseId) -> Vec<RmrkThemeName>, rmrk_api);690 pass_method!(691 theme(692 base_id: RmrkBaseId,693694 #[map(|n| n.into_bytes())]695 theme_name: String,696697 #[map(|keys| string_keys_to_bytes_keys(keys))]698 filter_keys: Option<Vec<String>>699 ) -> Option<Theme>, rmrk_api);700}701702fn string_keys_to_bytes_keys(keys: Option<Vec<String>>) -> Option<Vec<Vec<u8>>> {703 keys.map(|keys| keys.into_iter().map(|key| key.into_bytes()).collect())704}