tpl: Add strings.Repeat

This commit is contained in:
David E. Wheeler
2018-06-03 02:55:37 -04:00
committed by Bjørn Erik Pedersen
parent 07b96d16e8
commit 13435a6f60
5 changed files with 109 additions and 0 deletions

View File

@@ -158,6 +158,13 @@ func init() {
},
)
ns.AddMethodMapping(ctx.Repeat,
nil,
[][2]string{
{`{{ "yo" | strings.Repeat 4 }}`, `yoyoyoyo`},
},
)
ns.AddMethodMapping(ctx.ToUpper,
[]string{"upper"},
[][2]string{

View File

@@ -17,6 +17,7 @@ import (
"errors"
"fmt"
"html/template"
"math"
_strings "strings"
"unicode/utf8"
@@ -417,3 +418,23 @@ func (ns *Namespace) TrimSuffix(suffix, s interface{}) (string, error) {
return _strings.TrimSuffix(ss, sx), nil
}
// Repeat returns a new string consisting of count copies of the string s.
// The count is limited to an in16 value (up to 32767).
func (ns *Namespace) Repeat(n, s interface{}) (string, error) {
ss, err := cast.ToStringE(s)
if err != nil {
return "", err
}
sn, err := cast.ToIntE(n)
if err != nil {
return "", err
}
if sn > math.MaxInt16 {
return "", fmt.Errorf("Cannot repeat string more than %d times", math.MaxInt16)
}
return _strings.Repeat(ss, sn), nil
}

View File

@@ -16,6 +16,7 @@ package strings
import (
"fmt"
"html/template"
"math"
"testing"
"github.com/gohugoio/hugo/deps"
@@ -709,3 +710,38 @@ func TestTrimSuffix(t *testing.T) {
assert.Equal(t, test.expect, result, errMsg)
}
}
func TestRepeat(t *testing.T) {
t.Parallel()
for i, test := range []struct {
s interface{}
n interface{}
expect interface{}
}{
{"yo", "2", "yoyo"},
{"~", "16", "~~~~~~~~~~~~~~~~"},
{"<tag>", "0", ""},
{"yay", "1", "yay"},
{1221, "1", "1221"},
{1221, 2, "12211221"},
{template.HTML("<tag>"), "2", "<tag><tag>"},
{[]byte("<tag>"), 2, "<tag><tag>"},
// errors
{"", tstNoStringer{}, false},
{tstNoStringer{}, "", false},
{"hi", math.MaxInt16 + 1, false},
} {
errMsg := fmt.Sprintf("[%d] %v", i, test)
result, err := ns.Repeat(test.n, test.s)
if b, ok := test.expect.(bool); ok && !b {
require.Error(t, err, errMsg)
continue
}
require.NoError(t, err, errMsg)
assert.Equal(t, test.expect, result, errMsg)
}
}