1.4.5.2

EdgeNode/internal/js/lib_base64.go

@@ -0,0 +1,64 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build script
+// +build script
+
+package js
+
+import (
+	"encoding/base64"
+	teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
+)
+
+func init() {
+	if !teaconst.IsMain {
+		return
+	}
+
+	SharedLibraryManager.Register(&JSBase64Library{})
+}
+
+type JSBase64Library struct {
+	JSBaseLibrary
+}
+
+func (this *JSBase64Library) JSNamespace() string {
+	return "gojs.base64"
+}
+
+func (this *JSBase64Library) JSPrototype() string {
+	return `gojs.base64.encode = function (data) {
+	return $this.Encode(data)
+}
+
+gojs.base64.decode = function (data) {
+	return $this.Decode(data)
+}
+`
+}
+
+func (this *JSBase64Library) Encode(arguments *FunctionArguments) (string, error) {
+	s, ok := arguments.StringAt(0)
+	if ok {
+		return base64.StdEncoding.EncodeToString([]byte(s)), nil
+	}
+
+	b, ok := arguments.BytesAt(0)
+	if ok {
+		return base64.StdEncoding.EncodeToString(b), nil
+	}
+
+	return "", nil
+}
+
+func (this *JSBase64Library) Decode(arguments *FunctionArguments) (string, error) {
+	s, ok := arguments.StringAt(0)
+	if ok {
+		result, err := base64.StdEncoding.DecodeString(s)
+		if err != nil {
+			return "", err
+		}
+		return string(result), nil
+	}
+
+	return "", nil
+}