1use crate::{2 Config, Call, CreateItemBasket, VariableMetaDataBasket, ReFungibleTransferBasket,3 FungibleTransferBasket, NftTransferBasket, CreateItemData, CollectionMode, NftApproveBasket,4 FungibleApproveBasket, RefungibleApproveBasket,5};6use core::marker::PhantomData;7use up_sponsorship::SponsorshipHandler;8use frame_support::{9 traits::{IsSubType},10 storage::{StorageMap, StorageDoubleMap, StorageNMap},11};12use up_data_structs::{13 CollectionId, FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT, MetaUpdatePermission,14 NFT_SPONSOR_TRANSFER_TIMEOUT, REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT, TokenId,15};16use pallet_common::{CollectionHandle};17use pallet_common::account::CrossAccountId;1819pub fn withdraw_transfer<T: Config>(20 collection: &CollectionHandle<T>,21 who: &T::CrossAccountId,22 item_id: &TokenId,23) -> Option<()> {24 25 match collection.mode {26 CollectionMode::NFT => {27 let owner = pallet_nonfungible::TokenData::<T>::get((collection.id, item_id))?.owner;28 if !owner.conv_eq(who) {29 return None;30 }31 }32 CollectionMode::Fungible(_) => {33 if item_id != &TokenId::default() {34 return None;35 }36 if <pallet_fungible::Balance<T>>::get((collection.id, who)) == 0 {37 return None;38 }39 }40 CollectionMode::ReFungible => {41 if !<pallet_refungible::Owned<T>>::get((collection.id, who, item_id)) {42 return None;43 }44 }45 }4647 48 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;49 let limit = collection50 .limits51 .sponsor_transfer_timeout(match collection.mode {52 CollectionMode::NFT => NFT_SPONSOR_TRANSFER_TIMEOUT,53 CollectionMode::Fungible(_) => FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,54 CollectionMode::ReFungible => REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,55 });5657 let last_tx_block = match collection.mode {58 CollectionMode::NFT => <NftTransferBasket<T>>::get(collection.id, item_id),59 CollectionMode::Fungible(_) => {60 <FungibleTransferBasket<T>>::get(collection.id, who.as_sub())61 }62 CollectionMode::ReFungible => {63 <ReFungibleTransferBasket<T>>::get((collection.id, item_id, who.as_sub()))64 }65 };6667 if let Some(last_tx_block) = last_tx_block {68 let timeout = last_tx_block + limit.into();69 if block_number < timeout {70 return None;71 }72 }7374 match collection.mode {75 CollectionMode::NFT => <NftTransferBasket<T>>::insert(collection.id, item_id, block_number),76 CollectionMode::Fungible(_) => {77 <FungibleTransferBasket<T>>::insert(collection.id, who.as_sub(), block_number)78 }79 CollectionMode::ReFungible => <ReFungibleTransferBasket<T>>::insert(80 (collection.id, item_id, who.as_sub()),81 block_number,82 ),83 };8485 Some(())86}8788pub fn withdraw_create_item<T: Config>(89 collection: &CollectionHandle<T>,90 who: &T::AccountId,91 _properties: &CreateItemData,92) -> Option<()> {93 if _properties.data_size() as u32 > collection.limits.sponsored_data_size() {94 return None;95 }9697 98 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;99 let limit = collection100 .limits101 .sponsor_transfer_timeout(match _properties {102 CreateItemData::NFT(_) => NFT_SPONSOR_TRANSFER_TIMEOUT,103 CreateItemData::Fungible(_) => FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,104 CreateItemData::ReFungible(_) => REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,105 });106107 if let Some(last_tx_block) = <CreateItemBasket<T>>::get((collection.id, &who)) {108 let timeout = last_tx_block + limit.into();109 if block_number < timeout {110 return None;111 }112 }113114 CreateItemBasket::<T>::insert((collection.id, who.clone()), block_number);115116 Some(())117}118119pub fn withdraw_set_variable_meta_data<T: Config>(120 who: &T::CrossAccountId,121 collection: &CollectionHandle<T>,122 item_id: &TokenId,123 data: &[u8],124) -> Option<()> {125 126 if collection.meta_update_permission != MetaUpdatePermission::ItemOwner {127 return None;128 }129 130 match collection.mode {131 CollectionMode::NFT => {132 let owner = pallet_nonfungible::TokenData::<T>::get((collection.id, item_id))?.owner;133 if !owner.conv_eq(who) {134 return None;135 }136 }137 CollectionMode::Fungible(_) => {138 if item_id != &TokenId::default() {139 return None;140 }141 if <pallet_fungible::Balance<T>>::get((collection.id, who)) == 0 {142 return None;143 }144 }145 CollectionMode::ReFungible => {146 if !<pallet_refungible::Owned<T>>::get((collection.id, who, item_id)) {147 return None;148 }149 }150 }151152 153 154 if matches!(collection.mode, CollectionMode::Fungible(_)) {155 return None;156 }157 if data.len() > collection.limits.sponsored_data_size() as usize {158 return None;159 }160161 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;162 let limit = collection.limits.sponsored_data_rate_limit()?;163164 if let Some(last_tx_block) = VariableMetaDataBasket::<T>::get(collection.id, item_id) {165 let timeout = last_tx_block + limit.into();166 if block_number < timeout {167 return None;168 }169 }170171 <VariableMetaDataBasket<T>>::insert(collection.id, item_id, block_number);172173 Some(())174}175176pub fn withdraw_approve<T: Config>(177 collection: &CollectionHandle<T>,178 who: &T::AccountId,179 item_id: &TokenId,180) -> Option<()> {181 182 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;183 let limit = collection.limits.sponsor_approve_timeout();184185 let last_tx_block = match collection.mode {186 CollectionMode::NFT => <NftApproveBasket<T>>::get(collection.id, item_id),187 CollectionMode::Fungible(_) => <FungibleApproveBasket<T>>::get(collection.id, who),188 CollectionMode::ReFungible => {189 <RefungibleApproveBasket<T>>::get((collection.id, item_id, who))190 }191 };192193 if let Some(last_tx_block) = last_tx_block {194 let timeout = last_tx_block + limit.into();195 if block_number < timeout {196 return None;197 }198 }199200 match collection.mode {201 CollectionMode::NFT => <NftApproveBasket<T>>::insert(collection.id, item_id, block_number),202 CollectionMode::Fungible(_) => {203 <FungibleApproveBasket<T>>::insert(collection.id, who, block_number)204 }205 CollectionMode::ReFungible => {206 <RefungibleApproveBasket<T>>::insert((collection.id, item_id, who), block_number)207 }208 };209210 Some(())211}212213fn load<T: Config>(id: CollectionId) -> Option<(T::AccountId, CollectionHandle<T>)> {214 let collection = CollectionHandle::new(id)?;215 let sponsor = collection.sponsorship.sponsor().cloned()?;216 Some((sponsor, collection))217}218219pub struct UniqueSponsorshipHandler<T>(PhantomData<T>);220impl<T, C> SponsorshipHandler<T::AccountId, C> for UniqueSponsorshipHandler<T>221where222 T: Config,223 C: IsSubType<Call<T>>,224{225 fn get_sponsor(who: &T::AccountId, call: &C) -> Option<T::AccountId> {226 match IsSubType::<Call<T>>::is_sub_type(call)? {227 Call::create_item {228 collection_id,229 data,230 ..231 } => {232 let (sponsor, collection) = load(*collection_id)?;233 withdraw_create_item::<T>(&collection, who, data).map(|()| sponsor)234 }235 Call::transfer {236 collection_id,237 item_id,238 ..239 } => {240 let (sponsor, collection) = load(*collection_id)?;241 withdraw_transfer::<T>(242 &collection,243 &T::CrossAccountId::from_sub(who.clone()),244 item_id,245 )246 .map(|()| sponsor)247 }248 Call::transfer_from {249 collection_id,250 item_id,251 from,252 ..253 } => {254 let (sponsor, collection) = load(*collection_id)?;255 withdraw_transfer::<T>(&collection, from, item_id).map(|()| sponsor)256 }257 Call::approve {258 collection_id,259 item_id,260 ..261 } => {262 let (sponsor, collection) = load(*collection_id)?;263 withdraw_approve::<T>(&collection, who, item_id).map(|()| sponsor)264 }265 Call::set_variable_meta_data {266 collection_id,267 item_id,268 data,269 } => {270 let (sponsor, collection) = load(*collection_id)?;271 withdraw_set_variable_meta_data::<T>(272 &T::CrossAccountId::from_sub(who.clone()),273 &collection,274 item_id,275 data,276 )277 .map(|()| sponsor)278 }279 _ => None,280 }281 }282}