-
Notifications
You must be signed in to change notification settings - Fork 304
/
server.go
78 lines (69 loc) · 2.27 KB
/
server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package websocket
import (
"net/http"
"strconv"
"github.com/flutter-webrtc/flutter-webrtc-server/pkg/logger"
"github.com/gorilla/websocket"
)
type WebSocketServerConfig struct {
Host string
Port int
CertFile string
KeyFile string
HTMLRoot string
WebSocketPath string
TurnServerPath string
}
func DefaultConfig() WebSocketServerConfig {
return WebSocketServerConfig{
Host: "0.0.0.0",
Port: 8086,
HTMLRoot: "web",
WebSocketPath: "/ws",
TurnServerPath: "/api/turn",
}
}
type WebSocketServer struct {
handleWebSocket func(ws *WebSocketConn, request *http.Request)
handleTurnServer func(writer http.ResponseWriter, request *http.Request)
// Websocket upgrader
upgrader websocket.Upgrader
}
func NewWebSocketServer(
wsHandler func(ws *WebSocketConn, request *http.Request),
turnServerHandler func(writer http.ResponseWriter, request *http.Request)) *WebSocketServer {
var server = &WebSocketServer{
handleWebSocket: wsHandler,
handleTurnServer: turnServerHandler,
}
server.upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
}
return server
}
func (server *WebSocketServer) handleWebSocketRequest(writer http.ResponseWriter, request *http.Request) {
responseHeader := http.Header{}
//responseHeader.Add("Sec-WebSocket-Protocol", "protoo")
socket, err := server.upgrader.Upgrade(writer, request, responseHeader)
if err != nil {
logger.Panicf("%v", err)
}
wsTransport := NewWebSocketConn(socket)
server.handleWebSocket(wsTransport, request)
wsTransport.ReadMessage()
}
func (server *WebSocketServer) handleTurnServerRequest(writer http.ResponseWriter, request *http.Request) {
server.handleTurnServer(writer, request)
}
// Bind .
func (server *WebSocketServer) Bind(cfg WebSocketServerConfig) {
// Websocket handle func
http.HandleFunc(cfg.WebSocketPath, server.handleWebSocketRequest)
http.HandleFunc(cfg.TurnServerPath, server.handleTurnServerRequest)
http.Handle("/", http.FileServer(http.Dir(cfg.HTMLRoot)))
logger.Infof("Flutter WebRTC Server listening on: %s:%d", cfg.Host, cfg.Port)
// http.ListenAndServe(cfg.Host+":"+strconv.Itoa(cfg.Port), nil)
panic(http.ListenAndServeTLS(cfg.Host+":"+strconv.Itoa(cfg.Port), cfg.CertFile, cfg.KeyFile, nil))
}