fix: epoch nav computations for snapshots

Author: filo87

src/mappings/handlers/poolsHandlers.ts

@@ -186,6 +186,10 @@ async function _handleEpochExecuted(event: SubstrateEvent<EpochClosedExecutedEve
   const pool = await PoolService.getById(poolId.toString())
   if (!pool) throw missingPool
 
+  assertPropInitialized(pool, 'currencyId', 'string')
+  const currency = await CurrencyService.get(pool.currencyId!)
+  if (!currency) throw new Error(`Currency ${pool.currencyId!} not found for pool ${pool.id}`)
+
   const epoch = await EpochService.getById(poolId.toString(), epochId.toNumber())
   if (!epoch) throw new Error(`Epoch ${epochId.toString(10)} not found for pool ${poolId.toString(10)}`)
 
@@ -195,7 +199,6 @@ async function _handleEpochExecuted(event: SubstrateEvent<EpochClosedExecutedEve
   await pool.executeEpoch(epochId.toNumber())
   await pool.increaseInvestments(epoch.sumInvestedAmount!)
   await pool.increaseRedemptions(epoch.sumRedeemedAmount!)
-  await pool.save()
 
   // Compute and save aggregated order fulfillment
   const tranches = await TrancheService.getByPoolId(poolId.toString())
@@ -298,6 +301,12 @@ async function _handleEpochExecuted(event: SubstrateEvent<EpochClosedExecutedEve
   }
   await nextEpoch.saveWithStates()
 
+  // Update pool nav
+  const partialNavs = tranches.map((tranche) => tranche.computePartialNav())
+  await pool.setNav(partialNavs.reduce((nav, partialNav) => nav + partialNav, BigInt(0)))
+  await pool.updateNormalizedNAV(currency.decimals)
+  await pool.save()
+
   // Track investments and redemptions for onchain cash
   if (!event.extrinsic) throw new Error('Event has no extrinsic')
   const onChainCashAsset = await AssetService.getById(pool.id, ONCHAIN_CASH_ASSET_ID)

src/mappings/services/poolService.ts

@@ -286,6 +286,13 @@ export class PoolService extends Pool {
     } else {
       this.normalizedNAV = BigInt(this.netAssetValue!.toString(10).substring(0, WAD_DIGITS))
     }
+    logger.info(`Normalized NAV for pool ${this.id} set to ${this.normalizedNAV.toString(10)}`)
+    return this
+  }
+
+  public setNav(nav: bigint) {
+    logger.info(`Setting nav for pool ${this.id} to ${nav.toString(10)}`)
+    this.netAssetValue = nav
     return this
   }
 

src/mappings/services/trancheService.ts

@@ -287,6 +287,12 @@ export class TrancheService extends Tranche {
     }
     this.snapshot = snapshots.pop()
   }
+
+  public computePartialNav(): bigint {
+    const supply = bnToBn(this.tokenSupply)
+    const price = bnToBn(this.tokenPrice)
+    return nToBigInt(supply.mul(price).div(WAD))
+  }
 }
 
 type BigIntFields<T> = { [K in keyof Required<T>]: Required<T>[K] extends bigint ? K : never }[keyof Required<T>]