forked from Shiloh/githaven
acd5e5a868
* Add StatDir and replace com.StatDir * a nit * Remove wrong file Co-authored-by: 6543 <6543@obermui.de>
120 lines
3.0 KiB
Go
120 lines
3.0 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 (
|
|
"html/template"
|
|
"io/ioutil"
|
|
"path"
|
|
"strings"
|
|
texttmpl "text/template"
|
|
|
|
"code.gitea.io/gitea/modules/log"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
"code.gitea.io/gitea/modules/util"
|
|
|
|
"gitea.com/macaron/macaron"
|
|
)
|
|
|
|
var (
|
|
subjectTemplates = texttmpl.New("")
|
|
bodyTemplates = template.New("")
|
|
)
|
|
|
|
// HTMLRenderer implements the macaron handler for serving HTML templates.
|
|
func HTMLRenderer() macaron.Handler {
|
|
return macaron.Renderer(macaron.RenderOptions{
|
|
Funcs: NewFuncMap(),
|
|
Directory: path.Join(setting.StaticRootPath, "templates"),
|
|
AppendDirectories: []string{
|
|
path.Join(setting.CustomPath, "templates"),
|
|
},
|
|
})
|
|
}
|
|
|
|
// JSONRenderer implements the macaron handler for serving JSON templates.
|
|
func JSONRenderer() macaron.Handler {
|
|
return macaron.Renderer(macaron.RenderOptions{
|
|
Funcs: NewFuncMap(),
|
|
Directory: path.Join(setting.StaticRootPath, "templates"),
|
|
AppendDirectories: []string{
|
|
path.Join(setting.CustomPath, "templates"),
|
|
},
|
|
HTMLContentType: "application/json",
|
|
})
|
|
}
|
|
|
|
// Mailer provides the templates required for sending notification mails.
|
|
func Mailer() (*texttmpl.Template, *template.Template) {
|
|
for _, funcs := range NewTextFuncMap() {
|
|
subjectTemplates.Funcs(funcs)
|
|
}
|
|
for _, funcs := range NewFuncMap() {
|
|
bodyTemplates.Funcs(funcs)
|
|
}
|
|
|
|
staticDir := path.Join(setting.StaticRootPath, "templates", "mail")
|
|
|
|
isDir, err := util.IsDir(staticDir)
|
|
if err != nil {
|
|
log.Warn("Unable to check if templates dir %s is a directory. Error: %v", staticDir, err)
|
|
}
|
|
if isDir {
|
|
files, err := util.StatDir(staticDir)
|
|
|
|
if err != nil {
|
|
log.Warn("Failed to read %s templates dir. %v", staticDir, err)
|
|
} else {
|
|
for _, filePath := range files {
|
|
if !strings.HasSuffix(filePath, ".tmpl") {
|
|
continue
|
|
}
|
|
|
|
content, err := ioutil.ReadFile(path.Join(staticDir, filePath))
|
|
|
|
if err != nil {
|
|
log.Warn("Failed to read static %s template. %v", filePath, err)
|
|
continue
|
|
}
|
|
|
|
buildSubjectBodyTemplate(subjectTemplates, bodyTemplates, strings.TrimSuffix(filePath, ".tmpl"), content)
|
|
}
|
|
}
|
|
}
|
|
|
|
customDir := path.Join(setting.CustomPath, "templates", "mail")
|
|
|
|
isDir, err = util.IsDir(customDir)
|
|
if err != nil {
|
|
log.Warn("Unable to check if templates dir %s is a directory. Error: %v", customDir, err)
|
|
}
|
|
if isDir {
|
|
files, err := util.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
|
|
}
|
|
|
|
buildSubjectBodyTemplate(subjectTemplates, bodyTemplates, strings.TrimSuffix(filePath, ".tmpl"), content)
|
|
}
|
|
}
|
|
}
|
|
|
|
return subjectTemplates, bodyTemplates
|
|
}
|