Beanstalk 3 Subgraphs (BIP-50)

projects/subgraph-basin/manifests/arbitrum.yaml

@@ -0,0 +1,109 @@
+specVersion: 0.0.9
+schema:
+  file: ../schema.graphql
+dataSources:
+  ###
+  # INITIALIZATION
+  ###
+  - kind: ethereum/contract
+    name: InitVersion
+    network: arbitrum-one
+    source:
+      address: "0xBA51AAAa8C2f911AE672e783707Ceb2dA6E97521"
+      abi: Aquifer
+      startBlock: 235608025
+      endBlock: 235608025
+    mapping:
+      kind: ethereum/events
+      apiVersion: 0.0.7
+      language: wasm/assemblyscript
+      entities:
+        - Version
+      abis:
+        - name: Aquifer
+          file: ../../subgraph-core/abis/Aquifer.json
+      blockHandlers:
+        - handler: handleInitVersion
+          filter:
+            kind: once
+      file: ../src/utils/constants/Version.ts
+  - kind: ethereum/contract
+    name: Aquifer
+    network: arbitrum-one
+    source:
+      address: "0xBA51AAAa8C2f911AE672e783707Ceb2dA6E97521"
+      abi: Aquifer
+      startBlock: 235608025
+    mapping:
+      kind: ethereum/events
+      apiVersion: 0.0.7
+      language: wasm/assemblyscript
+      entities:
+        - Aquifer
+      abis:
+        - name: Aquifer
+          file: ../../subgraph-core/abis/Aquifer.json
+        - name: ERC20
+          file: ../../subgraph-core/abis/ERC20.json
+      eventHandlers:
+        - event: BoreWell(address,address,address[],(address,bytes),(address,bytes)[],bytes)
+          handler: handleBoreWell
+      file: ../src/templates/AquiferHandler.ts
+  - kind: ethereum/contract
+    name: Beanstalk
+    network: arbitrum-one
+    source:
+      address: "0xD1A0060ba708BC4BCD3DA6C37EFa8deDF015FB70"
+      abi: Beanstalk
+      startBlock: 235608025
+    mapping:
+      kind: ethereum/events
+      apiVersion: 0.0.7
+      language: wasm/assemblyscript
+      entities:
+        - Beanstalk
+      abis:
+        - name: Beanstalk
+          file: ../../subgraph-core/abis/Beanstalk/Beanstalk-BIP50.json
+      eventHandlers:
+        - event: Sunrise(indexed uint256)
+          handler: handleSunrise
+      file: ../src/BeanstalkHandler.ts
+templates:
+  - kind: ethereum/contract
+    name: Well
+    network: arbitrum-one
+    source:
+      abi: Well
+    mapping:
+      kind: ethereum/events
+      apiVersion: 0.0.7
+      language: wasm/assemblyscript
+      entities:
+        - Well
+      abis:
+        - name: Well
+          file: ../../subgraph-core/abis/Well.json
+        - name: BeanstalkPrice
+          file: ../../subgraph-core/abis/BeanstalkPrice.json
+        - name: ERC20
+          file: ../../subgraph-core/abis/ERC20.json
+      eventHandlers:
+        - event: AddLiquidity(uint256[],uint256,address)
+          handler: handleAddLiquidity
+        - event: RemoveLiquidity(uint256,uint256[],address)
+          handler: handleRemoveLiquidity
+        - event: RemoveLiquidityOneToken(uint256,address,uint256,address)
+          handler: handleRemoveLiquidityOneToken
+        - event: Swap(address,address,uint256,uint256,address)
+          handler: handleSwap
+        - event: Shift(uint256[],address,uint256,address)
+          handler: handleShift
+        - event: Sync(uint256[],uint256,address)
+          handler: handleSync
+      file: ../src/WellHandler.ts
+# features:
+#   - grafting
+# graft:
+#   base: QmWHi6hu2wXnyxBHHmQHwCQLoq5KkoTX2qLm8MdyVXTyTu
+#   block: 20216425

