xmlGroup.go

// Copyright 2020 - 2021 The xgen Authors. All rights reserved. Use of this
// source code is governed by a BSD-style license that can be found in the
// LICENSE file.
//
// Package xgen written in pure Go providing a set of functions that allow you
// to parse XSD (XML schema files). This library needs Go version 1.10 or
// later.

package xgen

import "encoding/xml"

// OnGroup handles parsing event on the group start elements. The group
// element is used to define a group of elements to be used in complex type
// definitions.
func (opt *Options) OnGroup(ele xml.StartElement, protoTree []interface{}) (err error) {
	group := Group{}
	for _, attr := range ele.Attr {
		if attr.Name.Local == "name" {
			group.Name = attr.Value
		}
		if attr.Name.Local == "ref" {
			group.Name = attr.Value
			group.Ref, err = opt.GetValueType(attr.Value, protoTree)
			if err != nil {
				return
			}
		}
		if attr.Name.Local == "maxOccurs" {
			if attr.Value != "0" {
				group.Plural = true
			}
		}
	}
	if opt.Choice.Len() > 0 {
		group.Plural = group.Plural || opt.Choice.Peek().(*Choice).Plural
	}

	if opt.ComplexType.Len() == 0 {
		if opt.InGroup == 0 {
			opt.InGroup++
			opt.CurrentEle = opt.InElement
			opt.Group.Push(&group)
			return
		}
		if opt.InGroup > 0 {
			opt.InGroup++
			opt.Group.Peek().(*Group).Groups = append(opt.Group.Peek().(*Group).Groups, group)
			return
		}

	}
	if opt.ComplexType.Len() > 0 {
		if !inGroups(&group, opt.ComplexType.Peek().(*ComplexType).Groups) {
			opt.ComplexType.Peek().(*ComplexType).Groups = append(opt.ComplexType.Peek().(*ComplexType).Groups, group)
		}
		return
	}
	return
}

// EndGroup handles parsing event on the group end elements.
func (opt *Options) EndGroup(ele xml.EndElement, protoTree []interface{}) (err error) {
	if ele.Name.Local == opt.CurrentEle && opt.InGroup == 1 {
		opt.ProtoTree = append(opt.ProtoTree, opt.Group.Pop())
		opt.CurrentEle = ""
		opt.InGroup--
	}
	if ele.Name.Local == opt.CurrentEle {
		opt.InGroup--
	}
	return
}

func inGroups(group *Group, groups []Group) bool {
	for _, g := range groups {
		if g.Name == group.Name {
			return true
		}
	}
	return false
}