imgproxy/imagetype_new/svg.go

package imagetype_new

import (
	"strings"

	"github.com/imgproxy/imgproxy/v3/bufreader"

	"github.com/tdewolff/parse/v2"
	"github.com/tdewolff/parse/v2/xml"
)

func init() {
	// Register SVG detector (needs at least 1000 bytes to reliably detect SVG)
	RegisterDetector(IsSVG)
}

func IsSVG(r bufreader.ReadPeeker) (Type, error) {
	l := xml.NewLexer(parse.NewInput(r))

	for {
		tt, _ := l.Next()

		switch tt {
		case xml.ErrorToken:
			return Unknown, nil

		case xml.StartTagToken:
			tag := strings.ToLower(string(l.Text()))
			if tag == "svg" || tag == "svg:svg" {
				return SVG, nil
			}
		}
	}
}