91 lines
1.8 KiB
Go
91 lines
1.8 KiB
Go
package handler
|
|
|
|
import (
|
|
"github.com/gorilla/websocket"
|
|
"github.com/loveuer/nf"
|
|
"github.com/loveuer/nf/nft/log"
|
|
"github.com/loveuer/nf/nft/resp"
|
|
"github.com/loveuer/ushare/internal/controller"
|
|
"net/http"
|
|
)
|
|
|
|
func LocalClients() nf.HandlerFunc {
|
|
return func(c *nf.Ctx) error {
|
|
list := controller.RoomController.List()
|
|
|
|
return resp.Resp200(c, list)
|
|
}
|
|
}
|
|
|
|
func LocalWS() nf.HandlerFunc {
|
|
upgrader := websocket.Upgrader{
|
|
ReadBufferSize: 1024,
|
|
WriteBufferSize: 1024,
|
|
CheckOrigin: func(r *http.Request) bool {
|
|
return true
|
|
},
|
|
}
|
|
|
|
return func(c *nf.Ctx) error {
|
|
var (
|
|
ip = c.IP(true)
|
|
ua = c.Get("User-Agent")
|
|
)
|
|
|
|
conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
|
|
if err != nil {
|
|
log.Error("LocalWS: failed to upgrade websocket connection, err = %s", err.Error())
|
|
return err
|
|
}
|
|
|
|
controller.RoomController.Register(conn, ip, ua)
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func LocalOffer() nf.HandlerFunc {
|
|
return func(c *nf.Ctx) error {
|
|
type Req struct {
|
|
Id string `json:"id"`
|
|
Offer *controller.RoomOffer `json:"offer"`
|
|
Candidate *controller.RoomCandidate `json:"candidate"`
|
|
}
|
|
|
|
var (
|
|
err error
|
|
req = new(Req)
|
|
)
|
|
|
|
if err = c.BodyParser(req); err != nil {
|
|
return c.Status(http.StatusBadRequest).JSON(map[string]string{"err": err.Error()})
|
|
}
|
|
|
|
controller.RoomController.Offer(req.Id, req.Offer, req.Candidate)
|
|
|
|
return resp.Resp200(c, req.Offer)
|
|
}
|
|
}
|
|
|
|
func LocalAnswer() nf.HandlerFunc {
|
|
return func(c *nf.Ctx) error {
|
|
type Req struct {
|
|
Id string `json:"id"`
|
|
Answer *controller.RoomOffer `json:"answer"`
|
|
}
|
|
|
|
var (
|
|
err error
|
|
req = new(Req)
|
|
)
|
|
|
|
if err = c.BodyParser(req); err != nil {
|
|
return c.Status(http.StatusBadRequest).JSON(map[string]string{"err": err.Error()})
|
|
}
|
|
|
|
controller.RoomController.Answer(req.Id, req.Answer)
|
|
|
|
return resp.Resp200(c, req)
|
|
}
|
|
}
|