Add WithRedirectURL option (#177)

palantir · Aug 29, 2022 · 409e16b · 409e16b
1 parent 16ec487
commit 409e16b
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/oauth2/handler.go b/oauth2/handler.go
@@ -125,10 +125,20 @@ func OnLogin(c LoginCallback) Param {
 	}
 }
 
+// WithRedirectURL sets a static redirect URL. By default, the redirect URL is
+// generated using the request path, the Host header, and the ForceTLS option.
+func WithRedirectURL(uri string) Param {
+	return func(h *handler) {
+		h.config.RedirectURL = uri
+	}
+}
+
 func (h *handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	// copy config for modification
 	conf := *h.config
-	conf.RedirectURL = redirectURL(r, h.forceTLS)
+	if conf.RedirectURL == "" {
+		conf.RedirectURL = redirectURL(r, h.forceTLS)
+	}
 
 	// if the provider returned an error, abort the processes
 	if r.FormValue(queryError) != "" {