mirror of
				https://github.com/go-gitea/gitea
				synced 2025-11-04 05:18:25 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			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/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
 | 
						|
}
 | 
						|
 | 
						|
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)))
 | 
						|
}
 |