diff --git a/site/src/pages/continue-page.tsx b/site/src/pages/continue-page.tsx
index e07cae34..1101b9b7 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 ;
}
- 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 a84bdeee..ff8e4f7c 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 ;
@@ -65,6 +65,10 @@ export const LoginPage = () => {
color: "green",
});
setTimeout(() => {
+ if (disableContinue) {
+ window.location.replace(redirectUri!);
+ return;
+ }
window.location.replace(`/continue?redirect_uri=${redirectUri}`);
});
},