1234567891011121314151617use std::sync::Arc;1819use codec::Decode;20use jsonrpc_core::{Error as RpcError, ErrorCode, Result};21use jsonrpc_derive::rpc;22use up_data_structs::{RpcCollection, CollectionId, CollectionStats, CollectionLimits, TokenId};23use sp_api::{BlockId, BlockT, ProvideRuntimeApi, ApiExt};24use sp_blockchain::HeaderBackend;25use up_rpc::UniqueApi as UniqueRuntimeApi;2627#[rpc]28pub trait UniqueApi<BlockHash, CrossAccountId, AccountId> {29 #[rpc(name = "unique_accountTokens")]30 fn account_tokens(31 &self,32 collection: CollectionId,33 account: CrossAccountId,34 at: Option<BlockHash>,35 ) -> Result<Vec<TokenId>>;36 #[rpc(name = "unique_collectionTokens")]37 fn collection_tokens(38 &self,39 collection: CollectionId,40 at: Option<BlockHash>,41 ) -> Result<Vec<TokenId>>;42 #[rpc(name = "unique_tokenExists")]43 fn token_exists(44 &self,45 collection: CollectionId,46 token: TokenId,47 at: Option<BlockHash>,48 ) -> Result<bool>;4950 #[rpc(name = "unique_tokenOwner")]51 fn token_owner(52 &self,53 collection: CollectionId,54 token: TokenId,55 at: Option<BlockHash>,56 ) -> Result<Option<CrossAccountId>>;57 #[rpc(name = "unique_topmostTokenOwner")]58 fn topmost_token_owner(59 &self,60 collection: CollectionId,61 token: TokenId,62 at: Option<BlockHash>,63 ) -> Result<Option<CrossAccountId>>;64 #[rpc(name = "unique_constMetadata")]65 fn const_metadata(66 &self,67 collection: CollectionId,68 token: TokenId,69 at: Option<BlockHash>,70 ) -> Result<Vec<u8>>;71 #[rpc(name = "unique_variableMetadata")]72 fn variable_metadata(73 &self,74 collection: CollectionId,75 token: TokenId,76 at: Option<BlockHash>,77 ) -> Result<Vec<u8>>;7879 #[rpc(name = "unique_totalSupply")]80 fn total_supply(81 &self,82 collection: CollectionId,83 at: Option<BlockHash>,84 ) -> Result<u32>;85 #[rpc(name = "unique_accountBalance")]86 fn account_balance(87 &self,88 collection: CollectionId,89 account: CrossAccountId,90 at: Option<BlockHash>,91 ) -> Result<u32>;92 #[rpc(name = "unique_balance")]93 fn balance(94 &self,95 collection: CollectionId,96 account: CrossAccountId,97 token: TokenId,98 at: Option<BlockHash>,99 ) -> Result<String>;100 #[rpc(name = "unique_allowance")]101 fn allowance(102 &self,103 collection: CollectionId,104 sender: CrossAccountId,105 spender: CrossAccountId,106 token: TokenId,107 at: Option<BlockHash>,108 ) -> Result<String>;109110 #[rpc(name = "unique_adminlist")]111 fn adminlist(112 &self,113 collection: CollectionId,114 at: Option<BlockHash>,115 ) -> Result<Vec<CrossAccountId>>;116 #[rpc(name = "unique_allowlist")]117 fn allowlist(118 &self,119 collection: CollectionId,120 at: Option<BlockHash>,121 ) -> Result<Vec<CrossAccountId>>;122 #[rpc(name = "unique_allowed")]123 fn allowed(124 &self,125 collection: CollectionId,126 user: CrossAccountId,127 at: Option<BlockHash>,128 ) -> Result<bool>;129 #[rpc(name = "unique_lastTokenId")]130 fn last_token_id(&self, collection: CollectionId, at: Option<BlockHash>) -> Result<TokenId>;131 #[rpc(name = "unique_collectionById")]132 fn collection_by_id(133 &self,134 collection: CollectionId,135 at: Option<BlockHash>,136 ) -> Result<Option<RpcCollection<AccountId>>>;137 #[rpc(name = "unique_collectionStats")]138 fn collection_stats(&self, at: Option<BlockHash>) -> Result<CollectionStats>;139140 #[rpc(name = "unique_nextSponsored")]141 fn next_sponsored(142 &self,143 collection: CollectionId,144 account: CrossAccountId,145 token: TokenId,146 at: Option<BlockHash>,147 ) -> Result<Option<u64>>;148 #[rpc(name = "unique_effectiveCollectionLimits")]149 fn effective_collection_limits(150 &self,151 collection_id: CollectionId,152 at: Option<BlockHash>,153 ) -> Result<Option<CollectionLimits>>;154}155156pub struct Unique<C, P> {157 client: Arc<C>,158 _marker: std::marker::PhantomData<P>,159}160161impl<C, P> Unique<C, P> {162 pub fn new(client: Arc<C>) -> Self {163 Self {164 client,165 _marker: Default::default(),166 }167 }168}169170pub enum Error {171 RuntimeError,172}173174impl From<Error> for i64 {175 fn from(e: Error) -> i64 {176 match e {177 Error::RuntimeError => 1,178 }179 }180}181182macro_rules! pass_method {183 (184 $method_name:ident($($name:ident: $ty:ty),* $(,)?) -> $result:ty $(=> $mapper:expr)?185 $(; changed_in $ver:expr, $changed_method_name:ident ($($changed_name:expr), * $(,)?) => $fixer:expr)*186 ) => {187 fn $method_name(188 &self,189 $(190 $name: $ty,191 )*192 at: Option<<Block as BlockT>::Hash>,193 ) -> Result<$result> {194 let api = self.client.runtime_api();195 let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));196 let _api_version = if let Ok(Some(api_version)) =197 api.api_version::<dyn UniqueRuntimeApi<Block, CrossAccountId, AccountId>>(&at)198 {199 api_version200 } else {201 202 return Err(RpcError {203 code: ErrorCode::InvalidParams,204 message: "Api is not available".into(),205 data: None,206 })207 };208209 let result = $(if _api_version < $ver {210 api.$changed_method_name(&at, $($changed_name),*).map(|r| r.map($fixer))211 } else)*212 { api.$method_name(&at, $($name),*) };213214 let result = result.map_err(|e| RpcError {215 code: ErrorCode::ServerError(Error::RuntimeError.into()),216 message: "Unable to query".into(),217 data: Some(format!("{:?}", e).into()),218 })?;219 result.map_err(|e| RpcError {220 code: ErrorCode::InvalidParams,221 message: "Runtime returned error".into(),222 data: Some(format!("{:?}", e).into()),223 })$(.map($mapper))?224 }225 };226}227228#[allow(deprecated)]229impl<C, Block, CrossAccountId, AccountId>230 UniqueApi<<Block as BlockT>::Hash, CrossAccountId, AccountId> for Unique<C, Block>231where232 Block: BlockT,233 AccountId: Decode,234 C: 'static + ProvideRuntimeApi<Block> + HeaderBackend<Block>,235 C::Api: UniqueRuntimeApi<Block, CrossAccountId, AccountId>,236 CrossAccountId: pallet_evm::account::CrossAccountId<AccountId>,237{238 pass_method!(account_tokens(collection: CollectionId, account: CrossAccountId) -> Vec<TokenId>);239 pass_method!(collection_tokens(collection: CollectionId) -> Vec<TokenId>);240 pass_method!(token_exists(collection: CollectionId, token: TokenId) -> bool);241 pass_method!(242 token_owner(collection: CollectionId, token: TokenId) -> Option<CrossAccountId>;243 changed_in 2, token_owner_before_version_2(collection, token) => |u| Some(u)244 );245 pass_method!(topmost_token_owner(collection: CollectionId, token: TokenId) -> Option<CrossAccountId>);246 pass_method!(const_metadata(collection: CollectionId, token: TokenId) -> Vec<u8>);247 pass_method!(variable_metadata(collection: CollectionId, token: TokenId) -> Vec<u8>);248249 pass_method!(total_supply(collection: CollectionId) -> u32);250 pass_method!(account_balance(collection: CollectionId, account: CrossAccountId) -> u32);251 pass_method!(balance(collection: CollectionId, account: CrossAccountId, token: TokenId) -> String => |v| v.to_string());252 pass_method!(allowance(collection: CollectionId, sender: CrossAccountId, spender: CrossAccountId, token: TokenId) -> String => |v| v.to_string());253254 pass_method!(adminlist(collection: CollectionId) -> Vec<CrossAccountId>);255 pass_method!(allowlist(collection: CollectionId) -> Vec<CrossAccountId>);256 pass_method!(allowed(collection: CollectionId, user: CrossAccountId) -> bool);257 pass_method!(last_token_id(collection: CollectionId) -> TokenId);258 pass_method!(collection_by_id(collection: CollectionId) -> Option<RpcCollection<AccountId>>);259 pass_method!(collection_stats() -> CollectionStats);260 pass_method!(next_sponsored(collection: CollectionId, account: CrossAccountId, token: TokenId) -> Option<u64>);261 pass_method!(effective_collection_limits(collection_id: CollectionId) -> Option<CollectionLimits>);262}