183 lines
3.6 KiB
Go
183 lines
3.6 KiB
Go
// +build bindata
|
|
|
|
// Copyright 2016 The Gitea 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"
|
|
"fmt"
|
|
"html/template"
|
|
"io"
|
|
"io/ioutil"
|
|
"path"
|
|
"strings"
|
|
|
|
"code.gitea.io/gitea/modules/log"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
"github.com/Unknwon/com"
|
|
"gopkg.in/macaron.v1"
|
|
)
|
|
|
|
var (
|
|
templates = template.New("")
|
|
)
|
|
|
|
type templateFileSystem struct {
|
|
files []macaron.TemplateFile
|
|
}
|
|
|
|
func (templates templateFileSystem) ListFiles() []macaron.TemplateFile {
|
|
return templates.files
|
|
}
|
|
|
|
func (templates templateFileSystem) Get(name string) (io.Reader, error) {
|
|
for i := range templates.files {
|
|
if templates.files[i].Name()+templates.files[i].Ext() == name {
|
|
return bytes.NewReader(templates.files[i].Data()), nil
|
|
}
|
|
}
|
|
|
|
return nil, fmt.Errorf("file '%s' not found", name)
|
|
}
|
|
|
|
// Renderer implements the macaron handler for serving the templates.
|
|
func Renderer() macaron.Handler {
|
|
fs := templateFileSystem{}
|
|
fs.files = make([]macaron.TemplateFile, 0, 10)
|
|
|
|
for _, assetPath := range AssetNames() {
|
|
if strings.HasPrefix(assetPath, "mail/") {
|
|
continue
|
|
}
|
|
|
|
if !strings.HasSuffix(assetPath, ".tmpl") {
|
|
continue
|
|
}
|
|
|
|
content, err := Asset(assetPath)
|
|
|
|
if err != nil {
|
|
log.Warn("Failed to read embedded %s template. %v", assetPath, err)
|
|
continue
|
|
}
|
|
|
|
fs.files = append(fs.files, macaron.NewTplFile(
|
|
strings.TrimSuffix(
|
|
assetPath,
|
|
".tmpl",
|
|
),
|
|
content,
|
|
".tmpl",
|
|
))
|
|
}
|
|
|
|
customDir := path.Join(setting.CustomPath, "templates")
|
|
|
|
if com.IsDir(customDir) {
|
|
files, err := com.StatDir(customDir)
|
|
|
|
if err != nil {
|
|
log.Warn("Failed to read %s templates dir. %v", customDir, err)
|
|
} else {
|
|
for _, filePath := range files {
|
|
if strings.HasPrefix(filePath, "mail/") {
|
|
continue
|
|
}
|
|
|
|
if !strings.HasSuffix(filePath, ".tmpl") {
|
|
continue
|
|
}
|
|
|
|
content, err := ioutil.ReadFile(path.Join(customDir, filePath))
|
|
|
|
if err != nil {
|
|
log.Warn("Failed to read custom %s template. %v", filePath, err)
|
|
continue
|
|
}
|
|
|
|
fs.files = append(fs.files, macaron.NewTplFile(
|
|
strings.TrimSuffix(
|
|
filePath,
|
|
".tmpl",
|
|
),
|
|
content,
|
|
".tmpl",
|
|
))
|
|
}
|
|
}
|
|
}
|
|
|
|
return macaron.Renderer(macaron.RenderOptions{
|
|
Funcs: NewFuncMap(),
|
|
TemplateFileSystem: fs,
|
|
})
|
|
}
|
|
|
|
// Mailer provides the templates required for sending notification mails.
|
|
func Mailer() *template.Template {
|
|
for _, funcs := range NewFuncMap() {
|
|
templates.Funcs(funcs)
|
|
}
|
|
|
|
for _, assetPath := range AssetNames() {
|
|
if !strings.HasPrefix(assetPath, "mail/") {
|
|
continue
|
|
}
|
|
|
|
if !strings.HasSuffix(assetPath, ".tmpl") {
|
|
continue
|
|
}
|
|
|
|
content, err := Asset(assetPath)
|
|
|
|
if err != nil {
|
|
log.Warn("Failed to read embedded %s template. %v", assetPath, err)
|
|
continue
|
|
}
|
|
|
|
templates.New(
|
|
strings.TrimPrefix(
|
|
strings.TrimSuffix(
|
|
assetPath,
|
|
".tmpl",
|
|
),
|
|
"mail/",
|
|
),
|
|
).Parse(string(content))
|
|
}
|
|
|
|
customDir := path.Join(setting.CustomPath, "templates", "mail")
|
|
|
|
if com.IsDir(customDir) {
|
|
files, err := com.StatDir(customDir)
|
|
|
|
if err != nil {
|
|
log.Warn("Failed to read %s templates dir. %v", customDir, err)
|
|
} else {
|
|
for _, filePath := range files {
|
|
if !strings.HasSuffix(filePath, ".tmpl") {
|
|
continue
|
|
}
|
|
|
|
content, err := ioutil.ReadFile(path.Join(customDir, filePath))
|
|
|
|
if err != nil {
|
|
log.Warn("Failed to read custom %s template. %v", filePath, err)
|
|
continue
|
|
}
|
|
|
|
templates.New(
|
|
strings.TrimSuffix(
|
|
filePath,
|
|
".tmpl",
|
|
),
|
|
).Parse(string(content))
|
|
}
|
|
}
|
|
}
|
|
|
|
return templates
|
|
}
|