snapshot of https://github.com/mattermost/mattermost.git (master, 2026-09-16; blobs > 131072B omitted)

server/cmd/mmctl/commands/websockets.go

@@ -0,0 +1,56 @@
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
+// See LICENSE.txt for license information.
+
+package commands
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/mattermost/mattermost/server/public/model"
+	"github.com/pkg/errors"
+	"github.com/spf13/cobra"
+)
+
+var WebsocketCmd = &cobra.Command{
+	Use:   "websocket",
+	Short: "Display websocket in a human-readable format",
+	RunE:  websocketCmdF,
+}
+
+func init() {
+	RootCmd.AddCommand(WebsocketCmd)
+}
+
+func websocketCmdF(cmd *cobra.Command, args []string) error {
+	c, err := InitWebSocketClient()
+	if err != nil {
+		return err
+	}
+	appErr := c.Connect()
+	if appErr != nil {
+		return errors.New(appErr.Error())
+	}
+
+	c.Listen()
+	fmt.Fprintln(os.Stderr, "Press CTRL+C to exit")
+	return processWebSocketEvents(c)
+}
+
+func processWebSocketEvents(c *model.WebSocketClient) error {
+	for event := range c.EventChannel {
+		if event == nil {
+			continue
+		}
+		data, err := event.ToJSON()
+		if err != nil {
+			fmt.Println(err.Error())
+		} else {
+			fmt.Println(string(data))
+		}
+	}
+	if c.ListenError != nil {
+		return errors.New(c.ListenError.Error())
+	}
+	return nil
+}