123456#![recursion_limit = "1024"]7#![cfg_attr(not(feature = "std"), no_std)]89#[cfg(feature = "std")]10pub use std::*;1112pub use serde::{Serialize, Deserialize};1314#[cfg(feature = "runtime-benchmarks")]15mod benchmarking;1617#[cfg(test)]18mod tests;1920pub use frame_support::{21 construct_runtime, decl_module, decl_storage, ensure,22 traits::{23 Currency, ExistenceRequirement, Get, Imbalance, KeyOwnerProofSystem, OnUnbalanced,24 Randomness, IsSubType, WithdrawReasons,25 },26 weights::{27 constants::{BlockExecutionWeight, ExtrinsicBaseWeight, RocksDbWeight, WEIGHT_PER_SECOND},28 DispatchInfo, GetDispatchInfo, IdentityFee, Pays, PostDispatchInfo, Weight,29 WeightToFeePolynomial, DispatchClass,30 },31 StorageValue, transactional,32};333435pub use frame_support::dispatch::DispatchResult;3637use sp_runtime::{38 Perbill,39 traits::{BlockNumberProvider, Zero},40};41use sp_std::convert::TryInto;4243use frame_system::{self as system};444546pub type BalanceOf<T> =47 <<T as Config>::Currency as Currency<<T as frame_system::Config>::AccountId>>::Balance;484950pub const YEAR: u32 = 2_629_800; 51pub const TOTAL_YEARS_UNTIL_FLAT: u32 = 9;52pub const START_INFLATION_PERCENT: u32 = 10;53pub const END_INFLATION_PERCENT: u32 = 4;5455pub trait Config: system::Config {56 type Currency: Currency<Self::AccountId>;57 type TreasuryAccountId: Get<Self::AccountId>;58 type InflationBlockInterval: Get<Self::BlockNumber>;5960 61 type BlockNumberProvider: BlockNumberProvider<BlockNumber = Self::BlockNumber>;62}6364decl_storage! {65 trait Store for Module<T: Config> as Inflation {66 67 pub StartingYearTotalIssuance get(fn starting_year_total_issuance): BalanceOf<T>;6869 70 pub BlockInflation get(fn block_inflation): BalanceOf<T>;71 }72}7374decl_module! {75 pub struct Module<T: Config> for enum Call76 where77 origin: T::Origin,78 {79 const InflationBlockInterval: T::BlockNumber = T::InflationBlockInterval::get();8081 fn on_initialize() -> Weight82 {83 let mut consumed_weight = 0;84 let mut add_weight = |reads, writes, weight| {85 consumed_weight += T::DbWeight::get().reads_writes(reads, writes);86 consumed_weight += weight;87 };8889 let block_interval: u32 = T::InflationBlockInterval::get().try_into().unwrap_or(0);90 let _now = T::BlockNumberProvider::current_block_number();9192 93 if (_now % T::BlockNumber::from(YEAR)).is_zero() || <BlockInflation<T>>::get().is_zero() {94 let current_year: u32 = (_now / T::BlockNumber::from(YEAR)).try_into().unwrap_or(0);9596 let one_percent = Perbill::from_percent(1);9798 if current_year <= TOTAL_YEARS_UNTIL_FLAT {99 let amount: BalanceOf<T> = Perbill::from_rational(100 block_interval * (START_INFLATION_PERCENT * TOTAL_YEARS_UNTIL_FLAT - current_year * (START_INFLATION_PERCENT - END_INFLATION_PERCENT)),101 YEAR * TOTAL_YEARS_UNTIL_FLAT102 ) * ( one_percent * T::Currency::total_issuance() );103 <BlockInflation<T>>::put(amount);104 }105 else {106 let amount: BalanceOf<T> = Perbill::from_rational(107 block_interval * END_INFLATION_PERCENT,108 YEAR109 ) * (one_percent * T::Currency::total_issuance());110 <BlockInflation<T>>::put(amount);111 }112 <StartingYearTotalIssuance<T>>::set(T::Currency::total_issuance());113114 115 T::Currency::deposit_creating(&T::TreasuryAccountId::get(), <BlockInflation<T>>::get());116117 add_weight(7, 6, 28_300_000);118 }119120 121 else if (_now % T::BlockNumber::from(block_interval)).is_zero() {122 T::Currency::deposit_into_existing(&T::TreasuryAccountId::get(), <BlockInflation<T>>::get()).ok();123124 add_weight(3, 2, 12_900_000);125 }126127 consumed_weight128 }129130 }131}