add _handleBondWithdrawn

ajna-finance · Sep 18, 2023 · d2e7b76 · d2e7b76
1 parent 8b707e6
commit d2e7b76
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 31 deletions.
diff --git a/src/mappings/base/base-pool.ts b/src/mappings/base/base-pool.ts
@@ -1,5 +1,5 @@
 import { Address, BigInt, Bytes, ethereum, log } from "@graphprotocol/graph-ts"
-import { Account, AddQuoteToken, Bucket, BucketBankruptcy, Flashloan, Lend, LoanStamped, MoveQuoteToken, Pool, PositionLend, RemoveQuoteToken, ReserveAuctionKick, ReserveAuctionTake, Token, TransferLP, UpdateInterestRate } from "../../../generated/schema"
+import { Account, AddQuoteToken, BondWithdrawn, Bucket, BucketBankruptcy, Flashloan, Lend, LoanStamped, MoveQuoteToken, Pool, PositionLend, RemoveQuoteToken, ReserveAuctionKick, ReserveAuctionTake, Token, TransferLP, UpdateInterestRate } from "../../../generated/schema"
 import {
     AddQuoteToken as AddQuoteTokenERC20Event,
     MoveQuoteToken as MoveQuoteTokenERC20Event,
@@ -572,6 +572,25 @@ export function _handleRevokeLPTransferors(event: ethereum.Event, lender: Addres
     entity.save()
 }
 
+/**********************************/
+/*** Liquidation Event Handlers ***/
+/**********************************/
+
+export function _handleBondWithdrawn(event: ethereum.Event, kicker: Address, reciever: Address, amount: BigInt): void {
+    const entity = new BondWithdrawn(
+        event.transaction.hash.concatI32(event.logIndex.toI32())
+    )
+    entity.kicker = kicker
+    entity.reciever = reciever
+    entity.amount = wadToDecimal(amount)
+
+    entity.blockNumber = event.block.number
+    entity.blockTimestamp = event.block.timestamp
+    entity.transactionHash = event.transaction.hash
+
+    entity.save()
+}
+
 /*******************************/
 /*** Reserves Event Handlers ***/
 /*******************************/

diff --git a/src/mappings/erc-20-pool.ts b/src/mappings/erc-20-pool.ts
@@ -32,7 +32,6 @@ import {
 import {
   AddCollateral,
   AuctionSettle,
-  BondWithdrawn,
   BucketTake,
   BucketTakeLPAwarded,
   DrawDebt,
@@ -56,7 +55,7 @@ import { getLiquidationAuctionId, getAuctionInfoERC20Pool, loadOrCreateLiquidati
 import { updatePool, addLiquidationToPool } from "../utils/pool/pool"
 import { lpbValueInQuote } from "../utils/pool/lend"
 import { incrementTokenTxCount } from "../utils/token-erc20"
-import { _handleAddQuoteToken, _handleApproveLPTransferors, _handleBucketBankruptcy, _handleDecreaseLPAllowance, _handleFlashLoan, _handleIncreaseLPAllowance, _handleInterestRateEvent, _handleLoanStamped, _handleMoveQuoteToken, _handleRemoveQuoteToken, _handleReserveAuctionKick, _handleReserveAuctionTake, _handleRevokeLPAllowance, _handleRevokeLPTransferors, _handleTransferLP } from "./base/base-pool"
+import { _handleAddQuoteToken, _handleApproveLPTransferors, _handleBondWithdrawn, _handleBucketBankruptcy, _handleDecreaseLPAllowance, _handleFlashLoan, _handleIncreaseLPAllowance, _handleInterestRateEvent, _handleLoanStamped, _handleMoveQuoteToken, _handleRemoveQuoteToken, _handleReserveAuctionKick, _handleReserveAuctionTake, _handleRevokeLPAllowance, _handleRevokeLPTransferors, _handleTransferLP } from "./base/base-pool"
 
 
 /*******************************/
@@ -333,20 +332,8 @@ export function handleRemoveQuoteToken(event: RemoveQuoteTokenEvent): void {
 /*** Liquidation Event Handlers ***/
 /**********************************/
 
-// TODO: move to base pool
 export function handleBondWithdrawn(event: BondWithdrawnEvent): void {
-  const entity = new BondWithdrawn(
-    event.transaction.hash.concatI32(event.logIndex.toI32())
-  )
-  entity.kicker = event.params.kicker
-  entity.reciever = event.params.reciever
-  entity.amount = wadToDecimal(event.params.amount)
-
-  entity.blockNumber = event.block.number
-  entity.blockTimestamp = event.block.timestamp
-  entity.transactionHash = event.transaction.hash
-
-  entity.save()
+  _handleBondWithdrawn(event, event.params.kicker, event.params.reciever, event.params.amount)
 }
 
 export function handleKick(event: KickEvent): void {

diff --git a/src/mappings/erc-721-pool.ts b/src/mappings/erc-721-pool.ts
@@ -34,7 +34,6 @@ import {
   AuctionNFTSettle,
   BucketTake,
   BucketTakeLPAwarded,
-  BondWithdrawn,
   DrawDebtNFT,
   LiquidationAuction,
   Loan,
@@ -57,7 +56,8 @@ import { getBorrowerInfoERC721Pool, getLoanId, loadOrCreateLoan, saveOrRemoveLoa
 import { getLiquidationAuctionId, loadOrCreateLiquidationAuction, updateLiquidationAuction, getAuctionStatus, loadOrCreateBucketTake, getAuctionInfoERC721Pool } from "../utils/pool/liquidation"
 import { updatePool, addLiquidationToPool, getLenderInfoERC721Pool } from "../utils/pool/pool"
 import { lpbValueInQuote } from "../utils/pool/lend"
-import { _handleAddQuoteToken, _handleApproveLPTransferors, _handleBucketBankruptcy, _handleDecreaseLPAllowance, _handleFlashLoan, _handleIncreaseLPAllowance, _handleInterestRateEvent, _handleLoanStamped, _handleMoveQuoteToken, _handleRemoveQuoteToken, _handleReserveAuctionKick, _handleReserveAuctionTake, _handleRevokeLPAllowance, _handleRevokeLPTransferors, _handleTransferLP } from "./base/base-pool"
+import { _handleAddQuoteToken, _handleApproveLPTransferors, _handleBondWithdrawn, _handleBucketBankruptcy, _handleDecreaseLPAllowance, _handleFlashLoan, _handleIncreaseLPAllowance, _handleInterestRateEvent, _handleLoanStamped, _handleMoveQuoteToken, _handleRemoveQuoteToken, _handleReserveAuctionKick, _handleReserveAuctionTake, _handleRevokeLPAllowance, _handleRevokeLPTransferors, _handleTransferLP } from "./base/base-pool"
+
 
 /*******************************/
 /*** Borrower Event Handlers ***/
@@ -419,20 +419,8 @@ export function handleMergeOrRemoveCollateralNFT(
 /*** Liquidation Event Handlers ***/
 /**********************************/
 
-// TODO: move to base pool
 export function handleBondWithdrawn(event: BondWithdrawnEvent): void {
-  const entity = new BondWithdrawn(
-    event.transaction.hash.concatI32(event.logIndex.toI32())
-  )
-  entity.kicker = event.params.kicker
-  entity.reciever = event.params.reciever
-  entity.amount = wadToDecimal(event.params.amount)
-
-  entity.blockNumber = event.block.number
-  entity.blockTimestamp = event.block.timestamp
-  entity.transactionHash = event.transaction.hash
-
-  entity.save()
+  _handleBondWithdrawn(event, event.params.kicker, event.params.reciever, event.params.amount)
 }
 
 // emitted concurrently with `Settle`