fix: don't reject when signing (#2666)

* fix: don't reject when signing

* fix: comment

src/components/tx-flow/index.tsx

@@ -19,9 +19,11 @@ export const TxModalContext = createContext<TxModalContextType>({
   setFullWidth: noop,
 })
 
+const shouldClose = () => confirm('Closing this window will discard your current progress.')
+
 export const TxModalProvider = ({ children }: { children: ReactNode }): ReactElement => {
   const [txFlow, setFlow] = useState<TxModalContextType['txFlow']>(undefined)
-  const [shouldWarn, setShouldWarn] = useState<boolean>(true)
+  const [shouldWarn, setShouldWarn] = useState<boolean>(false)
   const [, setOnClose] = useState<Parameters<TxModalContextType['setTxFlow']>[1]>(noop)
   const [fullWidth, setFullWidth] = useState<boolean>(false)
   const pathname = usePathname()
@@ -42,29 +44,27 @@ export const TxModalProvider = ({ children }: { children: ReactNode }): ReactEle
       return
     }
 
-    const ok = confirm('Closing this window will discard your current progress.')
-    if (!ok) return
+    if (!shouldClose()) return
 
-    // Reject if the flow is closed
     txDispatch(TxEvent.USER_QUIT, {})
 
     handleModalClose()
   }, [shouldWarn, handleModalClose])
 
   const setTxFlow = useCallback(
-    (newTxFlow: TxModalContextType['txFlow'], onClose?: () => void, shouldWarn?: boolean) => {
-      setFlow((prevFlow) => {
-        // Reject if a flow is open and the user changes to a different one
-        if (prevFlow && prevFlow !== newTxFlow && newTxFlow?.type !== SuccessScreen) {
-          txDispatch(TxEvent.USER_QUIT, {})
-        }
-
-        return newTxFlow
-      })
+    (newTxFlow: TxModalContextType['txFlow'], onClose?: () => void, newShouldWarn?: boolean) => {
+      // If flow is open and user opens a different one, show confirmation dialog if required
+      if (txFlow && newTxFlow && newTxFlow?.type !== SuccessScreen && shouldWarn) {
+        if (!shouldClose()) return
+
+        txDispatch(TxEvent.USER_QUIT, {})
+      }
+
+      setFlow(newTxFlow)
       setOnClose(() => onClose ?? noop)
-      setShouldWarn(shouldWarn ?? true)
+      setShouldWarn(newShouldWarn ?? true)
     },
-    [],
+    [txFlow, shouldWarn],
   )
 
   // Show the confirmation dialog if user navigates