projects/subgraph-basin/manifests/codegen-abis.yaml

@@ -26,7 +26,7 @@ dataSources:
         - name: ERC20
           file: ../../subgraph-core/abis/ERC20.json
         - name: Beanstalk
-          file: ../../subgraph-core/abis/Beanstalk/Beanstalk-Replanted.json
+          file: ../../subgraph-core/abis/Beanstalk/Beanstalk-BIP50.json
         - name: Well
           file: ../../subgraph-core/abis/Well.json
         - name: CurvePrice
@@ -37,7 +37,7 @@ dataSources:
         - handler: handleInitVersion
           filter:
             kind: once
-      file: ../src/utils/Init.ts
+      file: ../src/utils/constants/Version.ts
 templates:
   - kind: ethereum/contract
     name: Well
@@ -57,4 +57,4 @@ templates:
         - handler: handleInitVersion
           filter:
             kind: once
-      file: ../src/utils/Init.ts
+      file: ../src/utils/constants/Version.ts

projects/subgraph-basin/manifests/ethereum.yaml

@@ -6,7 +6,7 @@ dataSources:
   # INITIALIZATION
   ###
   - kind: ethereum/contract
-    name: Version
+    name: InitVersion
     network: mainnet
     source:
       address: "0xBA51AAAA95aeEFc1292515b36D86C51dC7877773"
@@ -26,7 +26,7 @@ dataSources:
         - handler: handleInitVersion
           filter:
             kind: once
-      file: ../src/utils/Init.ts
+      file: ../src/utils/constants/Version.ts
   - kind: ethereum/contract
     name: Aquifer
     network: mainnet
@@ -64,7 +64,7 @@ dataSources:
         - Beanstalk
       abis:
         - name: Beanstalk
-          file: ../../subgraph-core/abis/Beanstalk/Beanstalk-Replanted.json
+          file: ../../subgraph-core/abis/Beanstalk/Beanstalk-BIP50.json
       eventHandlers:
         - event: Sunrise(indexed uint256)
           handler: handleSunrise

projects/subgraph-basin/schema.graphql

