tpl/transform: Add CanHighlight

Closes #9573
This commit is contained in:
Bjørn Erik Pedersen 2022-02-27 17:09:11 +01:00
parent cff14144a6
commit 39261b689e
2 changed files with 16 additions and 0 deletions

View file

@ -18,6 +18,7 @@ import (
"html"
"html/template"
"github.com/alecthomas/chroma/lexers"
"github.com/gohugoio/hugo/cache/namedmemcache"
"github.com/gohugoio/hugo/common/herrors"
"github.com/gohugoio/hugo/markup/converter/hooks"
@ -90,6 +91,11 @@ func (ns *Namespace) HighlightCodeBlock(ctx hooks.CodeblockContext, opts ...inte
return hl.HighlightCodeBlock(ctx, optsv)
}
// CanHighlight returns whether the given language is supported by the Chroma highlighter.
func (ns *Namespace) CanHighlight(lang string) bool {
return lexers.Get(lang) != nil
}
// HTMLEscape returns a copy of s with reserved HTML characters escaped.
func (ns *Namespace) HTMLEscape(s interface{}) (string, error) {
ss, err := cast.ToStringE(s)

View file

@ -95,6 +95,16 @@ func TestHighlight(t *testing.T) {
}
}
func TestCanHighlight(t *testing.T) {
t.Parallel()
c := qt.New(t)
ns := &transform.Namespace{}
c.Assert(ns.CanHighlight("go"), qt.Equals, true)
c.Assert(ns.CanHighlight("foo"), qt.Equals, false)
}
func TestHTMLEscape(t *testing.T) {
t.Parallel()
b := hugolib.NewIntegrationTestBuilder(