hugolib: Add .Page.BundleType

I eturn either:

1. leaf
2. branch
3. empty string

The above sits well with constructs like:

```
{{ with .BundleType }}
// Now we know it is a bundle
{{ end }}
```

Fixes #4662

diff --git a/hugolib/page.go b/hugolib/page.go
index 5f9f86a0f47ccf89eb274e86c3874a1e098d3b95..ebaffa57f6bfbfcdff6e68d822121c4309dc9e20 100644 (file)
--- a/hugolib/page.go
+++ b/hugolib/page.go
@@ -451,6 +451,26 @@ func (p *Page) IsPage() bool {
        return p.Kind == KindPage
 }
 
+// BundleType returns the bundle type: "leaf", "branch" or an empty string if it is none.
+// See https://gohugo.io/content-management/page-bundles/
+func (p *Page) BundleType() string {
+       if p.IsNode() {
+               return "branch"
+       }
+
+       var source interface{} = p.Source.File
+       if fi, ok := source.(*fileInfo); ok {
+               switch fi.bundleTp {
+               case bundleBranch:
+                       return "branch"
+               case bundleLeaf:
+                       return "leaf"
+               }
+       }
+
+       return ""
+}
+
 type Source struct {
        Frontmatter []byte
        Content     []byte

diff --git a/hugolib/page_bundler_test.go b/hugolib/page_bundler_test.go
index a41069d52344e5bf54d6fccb3d28028db5e1e8ed..2e0ca9e4cb237175a1feebbd7d8ddfc6893e65fd 100644 (file)
--- a/hugolib/page_bundler_test.go
+++ b/hugolib/page_bundler_test.go
@@ -82,6 +82,7 @@ func TestPageBundlerSiteRegular(t *testing.T) {
                                assert.Len(s.RegularPages, 8)
 
                                singlePage := s.getPage(KindPage, "a/1.md")
+                               assert.Equal("", singlePage.BundleType())
 
                                assert.NotNil(singlePage)
                                assert.Equal(singlePage, s.getPage("page", "a/1"))
@@ -105,8 +106,12 @@ func TestPageBundlerSiteRegular(t *testing.T) {
 
                                leafBundle1 := s.getPage(KindPage, "b/my-bundle/index.md")
                                assert.NotNil(leafBundle1)
+                               assert.Equal("leaf", leafBundle1.BundleType())
                                assert.Equal("b", leafBundle1.Section())
-                               assert.NotNil(s.getPage(KindSection, "b"))
+                               sectionB := s.getPage(KindSection, "b")
+                               assert.NotNil(sectionB)
+                               home, _ := s.Info.Home()
+                               assert.Equal("branch", home.BundleType())
 
                                // This is a root bundle and should live in the "home section"
                                // See https://github.com/gohugoio/hugo/issues/4332