wxiaoguang
8a20fba8eb
Remove unmaintainable sanitizer rules. No need to add special "class" regexp rules anymore, use RenderInternal.SafeAttr instead, more details (and examples) are in the tests
76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
// 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/htmlutil"
|
|
"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
|
|
}
|
|
|
|
func MockIcon(icon string) func() {
|
|
if svgIcons == nil {
|
|
svgIcons = make(map[string]string)
|
|
}
|
|
orig, exist := svgIcons[icon]
|
|
svgIcons[icon] = fmt.Sprintf(`<svg class="svg %s" width="%d" height="%d"></svg>`, icon, defaultSize, defaultSize)
|
|
return func() {
|
|
if exist {
|
|
svgIcons[icon] = orig
|
|
} else {
|
|
delete(svgIcons, icon)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 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)
|
|
}
|
|
// during test (or something wrong happens), there is no SVG loaded, so use a dummy span to tell that the icon is missing
|
|
return template.HTML(fmt.Sprintf("<span>%s(%d/%s)</span>", template.HTMLEscapeString(icon), size, template.HTMLEscapeString(class)))
|
|
}
|