34 lines
704 B
Go
34 lines
704 B
Go
package builtins
|
|
|
|
import (
|
|
"monkey/internal/context"
|
|
"monkey/internal/object"
|
|
"monkey/internal/typing"
|
|
"strings"
|
|
)
|
|
|
|
// Split ...
|
|
func Split(ctx context.Context, args ...object.Object) object.Object {
|
|
if err := typing.Check(
|
|
"split", args,
|
|
typing.RangeOfArgs(1, 2),
|
|
typing.WithTypes(object.StringType, object.StringType),
|
|
); err != nil {
|
|
return newError(err.Error())
|
|
}
|
|
|
|
var sep string
|
|
s := args[0].(object.String).Value
|
|
|
|
if len(args) == 2 {
|
|
sep = args[1].(object.String).Value
|
|
}
|
|
|
|
tokens := strings.Split(s, sep)
|
|
elements := make([]object.Object, len(tokens))
|
|
for i, token := range tokens {
|
|
elements[i] = object.String{Value: token}
|
|
}
|
|
return &object.Array{Elements: elements}
|
|
}
|