gf/g/encoding/gxml/gxml.go

// Copyright 2017 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://gitee.com/johng/gf.

// Package gxml provides accessing and converting for XML content.
package gxml

import (
    "gitee.com/johng/gf/third/github.com/clbanning/mxj"
    "encoding/xml"
    "io"
	"gitee.com/johng/gf/g/util/gregex"
	"gitee.com/johng/gf/third/github.com/axgle/mahonia"
	"errors"
	"fmt"
	"strings"
)

// 将XML内容解析为map变量
func Decode(xmlbyte []byte) (map[string]interface{}, error) {
    prepare(xmlbyte)
    return mxj.NewMapXml(xmlbyte)
}

// 将map变量解析为XML格式内容
func Encode(v map[string]interface{}, rootTag...string) ([]byte, error) {
    return mxj.Map(v).Xml(rootTag...)
}

func EncodeWithIndent(v map[string]interface{}, rootTag...string) ([]byte, error) {
    return mxj.Map(v).XmlIndent("", "\t", rootTag...)
}

// XML格式内容直接转换为JSON格式内容
func ToJson(xmlbyte []byte) ([]byte, error) {
    prepare(xmlbyte)
	mv, err := mxj.NewMapXml(xmlbyte)
	if err == nil {
        return mv.Json()
    } else {
        return nil, err
    }
}

// XML字符集预处理
// @author wenzi1
// @date 20180604
func prepare(xmlbyte []byte) error {
	patten := `<\?xml.*encoding\s*=\s*['|"](.*?)['|"].*\?>`
	charsetReader := func(charset string, input io.Reader) (io.Reader, error) {
		reader := mahonia.GetCharset(charset)
		if reader == nil {
			return nil, errors.New(fmt.Sprintf("not support charset:%s", charset))
		}
		return reader.NewDecoder().NewReader(input), nil
	}

	matchStr, err := gregex.MatchString(patten, string(xmlbyte))
	if err != nil {
		return err
	}
	
	xmlEncode := "UTF-8"
	if len(matchStr) == 2 {
		xmlEncode = matchStr[1]	
	}

	charset := mahonia.GetCharset(xmlEncode)
	if charset == nil {
		return errors.New(fmt.Sprintf("not support charset:%s", xmlEncode))
	}

	if !strings.EqualFold(charset.Name, "UTF-8") {
		mxj.CustomDecoder = &xml.Decoder{Strict : false, CharsetReader : charsetReader}
	}
	return nil
}