Skip to content

Commit 153cd03

Browse files
committed
feat: improve usePrompt logic
1 parent b93feca commit 153cd03

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

src/Shared/Hooks/UsePrompt/UsePrompt.tsx

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,13 +42,18 @@ const usePrompt = ({ shouldPrompt, message = DEFAULT_ROUTE_PROMPT_MESSAGE }: Use
4242
}
4343
}, [handlePageLeave])
4444

45-
const blocker = useBlocker(() => shouldPrompt)
45+
const blocker = useBlocker(shouldPrompt)
4646

4747
useEffect(() => {
4848
if (!blocker || blocker.state !== 'blocked') {
4949
return
5050
}
5151

52+
if (!shouldPrompt) {
53+
blocker.proceed()
54+
return
55+
}
56+
5257
// eslint-disable-next-line no-alert
5358
const proceed = window.confirm(message)
5459

@@ -57,7 +62,7 @@ const usePrompt = ({ shouldPrompt, message = DEFAULT_ROUTE_PROMPT_MESSAGE }: Use
5762
} else {
5863
blocker.reset()
5964
}
60-
}, [blocker, message])
65+
}, [blocker, message, shouldPrompt])
6166
}
6267

6368
export default usePrompt

0 commit comments

Comments
 (0)