From 0ad0dae4a9d3f4c4a81a703b965ce743364283c1 Mon Sep 17 00:00:00 2001
From: Ronoaldo JLP <ronoaldo@gmail.com>
Date: Tue, 5 Jul 2016 20:02:00 +0000
Subject: [PATCH] Allow clients to add SOAPHeader in SOAPClient.

In order to allow the SOAPClient to send a
SOAPHeader without modification of generated code,
a new method SetHeader was added to both the
SOAPClient and to the generated operation.
The header, once set, is preserved between
invocations of other methods in the same instance.

This allows client code to call SetHeader()
with the desired header, including nil to remove
it from future calls in the same instance.

The SOAPHeader usually holds authentication info,
and we can benefit from reusing already set
credentials between invocations.
---
 gowsdl_test.go     | 16 ++++++++++++++++
 operations_tmpl.go |  4 ++++
 soap_tmpl.go       | 13 ++++++++++---
 3 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/gowsdl_test.go b/gowsdl_test.go
index fec0c68..8eb85ef 100644
--- a/gowsdl_test.go
+++ b/gowsdl_test.go
@@ -61,3 +61,19 @@ func TestVboxGeneratesWithoutSyntaxErrors(t *testing.T) {
 		}
 	}
 }
+
+func TestSOAPHeaderGeneratesWithoutErrors(t *testing.T) {
+	g := GoWSDL{
+		file:         "fixtures/ferry.wsdl",
+		pkg:          "myservice",
+		makePublicFn: makePublic,
+	}
+
+	resp, err := g.Start()
+	if err != nil {
+		t.Error(err)
+	}
+	if !strings.Contains(string(resp["operations"]), "SetHeader") {
+		t.Error("SetHeader method should be generated in the service operation")
+	}
+}
diff --git a/operations_tmpl.go b/operations_tmpl.go
index 415ae4f..f1f48dd 100644
--- a/operations_tmpl.go
+++ b/operations_tmpl.go
@@ -22,6 +22,10 @@ var opsTmpl = `
 		}
 	}
 
+	func (service *{{$portType}}) SetHeader(header interface{}) {
+		service.client.SetHeader(header)
+	}
+
 	{{range .Operations}}
 		{{$faults := len .Faults}}
 		{{$requestType := findType .Input.Message | replaceReservedWords | makePublic}}
diff --git a/soap_tmpl.go b/soap_tmpl.go
index 5a91a3a..63e349e 100644
--- a/soap_tmpl.go
+++ b/soap_tmpl.go
@@ -13,7 +13,7 @@ func dialTimeout(network, addr string) (net.Conn, error) {
 
 type SOAPEnvelope struct {
 	XMLName xml.Name ` + "`" + `xml:"http://schemas.xmlsoap.org/soap/envelope/ Envelope"` + "`" + `
-
+	Header *SOAPHeader
 	Body SOAPBody
 }
 
@@ -48,6 +48,7 @@ type SOAPClient struct {
 	url string
 	tls bool
 	auth *BasicAuth
+	header interface{}
 }
 
 func (b *SOAPBody) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
@@ -112,9 +113,15 @@ func NewSOAPClient(url string, tls bool, auth *BasicAuth) *SOAPClient {
 	}
 }
 
+func (s *SOAPClient) SetHeader(header interface{}) {
+	s.header = header
+}
+
 func (s *SOAPClient) Call(soapAction string, request, response interface{}) error {
-	envelope := SOAPEnvelope{
-	//Header:        SoapHeader{},
+	envelope := SOAPEnvelope{}
+
+	if s.header != nil {
+		envelope.Header = &SOAPHeader{Header: s.header}
 	}
 
 	envelope.Body.Content = request