1234567891011121314151617use frame_support::dispatch::{GetDispatchInfo, PostDispatchInfo, DispatchInfo};18use sp_runtime::{19 traits::{Dispatchable, Applyable, Member},20 generic::Era,21 transaction_validity::TransactionValidityError,22 DispatchErrorWithPostInfo,23};24use codec::Encode;25use crate::{Runtime, RuntimeCall, RuntimeOrigin};26use up_common::types::AccountId;27use fp_self_contained::SelfContainedCall;28use pallet_unique_scheduler::DispatchCall;29use pallet_transaction_payment::ChargeTransactionPayment;3031type SponsorshipChargeTransactionPayment =32 pallet_charge_transaction::ChargeTransactionPayment<Runtime>;333435pub type SignedExtraScheduler = (36 frame_system::CheckSpecVersion<Runtime>,37 frame_system::CheckGenesis<Runtime>,38 frame_system::CheckEra<Runtime>,39 frame_system::CheckNonce<Runtime>,40 frame_system::CheckWeight<Runtime>,41 ChargeTransactionPayment<Runtime>,42);4344fn get_signed_extras(from: <Runtime as frame_system::Config>::AccountId) -> SignedExtraScheduler {45 (46 frame_system::CheckSpecVersion::<Runtime>::new(),47 frame_system::CheckGenesis::<Runtime>::new(),48 frame_system::CheckEra::<Runtime>::from(Era::Immortal),49 frame_system::CheckNonce::<Runtime>::from(frame_system::Pallet::<Runtime>::account_nonce(50 from,51 )),52 frame_system::CheckWeight::<Runtime>::new(),53 ChargeTransactionPayment::<Runtime>::from(0),54 )55}5657pub struct SchedulerPaymentExecutor;5859impl<T: frame_system::Config + pallet_unique_scheduler::Config, SelfContainedSignedInfo>60 DispatchCall<T, SelfContainedSignedInfo> for SchedulerPaymentExecutor61where62 <T as frame_system::Config>::RuntimeCall: Member63 + Dispatchable<RuntimeOrigin = RuntimeOrigin, Info = DispatchInfo>64 + SelfContainedCall<SignedInfo = SelfContainedSignedInfo>65 + GetDispatchInfo66 + From<frame_system::Call<Runtime>>,67 SelfContainedSignedInfo: Send + Sync + 'static,68 RuntimeCall: From<<T as frame_system::Config>::RuntimeCall>69 + From<<T as pallet_unique_scheduler::Config>::RuntimeCall>70 + SelfContainedCall<SignedInfo = SelfContainedSignedInfo>,71 sp_runtime::AccountId32: From<<T as frame_system::Config>::AccountId>,72{73 fn dispatch_call(74 signer: Option<<T as frame_system::Config>::AccountId>,75 call: <T as pallet_unique_scheduler::Config>::RuntimeCall,76 ) -> Result<77 Result<PostDispatchInfo, DispatchErrorWithPostInfo<PostDispatchInfo>>,78 TransactionValidityError,79 > {80 let dispatch_info = call.get_dispatch_info();81 let len = call.encoded_size();8283 let signed = match signer {84 Some(signer) => fp_self_contained::CheckedSignature::Signed(85 signer.clone().into(),86 get_signed_extras(signer.into()),87 ),88 None => fp_self_contained::CheckedSignature::Unsigned,89 };9091 let extrinsic = fp_self_contained::CheckedExtrinsic::<92 AccountId,93 RuntimeCall,94 SignedExtraScheduler,95 SelfContainedSignedInfo,96 > {97 signed,98 function: call.into(),99 };100101 extrinsic.apply::<Runtime>(&dispatch_info, len)102 }103104 fn reserve_balance(105 id: [u8; 16],106 sponsor: <T as frame_system::Config>::AccountId,107 call: <T as pallet_unique_scheduler::Config>::RuntimeCall,108 count: u32,109 ) -> Result<(), DispatchError> {110 let dispatch_info = call.get_dispatch_info();111 let weight: Balance =112 SponsorshipChargeTransactionPayment::traditional_fee(0, &dispatch_info, 0)113 .saturating_mul(count.into());114115 <Balances as NamedReservableCurrency<AccountId>>::reserve_named(116 &id,117 &(sponsor.into()),118 weight,119 )120 }121122 fn pay_for_call(123 id: [u8; 16],124 sponsor: <T as frame_system::Config>::AccountId,125 call: <T as pallet_unique_scheduler::Config>::RuntimeCall,126 ) -> Result<u128, DispatchError> {127 let dispatch_info = call.get_dispatch_info();128 let weight: Balance =129 SponsorshipChargeTransactionPayment::traditional_fee(0, &dispatch_info, 0);130 Ok(131 <Balances as NamedReservableCurrency<AccountId>>::unreserve_named(132 &id,133 &(sponsor.into()),134 weight,135 ),136 )137 }138139 fn cancel_reserve(140 id: [u8; 16],141 sponsor: <T as frame_system::Config>::AccountId,142 ) -> Result<u128, DispatchError> {143 Ok(144 <Balances as NamedReservableCurrency<AccountId>>::unreserve_named(145 &id,146 &(sponsor.into()),147 u128::MAX,148 ),149 )150 }151}