From c4951ed359c72529445e9a9558fce6782dbc8f98 Mon Sep 17 00:00:00 2001 From: Gustavo Maronato Date: Sat, 9 Mar 2024 14:09:14 -0500 Subject: [PATCH] use current host if in prod --- internal/server/oidc/handler.go | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/internal/server/oidc/handler.go b/internal/server/oidc/handler.go index 6ff65ee..36f4a59 100644 --- a/internal/server/oidc/handler.go +++ b/internal/server/oidc/handler.go @@ -23,6 +23,7 @@ type OIDCHandler struct { oidc *oidcservice.OIDCService user *userservice.UserService uiHost string + prod bool } func NewOIDCHandler(cfg *config.Config, oidc *oidcservice.OIDCService, user *userservice.UserService) *OIDCHandler { @@ -32,6 +33,7 @@ func NewOIDCHandler(cfg *config.Config, oidc *oidcservice.OIDCService, user *use oidc: oidc, user: user, uiHost: uiHost, + prod: cfg.Prod, } } @@ -126,5 +128,11 @@ func (h *OIDCHandler) Callback(w http.ResponseWriter, r *http.Request) { Host: h.uiHost, Path: "/", } - http.Redirect(w, r, uiURL.String(), http.StatusFound) + + redirectURL := "/" + if !h.prod { + redirectURL = uiURL.String() + } + + http.Redirect(w, r, redirectURL, http.StatusFound) }