mirror of https://github.com/gogs/gogs.git
90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
// Copyright 2020 The Gogs Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package templates
|
|
|
|
import (
|
|
"bytes"
|
|
"embed"
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
"gopkg.in/macaron.v1"
|
|
|
|
"gogs.io/gogs/internal/osutil"
|
|
)
|
|
|
|
//go:embed *.tmpl **/*
|
|
var files embed.FS
|
|
|
|
// fileSystem implements the macaron.TemplateFileSystem interface.
|
|
type fileSystem struct {
|
|
files []macaron.TemplateFile
|
|
}
|
|
|
|
func (fs *fileSystem) ListFiles() []macaron.TemplateFile {
|
|
return fs.files
|
|
}
|
|
|
|
func (fs *fileSystem) Get(name string) (io.Reader, error) {
|
|
for i := range fs.files {
|
|
if fs.files[i].Name()+fs.files[i].Ext() == name {
|
|
return bytes.NewReader(fs.files[i].Data()), nil
|
|
}
|
|
}
|
|
return nil, fmt.Errorf("file %q not found", name)
|
|
}
|
|
|
|
func mustNames(fsys fs.FS) []string {
|
|
var names []string
|
|
walkDirFunc := func(path string, d fs.DirEntry, err error) error {
|
|
if !d.IsDir() {
|
|
names = append(names, path)
|
|
}
|
|
return nil
|
|
}
|
|
if err := fs.WalkDir(fsys, ".", walkDirFunc); err != nil {
|
|
panic("assetNames failure: " + err.Error())
|
|
}
|
|
return names
|
|
}
|
|
|
|
// NewTemplateFileSystem returns a macaron.TemplateFileSystem instance for embedded assets.
|
|
// The argument "dir" can be used to serve subset of embedded assets. Template file
|
|
// found under the "customDir" on disk has higher precedence over embedded assets.
|
|
func NewTemplateFileSystem(dir, customDir string) macaron.TemplateFileSystem {
|
|
if dir != "" && !strings.HasSuffix(dir, "/") {
|
|
dir += "/"
|
|
}
|
|
|
|
var err error
|
|
var tmplFiles []macaron.TemplateFile
|
|
names := mustNames(files)
|
|
for _, name := range names {
|
|
if !strings.HasPrefix(name, dir) {
|
|
continue
|
|
}
|
|
// Check if corresponding custom file exists
|
|
var data []byte
|
|
fpath := path.Join(customDir, name)
|
|
if osutil.IsFile(fpath) {
|
|
data, err = os.ReadFile(fpath)
|
|
} else {
|
|
data, err = files.ReadFile(name)
|
|
}
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
name = strings.TrimPrefix(name, dir)
|
|
ext := path.Ext(name)
|
|
name = strings.TrimSuffix(name, ext)
|
|
tmplFiles = append(tmplFiles, macaron.NewTplFile(name, data, ext))
|
|
}
|
|
return &fileSystem{files: tmplFiles}
|
|
}
|