sso-proxy: re-factor to help tests

buzzfeed · Oct 10, 2018 · 774d874 · 774d874
1 parent 6fe9d49
commit 774d874
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/internal/proxy/providers/sso.go b/internal/proxy/providers/sso.go
@@ -106,7 +106,16 @@ func (p *SSOProvider) Redeem(redirectURL, code string) (*SessionState, error) {
 	params.Add("code", code)
 	params.Add("grant_type", "authorization_code")
 
-	req, err := newRequest("POST", p.ProxyRedeemURL.String(), bytes.NewBufferString(params.Encode()))
+	// If the hosts dont match use the one in the function call
+	RedeemURL * url.URL
+	redirectURLhost, _ := url.Parse(redirectURL)
+	if redirectURLhost.Host != p.ProxyRedeemURL.Host {
+		RedeemURL = redirectURL
+	} else {
+		RedeemURL = p.ProxyRedeemURL
+	}
+
+	req, err := newRequest("POST", RedeemURL.String(), bytes.NewBufferString(params.Encode()))
 	if err != nil {
 		return nil, err
 	}