tpl/strings: Add findRESubmatch

Fixes #10594
This commit is contained in:
Bjørn Erik Pedersen 2023-01-17 09:35:16 +01:00
parent c6b3887696
commit 2fb40ece5d
4 changed files with 74 additions and 1 deletions

View file

@ -66,6 +66,11 @@ func (t *TemplateFuncsNamespace) AddMethodMapping(m any, aliases []string, examp
name := methodToName(m) name := methodToName(m)
// Rewrite §§ to ` in example commands.
for i, e := range examples {
examples[i][0] = strings.ReplaceAll(e[0], "§§", "`")
}
// sanity check // sanity check
for _, e := range examples { for _, e := range examples {
if e[0] == "" { if e[0] == "" {

View file

@ -78,12 +78,22 @@ func init() {
[]string{"findRE"}, []string{"findRE"},
[][2]string{ [][2]string{
{ {
`{{ findRE "[G|g]o" "Hugo is a static side generator written in Go." "1" }}`, `{{ findRE "[G|g]o" "Hugo is a static side generator written in Go." 1 }}`,
`[go]`, `[go]`,
}, },
}, },
) )
ns.AddMethodMapping(ctx.FindRESubmatch,
[]string{"findRESubmatch"},
[][2]string{
{
`{{ findRESubmatch §§<a\s*href="(.+?)">(.+?)</a>§§ §§<li><a href="#foo">Foo</a></li> <li><a href="#bar">Bar</a></li>§§ | print | safeHTML }}`,
"[[<a href=\"#foo\">Foo</a> #foo Foo] [<a href=\"#bar\">Bar</a> #bar Bar]]",
},
},
)
ns.AddMethodMapping(ctx.HasPrefix, ns.AddMethodMapping(ctx.HasPrefix,
[]string{"hasPrefix"}, []string{"hasPrefix"},
[][2]string{ [][2]string{

View file

@ -45,6 +45,31 @@ func (ns *Namespace) FindRE(expr string, content any, limit ...any) ([]string, e
return re.FindAllString(conv, lim), nil return re.FindAllString(conv, lim), nil
} }
// FindRESubmatch returns returns a slice of strings holding the text of the leftmost match of the regular expression in s and the matches, if any, of its subexpressions.
//
// By default all matches will be included. The number of matches can be limited with the optional limit parameter. A return value of nil indicates no match.
func (ns *Namespace) FindRESubmatch(expr string, content any, limit ...any) ([][]string, error) {
re, err := reCache.Get(expr)
if err != nil {
return nil, err
}
conv, err := cast.ToStringE(content)
if err != nil {
return nil, err
}
n := -1
if len(limit) > 0 {
n, err = cast.ToIntE(limit[0])
if err != nil {
return nil, err
}
}
return re.FindAllStringSubmatch(conv, n), nil
}
// ReplaceRE returns a copy of s, replacing all matches of the regular // ReplaceRE returns a copy of s, replacing all matches of the regular
// expression pattern with the replacement text repl. The number of replacements // expression pattern with the replacement text repl. The number of replacements
// can be limited with an optional fourth parameter. // can be limited with an optional fourth parameter.

View file

@ -50,6 +50,39 @@ func TestFindRE(t *testing.T) {
} }
} }
func TestFindRESubmatch(t *testing.T) {
t.Parallel()
c := qt.New(t)
for _, test := range []struct {
expr string
content any
limit any
expect any
}{
{`<a\s*href="(.+?)">(.+?)</a>`, `<li><a href="#foo">Foo</a></li><li><a href="#bar">Bar</a></li>`, -1, [][]string{
{"<a href=\"#foo\">Foo</a>", "#foo", "Foo"},
{"<a href=\"#bar\">Bar</a>", "#bar", "Bar"},
}},
// Some simple cases.
{"([G|g]o)", "Hugo is a static site generator written in Go.", -1, [][]string{{"go", "go"}, {"Go", "Go"}}},
{"([G|g]o)", "Hugo is a static site generator written in Go.", 1, [][]string{{"go", "go"}}},
// errors
{"([G|go", "Hugo is a static site generator written in Go.", nil, false},
{"([G|g]o)", t, nil, false},
} {
result, err := ns.FindRESubmatch(test.expr, test.content, test.limit)
if b, ok := test.expect.(bool); ok && !b {
c.Assert(err, qt.Not(qt.IsNil))
continue
}
c.Assert(err, qt.IsNil)
c.Check(result, qt.DeepEquals, test.expect)
}
}
func TestReplaceRE(t *testing.T) { func TestReplaceRE(t *testing.T) {
t.Parallel() t.Parallel()
c := qt.New(t) c := qt.New(t)