x

core/websocket.go

@@ -0,0 +1,79 @@
+package core
+
+import (
+	"net/http"
+	"strings"
+
+	"github.com/dop251/goja"
+	"github.com/gin-gonic/gin"
+	"github.com/gorilla/websocket"
+)
+
+func IsWebSocketRequest(req *http.Request) bool {
+	if req.Header.Get("Upgrade") != "websocket" {
+		return false
+	}
+	if !websocket.IsWebSocketUpgrade(req) {
+		return false
+	}
+	return true
+}
+
+func handleWebsocket(c *gin.Context) {
+	for _, function := range Functions {
+		if function.Http != nil {
+			path := function.Http.Path
+			method := function.Http.Method
+			if c.Request.URL.Path == path && strings.HasPrefix(method, "W") {
+				// connect
+				var req = &Request{
+					c: c,
+					// uuid: uuid,
+				}
+				var res = &Response{
+					c: c,
+				}
+				req._event = "connect"
+				function.Handle(&Faker{
+					Type: "*",
+				}, func(vm *goja.Runtime) {
+					vm.Set("res", res)
+					vm.Set("req", req)
+				})
+				// message
+				var upGrader = websocket.Upgrader{
+					CheckOrigin: func(r *http.Request) bool {
+						return true
+					},
+				}
+				ws, err := upGrader.Upgrade(c.Writer, c.Request, nil)
+				if err != nil {
+					c.Writer.Write([]byte(err.Error()))
+					return
+				}
+				for {
+					_, data, err := ws.ReadMessage()
+					if err != nil { // disconnect
+						req._event = "disconnect"
+						function.Handle(&Faker{
+							Type: "*",
+						}, func(vm *goja.Runtime) {
+							vm.Set("res", res)
+							vm.Set("req", req)
+						})
+						ws.Close()
+						break
+					}
+					req.bodyData = data
+					req._event = "message"
+					function.Handle(&Faker{
+						Type: "*",
+					}, func(vm *goja.Runtime) {
+						vm.Set("res", res)
+						vm.Set("req", req)
+					})
+				}
+			}
+		}
+	}
+}