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_tokenExists")]37 fn token_exists(38 &self,39 collection: CollectionId,40 token: TokenId,41 at: Option<BlockHash>,42 ) -> Result<bool>;4344 #[rpc(name = "unique_tokenOwner")]45 fn token_owner(46 &self,47 collection: CollectionId,48 token: TokenId,49 at: Option<BlockHash>,50 ) -> Result<Option<CrossAccountId>>;51 #[rpc(name = "unique_topmostTokenOwner")]52 fn topmost_token_owner(53 &self,54 collection: CollectionId,55 token: TokenId,56 at: Option<BlockHash>,57 ) -> Result<Option<CrossAccountId>>;58 #[rpc(name = "unique_constMetadata")]59 fn const_metadata(60 &self,61 collection: CollectionId,62 token: TokenId,63 at: Option<BlockHash>,64 ) -> Result<Vec<u8>>;65 #[rpc(name = "unique_variableMetadata")]66 fn variable_metadata(67 &self,68 collection: CollectionId,69 token: TokenId,70 at: Option<BlockHash>,71 ) -> Result<Vec<u8>>;7273 #[rpc(name = "unique_collectionTokens")]74 fn collection_tokens(&self, collection: CollectionId, at: Option<BlockHash>) -> Result<u32>;75 #[rpc(name = "unique_accountBalance")]76 fn account_balance(77 &self,78 collection: CollectionId,79 account: CrossAccountId,80 at: Option<BlockHash>,81 ) -> Result<u32>;82 #[rpc(name = "unique_balance")]83 fn balance(84 &self,85 collection: CollectionId,86 account: CrossAccountId,87 token: TokenId,88 at: Option<BlockHash>,89 ) -> Result<String>;90 #[rpc(name = "unique_allowance")]91 fn allowance(92 &self,93 collection: CollectionId,94 sender: CrossAccountId,95 spender: CrossAccountId,96 token: TokenId,97 at: Option<BlockHash>,98 ) -> Result<String>;99100 #[rpc(name = "unique_adminlist")]101 fn adminlist(102 &self,103 collection: CollectionId,104 at: Option<BlockHash>,105 ) -> Result<Vec<CrossAccountId>>;106 #[rpc(name = "unique_allowlist")]107 fn allowlist(108 &self,109 collection: CollectionId,110 at: Option<BlockHash>,111 ) -> Result<Vec<CrossAccountId>>;112 #[rpc(name = "unique_allowed")]113 fn allowed(114 &self,115 collection: CollectionId,116 user: CrossAccountId,117 at: Option<BlockHash>,118 ) -> Result<bool>;119 #[rpc(name = "unique_lastTokenId")]120 fn last_token_id(&self, collection: CollectionId, at: Option<BlockHash>) -> Result<TokenId>;121 #[rpc(name = "unique_collectionById")]122 fn collection_by_id(123 &self,124 collection: CollectionId,125 at: Option<BlockHash>,126 ) -> Result<Option<RpcCollection<AccountId>>>;127 #[rpc(name = "unique_collectionStats")]128 fn collection_stats(&self, at: Option<BlockHash>) -> Result<CollectionStats>;129130 #[rpc(name = "unique_nextSponsored")]131 fn next_sponsored(132 &self,133 collection: CollectionId,134 account: CrossAccountId,135 token: TokenId,136 at: Option<BlockHash>,137 ) -> Result<Option<u64>>;138 #[rpc(name = "unique_effectiveCollectionLimits")]139 fn effective_collection_limits(140 &self,141 collection_id: CollectionId,142 at: Option<BlockHash>,143 ) -> Result<Option<CollectionLimits>>;144}145146pub struct Unique<C, P> {147 client: Arc<C>,148 _marker: std::marker::PhantomData<P>,149}150151impl<C, P> Unique<C, P> {152 pub fn new(client: Arc<C>) -> Self {153 Self {154 client,155 _marker: Default::default(),156 }157 }158}159160pub enum Error {161 RuntimeError,162}163164impl From<Error> for i64 {165 fn from(e: Error) -> i64 {166 match e {167 Error::RuntimeError => 1,168 }169 }170}171172macro_rules! pass_method {173 (174 $method_name:ident($($name:ident: $ty:ty),* $(,)?) -> $result:ty $(=> $mapper:expr)?175 $(; changed_in $ver:expr, $changed_method_name:ident ($($changed_name:expr), * $(,)?) => $fixer:expr)*176 ) => {177 fn $method_name(178 &self,179 $(180 $name: $ty,181 )*182 at: Option<<Block as BlockT>::Hash>,183 ) -> Result<$result> {184 let api = self.client.runtime_api();185 let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));186 let _api_version = if let Ok(Some(api_version)) =187 api.api_version::<dyn UniqueRuntimeApi<Block, CrossAccountId, AccountId>>(&at)188 {189 api_version190 } else {191 192 return Err(RpcError {193 code: ErrorCode::InvalidParams,194 message: "Api is not available".into(),195 data: None,196 })197 };198199 let result = $(if _api_version < $ver {200 api.$changed_method_name(&at, $($changed_name),*).map(|r| r.map($fixer))201 } else)*202 { api.$method_name(&at, $($name),*) };203204 let result = result.map_err(|e| RpcError {205 code: ErrorCode::ServerError(Error::RuntimeError.into()),206 message: "Unable to query".into(),207 data: Some(format!("{:?}", e).into()),208 })?;209 result.map_err(|e| RpcError {210 code: ErrorCode::InvalidParams,211 message: "Runtime returned error".into(),212 data: Some(format!("{:?}", e).into()),213 })$(.map($mapper))?214 }215 };216}217218#[allow(deprecated)]219impl<C, Block, CrossAccountId, AccountId>220 UniqueApi<<Block as BlockT>::Hash, CrossAccountId, AccountId> for Unique<C, Block>221where222 Block: BlockT,223 AccountId: Decode,224 C: 'static + ProvideRuntimeApi<Block> + HeaderBackend<Block>,225 C::Api: UniqueRuntimeApi<Block, CrossAccountId, AccountId>,226 CrossAccountId: pallet_evm::account::CrossAccountId<AccountId>,227{228 pass_method!(account_tokens(collection: CollectionId, account: CrossAccountId) -> Vec<TokenId>);229 pass_method!(token_exists(collection: CollectionId, token: TokenId) -> bool);230 pass_method!(231 token_owner(collection: CollectionId, token: TokenId) -> Option<CrossAccountId>;232 changed_in 2, token_owner_before_version_2(collection, token) => |u| Some(u)233 );234 pass_method!(topmost_token_owner(collection: CollectionId, token: TokenId) -> Option<CrossAccountId>);235 pass_method!(const_metadata(collection: CollectionId, token: TokenId) -> Vec<u8>);236 pass_method!(variable_metadata(collection: CollectionId, token: TokenId) -> Vec<u8>);237 pass_method!(collection_tokens(collection: CollectionId) -> u32);238 pass_method!(account_balance(collection: CollectionId, account: CrossAccountId) -> u32);239 pass_method!(balance(collection: CollectionId, account: CrossAccountId, token: TokenId) -> String => |v| v.to_string());240 pass_method!(allowance(collection: CollectionId, sender: CrossAccountId, spender: CrossAccountId, token: TokenId) -> String => |v| v.to_string());241242 pass_method!(adminlist(collection: CollectionId) -> Vec<CrossAccountId>);243 pass_method!(allowlist(collection: CollectionId) -> Vec<CrossAccountId>);244 pass_method!(allowed(collection: CollectionId, user: CrossAccountId) -> bool);245 pass_method!(last_token_id(collection: CollectionId) -> TokenId);246 pass_method!(collection_by_id(collection: CollectionId) -> Option<RpcCollection<AccountId>>);247 pass_method!(collection_stats() -> CollectionStats);248 pass_method!(next_sponsored(collection: CollectionId, account: CrossAccountId, token: TokenId) -> Option<u64>);249 pass_method!(effective_collection_limits(collection_id: CollectionId) -> Option<CollectionLimits>);250}