Merge pull request #231 from VenusProtocol/update-underlying-names

Update underlying names

Author: coreyar

packages/isolated-pools-abis/Bep20.json

@@ -218,5 +218,18 @@
     ],
     "name": "Transfer",
     "type": "event"
+  },
+  {
+    "anonymous": false,
+    "inputs": [
+      {
+        "indexed": true,
+        "internalType": "address",
+        "name": "implementation",
+        "type": "address"
+      }
+    ],
+    "name": "Upgraded",
+    "type": "event"
   }
 ]

packages/utils/src/testing.ts

@@ -13,7 +13,7 @@ export const assertEqual = (
       subgraphValue,
       `
       ${key} incorrect on ${
-        entity.id
+        entity.address
       } Contract: ${contractValue?.toString()}; Subgraph: ${subgraphValue};`,
     );
   } catch (e) {

subgraphs/isolated-pools/schema.graphql

@@ -38,7 +38,7 @@ type Pool @entity {
 """
 ERC20 Token
 """
-type Token @entity(immutable: true) {
+type Token @entity {
     "Address of the asset"
     id: Bytes!
     "Address of the asset"

subgraphs/isolated-pools/src/mappings/underlying.ts

@@ -0,0 +1,11 @@
+import { Upgraded } from '../../generated/templates/Underlying/BEP20';
+import { BEP20 } from '../../generated/PoolRegistry/BEP20';
+import { getOrCreateToken } from '../operations/getOrCreate';
+
+export function handleUpgraded(event: Upgraded): void {
+  const token = getOrCreateToken(event.address);
+  const erc20 = BEP20.bind(event.address);
+  token.name = erc20.name();
+  token.symbol = erc20.symbol();
+  token.save();
+}

subgraphs/isolated-pools/src/operations/create.ts

@@ -2,7 +2,10 @@ import { Address, BigInt } from '@graphprotocol/graph-ts';
 
 import { Comptroller as ComptrollerContract } from '../../generated/PoolRegistry/Comptroller';
 import { PoolRegistry as PoolRegistryContract } from '../../generated/PoolRegistry/PoolRegistry';
-import { VToken as VTokenDataSource } from '../../generated/templates';
+import {
+  Underlying as UnderlyingDataSource,
+  VToken as VTokenDataSource,
+} from '../../generated/templates';
 import {
   BadDebtIncreased,
   Borrow,
@@ -111,6 +114,7 @@ export function createMarket(
   const vTokenContract = VTokenContract.bind(vTokenAddress);
   const poolComptroller = Comptroller.bind(comptroller);
   const underlyingAddress = vTokenContract.underlying();
+  UnderlyingDataSource.create(underlyingAddress);
 
   const market = new Market(vTokenAddress);
 

subgraphs/isolated-pools/src/operations/get.ts

@@ -11,7 +11,7 @@ import {
 export const getPool = (comptroller: Address): Pool | null => {
   const pool = Pool.load(getPoolId(comptroller));
   if (!pool) {
-    log.error('Pool {} not found', [comptroller.toString()]);
+    log.error('Pool {} not found', [comptroller.toHexString()]);
   }
   return pool as Pool;
 };

subgraphs/isolated-pools/template.yaml

@@ -155,6 +155,24 @@ templates:
           handler: handleBadDebtRecovered
         - event: HealBorrow(indexed address,indexed address,uint256)
           handler: handleHealBorrow
+  - name: Underlying
+    kind: ethereum/contract
+    network: {{ network }}
+    source:
+      abi: BEP20
+    mapping:
+      abis:
+        - name: BEP20
+          file: ../../packages/isolated-pools-abis/Bep20.json
+      kind: ethereum/events
+      apiVersion: 0.0.9
+      language: wasm/assemblyscript
+      file: ./src/mappings/underlying.ts
+      entities:
+        - Token
+      eventHandlers:
+        - event: Upgraded(indexed address)
+          handler: handleUpgraded
   - name: RewardsDistributor
     kind: ethereum/contract
     network: {{ network }}