hubolib: Add HasShortcode

Fixes #3707

diff --git a/hugolib/page.go b/hugolib/page.go
index 0f44b8b99cf1ee9689a6d04b98eb5ec0a26a1ee1..9937be325964d97c7ca7360c2de3874ead8be63b 100644 (file)
--- a/hugolib/page.go
+++ b/hugolib/page.go
@@ -791,6 +791,17 @@ func (p *Page) Extension() string {
        return p.extension
 }
 
+// HasShortcode return whether the page has a shortcode with the given name.
+// This method is mainly motivated with the Hugo Docs site's need for a list
+// of pages with the `todo` shortcode in it.
+func (p *Page) HasShortcode(name string) bool {
+       if p.shortcodeState == nil {
+               return false
+       }
+
+       return p.shortcodeState.nameSet[name]
+}
+
 // AllTranslations returns all translations, including the current Page.
 func (p *Page) AllTranslations() Pages {
        return p.translations

diff --git a/hugolib/site_output_test.go b/hugolib/site_output_test.go
index 8455a13f7ecc10030f1657741c2a3f8787cdccf7..941387e3546681bf1bb192a0b66d2605aa76ba11 100644 (file)
--- a/hugolib/site_output_test.go
+++ b/hugolib/site_output_test.go
@@ -254,6 +254,9 @@ Content: {{ .Content }}
                require.Equal(t, "webcal://example.com/blog/index.ics", cal.Permalink())
        }
 
+       require.True(t, home.HasShortcode("myShort"))
+       require.False(t, home.HasShortcode("doesNotExist"))
+
 }
 
 // Issue #3447