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

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 nft_data_structs::{13	CollectionId, FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT, NFT_SPONSOR_TRANSFER_TIMEOUT,14	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::AccountId,22	item_id: &TokenId,23) -> Option<()> {2425	// preliminary sponsoring correctness check26	if !((pallet_nonfungible::TokenData::<T>::get((collection.id, item_id))?.owner).as_sub()27		== who) || (pallet_refungible::Owned::<T>::get((28		collection.id,29		T::CrossAccountId::from_sub(who.clone()),30		item_id,31	))) {32		return None;33	}34	35	// sponsor timeout36	let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;37	let limit = collection38		.limits39		.sponsor_transfer_timeout(match collection.mode {40			CollectionMode::NFT => NFT_SPONSOR_TRANSFER_TIMEOUT,41			CollectionMode::Fungible(_) => FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,42			CollectionMode::ReFungible => REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,43		});4445	let last_tx_block = match collection.mode {46		CollectionMode::NFT => <NftTransferBasket<T>>::get(collection.id, item_id),47		CollectionMode::Fungible(_) => <FungibleTransferBasket<T>>::get(collection.id, who),48		CollectionMode::ReFungible => {49			<ReFungibleTransferBasket<T>>::get((collection.id, item_id, who))50		}51	};5253	if let Some(last_tx_block) = last_tx_block {54		let timeout = last_tx_block + limit.into();55		if block_number < timeout {56			return None;57		}58	}5960	match collection.mode {61		CollectionMode::NFT => <NftTransferBasket<T>>::insert(collection.id, item_id, block_number),62		CollectionMode::Fungible(_) => {63			<FungibleTransferBasket<T>>::insert(collection.id, who, block_number)64		}65		CollectionMode::ReFungible => {66			<ReFungibleTransferBasket<T>>::insert((collection.id, item_id, who), block_number)67		}68	};6970	Some(())71}7273pub fn withdraw_create_item<T: Config>(74	collection: &CollectionHandle<T>,75	who: &T::AccountId,76	_properties: &CreateItemData,77) -> Option<()> {78	if _properties.data_size() as u32 > collection.limits.sponsored_data_size() {79		return None;80	}8182	// sponsor timeout83	let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;84	let limit = collection85		.limits86		.sponsor_transfer_timeout(match _properties {87			CreateItemData::NFT(_) => NFT_SPONSOR_TRANSFER_TIMEOUT,88			CreateItemData::Fungible(_) => FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,89			CreateItemData::ReFungible(_) => REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,90		});9192	if let Some(last_tx_block) = <CreateItemBasket<T>>::get((collection.id, &who)) {93		let timeout = last_tx_block + limit.into();94		if block_number < timeout {95			return None;96		}97	}9899	CreateItemBasket::<T>::insert((collection.id, who.clone()), block_number);100101	Some(())102}103104pub fn withdraw_set_variable_meta_data<T: Config>(105	who: &T::AccountId,106	collection: &CollectionHandle<T>,107	item_id: &TokenId,108	data: &[u8],109) -> Option<()> {110111	// preliminary sponsoring correctness check112	if !((pallet_nonfungible::TokenData::<T>::get((collection.id, item_id))?.owner).as_sub()113		== who) || (pallet_refungible::Owned::<T>::get((114		collection.id,115		T::CrossAccountId::from_sub(who.clone()),116		item_id,117	))) {118		return None;119	}120121	// Can't sponsor fungible collection, this tx will be rejected122	// as invalid123	if matches!(collection.mode, CollectionMode::Fungible(_)) {124		return None;125	}126	if data.len() > collection.limits.sponsored_data_size() as usize {127		return None;128	}129130	let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;131	let limit = collection.limits.sponsored_data_rate_limit()?;132133	if let Some(last_tx_block) = VariableMetaDataBasket::<T>::get(collection.id, item_id) {134		let timeout = last_tx_block + limit.into();135		if block_number < timeout {136			return None;137		}138	}139140	<VariableMetaDataBasket<T>>::insert(collection.id, item_id, block_number);141142	Some(())143}144145pub fn withdraw_approve<T: Config>(146	collection: &CollectionHandle<T>,147	who: &T::AccountId,148	item_id: &TokenId,149) -> Option<()> {150	// sponsor timeout151	let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;152	let limit = collection.limits.sponsor_approve_timeout();153154	let last_tx_block = match collection.mode {155		CollectionMode::NFT => <NftApproveBasket<T>>::get(collection.id, item_id),156		CollectionMode::Fungible(_) => <FungibleApproveBasket<T>>::get(collection.id, who),157		CollectionMode::ReFungible => {158			<RefungibleApproveBasket<T>>::get((collection.id, item_id, who))159		}160	};161162	if let Some(last_tx_block) = last_tx_block {163		let timeout = last_tx_block + limit.into();164		if block_number < timeout {165			return None;166		}167	}168169	match collection.mode {170		CollectionMode::NFT => <NftApproveBasket<T>>::insert(collection.id, item_id, block_number),171		CollectionMode::Fungible(_) => {172			<FungibleApproveBasket<T>>::insert(collection.id, who, block_number)173		}174		CollectionMode::ReFungible => {175			<RefungibleApproveBasket<T>>::insert((collection.id, item_id, who), block_number)176		}177	};178179	Some(())180}181182fn load<T: Config>(id: CollectionId) -> Option<(T::AccountId, CollectionHandle<T>)> {183	let collection = CollectionHandle::new(id)?;184	let sponsor = collection.sponsorship.sponsor().cloned()?;185	Some((sponsor, collection))186}187188pub struct NftSponsorshipHandler<T>(PhantomData<T>);189impl<T, C> SponsorshipHandler<T::AccountId, C> for NftSponsorshipHandler<T>190where191	T: Config,192	C: IsSubType<Call<T>>,193{194	fn get_sponsor(who: &T::AccountId, call: &C) -> Option<T::AccountId> {195		match IsSubType::<Call<T>>::is_sub_type(call)? {196			Call::create_item {197				collection_id,198				data,199				..200			} => {201				let (sponsor, collection) = load(*collection_id)?;202				withdraw_create_item::<T>(&collection, who, data).map(|()| sponsor)203			}204			Call::transfer {205				collection_id,206				item_id,207				..208			}209			| Call::transfer_from {210				collection_id,211				item_id,212				..213			} => {214				let (sponsor, collection) = load(*collection_id)?;215				withdraw_transfer::<T>(&collection, who, item_id).map(|()| sponsor)216			}217			Call::approve {218				collection_id,219				item_id,220				..221			} => {222				let (sponsor, collection) = load(*collection_id)?;223				withdraw_approve::<T>(&collection, who, item_id).map(|()| sponsor)224			}225			Call::set_variable_meta_data {226				collection_id,227				item_id,228				data,229			} => {230				let (sponsor, collection) = load(*collection_id)?;231				withdraw_set_variable_meta_data::<T>(&who, &collection, item_id, data).map(|()| sponsor)232			}233			_ => None,234		}235	}236}