made selected binding configureable (#144)

crewjam · Aug 20, 2018 · 42c78b1 · 42c78b1
1 parent e8774a4
commit 42c78b1
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/samlsp/middleware.go b/samlsp/middleware.go
@@ -48,6 +48,7 @@ type Middleware struct {
 	TokenMaxAge       time.Duration
 	ClientState       ClientState
 	ClientToken       ClientToken
+	Binding           string
 }
 
 var jwtSigningMethod = jwt.SigningMethodHS256
@@ -110,11 +111,17 @@ func (m *Middleware) RequireAccount(handler http.Handler) http.Handler {
 			panic("don't wrap Middleware with RequireAccount")
 		}
 
-		binding := saml.HTTPRedirectBinding
-		bindingLocation := m.ServiceProvider.GetSSOBindingLocation(binding)
-		if bindingLocation == "" {
-			binding = saml.HTTPPostBinding
+		var binding, bindingLocation string
+		if m.Binding != "" {
+			binding = m.Binding
 			bindingLocation = m.ServiceProvider.GetSSOBindingLocation(binding)
+		} else {
+			binding = saml.HTTPRedirectBinding
+			bindingLocation = m.ServiceProvider.GetSSOBindingLocation(binding)
+			if bindingLocation == "" {
+				binding = saml.HTTPPostBinding
+				bindingLocation = m.ServiceProvider.GetSSOBindingLocation(binding)
+			}
 		}
 
 		req, err := m.ServiceProvider.MakeAuthenticationRequest(bindingLocation)