git.delta.rocks / unique-network / refs/commits / 5832391751b5

17#![cfg_attr(not(feature = "std"), no_std)]17#![cfg_attr(not(feature = "std"), no_std)]
1818
19pub use pallet::*;19pub use pallet::*;
20use frame_support::pallet_prelude::*;
21use frame_system::pallet_prelude::*;
2022
21#[frame_support::pallet]23#[frame_support::pallet]
22pub mod pallet {24pub mod pallet {
40	#[pallet::generate_store(pub(super) trait Store)]42	#[pallet::generate_store(pub(super) trait Store)]
41	pub struct Pallet<T>(_);43	pub struct Pallet<T>(_);
44
45	#[pallet::storage]
46	#[pallet::getter(fn is_enabled)]
47	pub type Enabled<T> = StorageValue<_, bool, ValueQuery>;
4248
43	#[pallet::storage]49	#[pallet::storage]
44	#[pallet::getter(fn test_value)]50	#[pallet::getter(fn test_value)]
45	pub type TestValue<T> = StorageValue<_, u32, ValueQuery>;51	pub type TestValue<T> = StorageValue<_, u32, ValueQuery>;
4652
47	#[pallet::error]53	#[pallet::error]
48	pub enum Error<T> {54	pub enum Error<T> {
55		TestPalletDisabled,
49		TriggerRollback,56		TriggerRollback,
50	}57	}
5158
52	#[pallet::call]59	#[pallet::call]
53	impl<T: Config> Pallet<T> {60	impl<T: Config> Pallet<T> {
61		#[pallet::weight(10_000)]
62		pub fn enable(origin: OriginFor<T>) -> DispatchResult {
63			ensure_root(origin)?;
64			<Enabled<T>>::set(true);
65
66			Ok(())
67		}
68
54		#[pallet::weight(10_000)]69		#[pallet::weight(10_000)]
55		pub fn set_test_value(origin: OriginFor<T>, value: u32) -> DispatchResult {70		pub fn set_test_value(origin: OriginFor<T>, value: u32) -> DispatchResult {
56			ensure_signed(origin)?;71			Self::ensure_origin_and_enabled(origin)?;
5772
58			<TestValue<T>>::put(value);73			<TestValue<T>>::put(value);
5974
82			id: ScheduledId,97			id: ScheduledId,
83			max_test_value: u32,98			max_test_value: u32,
84		) -> DispatchResult {99		) -> DispatchResult {
100			Self::ensure_origin_and_enabled(origin.clone())?;
101
85			if <TestValue<T>>::get() < max_test_value {102			if <TestValue<T>>::get() < max_test_value {
86				Self::inc_test_value(origin)?;103				Self::inc_test_value(origin)?;
92		}109		}
93110
94		#[pallet::weight(100_000_000)]111		#[pallet::weight(100_000_000)]
95		pub fn just_take_fee(_origin: OriginFor<T>) -> DispatchResult {112		pub fn just_take_fee(origin: OriginFor<T>) -> DispatchResult {
113			Self::ensure_origin_and_enabled(origin)?;
96			Ok(())114			Ok(())
97		}115		}
98	}116	}
99}117}
118
119impl<T: Config> Pallet<T> {
120	fn ensure_origin_and_enabled(origin: OriginFor<T>) -> DispatchResult {
121		ensure_signed(origin)?;
122		<Enabled<T>>::get().then(|| ()).ok_or(<Error<T>>::TestPalletDisabled.into())
123	}
124}
100125