feat: added basic delegation pool management

Author: juanmardefago

src/mappings/helpers/helpers.ts

@@ -909,6 +909,18 @@ export function updateAdvancedIndexerMetrics(indexer: Indexer): Indexer {
   return indexer as Indexer
 }
 
+export function updateAdvancedIndexerMetricsHorizon(indexer: Indexer, provision: Provision): Indexer {
+  indexer.ownStakeRatio = calculateOwnStakeRatio(indexer as Indexer)
+  indexer.delegatedStakeRatio = calculateDelegatedStakeRatio(indexer as Indexer)
+  indexer.indexingRewardEffectiveCut = provision.indexingRewardsCut.toBigDecimal()
+  indexer.queryFeeEffectiveCut = provision.queryFeeCut.toBigDecimal()
+  indexer.indexerRewardsOwnGenerationRatio = calculateIndexerRewardOwnGenerationRatio(
+    indexer as Indexer,
+  )
+  indexer.overDelegationDilution = calculateOverdelegationDilution(indexer as Indexer)
+  return indexer as Indexer
+}
+
 export function updateDelegationExchangeRate(indexer: Indexer): Indexer {
   indexer.delegationExchangeRate = indexer.delegatedTokens
     .toBigDecimal()

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 { calculateCapacities, createOrLoadDataService, createOrLoadEpoch, createOrLoadGraphAccount, createOrLoadGraphNetwork, createOrLoadIndexer, createOrLoadOperator, createOrLoadProvision, updateAdvancedIndexerMetrics, updateAdvancedIndexerMetricsHorizon, updateDelegationExchangeRate } 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,7 +20,7 @@ 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 = updateAdvancedIndexerMetrics(indexer as Indexer) // Can't rely on provision here, will need to figure out a way
     indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
@@ -48,7 +46,7 @@ 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 = updateAdvancedIndexerMetrics(indexer as Indexer) // Can't rely on provision here, will need to figure out a way
     indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
@@ -67,7 +65,7 @@ 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 = updateAdvancedIndexerMetrics(indexer as Indexer) // Can't rely on provision here, will need to figure out a way
     indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
@@ -270,4 +268,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)
+    indexer.delegatedTokens = indexer.delegatedTokens.plus(event.params.tokens)
+    indexer = updateAdvancedIndexerMetricsHorizon(indexer, provision)
+    indexer = updateDelegationExchangeRate(indexer)
+    indexer = calculateCapacities(indexer)
+    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}}