1.4.5.2

EdgeNode/internal/nodes/http_request_events_script_plus.go

@@ -0,0 +1,124 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build script
+// +build script
+
+package nodes
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
+	"github.com/iwind/TeaGo/maps"
+	"github.com/iwind/TeaGo/types"
+	"rogchap.com/v8go"
+	"strings"
+)
+
+func (this *HTTPRequest) onInit() {
+	this.fireScriptEvent("init")
+}
+
+func (this *HTTPRequest) onRequest() {
+	if this.isDone || this.writer.isFinished {
+		return
+	}
+	this.fireScriptEvent("request")
+}
+
+// 触发事件
+func (this *HTTPRequest) fireScriptEvent(eventType string) {
+	if SharedJSPool == nil {
+		return
+	}
+
+	if this.web.RequestScripts == nil {
+		return
+	}
+
+	var group *serverconfigs.ScriptGroupConfig
+	if eventType == "init" && this.web.RequestScripts.InitGroup != nil && this.web.RequestScripts.InitGroup.IsOn {
+		group = this.web.RequestScripts.InitGroup
+	} else if eventType == "request" && this.web.RequestScripts.RequestGroup != nil && this.web.RequestScripts.RequestGroup.IsOn {
+		group = this.web.RequestScripts.RequestGroup
+	}
+
+	if group == nil {
+		return
+	}
+
+	for _, script := range group.Scripts {
+		if this.isDone {
+			return
+		}
+
+		if !script.IsOn {
+			continue
+		}
+
+		if len(script.RealCode()) > 0 {
+			this.runScript(eventType, script.RealCode())
+		}
+	}
+}
+
+func (this *HTTPRequest) runScript(eventType string, script string) {
+	ctx, err := SharedJSPool.GetContext()
+	if err != nil {
+		remotelogs.Error("SCRIPT", "get context failed: "+err.Error())
+		return
+	}
+	ctx.SetServerId(this.ReqServer.Id)
+	defer SharedJSPool.PutContext(ctx)
+
+	var reqObjectId = ctx.AddGoRequestObject(this)
+	var respObjectId = ctx.AddGoResponseObject(this.writer)
+
+	script = `(function () {
+	let req = new gojs.net.http.Request()
+	req.setGoObject(` + types.String(reqObjectId) + `)
+
+	let resp = new gojs.net.http.Response()
+	resp.setGoObject(` + types.String(respObjectId) + `)
+` + script + `
+})()`
+	_, err = ctx.Run(script, "request."+eventType+".js")
+	if err != nil {
+		var errString = ""
+		jsErr, ok := err.(*v8go.JSError)
+		if ok {
+			var pieces = strings.Split(jsErr.Location, ":")
+			if len(pieces) < 3 {
+				errString = err.Error()
+			} else {
+				if strings.HasPrefix(pieces[0], "gojs") {
+					var line = types.Int(pieces[len(pieces)-2])
+					var scriptLine = ctx.ReadLineFromLibrary(pieces[0], line)
+					if len(scriptLine) > 0 {
+						if len(scriptLine) > 256 {
+							scriptLine = scriptLine[:256] + "..."
+						}
+
+						errString = jsErr.Error() + ", location: " + strings.Join(pieces[:len(pieces)-2], ":") + ":" + types.String(line) + ":" + pieces[len(pieces)-1] + ", code: " + scriptLine
+					}
+				}
+				if len(errString) == 0 {
+					var line = types.Int(pieces[len(pieces)-2])
+					var scriptLines = strings.Split(script, "\n")
+					var scriptLine = ""
+					if len(scriptLines) > line-1 {
+						scriptLine = scriptLines[line-1]
+						if len(scriptLine) > 256 {
+							scriptLine = scriptLine[:256] + "..."
+						}
+					}
+					line -= 6 /* 6是req和resp构造用的行数 */
+					errString = jsErr.Error() + ", location: " + strings.Join(pieces[:len(pieces)-2], ":") + ":" + types.String(line) + ":" + pieces[len(pieces)-1] + ", code: " + scriptLine
+				}
+			}
+		} else {
+			errString = err.Error()
+		}
+		remotelogs.ServerError(this.ReqServer.Id, "SCRIPT", "run "+eventType+" script failed: "+errString, nodeconfigs.NodeLogTypeRunScriptFailed, maps.Map{})
+		return
+	}
+}