1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#![cfg_attr(not(feature = "std"), no_std)]
use frame_support::{
traits::{Currency, ReservableCurrency},
PalletId,
};
use sp_runtime::traits::AccountIdConversion;
pub use pallet::*;
pub mod runtime_api;
pub type BalanceOf<T> =
<<T as Config>::Currency as Currency<<T as frame_system::Config>::AccountId>>::Balance;
#[cfg(test)]
mod tests;
const TREASURY_ID: PalletId = PalletId(*b"Treasury");
#[frame_support::pallet]
pub mod pallet {
use super::*;
use frame_support::pallet_prelude::*;
#[pallet::config]
pub trait Config: frame_system::Config {
type Currency: ReservableCurrency<Self::AccountId>;
}
#[pallet::pallet]
#[pallet::generate_store(pub(super) trait Store)]
pub struct Pallet<T>(_);
#[pallet::storage]
#[pallet::getter(fn emission_supply)]
pub type EmissionSupply<T> = StorageValue<_, BalanceOf<T>, ValueQuery>;
#[pallet::storage]
#[pallet::getter(fn poa_last_block)]
pub type PoALastBlock<T: Config> = StorageValue<_, T::Hash>;
#[pallet::genesis_config]
pub struct GenesisConfig<T: Config> {
pub emission_supply: BalanceOf<T>,
pub poa_last_block: T::Hash,
pub _marker: PhantomData<T>,
}
#[cfg(feature = "std")]
impl<T: Config> Default for GenesisConfig<T> {
fn default() -> Self {
GenesisConfig {
emission_supply: Default::default(),
poa_last_block: Default::default(),
_marker: PhantomData,
}
}
}
#[pallet::genesis_build]
impl<T: Config> GenesisBuild<T> for GenesisConfig<T> {
fn build(&self) {
EmissionSupply::<T>::put(self.emission_supply);
PoALastBlock::<T>::put(self.poa_last_block);
}
}
impl<T: Config> Pallet<T> {
pub fn treasury_account() -> T::AccountId {
TREASURY_ID.into_account_truncating()
}
pub fn treasury_balance() -> BalanceOf<T> {
T::Currency::free_balance(&Self::treasury_account())
}
}
}