@@ -1,7 +1,16 @@
-# Custommized base schema from Messari
-# Subgraph Schema: DEX AMM (Extended)
-# Version: 3.0.3
-# See https://github.com/messari/subgraphs/blob/master/docs/Schema.md for details
+# This same entity schema is intended for use across the subgraphs
+type Version @entity {
+  "= 'subgraph'"
+  id: ID!
+  "= 'beanstalk'"
+  subgraphName: String!
+  "Verison number of the subgraph"
+  versionNumber: String!
+  "Address of Beanstalk protocol"
+  protocolAddress: Bytes!
+  "Which blockchain is being indexed, i.e. 'ethereum', 'arbitrum', etc."
+  chain: String!
+}
 
 enum LiquidityEvent {
   # Additions
@@ -17,18 +26,6 @@ enum SwapEvent {
   SHIFT
 }
 
-# This same entity schema is intended for use across the subgraphs
-type Version @entity {
-  "= 'subgraph'"
-  id: ID!
-  "= 'beanstalk'"
-  subgraphName: String!
-  "Verison number of the subgraph"
-  versionNumber: String!
-  "Which blockchain is being indexed, i.e. 'ethereum', 'arbitrum', etc."
-  chain: String!
-}
-
 type Token @entity {
   " Smart contract address of the token "
   id: Bytes!

projects/subgraph-basin/src/BeanstalkHandler.ts

@@ -1,14 +1,15 @@
 import { Address } from "@graphprotocol/graph-ts";
-import { AQUIFER } from "../../subgraph-core/utils/Constants";
 import { Sunrise } from "../generated/Basin-ABIs/Beanstalk";
 import { loadOrCreateAquifer } from "./utils/Aquifer";
 import { checkForSnapshot } from "./utils/Well";
+import { getAquifer } from "../../subgraph-core/constants/RuntimeConstants";
+import { v } from "./utils/constants/Version";
 
 export function handleSunrise(event: Sunrise): void {
   // Right now this is a manual list of aquifers that are checked for deployments and wells updated
   // Keeping this manual is reasonable as each aquifer has to be defined as a datasource in subgraph.yaml
 
-  let aquifer = loadOrCreateAquifer(AQUIFER);
+  let aquifer = loadOrCreateAquifer(getAquifer(v()));
 
   for (let i = 0; i < aquifer.wells.length; i++) {
     checkForSnapshot(Address.fromBytes(aquifer.wells[i]), event.block.timestamp, event.block.number);

projects/subgraph-basin/src/utils/BeanstalkPrice.ts

@@ -1,17 +1,30 @@
 // Unfortunately this file must be copied across the various subgraph projects. This is due to the codegen
-import { Address, BigInt } from "@graphprotocol/graph-ts";
+import { BigDecimal, BigInt } from "@graphprotocol/graph-ts";
 import { BeanstalkPrice } from "../../generated/Basin-ABIs/BeanstalkPrice";
-import { BEANSTALK_PRICE_1, BEANSTALK_PRICE_2, PRICE_2_BLOCK } from "../../../subgraph-core/utils/Constants";
+import { getBeanstalkPriceAddress } from "../../../subgraph-core/constants/RuntimeConstants";
+import { v } from "./constants/Version";
+import { toDecimal } from "../../../subgraph-core/utils/Decimals";
+import { CurvePrice } from "../../generated/Basin-ABIs/CurvePrice";
+import { CURVE_PRICE } from "../../../subgraph-core/constants/raw/BeanstalkEthConstants";
 
 // Gets the BeanstalkPrice contract, bound to the appropriate instance of the contract.
-// Note: Will bind to PRICE_1 even if that contract has not been deployed yet
-// Thus the caller still needs to check for reverts.
 export function getBeanstalkPrice(blockNumber: BigInt): BeanstalkPrice {
-  let contractAddress: Address;
-  if (blockNumber < PRICE_2_BLOCK) {
-    contractAddress = BEANSTALK_PRICE_1;
+  return BeanstalkPrice.bind(getBeanstalkPriceAddress(v(), blockNumber));
+}
+
+export function getBeanPrice(blockNumber: BigInt): BigDecimal | null {
+  let beanstalkPrice = getBeanstalkPrice(blockNumber);
+  let price = beanstalkPrice.try_price();
+  if (!price.reverted) {
+    return toDecimal(price.value.price);
   } else {
-    contractAddress = BEANSTALK_PRICE_2;
+    // Fetch price on Curve
+    let curvePrice = CurvePrice.bind(CURVE_PRICE);
+    let curve = curvePrice.try_getCurve();
+
+    if (curve.reverted) {
+      return null;
+    }
+    return toDecimal(curve.value.price);
   }
-  return BeanstalkPrice.bind(contractAddress);
 }

projects/subgraph-basin/src/utils/Token.ts

@@ -1,10 +1,10 @@
 import { Address, BigDecimal, BigInt, log } from "@graphprotocol/graph-ts";
 import { ERC20 } from "../../generated/Basin-ABIs/ERC20";
 import { Token } from "../../generated/schema";
-import { CurvePrice } from "../../generated/Basin-ABIs/CurvePrice";
-import { BEAN_ERC20, CURVE_PRICE } from "../../../subgraph-core/utils/Constants";
-import { toDecimal, ZERO_BD, ZERO_BI } from "../../../subgraph-core/utils/Decimals";
-import { getBeanstalkPrice } from "./BeanstalkPrice";
+import { BI_MAX, ZERO_BD, ZERO_BI } from "../../../subgraph-core/utils/Decimals";
+import { getBeanPrice } from "./BeanstalkPrice";
+import { getProtocolToken } from "../../../subgraph-core/constants/RuntimeConstants";
+import { v } from "./constants/Version";
 
 export function loadOrCreateToken(tokenAddress: Address): Token {
   let token = Token.load(tokenAddress);
@@ -45,7 +45,7 @@ export function loadToken(tokenAddress: Address): Token {
 }
 
 export function getBeanPriceUDSC(): BigDecimal {
-  let token = loadToken(BEAN_ERC20);
+  let token = loadToken(getProtocolToken(v(), BI_MAX));
   return token.lastPriceUSD;
 }
 
@@ -56,23 +56,12 @@ export function getTokenDecimals(tokenAddress: Address): i32 {
 
 export function updateTokenUSD(tokenAddress: Address, blockNumber: BigInt, beanPrice: BigDecimal = ZERO_BD): void {
   let token = loadToken(tokenAddress);
-  if (tokenAddress == BEAN_ERC20) {
-    // Attempt to use Beanstalk price contract first
-
-    let beanstalkPrice = getBeanstalkPrice(blockNumber);
-    let price = beanstalkPrice.try_price();
-    if (!price.reverted) {
-      token.lastPriceUSD = toDecimal(price.value.price);
-    } else {
-      // Fetch price on Curve
-      let curvePrice = CurvePrice.bind(CURVE_PRICE);
-      let curve = curvePrice.try_getCurve();
-
-      if (curve.reverted) {
-        return;
-      }
-      token.lastPriceUSD = toDecimal(curve.value.price);
+  if (tokenAddress == getProtocolToken(v(), BI_MAX)) {
+    const beanPrice = getBeanPrice(blockNumber);
+    if (beanPrice === null) {
+      return;
     }
+    token.lastPriceUSD = beanPrice;
   } else {
     token.lastPriceUSD = beanPrice.times(getBeanPriceUDSC());
   }

projects/subgraph-basin/src/utils/Well.ts

@@ -2,7 +2,6 @@ import { Address, BigDecimal, BigInt, Bytes, log } from "@graphprotocol/graph-ts
 import { BoreWellWellFunctionStruct } from "../../generated/Basin-ABIs/Aquifer";
 import { Well, WellDailySnapshot, WellFunction, WellHourlySnapshot } from "../../generated/schema";
 import { ERC20 } from "../../generated/Basin-ABIs/ERC20";
-import { BEAN_ERC20 } from "../../../subgraph-core/utils/Constants";
 import { dayFromTimestamp, hourFromTimestamp } from "../../../subgraph-core/utils/Dates";
 import {
   deltaBigDecimalArray,
@@ -15,6 +14,8 @@ import {
   ZERO_BI
 } from "../../../subgraph-core/utils/Decimals";
 import { getTokenDecimals, loadToken, updateTokenUSD } from "./Token";
+import { getProtocolToken } from "../../../subgraph-core/constants/RuntimeConstants";
+import { v } from "./constants/Version";
 
 export function createWell(wellAddress: Address, implementation: Address, inputTokens: Address[]): Well {
   let well = Well.load(wellAddress);
@@ -135,8 +136,9 @@ export function updateWellTokenUSDPrices(wellAddress: Address, blockNumber: BigI
   let well = loadWell(wellAddress);
 
   // Update the BEAN price first as it is the reference for other USD calculations
-  updateTokenUSD(BEAN_ERC20, blockNumber, BigDecimal.fromString("1"));
-  let beanIndex = well.tokens.indexOf(BEAN_ERC20);
+  const beanToken = getProtocolToken(v(), blockNumber);
+  updateTokenUSD(beanToken, blockNumber, BigDecimal.fromString("1"));
+  let beanIndex = well.tokens.indexOf(beanToken);
   // Curretly only supporting USD values for Wells with BEAN as a token.
   if (beanIndex == -1) {
     return;