1 files changed, 96 insertions, 0 deletions
diff --git a/app/websocket_router.go b/app/websocket_router.go
new file mode 100644
index 000000000..984b9d17e
--- /dev/null
+++ b/app/websocket_router.go
@@ -0,0 +1,96 @@
+// Copyright (c) 2016 Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package app
+
+import (
+	l4g "github.com/alecthomas/log4go"
+
+	"github.com/mattermost/platform/model"
+	"github.com/mattermost/platform/utils"
+)
+
+type webSocketHandler interface {
+	ServeWebSocket(*WebConn, *model.WebSocketRequest)
+}
+
+type WebSocketRouter struct {
+	handlers map[string]webSocketHandler
+}
+
+func NewWebSocketRouter() *WebSocketRouter {
+	router := &WebSocketRouter{}
+	router.handlers = make(map[string]webSocketHandler)
+	return router
+}
+
+func (wr *WebSocketRouter) Handle(action string, handler webSocketHandler) {
+	wr.handlers[action] = handler
+}
+
+func (wr *WebSocketRouter) ServeWebSocket(conn *WebConn, r *model.WebSocketRequest) {
+	if r.Action == "" {
+		err := model.NewLocAppError("ServeWebSocket", "api.web_socket_router.no_action.app_error", nil, "")
+		ReturnWebSocketError(conn, r, err)
+		return
+	}
+
+	if r.Seq <= 0 {
+		err := model.NewLocAppError("ServeWebSocket", "api.web_socket_router.bad_seq.app_error", nil, "")
+		ReturnWebSocketError(conn, r, err)
+		return
+	}
+
+	if r.Action == model.WEBSOCKET_AUTHENTICATION_CHALLENGE {
+		token, ok := r.Data["token"].(string)
+		if !ok {
+			conn.WebSocket.Close()
+			return
+		}
+
+		session, err := GetSession(token)
+
+		if err != nil {
+			conn.WebSocket.Close()
+		} else {
+			go SetStatusOnline(session.UserId, session.Id, false)
+
+			conn.SessionToken = session.Token
+			conn.UserId = session.UserId
+
+			resp := model.NewWebSocketResponse(model.STATUS_OK, r.Seq, nil)
+			resp.DoPreComputeJson()
+			conn.Send <- resp
+			conn.SendHello()
+		}
+
+		return
+	}
+
+	if !conn.IsAuthenticated() {
+		err := model.NewLocAppError("ServeWebSocket", "api.web_socket_router.not_authenticated.app_error", nil, "")
+		ReturnWebSocketError(conn, r, err)
+		return
+	}
+
+	var handler webSocketHandler
+	if h, ok := wr.handlers[r.Action]; !ok {
+		err := model.NewLocAppError("ServeWebSocket", "api.web_socket_router.bad_action.app_error", nil, "")
+		ReturnWebSocketError(conn, r, err)
+		return
+	} else {
+		handler = h
+	}
+
+	handler.ServeWebSocket(conn, r)
+}
+
+func ReturnWebSocketError(conn *WebConn, r *model.WebSocketRequest, err *model.AppError) {
+	l4g.Error(utils.T("api.web_socket_router.log.error"), r.Seq, conn.UserId, err.SystemMessage(utils.T), err.DetailedError)
+
+	err.DetailedError = ""
+	errorResp := model.NewWebSocketError(r.Seq, err)
+	errorResp.DoPreComputeJson()
+
+	conn.Send <- errorResp
+}