Fix poolSnapshot and fixed rate balance (#357)

* fix

* fix lint

* fix import

Author: mihaisc

package-lock.json

@@ -1,4 +1,3 @@
-import { log } from '@graphprotocol/graph-ts'
 import { NFTCreated, TokenCreated } from '../@types/ERC721Factory/ERC721Factory'
 import { decimal } from './utils/constants'
 import { weiToDecimal } from './utils/generic'
@@ -8,7 +7,6 @@ import { getToken, getNftToken } from './utils/tokenUtils'
 import { addDatatoken } from './utils/globalUtils'
 
 export function handleNftCreated(event: NFTCreated): void {
-  log.warning('nft handleNftCreated {}', [event.params.tokenURI.toString()])
   // const nft = new Nft(event.params.newTokenAddress.toHexString())
   const nft = getNftToken(event.params.newTokenAddress)
   const user = getUser(event.params.admin.toHexString())

src/mappings/erc721Factory.ts

@@ -7,7 +7,8 @@ import {
   ExchangeMintStateChanged,
   ExchangeRateChanged,
   Swapped,
-  PublishMarketFeeChanged
+  PublishMarketFeeChanged,
+  TokenCollected
 } from '../@types/templates/FixedRateExchange/FixedRateExchange'
 import {
   FixedRateExchange,
@@ -218,3 +219,21 @@ export function handlePublishMarketFeeChanged(
     fixedRateExchange.save()
   }
 }
+
+export function handleTokenCollected(event: TokenCollected): void {
+  const fixedRateId = getFixedRateGraphID(
+    event.params.exchangeId.toHexString(),
+    event.address
+  )
+  const fixedRateExchange = getFixedRateExchange(fixedRateId)
+
+  if (event.params.token.toHexString() == fixedRateExchange.baseToken) {
+    const baseToken = getToken(event.params.token, false)
+    fixedRateExchange.baseTokenBalance =
+      fixedRateExchange.baseTokenBalance.minus(
+        weiToDecimal(event.params.amount.toBigDecimal(), baseToken.decimals)
+      )
+
+    fixedRateExchange.save()
+  }
+}

src/mappings/fixedRateExchange.ts

@@ -36,7 +36,7 @@ export function handleJoin(event: LOG_JOIN): void {
   pool.joinCount = pool.joinCount.plus(integer.ONE)
 
   // get token,  update pool transaction, poolSnapshot
-  const poolSnapshot = getPoolSnapshot(pool.id, event.block.timestamp.toI32())
+
   const token = getToken(event.params.tokenIn, false)
   const ammount = weiToDecimal(
     event.params.tokenAmountIn.toBigDecimal(),
@@ -45,24 +45,36 @@ export function handleJoin(event: LOG_JOIN): void {
   if (token.isDatatoken) {
     poolTx.datatoken = token.id
     poolTx.datatokenValue = ammount
-
-    poolSnapshot.datatokenLiquidity =
-      poolSnapshot.datatokenLiquidity.plus(ammount)
+    if (pool.isFinalized) {
+      const poolSnapshot = getPoolSnapshot(
+        pool.id,
+        event.block.timestamp.toI32()
+      )
+      poolSnapshot.datatokenLiquidity =
+        poolSnapshot.datatokenLiquidity.plus(ammount)
+
+      poolSnapshot.save()
+    }
 
     pool.datatokenLiquidity = pool.datatokenLiquidity.plus(ammount)
   } else {
     poolTx.baseToken = token.id
     poolTx.baseTokenValue = ammount
 
-    poolSnapshot.baseTokenLiquidity =
-      poolSnapshot.baseTokenLiquidity.plus(ammount)
-
+    if (pool.isFinalized) {
+      const poolSnapshot = getPoolSnapshot(
+        pool.id,
+        event.block.timestamp.toI32()
+      )
+      poolSnapshot.baseTokenLiquidity =
+        poolSnapshot.baseTokenLiquidity.plus(ammount)
+      poolSnapshot.save()
+    }
     pool.baseTokenLiquidity = pool.baseTokenLiquidity.plus(ammount)
 
     addLiquidity(token.id, ammount)
   }
 
-  poolSnapshot.save()
   poolTx.save()
   pool.save()
 }
@@ -225,15 +237,19 @@ export function handleSetup(event: LOG_SETUP): void {
   poolTx.type = PoolTransactionType.SETUP
   poolTx.baseToken = token.id
   poolTx.datatoken = datatoken.id
+  pool.save()
+  poolTx.save()
+
   const poolSnapshot = getPoolSnapshot(pool.id, event.block.timestamp.toI32())
   poolSnapshot.spotPrice = spotPrice
+  poolSnapshot.baseTokenLiquidity = pool.baseTokenLiquidity
+  poolSnapshot.datatokenLiquidity = pool.datatokenLiquidity
+  poolSnapshot.totalShares = pool.totalShares
 
-  poolTx.save()
   poolSnapshot.save()
   const globalStats = getGlobalStats()
   globalStats.poolCount = globalStats.poolCount + 1
   globalStats.save()
-  pool.save()
   datatoken.save()
 }
 
@@ -243,10 +259,6 @@ export function handlerBptTransfer(event: Transfer): void {
   const poolAddress = event.address.toHex()
   const caller = getUser(event.transaction.from.toHex())
   const poolTx = getPoolTransaction(event, caller.id, PoolTransactionType.SWAP)
-  const poolSnapshot = getPoolSnapshot(
-    poolAddress,
-    event.block.timestamp.toI32()
-  )
 
   // btoken has 18 decimals
   const ammount = weiToDecimal(event.params.amt.toBigDecimal(), 18)
@@ -261,7 +273,15 @@ export function handlerBptTransfer(event: Transfer): void {
     pool.totalShares = pool.totalShares.plus(ammount)
 
     // check tx?
-    poolSnapshot.totalShares = pool.totalShares
+    if (pool.isFinalized) {
+      const poolSnapshot = getPoolSnapshot(
+        poolAddress,
+        event.block.timestamp.toI32()
+      )
+      poolSnapshot.totalShares = pool.totalShares
+      poolSnapshot.save()
+    }
+
     pool.save()
   } else {
     if (poolAddress != fromAddress) {
@@ -275,7 +295,14 @@ export function handlerBptTransfer(event: Transfer): void {
     // remove
     const pool = getPool(poolAddress)
     pool.totalShares = pool.totalShares.minus(ammount)
-    poolSnapshot.totalShares = pool.totalShares
+    if (pool.isFinalized) {
+      const poolSnapshot = getPoolSnapshot(
+        poolAddress,
+        event.block.timestamp.toI32()
+      )
+      poolSnapshot.totalShares = pool.totalShares
+      poolSnapshot.save()
+    }
     pool.save()
   } else {
     if (poolAddress != toAddress) {
@@ -286,7 +313,6 @@ export function handlerBptTransfer(event: Transfer): void {
   }
 
   poolTx.save()
-  poolSnapshot.save()
 }
 
 export function handlePublishMarketFeeChanged(

src/mappings/pool.ts

@@ -99,7 +99,6 @@ export function createPoolSnapshot(
   const snapshotId = getPoolSnapshotId(poolAddress, timestamp)
 
   const pool = getPool(poolAddress)
-
   const snapshot = new PoolSnapshot(snapshotId)
 
   snapshot.pool = poolAddress

src/mappings/utils/poolUtils.ts

@@ -28,7 +28,7 @@ dataSources:
           handler: handleNftCreated
         - event: TokenCreated(indexed address,indexed address,string,string,uint256,address)
           handler: handleNewToken
-  
+
   - kind: ethereum/contract
     name: FactoryRouter
     network: __NETWORK__
@@ -71,7 +71,6 @@ dataSources:
           handler: handleDispenserContractRemoved
 
 templates:
-
   - name: ERC20Template
     kind: ethereum/contract
     network: __NETWORK__
@@ -262,7 +261,9 @@ templates:
           handler: handleSwap
         - event: PublishMarketFeeChanged(indexed bytes32,address,address,uint256)
           handler: handlePublishMarketFeeChanged
-  
+        - event: TokenCollected(indexed bytes32,indexed address,indexed address,uint256)
+          handler: handleTokenCollected
+
   - name: SSContract
     kind: ethereum/contract
     network: __NETWORK__
@@ -277,9 +278,9 @@ templates:
         - SSContract
       abis:
         - name: SSContract
-          file: ./node_modules/@oceanprotocol/contracts/artifacts/contracts/pools/ssContracts/SideStaking.sol/SideStaking.json 
+          file: ./node_modules/@oceanprotocol/contracts/artifacts/contracts/pools/ssContracts/SideStaking.sol/SideStaking.json
       eventHandlers:
         - event: VestingCreated(indexed address,indexed address,uint256,uint256)
           handler: handleVestingCreated
         - event: Vesting(indexed address,indexed address,indexed address,uint256)
-          handler: handleVesting
+          handler: handleVesting