1234567891011121314151617use frame_support::{traits::{PrivilegeCmp, EnsureOrigin}, weights::Weight, parameter_types};18use frame_system::{EnsureRoot, RawOrigin};19use sp_runtime::Perbill;20use core::cmp::Ordering;21use codec::Decode;22use crate::{23 runtime_common::{scheduler::SchedulerPaymentExecutor, config::substrate::RuntimeBlockWeights},24 Runtime, Call, Event, Origin, OriginCaller, Balances,25};26use pallet_unique_scheduler::ScheduledEnsureOriginSuccess;27use up_common::types::AccountId;2829parameter_types! {30 pub MaximumSchedulerWeight: Weight = Perbill::from_percent(50) *31 RuntimeBlockWeights::get().max_block;32 pub const MaxScheduledPerBlock: u32 = 50;3334 pub const NoPreimagePostponement: Option<u32> = Some(10);35 pub const Preimage: Option<u32> = Some(10);36}3738pub struct EnsureSignedOrRoot<AccountId>(sp_std::marker::PhantomData<AccountId>);39impl<O: Into<Result<RawOrigin<AccountId>, O>> + From<RawOrigin<AccountId>>, AccountId: Decode>40 EnsureOrigin<O> for EnsureSignedOrRoot<AccountId> {41 type Success = ScheduledEnsureOriginSuccess<AccountId>;42 fn try_origin(o: O) -> Result<Self::Success, O> {43 o.into().and_then(|o| match o {44 RawOrigin::Root => Ok(ScheduledEnsureOriginSuccess::Root),45 RawOrigin::Signed(who) => Ok(ScheduledEnsureOriginSuccess::Signed(who)),46 r => Err(O::from(r)),47 })48 }49}5051pub struct EqualOrRootOnly;52impl PrivilegeCmp<OriginCaller> for EqualOrRootOnly {53 fn cmp_privilege(left: &OriginCaller, right: &OriginCaller) -> Option<Ordering> {54 use RawOrigin::*;5556 let left = left.clone().try_into().ok()?;57 let right = right.clone().try_into().ok()?;5859 match (left, right) {60 (Root, Root) => Some(Ordering::Equal),61 (Root, _) => Some(Ordering::Greater),62 (_, Root) => Some(Ordering::Less),63 lr @ _ => (lr.0 == lr.1).then(|| Ordering::Equal)64 }65 }66}6768impl pallet_unique_scheduler::Config for Runtime {69 type RuntimeEvent = RuntimeEvent;70 type RuntimeOrigin = RuntimeOrigin;71 type Currency = Balances;72 type PalletsOrigin = OriginCaller;73 type RuntimeCall = RuntimeCall;74 type MaximumWeight = MaximumSchedulerWeight;75 type ScheduleOrigin = EnsureSignedOrRoot<AccountId>;76 type PrioritySetOrigin = EnsureRoot<AccountId>;77 type MaxScheduledPerBlock = MaxScheduledPerBlock;78 type WeightInfo = ();79 type CallExecutor = SchedulerPaymentExecutor;80 type OriginPrivilegeCmp = EqualOrRootOnly;81 type PreimageProvider = ();82 type NoPreimagePostponement = NoPreimagePostponement;83}