fix: support calculate capacity pre horizon

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

Author: tmigone

src/mappings/helpers/helpers.ts

@@ -1154,6 +1154,14 @@ export function updateDelegationExchangeRateForProvision(provision: Provision):
 // TODO - this is broken if we change the delegatio ratio
 // Need to remove, or find a fix
 export function calculateCapacities(indexer: Indexer): Indexer {
+  if (indexer.provisionedTokens.gt(BigInt.fromI32(0))) {
+    return calculateCapacitiesHorizon(indexer)
+  } else {
+    return calculateCapacitiesLegacy(indexer)
+  }
+}
+
+export function calculateCapacitiesHorizon(indexer: Indexer): Indexer {
   let graphNetwork = GraphNetwork.load('1')!
   let tokensDelegatedMax = indexer.provisionedTokens.times(BigInt.fromI32(graphNetwork.delegationRatio))
 
@@ -1170,6 +1178,21 @@ export function calculateCapacities(indexer: Indexer): Indexer {
   return indexer
 }
 
+export function calculateCapacitiesLegacy(indexer: Indexer): Indexer {
+  let graphNetwork = GraphNetwork.load('1')!
+  let tokensDelegatedMax = indexer.stakedTokens.times(BigInt.fromI32(graphNetwork.delegationRatio))
+
+  // Eligible to add to the capacity
+  indexer.delegatedCapacity =
+    indexer.delegatedTokens < tokensDelegatedMax ? indexer.delegatedTokens : tokensDelegatedMax
+
+  indexer.tokenCapacity = indexer.stakedTokens.plus(indexer.delegatedCapacity)
+  indexer.availableStake = indexer.tokenCapacity
+    .minus(indexer.allocatedTokens)
+    .minus(indexer.lockedTokens)
+  return indexer
+}
+
 export function calculatePricePerShare(deployment: SubgraphDeployment): BigDecimal {
   // TODO check why there's a deviation from the values of the bancor formula
   // Ideally this would be a 1 to 1 recreation of the share sell formula, but due to