From 81d25061dfd4ffe44cac13a1a9e237b93d72b816 Mon Sep 17 00:00:00 2001
From: Stavros <steveiliop56@gmail.com>
Date: Sat, 25 Jan 2025 21:11:09 +0200
Subject: [PATCH] refactor: move disable continue logic in login screen

---
 site/src/pages/continue-page.tsx | 6 +-----
 site/src/pages/login-page.tsx    | 6 +++++-
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/site/src/pages/continue-page.tsx b/site/src/pages/continue-page.tsx
index e07cae3..1101b9b 100644
--- a/site/src/pages/continue-page.tsx
+++ b/site/src/pages/continue-page.tsx
@@ -9,16 +9,12 @@ export const ContinuePage = () => {
   const params = new URLSearchParams(queryString);
   const redirectUri = params.get("redirect_uri");
 
-  const { isLoggedIn, disableContinue } = useUserContext();
+  const { isLoggedIn } = useUserContext();
 
   if (!isLoggedIn) {
     return <Navigate to="/login" />;
   }
 
-  if (disableContinue && redirectUri !== "null") {
-    window.location.replace(redirectUri!);
-  }
-
   const redirect = () => {
     notifications.show({
       title: "Redirecting",
diff --git a/site/src/pages/login-page.tsx b/site/src/pages/login-page.tsx
index a84bdee..ff8e4f7 100644
--- a/site/src/pages/login-page.tsx
+++ b/site/src/pages/login-page.tsx
@@ -25,7 +25,7 @@ export const LoginPage = () => {
   const params = new URLSearchParams(queryString);
   const redirectUri = params.get("redirect_uri");
 
-  const { isLoggedIn, configuredProviders } = useUserContext();
+  const { isLoggedIn, configuredProviders, disableContinue } = useUserContext();
 
   if (isLoggedIn) {
     return <Navigate to="/logout" />;
@@ -65,6 +65,10 @@ export const LoginPage = () => {
         color: "green",
       });
       setTimeout(() => {
+        if (disableContinue) {
+          window.location.replace(redirectUri!);
+          return;
+        }
         window.location.replace(`/continue?redirect_uri=${redirectUri}`);
       });
     },