Skip to content

Commit

Permalink
Fix baseof with regular define regression
Browse files Browse the repository at this point in the history
Fixes #6790
  • Loading branch information
bep committed Jan 23, 2020
1 parent 7ed22e9 commit da3a230
Show file tree
Hide file tree
Showing 4 changed files with 112 additions and 17 deletions.
37 changes: 37 additions & 0 deletions hugolib/template_test.go
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
55 changes: 38 additions & 17 deletions tpl/tplimpl/template.go
Expand Up @@ -72,7 +72,14 @@ 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.
// If a base template does not exist, we will handle that when it's used.
var baseTemplateDefineRe = regexp.MustCompile(`^\s*{{-?\s?define`)

func isBaseTemplate(templ string) bool {
return baseTemplateDefineRe.MatchString(templ)
}

func newIdentity(name string) identity.Manager {
return identity.NewManager(identity.NewPathIdentity(files.ComponentFolderLayouts, name))
Expand Down Expand Up @@ -379,20 +386,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 @@ -537,13 +543,13 @@ func (t *templateHandler) addTemplateFile(name, path string) error {
return err
}

if isBaseTemplate(name) {
if isBaseTemplatePath(name) {
// Store it for later.
t.baseof[name] = tinfo
return nil
}

needsBaseof := !t.noBaseNeeded(name) && defineRe.MatchString(tinfo.template)
needsBaseof := !t.noBaseNeeded(name) && baseTemplateDefineRe.MatchString(tinfo.template)
if needsBaseof {
t.needsBaseof[name] = tinfo
return nil
Expand All @@ -565,20 +571,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 Expand Up @@ -890,7 +911,7 @@ func isBackupFile(path string) bool {
return path[len(path)-1] == '~'
}

func isBaseTemplate(path string) bool {
func isBaseTemplatePath(path string) bool {
return strings.Contains(filepath.Base(path), baseFileBase)
}

Expand Down
4 changes: 4 additions & 0 deletions tpl/tplimpl/template_errors.go
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
33 changes: 33 additions & 0 deletions tpl/tplimpl/template_test.go
@@ -0,0 +1,33 @@
// Copyright 2019 The Hugo Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package tplimpl

import (
"testing"

qt "github.com/frankban/quicktest"
)

func TestIsBaseTemplate(t *testing.T) {
c := qt.New(t)

c.Assert(isBaseTemplate(`{{ define "main" }}`), qt.Equals, true)
c.Assert(isBaseTemplate(`{{define "main" }}`), qt.Equals, true)
c.Assert(isBaseTemplate(`{{- define "main" }}`), qt.Equals, true)
c.Assert(isBaseTemplate(`{{-define "main" }}`), qt.Equals, true)
c.Assert(isBaseTemplate(` {{ define "main" }}`), qt.Equals, true)
c.Assert(isBaseTemplate(`
{{ define "main" }}`), qt.Equals, true)
c.Assert(isBaseTemplate(` A {{ define "main" }}`), qt.Equals, false)

}

0 comments on commit da3a230

Please sign in to comment.