gitea/modules/fileicon/material.go

// Copyright 2025 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package fileicon

import (
	"html/template"
	"path"
	"strings"
	"sync"

	"code.gitea.io/gitea/modules/git"
	"code.gitea.io/gitea/modules/json"
	"code.gitea.io/gitea/modules/log"
	"code.gitea.io/gitea/modules/options"
	"code.gitea.io/gitea/modules/reqctx"
	"code.gitea.io/gitea/modules/svg"
)

type materialIconRulesData struct {
	FileNames      map[string]string `json:"fileNames"`
	FolderNames    map[string]string `json:"folderNames"`
	FileExtensions map[string]string `json:"fileExtensions"`
}

type MaterialIconProvider struct {
	once  sync.Once
	rules *materialIconRulesData
	svgs  map[string]string
}

var materialIconProvider MaterialIconProvider

func DefaultMaterialIconProvider() *MaterialIconProvider {
	materialIconProvider.once.Do(materialIconProvider.loadData)
	return &materialIconProvider
}

func (m *MaterialIconProvider) loadData() {
	buf, err := options.AssetFS().ReadFile("fileicon/material-icon-rules.json")
	if err != nil {
		log.Error("Failed to read material icon rules: %v", err)
		return
	}
	err = json.Unmarshal(buf, &m.rules)
	if err != nil {
		log.Error("Failed to unmarshal material icon rules: %v", err)
		return
	}

	buf, err = options.AssetFS().ReadFile("fileicon/material-icon-svgs.json")
	if err != nil {
		log.Error("Failed to read material icon rules: %v", err)
		return
	}
	err = json.Unmarshal(buf, &m.svgs)
	if err != nil {
		log.Error("Failed to unmarshal material icon rules: %v", err)
		return
	}
	log.Debug("Loaded material icon rules and SVG images")
}

func (m *MaterialIconProvider) renderFileIconSVG(ctx reqctx.RequestContext, name, svg string) template.HTML {
	data := ctx.GetData()
	renderedSVGs, _ := data["_RenderedSVGs"].(map[string]bool)
	if renderedSVGs == nil {
		renderedSVGs = make(map[string]bool)
		data["_RenderedSVGs"] = renderedSVGs
	}
	// This part is a bit hacky, but it works really well. It should be safe to do so because all SVG icons are generated by us.
	// Will try to refactor this in the future.
	if !strings.HasPrefix(svg, "<svg") {
		panic("Invalid SVG icon")
	}
	svgID := "svg-mfi-" + name
	svgCommonAttrs := `class="svg fileicon" width="16" height="16" aria-hidden="true"`
	posOuterBefore := strings.IndexByte(svg, '>')
	if renderedSVGs[svgID] && posOuterBefore != -1 {
		return template.HTML(`<svg ` + svgCommonAttrs + `><use xlink:href="#` + svgID + `"></use></svg>`)
	}
	svg = `<svg id="` + svgID + `" ` + svgCommonAttrs + svg[4:]
	renderedSVGs[svgID] = true
	return template.HTML(svg)
}

func (m *MaterialIconProvider) FileIcon(ctx reqctx.RequestContext, entry *git.TreeEntry) template.HTML {
	if m.rules == nil {
		return BasicThemeIcon(entry)
	}

	if entry.IsLink() {
		if te, err := entry.FollowLink(); err == nil && te.IsDir() {
			return svg.RenderHTML("material-folder-symlink")
		}
		return svg.RenderHTML("octicon-file-symlink-file") // TODO: find some better icons for them
	}

	name := m.findIconNameByGit(entry)
	if name == "folder" {
		// the material icon pack's "folder" icon doesn't look good, so use our built-in one
		return svg.RenderHTML("material-folder-generic")
	}
	if iconSVG, ok := m.svgs[name]; ok && iconSVG != "" {
		return m.renderFileIconSVG(ctx, name, iconSVG)
	}
	return svg.RenderHTML("octicon-file")
}

func (m *MaterialIconProvider) FindIconName(name string, isDir bool) string {
	iconsData := m.rules
	fileNameLower := strings.ToLower(path.Base(name))
	if isDir {
		if s, ok := iconsData.FolderNames[fileNameLower]; ok {
			return s
		}
		return "folder"
	}

	if s, ok := iconsData.FileNames[fileNameLower]; ok {
		return s
	}

	for i := len(fileNameLower) - 1; i >= 0; i-- {
		if fileNameLower[i] == '.' {
			ext := fileNameLower[i+1:]
			if s, ok := iconsData.FileExtensions[ext]; ok {
				return s
			}
		}
	}

	return "file"
}

func (m *MaterialIconProvider) findIconNameByGit(entry *git.TreeEntry) string {
	if entry.IsSubModule() {
		return "folder-git"
	}
	return m.FindIconName(entry.Name(), entry.IsDir())
}