fix: allow buttons in edit and preview mode to reload when clicked

opengovsg · Feb 20, 2024 · ef3e717 · ef3e717
1 parent 36b89ae
commit ef3e717
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/frontend/src/features/admin-form/create/end-page/PaymentEndPageBlock.tsx b/frontend/src/features/admin-form/create/end-page/PaymentEndPageBlock.tsx
@@ -44,6 +44,10 @@ export const PaymentEndPageBlock = ({
     return 'You have successfully submitted your response.'
   }, [form?.title])
 
+  const handleButtonClick = () => {
+    window.location.reload()
+  }
+
   if (!isPaymentEnabled) {
     return <></>
   }
@@ -68,6 +72,7 @@ export const PaymentEndPageBlock = ({
         products={paymentProducts}
         name={'Product/ Service'}
         paymentDate={new Date()}
+        onClick={handleButtonClick}
       />
     </Box>
   )

diff --git a/frontend/src/features/public-form/components/FormEndPage/components/EndPageBlock.tsx b/frontend/src/features/public-form/components/FormEndPage/components/EndPageBlock.tsx
@@ -1,5 +1,5 @@
 import { useEffect, useMemo, useRef } from 'react'
-import { Box, Container, Text, VisuallyHidden } from '@chakra-ui/react'
+import { Box, Text, VisuallyHidden } from '@chakra-ui/react'
 import { format } from 'date-fns'
 
 import { FormColorTheme, FormDto } from '~shared/types/form'

diff --git a/...eatures/public-form/components/FormPaymentPage/stripe/components/DownloadReceiptBlock.tsx b/...eatures/public-form/components/FormPaymentPage/stripe/components/DownloadReceiptBlock.tsx
@@ -21,6 +21,7 @@ type DownloadReceiptBlockProps = {
   name: string
   paymentDate: Date | null
   endPage: FormDto['endPage']
+  onClick?: () => void
 }
 
 const PaymentDetailsRow = ({
@@ -111,6 +112,7 @@ export const DownloadReceiptBlock = ({
   products,
   paymentDate,
   endPage,
+  onClick,
 }: DownloadReceiptBlockProps) => {
   const toast = useToast({ status: 'success', isClosable: true })
 
@@ -194,7 +196,7 @@ export const DownloadReceiptBlock = ({
           <Button
             w="100%"
             leftIcon={<BiDownload fontSize="1.5rem" />}
-            onClick={handleInvoiceClick}
+            onClick={onClick || handleInvoiceClick}
           >
             Save proof of payment
           </Button>