From 9d20324c76887335cbaee01e06e17caa618d4e27 Mon Sep 17 00:00:00 2001
From: Majorfi <major@ycorpo.com>
Date: Wed, 17 Jan 2024 12:40:05 +0100
Subject: [PATCH] build: bump

---
 package.json                             |  2 +-
 src/contexts/useYearn.tsx                | 11 ++++++++++-
 src/hooks/useYearnTokens.ts              |  7 +++++--
 src/utils/schemas/yDaemonTokensSchema.ts |  3 ++-
 4 files changed, 18 insertions(+), 5 deletions(-)

diff --git a/package.json b/package.json
index 59ac1876..d37e7786 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@yearn-finance/web-lib",
-  "version": "3.0.115",
+  "version": "3.0.116",
   "main": "./dist/index.js",
   "types": "./dist/index.d.js",
   "files": [
diff --git a/src/contexts/useYearn.tsx b/src/contexts/useYearn.tsx
index 32a7f312..d354ec31 100755
--- a/src/contexts/useYearn.tsx
+++ b/src/contexts/useYearn.tsx
@@ -1,5 +1,6 @@
-import {createContext, memo, useContext} from 'react';
+import {createContext, memo, useContext, useEffect} from 'react';
 import {deserialize, serialize} from 'wagmi';
+import useWallet from '@builtbymom/web3/contexts/useWallet';
 import {useLocalStorageValue} from '@react-hookz/web';
 import {toAddress} from '@yearn-finance/web-lib/utils/address';
 
@@ -67,6 +68,7 @@ const YearnContext = createContext<TYearnContext>({
 });
 
 export const YearnContextApp = memo(function YearnContextApp({children}: {children: ReactElement}): ReactElement {
+	const {refresh} = useWallet();
 	const {value: maxLoss, set: set_maxLoss} = useLocalStorageValue<bigint>('yearn.fi/max-loss', {
 		defaultValue: DEFAULT_MAX_LOSS,
 		parse: (str: string, fallback: bigint): bigint => (str ? deserialize(str) : fallback),
@@ -90,6 +92,13 @@ export const YearnContextApp = memo(function YearnContextApp({children}: {childr
 	const earned = useYearnEarned();
 	const {vaults, vaultsMigrations, vaultsRetired, isLoading, mutate} = useYearnVaults();
 
+	useEffect(() => {
+		const tokensToRefresh = Object.values(tokens).map(token => {
+			return {token: toAddress(token?.address)};
+		});
+		refresh(tokensToRefresh);
+	}, [tokens, refresh]);
+
 	return (
 		<YearnContext.Provider
 			value={{
diff --git a/src/hooks/useYearnTokens.ts b/src/hooks/useYearnTokens.ts
index 30a88d87..6330b6ad 100644
--- a/src/hooks/useYearnTokens.ts
+++ b/src/hooks/useYearnTokens.ts
@@ -19,10 +19,13 @@ function useYearnTokens(): TYDaemonTokens {
 			return {};
 		}
 		const _tokens: TYDaemonTokens = {};
-		for (const [, tokensData] of Object.entries(tokens)) {
+		for (const [chainID, tokensData] of Object.entries(tokens)) {
 			for (const [tokenAddress, token] of Object.entries(tokensData)) {
 				if (token) {
-					_tokens[toAddress(tokenAddress)] = token;
+					_tokens[toAddress(tokenAddress)] = {
+						...token,
+						chainID: Number(chainID)
+					};
 				}
 			}
 		}
diff --git a/src/utils/schemas/yDaemonTokensSchema.ts b/src/utils/schemas/yDaemonTokensSchema.ts
index dfa25b43..ea5523da 100644
--- a/src/utils/schemas/yDaemonTokensSchema.ts
+++ b/src/utils/schemas/yDaemonTokensSchema.ts
@@ -8,7 +8,8 @@ export const yDaemonTokenSchema = z.object({
 	symbol: z.string(),
 	decimals: z.number(),
 	isVault: z.boolean(),
-	underlyingTokens: z.array(addressSchema).optional()
+	underlyingTokens: z.array(addressSchema).optional(),
+	chainID: z.number().optional()
 });
 
 export const yDaemonTokensSchema = z.record(addressSchema, yDaemonTokenSchema);