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

1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617use frame_support::{18	traits::NamedReservableCurrency,19	dispatch::{GetDispatchInfo, PostDispatchInfo, DispatchInfo},20};21use sp_runtime::{22	traits::{Dispatchable, Applyable, Member},23	generic::Era,24	transaction_validity::TransactionValidityError,25	DispatchErrorWithPostInfo, DispatchError,26};27use codec::Encode;28use crate::{Runtime, RuntimeCall, RuntimeOrigin, Balances};29use up_common::types::{AccountId, Balance};30use fp_self_contained::SelfContainedCall;31use pallet_unique_scheduler::DispatchCall;32use pallet_transaction_payment::ChargeTransactionPayment;3334type SponsorshipChargeTransactionPayment =35	pallet_charge_transaction::ChargeTransactionPayment<Runtime>;3637/// The SignedExtension to the basic transaction logic.38pub type SignedExtraScheduler = (39	frame_system::CheckSpecVersion<Runtime>,40	frame_system::CheckGenesis<Runtime>,41	frame_system::CheckEra<Runtime>,42	frame_system::CheckNonce<Runtime>,43	frame_system::CheckWeight<Runtime>,44	ChargeTransactionPayment<Runtime>,45);4647fn get_signed_extras(from: <Runtime as frame_system::Config>::AccountId) -> SignedExtraScheduler {48	(49		frame_system::CheckSpecVersion::<Runtime>::new(),50		frame_system::CheckGenesis::<Runtime>::new(),51		frame_system::CheckEra::<Runtime>::from(Era::Immortal),52		frame_system::CheckNonce::<Runtime>::from(frame_system::Pallet::<Runtime>::account_nonce(53			from,54		)),55		frame_system::CheckWeight::<Runtime>::new(),56		ChargeTransactionPayment::<Runtime>::from(0),57	)58}5960pub struct SchedulerPaymentExecutor;6162impl<T: frame_system::Config + pallet_unique_scheduler::Config, SelfContainedSignedInfo>63	DispatchCall<T, SelfContainedSignedInfo> for SchedulerPaymentExecutor64where65	<T as frame_system::Config>::RuntimeCall: Member66		+ Dispatchable<RuntimeOrigin = RuntimeOrigin, Info = DispatchInfo>67		+ SelfContainedCall<SignedInfo = SelfContainedSignedInfo>68		+ GetDispatchInfo69		+ From<frame_system::Call<Runtime>>,70	SelfContainedSignedInfo: Send + Sync + 'static,71	RuntimeCall: From<<T as frame_system::Config>::RuntimeCall>72		+ From<<T as pallet_unique_scheduler::Config>::RuntimeCall>73		+ SelfContainedCall<SignedInfo = SelfContainedSignedInfo>,74	sp_runtime::AccountId32: From<<T as frame_system::Config>::AccountId>,75{76	fn dispatch_call(77		signer: Option<<T as frame_system::Config>::AccountId>,78		call: <T as pallet_unique_scheduler::Config>::RuntimeCall,79	) -> Result<80		Result<PostDispatchInfo, DispatchErrorWithPostInfo<PostDispatchInfo>>,81		TransactionValidityError,82	> {83		let dispatch_info = call.get_dispatch_info();84		let len = call.encoded_size();8586		let signed = match signer {87			Some(signer) => fp_self_contained::CheckedSignature::Signed(88				signer.clone().into(),89				get_signed_extras(signer.into()),90			),91			None => fp_self_contained::CheckedSignature::Unsigned,92		};9394		let extrinsic = fp_self_contained::CheckedExtrinsic::<95			AccountId,96			RuntimeCall,97			SignedExtraScheduler,98			SelfContainedSignedInfo,99		> {100			signed,101			function: call.into(),102		};103104		extrinsic.apply::<Runtime>(&dispatch_info, len)105	}106107	fn reserve_balance(108		id: [u8; 16],109		sponsor: <T as frame_system::Config>::AccountId,110		call: <T as pallet_unique_scheduler::Config>::RuntimeCall,111		count: u32,112	) -> Result<(), DispatchError> {113		let dispatch_info = call.get_dispatch_info();114		let weight: Balance =115			SponsorshipChargeTransactionPayment::traditional_fee(0, &dispatch_info, 0)116				.saturating_mul(count.into());117118		<Balances as NamedReservableCurrency<AccountId>>::reserve_named(119			&id,120			&(sponsor.into()),121			weight,122		)123	}124125	fn pay_for_call(126		id: [u8; 16],127		sponsor: <T as frame_system::Config>::AccountId,128		call: <T as pallet_unique_scheduler::Config>::RuntimeCall,129	) -> Result<u128, DispatchError> {130		let dispatch_info = call.get_dispatch_info();131		let weight: Balance =132			SponsorshipChargeTransactionPayment::traditional_fee(0, &dispatch_info, 0);133		Ok(134			<Balances as NamedReservableCurrency<AccountId>>::unreserve_named(135				&id,136				&(sponsor.into()),137				weight,138			),139		)140	}141142	fn cancel_reserve(143		id: [u8; 16],144		sponsor: <T as frame_system::Config>::AccountId,145	) -> Result<u128, DispatchError> {146		Ok(147			<Balances as NamedReservableCurrency<AccountId>>::unreserve_named(148				&id,149				&(sponsor.into()),150				u128::MAX,151			),152		)153	}154}