Nomination pools rewards

project-aleph-zero.yaml

@@ -42,6 +42,12 @@ dataSources:
             module: staking
             method: Rewarded
 
+        - handler: handleAlephZeroPoolStakingReward
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: PaidOut
+
         - handler: handleAlephZeroStakingSlash
           kind: substrate/EventHandler
           filter:
@@ -53,3 +59,15 @@ dataSources:
           filter:
             module: staking
             method: Slashed
+
+        - handler: handleAlephZeroPoolStakingBondedSlash
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: PoolSlashed
+
+        - handler: handleAlephZeroPoolStakingUnbondingSlash
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: UnbondingPoolSlashed

project-kusama.yaml

@@ -41,6 +41,12 @@ dataSources:
             module: staking
             method: Rewarded
 
+        - handler: handleKusamaPoolStakingReward
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: PaidOut
+
         - handler: handleKusamaStakingSlash
           kind: substrate/EventHandler
           filter:
@@ -52,3 +58,15 @@ dataSources:
           filter:
             module: staking
             method: Slashed
+
+        - handler: handleKusamaPoolStakingBondedSlash
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: PoolSlashed
+
+        - handler: handleKusamaPoolStakingUnbondingSlash
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: UnbondingPoolSlashed

project-polkadot.yaml

@@ -41,6 +41,12 @@ dataSources:
             module: staking
             method: Rewarded
 
+        - handler: handlePolkadotPoolStakingReward
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: PaidOut
+
         - handler: handlePolkadotStakingSlash
           kind: substrate/EventHandler
           filter:
@@ -52,3 +58,15 @@ dataSources:
           filter:
             module: staking
             method: Slashed
+
+        - handler: handlePolkadotPoolStakingBondedSlash
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: PoolSlashed
+
+        - handler: handlePolkadotPoolStakingUnbondingSlash
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: UnbondingPoolSlashed

project-westend.yaml

@@ -42,6 +42,12 @@ dataSources:
             module: staking
             method: Rewarded
 
+        - handler: handleWestendPoolStakingReward
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: PaidOut
+
         - handler: handleWestendStakingSlash
           kind: substrate/EventHandler
           filter:
@@ -53,3 +59,15 @@ dataSources:
           filter:
             module: staking
             method: Slashed
+
+        - handler: handleWestendPoolStakingBondedSlash
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: PoolSlashed
+
+        - handler: handleWestendPoolStakingUnbondingSlash
+          kind: substrate/EventHandler
+          filter:
+            module: nominationPools
+            method: UnbondingPoolSlashed

schema.graphql

@@ -60,4 +60,5 @@ type Reward @entity {
   amount: BigInt!
   accumulatedAmount: BigInt!
   type: RewardType!
+  poolId: Int
 }

src/mappings/aleph-zero.ts

@@ -4,10 +4,17 @@ import {AlephZeroRewardCalculator} from "./rewards/AlephZero";
 import {ValidatorEraInfoDataSource} from "./era/ValidatorEraInfoDataSource";
 import {Codec} from "@polkadot/types/types";
 import {INumber} from "@polkadot/types-codec/types/interfaces";
-import {handleRelaychainStakingReward, handleRelaychainStakingSlash} from "./rewards/history/relaychain";
+import {
+    handleRelaychainStakingReward, 
+    handleRelaychainStakingSlash, 
+    handleRelaychainPooledStakingReward, 
+    handleRelaychainPooledStakingBondedSlash,
+    handleRelaychainPooledStakingUnbondingSlash
+} from "./rewards/history/relaychain";
 
 const ALEPH_ZERO_GENESIS = "0x70255b4d28de0fc4e1a193d7e175ad1ccef431598211c55538f1018651a0344e"
