83 lines
2.2 KiB
Rust
83 lines
2.2 KiB
Rust
use crate::{Config, Key};
|
|
use codec::{Decode, Encode};
|
|
use frame_support::{dispatch::DispatchInfo, ensure};
|
|
use scale_info::TypeInfo;
|
|
use sp_runtime::{
|
|
traits::{DispatchInfoOf, Dispatchable, SignedExtension},
|
|
transaction_validity::{
|
|
InvalidTransaction, TransactionPriority, TransactionValidity, TransactionValidityError,
|
|
UnknownTransaction, ValidTransaction,
|
|
},
|
|
};
|
|
use sp_std::{fmt, marker::PhantomData};
|
|
|
|
#[derive(Clone, Eq, PartialEq, Encode, Decode, TypeInfo)]
|
|
#[scale_info(skip_type_params(T))]
|
|
pub struct CheckOnlySudoAccount<T: Config + Send + Sync>(PhantomData<T>);
|
|
|
|
impl<T: Config + Send + Sync> Default for CheckOnlySudoAccount<T> {
|
|
fn default() -> Self {
|
|
Self(Default::default())
|
|
}
|
|
}
|
|
|
|
impl<T: Config + Send + Sync> fmt::Debug for CheckOnlySudoAccount<T> {
|
|
#[cfg(feature = "std")]
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(f, "CheckOnlySudoAccount")
|
|
}
|
|
|
|
#[cfg(not(feature = "std"))]
|
|
fn fmt(&self, _: &mut fmt::Formatter) -> fmt::Result {
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl<T: Config + Send + Sync> CheckOnlySudoAccount<T> {
|
|
/// Creates new `SignedExtension` to check sudo key.
|
|
pub fn new() -> Self {
|
|
Self::default()
|
|
}
|
|
}
|
|
|
|
impl<T: Config + Send + Sync> SignedExtension for CheckOnlySudoAccount<T>
|
|
where
|
|
<T as Config>::RuntimeCall: Dispatchable<Info = DispatchInfo>,
|
|
{
|
|
const IDENTIFIER: &'static str = "CheckOnlySudoAccount";
|
|
type AccountId = T::AccountId;
|
|
type Call = <T as Config>::RuntimeCall;
|
|
type AdditionalSigned = ();
|
|
type Pre = ();
|
|
|
|
fn additional_signed(&self) -> Result<Self::AdditionalSigned, TransactionValidityError> {
|
|
Ok(())
|
|
}
|
|
|
|
fn validate(
|
|
&self,
|
|
who: &Self::AccountId,
|
|
_call: &Self::Call,
|
|
info: &DispatchInfoOf<Self::Call>,
|
|
_len: usize,
|
|
) -> TransactionValidity {
|
|
let sudo_key: T::AccountId = Key::<T>::get().ok_or(UnknownTransaction::CannotLookup)?;
|
|
ensure!(*who == sudo_key, InvalidTransaction::BadSigner);
|
|
|
|
Ok(ValidTransaction {
|
|
priority: info.weight.ref_time() as TransactionPriority,
|
|
..Default::default()
|
|
})
|
|
}
|
|
|
|
fn pre_dispatch(
|
|
self,
|
|
who: &Self::AccountId,
|
|
call: &Self::Call,
|
|
info: &DispatchInfoOf<Self::Call>,
|
|
len: usize,
|
|
) -> Result<Self::Pre, TransactionValidityError> {
|
|
self.validate(who, call, info, len).map(|_| ())
|
|
}
|
|
}
|