git.delta.rocks / unique-network / refs/commits / b70e369d3c1a

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}