fix: fix issue with return url

flanksource · Nov 17, 2023 · 47aeba7 · 47aeba7
1 parent e68010c
commit 47aeba7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/components/ory/hooks.ts b/src/components/ory/hooks.ts
@@ -172,10 +172,13 @@ export function useCreateLogoutHandler(deps?: DependencyList) {
   const [logoutToken, setLogoutToken] = useState<string>("");
   const handleError = HandleError();
   const { push } = useRouter();
+  const returnTo = window.location.pathname + window.location.search;
 
   useEffect(() => {
     ory
-      .createBrowserLogoutFlow()
+      .createBrowserLogoutFlow({
+        returnTo: returnTo
+      })
       .then(({ data }) => {
         setLogoutToken(data.logout_token);
       })
@@ -187,7 +190,8 @@ export function useCreateLogoutHandler(deps?: DependencyList) {
     if (logoutToken) {
       ory
         .updateLogoutFlow({
-          token: logoutToken
+          token: logoutToken,
+          returnTo: returnTo
         })
         .then(() => push("/"))
         .catch((error) => {