[Horizon] DelegationPoolManagement implementation

schema.graphql

@@ -890,6 +890,14 @@ type Provision @entity {
   indexingRewardsCut: BigInt!
 
   # Might want to add polymorphic handling of different fee cuts through a generic hashmap in the future
+
+  # Delegation
+  "Tokens delegated to the provision"
+  delegatedTokens: BigInt!
+  "Total shares of the delegator pool"
+  delegatorShares: BigInt!
+  "Exchange rate of of tokens received for each share"
+  delegationExchangeRate: BigDecimal!
 }
 
 enum PaymentTypes {

src/mappings/helpers/helpers.ts

@@ -209,6 +209,9 @@ export function createOrLoadProvision(indexerAddress: Bytes, verifierAddress: By
     provision.queryFeeCut = BigInt.fromI32(0)
     provision.indexingFeeCut = BigInt.fromI32(0)
     provision.indexingRewardsCut = BigInt.fromI32(0)
+    provision.delegatedTokens = BigInt.fromI32(0)
+    provision.delegatorShares = BigInt.fromI32(0)
+    provision.delegationExchangeRate = BigInt.fromI32(0).toBigDecimal()
     provision.save()
   }
 

src/mappings/horizonStaking.ts

@@ -1,8 +1,8 @@
-import { BigInt, log } from '@graphprotocol/graph-ts'
+import { BigInt } from '@graphprotocol/graph-ts'
 import { addresses } from '../../config/addresses'
-import { DelegationFeeCutSet, HorizonStakeDeposited, HorizonStakeLocked, HorizonStakeWithdrawn, OperatorSet, TokensDeprovisioned } from '../types/HorizonStaking/HorizonStaking'
-import { Indexer, GraphNetwork, ThawRequest } from '../types/schema'
-import { calculateCapacities, createOrLoadDataService, createOrLoadEpoch, createOrLoadGraphAccount, createOrLoadGraphNetwork, createOrLoadIndexer, createOrLoadOperator, createOrLoadProvision, updateAdvancedIndexerMetrics } from './helpers/helpers'
+import { DelegationFeeCutSet, HorizonStakeDeposited, HorizonStakeLocked, HorizonStakeWithdrawn, OperatorSet, TokensDeprovisioned, TokensToDelegationPoolAdded } from '../types/HorizonStaking/HorizonStaking'
+import { Indexer, ThawRequest } from '../types/schema'
+import { createOrLoadDataService, createOrLoadEpoch, createOrLoadGraphAccount, createOrLoadGraphNetwork, createOrLoadIndexer, createOrLoadOperator, createOrLoadProvision } from './helpers/helpers'
 import {
     ProvisionCreated,
     ProvisionIncreased,
@@ -12,8 +12,6 @@ import {
     ProvisionThawed,
     ThawRequestCreated,
     ThawRequestFulfilled,
-    ThawRequestsFulfilled,
-    ThawingPeriodCleared
 } from '../types/HorizonStaking/HorizonStaking'
 
 export function handleHorizonStakeDeposited(event: HorizonStakeDeposited): void {
@@ -22,8 +20,6 @@ export function handleHorizonStakeDeposited(event: HorizonStakeDeposited): void
     let indexer = createOrLoadIndexer(event.params.serviceProvider, event.block.timestamp)
     let previousStake = indexer.stakedTokens
     indexer.stakedTokens = indexer.stakedTokens.plus(event.params.tokens)
-    indexer = updateAdvancedIndexerMetrics(indexer as Indexer)
-    indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
     // Update graph network
@@ -48,8 +44,6 @@ export function handleHorizonStakeLocked(event: HorizonStakeLocked): void {
     let indexer = Indexer.load(id)!
     indexer.lockedTokens = event.params.tokens
     indexer.tokensLockedUntil = event.params.until.toI32()
-    indexer = updateAdvancedIndexerMetrics(indexer as Indexer)
-    indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
     // update graph network
@@ -67,8 +61,6 @@ export function handleHorizonStakeWithdrawn(event: HorizonStakeWithdrawn): void
     indexer.stakedTokens = indexer.stakedTokens.minus(event.params.tokens)
     indexer.lockedTokens = BigInt.fromI32(0) // set to 0 to prevent issues when Stage 2 comes
     indexer.tokensLockedUntil = 0 // always set to 0 when withdrawn
-    indexer = updateAdvancedIndexerMetrics(indexer as Indexer)
-    indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
     // Update graph network
@@ -270,4 +262,14 @@ export function handleThawRequestFulfilled(event: ThawRequestFulfilled): void {
     request.tokens = event.params.tokens
     request.valid = event.params.valid
     request.save()
+}
+
+export function handleTokensToDelegationPoolAdded(event: TokensToDelegationPoolAdded): void {
+    let indexer = Indexer.load(event.params.serviceProvider.toHexString())!
+    let provision = createOrLoadProvision(event.params.serviceProvider, event.params.verifier, event.block.timestamp)
+    provision.delegatedTokens = provision.delegatedTokens.plus(event.params.tokens)
+    provision.save()
+
+    indexer.delegatedTokens = indexer.delegatedTokens.plus(event.params.tokens) // this only serves as a general tracker, but the real deal is per provision
+    indexer.save()
 }

subgraph.template.yaml

@@ -427,6 +427,8 @@ dataSources:
           handler: handleOperatorSet
         - event: DelegationFeeCutSet(indexed address,indexed address,indexed uint8,uint256)
           handler: handleDelegationFeeCutSet
+        - event: TokensToDelegationPoolAdded(indexed address,indexed address,uint256)
+          handler: handleTokensToDelegationPoolAdded
   - kind: ethereum/contract
     name: Curation
     network: {{network}}