You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package operation
|
|
|
|
import (
|
|
"html/template"
|
|
"log"
|
|
"mime"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
gopath "path"
|
|
|
|
"github.com/alecthomas/repr"
|
|
"github.com/aziis98/cabret"
|
|
"github.com/aziis98/cabret/operation/layout"
|
|
"github.com/aziis98/cabret/util"
|
|
)
|
|
|
|
var HtmlMimeType = mime.TypeByExtension(".html")
|
|
|
|
var _ cabret.FlatMapOperation = Layout{}
|
|
|
|
type Layout struct {
|
|
// TemplateFilesPattern is a comma separated list of unix glob patterns
|
|
TemplateFilesPattern string
|
|
|
|
Options map[string]any
|
|
}
|
|
|
|
func (op Layout) FlatMap(content cabret.Content) (*cabret.Content, error) {
|
|
var tmpl layout.Template
|
|
|
|
patterns := strings.Split(op.TemplateFilesPattern, ",")
|
|
|
|
tmplFiles := []string{}
|
|
for _, pat := range patterns {
|
|
files, err := filepath.Glob(strings.TrimSpace(pat))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
tmplFiles = append(tmplFiles, files...)
|
|
}
|
|
|
|
log.Printf(`[Layout] template pattern "%s" expanded to %s`, op.TemplateFilesPattern, repr.String(tmplFiles))
|
|
|
|
if gopath.Ext(tmplFiles[0]) == ".html" {
|
|
var err error
|
|
if tmpl, err = layout.NewHtmlTemplate(tmplFiles...); err != nil {
|
|
return nil, err
|
|
}
|
|
} else {
|
|
var err error
|
|
if tmpl, err = layout.NewTextTemplate(tmplFiles...); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
ctx := util.CloneMap(content.Metadata)
|
|
|
|
if content.Type == HtmlMimeType {
|
|
ctx["Content"] = template.HTML(content.Data)
|
|
} else {
|
|
ctx["Content"] = content.Data
|
|
}
|
|
|
|
data, err := tmpl.Render(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
content.Data = data
|
|
return &content, nil
|
|
}
|