mirror of https://github.com/miniflux/v2.git
122 lines
2.9 KiB
Go
122 lines
2.9 KiB
Go
// Copyright 2017 Frédéric Guillot. All rights reserved.
|
|
// Use of this source code is governed by the Apache 2.0
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package template // import "miniflux.app/template"
|
|
|
|
import (
|
|
"bytes"
|
|
"embed"
|
|
"html/template"
|
|
"strings"
|
|
"time"
|
|
|
|
"miniflux.app/errors"
|
|
"miniflux.app/locale"
|
|
"miniflux.app/logger"
|
|
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
//go:embed templates/common/*.html
|
|
var commonTemplateFiles embed.FS
|
|
|
|
//go:embed templates/views/*.html
|
|
var viewTemplateFiles embed.FS
|
|
|
|
// Engine handles the templating system.
|
|
type Engine struct {
|
|
templates map[string]*template.Template
|
|
funcMap *funcMap
|
|
}
|
|
|
|
// NewEngine returns a new template engine.
|
|
func NewEngine(router *mux.Router) *Engine {
|
|
return &Engine{
|
|
templates: make(map[string]*template.Template),
|
|
funcMap: &funcMap{router},
|
|
}
|
|
}
|
|
|
|
// ParseTemplates parses template files embed into the application.
|
|
func (e *Engine) ParseTemplates() error {
|
|
var commonTemplateContents strings.Builder
|
|
|
|
dirEntries, err := commonTemplateFiles.ReadDir("templates/common")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, dirEntry := range dirEntries {
|
|
fileData, err := commonTemplateFiles.ReadFile("templates/common/" + dirEntry.Name())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
commonTemplateContents.Write(fileData)
|
|
}
|
|
|
|
dirEntries, err = viewTemplateFiles.ReadDir("templates/views")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, dirEntry := range dirEntries {
|
|
templateName := dirEntry.Name()
|
|
fileData, err := viewTemplateFiles.ReadFile("templates/views/" + dirEntry.Name())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var templateContents strings.Builder
|
|
templateContents.WriteString(commonTemplateContents.String())
|
|
templateContents.Write(fileData)
|
|
|
|
logger.Debug("[Template] Parsing: %s", templateName)
|
|
e.templates[templateName] = template.Must(template.New("main").Funcs(e.funcMap.Map()).Parse(templateContents.String()))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Render process a template.
|
|
func (e *Engine) Render(name, language string, data interface{}) []byte {
|
|
tpl, ok := e.templates[name]
|
|
if !ok {
|
|
logger.Fatal("[Template] The template %s does not exists", name)
|
|
}
|
|
|
|
printer := locale.NewPrinter(language)
|
|
|
|
// Functions that need to be declared at runtime.
|
|
tpl.Funcs(template.FuncMap{
|
|
"elapsed": func(timezone string, t time.Time) string {
|
|
return elapsedTime(printer, timezone, t)
|
|
},
|
|
"t": func(key interface{}, args ...interface{}) string {
|
|
switch k := key.(type) {
|
|
case string:
|
|
return printer.Printf(k, args...)
|
|
case errors.LocalizedError:
|
|
return k.Localize(printer)
|
|
case *errors.LocalizedError:
|
|
return k.Localize(printer)
|
|
case error:
|
|
return k.Error()
|
|
default:
|
|
return ""
|
|
}
|
|
},
|
|
"plural": func(key string, n int, args ...interface{}) string {
|
|
return printer.Plural(key, n, args...)
|
|
},
|
|
})
|
|
|
|
var b bytes.Buffer
|
|
err := tpl.ExecuteTemplate(&b, "base", data)
|
|
if err != nil {
|
|
logger.Fatal("[Template] Unable to render template: %v", err)
|
|
}
|
|
|
|
return b.Bytes()
|
|
}
|