fix(suite): reset amount, send max and draft on token change

trezor · Dec 18, 2024 · 058075e · 058075e
1 parent 280c7d1
commit 058075e
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/packages/suite/src/hooks/wallet/useSendForm.ts b/packages/suite/src/hooks/wallet/useSendForm.ts
@@ -226,6 +226,10 @@ export const useSendForm = (props: UseSendFormProps): SendContextValues => {
         setState(getStateFromProps(props)); // resetting state will trigger "loadDraft" useEffect block, which will reset FormState to default
     }, [dispatch, props, setComposedLevels]);
 
+    const resetDraft = useCallback(() => {
+        dispatch(removeSendFormDraftThunk());
+    }, [dispatch]);
+
     // declare useSendFormImport, sub-hook of useSendForm
     const { importTransaction, validateImportedTransaction } = useSendFormImport({
         network: state.network,
@@ -393,6 +397,7 @@ export const useSendForm = (props: UseSendFormProps): SendContextValues => {
         loadTransaction,
         signTransaction: sign,
         setDraftSaveRequest,
+        resetDraft,
         utxoSelection,
         ...sendFormUtils,
         ...sendFormOutputs,

diff --git a/packages/suite/src/types/wallet/sendForm.ts b/packages/suite/src/types/wallet/sendForm.ts
@@ -76,6 +76,7 @@ export type SendContextValues<TFormValues extends FormState = FormState> =
             outputs: Partial<Output & { id: string }>[]; // useFieldArray fields
             updateContext: (value: Partial<UseSendFormState>) => void;
             resetContext: () => void;
+            resetDraft: () => void;
             composeTransaction: (field?: FieldPath<TFormValues>) => void;
             loadTransaction: () => Promise<void>;
             signTransaction: () => void;

diff --git a/packages/suite/src/views/wallet/send/Outputs/TokenSelect/TokenSelect.tsx b/packages/suite/src/views/wallet/send/Outputs/TokenSelect/TokenSelect.tsx
@@ -124,6 +124,8 @@ export const TokenSelect = ({ outputId }: TokenSelectProps) => {
         watch,
         setValue,
         setDraftSaveRequest,
+        resetDraft,
+        setMax,
     } = useSendFormContext();
 
     const dispatch = useDispatch();
@@ -215,9 +217,20 @@ export const TokenSelect = ({ outputId }: TokenSelectProps) => {
             token => token.contract === selectedAsset.contractAddress,
         );
 
+        resetDraft();
+
         setValue(tokenInputName, newlySelectedToken?.contract || account.symbol, {
             shouldDirty: true,
         });
+        setValue(amountInputName, '', {
+            shouldDirty: true,
+        });
+
+        const isSetMaxActive = getDefaultValue('setMaxOutputId') === outputId;
+
+        if (isSetMaxActive) {
+            setMax(outputId, isSetMaxActive);
+        }
 
         setIsTokensModalActive(false);