change input amount when selected token changes

massalabs · Jun 20, 2024 · 5f629fc · 5f629fc
1 parent c4cce52
commit 5f629fc
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/store/tokenStore.ts b/src/store/tokenStore.ts
@@ -135,6 +135,21 @@ export const useTokenStore = create<TokenStoreState>((set, get) => ({
     if (!selectedToken) {
       set({ selectedToken });
     }
+    const currentSelectedToken = get().selectedToken;
+    const inputAmount = useOperationStore.getState().inputAmount;
+    if (
+      inputAmount &&
+      currentSelectedToken?.decimals !== selectedToken?.decimals
+    ) {
+      const decsDiff = selectedToken!.decimals - currentSelectedToken!.decimals;
+      let newAmount: bigint;
+      if (decsDiff > 0) {
+        newAmount = inputAmount * 10n ** BigInt(decsDiff);
+      } else {
+        newAmount = inputAmount / 10n ** BigInt(-decsDiff);
+      }
+      useOperationStore.setState({ inputAmount: newAmount });
+    }
     // if selected token is not in the list, set first selectable token
     // selectable is the token supported on the current evm chain selected
     const selectable = get().getTokens();