cdle
80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
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)
|
|
})
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|