123456789101112131415161718use std::sync::Arc;1920use codec::{Decode, Encode};21use jsonrpsee::{22 core::{RpcResult as Result},23 proc_macros::rpc,24};25use anyhow::anyhow;26use up_data_structs::{27 RpcCollection, CollectionId, CollectionStats, CollectionLimits, TokenId, Property,28 PropertyKeyPermission, TokenData, TokenChild,29};30use sp_api::{BlockId, BlockT, ProvideRuntimeApi, ApiExt};31use sp_blockchain::HeaderBackend;32use up_rpc::UniqueApi as UniqueRuntimeApi;333435use rmrk_rpc::RmrkApi as RmrkRuntimeApi;36use up_data_structs::{37 RmrkCollectionId, RmrkNftId, RmrkBaseId, RmrkNftChild, RmrkThemeName, RmrkResourceId,38};3940pub use rmrk_unique_rpc::RmrkApiServer;4142#[rpc(server)]43#[async_trait]44pub trait UniqueApi<BlockHash, CrossAccountId, AccountId> {45 46 #[method(name = "unique_accountTokens")]47 fn account_tokens(48 &self,49 collection: CollectionId,50 account: CrossAccountId,51 at: Option<BlockHash>,52 ) -> Result<Vec<TokenId>>;5354 55 #[method(name = "unique_collectionTokens")]56 fn collection_tokens(57 &self,58 collection: CollectionId,59 at: Option<BlockHash>,60 ) -> Result<Vec<TokenId>>;6162 63 #[method(name = "unique_tokenExists")]64 fn token_exists(65 &self,66 collection: CollectionId,67 token: TokenId,68 at: Option<BlockHash>,69 ) -> Result<bool>;7071 72 #[method(name = "unique_tokenOwner")]73 fn token_owner(74 &self,75 collection: CollectionId,76 token: TokenId,77 at: Option<BlockHash>,78 ) -> Result<Option<CrossAccountId>>;7980 81 #[method(name = "unique_topmostTokenOwner")]82 fn topmost_token_owner(83 &self,84 collection: CollectionId,85 token: TokenId,86 at: Option<BlockHash>,87 ) -> Result<Option<CrossAccountId>>;8889 90 #[method(name = "unique_tokenChildren")]91 fn token_children(92 &self,93 collection: CollectionId,94 token: TokenId,95 at: Option<BlockHash>,96 ) -> Result<Vec<TokenChild>>;9798 99 #[method(name = "unique_collectionProperties")]100 fn collection_properties(101 &self,102 collection: CollectionId,103 keys: Option<Vec<String>>,104 at: Option<BlockHash>,105 ) -> Result<Vec<Property>>;106107 108 #[method(name = "unique_tokenProperties")]109 fn token_properties(110 &self,111 collection: CollectionId,112 token_id: TokenId,113 keys: Option<Vec<String>>,114 at: Option<BlockHash>,115 ) -> Result<Vec<Property>>;116117 118 #[method(name = "unique_propertyPermissions")]119 fn property_permissions(120 &self,121 collection: CollectionId,122 keys: Option<Vec<String>>,123 at: Option<BlockHash>,124 ) -> Result<Vec<PropertyKeyPermission>>;125126 127 #[method(name = "unique_tokenData")]128 fn token_data(129 &self,130 collection: CollectionId,131 token_id: TokenId,132 keys: Option<Vec<String>>,133 at: Option<BlockHash>,134 ) -> Result<TokenData<CrossAccountId>>;135136 137 #[method(name = "unique_totalSupply")]138 fn total_supply(&self, collection: CollectionId, at: Option<BlockHash>) -> Result<u32>;139140 141 #[method(name = "unique_accountBalance")]142 fn account_balance(143 &self,144 collection: CollectionId,145 account: CrossAccountId,146 at: Option<BlockHash>,147 ) -> Result<u32>;148149 150 #[method(name = "unique_balance")]151 fn balance(152 &self,153 collection: CollectionId,154 account: CrossAccountId,155 token: TokenId,156 at: Option<BlockHash>,157 ) -> Result<String>;158159 160 #[method(name = "unique_allowance")]161 fn allowance(162 &self,163 collection: CollectionId,164 sender: CrossAccountId,165 spender: CrossAccountId,166 token: TokenId,167 at: Option<BlockHash>,168 ) -> Result<String>;169170 171 #[method(name = "unique_adminlist")]172 fn adminlist(173 &self,174 collection: CollectionId,175 at: Option<BlockHash>,176 ) -> Result<Vec<CrossAccountId>>;177178 179 #[method(name = "unique_allowlist")]180 fn allowlist(181 &self,182 collection: CollectionId,183 at: Option<BlockHash>,184 ) -> Result<Vec<CrossAccountId>>;185186 187 #[method(name = "unique_allowed")]188 fn allowed(189 &self,190 collection: CollectionId,191 user: CrossAccountId,192 at: Option<BlockHash>,193 ) -> Result<bool>;194195 196 #[method(name = "unique_lastTokenId")]197 fn last_token_id(&self, collection: CollectionId, at: Option<BlockHash>) -> Result<TokenId>;198199 200 #[method(name = "unique_collectionById")]201 fn collection_by_id(202 &self,203 collection: CollectionId,204 at: Option<BlockHash>,205 ) -> Result<Option<RpcCollection<AccountId>>>;206207 208 #[method(name = "unique_collectionStats")]209 fn collection_stats(&self, at: Option<BlockHash>) -> Result<CollectionStats>;210211 212 #[method(name = "unique_nextSponsored")]213 fn next_sponsored(214 &self,215 collection: CollectionId,216 account: CrossAccountId,217 token: TokenId,218 at: Option<BlockHash>,219 ) -> Result<Option<u64>>;220221 222 #[method(name = "unique_effectiveCollectionLimits")]223 fn effective_collection_limits(224 &self,225 collection_id: CollectionId,226 at: Option<BlockHash>,227 ) -> Result<Option<CollectionLimits>>;228229 230 #[method(name = "unique_totalPieces")]231 fn total_pieces(232 &self,233 collection_id: CollectionId,234 token_id: TokenId,235 at: Option<BlockHash>,236 ) -> Result<Option<u128>>;237}238239mod rmrk_unique_rpc {240 use super::*;241242 #[rpc(server)]243 #[async_trait]244 pub trait RmrkApi<245 BlockHash,246 AccountId,247 CollectionInfo,248 NftInfo,249 ResourceInfo,250 PropertyInfo,251 BaseInfo,252 PartType,253 Theme,254 >255 {256 257 #[method(name = "rmrk_lastCollectionIdx")]258 fn last_collection_idx(&self, at: Option<BlockHash>) -> Result<RmrkCollectionId>;259260 261 #[method(name = "rmrk_collectionById")]262 fn collection_by_id(263 &self,264 id: RmrkCollectionId,265 at: Option<BlockHash>,266 ) -> Result<Option<CollectionInfo>>;267268 269 #[method(name = "rmrk_nftById")]270 fn nft_by_id(271 &self,272 collection_id: RmrkCollectionId,273 nft_id: RmrkNftId,274 at: Option<BlockHash>,275 ) -> Result<Option<NftInfo>>;276277 278 #[method(name = "rmrk_accountTokens")]279 fn account_tokens(280 &self,281 account_id: AccountId,282 collection_id: RmrkCollectionId,283 at: Option<BlockHash>,284 ) -> Result<Vec<RmrkNftId>>;285286 287 #[method(name = "rmrk_nftChildren")]288 fn nft_children(289 &self,290 collection_id: RmrkCollectionId,291 nft_id: RmrkNftId,292 at: Option<BlockHash>,293 ) -> Result<Vec<RmrkNftChild>>;294295 296 #[method(name = "rmrk_collectionProperties")]297 fn collection_properties(298 &self,299 collection_id: RmrkCollectionId,300 filter_keys: Option<Vec<String>>,301 at: Option<BlockHash>,302 ) -> Result<Vec<PropertyInfo>>;303304 305 #[method(name = "rmrk_nftProperties")]306 fn nft_properties(307 &self,308 collection_id: RmrkCollectionId,309 nft_id: RmrkNftId,310 filter_keys: Option<Vec<String>>,311 at: Option<BlockHash>,312 ) -> Result<Vec<PropertyInfo>>;313314 315 #[method(name = "rmrk_nftResources")]316 fn nft_resources(317 &self,318 collection_id: RmrkCollectionId,319 nft_id: RmrkNftId,320 at: Option<BlockHash>,321 ) -> Result<Vec<ResourceInfo>>;322323 324 #[method(name = "rmrk_nftResourcePriority")]325 fn nft_resource_priority(326 &self,327 collection_id: RmrkCollectionId,328 nft_id: RmrkNftId,329 resource_id: RmrkResourceId,330 at: Option<BlockHash>,331 ) -> Result<Option<u32>>;332333 334 #[method(name = "rmrk_base")]335 fn base(&self, base_id: RmrkBaseId, at: Option<BlockHash>) -> Result<Option<BaseInfo>>;336337 338 #[method(name = "rmrk_baseParts")]339 fn base_parts(&self, base_id: RmrkBaseId, at: Option<BlockHash>) -> Result<Vec<PartType>>;340341 342 #[method(name = "rmrk_themeNames")]343 fn theme_names(344 &self,345 base_id: RmrkBaseId,346 at: Option<BlockHash>,347 ) -> Result<Vec<RmrkThemeName>>;348349 350 #[method(name = "rmrk_themes")]351 fn theme(352 &self,353 base_id: RmrkBaseId,354 theme_name: String,355 filter_keys: Option<Vec<String>>,356 at: Option<BlockHash>,357 ) -> Result<Option<Theme>>;358 }359}360361pub struct Unique<C, P> {362 client: Arc<C>,363 _marker: std::marker::PhantomData<P>,364}365366impl<C, P> Unique<C, P> {367 pub fn new(client: Arc<C>) -> Self {368 Self {369 client,370 _marker: Default::default(),371 }372 }373}374375pub struct Rmrk<C, P> {376 client: Arc<C>,377 _marker: std::marker::PhantomData<P>,378}379380impl<C, P> Rmrk<C, P> {381 pub fn new(client: Arc<C>) -> Self {382 Self {383 client,384 _marker: Default::default(),385 }386 }387}388389macro_rules! pass_method {390 (391 $method_name:ident(392 $($(#[map(|$map_arg:ident| $map:expr)])? $name:ident: $ty:ty),* $(,)?393 ) -> $result:ty $(=> $mapper:expr)?,394 395 $runtime_api_macro:ident396 $(; changed_in $ver:expr, $changed_method_name:ident ($($changed_name:expr), * $(,)?) => $fixer:expr)*397 ) => {398 fn $method_name(399 &self,400 $(401 $name: $ty,402 )*403 at: Option<<Block as BlockT>::Hash>,404 ) -> Result<$result> {405 let api = self.client.runtime_api();406 let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));407 let _api_version = if let Ok(Some(api_version)) =408 api.api_version::<$runtime_api_macro!()>(&at)409 {410 api_version411 } else {412 413 return Err(anyhow!("api is not available").into())414 };415416 let result = $(if _api_version < $ver {417 api.$changed_method_name(&at, $($changed_name),*).map(|r| r.map($fixer))418 } else)*419 { api.$method_name(&at, $($((|$map_arg: $ty| $map))? ($name)),*) };420421 Ok(result422 .map_err(|e| anyhow!("unable to query: {e}"))?423 .map_err(|e| anyhow!("runtime error: {e:?}"))$(.map($mapper))??)424 }425 };426}427428macro_rules! unique_api {429 () => {430 dyn UniqueRuntimeApi<Block, CrossAccountId, AccountId>431 };432}433434macro_rules! rmrk_api {435 () => {436 dyn RmrkRuntimeApi<Block, AccountId, CollectionInfo, NftInfo, ResourceInfo, PropertyInfo, BaseInfo, PartType, Theme>437 };438}439440#[allow(deprecated)]441impl<C, Block, CrossAccountId, AccountId>442 UniqueApiServer<<Block as BlockT>::Hash, CrossAccountId, AccountId> for Unique<C, Block>443where444 Block: BlockT,445 AccountId: Decode,446 C: 'static + ProvideRuntimeApi<Block> + HeaderBackend<Block>,447 C::Api: UniqueRuntimeApi<Block, CrossAccountId, AccountId>,448 CrossAccountId: pallet_evm::account::CrossAccountId<AccountId>,449{450 pass_method!(451 account_tokens(collection: CollectionId, account: CrossAccountId) -> Vec<TokenId>, unique_api452 );453 pass_method!(454 collection_tokens(collection: CollectionId) -> Vec<TokenId>, unique_api455 );456 pass_method!(457 token_exists(collection: CollectionId, token: TokenId) -> bool, unique_api458 );459 pass_method!(460 token_owner(collection: CollectionId, token: TokenId) -> Option<CrossAccountId>, unique_api461 );462 pass_method!(463 topmost_token_owner(collection: CollectionId, token: TokenId) -> Option<CrossAccountId>, unique_api464 );465 pass_method!(token_children(collection: CollectionId, token: TokenId) -> Vec<TokenChild>, unique_api);466 pass_method!(total_supply(collection: CollectionId) -> u32, unique_api);467 pass_method!(account_balance(collection: CollectionId, account: CrossAccountId) -> u32, unique_api);468 pass_method!(balance(collection: CollectionId, account: CrossAccountId, token: TokenId) -> String => |v| v.to_string(), unique_api);469 pass_method!(470 allowance(collection: CollectionId, sender: CrossAccountId, spender: CrossAccountId, token: TokenId) -> String => |v| v.to_string(),471 unique_api472 );473474 pass_method!(collection_properties(475 collection: CollectionId,476477 #[map(|keys| string_keys_to_bytes_keys(keys))]478 keys: Option<Vec<String>>479 ) -> Vec<Property>, unique_api);480481 pass_method!(token_properties(482 collection: CollectionId,483 token_id: TokenId,484485 #[map(|keys| string_keys_to_bytes_keys(keys))]486 keys: Option<Vec<String>>487 ) -> Vec<Property>, unique_api);488489 pass_method!(property_permissions(490 collection: CollectionId,491492 #[map(|keys| string_keys_to_bytes_keys(keys))]493 keys: Option<Vec<String>>494 ) -> Vec<PropertyKeyPermission>, unique_api);495496 pass_method!(token_data(497 collection: CollectionId,498 token_id: TokenId,499500 #[map(|keys| string_keys_to_bytes_keys(keys))]501 keys: Option<Vec<String>>,502 ) -> TokenData<CrossAccountId>, unique_api);503504 pass_method!(adminlist(collection: CollectionId) -> Vec<CrossAccountId>, unique_api);505 pass_method!(allowlist(collection: CollectionId) -> Vec<CrossAccountId>, unique_api);506 pass_method!(allowed(collection: CollectionId, user: CrossAccountId) -> bool, unique_api);507 pass_method!(last_token_id(collection: CollectionId) -> TokenId, unique_api);508 pass_method!(collection_by_id(collection: CollectionId) -> Option<RpcCollection<AccountId>>, unique_api);509 pass_method!(collection_stats() -> CollectionStats, unique_api);510 pass_method!(next_sponsored(collection: CollectionId, account: CrossAccountId, token: TokenId) -> Option<u64>, unique_api);511 pass_method!(effective_collection_limits(collection_id: CollectionId) -> Option<CollectionLimits>, unique_api);512 pass_method!(total_pieces(collection_id: CollectionId, token_id: TokenId) -> Option<u128>, unique_api);513}514515#[allow(deprecated)]516impl<517 C,518 Block,519 AccountId,520 CollectionInfo,521 NftInfo,522 ResourceInfo,523 PropertyInfo,524 BaseInfo,525 PartType,526 Theme,527 >528 rmrk_unique_rpc::RmrkApiServer<529 <Block as BlockT>::Hash,530 AccountId,531 CollectionInfo,532 NftInfo,533 ResourceInfo,534 PropertyInfo,535 BaseInfo,536 PartType,537 Theme,538 > for Rmrk<C, Block>539where540 C: Send + Sync + 'static + ProvideRuntimeApi<Block> + HeaderBackend<Block>,541 C::Api: RmrkRuntimeApi<542 Block,543 AccountId,544 CollectionInfo,545 NftInfo,546 ResourceInfo,547 PropertyInfo,548 BaseInfo,549 PartType,550 Theme,551 >,552 AccountId: Decode + Encode,553 CollectionInfo: Decode,554 NftInfo: Decode,555 ResourceInfo: Decode,556 PropertyInfo: Decode,557 BaseInfo: Decode,558 PartType: Decode,559 Theme: Decode,560 Block: BlockT,561{562 pass_method!(last_collection_idx() -> RmrkCollectionId, rmrk_api);563 pass_method!(collection_by_id(id: RmrkCollectionId) -> Option<CollectionInfo>, rmrk_api);564 pass_method!(nft_by_id(collection_id: RmrkCollectionId, nft_id: RmrkNftId) -> Option<NftInfo>, rmrk_api);565 pass_method!(account_tokens(account_id: AccountId, collection_id: RmrkCollectionId) -> Vec<RmrkNftId>, rmrk_api);566 pass_method!(nft_children(collection_id: RmrkCollectionId, nft_id: RmrkNftId) -> Vec<RmrkNftChild>, rmrk_api);567 pass_method!(568 collection_properties(569 collection_id: RmrkCollectionId,570571 #[map(|keys| string_keys_to_bytes_keys(keys))]572 filter_keys: Option<Vec<String>>573 ) -> Vec<PropertyInfo>,574 rmrk_api575 );576 pass_method!(577 nft_properties(578 collection_id: RmrkCollectionId,579 nft_id: RmrkNftId,580581 #[map(|keys| string_keys_to_bytes_keys(keys))]582 filter_keys: Option<Vec<String>>583 ) -> Vec<PropertyInfo>,584 rmrk_api585 );586 pass_method!(nft_resources(collection_id: RmrkCollectionId, nft_id: RmrkNftId) -> Vec<ResourceInfo>, rmrk_api);587 pass_method!(nft_resource_priority(collection_id: RmrkCollectionId, nft_id: RmrkNftId, resource_id: RmrkResourceId) -> Option<u32>, rmrk_api);588 pass_method!(base(base_id: RmrkBaseId) -> Option<BaseInfo>, rmrk_api);589 pass_method!(base_parts(base_id: RmrkBaseId) -> Vec<PartType>, rmrk_api);590 pass_method!(theme_names(base_id: RmrkBaseId) -> Vec<RmrkThemeName>, rmrk_api);591 pass_method!(592 theme(593 base_id: RmrkBaseId,594595 #[map(|n| n.into_bytes())]596 theme_name: String,597598 #[map(|keys| string_keys_to_bytes_keys(keys))]599 filter_keys: Option<Vec<String>>600 ) -> Option<Theme>, rmrk_api);601}602603fn string_keys_to_bytes_keys(keys: Option<Vec<String>>) -> Option<Vec<Vec<u8>>> {604 keys.map(|keys| keys.into_iter().map(|key| key.into_bytes()).collect())605}