feat: use rpc prices from trancheTokenPrices() when investor transact…

schema.graphql

@@ -83,6 +83,7 @@ type PoolSnapshot @entity {
 
 type Tranche @entity {
   id: ID! #poolId-trancheId
+  index: Int!
   type: String! @index
   pool: Pool! @index
 

src/mappings/handlers/ormlTokensHandlers.ts

@@ -29,6 +29,10 @@ async function _handleTokenTransfer(event: SubstrateEvent<TokensTransferEvent>):
     const pool = await PoolService.getById(poolId.toString())
     const tranche = await TrancheService.getById(poolId.toString(), trancheId.toHex())
 
+    // Update tranche price
+    await tranche.updatePriceFromRpc()
+    await tranche.save()
+
     const orderData = {
       poolId: poolId.toString(),
       trancheId: trancheId.toString(),

src/mappings/handlers/poolsHandlers.ts

@@ -29,7 +29,7 @@ async function _handlePoolCreated(event: SubstrateEvent<PoolEvent>): Promise<voi
   for (const [index, trancheData] of tranches.entries()) {
     const trancheId = trancheIds[index]
     logger.info(`Creating tranche with id: ${trancheId}`)
-    const trancheService = await TrancheService.init(trancheId, poolId.toString(), trancheData)
+    const trancheService = await TrancheService.init(poolId.toString(), trancheId, index, trancheData)
     await trancheService.updateSupply()
     await trancheService.updateDebt(trancheData.debt.toBigInt())
     await trancheService.save()
@@ -181,6 +181,10 @@ async function _handleInvestOrderUpdated(event: SubstrateEvent<OrderEvent>): Pro
   const pool = await PoolService.getById(poolId.toString())
   const tranche = await TrancheService.getById(poolId.toString(), trancheId.toHex())
 
+  // Update tranche price
+  await tranche.updatePriceFromRpc()
+  await tranche.save()
+
   const orderData: InvestorTransactionData = {
     poolId: poolId.toString(),
     trancheId: trancheId.toString(),
@@ -278,6 +282,11 @@ async function _handleOrdersCollected(event: SubstrateEvent<OrdersCollectedEvent
 
   const pool = await PoolService.getById(poolId.toString())
   const tranche = await TrancheService.getById(poolId.toString(), trancheId.toHex())
+
+  // Update tranche price
+  await tranche.updatePriceFromRpc()
+  await tranche.save()
+
   const { payoutTokenAmount, payoutCurrencyAmount } = outstandingCollections
 
   const orderData = {

src/mappings/services/trancheService.ts

@@ -1,5 +1,5 @@
 import { u128, u64 } from '@polkadot/types'
-import { bnToBn, hexToU8a, nToBigInt } from '@polkadot/util'
+import { bnToBn, nToBigInt } from '@polkadot/util'
 import { RAY } from '../../config'
 import { errorHandler } from '../../helpers/errorHandler'
 import { ExtendedRpc, TrancheDetails } from '../../helpers/types'
@@ -14,7 +14,7 @@ export class TrancheService {
     this.trancheState = trancheState
   }
 
-  static init = async (trancheId: string, poolId: string, trancheData: TrancheDetails) => {
+  static init = async (poolId: string, trancheId: string, index: number, trancheData: TrancheDetails) => {
     const trancheState = new TrancheState(`${poolId}-${trancheId}`)
     trancheState.type = 'ALL'
 
@@ -28,6 +28,7 @@ export class TrancheService {
     tranche.type = 'ALL'
     tranche.poolId = poolId
     tranche.trancheId = trancheId
+    tranche.index = index
     tranche.isResidual = trancheData.trancheType.isResidual
     tranche.seniority = trancheData.seniority.toNumber()
 
@@ -81,10 +82,8 @@ export class TrancheService {
   private _updatePricefromRpc = async () => {
     logger.info(`Qerying RPC price for tranche ${this.tranche.id}`)
     const poolId = new u64(api.registry, this.tranche.poolId)
-    const trancheId = Array.from(hexToU8a(this.tranche.trancheId, 128))
-    logger.info(`Compact trancheId: ${trancheId}`)
-    const tokenPrice = await (api.rpc as ExtendedRpc).pools.trancheTokenPrice(poolId, trancheId)
-    this.updatePrice(tokenPrice.toBigInt())
+    const tokenPrices = await (api.rpc as ExtendedRpc).pools.trancheTokenPrices(poolId)
+    this.updatePrice(tokenPrices[this.tranche.index].toBigInt())
     return this
   }
   public updatePriceFromRpc = errorHandler(this._updatePricefromRpc)