From 4174983d3e3a0018c49969ff97f279843c34fcfa Mon Sep 17 00:00:00 2001
From: Filippo Fontana <filippo@embrio.tech>
Date: Fri, 7 Jun 2024 13:16:19 +0200
Subject: [PATCH] feat: track realizedProfit in pool entity

---
 schema.graphql                         | 4 ++++
 src/mappings/handlers/loansHandlers.ts | 9 ++++++---
 src/mappings/services/poolService.ts   | 6 ++++++
 3 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/schema.graphql b/schema.graphql
index 10fc29fd..f0a963bd 100644
--- a/schema.graphql
+++ b/schema.graphql
@@ -67,6 +67,8 @@ type Pool @entity {
   sumPoolFeesPaidAmount: BigInt #Applies to All 
   sumPoolFeesPendingAmount: BigInt #Applies to All 
 
+  sumRealizedProfitFifoByPeriod: BigInt
+
   # Cumulated transaction data since pool creation
   sumBorrowedAmount: BigInt
   sumRepaidAmount: BigInt
@@ -123,6 +125,8 @@ type PoolSnapshot @entity {
   sumPoolFeesPaidAmount: BigInt #Applies to All 
   sumPoolFeesPendingAmount: BigInt #Applies to All 
 
+  sumRealizedProfitFifoByPeriod: BigInt
+
   # Cumulated transaction data since pool creation
   sumBorrowedAmount: BigInt
   sumRepaidAmount: BigInt
diff --git a/src/mappings/handlers/loansHandlers.ts b/src/mappings/handlers/loansHandlers.ts
index 41e6564f..a5fbae00 100644
--- a/src/mappings/handlers/loansHandlers.ts
+++ b/src/mappings/handlers/loansHandlers.ts
@@ -222,6 +222,7 @@ async function _handleLoanRepaid(event: SubstrateEvent<LoanRepaidEvent>) {
         quantity.toBigInt(),
         settlementPrice.toBigInt()
       )
+      await pool.increaseRealizedProfitFifo(realizedProfitFifo)
     }
 
     const at = await AssetTransactionService.repaid({ ...assetTransactionBaseData, realizedProfitFifo })
@@ -337,6 +338,7 @@ async function _handleLoanDebtTransferred(event: SubstrateEvent<LoanDebtTransfer
         quantity.toBigInt(),
         settlementPrice.toBigInt()
       )
+      await pool.increaseRealizedProfitFifo(realizedProfitFifo)
     }
     await fromAsset.updateIpfsAssetName()
     await fromAsset.save()
@@ -443,15 +445,16 @@ async function _handleLoanDebtTransferred1024(event: SubstrateEvent<LoanDebtTran
     await fromAsset.repay(amount)
     await fromAsset.save()
 
+    const quantity = nToBigInt(bnToBn(amount).mul(WAD).div(bnToBn(fromAsset.currentPrice)))
+    const realizedProfitFifo = await AssetPositionService.sellFifo(toAsset.id, quantity, toAsset.currentPrice)
+
+    await pool.increaseRealizedProfitFifo(realizedProfitFifo)
     await pool.increaseRepayments1024(amount)
     await pool.save()
 
     await epoch.increaseRepayments(amount)
     await epoch.save()
 
-    const quantity = nToBigInt(bnToBn(amount).mul(WAD).div(bnToBn(fromAsset.currentPrice)))
-    const realizedProfitFifo = await AssetPositionService.sellFifo(toAsset.id, quantity, toAsset.currentPrice)
-
     // principal repayment transaction
     const principalRepayment = await AssetTransactionService.repaid({
       ...txData,
diff --git a/src/mappings/services/poolService.ts b/src/mappings/services/poolService.ts
index e72a0488..9bb6877c 100644
--- a/src/mappings/services/poolService.ts
+++ b/src/mappings/services/poolService.ts
@@ -72,6 +72,7 @@ export class PoolService extends Pool {
     this.sumPoolFeesPaidAmountByPeriod = BigInt(0)
     this.deltaPortfolioValuationByPeriod = BigInt(0)
     this.sumInterestAccruedByPeriod = BigInt(0)
+    this.sumRealizedProfitFifoByPeriod = BigInt(0)
 
     this.sumBorrowedAmount = BigInt(0)
     this.sumRepaidAmount = BigInt(0)
@@ -403,6 +404,11 @@ export class PoolService extends Pool {
     logger.info(`Updating sumPoolFeesPendingAmount for pool ${this.id} to ${pendingAmount.toString(10)}`)
     this.sumPoolFeesPendingAmount = pendingAmount
   }
+
+  public increaseRealizedProfitFifo(amount: bigint) {
+    logger.info(`Increasing umRealizedProfitFifoByPeriod for pool ${this.id} by ${amount.toString(10)}`)
+    this.sumRealizedProfitFifoByPeriod += amount
+  }
 }
 
 export interface ActiveLoanData {