electronwall/rules/rules.go

package rules

import (
	"fmt"
	"log"
	"os"

	"github.com/callebtc/electronwall/types"
	"github.com/dop251/goja"
)

func Apply(s interface{}, decision_chan chan bool) (accept bool, err error) {
	vm := goja.New()

	var js_script []byte

	// load script according to event type
	switch s.(type) {
	case types.HtlcForwardEvent:
		vm.Set("HtlcForward", s)
		js_script, err = os.ReadFile("rules/HtlcForward.js")
		if err != nil {
			log.Fatal(err)
		}
	case types.ChannelAcceptEvent:
		vm.Set("ChannelAccept", s)
		js_script, err = os.ReadFile("rules/ChannelAccept.js")
		if err != nil {
			log.Fatal(err)
		}
	// case *routerrpc.ForwardHtlcInterceptRequest:
	// 	vm.Set("HtlcForwardEvent", s)
	// 	js_script, err = os.ReadFile("rules/ForwardHtlcInterceptRequest.js")
	// 	if err != nil {
	// 		log.Fatal(err)
	// 	}
	default:
		return false, fmt.Errorf("no rule found for event type")
	}

	// execute script
	v, err := vm.RunString(string(js_script))
	if err != nil {
		fmt.Print(err.Error())
		return
	}

	accept = v.Export().(bool)
	decision_chan <- accept
	return accept, nil
}