mirror of
https://github.com/mattermost/mattermost.git
synced 2026-02-03 20:40:00 -05:00
60 lines
1.9 KiB
Go
60 lines
1.9 KiB
Go
// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package wsapi
|
|
|
|
import (
|
|
l4g "github.com/alecthomas/log4go"
|
|
|
|
"net/http"
|
|
|
|
"github.com/mattermost/mattermost-server/app"
|
|
"github.com/mattermost/mattermost-server/model"
|
|
"github.com/mattermost/mattermost-server/utils"
|
|
)
|
|
|
|
func ApiWebSocketHandler(wh func(*model.WebSocketRequest) (map[string]interface{}, *model.AppError)) webSocketHandler {
|
|
return webSocketHandler{wh}
|
|
}
|
|
|
|
type webSocketHandler struct {
|
|
handlerFunc func(*model.WebSocketRequest) (map[string]interface{}, *model.AppError)
|
|
}
|
|
|
|
func (wh webSocketHandler) ServeWebSocket(conn *app.WebConn, r *model.WebSocketRequest) {
|
|
l4g.Debug("/api/v3/users/websocket:%s", r.Action)
|
|
|
|
session, sessionErr := app.Global().GetSession(conn.GetSessionToken())
|
|
if sessionErr != nil {
|
|
l4g.Error(utils.T("api.web_socket_handler.log.error"), "/api/v3/users/websocket", r.Action, r.Seq, conn.UserId, sessionErr.SystemMessage(utils.T), sessionErr.Error())
|
|
sessionErr.DetailedError = ""
|
|
errResp := model.NewWebSocketError(r.Seq, sessionErr)
|
|
|
|
conn.Send <- errResp
|
|
return
|
|
}
|
|
|
|
r.Session = *session
|
|
r.T = conn.T
|
|
r.Locale = conn.Locale
|
|
|
|
var data map[string]interface{}
|
|
var err *model.AppError
|
|
|
|
if data, err = wh.handlerFunc(r); err != nil {
|
|
l4g.Error(utils.T("api.web_socket_handler.log.error"), "/api/v3/users/websocket", r.Action, r.Seq, r.Session.UserId, err.SystemMessage(utils.T), err.DetailedError)
|
|
err.DetailedError = ""
|
|
errResp := model.NewWebSocketError(r.Seq, err)
|
|
|
|
conn.Send <- errResp
|
|
return
|
|
}
|
|
|
|
resp := model.NewWebSocketResponse(model.STATUS_OK, r.Seq, data)
|
|
|
|
conn.Send <- resp
|
|
}
|
|
|
|
func NewInvalidWebSocketParamError(action string, name string) *model.AppError {
|
|
return model.NewAppError("/api/v3/users/websocket:"+action, "api.websocket_handler.invalid_param.app_error", map[string]interface{}{"Name": name}, "", http.StatusBadRequest)
|
|
}
|