1234567891011121314151617use std::sync::Arc;1819use codec::Decode;20use jsonrpc_core::{Error as RpcError, ErrorCode, Result};21use jsonrpc_derive::rpc;22use up_data_structs::{23 RpcCollection, CollectionId, CollectionStats, CollectionLimits, TokenId, Property,24 PropertyKeyPermission, TokenData,25};26use sp_api::{BlockId, BlockT, ProvideRuntimeApi, ApiExt};27use sp_blockchain::HeaderBackend;28use up_rpc::UniqueApi as UniqueRuntimeApi;2930#[rpc]31pub trait UniqueApi<BlockHash, CrossAccountId, AccountId> {32 #[rpc(name = "unique_accountTokens")]33 fn account_tokens(34 &self,35 collection: CollectionId,36 account: CrossAccountId,37 at: Option<BlockHash>,38 ) -> Result<Vec<TokenId>>;39 #[rpc(name = "unique_collectionTokens")]40 fn collection_tokens(41 &self,42 collection: CollectionId,43 at: Option<BlockHash>,44 ) -> Result<Vec<TokenId>>;45 #[rpc(name = "unique_tokenExists")]46 fn token_exists(47 &self,48 collection: CollectionId,49 token: TokenId,50 at: Option<BlockHash>,51 ) -> Result<bool>;5253 #[rpc(name = "unique_tokenOwner")]54 fn token_owner(55 &self,56 collection: CollectionId,57 token: TokenId,58 at: Option<BlockHash>,59 ) -> Result<Option<CrossAccountId>>;60 #[rpc(name = "unique_topmostTokenOwner")]61 fn topmost_token_owner(62 &self,63 collection: CollectionId,64 token: TokenId,65 at: Option<BlockHash>,66 ) -> Result<Option<CrossAccountId>>;67 #[rpc(name = "unique_constMetadata")]68 fn const_metadata(69 &self,70 collection: CollectionId,71 token: TokenId,72 at: Option<BlockHash>,73 ) -> Result<Vec<u8>>;74 #[rpc(name = "unique_variableMetadata")]75 fn variable_metadata(76 &self,77 collection: CollectionId,78 token: TokenId,79 at: Option<BlockHash>,80 ) -> Result<Vec<u8>>;8182 #[rpc(name = "unique_collectionProperties")]83 fn collection_properties(84 &self,85 collection: CollectionId,86 keys: Vec<String>,87 at: Option<BlockHash>,88 ) -> Result<Vec<Property>>;8990 #[rpc(name = "unique_tokenProperties")]91 fn token_properties(92 &self,93 collection: CollectionId,94 token_id: TokenId,95 properties: Vec<String>,96 at: Option<BlockHash>,97 ) -> Result<Vec<Property>>;9899 #[rpc(name = "unique_propertyPermissions")]100 fn property_permissions(101 &self,102 collection: CollectionId,103 keys: Vec<String>,104 at: Option<BlockHash>,105 ) -> Result<Vec<PropertyKeyPermission>>;106107 #[rpc(name = "unique_tokenData")]108 fn token_data(109 &self,110 collection: CollectionId,111 token_id: TokenId,112 keys: Vec<String>,113 at: Option<BlockHash>,114 ) -> Result<TokenData<CrossAccountId>>;115116 #[rpc(name = "unique_totalSupply")]117 fn total_supply(&self, collection: CollectionId, at: Option<BlockHash>) -> Result<u32>;118 #[rpc(name = "unique_accountBalance")]119 fn account_balance(120 &self,121 collection: CollectionId,122 account: CrossAccountId,123 at: Option<BlockHash>,124 ) -> Result<u32>;125 #[rpc(name = "unique_balance")]126 fn balance(127 &self,128 collection: CollectionId,129 account: CrossAccountId,130 token: TokenId,131 at: Option<BlockHash>,132 ) -> Result<String>;133 #[rpc(name = "unique_allowance")]134 fn allowance(135 &self,136 collection: CollectionId,137 sender: CrossAccountId,138 spender: CrossAccountId,139 token: TokenId,140 at: Option<BlockHash>,141 ) -> Result<String>;142143 #[rpc(name = "unique_adminlist")]144 fn adminlist(145 &self,146 collection: CollectionId,147 at: Option<BlockHash>,148 ) -> Result<Vec<CrossAccountId>>;149 #[rpc(name = "unique_allowlist")]150 fn allowlist(151 &self,152 collection: CollectionId,153 at: Option<BlockHash>,154 ) -> Result<Vec<CrossAccountId>>;155 #[rpc(name = "unique_allowed")]156 fn allowed(157 &self,158 collection: CollectionId,159 user: CrossAccountId,160 at: Option<BlockHash>,161 ) -> Result<bool>;162 #[rpc(name = "unique_lastTokenId")]163 fn last_token_id(&self, collection: CollectionId, at: Option<BlockHash>) -> Result<TokenId>;164 #[rpc(name = "unique_collectionById")]165 fn collection_by_id(166 &self,167 collection: CollectionId,168 at: Option<BlockHash>,169 ) -> Result<Option<RpcCollection<AccountId>>>;170 #[rpc(name = "unique_collectionStats")]171 fn collection_stats(&self, at: Option<BlockHash>) -> Result<CollectionStats>;172173 #[rpc(name = "unique_nextSponsored")]174 fn next_sponsored(175 &self,176 collection: CollectionId,177 account: CrossAccountId,178 token: TokenId,179 at: Option<BlockHash>,180 ) -> Result<Option<u64>>;181 #[rpc(name = "unique_effectiveCollectionLimits")]182 fn effective_collection_limits(183 &self,184 collection_id: CollectionId,185 at: Option<BlockHash>,186 ) -> Result<Option<CollectionLimits>>;187}188189pub struct Unique<C, P> {190 client: Arc<C>,191 _marker: std::marker::PhantomData<P>,192}193194impl<C, P> Unique<C, P> {195 pub fn new(client: Arc<C>) -> Self {196 Self {197 client,198 _marker: Default::default(),199 }200 }201}202203pub enum Error {204 RuntimeError,205}206207impl From<Error> for i64 {208 fn from(e: Error) -> i64 {209 match e {210 Error::RuntimeError => 1,211 }212 }213}214215macro_rules! pass_method {216 (217 $method_name:ident(218 $($(#[map(|$map_arg:ident| $map:expr)])? $name:ident: $ty:ty),* $(,)?219 ) -> $result:ty $(=> $mapper:expr)?220 $(; changed_in $ver:expr, $changed_method_name:ident ($($changed_name:expr), * $(,)?) => $fixer:expr)*221 ) => {222 fn $method_name(223 &self,224 $(225 $name: $ty,226 )*227 at: Option<<Block as BlockT>::Hash>,228 ) -> Result<$result> {229 let api = self.client.runtime_api();230 let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));231 let _api_version = if let Ok(Some(api_version)) =232 api.api_version::<dyn UniqueRuntimeApi<Block, CrossAccountId, AccountId>>(&at)233 {234 api_version235 } else {236 237 return Err(RpcError {238 code: ErrorCode::InvalidParams,239 message: "Api is not available".into(),240 data: None,241 })242 };243244 let result = $(if _api_version < $ver {245 api.$changed_method_name(&at, $($changed_name),*).map(|r| r.map($fixer))246 } else)*247 { api.$method_name(&at, $($((|$map_arg: $ty| $map))? ($name)),*) };248249 let result = result.map_err(|e| RpcError {250 code: ErrorCode::ServerError(Error::RuntimeError.into()),251 message: "Unable to query".into(),252 data: Some(format!("{:?}", e).into()),253 })?;254 result.map_err(|e| RpcError {255 code: ErrorCode::InvalidParams,256 message: "Runtime returned error".into(),257 data: Some(format!("{:?}", e).into()),258 })$(.map($mapper))?259 }260 };261}262263#[allow(deprecated)]264impl<C, Block, CrossAccountId, AccountId>265 UniqueApi<<Block as BlockT>::Hash, CrossAccountId, AccountId> for Unique<C, Block>266where267 Block: BlockT,268 AccountId: Decode,269 C: 'static + ProvideRuntimeApi<Block> + HeaderBackend<Block>,270 C::Api: UniqueRuntimeApi<Block, CrossAccountId, AccountId>,271 CrossAccountId: pallet_evm::account::CrossAccountId<AccountId>,272{273 pass_method!(account_tokens(collection: CollectionId, account: CrossAccountId) -> Vec<TokenId>);274 pass_method!(collection_tokens(collection: CollectionId) -> Vec<TokenId>);275 pass_method!(token_exists(collection: CollectionId, token: TokenId) -> bool);276 pass_method!(277 token_owner(collection: CollectionId, token: TokenId) -> Option<CrossAccountId>;278 changed_in 2, token_owner_before_version_2(collection, token) => |u| Some(u)279 );280 pass_method!(topmost_token_owner(collection: CollectionId, token: TokenId) -> Option<CrossAccountId>);281 pass_method!(const_metadata(collection: CollectionId, token: TokenId) -> Vec<u8>);282 pass_method!(variable_metadata(collection: CollectionId, token: TokenId) -> Vec<u8>);283284 pass_method!(collection_properties(285 collection: CollectionId,286287 #[map(|keys| string_keys_to_bytes_keys(keys))]288 keys: Vec<String>289 ) -> Vec<Property>);290291 pass_method!(token_properties(292 collection: CollectionId,293 token_id: TokenId,294295 #[map(|keys| string_keys_to_bytes_keys(keys))]296 properties: Vec<String>297 ) -> Vec<Property>);298299 pass_method!(property_permissions(300 collection: CollectionId,301302 #[map(|keys| string_keys_to_bytes_keys(keys))]303 keys: Vec<String>304 ) -> Vec<PropertyKeyPermission>);305306 pass_method!(token_data(307 collection: CollectionId,308 token_id: TokenId,309310 #[map(|keys| string_keys_to_bytes_keys(keys))]311 keys: Vec<String>,312 ) -> TokenData<CrossAccountId>);313314 pass_method!(total_supply(collection: CollectionId) -> u32);315 pass_method!(account_balance(collection: CollectionId, account: CrossAccountId) -> u32);316 pass_method!(balance(collection: CollectionId, account: CrossAccountId, token: TokenId) -> String => |v| v.to_string());317 pass_method!(allowance(collection: CollectionId, sender: CrossAccountId, spender: CrossAccountId, token: TokenId) -> String => |v| v.to_string());318319 pass_method!(adminlist(collection: CollectionId) -> Vec<CrossAccountId>);320 pass_method!(allowlist(collection: CollectionId) -> Vec<CrossAccountId>);321 pass_method!(allowed(collection: CollectionId, user: CrossAccountId) -> bool);322 pass_method!(last_token_id(collection: CollectionId) -> TokenId);323 pass_method!(collection_by_id(collection: CollectionId) -> Option<RpcCollection<AccountId>>);324 pass_method!(collection_stats() -> CollectionStats);325 pass_method!(next_sponsored(collection: CollectionId, account: CrossAccountId, token: TokenId) -> Option<u64>);326 pass_method!(effective_collection_limits(collection_id: CollectionId) -> Option<CollectionLimits>);327}328329fn string_keys_to_bytes_keys(keys: Vec<String>) -> Vec<Vec<u8>> {330 keys.into_iter().map(|key| key.into_bytes()).collect()331}