1234567891011121314151617use core::marker::PhantomData;1819use frame_support::{dispatch::DispatchResultWithPostInfo, ensure, fail, weights::Weight, BoundedVec};20use up_data_structs::{TokenId, CollectionId, CreateItemExData, budget::Budget};21use pallet_common::{CommonCollectionOperations, CommonWeightInfo, with_weight};22use sp_runtime::ArithmeticError;23use sp_std::{vec::Vec, vec};24use up_data_structs::{CustomDataLimit, Property, PropertyKeyPermission,};2526use crate::{27 Allowance, Balance, Config, Error, FungibleHandle, Pallet, SelfWeightOf, weights::WeightInfo,28};2930pub struct CommonWeights<T: Config>(PhantomData<T>);31impl<T: Config> CommonWeightInfo<T::CrossAccountId> for CommonWeights<T> {32 fn create_item() -> Weight {33 <SelfWeightOf<T>>::create_item()34 }3536 fn create_multiple_items(_amount: u32) -> Weight {37 Self::create_item()38 }3940 fn create_multiple_items_ex(data: &CreateItemExData<T::CrossAccountId>) -> Weight {41 match data {42 CreateItemExData::Fungible(f) => {43 <SelfWeightOf<T>>::create_multiple_items_ex(f.len() as u32)44 }45 _ => 0,46 }47 }4849 fn burn_item() -> Weight {50 <SelfWeightOf<T>>::burn_item()51 }5253 fn set_collection_properties(amount: u32) -> Weight {54 <SelfWeightOf<T>>::set_collection_properties(amount)55 }5657 fn set_token_properties(amount: u32) -> Weight {58 <SelfWeightOf<T>>::set_token_properties(amount)59 }6061 fn set_property_permissions(amount: u32) -> Weight {62 <SelfWeightOf<T>>::set_property_permissions(amount)63 }6465 fn transfer() -> Weight {66 <SelfWeightOf<T>>::transfer()67 }6869 fn approve() -> Weight {70 <SelfWeightOf<T>>::approve()71 }7273 fn transfer_from() -> Weight {74 <SelfWeightOf<T>>::transfer_from()75 }7677 fn burn_from() -> Weight {78 <SelfWeightOf<T>>::burn_from()79 }8081 fn set_variable_metadata(_bytes: u32) -> Weight {82 83 084 }85}8687impl<T: Config> CommonCollectionOperations<T> for FungibleHandle<T> {88 fn create_item(89 &self,90 sender: T::CrossAccountId,91 to: T::CrossAccountId,92 data: up_data_structs::CreateItemData,93 nesting_budget: &dyn Budget,94 ) -> DispatchResultWithPostInfo {95 match data {96 up_data_structs::CreateItemData::Fungible(data) => with_weight(97 <Pallet<T>>::create_item(self, &sender, (to, data.value), nesting_budget),98 <CommonWeights<T>>::create_item(),99 ),100 _ => fail!(<Error<T>>::NotFungibleDataUsedToMintFungibleCollectionToken),101 }102 }103104 fn create_multiple_items(105 &self,106 sender: T::CrossAccountId,107 to: T::CrossAccountId,108 data: Vec<up_data_structs::CreateItemData>,109 nesting_budget: &dyn Budget,110 ) -> DispatchResultWithPostInfo {111 let mut sum: u128 = 0;112 for data in data {113 match data {114 up_data_structs::CreateItemData::Fungible(data) => {115 sum = sum116 .checked_add(data.value)117 .ok_or(ArithmeticError::Overflow)?;118 }119 _ => fail!(<Error<T>>::NotFungibleDataUsedToMintFungibleCollectionToken),120 }121 }122123 with_weight(124 <Pallet<T>>::create_item(self, &sender, (to, sum), nesting_budget),125 <CommonWeights<T>>::create_item(),126 )127 }128129 fn create_multiple_items_ex(130 &self,131 sender: <T>::CrossAccountId,132 data: up_data_structs::CreateItemExData<<T>::CrossAccountId>,133 nesting_budget: &dyn Budget,134 ) -> DispatchResultWithPostInfo {135 let weight = <CommonWeights<T>>::create_multiple_items_ex(&data);136 let data = match data {137 up_data_structs::CreateItemExData::Fungible(f) => f,138 _ => fail!(<Error<T>>::NotFungibleDataUsedToMintFungibleCollectionToken),139 };140141 with_weight(142 <Pallet<T>>::create_multiple_items(self, &sender, data.into_inner(), nesting_budget),143 weight,144 )145 }146147 fn burn_item(148 &self,149 sender: T::CrossAccountId,150 token: TokenId,151 amount: u128,152 ) -> DispatchResultWithPostInfo {153 ensure!(154 token == TokenId::default(),155 <Error<T>>::FungibleItemsHaveNoId156 );157158 with_weight(159 <Pallet<T>>::burn(self, &sender, amount),160 <CommonWeights<T>>::burn_item(),161 )162 }163164 fn transfer(165 &self,166 from: T::CrossAccountId,167 to: T::CrossAccountId,168 token: TokenId,169 amount: u128,170 nesting_budget: &dyn Budget,171 ) -> DispatchResultWithPostInfo {172 ensure!(173 token == TokenId::default(),174 <Error<T>>::FungibleItemsHaveNoId175 );176177 with_weight(178 <Pallet<T>>::transfer(self, &from, &to, amount, nesting_budget),179 <CommonWeights<T>>::transfer(),180 )181 }182183 fn approve(184 &self,185 sender: T::CrossAccountId,186 spender: T::CrossAccountId,187 token: TokenId,188 amount: u128,189 ) -> DispatchResultWithPostInfo {190 ensure!(191 token == TokenId::default(),192 <Error<T>>::FungibleItemsHaveNoId193 );194195 with_weight(196 <Pallet<T>>::set_allowance(self, &sender, &spender, amount),197 <CommonWeights<T>>::approve(),198 )199 }200201 fn transfer_from(202 &self,203 sender: T::CrossAccountId,204 from: T::CrossAccountId,205 to: T::CrossAccountId,206 token: TokenId,207 amount: u128,208 nesting_budget: &dyn Budget,209 ) -> DispatchResultWithPostInfo {210 ensure!(211 token == TokenId::default(),212 <Error<T>>::FungibleItemsHaveNoId213 );214215 with_weight(216 <Pallet<T>>::transfer_from(self, &sender, &from, &to, amount, nesting_budget),217 <CommonWeights<T>>::transfer_from(),218 )219 }220221 fn burn_from(222 &self,223 sender: T::CrossAccountId,224 from: T::CrossAccountId,225 token: TokenId,226 amount: u128,227 nesting_budget: &dyn Budget,228 ) -> DispatchResultWithPostInfo {229 ensure!(230 token == TokenId::default(),231 <Error<T>>::FungibleItemsHaveNoId232 );233234 with_weight(235 <Pallet<T>>::burn_from(self, &sender, &from, amount, nesting_budget),236 <CommonWeights<T>>::burn_from(),237 )238 }239240 fn set_collection_properties(241 &self,242 _sender: T::CrossAccountId,243 _property: Vec<Property>,244 ) -> DispatchResultWithPostInfo {245 fail!(<Error<T>>::PropertiesNotAllowed)246 }247248 fn set_token_properties(249 &self,250 _sender: T::CrossAccountId,251 _token_id: TokenId,252 _property: Vec<Property>,253 ) -> DispatchResultWithPostInfo {254 fail!(<Error<T>>::PropertiesNotAllowed)255 }256257 fn set_property_permissions(258 &self,259 _sender: &T::CrossAccountId,260 _property_permissions: Vec<PropertyKeyPermission>,261 ) -> DispatchResultWithPostInfo {262 fail!(<Error<T>>::PropertiesNotAllowed)263 }264265 fn set_variable_metadata(266 &self,267 _sender: T::CrossAccountId,268 _token: TokenId,269 _data: BoundedVec<u8, CustomDataLimit>,270 ) -> DispatchResultWithPostInfo {271 fail!(<Error<T>>::FungibleItemsDontHaveData)272 }273274 fn check_nesting(275 &self,276 _sender: <T>::CrossAccountId,277 _from: (CollectionId, TokenId),278 _under: TokenId,279 _budget: &dyn Budget,280 ) -> sp_runtime::DispatchResult {281 fail!(<Error<T>>::FungibleDisallowsNesting)282 }283284 fn collection_tokens(&self) -> Vec<TokenId> {285 vec![TokenId::default()]286 }287288 fn account_tokens(&self, account: T::CrossAccountId) -> Vec<TokenId> {289 if <Balance<T>>::get((self.id, account)) != 0 {290 vec![TokenId::default()]291 } else {292 vec![]293 }294 }295296 fn token_exists(&self, token: TokenId) -> bool {297 token == TokenId::default()298 }299300 fn last_token_id(&self) -> TokenId {301 TokenId::default()302 }303304 fn token_owner(&self, _token: TokenId) -> Option<T::CrossAccountId> {305 None306 }307 fn const_metadata(&self, _token: TokenId) -> Vec<u8> {308 Vec::new()309 }310 fn variable_metadata(&self, _token: TokenId) -> Vec<u8> {311 Vec::new()312 }313314 fn total_supply(&self) -> u32 {315 1316 }317318 fn account_balance(&self, account: T::CrossAccountId) -> u32 {319 if <Balance<T>>::get((self.id, account)) != 0 {320 1321 } else {322 0323 }324 }325326 fn balance(&self, account: T::CrossAccountId, token: TokenId) -> u128 {327 if token != TokenId::default() {328 return 0;329 }330 <Balance<T>>::get((self.id, account))331 }332333 fn allowance(334 &self,335 sender: T::CrossAccountId,336 spender: T::CrossAccountId,337 token: TokenId,338 ) -> u128 {339 if token != TokenId::default() {340 return 0;341 }342 <Allowance<T>>::get((self.id, sender, spender))343 }344}