neel4os
diff --git a/‎identity_provider.go‎
Lines changed: 16 additions & 10 deletions b/‎identity_provider.go‎
Lines changed: 16 additions & 10 deletions
@@ -8,13 +8,13 @@ import (
 	"encoding/base64"
 	"encoding/xml"
 	"fmt"
+	"html/template"
 	"io"
 	"net/http"
 	"net/url"
 	"os"
 	"regexp"
 	"strconv"
-	"text/template"
 	"time"
 
 	"github.com/beevik/etree"
@@ -108,6 +108,7 @@ type IdentityProvider struct {
 	AssertionMaker          AssertionMaker
 	SignatureMethod         string
 	ValidDuration           *time.Duration
+	ResponseFormTemplate    *template.Template
 }
 
 // Metadata returns the metadata structure for this identity provider.
@@ -942,6 +943,16 @@ func (req *IdpAuthnRequest) PostBinding() (IdpAuthnRequestForm, error) {
 	return form, nil
 }
 
+var defaultResponseFormTemplate = template.Must(template.New("saml-post-form").Parse(`<html>` +
+	`<form method="post" action="{{.URL}}" id="SAMLResponseForm">` +
+	`<input type="hidden" name="SAMLResponse" value="{{.SAMLResponse}}" />` +
+	`<input type="hidden" name="RelayState" value="{{.RelayState}}" />` +
+	`<input id="SAMLSubmitButton" type="submit" value="Continue" />` +
+	`</form>` +
+	`<script>document.getElementById('SAMLSubmitButton').style.visibility='hidden';</script>` +
+	`<script>document.getElementById('SAMLResponseForm').submit();</script>` +
+	`</html>`))
+
 // WriteResponse writes the `Response` to the http.ResponseWriter. If
 // `Response` is not already set, it calls MakeResponse to produce it.
 func (req *IdpAuthnRequest) WriteResponse(w http.ResponseWriter) error {
@@ -950,15 +961,10 @@ func (req *IdpAuthnRequest) WriteResponse(w http.ResponseWriter) error {
 		return err
 	}
 
-	tmpl := template.Must(template.New("saml-post-form").Parse(`<html>` +
-		`<form method="post" action="{{.URL}}" id="SAMLResponseForm">` +
-		`<input type="hidden" name="SAMLResponse" value="{{.SAMLResponse}}" />` +
-		`<input type="hidden" name="RelayState" value="{{.RelayState}}" />` +
-		`<input id="SAMLSubmitButton" type="submit" value="Continue" />` +
-		`</form>` +
-		`<script>document.getElementById('SAMLSubmitButton').style.visibility='hidden';</script>` +
-		`<script>document.getElementById('SAMLResponseForm').submit();</script>` +
-		`</html>`))
+	tmpl := req.IDP.ResponseFormTemplate
+	if tmpl == nil {
+		tmpl = defaultResponseFormTemplate
+	}
 
 	buf := bytes.NewBuffer(nil)
 	if err := tmpl.Execute(buf, form); err != nil {