fix: fix issue with return url

flanksource · Nov 20, 2023 · c43986c · c43986c
1 parent 896b41a
commit c43986c
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/components/ory/hooks.ts b/src/components/ory/hooks.ts
@@ -1,6 +1,6 @@
 import { AxiosError } from "axios";
 import Router, { useRouter } from "next/router";
-import { DependencyList, useEffect, useState } from "react";
+import { DependencyList, useEffect, useMemo, useState } from "react";
 import ory from "./sdk";
 
 export const HandleError = (
@@ -173,9 +173,18 @@ export function useCreateLogoutHandler(deps?: DependencyList) {
   const handleError = HandleError();
   const { push } = useRouter();
 
+  const returnTo = useMemo(() => {
+    if (typeof window !== "undefined") {
+      return window.location.pathname + window.location.search;
+    }
+    return undefined;
+  }, []);
+
   useEffect(() => {
     ory
-      .createBrowserLogoutFlow()
+      .createBrowserLogoutFlow({
+        returnTo: returnTo
+      })
       .then(({ data }) => {
         setLogoutToken(data.logout_token);
       })
@@ -187,7 +196,8 @@ export function useCreateLogoutHandler(deps?: DependencyList) {
     if (logoutToken) {
       ory
         .updateLogoutFlow({
-          token: logoutToken
+          token: logoutToken,
+          returnTo: returnTo
         })
         .then(() => push("/"))
         .catch((error) => {