Function ParseEntityMap

xmlparser/entities.go:14–53

ParseEntityMap searches for a DOCTYPE directive in the given nodes and extracts entity declarations into a map.

(nodes []Token)

// ParseEntityMap searches for a DOCTYPE directive in the given nodes
// and extracts entity declarations into a map.
func ParseEntityMap(nodes []Token) map[string][]byte {
	// Find doctype
	var doctype *Directive
	for _, node := range nodes {
		if dir, ok := node.(*Directive); ok {
			doctype = dir
			break
		}
	}

	if doctype == nil {
		return nil
	}

	start := bytes.IndexByte(doctype.Data, '[')
	if start == -1 {
		return nil
	}

	data := doctype.Data[start:]

	// Remove comments to avoid false positives
	if bytes.Contains(data, commentStart) {
		data = entityCommentRe.ReplaceAll(data, []byte{})
	}

	matches := entityDeclRe.FindAllSubmatch(data, -1)
	if matches == nil {
		return nil
	}

	em := make(map[string][]byte, len(matches))
	for _, match := range matches {
		name := match[1]
		value := bytes.Trim(match[2], `"'`)
		em[string(name)] = value
	}

	return em
}

// ReplaceEntitiesBytes replaces XML entities in the given byte slice
// according to the provided entity map.