Skip to content

Commit 386ff5c

Browse files
committed
Allow trim* func to take only one arg
1 parent 31a57fa commit 386ff5c

File tree

1 file changed

+18
-4
lines changed

1 file changed

+18
-4
lines changed

builtin/builtin.go

Lines changed: 18 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -122,16 +122,30 @@ var Functions = []*Function{
122122
{
123123
Name: "trimPrefix",
124124
Func: func(args ...interface{}) (interface{}, error) {
125-
return strings.TrimPrefix(args[0].(string), args[1].(string)), nil
125+
s := " "
126+
if len(args) == 2 {
127+
s = args[1].(string)
128+
}
129+
return strings.TrimPrefix(args[0].(string), s), nil
126130
},
127-
Types: types(strings.TrimPrefix),
131+
Types: types(
132+
strings.TrimPrefix,
133+
new(func(string) string),
134+
),
128135
},
129136
{
130137
Name: "trimSuffix",
131138
Func: func(args ...interface{}) (interface{}, error) {
132-
return strings.TrimSuffix(args[0].(string), args[1].(string)), nil
139+
s := " "
140+
if len(args) == 2 {
141+
s = args[1].(string)
142+
}
143+
return strings.TrimSuffix(args[0].(string), s), nil
133144
},
134-
Types: types(strings.TrimSuffix),
145+
Types: types(
146+
strings.TrimSuffix,
147+
new(func(string) string),
148+
),
135149
},
136150
{
137151
Name: "upper",

0 commit comments

Comments
 (0)