Function ParseWithOptions

parse.go:40–93

ParseWithOptions is like parse, but with custom options

(r io.Reader, options ParserOptions)

// ParseWithOptions is like parse, but with custom options
func ParseWithOptions(r io.Reader, options ParserOptions) (*Node, error) {
	var lineStarts []int
	// If line numbers are requested, read all data for position tracking
	if options.WithLineNumbers {
		var err error
		var data []byte
		data, err = io.ReadAll(r)
		if err != nil {
			return nil, err
		}
		r = bytes.NewReader(data)

		// Pre-calculate line starts
		lineStarts = []int{0}
		for i, b := range data {
			if b == '\n' {
				lineStarts = append(lineStarts, i+1)
			}
		}
	}

	p := createParser(r)
	if options.WithLineNumbers {
		p.lineStarts = lineStarts
	}
	options.apply(p)

	var err error
	for err == nil {
		_, err = p.parse()
	}

	if err == io.EOF {
		// additional check for validity
		// according to: https://www.w3.org/TR/xml
		// the document MUST contain at least ONE element
		valid := false
		for doc := p.doc; doc != nil; doc = doc.NextSibling {
			for node := doc.FirstChild; node != nil; node = node.NextSibling {
				if node.Type == ElementNode {
					valid = true
					break
				}
			}
		}
		if !valid {
			return nil, fmt.Errorf("xmlquery: invalid XML document")
		}

		return p.doc, nil
	}

	return nil, err
}

type parser struct {
	decoder             *xml.Decoder