refactor: read underlying decimals to calculate exchange rate

subgraphs/etherfi-promo/schema.graphql

@@ -10,8 +10,12 @@ type TVL @entity {
 Entity to iterate over suppliers
 """
 type Supply @entity {
-    "Token Address"
+    "VToken Address"
     id: Bytes!
+    "Address of the asset"
+    underlyingAddress: Bytes!
+    "Decimals of the asset"
+    underlyingDecimals: Int!
     suppliers: [SupplierAccount!]! @derivedFrom(field: "token")
 }
 
@@ -21,6 +25,10 @@ Entity to iterate over borrowers
 type Borrow @entity {
     "Token Address"
     id: Bytes!
+    "Address of the asset"
+    underlyingAddress: Bytes!
+    "Decimals of the asset"
+    underlyingDecimals: Int!
     borrowers: [BorrowerAccount!]! @derivedFrom(field: "token")
 }
 

subgraphs/etherfi-promo/src/mappings/vToken.ts

@@ -7,27 +7,28 @@ import { getBorrow, getSupply } from '../operations/get';
 import { getOrCreateBorrowerAccount, getOrCreateSupplierAccount } from '../operations/getOrCreate';
 import { updateBorrowerAccount, updateSupplierAccount, updateTvl } from '../operations/update';
 import exponentToBigDecimal from '../utilities/exponentToBigDecimal';
-import exponentToBigInt from '../utilities/exponentToBigInt';
 
 export function handleMint(event: Mint): void {
   const minter = event.params.minter;
   const supplierAccount = getOrCreateSupplierAccount(minter, event.address);
+  const vToken = getSupply(event.address);
   updateSupplierAccount(
     minter,
     event.address,
     supplierAccount.effective_balance.plus(
-      event.params.mintAmount.toBigDecimal().div(exponentToBigDecimal(18)),
+      event.params.mintAmount.toBigDecimal().div(exponentToBigDecimal(vToken.underlyingDecimals)),
     ),
   );
 }
 
 export function handleBorrow(event: Borrow): void {
   const borrower = event.params.borrower;
+  const vToken = getBorrow(event.address);
   getOrCreateBorrowerAccount(borrower, event.address);
   updateBorrowerAccount(
     borrower,
     event.address,
-    event.params.accountBorrows.toBigDecimal().div(exponentToBigDecimal(18)),
+    event.params.accountBorrows.toBigDecimal().div(exponentToBigDecimal(vToken.underlyingDecimals)),
   );
 }
 
@@ -37,6 +38,7 @@ export function handleTransfer(event: Transfer): void {
   const fromAccountAddress = event.params.from;
   // If the to account is the vToken address we assume it was a redeem
   const toAccountAddress = event.params.to;
+  const vToken = getSupply(event.address);
 
   if (
     fromAccountAddress.notEqual(event.address) &&
@@ -45,25 +47,23 @@ export function handleTransfer(event: Transfer): void {
   ) {
     const vTokenContract = VTokenContract.bind(event.address);
     const exchangeRateMantissa = vTokenContract.exchangeRateCurrent();
+
     const amountUnderlying = exchangeRateMantissa
       .times(event.params.amount)
-      .div(exponentToBigInt(18));
+      .toBigDecimal()
+      .div(exponentToBigDecimal(18 + vToken.underlyingDecimals));
     const fromAccount = getOrCreateSupplierAccount(fromAccountAddress, event.address);
     updateSupplierAccount(
       fromAccountAddress,
       event.address,
-      fromAccount.effective_balance.minus(
-        amountUnderlying.toBigDecimal().div(exponentToBigDecimal(18)),
-      ),
+      fromAccount.effective_balance.minus(amountUnderlying),
     );
     // To
     const toAccount = getOrCreateSupplierAccount(toAccountAddress, event.address);
     updateSupplierAccount(
       toAccountAddress,
       event.address,
-      toAccount.effective_balance.plus(
-        amountUnderlying.toBigDecimal().div(exponentToBigDecimal(18)),
-      ),
+      toAccount.effective_balance.plus(amountUnderlying),
     );
   }
 }
@@ -72,23 +72,28 @@ export function handleAccrueInterest(event: AccrueInterest): void {
   const supply = getSupply(event.address);
   supply.suppliers.load().forEach(supplier => {
     const vTokenContract = VTokenContract.bind(Address.fromBytes(supplier.token));
+    const vToken = getSupply(Address.fromBytes(supplier.token));
 
     const exchangeRateMantissa = vTokenContract.exchangeRateCurrent();
     const vTokenBalance = vTokenContract.balanceOf(Address.fromBytes(supplier.address));
-    const amountUnderlying = exchangeRateMantissa.times(vTokenBalance).div(exponentToBigInt(18));
-    supplier.effective_balance = amountUnderlying.toBigDecimal().div(exponentToBigDecimal(18));
+    const amountUnderlying = exchangeRateMantissa
+      .times(vTokenBalance)
+      .toBigDecimal()
+      .div(exponentToBigDecimal(18 + vToken.underlyingDecimals));
+    supplier.effective_balance = amountUnderlying;
     supplier.save();
   });
 
   const borrow = getBorrow(event.address);
   borrow.borrowers.load().forEach(borrower => {
     const vTokenContract = VTokenContract.bind(Address.fromBytes(borrower.token));
+    const vToken = getBorrow(Address.fromBytes(borrower.token));
     const underlyingBorrowBalance = vTokenContract.borrowBalanceCurrent(
       Address.fromBytes(borrower.address),
     );
     borrower.effective_balance = underlyingBorrowBalance
       .toBigDecimal()
-      .div(exponentToBigDecimal(18));
+      .div(exponentToBigDecimal(vToken.underlyingDecimals));
     borrower.save();
   });
 

subgraphs/etherfi-promo/src/operations/get.ts

@@ -1,5 +1,5 @@
 import { Address } from '@graphprotocol/graph-ts';
-
+import { VToken } from '../../generated/vWeETH/VToken';
 import { Borrow, BorrowerAccount, SupplierAccount, Supply, TVL } from '../../generated/schema';
 import { zeroBigDecimal } from '../constants';
 import { getPositionId } from '../utilities/ids';
@@ -19,6 +19,13 @@ export const getSupply = (tokenAddress: Address): Supply => {
   let supply = Supply.load(tokenAddress);
   if (!supply) {
     supply = new Supply(tokenAddress);
+
+    const vTokenContract = VToken.bind(tokenAddress);
+    const underlyingAddress = vTokenContract.underlying();
+    const erc20 = VToken.bind(underlyingAddress);
+
+    supply.underlyingAddress = underlyingAddress;
+    supply.underlyingDecimals = erc20.decimals();
   }
   supply.save();
   return supply;
@@ -36,6 +43,12 @@ export const getBorrow = (tokenAddress: Address): Borrow => {
   let borrow = Borrow.load(tokenAddress);
   if (!borrow) {
     borrow = new Borrow(tokenAddress);
+    const vTokenContract = VToken.bind(tokenAddress);
+    const underlyingAddress = vTokenContract.underlying();
+    const erc20 = VToken.bind(underlyingAddress);
+
+    borrow.underlyingAddress = underlyingAddress;
+    borrow.underlyingDecimals = erc20.decimals();
   }
   borrow.save();
   return borrow;

subgraphs/etherfi-promo/src/operations/update.ts

@@ -3,7 +3,7 @@ import { Address, BigDecimal, ethereum } from '@graphprotocol/graph-ts';
 import { BorrowerAccount, SupplierAccount, TVL } from '../../generated/schema';
 import { ERC20 as ERC20Contract } from '../../generated/vWeETH/ERC20';
 import { VToken as VTokenContract } from '../../generated/vWeETH/VToken';
-import { getBorrowerAccount, getSupplierAccount, getTvl } from './get';
+import { getBorrowerAccount, getSupplierAccount, getTvl, getSupply } from './get';
 import exponentToBigDecimal from '../utilities/exponentToBigDecimal';
 
 export function updateSupplierAccount(
@@ -31,6 +31,7 @@ export function updateBorrowerAccount(
 export function updateTvl(event: ethereum.Event): TVL {
   const vTokenContract = VTokenContract.bind(event.address);
   const underlyingAddress = vTokenContract.underlying();
+  const vToken = getSupply(event.address);
   const underlyingContract = ERC20Contract.bind(underlyingAddress);
   const tvl = getTvl(event.address);
   const totalBorrows = vTokenContract.totalBorrowsCurrent();
@@ -40,7 +41,7 @@ export function updateTvl(event: ethereum.Event): TVL {
     .plus(totalBorrows)
     .minus(totalReserves)
     .toBigDecimal()
-    .div(exponentToBigDecimal(18));
+    .div(exponentToBigDecimal(vToken.underlyingDecimals));
   tvl.save();
   return tvl;
 }