1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
// 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}
}
|