xml.go

package gokalkan

import (
	"strings"

	"github.com/beevik/etree"
)

// XMLSignOptions - опции подписи XML, аналог полей XmlSignRequest в NCANode.
type XMLSignOptions struct {
	// ReferenceURI - id тэга, который нужно подписать (Reference URI="#id").
	// Пусто - подписывается документ целиком (Reference URI="").
	ReferenceURI string
	// ClearSignatures - удалить существующие ds:Signature перед подписью.
	ClearSignatures bool
	// TrimXML - удалить текстовые узлы, состоящие только из пробельных
	// символов, перед подписью (влияет на канонизированные байты).
	TrimXML bool
}

// SignXMLWithOptions - расширенная версия SignXML с поддержкой опций,
// аналогичных NCANode (/xml/sign): referenceUri, clearSignatures, trimXml.
// Для нескольких подписантов вызывайте последовательно (загружая между
// вызовами очередной ключ через LoadKeyStore) - каждый новый вызов
// добавляет ещё один ds:Signature, не трогая предыдущие.
func (cli *Client) SignXMLWithOptions(xmlData string, opts XMLSignOptions) (string, error) {
	var err error

	if opts.ClearSignatures {
		xmlData, err = clearXMLSignatures(xmlData)
		if err != nil {
			return "", err
		}
	}

	if opts.TrimXML {
		xmlData, err = trimXMLWhitespace(xmlData)
		if err != nil {
			return "", err
		}
	}

	return cli.kc.SignXML(xmlData, "", 0, opts.ReferenceURI, "", "")
}

// clearXMLSignatures удаляет все ds:Signature элементы документа (на любом
// уровне вложенности, как в NCANode XmlService.read(removeSignatures=true)).
func clearXMLSignatures(xmlData string) (string, error) {
	doc := etree.NewDocument()
	if err := doc.ReadFromString(xmlData); err != nil {
		return "", err
	}

	for _, el := range doc.FindElements("//Signature") {
		if parent := el.Parent(); parent != nil {
			parent.RemoveChild(el)
		}
	}

	return doc.WriteToString()
}

// TrimXML удаляет текстовые узлы, состоящие только из пробельных символов
// (аналог NCANode XmlService.removeWhitespace). Экспортировано отдельно от
// SignXMLWithOptions.TrimXML для использующих SignWSSE - тот принимает не
// полный документ, а фрагмент (см. SignWSSE), и оборачивает его в SOAP
// конверт сам, поэтому подрезать пробелы нужно до вызова SignWSSE.
func TrimXML(xmlData string) (string, error) {
	return trimXMLWhitespace(xmlData)
}

// trimXMLWhitespace удаляет текстовые узлы, состоящие только из пробельных
// символов (аналог NCANode XmlService.removeWhitespace).
func trimXMLWhitespace(xmlData string) (string, error) {
	doc := etree.NewDocument()
	if err := doc.ReadFromString(xmlData); err != nil {
		return "", err
	}

	removeWhitespaceOnlyText(&doc.Element)

	return doc.WriteToString()
}

func removeWhitespaceOnlyText(el *etree.Element) {
	kept := el.Child[:0]

	for _, child := range el.Child {
		if cd, ok := child.(*etree.CharData); ok {
			if strings.TrimSpace(cd.Data) == "" {
				continue
			}
		}

		kept = append(kept, child)
	}

	el.Child = kept

	for _, child := range el.ChildElements() {
		removeWhitespaceOnlyText(child)
	}
}