-const STAKING_TYPE = "aleph-zero"
+const DIRECT_STAKING_TYPE = "aleph-zero"
+const POOLED_STAKING_TYPE = "nomination-pool"
 
 export async function handleAlephZeroNewEra(_: SubstrateEvent): Promise<void> {
     let validatorEraInfoDataSource = new ValidatorEraInfoDataSource();
@@ -16,18 +23,36 @@ export async function handleAlephZeroNewEra(_: SubstrateEvent): Promise<void> {
         validatorEraInfoDataSource,
         new AlephZeroRewardCalculator(validatorEraInfoDataSource),
         ALEPH_ZERO_GENESIS,
-        STAKING_TYPE
+        DIRECT_STAKING_TYPE
     )
 }
 
 export async function handleAlephZeroStakingReward(
     event: SubstrateEvent<[accountId: Codec, reward: INumber]>,
 ): Promise<void> {
-    await handleRelaychainStakingReward(event, ALEPH_ZERO_GENESIS, STAKING_TYPE)
+    await handleRelaychainStakingReward(event, ALEPH_ZERO_GENESIS, DIRECT_STAKING_TYPE)
 }
 
 export async function handleAlephZeroStakingSlash(
     event: SubstrateEvent<[account: Codec, slash: INumber]>,
 ): Promise<void> {
-    await handleRelaychainStakingSlash(event, ALEPH_ZERO_GENESIS, STAKING_TYPE)
+    await handleRelaychainStakingSlash(event, ALEPH_ZERO_GENESIS, DIRECT_STAKING_TYPE)
+}
+
+export async function handleAlephZeroPoolStakingReward(
+    event: SubstrateEvent<[accountId: Codec, poolId: INumber, reward: INumber]>,
+): Promise<void> {
+    await handleRelaychainPooledStakingReward(event, ALEPH_ZERO_GENESIS, POOLED_STAKING_TYPE)
+}
+
+export async function handleAlephZeroPoolStakingBondedSlash(
+    event: SubstrateEvent<[poolId: INumber, slash: INumber]>,
+): Promise<void> {
+    await handleRelaychainPooledStakingBondedSlash(event, ALEPH_ZERO_GENESIS, POOLED_STAKING_TYPE)
+}
+
+export async function handleAlephZeroPoolStakingUnbondingSlash(
+    event: SubstrateEvent<[era: INumber, poolId: INumber, slash: INumber]>,
+): Promise<void> {
+    await handleRelaychainPooledStakingUnbondingSlash(event, ALEPH_ZERO_GENESIS, POOLED_STAKING_TYPE)
 }

src/mappings/kusama.ts

@@ -4,10 +4,17 @@ import {RelaychainRewardCalculator} from "./rewards/Relaychain";
 import {ValidatorEraInfoDataSource} from "./era/ValidatorEraInfoDataSource";
 import {Codec} from "@polkadot/types/types";
 import {INumber} from "@polkadot/types-codec/types/interfaces";
-import {handleRelaychainStakingReward, handleRelaychainStakingSlash} from "./rewards/history/relaychain";
+import {
+    handleRelaychainStakingReward, 
+    handleRelaychainStakingSlash, 
+    handleRelaychainPooledStakingReward, 
+    handleRelaychainPooledStakingBondedSlash,
+    handleRelaychainPooledStakingUnbondingSlash
+} from "./rewards/history/relaychain";
 
 const KUSAMA_GENESIS = "0xb0a8d493285c2df73290dfb7e61f870f17b41801197a149ca93654499ea3dafe"
-const STAKING_TYPE = "relaychain"
+const DIRECT_STAKING_TYPE = "relaychain"
+const POOLED_STAKING_TYPE = "nomination-pool"
 
 export async function handleKusamaNewEra(_: SubstrateEvent): Promise<void> {
     let validatorEraInfoDataSource = new ValidatorEraInfoDataSource();
@@ -16,18 +23,36 @@ export async function handleKusamaNewEra(_: SubstrateEvent): Promise<void> {
         validatorEraInfoDataSource,
         await RelaychainRewardCalculator(validatorEraInfoDataSource),
         KUSAMA_GENESIS,
-        STAKING_TYPE
+        DIRECT_STAKING_TYPE
     )
 }
 
 export async function handleKusamaStakingReward(
     event: SubstrateEvent<[accountId: Codec, reward: INumber]>,
 ): Promise<void> {
-    await handleRelaychainStakingReward(event, KUSAMA_GENESIS, STAKING_TYPE)
+    await handleRelaychainStakingReward(event, KUSAMA_GENESIS, DIRECT_STAKING_TYPE)
 }
 
 export async function handleKusamaStakingSlash(
     event: SubstrateEvent<[account: Codec, slash: INumber]>,
 ): Promise<void> {
-    await handleRelaychainStakingSlash(event, KUSAMA_GENESIS, STAKING_TYPE)
+    await handleRelaychainStakingSlash(event, KUSAMA_GENESIS, DIRECT_STAKING_TYPE)
+}
+
+export async function handleKusamaPoolStakingReward(
+    event: SubstrateEvent<[accountId: Codec, poolId: INumber, reward: INumber]>,
+): Promise<void> {
+    await handleRelaychainPooledStakingReward(event, KUSAMA_GENESIS, POOLED_STAKING_TYPE)
+}
+
+export async function handleKusamaPoolStakingBondedSlash(
+    event: SubstrateEvent<[poolId: INumber, slash: INumber]>,
+): Promise<void> {
+    await handleRelaychainPooledStakingBondedSlash(event, KUSAMA_GENESIS, POOLED_STAKING_TYPE)
+}
+
+export async function handleKusamaPoolStakingUnbondingSlash(
+    event: SubstrateEvent<[era: INumber, poolId: INumber, slash: INumber]>,
+): Promise<void> {
+    await handleRelaychainPooledStakingUnbondingSlash(event, KUSAMA_GENESIS, POOLED_STAKING_TYPE)
 }

