add WS handlers

server/internal/broadcast/manager.go

@@ -13,6 +13,8 @@ type BroadcastManager struct {
 	pipeline *gst.Pipeline
 	remote   *config.Remote
 	config   *config.Broadcast
+	enabled  bool
+	url      string
 }
 
 func New(remote *config.Remote, config *config.Broadcast) *BroadcastManager {
@@ -20,11 +22,13 @@ func New(remote *config.Remote, config *config.Broadcast) *BroadcastManager {
 		logger:  log.With().Str("module", "remote").Logger(),
 		remote:  remote,
 		config:  config,
+		enabled: false,
+		url:     "",
 	}
 }
 
 func (manager *BroadcastManager) Start() {
-	if !manager.config.Enabled {
+	if !manager.enabled || manager.IsActive() {
 		return
 	}
 
@@ -32,7 +36,7 @@ func (manager *BroadcastManager) Start() {
 	manager.pipeline, err = gst.CreateRTMPPipeline(
 		manager.remote.Device,
 		manager.remote.Display,
-		manager.config.RTMP,
+		manager.url,
 	)
 
 	manager.logger.Info().
@@ -50,10 +54,29 @@ func (manager *BroadcastManager) Start() {
 }
 
 func (manager *BroadcastManager) Stop() {
-	if manager.pipeline == nil {
+	if !manager.IsActive() {
 		return
 	}
 
 	manager.pipeline.Stop()
 	manager.pipeline = nil
 }
+
+func (manager *BroadcastManager) IsActive() bool {
+	return manager.pipeline != nil
+}
+
+func (manager *BroadcastManager) Create(url string) {
+	manager.url = url
+	manager.enabled = true
+	manager.Start()
+}
+
+func (manager *BroadcastManager) Destroy() {
+	manager.Stop()
+	manager.enabled = false
+}
+
+func (manager *BroadcastManager) GetUrl() string {
+	return manager.url
+}

server/internal/types/broadcast.go

@@ -3,4 +3,8 @@ package types
 type BroadcastManager interface {
 	Start()
 	Stop()
+	IsActive() bool
+	Create(url string)
+	Destroy()
+	GetUrl() string
 }

server/internal/types/event/events.go

@@ -36,6 +36,12 @@ const (
 	SCREEN_SET            = "screen/set"
 )
 
+const (
+	BORADCAST_STATUS  = "broadcast/status"
+	BORADCAST_CREATE  = "broadcast/create"
+	BORADCAST_DESTROY = "broadcast/destroy"
+)
+
 const (
 	ADMIN_BAN     = "admin/ban"
 	ADMIN_KICK    = "admin/kick"

server/internal/types/message/messages.go

@@ -110,3 +110,14 @@ type ScreenConfigurations struct {
 	Event          string                            `json:"event"`
 	Configurations map[int]types.ScreenConfiguration `json:"configurations"`
 }
+
+type BroadcastStatus struct {
+	Event    string `json:"event"`
+	URL      string `json:"url"`
+	IsActive bool   `json:"isActive"`
+}
+
+type BroadcastCreate struct {
+	Event  string `json:"event"`
+	URL    string `json:"url"`
+}

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

@@ -17,6 +17,7 @@ type MessageHandler struct {
 	sessions  types.SessionManager
 	webrtc    types.WebRTCManager
 	remote    types.RemoteManager
+	broadcast types.BroadcastManager
 	banned    map[string]bool
 	locked    bool
 }
@@ -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,7 +16,7 @@ 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{
@@ -32,6 +32,7 @@ func New(sessions types.SessionManager, remote types.RemoteManager, webrtc types
 		handler: &MessageHandler{
 			logger:    logger.With().Str("subsystem", "handler").Logger(),
 			remote:    remote,
+			broadcast: broadcast,
 			sessions:  sessions,
 			webrtc:    webrtc,
 			banned:    make(map[string]bool),

server/neko.go

@@ -130,7 +130,7 @@ func (neko *Neko) Start() {
 	webRTCManager := webrtc.New(sessionManager, remoteManager, neko.WebRTC)
 	webRTCManager.Start()
 
-	webSocketHandler := websocket.New(sessionManager, remoteManager, webRTCManager, neko.WebSocket)
+	webSocketHandler := websocket.New(sessionManager, remoteManager, broadcastManager, webRTCManager, neko.WebSocket)
 	webSocketHandler.Start()
 
 	server := http.New(neko.Server, webSocketHandler)