1use core::marker::PhantomData;23use frame_support::{dispatch::DispatchResultWithPostInfo, ensure, fail, weights::Weight};4use nft_data_structs::TokenId;5use pallet_common::{CommonCollectionOperations, CommonWeightInfo, with_weight};6use sp_runtime::DispatchError;7use sp_std::vec::Vec;89use crate::{10 AccountBalance, Allowance, Config, CreateItemData, Error, NonfungibleHandle, Owned, Pallet,11 SelfWeightOf, TokenData, weights::WeightInfo, TokensMinted,12};1314pub struct CommonWeights<T: Config>(PhantomData<T>);15impl<T: Config> CommonWeightInfo for CommonWeights<T> {16 fn create_item() -> Weight {17 <SelfWeightOf<T>>::create_item()18 }1920 fn create_multiple_items(amount: u32) -> Weight {21 <SelfWeightOf<T>>::create_multiple_items(amount)22 }2324 fn burn_item() -> Weight {25 <SelfWeightOf<T>>::burn_item()26 }2728 fn transfer() -> Weight {29 <SelfWeightOf<T>>::transfer()30 }3132 fn approve() -> Weight {33 <SelfWeightOf<T>>::approve()34 }3536 fn transfer_from() -> Weight {37 <SelfWeightOf<T>>::transfer_from()38 }3940 fn burn_from() -> Weight {41 042 }4344 fn set_variable_metadata(bytes: u32) -> Weight {45 <SelfWeightOf<T>>::set_variable_metadata(bytes)46 }47}4849fn map_create_data<T: Config>(50 data: nft_data_structs::CreateItemData,51 to: &T::CrossAccountId,52) -> Result<CreateItemData<T>, DispatchError> {53 match data {54 nft_data_structs::CreateItemData::NFT(data) => Ok(CreateItemData {55 const_data: data.const_data,56 variable_data: data.variable_data,57 owner: to.clone(),58 }),59 _ => fail!(<Error<T>>::NotNonfungibleDataUsedToMintFungibleCollectionToken),60 }61}6263impl<T: Config> CommonCollectionOperations<T> for NonfungibleHandle<T> {64 fn create_item(65 &self,66 sender: T::CrossAccountId,67 to: T::CrossAccountId,68 data: nft_data_structs::CreateItemData,69 ) -> DispatchResultWithPostInfo {70 with_weight(71 <Pallet<T>>::create_item(self, &sender, map_create_data(data, &to)?),72 <CommonWeights<T>>::create_item(),73 )74 }7576 fn create_multiple_items(77 &self,78 sender: T::CrossAccountId,79 to: T::CrossAccountId,80 data: Vec<nft_data_structs::CreateItemData>,81 ) -> DispatchResultWithPostInfo {82 let data = data83 .into_iter()84 .map(|d| map_create_data::<T>(d, &to))85 .collect::<Result<Vec<_>, DispatchError>>()?;8687 let amount = data.len();88 with_weight(89 <Pallet<T>>::create_multiple_items(self, &sender, data),90 <CommonWeights<T>>::create_multiple_items(amount as u32),91 )92 }9394 fn burn_item(95 &self,96 sender: T::CrossAccountId,97 token: TokenId,98 amount: u128,99 ) -> DispatchResultWithPostInfo {100 ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);101 if amount == 1 {102 with_weight(103 <Pallet<T>>::burn(&self, &sender, token),104 <CommonWeights<T>>::burn_item(),105 )106 } else {107 Ok(().into())108 }109 }110111 fn transfer(112 &self,113 from: T::CrossAccountId,114 to: T::CrossAccountId,115 token: TokenId,116 amount: u128,117 ) -> DispatchResultWithPostInfo {118 ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);119 if amount == 1 {120 with_weight(121 <Pallet<T>>::transfer(&self, &from, &to, token),122 <CommonWeights<T>>::transfer(),123 )124 } else {125 Ok(().into())126 }127 }128129 fn approve(130 &self,131 sender: T::CrossAccountId,132 spender: T::CrossAccountId,133 token: TokenId,134 amount: u128,135 ) -> DispatchResultWithPostInfo {136 ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);137138 with_weight(139 if amount == 1 {140 <Pallet<T>>::set_allowance(&self, &sender, token, Some(&spender))141 } else {142 <Pallet<T>>::set_allowance(&self, &sender, token, None)143 },144 <CommonWeights<T>>::approve(),145 )146 }147148 fn transfer_from(149 &self,150 sender: T::CrossAccountId,151 from: T::CrossAccountId,152 to: T::CrossAccountId,153 token: TokenId,154 amount: u128,155 ) -> DispatchResultWithPostInfo {156 ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);157158 if amount == 1 {159 with_weight(160 <Pallet<T>>::transfer_from(&self, &sender, &from, &to, token),161 <CommonWeights<T>>::transfer_from(),162 )163 } else {164 Ok(().into())165 }166 }167168 fn burn_from(169 &self,170 sender: T::CrossAccountId,171 from: T::CrossAccountId,172 token: TokenId,173 amount: u128,174 ) -> DispatchResultWithPostInfo {175 ensure!(amount <= 1, <Error<T>>::NonfungibleItemsHaveNoAmount);176177 if amount == 1 {178 with_weight(179 <Pallet<T>>::burn_from(&self, &sender, &from, token),180 <CommonWeights<T>>::burn_from(),181 )182 } else {183 Ok(().into())184 }185 }186187 fn set_variable_metadata(188 &self,189 sender: T::CrossAccountId,190 token: TokenId,191 data: Vec<u8>,192 ) -> DispatchResultWithPostInfo {193 let len = data.len();194 with_weight(195 <Pallet<T>>::set_variable_metadata(&self, &sender, token, data),196 <CommonWeights<T>>::set_variable_metadata(len as u32),197 )198 }199200 fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId> {201 <Owned<T>>::iter_prefix((self.id, account))202 .map(|(id, _)| id)203 .collect()204 }205206 fn token_exists(&self, token: TokenId) -> bool {207 <Pallet<T>>::token_exists(self, token)208 }209210 fn last_token_id(&self) -> TokenId {211 TokenId(<TokensMinted<T>>::get(self.id))212 }213214 fn token_owner(&self, token: TokenId) -> T::CrossAccountId {215 <TokenData<T>>::get((self.id, token))216 .map(|t| t.owner)217 .unwrap_or_default()218 }219 fn const_metadata(&self, token: TokenId) -> Vec<u8> {220 <TokenData<T>>::get((self.id, token))221 .map(|t| t.const_data.clone())222 .unwrap_or_default()223 }224 fn variable_metadata(&self, token: TokenId) -> Vec<u8> {225 <TokenData<T>>::get((self.id, token))226 .map(|t| t.variable_data.clone())227 .unwrap_or_default()228 }229230 fn collection_tokens(&self) -> u32 {231 <Pallet<T>>::total_supply(self)232 }233234 fn account_balance(&self, account: T::CrossAccountId) -> u32 {235 <AccountBalance<T>>::get((self.id, account))236 }237238 fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128 {239 if <TokenData<T>>::get((self.id, token))240 .map(|a| a.owner == account)241 .unwrap_or(false)242 {243 1244 } else {245 0246 }247 }248249 fn allowance(250 &self,251 sender: T::CrossAccountId,252 spender: T::CrossAccountId,253 token: TokenId,254 ) -> u128 {255 if <TokenData<T>>::get((self.id, token))256 .map(|a| a.owner != sender)257 .unwrap_or(true)258 {259 0260 } else if <Allowance<T>>::get((self.id, token)) == Some(spender) {261 1262 } else {263 0264 }265 }266}