src/mappings/polkadot.ts

@@ -4,10 +4,17 @@ import {RelaychainRewardCalculator} from "./rewards/Relaychain";
 import {ValidatorEraInfoDataSource} from "./era/ValidatorEraInfoDataSource";
 import {Codec} from "@polkadot/types/types";
 import {INumber} from "@polkadot/types-codec/types/interfaces";
-import {handleRelaychainStakingReward, handleRelaychainStakingSlash} from "./rewards/history/relaychain";
+import {
+    handleRelaychainStakingReward, 
+    handleRelaychainStakingSlash, 
+    handleRelaychainPooledStakingReward, 
+    handleRelaychainPooledStakingBondedSlash,
+    handleRelaychainPooledStakingUnbondingSlash
+} from "./rewards/history/relaychain";
 
 const POLKADOT_GENESIS = "0x91b171bb158e2d3848fa23a9f1c25182fb8e20313b2c1eb49219da7a70ce90c3"
-const STAKING_TYPE = "relaychain"
+const DIRECT_STAKING_TYPE = "relaychain"
+const POOLED_STAKING_TYPE = "nomination-pool"
 
 export async function handlePolkadotNewEra(_: SubstrateEvent): Promise<void> {
     let validatorEraInfoDataSource = new ValidatorEraInfoDataSource();
@@ -16,18 +23,36 @@ export async function handlePolkadotNewEra(_: SubstrateEvent): Promise<void> {
         validatorEraInfoDataSource,
         await RelaychainRewardCalculator(validatorEraInfoDataSource),
         POLKADOT_GENESIS,
-        STAKING_TYPE
+        DIRECT_STAKING_TYPE
     )
 }
 
 export async function handlePolkadotStakingReward(
     event: SubstrateEvent<[accountId: Codec, reward: INumber]>,
 ): Promise<void> {
-    await handleRelaychainStakingReward(event, POLKADOT_GENESIS, STAKING_TYPE)
+    await handleRelaychainStakingReward(event, POLKADOT_GENESIS, DIRECT_STAKING_TYPE)
 }
 
 export async function handlePolkadotStakingSlash(
     event: SubstrateEvent<[account: Codec, slash: INumber]>,
 ): Promise<void> {
-    await handleRelaychainStakingSlash(event, POLKADOT_GENESIS, STAKING_TYPE)
+    await handleRelaychainStakingSlash(event, POLKADOT_GENESIS, DIRECT_STAKING_TYPE)
+}
+
+export async function handlePolkadotPoolStakingReward(
+    event: SubstrateEvent<[accountId: Codec, poolId: INumber, reward: INumber]>,
+): Promise<void> {
+    await handleRelaychainPooledStakingReward(event, POLKADOT_GENESIS, POOLED_STAKING_TYPE)
+}
+
+export async function handlePolkadotPoolStakingBondedSlash(
+    event: SubstrateEvent<[poolId: INumber, slash: INumber]>,
+): Promise<void> {
+    await handleRelaychainPooledStakingBondedSlash(event, POLKADOT_GENESIS, POOLED_STAKING_TYPE)
+}
+
+export async function handlePolkadotPoolStakingUnbondingSlash(
+    event: SubstrateEvent<[era: INumber, poolId: INumber, slash: INumber]>,
+): Promise<void> {
+    await handleRelaychainPooledStakingUnbondingSlash(event, POLKADOT_GENESIS, POOLED_STAKING_TYPE)
 }

src/mappings/rewards/history/common.ts

@@ -12,6 +12,8 @@ export interface RewardArgs {
     chainId: string
 
     stakingType: string
+
+    poolId?: number
 }
 
 export async function handleReward(rewardProps: RewardArgs, event: SubstrateEvent) {
@@ -29,9 +31,13 @@ export async function handleReward(rewardProps: RewardArgs, event: SubstrateEven
         timestamp: timestamp(event.block),
         blockNumber: blockNumber(event),
         networkId: rewardProps.chainId,
-        stakingType: rewardProps.stakingType
+        stakingType: rewardProps.stakingType,
     });
 
+    if (rewardProps.poolId !== undefined) {
+        accountReward.poolId = rewardProps.poolId
+    }
+
     await accountReward.save()
 }
 

src/mappings/rewards/history/parachain.ts

@@ -17,7 +17,7 @@ export async function handleParachainStakingReward(
         address: accountId.toString(),
         type: RewardType.reward,
         chainId: chainId,
-        stakingType: stakingType
+        stakingType: stakingType,
     }
 
     await handleReward(rewardProps, event)