72 lines
1.9 KiB
Go
72 lines
1.9 KiB
Go
package ws
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
mmModel "github.com/mattermost/mattermost-server/v6/model"
|
|
)
|
|
|
|
type ClusterMessage struct {
|
|
TeamID string
|
|
BoardID string
|
|
Payload map[string]interface{}
|
|
EnsureUsers []string
|
|
}
|
|
|
|
//nolint:unparam // the `id` param is to key this function generic and handle more than just websocket messages
|
|
func (pa *PluginAdapter) sendMessageToCluster(id string, clusterMessage *ClusterMessage) {
|
|
b, err := json.Marshal(clusterMessage)
|
|
if err != nil {
|
|
pa.api.LogError("couldn't get JSON bytes from cluster message",
|
|
"id", id,
|
|
"err", err,
|
|
)
|
|
return
|
|
}
|
|
|
|
event := mmModel.PluginClusterEvent{Id: id, Data: b}
|
|
opts := mmModel.PluginClusterEventSendOptions{
|
|
SendType: mmModel.PluginClusterEventSendTypeReliable,
|
|
}
|
|
|
|
if err := pa.api.PublishPluginClusterEvent(event, opts); err != nil {
|
|
pa.api.LogError("error publishing cluster event",
|
|
"id", id,
|
|
"err", err,
|
|
)
|
|
}
|
|
}
|
|
|
|
func (pa *PluginAdapter) HandleClusterEvent(ev mmModel.PluginClusterEvent) {
|
|
pa.api.LogDebug("received cluster event", "id", ev.Id)
|
|
|
|
var clusterMessage ClusterMessage
|
|
if err := json.Unmarshal(ev.Data, &clusterMessage); err != nil {
|
|
pa.api.LogError("cannot unmarshal cluster message data",
|
|
"id", ev.Id,
|
|
"err", err,
|
|
)
|
|
return
|
|
}
|
|
|
|
if clusterMessage.BoardID != "" {
|
|
pa.sendBoardMessageSkipCluster(clusterMessage.TeamID, clusterMessage.BoardID, clusterMessage.Payload, clusterMessage.EnsureUsers...)
|
|
return
|
|
}
|
|
|
|
var action string
|
|
if actionRaw, ok := clusterMessage.Payload["action"]; ok {
|
|
if s, ok := actionRaw.(string); ok {
|
|
action = s
|
|
}
|
|
}
|
|
if action == "" {
|
|
// no action was specified in the event; assume block change and warn.
|
|
pa.api.LogWarn("cannot determine action from cluster message data",
|
|
"id", ev.Id,
|
|
"payload", clusterMessage.Payload,
|
|
)
|
|
}
|
|
|
|
pa.sendTeamMessageSkipCluster(websocketActionUpdateBlock, clusterMessage.TeamID, clusterMessage.Payload)
|
|
}
|