git.delta.rocks / unique-network / refs/commits / 1cfabfc42ed0

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<()> {24	// preliminary sponsoring correctness check25	if !((pallet_nonfungible::TokenData::<T>::get((collection.id, item_id))?.owner).as_sub() == who)26		|| (pallet_refungible::Owned::<T>::get((27			collection.id,28			T::CrossAccountId::from_sub(who.clone()),29			item_id,30		))) {31		return None;32	}3334	// sponsor timeout35	let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;36	let limit = collection37		.limits38		.sponsor_transfer_timeout(match collection.mode {39			CollectionMode::NFT => NFT_SPONSOR_TRANSFER_TIMEOUT,40			CollectionMode::Fungible(_) => FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,41			CollectionMode::ReFungible => REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,42		});4344	let last_tx_block = match collection.mode {45		CollectionMode::NFT => <NftTransferBasket<T>>::get(collection.id, item_id),46		CollectionMode::Fungible(_) => <FungibleTransferBasket<T>>::get(collection.id, who),47		CollectionMode::ReFungible => {48			<ReFungibleTransferBasket<T>>::get((collection.id, item_id, who))49		}50	};5152	if let Some(last_tx_block) = last_tx_block {53		let timeout = last_tx_block + limit.into();54		if block_number < timeout {55			return None;56		}57	}5859	match collection.mode {60		CollectionMode::NFT => <NftTransferBasket<T>>::insert(collection.id, item_id, block_number),61		CollectionMode::Fungible(_) => {62			<FungibleTransferBasket<T>>::insert(collection.id, who, block_number)63		}64		CollectionMode::ReFungible => {65			<ReFungibleTransferBasket<T>>::insert((collection.id, item_id, who), block_number)66		}67	};6869	Some(())70}7172pub fn withdraw_create_item<T: Config>(73	collection: &CollectionHandle<T>,74	who: &T::AccountId,75	_properties: &CreateItemData,76) -> Option<()> {77	if _properties.data_size() as u32 > collection.limits.sponsored_data_size() {78		return None;79	}8081	// sponsor timeout82	let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;83	let limit = collection84		.limits85		.sponsor_transfer_timeout(match _properties {86			CreateItemData::NFT(_) => NFT_SPONSOR_TRANSFER_TIMEOUT,87			CreateItemData::Fungible(_) => FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,88			CreateItemData::ReFungible(_) => REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT,89		});9091	if let Some(last_tx_block) = <CreateItemBasket<T>>::get((collection.id, &who)) {92		let timeout = last_tx_block + limit.into();93		if block_number < timeout {94			return None;95		}96	}9798	CreateItemBasket::<T>::insert((collection.id, who.clone()), block_number);99100	Some(())101}102103pub fn withdraw_set_variable_meta_data<T: Config>(104	who: &T::AccountId,105	collection: &CollectionHandle<T>,106	item_id: &TokenId,107	data: &[u8],108) -> Option<()> {109	// preliminary sponsoring correctness check110	if !((pallet_nonfungible::TokenData::<T>::get((collection.id, item_id))?.owner).as_sub() == who)111		|| (pallet_refungible::Owned::<T>::get((112			collection.id,113			T::CrossAccountId::from_sub(who.clone()),114			item_id,115		))) {116		return None;117	}118119	// Can't sponsor fungible collection, this tx will be rejected120	// as invalid121	if matches!(collection.mode, CollectionMode::Fungible(_)) {122		return None;123	}124	if data.len() > collection.limits.sponsored_data_size() as usize {125		return None;126	}127128	let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;129	let limit = collection.limits.sponsored_data_rate_limit()?;130131	if let Some(last_tx_block) = VariableMetaDataBasket::<T>::get(collection.id, item_id) {132		let timeout = last_tx_block + limit.into();133		if block_number < timeout {134			return None;135		}136	}137138	<VariableMetaDataBasket<T>>::insert(collection.id, item_id, block_number);139140	Some(())141}142143pub fn withdraw_approve<T: Config>(144	collection: &CollectionHandle<T>,145	who: &T::AccountId,146	item_id: &TokenId,147) -> Option<()> {148	// sponsor timeout149	let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;150	let limit = collection.limits.sponsor_approve_timeout();151152	let last_tx_block = match collection.mode {153		CollectionMode::NFT => <NftApproveBasket<T>>::get(collection.id, item_id),154		CollectionMode::Fungible(_) => <FungibleApproveBasket<T>>::get(collection.id, who),155		CollectionMode::ReFungible => {156			<RefungibleApproveBasket<T>>::get((collection.id, item_id, who))157		}158	};159160	if let Some(last_tx_block) = last_tx_block {161		let timeout = last_tx_block + limit.into();162		if block_number < timeout {163			return None;164		}165	}166167	match collection.mode {168		CollectionMode::NFT => <NftApproveBasket<T>>::insert(collection.id, item_id, block_number),169		CollectionMode::Fungible(_) => {170			<FungibleApproveBasket<T>>::insert(collection.id, who, block_number)171		}172		CollectionMode::ReFungible => {173			<RefungibleApproveBasket<T>>::insert((collection.id, item_id, who), block_number)174		}175	};176177	Some(())178}179180fn load<T: Config>(id: CollectionId) -> Option<(T::AccountId, CollectionHandle<T>)> {181	let collection = CollectionHandle::new(id)?;182	let sponsor = collection.sponsorship.sponsor().cloned()?;183	Some((sponsor, collection))184}185186pub struct NftSponsorshipHandler<T>(PhantomData<T>);187impl<T, C> SponsorshipHandler<T::AccountId, C> for NftSponsorshipHandler<T>188where189	T: Config,190	C: IsSubType<Call<T>>,191{192	fn get_sponsor(who: &T::AccountId, call: &C) -> Option<T::AccountId> {193		match IsSubType::<Call<T>>::is_sub_type(call)? {194			Call::create_item {195				collection_id,196				data,197				..198			} => {199				let (sponsor, collection) = load(*collection_id)?;200				withdraw_create_item::<T>(&collection, who, data).map(|()| sponsor)201			}202			Call::transfer {203				collection_id,204				item_id,205				..206			}207			| Call::transfer_from {208				collection_id,209				item_id,210				..211			} => {212				let (sponsor, collection) = load(*collection_id)?;213				withdraw_transfer::<T>(&collection, who, item_id).map(|()| sponsor)214			}215			Call::approve {216				collection_id,217				item_id,218				..219			} => {220				let (sponsor, collection) = load(*collection_id)?;221				withdraw_approve::<T>(&collection, who, item_id).map(|()| sponsor)222			}223			Call::set_variable_meta_data {224				collection_id,225				item_id,226				data,227			} => {228				let (sponsor, collection) = load(*collection_id)?;229				withdraw_set_variable_meta_data::<T>(&who, &collection, item_id, data)230					.map(|()| sponsor)231			}232			_ => None,233		}234	}235}