diff options
author | Daniel Baumann <daniel@debian.org> | 2024-10-18 20:33:49 +0200 |
---|---|---|
committer | Daniel Baumann <daniel@debian.org> | 2024-12-12 23:57:56 +0100 |
commit | e68b9d00a6e05b3a941f63ffb696f91e554ac5ec (patch) | |
tree | 97775d6c13b0f416af55314eb6a89ef792474615 /modules/svg | |
parent | Initial commit. (diff) | |
download | forgejo-e68b9d00a6e05b3a941f63ffb696f91e554ac5ec.tar.xz forgejo-e68b9d00a6e05b3a941f63ffb696f91e554ac5ec.zip |
Adding upstream version 9.0.3.
Signed-off-by: Daniel Baumann <daniel@debian.org>
Diffstat (limited to '')
-rw-r--r-- | modules/svg/processor.go | 59 | ||||
-rw-r--r-- | modules/svg/processor_test.go | 29 | ||||
-rw-r--r-- | modules/svg/svg.go | 59 |
3 files changed, 147 insertions, 0 deletions
diff --git a/modules/svg/processor.go b/modules/svg/processor.go new file mode 100644 index 0000000..82248fb --- /dev/null +++ b/modules/svg/processor.go @@ -0,0 +1,59 @@ +// Copyright 2023 The Gitea Authors. All rights reserved. +// SPDX-License-Identifier: MIT + +package svg + +import ( + "bytes" + "fmt" + "regexp" + "sync" +) + +type normalizeVarsStruct struct { + reXMLDoc, + reComment, + reAttrXMLNs, + reAttrSize, + reAttrClassPrefix *regexp.Regexp +} + +var ( + normalizeVars *normalizeVarsStruct + normalizeVarsOnce sync.Once +) + +// Normalize normalizes the SVG content: set default width/height, remove unnecessary tags/attributes +// It's designed to work with valid SVG content. For invalid SVG content, the returned content is not guaranteed. +func Normalize(data []byte, size int) []byte { + normalizeVarsOnce.Do(func() { + normalizeVars = &normalizeVarsStruct{ + reXMLDoc: regexp.MustCompile(`(?s)<\?xml.*?>`), + reComment: regexp.MustCompile(`(?s)<!--.*?-->`), + + reAttrXMLNs: regexp.MustCompile(`(?s)\s+xmlns\s*=\s*"[^"]*"`), + reAttrSize: regexp.MustCompile(`(?s)\s+(width|height)\s*=\s*"[^"]+"`), + reAttrClassPrefix: regexp.MustCompile(`(?s)\s+class\s*=\s*"`), + } + }) + data = normalizeVars.reXMLDoc.ReplaceAll(data, nil) + data = normalizeVars.reComment.ReplaceAll(data, nil) + + data = bytes.TrimSpace(data) + svgTag, svgRemaining, ok := bytes.Cut(data, []byte(">")) + if !ok || !bytes.HasPrefix(svgTag, []byte(`<svg`)) { + return data + } + normalized := bytes.Clone(svgTag) + normalized = normalizeVars.reAttrXMLNs.ReplaceAll(normalized, nil) + normalized = normalizeVars.reAttrSize.ReplaceAll(normalized, nil) + normalized = normalizeVars.reAttrClassPrefix.ReplaceAll(normalized, []byte(` class="`)) + normalized = bytes.TrimSpace(normalized) + normalized = fmt.Appendf(normalized, ` width="%d" height="%d"`, size, size) + if !bytes.Contains(normalized, []byte(` class="`)) { + normalized = append(normalized, ` class="svg"`...) + } + normalized = append(normalized, '>') + normalized = append(normalized, svgRemaining...) + return normalized +} diff --git a/modules/svg/processor_test.go b/modules/svg/processor_test.go new file mode 100644 index 0000000..a028666 --- /dev/null +++ b/modules/svg/processor_test.go @@ -0,0 +1,29 @@ +// Copyright 2023 The Gitea Authors. All rights reserved. +// SPDX-License-Identifier: MIT + +package svg + +import ( + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestNormalize(t *testing.T) { + res := Normalize([]byte("foo"), 1) + assert.Equal(t, "foo", string(res)) + + res = Normalize([]byte(`<?xml version="1.0"?> +<!-- +comment +--> +<svg xmlns = "...">content</svg>`), 1) + assert.Equal(t, `<svg width="1" height="1" class="svg">content</svg>`, string(res)) + + res = Normalize([]byte(`<svg +width="100" +class="svg-icon" +>content</svg>`), 16) + + assert.Equal(t, `<svg class="svg-icon" width="16" height="16">content</svg>`, string(res)) +} diff --git a/modules/svg/svg.go b/modules/svg/svg.go new file mode 100644 index 0000000..016e1dc --- /dev/null +++ b/modules/svg/svg.go @@ -0,0 +1,59 @@ +// Copyright 2020 The Gitea Authors. All rights reserved. +// SPDX-License-Identifier: MIT + +package svg + +import ( + "fmt" + "html/template" + "path" + "strings" + + gitea_html "code.gitea.io/gitea/modules/html" + "code.gitea.io/gitea/modules/log" + "code.gitea.io/gitea/modules/public" +) + +var svgIcons map[string]string + +const defaultSize = 16 + +// Init discovers SVG icons and populates the `svgIcons` variable +func Init() error { + const svgAssetsPath = "assets/img/svg" + files, err := public.AssetFS().ListFiles(svgAssetsPath) + if err != nil { + return err + } + + svgIcons = make(map[string]string, len(files)) + for _, file := range files { + if path.Ext(file) != ".svg" { + continue + } + bs, err := public.AssetFS().ReadFile(svgAssetsPath, file) + if err != nil { + log.Error("Failed to read SVG file %s: %v", file, err) + } else { + svgIcons[file[:len(file)-4]] = string(Normalize(bs, defaultSize)) + } + } + return nil +} + +// RenderHTML renders icons - arguments icon name (string), size (int), class (string) +func RenderHTML(icon string, others ...any) template.HTML { + size, class := gitea_html.ParseSizeAndClass(defaultSize, "", others...) + if svgStr, ok := svgIcons[icon]; ok { + // the code is somewhat hacky, but it just works, because the SVG contents are all normalized + if size != defaultSize { + svgStr = strings.Replace(svgStr, fmt.Sprintf(`width="%d"`, defaultSize), fmt.Sprintf(`width="%d"`, size), 1) + svgStr = strings.Replace(svgStr, fmt.Sprintf(`height="%d"`, defaultSize), fmt.Sprintf(`height="%d"`, size), 1) + } + if class != "" { + svgStr = strings.Replace(svgStr, `class="`, fmt.Sprintf(`class="%s `, class), 1) + } + return template.HTML(svgStr) + } + return "" +} |