From d6e9dca52fd0295dc4dd99802d558a4f8ce2b7f4 Mon Sep 17 00:00:00 2001
From: eli-d <64763513+eli-d@users.noreply.github.com>
Date: Tue, 17 Dec 2024 16:01:11 +1030
Subject: [PATCH] invalidate all stake token amounts when changing either token

---
 web/src/stores/useStakeStore.ts | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/web/src/stores/useStakeStore.ts b/web/src/stores/useStakeStore.ts
index 81786970..be89e8a0 100644
--- a/web/src/stores/useStakeStore.ts
+++ b/web/src/stores/useStakeStore.ts
@@ -75,10 +75,26 @@ export const useStakeStore = create<StakeStore>((set) => ({
   setMultiSingleToken: (multiSingleToken) => set({ multiSingleToken }),
 
   token0: EmptyToken,
-  setToken0: (token0) => set({ token0 }),
+  // changing token invalidates all amounts
+  setToken0: (token0) =>
+    set({
+      token0,
+      token0Amount: "0",
+      token0AmountRaw: "0",
+      token1Amount: "0",
+      token1AmountRaw: "0",
+    }),
 
   token1: EmptyToken,
-  setToken1: (token1) => set({ token1 }),
+  // changing token invalidates all amounts
+  setToken1: (token1) =>
+    set({
+      token1,
+      token0Amount: "0",
+      token0AmountRaw: "0",
+      token1Amount: "0",
+      token1AmountRaw: "0",
+    }),
 
   token0Amount: "",
   token1Amount: "",