1234567891011121314151617use 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, maintenance};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>;363738pub 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 maintenance::CheckMaintenance,45 ChargeTransactionPayment<Runtime>,46);4748fn get_signed_extras(from: <Runtime as frame_system::Config>::AccountId) -> SignedExtraScheduler {49 (50 frame_system::CheckSpecVersion::<Runtime>::new(),51 frame_system::CheckGenesis::<Runtime>::new(),52 frame_system::CheckEra::<Runtime>::from(Era::Immortal),53 frame_system::CheckNonce::<Runtime>::from(frame_system::Pallet::<Runtime>::account_nonce(54 from,55 )),56 frame_system::CheckWeight::<Runtime>::new(),57 maintenance::CheckMaintenance,58 ChargeTransactionPayment::<Runtime>::from(0),59 )60}6162pub struct SchedulerPaymentExecutor;6364impl<T: frame_system::Config + pallet_unique_scheduler::Config, SelfContainedSignedInfo>65 DispatchCall<T, SelfContainedSignedInfo> for SchedulerPaymentExecutor66where67 <T as frame_system::Config>::RuntimeCall: Member68 + Dispatchable<RuntimeOrigin = RuntimeOrigin, Info = DispatchInfo>69 + SelfContainedCall<SignedInfo = SelfContainedSignedInfo>70 + GetDispatchInfo71 + From<frame_system::Call<Runtime>>,72 SelfContainedSignedInfo: Send + Sync + 'static,73 RuntimeCall: From<<T as frame_system::Config>::RuntimeCall>74 + From<<T as pallet_unique_scheduler::Config>::RuntimeCall>75 + SelfContainedCall<SignedInfo = SelfContainedSignedInfo>,76 sp_runtime::AccountId32: From<<T as frame_system::Config>::AccountId>,77{78 fn dispatch_call(79 signer: Option<<T as frame_system::Config>::AccountId>,80 call: <T as pallet_unique_scheduler::Config>::RuntimeCall,81 ) -> Result<82 Result<PostDispatchInfo, DispatchErrorWithPostInfo<PostDispatchInfo>>,83 TransactionValidityError,84 > {85 let dispatch_info = call.get_dispatch_info();86 let len = call.encoded_size();8788 let signed = match signer {89 Some(signer) => fp_self_contained::CheckedSignature::Signed(90 signer.clone().into(),91 get_signed_extras(signer.into()),92 ),93 None => fp_self_contained::CheckedSignature::Unsigned,94 };9596 let extrinsic = fp_self_contained::CheckedExtrinsic::<97 AccountId,98 RuntimeCall,99 SignedExtraScheduler,100 SelfContainedSignedInfo,101 > {102 signed,103 function: call.into(),104 };105106 extrinsic.apply::<Runtime>(&dispatch_info, len)107 }108109 fn reserve_balance(110 id: [u8; 16],111 sponsor: <T as frame_system::Config>::AccountId,112 call: <T as pallet_unique_scheduler::Config>::RuntimeCall,113 count: u32,114 ) -> Result<(), DispatchError> {115 let dispatch_info = call.get_dispatch_info();116 let weight: Balance =117 SponsorshipChargeTransactionPayment::traditional_fee(0, &dispatch_info, 0)118 .saturating_mul(count.into());119120 <Balances as NamedReservableCurrency<AccountId>>::reserve_named(121 &id,122 &(sponsor.into()),123 weight,124 )125 }126127 fn pay_for_call(128 id: [u8; 16],129 sponsor: <T as frame_system::Config>::AccountId,130 call: <T as pallet_unique_scheduler::Config>::RuntimeCall,131 ) -> Result<u128, DispatchError> {132 let dispatch_info = call.get_dispatch_info();133 let weight: Balance =134 SponsorshipChargeTransactionPayment::traditional_fee(0, &dispatch_info, 0);135 Ok(136 <Balances as NamedReservableCurrency<AccountId>>::unreserve_named(137 &id,138 &(sponsor.into()),139 weight,140 ),141 )142 }143144 fn cancel_reserve(145 id: [u8; 16],146 sponsor: <T as frame_system::Config>::AccountId,147 ) -> Result<u128, DispatchError> {148 Ok(149 <Balances as NamedReservableCurrency<AccountId>>::unreserve_named(150 &id,151 &(sponsor.into()),152 u128::MAX,153 ),154 )155 }156}