add WS handlers

server/internal/websocket/broadcast.go

@@ -0,0 +1,56 @@
+package websocket
+
+import (
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+)
+
+func (h *MessageHandler) boradcastCreate(session types.Session, payload *message.BroadcastCreate) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	h.broadcast.Create(payload.URL)
+
+	if err := h.boradcastStatus(session); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) boradcastDestroy(session types.Session) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	h.broadcast.Destroy()
+
+	if err := h.boradcastStatus(session); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) boradcastStatus(session types.Session) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	if err := session.Send(
+		message.BroadcastStatus{
+			Event:    event.BORADCAST_STATUS,
+			IsActive: h.broadcast.IsActive(),
+			URL:      h.broadcast.GetUrl(),
+		}); err != nil {
+		h.logger.Warn().Err(err).Msgf("sending event %s has failed", event.BORADCAST_STATUS)
+		return err
+	}
+
+	return nil
+}

server/internal/websocket/handler.go

@@ -13,12 +13,13 @@ import (
 )
 
 type MessageHandler struct {
-	logger   zerolog.Logger
-	sessions types.SessionManager
-	webrtc   types.WebRTCManager
-	remote   types.RemoteManager
-	banned   map[string]bool
-	locked   bool
+	logger    zerolog.Logger
+	sessions  types.SessionManager
+	webrtc    types.WebRTCManager
+	remote    types.RemoteManager
+	broadcast types.BroadcastManager
+	banned    map[string]bool
+	locked    bool
 }
 
 func (h *MessageHandler) Connected(id string, socket *WebSocket) (bool, string, error) {
@@ -123,6 +124,16 @@ func (h *MessageHandler) Message(id string, raw []byte) error {
 				return h.screenSet(id, session, payload)
 			}), "%s failed", header.Event)
 
+	// Boradcast Events
+	case event.BORADCAST_CREATE:
+		payload := &message.BroadcastCreate{}
+		return errors.Wrapf(
+			utils.Unmarshal(payload, raw, func() error {
+				return h.boradcastCreate(session, payload)
+			}), "%s failed", header.Event)
+	case event.BORADCAST_DESTROY:
+		return errors.Wrapf(h.boradcastDestroy(session), "%s failed", header.Event)
+
 	// Admin Events
 	case event.ADMIN_LOCK:
 		return errors.Wrapf(h.adminLock(id, session), "%s failed", header.Event)

server/internal/websocket/session.go

@@ -17,6 +17,11 @@ func (h *MessageHandler) SessionCreated(id string, session types.Session) error
 		if err := h.screenConfigurations(id, session); err != nil {
 			return err
 		}
+
+		// send broadcast status if admin
+		if err := h.boradcastStatus(session); err != nil {
+			return err
+		}
 	}
 
 	return nil

server/internal/websocket/websocket.go

@@ -16,26 +16,27 @@ import (
 	"n.eko.moe/neko/internal/utils"
 )
 
-func New(sessions types.SessionManager, remote types.RemoteManager, webrtc types.WebRTCManager, conf *config.WebSocket) *WebSocketHandler {
+func New(sessions types.SessionManager, remote types.RemoteManager, broadcast types.BroadcastManager, webrtc types.WebRTCManager, conf *config.WebSocket) *WebSocketHandler {
 	logger := log.With().Str("module", "websocket").Logger()
 
 	return &WebSocketHandler{
-		logger:   logger,
-		conf:     conf,
-		sessions: sessions,
-		remote:   remote,
-		upgrader: websocket.Upgrader{
+		logger:    logger,
+		conf:      conf,
+		sessions:  sessions,
+		remote:    remote,
+		upgrader:  websocket.Upgrader{
 			CheckOrigin: func(r *http.Request) bool {
 				return true
 			},
 		},
 		handler: &MessageHandler{
-			logger:   logger.With().Str("subsystem", "handler").Logger(),
-			remote:   remote,
-			sessions: sessions,
-			webrtc:   webrtc,
-			banned:   make(map[string]bool),
-			locked:   false,
+			logger:    logger.With().Str("subsystem", "handler").Logger(),
+			remote:    remote,
+			broadcast: broadcast,
+			sessions:  sessions,
+			webrtc:    webrtc,
+			banned:    make(map[string]bool),
+			locked:    false,
 		},
 	}
 }
@@ -44,13 +45,13 @@ func New(sessions types.SessionManager, remote types.RemoteManager, webrtc types
 const pingPeriod = 60 * time.Second
 
 type WebSocketHandler struct {
-	logger   zerolog.Logger
-	upgrader websocket.Upgrader
-	sessions types.SessionManager
-	remote   types.RemoteManager
-	conf     *config.WebSocket
-	handler  *MessageHandler
-	shutdown chan bool
+	logger    zerolog.Logger
+	upgrader  websocket.Upgrader
+	sessions  types.SessionManager
+	remote    types.RemoteManager
+	conf      *config.WebSocket
+	handler   *MessageHandler
+	shutdown  chan bool
 }
 
 func (ws *WebSocketHandler) Start() error {