package nodes

import (
	"github.com/dop251/goja"
	"github.com/sirupsen/logrus"
	"pandax/pkg/rule_engine/message"
)

type ScriptEngine interface {
	ScriptOnMessage(msg message.Message, script string) (message.Message, error)
	//used by filter_switch_node
	ScriptOnSwitch(msg message.Message, script string) ([]string, error)
	//used by filter_script_node
	ScriptOnFilter(msg message.Message, script string) (bool, error)
	ScriptToString(msg message.Message, script string) (string, error)
}

type baseScriptEngine struct {
}

func NewScriptEngine() ScriptEngine {
	return &baseScriptEngine{}
}

func (bse *baseScriptEngine) ScriptOnMessage(msg message.Message, script string) (message.Message, error) {
	vm := goja.New()
	_, err := vm.RunString(script)
	if err != nil {
		logrus.Info("JS代码有问题")
		return nil, err
	}
	var fn func(map[string]interface{}, map[string]interface{}, string) map[string]interface{}
	err = vm.ExportTo(vm.Get("Transform"), &fn)
	if err != nil {
		logrus.Info("Js函数映射到 Go 函数失败！")
		return nil, err
	}
	datas := fn(msg.GetMsg(), msg.GetMetadata().GetValues(), msg.GetType())
	msg.SetMsg(datas["msg"].(map[string]interface{}))
	msg.SetMetadata(message.NewDefaultMetadata(datas["metadata"].(map[string]interface{})))
	msg.SetType(datas["msgType"].(string))
	return msg, nil

	return nil, nil
}

func (bse *baseScriptEngine) ScriptOnSwitch(msg message.Message, script string) ([]string, error) {
	vm := goja.New()
	_, err := vm.RunString(script)
	if err != nil {
		logrus.Info("JS代码有问题")
		return nil, err
	}
	var fn func(map[string]interface{}, map[string]interface{}, string) []string
	err = vm.ExportTo(vm.Get("Switch"), &fn)
	if err != nil {
		logrus.Info("Js函数映射到 Go 函数失败！")
		return nil, err
	}
	datas := fn(msg.GetMsg(), msg.GetMetadata().GetValues(), msg.GetType())
	return datas, nil
}

func (bse *baseScriptEngine) ScriptOnFilter(msg message.Message, script string) (bool, error) {
	vm := goja.New()
	_, err := vm.RunString(script)
	if err != nil {
		logrus.Info("JS代码有问题")
		return false, err
	}
	var fn func(map[string]interface{}, map[string]interface{}, string) bool
	err = vm.ExportTo(vm.Get("Filter"), &fn)
	if err != nil {
		logrus.Info("Js函数映射到 Go 函数失败！")
		return false, err
	}
	datas := fn(msg.GetMsg(), msg.GetMetadata().GetValues(), msg.GetType())
	return datas, nil
}

func (bse *baseScriptEngine) ScriptToString(msg message.Message, script string) (string, error) {

	return "", nil
}