diff --git a/src/components/tx-flow/common/TxButton.tsx b/src/components/tx-flow/common/TxButton.tsx index 032437807c..c0f03ac000 100644 --- a/src/components/tx-flow/common/TxButton.tsx +++ b/src/components/tx-flow/common/TxButton.tsx @@ -6,6 +6,8 @@ import { useTxBuilderApp } from '@/hooks/safe-apps/useTxBuilderApp' import { AppRoutes } from '@/config/routes' import Track from '@/components/common/Track' import { MODALS_EVENTS } from '@/services/analytics' +import { useContext } from 'react' +import { TxModalContext } from '..' const buttonSx = { height: '58px', @@ -24,11 +26,15 @@ export const SendTokensButton = ({ onClick, sx }: { onClick: () => void; sx?: Bu export const SendNFTsButton = () => { const router = useRouter() + const { setTxFlow } = useContext(TxModalContext) + + const isNftPage = router.pathname === AppRoutes.balances.nfts + const onClick = isNftPage ? () => setTxFlow(undefined) : undefined return ( - @@ -38,12 +44,18 @@ export const SendNFTsButton = () => { export const TxBuilderButton = () => { const txBuilder = useTxBuilderApp() + const router = useRouter() + const { setTxFlow } = useContext(TxModalContext) + if (!txBuilder?.app) return null + const isTxBuilder = typeof txBuilder.link.query === 'object' && router.query.appUrl === txBuilder.link.query?.appUrl + const onClick = isTxBuilder ? () => setTxFlow(undefined) : undefined + return ( -