fix: capacity calculations

Signed-off-by: Tomás Migone <[email protected]>

Author: tmigone

src/mappings/helpers/helpers.ts

@@ -1155,16 +1155,18 @@ export function updateDelegationExchangeRateForProvision(provision: Provision):
 // Need to remove, or find a fix
 export function calculateCapacities(indexer: Indexer): Indexer {
   let graphNetwork = GraphNetwork.load('1')!
-  let tokensDelegatedMax = indexer.stakedTokens.times(BigInt.fromI32(graphNetwork.delegationRatio))
+  let tokensDelegatedMax = indexer.provisionedTokens.times(BigInt.fromI32(graphNetwork.delegationRatio))
 
   // Eligible to add to the capacity
+  let delegatedTokens = indexer.delegatedTokens.minus(indexer.delegatedThawingTokens)
   indexer.delegatedCapacity =
-    indexer.delegatedTokens < tokensDelegatedMax ? indexer.delegatedTokens : tokensDelegatedMax
+    delegatedTokens < tokensDelegatedMax ? delegatedTokens : tokensDelegatedMax
 
-  indexer.tokenCapacity = indexer.stakedTokens.plus(indexer.delegatedCapacity)
+  indexer.tokenCapacity = indexer.provisionedTokens
+    .minus(indexer.thawingTokens)
+    .plus(indexer.delegatedCapacity)
   indexer.availableStake = indexer.tokenCapacity
     .minus(indexer.allocatedTokens)
-    .minus(indexer.lockedTokens)
   return indexer
 }
 

src/mappings/horizonStaking.ts

@@ -2,7 +2,7 @@ import { BigInt } from '@graphprotocol/graph-ts'
 import { addresses } from '../../config/addresses'
 import { AllowedLockedVerifierSet, DelegatedTokensWithdrawn, DelegationFeeCutSet, DelegationSlashed, DelegationSlashingEnabled, HorizonStakeDeposited, HorizonStakeLocked, HorizonStakeWithdrawn, MaxThawingPeriodSet, OperatorSet, StakeDelegatedWithdrawn, ThawingPeriodCleared, TokensDelegated, TokensDeprovisioned, TokensToDelegationPoolAdded, TokensUndelegated } from '../types/HorizonStaking/HorizonStaking'
 import { DelegatedStake, Delegator, Indexer, Provision, ThawRequest } from '../types/schema'
-import { createOrLoadDataService, createOrLoadDelegatedStakeForProvision, createOrLoadDelegator, createOrLoadEpoch, createOrLoadGraphAccount, createOrLoadGraphNetwork, createOrLoadHorizonOperator, createOrLoadIndexer, createOrLoadProvision, joinID, updateAdvancedIndexerMetrics, updateAdvancedProvisionMetrics, updateDelegationExchangeRate, updateDelegationExchangeRateForProvision } from './helpers/helpers'
+import { calculateCapacities, createOrLoadDataService, createOrLoadDelegatedStakeForProvision, createOrLoadDelegator, createOrLoadEpoch, createOrLoadGraphAccount, createOrLoadGraphNetwork, createOrLoadHorizonOperator, createOrLoadIndexer, createOrLoadProvision, joinID, updateAdvancedIndexerMetrics, updateAdvancedProvisionMetrics, updateDelegationExchangeRate, updateDelegationExchangeRateForProvision } from './helpers/helpers'
 import {
     ProvisionCreated,
     ProvisionIncreased,
@@ -126,6 +126,7 @@ export function handleProvisionThawed(event: ProvisionThawed): void {
     let provision = createOrLoadProvision(event.params.serviceProvider, event.params.verifier, event.block.timestamp)
 
     indexer.thawingTokens = indexer.thawingTokens.plus(event.params.tokens)
+    indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
     dataService.totalTokensThawing = dataService.totalTokensThawing.plus(event.params.tokens)
@@ -146,6 +147,7 @@ export function handleTokensDeprovisioned(event: TokensDeprovisioned): void {
 
     indexer.provisionedTokens = indexer.provisionedTokens.minus(event.params.tokens)
     indexer.thawingTokens = indexer.thawingTokens.minus(event.params.tokens)
+    indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
     dataService.totalTokensProvisioned = dataService.totalTokensProvisioned.minus(event.params.tokens)

src/mappings/subgraphService.ts

@@ -1,6 +1,6 @@
 import { BigDecimal, BigInt, Bytes, ethereum, log } from "@graphprotocol/graph-ts"
 import { AllocationClosed, AllocationCreated, AllocationResized, CurationCutSet, DelegationRatioSet, IndexingRewardsCollected, MaxPOIStalenessSet, ProvisionTokensRangeSet, QueryFeesCollected, RewardsDestinationSet, ServiceProviderRegistered, StakeToFeesRatioSet, ThawingPeriodRangeSet, VerifierCutRangeSet } from "../types/SubgraphService/SubgraphService"
-import { batchUpdateSubgraphSignalledTokens, calculatePricePerShare, createOrLoadDataService, createOrLoadGraphNetwork, createOrLoadEpoch,createOrLoadIndexerQueryFeePaymentAggregation, createOrLoadPaymentSource, createOrLoadProvision, createOrLoadSubgraphDeployment, joinID, updateDelegationExchangeRate } from "./helpers/helpers"
+import { batchUpdateSubgraphSignalledTokens, calculatePricePerShare, createOrLoadDataService, createOrLoadGraphNetwork, createOrLoadEpoch,createOrLoadIndexerQueryFeePaymentAggregation, createOrLoadPaymentSource, createOrLoadProvision, createOrLoadSubgraphDeployment, joinID, updateDelegationExchangeRate, calculateCapacities } from "./helpers/helpers"
 import { Allocation, Indexer, PoiSubmission, SubgraphDeployment } from "../types/schema"
 import { addresses } from "../../config/addresses"
 import { tuplePrefixBytes } from "./helpers/decoder"
@@ -64,6 +64,7 @@ export function handleAllocationCreated(event: AllocationCreated): void {
     indexer.allocatedTokens = indexer.allocatedTokens.plus(event.params.tokens)
     indexer.totalAllocationCount = indexer.totalAllocationCount.plus(BigInt.fromI32(1))
     indexer.allocationCount = indexer.allocationCount + 1
+    indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
     // update provision
@@ -140,6 +141,7 @@ export function handleAllocationClosed(event: AllocationClosed): void {
 
     indexer.allocatedTokens = indexer.allocatedTokens.minus(event.params.tokens)
     indexer.allocationCount = indexer.allocationCount - 1
+    indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
     // update provision
@@ -189,6 +191,7 @@ export function handleAllocationResized(event: AllocationResized): void {
     // update indexer
     let indexer = Indexer.load(indexerID)!
     indexer.allocatedTokens = indexer.allocatedTokens.plus(diffTokens)
+    indexer = calculateCapacities(indexer as Indexer)
     indexer.save()
 
     // update provision