Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Fix language handling in ExecuteAsTemplate
  • Loading branch information
bep committed Nov 26, 2019
1 parent 03b369e commit eda9d5b
Show file tree
Hide file tree
Showing 7 changed files with 65 additions and 26 deletions.
10 changes: 10 additions & 0 deletions helpers/general.go
Expand Up @@ -482,3 +482,13 @@ func PrintFs(fs afero.Fs, path string, w io.Writer) {
return nil
})
}

type Debugger interface {
Debug() string
}

type DebuggerFn func() string

func (fn DebuggerFn) Debug() string {
return fn()
}
12 changes: 12 additions & 0 deletions hugolib/hugo_sites_build_test.go
Expand Up @@ -1445,5 +1445,17 @@ weight: 2002
PNG Data
`)

i18nContent := func(id, value string) string {
return fmt.Sprintf(`
[%s]
other = %q
`, id, value)
}

b.WithSourceFile("i18n/en.toml", i18nContent("hello", "Hello"))
b.WithSourceFile("i18n/fr.toml", i18nContent("hello", "Bonjour"))
b.WithSourceFile("i18n/nb.toml", i18nContent("hello", "Hallo"))
b.WithSourceFile("i18n/nn.toml", i18nContent("hello", "Hallo"))

return &multiSiteTestBuilder{sitesBuilder: b, configFormat: configFormat, config: config, configData: configData}
}
30 changes: 28 additions & 2 deletions hugolib/resource_chain_test.go
Expand Up @@ -418,8 +418,7 @@ Fingerprinted: {{ $fingerprinted.RelPermalink }}

}},
{"execute-as-template", func() bool {
// TODO(bep) eventually remove
return isGo111()
return true
}, func(b *sitesBuilder) {
b.WithTemplates("home.html", `
{{ $var := "Hugo Page" }}
Expand Down Expand Up @@ -668,3 +667,30 @@ JSON: {{ $json.RelPermalink }}: {{ $json.Content }}
"JSON: /jsons/data1.json: json1 content",
"JSONS: 2", "/jsons/data1.json: json1 content")
}

func TestExecuteAsTemplateWithLanguage(t *testing.T) {
b := newMultiSiteTestDefaultBuilder(t)
indexContent := `
Lang: {{ site.Language.Lang }}
{{ $templ := "{{T \"hello\"}}" | resources.FromString "f1.html" }}
{{ $helloResource := $templ | resources.ExecuteAsTemplate (print "f%s.html" .Lang) . }}
Hello1: {{T "hello"}}
Hello2: {{ $helloResource.Content }}
LangURL: {{ relLangURL "foo" }}
`
b.WithTemplatesAdded("index.html", indexContent)
b.WithTemplatesAdded("index.fr.html", indexContent)

b.Build(BuildCfg{})

b.AssertFileContent("public/en/index.html", `
Hello1: Hello
Hello2: Hello
`)

b.AssertFileContent("public/fr/index.html", `
Hello1: Bonjour
Hello2: Bonjour
`)

}
4 changes: 4 additions & 0 deletions tpl/resources/resources.go
Expand Up @@ -35,6 +35,10 @@ import (
"github.com/spf13/cast"
)

type debug interface {
Debug() string
}

