Skip to content

Commit fe3ba9b

Browse files
committed
Add underlying token
1 parent 59f08d8 commit fe3ba9b

File tree

3 files changed

+10
-0
lines changed

3 files changed

+10
-0
lines changed

schema.graphql

+3
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,9 @@ type BeefyVault @entity {
4444
"Moo token address"
4545
sharesToken: Token!
4646

47+
"Underlying token address"
48+
underlyingToken: Token!
49+
4750
"The vault's strategy"
4851
strategy: BeefyStrategy!
4952

src/entity/vault.ts

+1
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ export function getBeefyVault(vaultAddress: Bytes): BeefyVault {
2424
if (!vault) {
2525
vault = new BeefyVault(vaultAddress)
2626
vault.sharesToken = ADDRESS_ZERO
27+
vault.underlyingToken = ADDRESS_ZERO
2728
vault.strategy = ADDRESS_ZERO
2829
vault.isInitialized = false
2930
vault.initializedAtBlockNumber = ZERO_BI

src/vault-lifecycle.ts

+6
Original file line numberDiff line numberDiff line change
@@ -100,9 +100,15 @@ export function handleStrategyInitialized(event: ethereum.Event): void {
100100
*/
101101
function fetchInitialVaultData(vault: BeefyVault): BeefyVault {
102102
const vaultAddress = Address.fromBytes(vault.id)
103+
const vaultContract = BeefyVaultV7Contract.bind(vaultAddress)
104+
105+
const underlyingTokenAddress = vaultContract.want()
106+
103107
const sharesToken = getTokenAndInitIfNeeded(vaultAddress)
108+
const underlyingToken = getTokenAndInitIfNeeded(underlyingTokenAddress)
104109

105110
vault.sharesToken = sharesToken.id
111+
vault.underlyingToken = underlyingToken.id
106112
vault.lifecycle = BEEFY_VAULT_LIFECYCLE_RUNNING
107113

108114
return vault

0 commit comments

Comments
 (0)