Skip to content

Commit

Permalink
Fix baseof with regular define regression
Browse files Browse the repository at this point in the history
  • Loading branch information
bep committed Jan 23, 2020
1 parent 7ed22e9 commit 50bc103
Show file tree
Hide file tree
Showing 3 changed files with 71 additions and 14 deletions.
37 changes: 37 additions & 0 deletions hugolib/template_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -361,6 +361,43 @@ Base %d: {{ block "main" . }}FOO{{ end }}

}

// https://github.com/gohugoio/hugo/issues/6790
func TestTemplateNoBasePlease(t *testing.T) {
t.Parallel()
b := newTestSitesBuilder(t).WithSimpleConfigFile()

b.WithTemplates("_default/list.html", `
{{ define "main" }}
Bonjour
{{ end }}
{{ printf "list" }}
`)

b.WithTemplates(
"_default/single.html", `
{{ printf "single" }}
{{ define "main" }}
Bonjour
{{ end }}
`)

b.WithContent("blog/p1.md", `---
title: The Page
---
`)

b.Build(BuildCfg{})

b.AssertFileContent("public/blog/p1/index.html", `single`)
b.AssertFileContent("public/blog/index.html", `list`)

}

func TestTemplateLookupSite(t *testing.T) {
t.Run("basic", func(t *testing.T) {
t.Parallel()
Expand Down
44 changes: 30 additions & 14 deletions tpl/tplimpl/template.go
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,9 @@ var (
_ tpl.Info = (*templateState)(nil)
)

var defineRe = regexp.MustCompile(`{{-?\s?define`)
// A template needing a base template is a template with only define sections,
// but we check only for the start.
var defineRe = regexp.MustCompile(`^\s*{{-?\s?define`)

func newIdentity(name string) identity.Manager {
return identity.NewManager(identity.NewPathIdentity(files.ComponentFolderLayouts, name))
Expand Down Expand Up @@ -379,20 +381,19 @@ func (t *templateHandler) findLayout(d output.LayoutDescriptor, f output.Format)
}
}

if !found {
return nil, false, errors.Errorf("no baseof layout found for %q:", name)
}

templ, err := t.applyBaseTemplate(overlay, base)
if err != nil {
return nil, false, err
}

ts := newTemplateState(templ, overlay)
ts.baseInfo = base

// Add the base identity to detect changes
ts.Add(identity.NewPathIdentity(files.ComponentFolderLayouts, base.name))
if found {
ts.baseInfo = base

// Add the base identity to detect changes
ts.Add(identity.NewPathIdentity(files.ComponentFolderLayouts, base.name))
}

t.applyTemplateTransformers(t.main, ts)

Expand Down Expand Up @@ -565,20 +566,35 @@ func (t *templateHandler) addTemplateTo(info templateInfo, to *templateNamespace

func (t *templateHandler) applyBaseTemplate(overlay, base templateInfo) (tpl.Template, error) {
if overlay.isText {
templ, err := t.main.prototypeTextClone.New(overlay.name).Parse(base.template)
if err != nil {
return nil, base.errWithFileContext("parse failed", err)
var (
templ = t.main.prototypeTextClone.New(overlay.name)
err error
)

if !base.IsZero() {
templ, err = templ.Parse(base.template)
if err != nil {
return nil, base.errWithFileContext("parse failed", err)
}
}

templ, err = templ.Parse(overlay.template)
if err != nil {
return nil, overlay.errWithFileContext("parse failed", err)
}
return templ, nil
}

templ, err := t.main.prototypeHTMLClone.New(overlay.name).Parse(base.template)
if err != nil {
return nil, base.errWithFileContext("parse failed", err)
var (
templ = t.main.prototypeHTMLClone.New(overlay.name)
err error
)

if !base.IsZero() {
templ, err = templ.Parse(base.template)
if err != nil {
return nil, base.errWithFileContext("parse failed", err)
}
}

templ, err = htmltemplate.Must(templ.Clone()).Parse(overlay.template)
Expand Down
4 changes: 4 additions & 0 deletions tpl/tplimpl/template_errors.go
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,10 @@ type templateInfo struct {
realFilename string
}

func (t templateInfo) IsZero() bool {
return t.name == ""
}

func (t templateInfo) resolveType() templateType {
return resolveTemplateType(t.name)
}
Expand Down

0 comments on commit 50bc103

Please sign in to comment.