Update to tengo v2 (#976)

1 files changed, 146 insertions, 0 deletions

diff --git a/vendor/github.com/d5/tengo/v2/stdlib/json.go b/vendor/github.com/d5/tengo/v2/stdlib/json.go
new file mode 100644
index 00000000..be2185db
--- /dev/null
+++ b/vendor/github.com/d5/tengo/v2/stdlib/json.go
@@ -0,0 +1,146 @@
+package stdlib
+
+import (
+	"bytes"
+	gojson "encoding/json"
+
+	"github.com/d5/tengo/v2"
+	"github.com/d5/tengo/v2/stdlib/json"
+)
+
+var jsonModule = map[string]tengo.Object{
+	"decode": &tengo.UserFunction{
+		Name:  "decode",
+		Value: jsonDecode,
+	},
+	"encode": &tengo.UserFunction{
+		Name:  "encode",
+		Value: jsonEncode,
+	},
+	"indent": &tengo.UserFunction{
+		Name:  "encode",
+		Value: jsonIndent,
+	},
+	"html_escape": &tengo.UserFunction{
+		Name:  "html_escape",
+		Value: jsonHTMLEscape,
+	},
+}
+
+func jsonDecode(args ...tengo.Object) (ret tengo.Object, err error) {
+	if len(args) != 1 {
+		return nil, tengo.ErrWrongNumArguments
+	}
+
+	switch o := args[0].(type) {
+	case *tengo.Bytes:
+		v, err := json.Decode(o.Value)
+		if err != nil {
+			return &tengo.Error{
+				Value: &tengo.String{Value: err.Error()},
+			}, nil
+		}
+		return v, nil
+	case *tengo.String:
+		v, err := json.Decode([]byte(o.Value))
+		if err != nil {
+			return &tengo.Error{
+				Value: &tengo.String{Value: err.Error()},
+			}, nil
+		}
+		return v, nil
+	default:
+		return nil, tengo.ErrInvalidArgumentType{
+			Name:     "first",
+			Expected: "bytes/string",
+			Found:    args[0].TypeName(),
+		}
+	}
+}
+
+func jsonEncode(args ...tengo.Object) (ret tengo.Object, err error) {
+	if len(args) != 1 {
+		return nil, tengo.ErrWrongNumArguments
+	}
+
+	b, err := json.Encode(args[0])
+	if err != nil {
+		return &tengo.Error{Value: &tengo.String{Value: err.Error()}}, nil
+	}
+
+	return &tengo.Bytes{Value: b}, nil
+}
+
+func jsonIndent(args ...tengo.Object) (ret tengo.Object, err error) {
+	if len(args) != 3 {
+		return nil, tengo.ErrWrongNumArguments
+	}
+
+	prefix, ok := tengo.ToString(args[1])
+	if !ok {
+		return nil, tengo.ErrInvalidArgumentType{
+			Name:     "prefix",
+			Expected: "string(compatible)",
+			Found:    args[1].TypeName(),
+		}
+	}
+
+	indent, ok := tengo.ToString(args[2])
+	if !ok {
+		return nil, tengo.ErrInvalidArgumentType{
+			Name:     "indent",
+			Expected: "string(compatible)",
+			Found:    args[2].TypeName(),
+		}
+	}
+
+	switch o := args[0].(type) {
+	case *tengo.Bytes:
+		var dst bytes.Buffer
+		err := gojson.Indent(&dst, o.Value, prefix, indent)
+		if err != nil {
+			return &tengo.Error{
+				Value: &tengo.String{Value: err.Error()},
+			}, nil
+		}
+		return &tengo.Bytes{Value: dst.Bytes()}, nil
+	case *tengo.String:
+		var dst bytes.Buffer
+		err := gojson.Indent(&dst, []byte(o.Value), prefix, indent)
+		if err != nil {
+			return &tengo.Error{
+				Value: &tengo.String{Value: err.Error()},
+			}, nil
+		}
+		return &tengo.Bytes{Value: dst.Bytes()}, nil
+	default:
+		return nil, tengo.ErrInvalidArgumentType{
+			Name:     "first",
+			Expected: "bytes/string",
+			Found:    args[0].TypeName(),
+		}
+	}
+}
+
+func jsonHTMLEscape(args ...tengo.Object) (ret tengo.Object, err error) {
+	if len(args) != 1 {
+		return nil, tengo.ErrWrongNumArguments
+	}
+
+	switch o := args[0].(type) {
+	case *tengo.Bytes:
+		var dst bytes.Buffer
+		gojson.HTMLEscape(&dst, o.Value)
+		return &tengo.Bytes{Value: dst.Bytes()}, nil
+	case *tengo.String:
+		var dst bytes.Buffer
+		gojson.HTMLEscape(&dst, []byte(o.Value))
+		return &tengo.Bytes{Value: dst.Bytes()}, nil
+	default:
+		return nil, tengo.ErrInvalidArgumentType{
+			Name:     "first",
+			Expected: "bytes/string",
+			Found:    args[0].TypeName(),
+		}
+	}
+}