fix: disable selection of the same tokens

lifinance · Jun 7, 2022 · 9fb33e9 · 9fb33e9
1 parent 70a1691
commit 9fb33e9
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/packages/widget/src/components/TokenList/TokenList.tsx b/packages/widget/src/components/TokenList/TokenList.tsx
@@ -21,7 +21,7 @@ export const TokenList: FC<TokenListProps> = ({
 }) => {
   const { t } = useTranslation();
   const { account } = useWallet();
-  const { setValue } = useFormContext();
+  const { setValue, getValues } = useFormContext();
   const [selectedChainId] = useWatch({
     name: [SwapFormKeyHelper.getChainKey(formType)],
   });
@@ -30,7 +30,7 @@ export const TokenList: FC<TokenListProps> = ({
     250,
   );
 
-  const { tokens, isLoading, isFetching } = useTokenBalances(selectedChainId);
+  const { tokens, isLoading } = useTokenBalances(selectedChainId);
 
   const chainTokens = useMemo(() => {
     let chainTokens = tokens ?? [];
@@ -67,9 +67,16 @@ export const TokenList: FC<TokenListProps> = ({
     (tokenAddress: string) => {
       setValue(SwapFormKeyHelper.getTokenKey(formType), tokenAddress);
       setValue(SwapFormKeyHelper.getAmountKey(formType), '');
+      const oppositeFormType = formType === 'from' ? 'to' : 'from';
+      const [selectedOppositeToken] = getValues([
+        SwapFormKeyHelper.getTokenKey(oppositeFormType),
+      ]);
+      if (selectedOppositeToken === tokenAddress) {
+        setValue(SwapFormKeyHelper.getTokenKey(oppositeFormType), '');
+      }
       onClick?.();
     },
-    [formType, onClick, setValue],
+    [formType, getValues, onClick, setValue],
   );
 
   return (