feat: track thawing delegation tokens

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

Author: tmigone

schema.graphql

@@ -815,6 +815,8 @@ type Indexer @entity(immutable: false) {
   delegators: [DelegatedStake!]! @derivedFrom(field: "indexer")
   "CURRENT tokens delegated to the indexer"
   delegatedTokens: BigInt!
+  "Tokens delegated that are thawing from the provision"
+  delegatedThawingTokens: BigInt!
   "Ratio between the amount of the indexers own stake over the total usable stake."
   ownStakeRatio: BigDecimal!
   "Ratio between the amount of delegated stake over the total usable stake."
@@ -998,6 +1000,8 @@ type Provision @entity(immutable: false) {
   # Delegation
   "Tokens delegated to the provision"
   delegatedTokens: BigInt!
+  "Tokens delegated that are thawing from the provision"
+  delegatedThawingTokens: BigInt!
   "Total shares of the delegator pool"
   delegatorShares: BigInt!
   "Exchange rate of tokens received for each share"

src/mappings/helpers/helpers.ts

@@ -163,6 +163,7 @@ export function createOrLoadIndexer(indexerAddress: Bytes, timestamp: BigInt ):
     indexer.availableStake = BigInt.fromI32(0)
 
     indexer.delegatedTokens = BigInt.fromI32(0)
+    indexer.delegatedThawingTokens = BigInt.fromI32(0)
     indexer.ownStakeRatio = BigDecimal.fromString('0')
     indexer.delegatedStakeRatio = BigDecimal.fromString('0')
     indexer.delegatorShares = BigInt.fromI32(0)
@@ -244,6 +245,7 @@ export function createOrLoadProvision(indexerAddress: Bytes, verifierAddress: By
     provision.indexerQueryFees = BigInt.fromI32(0)
     provision.delegatorQueryFees = BigInt.fromI32(0)
     provision.delegatedTokens = BigInt.fromI32(0)
+    provision.delegatedThawingTokens = BigInt.fromI32(0)
     provision.delegatorShares = BigInt.fromI32(0)
     provision.delegationExchangeRate = BigInt.fromI32(0).toBigDecimal()
     provision.thawingUntil = BigInt.fromI32(0)

src/mappings/horizonStaking.ts

@@ -466,7 +466,6 @@ export function handleTokensUndelegated(event: TokensUndelegated): void {
 
     let beforeUpdateDelegationExchangeRate = provision.delegationExchangeRate
 
-    provision.delegatedTokens = provision.delegatedTokens.minus(event.params.tokens)
     provision.delegatorShares = provision.delegatorShares.minus(event.params.shares)
     if (provision.delegatorShares != BigInt.fromI32(0)) {
         provision = updateDelegationExchangeRateForProvision(provision as Provision)
@@ -477,7 +476,6 @@ export function handleTokensUndelegated(event: TokensUndelegated): void {
     // update indexer
     let indexerID = event.params.serviceProvider.toHexString()
     let indexer = Indexer.load(indexerID)!
-    indexer.delegatedTokens = indexer.delegatedTokens.minus(event.params.tokens)
     indexer.delegatorShares = indexer.delegatorShares.minus(event.params.shares)
     if (indexer.delegatorShares != BigInt.fromI32(0)) {
         indexer = updateDelegationExchangeRate(indexer as Indexer)
@@ -534,8 +532,15 @@ export function handleTokensUndelegated(event: TokensUndelegated): void {
 export function handleDelegatedTokensWithdrawn(event: DelegatedTokensWithdrawn): void {
     let provision = createOrLoadProvision(event.params.serviceProvider, event.params.verifier, event.block.timestamp)
     // might want to track locked/thawing tokens in provision too
+    provision.delegatedTokens = provision.delegatedTokens.minus(event.params.tokens)
     provision.save()
 
+    let indexerID = event.params.serviceProvider.toHexString()
+    let indexer = Indexer.load(indexerID)!
+
+    indexer.delegatedTokens = indexer.delegatedTokens.minus(event.params.tokens)
+    indexer.save()
+
     // update delegated stake
     let delegatorID = event.params.delegator.toHexString()
     let id = joinID([delegatorID, provision.id])