// New returns a new instance of the resources-namespaced template functions.
func New(deps *deps.Deps) (*Namespace, error) {
if deps.ResourceSpec == nil {
Expand Down
2 changes: 0 additions & 2 deletions tpl/template.go
Expand Up @@ -50,8 +50,6 @@ type TemplateHandler interface {
AddLateTemplate(name, tpl string) error
LoadTemplates(prefix string) error

NewTextTemplate() TemplateParseFinder

MarkReady() error
RebuildClone()
}
Expand Down
27 changes: 10 additions & 17 deletions tpl/tplimpl/template.go
Expand Up @@ -98,8 +98,6 @@ type templateHandler struct {
text *textTemplates
html *htmlTemplates

extTextTemplates []*textTemplate

amberFuncMap template.FuncMap

errors []*templateErr
Expand Down Expand Up @@ -153,15 +151,7 @@ func (t *templateHandler) addShortcodeVariant(name string, info tpl.Info, templ
}
}

// NewTextTemplate provides a text template parser that has all the Hugo
// template funcs etc. built-in.
func (t *templateHandler) NewTextTemplate() tpl.TemplateParseFinder {
t.mu.Lock()
defer t.mu.Unlock()

tt := &textTemplate{t: texttemplate.New("")}
t.extTextTemplates = append(t.extTextTemplates, tt)

func (t *templateHandler) wrapTextTemplate(tt *textTemplate) tpl.TemplateParseFinder {
return struct {
tpl.TemplateParser
tpl.TemplateLookup
Expand Down Expand Up @@ -283,8 +273,11 @@ func (t *templateHandler) clone(d *deps.Deps) *templateHandler {
shortcodes: make(map[string]*shortcodeTemplates),
templateInfo: t.templateInfo,
html: &htmlTemplates{t: template.Must(t.html.t.Clone()), overlays: make(map[string]*template.Template), templatesCommon: t.html.templatesCommon},
text: &textTemplates{textTemplate: &textTemplate{t: texttemplate.Must(t.text.t.Clone())}, overlays: make(map[string]*texttemplate.Template), templatesCommon: t.text.templatesCommon},
errors: make([]*templateErr, 0),
text: &textTemplates{
textTemplate: &textTemplate{t: texttemplate.Must(t.text.t.Clone())},
standalone: &textTemplate{t: texttemplate.New("")},
overlays: make(map[string]*texttemplate.Template), templatesCommon: t.text.templatesCommon},
errors: make([]*templateErr, 0),
}

for k, v := range t.shortcodes {
Expand All @@ -302,6 +295,7 @@ func (t *templateHandler) clone(d *deps.Deps) *templateHandler {
}

d.Tmpl = c
d.TextTmpl = c.wrapTextTemplate(c.text.standalone)

c.initFuncs()

Expand Down Expand Up @@ -339,6 +333,7 @@ func newTemplateAdapter(deps *deps.Deps) *templateHandler {
}
textT := &textTemplates{
textTemplate: &textTemplate{t: texttemplate.New("")},
standalone: &textTemplate{t: texttemplate.New("")},
overlays: make(map[string]*texttemplate.Template),
templatesCommon: common,
}
Expand Down Expand Up @@ -431,6 +426,7 @@ func (t *textTemplates) setTemplateFuncster(f *templateFuncster) {
type textTemplates struct {
*templatesCommon
*textTemplate
standalone *textTemplate
clone *texttemplate.Template
cloneClone *texttemplate.Template

Expand Down Expand Up @@ -468,6 +464,7 @@ func (t *textTemplates) lookup(name string) *texttemplate.Template {
func (t *templateHandler) setFuncs(funcMap map[string]interface{}) {
t.html.setFuncs(funcMap)
t.text.setFuncs(funcMap)
t.setFuncMapInTemplate(t.text.standalone.t, funcMap)
}

// SetFuncs replaces the funcs in the func maps with new definitions.
Expand Down Expand Up @@ -781,10 +778,6 @@ func (t *templateHandler) initFuncs() {
}
}

for _, extText := range t.extTextTemplates {
extText.t.Funcs(funcMap)
}

// Amber is HTML only.
t.amberFuncMap = template.FuncMap{}

Expand Down
6 changes: 1 addition & 5 deletions tpl/tplimpl/templateProvider.go
Expand Up @@ -26,11 +26,9 @@ var DefaultTemplateProvider *TemplateProvider
// Update updates the Hugo Template System in the provided Deps
// with all the additional features, templates & functions.
func (*TemplateProvider) Update(deps *deps.Deps) error {

newTmpl := newTemplateAdapter(deps)
deps.Tmpl = newTmpl

deps.TextTmpl = newTmpl.NewTextTemplate()
deps.TextTmpl = newTmpl.wrapTextTemplate(newTmpl.text.standalone)

newTmpl.initFuncs()

Expand All @@ -56,8 +54,6 @@ func (*TemplateProvider) Clone(d *deps.Deps) error {
t := d.Tmpl.(*templateHandler)
clone := t.clone(d)

d.Tmpl = clone

return clone.MarkReady()

}

0 comments on commit eda9d5b

Please sign in to comment.