package soap

import (
	"github.com/masterzen/simplexml/dom"
	"strconv"
)

type HeaderOption struct {
	key   string
	value string
}

func NewHeaderOption(name string, value string) *HeaderOption {
	return &HeaderOption{key: name, value: value}
}

type SoapHeader struct {
	to              string
	replyTo         string
	maxEnvelopeSize string
	timeout         string
	locale          string
	id              string
	action          string
	shellId         string
	resourceURI     string
	options         []HeaderOption
	message         *SoapMessage
}

type HeaderBuilder interface {
	To(string) *SoapHeader
	ReplyTo(string) *SoapHeader
	MaxEnvelopeSize(int) *SoapHeader
	Timeout(string) *SoapHeader
	Locale(string) *SoapHeader
	Id(string) *SoapHeader
	Action(string) *SoapHeader
	ShellId(string) *SoapHeader
	resourceURI(string) *SoapHeader
	AddOption(*HeaderOption) *SoapHeader
	Options([]HeaderOption) *SoapHeader
	Build(*SoapMessage) *SoapMessage
}

func (self *SoapHeader) To(uri string) *SoapHeader {
	self.to = uri
	return self
}

func (self *SoapHeader) ReplyTo(uri string) *SoapHeader {
	self.replyTo = uri
	return self
}

func (self *SoapHeader) MaxEnvelopeSize(size int) *SoapHeader {
	self.maxEnvelopeSize = strconv.Itoa(size)
	return self
}

func (self *SoapHeader) Timeout(timeout string) *SoapHeader {
	self.timeout = timeout
	return self
}

func (self *SoapHeader) Id(id string) *SoapHeader {
	self.id = id
	return self
}

func (self *SoapHeader) Action(action string) *SoapHeader {
	self.action = action
	return self
}

func (self *SoapHeader) Locale(locale string) *SoapHeader {
	self.locale = locale
	return self
}

func (self *SoapHeader) ShellId(shellId string) *SoapHeader {
	self.shellId = shellId
	return self
}

func (self *SoapHeader) ResourceURI(resourceURI string) *SoapHeader {
	self.resourceURI = resourceURI
	return self
}

func (self *SoapHeader) AddOption(option *HeaderOption) *SoapHeader {
	self.options = append(self.options, *option)
	return self
}

func (self *SoapHeader) Options(options []HeaderOption) *SoapHeader {
	self.options = options
	return self
}

func (self *SoapHeader) Build() *SoapMessage {
	header := self.createElement(self.message.envelope, "Header", NS_SOAP_ENV)

	if self.to != "" {
		to := self.createElement(header, "To", NS_ADDRESSING)
		to.SetContent(self.to)
	}

	if self.replyTo != "" {
		replyTo := self.createElement(header, "ReplyTo", NS_ADDRESSING)
		a := self.createMUElement(replyTo, "Address", NS_ADDRESSING, true)
		a.SetContent(self.replyTo)
	}

	if self.maxEnvelopeSize != "" {
		envelope := self.createMUElement(header, "MaxEnvelopeSize", NS_WSMAN_DMTF, true)
		envelope.SetContent(self.maxEnvelopeSize)
	}

	if self.timeout != "" {
		timeout := self.createElement(header, "OperationTimeout", NS_WSMAN_DMTF)
		timeout.SetContent(self.timeout)
	}

	if self.id != "" {
		id := self.createElement(header, "MessageID", NS_ADDRESSING)
		id.SetContent(self.id)
	}

	if self.locale != "" {
		locale := self.createMUElement(header, "Locale", NS_WSMAN_DMTF, false)
		locale.SetAttr("xml:lang", self.locale)
		datalocale := self.createMUElement(header, "DataLocale", NS_WSMAN_MSFT, false)
		datalocale.SetAttr("xml:lang", self.locale)
	}

	if self.action != "" {
		action := self.createMUElement(header, "Action", NS_ADDRESSING, true)
		action.SetContent(self.action)
	}

	if self.shellId != "" {
		selectorSet := self.createElement(header, "SelectorSet", NS_WSMAN_DMTF)
		selector := self.createElement(selectorSet, "Selector", NS_WSMAN_DMTF)
		selector.SetAttr("Name", "ShellId")
		selector.SetContent(self.shellId)
	}

	if self.resourceURI != "" {
		resource := self.createMUElement(header, "ResourceURI", NS_WSMAN_DMTF, true)
		resource.SetContent(self.resourceURI)
	}

	if len(self.options) > 0 {
		set := self.createElement(header, "OptionSet", NS_WSMAN_DMTF)
		for _, option := range self.options {
			e := self.createElement(set, "Option", NS_WSMAN_DMTF)
			e.SetAttr("Name", option.key)
			e.SetContent(option.value)
		}
	}

	return self.message
}

func (self *SoapHeader) createElement(parent *dom.Element, name string, ns dom.Namespace) (element *dom.Element) {
	element = dom.CreateElement(name)
	parent.AddChild(element)
	ns.SetTo(element)
	return
}

func (self *SoapHeader) createMUElement(parent *dom.Element, name string, ns dom.Namespace, mustUnderstand bool) (element *dom.Element) {
	element = self.createElement(parent, name, ns)
	value := "false"
	if mustUnderstand {
		value = "true"
	}
	element.SetAttr("mustUnderstand", value)
	return
}