feat: add fee parameter to calculateStakingRewards

packages/nimiq-rewards-calculator/src/index.ts

@@ -31,6 +31,12 @@ export interface CalculateStakingRewardsParams {
    * @default 'main-albatross'
    */
   network?: 'main-albatross' | 'test-albatross'
+
+  /**
+   * The fee percentage that the pool charges for staking.
+   *  @default 0
+   */
+  fee: number
 }
 
 export interface CalculateStakingRewardsResult {
@@ -61,7 +67,7 @@ export interface CalculateStakingRewardsResult {
  * @returns {CalculateStakingRewardsResult} The result of the calculation.
  */
 export function calculateStakingRewards(params: CalculateStakingRewardsParams): CalculateStakingRewardsResult {
-  const { amount, durationInDays, autoRestake = true, stakedSupplyRatio, network = 'main-albatross' } = params
+  const { amount, durationInDays, autoRestake = true, stakedSupplyRatio, network = 'main-albatross', fee = 0 } = params
   const genesisSupply = network === 'main-albatross' ? SUPPLY_AT_PROOF_OF_STAKE_FORK_DATE : SUPPLY_AT_PROOF_OF_STAKE_FORK_DATE_TESTNET
 
   const initialRewardsPerDay = posSupplyAt(24 * 60 * 60 * 1000) - genesisSupply
@@ -82,6 +88,7 @@ export function calculateStakingRewards(params: CalculateStakingRewardsParams):
       - Math.log(DECAY_PER_DAY * genesisSupply)
     )
   }
+  gainRatio = gainRatio * (1 - fee) / 1e5
   const totalAmount = amount * (1 + gainRatio)
   return { totalAmount, gain: totalAmount - amount, gainRatio }
 }