1use std::sync::Arc;23use jsonrpc_core::{Error as RpcError, ErrorCode, Result};4use jsonrpc_derive::rpc;5use nft_data_structs::{CollectionId, TokenId};6use sp_api::{BlockId, BlockT, ProvideRuntimeApi};7use sp_blockchain::HeaderBackend;8use up_rpc::NftApi as NftRuntimeApi;910#[rpc]11pub trait NftApi<BlockHash, CrossAccountId, AccountId> {12 #[rpc(name = "nft_accountTokens")]13 fn account_tokens(14 &self,15 collection: CollectionId,16 account: CrossAccountId,17 at: Option<BlockHash>,18 ) -> Result<Vec<TokenId>>;19 #[rpc(name = "nft_tokenExists")]20 fn token_exists(21 &self,22 collection: CollectionId,23 token: TokenId,24 at: Option<BlockHash>,25 ) -> Result<bool>;2627 #[rpc(name = "nft_tokenOwner")]28 fn token_owner(29 &self,30 collection: CollectionId,31 token: TokenId,32 at: Option<BlockHash>,33 ) -> Result<CrossAccountId>;34 #[rpc(name = "nft_constMetadata")]35 fn const_metadata(36 &self,37 collection: CollectionId,38 token: TokenId,39 at: Option<BlockHash>,40 ) -> Result<Vec<u8>>;41 #[rpc(name = "nft_variableMetadata")]42 fn variable_metadata(43 &self,44 collection: CollectionId,45 token: TokenId,46 at: Option<BlockHash>,47 ) -> Result<Vec<u8>>;4849 #[rpc(name = "nft_collectionTokens")]50 fn collection_tokens(&self, collection: CollectionId, at: Option<BlockHash>) -> Result<u32>;51 #[rpc(name = "nft_accountBalance")]52 fn account_balance(53 &self,54 collection: CollectionId,55 account: CrossAccountId,56 at: Option<BlockHash>,57 ) -> Result<u32>;58 #[rpc(name = "nft_balance")]59 fn balance(60 &self,61 collection: CollectionId,62 account: CrossAccountId,63 token: TokenId,64 at: Option<BlockHash>,65 ) -> Result<u128>;66 #[rpc(name = "nft_allowance")]67 fn allowance(68 &self,69 collection: CollectionId,70 sender: CrossAccountId,71 spender: CrossAccountId,72 token: TokenId,73 at: Option<BlockHash>,74 ) -> Result<u128>;75}7677pub struct Nft<C, P> {78 client: Arc<C>,79 _marker: std::marker::PhantomData<P>,80}8182impl<C, P> Nft<C, P> {83 pub fn new(client: Arc<C>) -> Self {84 Self {85 client,86 _marker: Default::default(),87 }88 }89}9091pub enum Error {92 RuntimeError,93}9495impl From<Error> for i64 {96 fn from(e: Error) -> i64 {97 match e {98 Error::RuntimeError => 1,99 }100 }101}102103macro_rules! pass_method {104 ($method_name:ident($($name:ident: $ty:ty),* $(,)?) -> $result:ty) => {105 fn $method_name(106 &self,107 $(108 $name: $ty,109 )*110 at: Option<<Block as BlockT>::Hash>,111 ) -> Result<$result> {112 let api = self.client.runtime_api();113 let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));114115 api.$method_name(&at, $($name),*).map_err(|e| RpcError {116 code: ErrorCode::ServerError(Error::RuntimeError.into()),117 message: "Unable to query".into(),118 data: Some(format!("{:?}", e).into()),119 })120 }121 };122}123124impl<C, Block, CrossAccountId, AccountId> NftApi<<Block as BlockT>::Hash, CrossAccountId, AccountId>125 for Nft<C, Block>126where127 Block: BlockT,128 C: 'static + ProvideRuntimeApi<Block> + HeaderBackend<Block>,129 C::Api: NftRuntimeApi<Block, CrossAccountId, AccountId>,130 CrossAccountId: pallet_common::account::CrossAccountId<AccountId>,131{132 pass_method!(account_tokens(collection: CollectionId, account: CrossAccountId) -> Vec<TokenId>);133 pass_method!(token_exists(collection: CollectionId, token: TokenId) -> bool);134 pass_method!(token_owner(collection: CollectionId, token: TokenId) -> CrossAccountId);135 pass_method!(const_metadata(collection: CollectionId, token: TokenId) -> Vec<u8>);136 pass_method!(variable_metadata(collection: CollectionId, token: TokenId) -> Vec<u8>);137 pass_method!(collection_tokens(collection: CollectionId) -> u32);138 pass_method!(account_balance(collection: CollectionId, account: CrossAccountId) -> u32);139 pass_method!(balance(collection: CollectionId, account: CrossAccountId, token: TokenId) -> u128);140 pass_method!(allowance(collection: CollectionId, sender: CrossAccountId, spender: CrossAccountId, token: TokenId) -> u128);141}