`1ae086c66ec9`

source

pallets/maintenance/src/lib.rs4.1 KiBsource history
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/>.1617#![cfg_attr(not(feature = "std"), no_std)]1819pub use pallet::*;2021#[cfg(feature = "runtime-benchmarks")]22pub mod benchmarking;2324pub mod weights;2526#[frame_support::pallet]27pub mod pallet {28	use frame_support::{dispatch::*, pallet_prelude::*};29	use frame_support::{30		traits::{QueryPreimage, StorePreimage},31	};32	use frame_system::pallet_prelude::*;33	use sp_core::H256;3435	use crate::weights::WeightInfo;3637	#[pallet::config]38	pub trait Config: frame_system::Config {39		/// The overarching event type.40		type RuntimeEvent: From<Event<Self>> + IsType<<Self as frame_system::Config>::RuntimeEvent>;4142		/// The runtime origin type.43		type RuntimeOrigin: From<RawOrigin<Self::AccountId>>44			+ IsType<<Self as frame_system::Config>::RuntimeOrigin>;4546		/// The aggregated call type.47		type RuntimeCall: Parameter48			+ Dispatchable<49				RuntimeOrigin = <Self as Config>::RuntimeOrigin,50				PostInfo = PostDispatchInfo,51			> + GetDispatchInfo52			+ From<frame_system::Call<Self>>;5354		/// The preimage provider with which we look up call hashes to get the call.55		type Preimages: QueryPreimage + StorePreimage;5657		/// Weight information for extrinsics in this pallet.58		type WeightInfo: WeightInfo;59	}6061	#[pallet::event]62	#[pallet::generate_deposit(pub(super) fn deposit_event)]63	pub enum Event<T: Config> {64		MaintenanceEnabled,65		MaintenanceDisabled,66	}6768	#[pallet::pallet]69	#[pallet::generate_store(pub(super) trait Store)]70	pub struct Pallet<T>(_);7172	#[pallet::storage]73	#[pallet::getter(fn is_enabled)]74	pub type Enabled<T> = StorageValue<_, bool, ValueQuery>;7576	#[pallet::error]77	pub enum Error<T> {}7879	#[pallet::call]80	impl<T: Config> Pallet<T> {81		#[pallet::call_index(0)]82		#[pallet::weight(<T as Config>::WeightInfo::enable())]83		pub fn enable(origin: OriginFor<T>) -> DispatchResult {84			ensure_root(origin)?;8586			<Enabled<T>>::set(true);8788			Self::deposit_event(Event::MaintenanceEnabled);8990			Ok(())91		}9293		#[pallet::call_index(1)]94		#[pallet::weight(<T as Config>::WeightInfo::disable())]95		pub fn disable(origin: OriginFor<T>) -> DispatchResult {96			ensure_root(origin)?;9798			<Enabled<T>>::set(false);99100			Self::deposit_event(Event::MaintenanceDisabled);101102			Ok(())103		}104105		#[pallet::call_index(2)]106		#[pallet::weight(<T as Config>::WeightInfo::execute_preimage())]107		pub fn execute_preimage(108			origin: OriginFor<T>,109			hash: H256,110			preimage_length: Option<u32>,111			weight_bound: Weight,112		) -> DispatchResultWithPostInfo {113			use codec::Decode;114115			ensure_root(origin)?;116117			let data = T::Preimages::fetch(&hash, preimage_length)?;118			weight_bound.set_proof_size(119				weight_bound120					.proof_size()121					.checked_sub(122						data.len()123							.try_into()124							.map_err(|_| DispatchError::Corruption)?,125					)126					.ok_or(DispatchError::Exhausted)?,127			);128129			let call = <T as Config>::RuntimeCall::decode(&mut &data[..])130				.map_err(|_| DispatchError::Corruption)?;131132			ensure!(133				call.get_dispatch_info().weight.all_lte(weight_bound),134				DispatchError::Exhausted135			);136137			match call.dispatch(frame_system::RawOrigin::Root.into()) {138				Ok(post_info) => Ok(PostDispatchInfo {139					actual_weight: post_info.actual_weight,140					pays_fee: Pays::No,141				}),142				Err(error_and_info) => Err(DispatchErrorWithPostInfo {143					post_info: PostDispatchInfo {144						actual_weight: error_and_info.post_info.actual_weight,145						pays_fee: Pays::No,146					},147					error: error_and_info.error,148				}),149			}150		}151	}152}