You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
252 lines
5.3 KiB
Go
252 lines
5.3 KiB
Go
6 years ago
|
package stdlib
|
||
|
|
||
|
import (
|
||
|
"regexp"
|
||
|
|
||
5 years ago
|
"github.com/d5/tengo/v2"
|
||
6 years ago
|
)
|
||
|
|
||
5 years ago
|
func makeTextRegexp(re *regexp.Regexp) *tengo.ImmutableMap {
|
||
|
return &tengo.ImmutableMap{
|
||
|
Value: map[string]tengo.Object{
|
||
6 years ago
|
// match(text) => bool
|
||
5 years ago
|
"match": &tengo.UserFunction{
|
||
|
Value: func(args ...tengo.Object) (
|
||
|
ret tengo.Object,
|
||
|
err error,
|
||
|
) {
|
||
6 years ago
|
if len(args) != 1 {
|
||
5 years ago
|
err = tengo.ErrWrongNumArguments
|
||
6 years ago
|
return
|
||
|
}
|
||
|
|
||
5 years ago
|
s1, ok := tengo.ToString(args[0])
|
||
6 years ago
|
if !ok {
|
||
5 years ago
|
err = tengo.ErrInvalidArgumentType{
|
||
6 years ago
|
Name: "first",
|
||
|
Expected: "string(compatible)",
|
||
|
Found: args[0].TypeName(),
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if re.MatchString(s1) {
|
||
5 years ago
|
ret = tengo.TrueValue
|
||
6 years ago
|
} else {
|
||
5 years ago
|
ret = tengo.FalseValue
|
||
6 years ago
|
}
|
||
|
|
||
|
return
|
||
|
},
|
||
|
},
|
||
|
|
||
|
// find(text) => array(array({text:,begin:,end:}))/undefined
|
||
|
// find(text, maxCount) => array(array({text:,begin:,end:}))/undefined
|
||
5 years ago
|
"find": &tengo.UserFunction{
|
||
|
Value: func(args ...tengo.Object) (
|
||
|
ret tengo.Object,
|
||
|
err error,
|
||
|
) {
|
||
6 years ago
|
numArgs := len(args)
|
||
|
if numArgs != 1 && numArgs != 2 {
|
||
5 years ago
|
err = tengo.ErrWrongNumArguments
|
||
6 years ago
|
return
|
||
|
}
|
||
|
|
||
5 years ago
|
s1, ok := tengo.ToString(args[0])
|
||
6 years ago
|
if !ok {
|
||
5 years ago
|
err = tengo.ErrInvalidArgumentType{
|
||
6 years ago
|
Name: "first",
|
||
|
Expected: "string(compatible)",
|
||
|
Found: args[0].TypeName(),
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if numArgs == 1 {
|
||
|
m := re.FindStringSubmatchIndex(s1)
|
||
|
if m == nil {
|
||
5 years ago
|
ret = tengo.UndefinedValue
|
||
6 years ago
|
return
|
||
|
}
|
||
|
|
||
5 years ago
|
arr := &tengo.Array{}
|
||
6 years ago
|
for i := 0; i < len(m); i += 2 {
|
||
5 years ago
|
arr.Value = append(arr.Value,
|
||
|
&tengo.ImmutableMap{
|
||
|
Value: map[string]tengo.Object{
|
||
|
"text": &tengo.String{
|
||
|
Value: s1[m[i]:m[i+1]],
|
||
|
},
|
||
|
"begin": &tengo.Int{
|
||
|
Value: int64(m[i]),
|
||
|
},
|
||
|
"end": &tengo.Int{
|
||
|
Value: int64(m[i+1]),
|
||
|
},
|
||
|
}})
|
||
6 years ago
|
}
|
||
|
|
||
5 years ago
|
ret = &tengo.Array{Value: []tengo.Object{arr}}
|
||
6 years ago
|
|
||
|
return
|
||
|
}
|
||
|
|
||
5 years ago
|
i2, ok := tengo.ToInt(args[1])
|
||
6 years ago
|
if !ok {
|
||
5 years ago
|
err = tengo.ErrInvalidArgumentType{
|
||
6 years ago
|
Name: "second",
|
||
|
Expected: "int(compatible)",
|
||
|
Found: args[1].TypeName(),
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
m := re.FindAllStringSubmatchIndex(s1, i2)
|
||
|
if m == nil {
|
||
5 years ago
|
ret = tengo.UndefinedValue
|
||
6 years ago
|
return
|
||
|
}
|
||
|
|
||
5 years ago
|
arr := &tengo.Array{}
|
||
6 years ago
|
for _, m := range m {
|
||
5 years ago
|
subMatch := &tengo.Array{}
|
||
6 years ago
|
for i := 0; i < len(m); i += 2 {
|
||
5 years ago
|
subMatch.Value = append(subMatch.Value,
|
||
|
&tengo.ImmutableMap{
|
||
|
Value: map[string]tengo.Object{
|
||
|
"text": &tengo.String{
|
||
|
Value: s1[m[i]:m[i+1]],
|
||
|
},
|
||
|
"begin": &tengo.Int{
|
||
|
Value: int64(m[i]),
|
||
|
},
|
||
|
"end": &tengo.Int{
|
||
|
Value: int64(m[i+1]),
|
||
|
},
|
||
|
}})
|
||
6 years ago
|
}
|
||
|
|
||
|
arr.Value = append(arr.Value, subMatch)
|
||
|
}
|
||
|
|
||
|
ret = arr
|
||
|
|
||
|
return
|
||
|
},
|
||
|
},
|
||
|
|
||
|
// replace(src, repl) => string
|
||
5 years ago
|
"replace": &tengo.UserFunction{
|
||
|
Value: func(args ...tengo.Object) (
|
||
|
ret tengo.Object,
|
||
|
err error,
|
||
|
) {
|
||
6 years ago
|
if len(args) != 2 {
|
||
5 years ago
|
err = tengo.ErrWrongNumArguments
|
||
6 years ago
|
return
|
||
|
}
|
||
|
|
||
5 years ago
|
s1, ok := tengo.ToString(args[0])
|
||
6 years ago
|
if !ok {
|
||
5 years ago
|
err = tengo.ErrInvalidArgumentType{
|
||
6 years ago
|
Name: "first",
|
||
|
Expected: "string(compatible)",
|
||
|
Found: args[0].TypeName(),
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
5 years ago
|
s2, ok := tengo.ToString(args[1])
|
||
6 years ago
|
if !ok {
|
||
5 years ago
|
err = tengo.ErrInvalidArgumentType{
|
||
6 years ago
|
Name: "second",
|
||
|
Expected: "string(compatible)",
|
||
|
Found: args[1].TypeName(),
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s, ok := doTextRegexpReplace(re, s1, s2)
|
||
|
if !ok {
|
||
5 years ago
|
return nil, tengo.ErrStringLimit
|
||
6 years ago
|
}
|
||
|
|
||
5 years ago
|
ret = &tengo.String{Value: s}
|
||
6 years ago
|
|
||
|
return
|
||
|
},
|
||
|
},
|
||
|
|
||
|
// split(text) => array(string)
|
||
|
// split(text, maxCount) => array(string)
|
||
5 years ago
|
"split": &tengo.UserFunction{
|
||
|
Value: func(args ...tengo.Object) (
|
||
|
ret tengo.Object,
|
||
|
err error,
|
||
|
) {
|
||
6 years ago
|
numArgs := len(args)
|
||
|
if numArgs != 1 && numArgs != 2 {
|
||
5 years ago
|
err = tengo.ErrWrongNumArguments
|
||
6 years ago
|
return
|
||
|
}
|
||
|
|
||
5 years ago
|
s1, ok := tengo.ToString(args[0])
|
||
6 years ago
|
if !ok {
|
||
5 years ago
|
err = tengo.ErrInvalidArgumentType{
|
||
6 years ago
|
Name: "first",
|
||
|
Expected: "string(compatible)",
|
||
|
Found: args[0].TypeName(),
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
|
var i2 = -1
|
||
|
if numArgs > 1 {
|
||
5 years ago
|
i2, ok = tengo.ToInt(args[1])
|
||
6 years ago
|
if !ok {
|
||
5 years ago
|
err = tengo.ErrInvalidArgumentType{
|
||
6 years ago
|
Name: "second",
|
||
|
Expected: "int(compatible)",
|
||
|
Found: args[1].TypeName(),
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
|
||
5 years ago
|
arr := &tengo.Array{}
|
||
6 years ago
|
for _, s := range re.Split(s1, i2) {
|
||
5 years ago
|
arr.Value = append(arr.Value,
|
||
|
&tengo.String{Value: s})
|
||
6 years ago
|
}
|
||
|
|
||
|
ret = arr
|
||
|
|
||
|
return
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Size-limit checking implementation of regexp.ReplaceAllString.
|
||
|
func doTextRegexpReplace(re *regexp.Regexp, src, repl string) (string, bool) {
|
||
|
idx := 0
|
||
|
out := ""
|
||
|
for _, m := range re.FindAllStringSubmatchIndex(src, -1) {
|
||
|
var exp []byte
|
||
|
exp = re.ExpandString(exp, repl, src, m)
|
||
|
if len(out)+m[0]-idx+len(exp) > tengo.MaxStringLen {
|
||
|
return "", false
|
||
|
}
|
||
|
out += src[idx:m[0]] + string(exp)
|
||
|
idx = m[1]
|
||
|
}
|
||
|
if idx < len(src) {
|
||
|
if len(out)+len(src)-idx > tengo.MaxStringLen {
|
||
|
return "", false
|
||
|
}
|
||
|
out += src[idx:]
|
||
|
}
|
||
5 years ago
|
return out, true
|
||
6 years ago
|
}
|