From f969ad3cfe581bc98318af4668809643da7b048b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20FAMIBELLE-PRONZOLA?= <c.pronzola@tuta.io>
Date: Tue, 23 Jan 2024 05:10:54 +0400
Subject: [PATCH] Wrap useSearchParams into Suspense

---
 app/pwopose/page.js | 37 +++++++++++++++++++++++++++----------
 1 file changed, 27 insertions(+), 10 deletions(-)

diff --git a/app/pwopose/page.js b/app/pwopose/page.js
index 1a8d3a9..9ffd5d6 100644
--- a/app/pwopose/page.js
+++ b/app/pwopose/page.js
@@ -1,6 +1,7 @@
 'use client'
 
-import {useState, useEffect, forwardRef} from 'react'
+import PropTypes from 'prop-types'
+import {useState, useEffect, forwardRef, Suspense} from 'react'
 import {useSession} from 'next-auth/react'
 import MuiAlert from '@mui/material/Alert'
 import Snackbar from '@mui/material/Snackbar'
@@ -19,9 +20,26 @@ const Alert = forwardRef(function Alert(props, ref) {
   return <MuiAlert ref={ref} elevation={6} variant='filled' {...props} />
 })
 
-export default function Pwopose() {
+function HandleSession({session}) {
   const params = useSearchParams()
   const {code} = params
+
+  if (!session && !code) {
+    return (
+      <Koneksyon
+        chimen='/pwopose'
+      />
+    )
+  }
+
+  if (!session && code) {
+    return (
+      <NewPassword code={code} />
+    )
+  }
+}
+
+export default function Pwopose() {
   const {data: session} = useSession()
   const [localUsername, setLocalUsername] = useState(null)
   const [username, setUsername] = useState(null)
@@ -75,15 +93,10 @@ export default function Pwopose() {
   return (
     <Box sx={{display: 'flex', flexDirection: 'column', minHeight: '100vh'}}>
       <Box sx={{flexGrow: 1, marginTop: 1, marginBottom: 10}}>
-        {!session && !code && (
-          <Koneksyon
-            chimen='/pwopose'
-          />
-        )}
+        <Suspense>
+          <HandleSession session={session} />
+        </Suspense>
 
-        {!session && code && (
-          <NewPassword code={code} />
-        )}
         {session && session.user && (
           <>
             <Dekoneksyon position='absolute' top={95} left={5} chimen='/pwopose' />
@@ -108,3 +121,7 @@ export default function Pwopose() {
     </Box>
   )
 }
+
+HandleSession.propTypes = {
+  session: PropTypes.object.isRequired
+}