system init WS.

internal/websocket/handler/session.go

@@ -7,34 +7,12 @@ import (
 )
 
 func (h *MessageHandlerCtx) SessionConnected(session types.Session) error {
-	// create member list
-	members := []*message.MembersListEntry{}
-	for _, session := range h.sessions.Members() {
-		members = append(members, &message.MembersListEntry{
-			ID:    session.ID(),
-			Name:  session.Name(),
-			Admin: session.Admin(),
-		})
-	}
-
-	// send list of members to session
-	if err := session.Send(
-		message.MembersList{
-			Event:    event.MEMBER_LIST,
-			Memebers: members,
-		}); err != nil {
+	if err := h.systemInit(session); err != nil {
 		return err
 	}
 
-	// tell session there is a host
-	host := h.sessions.GetHost()
-	if host != nil {
-		if err := session.Send(
-			message.ControlHost{
-				Event:   event.CONTROL_HOST,
-				HasHost: true,
-				HostID:  host.ID(),
-			}); err != nil {
+	if session.Admin() {
+		if err := h.systemAdmin(session); err != nil {
 			return err
 		}
 	}
@@ -42,7 +20,7 @@ func (h *MessageHandlerCtx) SessionConnected(session types.Session) error {
 	// let everyone know there is a new session
 	h.sessions.Broadcast(
 		message.Member{
-			Event:  event.MEMBER_CONNECTED,
+			Event: event.MEMBER_CONNECTED,
 			ID:    session.ID(),
 			Name:  session.Name(),
 			Admin: session.Admin(),

internal/websocket/handler/system.go

@@ -0,0 +1,69 @@
+package handler
+
+import (
+	"demodesk/neko/internal/types"
+	"demodesk/neko/internal/types/event"
+	"demodesk/neko/internal/types/message"
+)
+
+func (h *MessageHandlerCtx) systemInit(session types.Session) error {
+	host := h.sessions.GetHost()
+
+	controlHost := message.ControlHost{
+		HasHost: host != nil,
+	}
+
+	if controlHost.HasHost {
+		controlHost.HostID = host.ID()
+	}
+
+	size := h.desktop.GetScreenSize()
+	if size == nil {
+		h.logger.Debug().Msg("could not get screen size")
+		return nil
+	}
+
+	members := []message.MemberData{}
+	for _, session := range h.sessions.Members() {
+		members = append(members, message.MemberData{
+			ID:      session.ID(),
+			Name:    session.Name(),
+			IsAdmin: session.Admin(),
+		})
+	}
+
+	return session.Send(
+		message.SystemInit{
+			Event:       event.SYSTEM_INIT,
+			ControlHost: controlHost,
+			Members:     members,
+			ScreenSize:  message.ScreenSize{
+				Width:  size.Width,
+				Height: size.Height,
+				Rate:   int(size.Rate),
+			},
+		})
+}
+
+func (h *MessageHandlerCtx) systemAdmin(session types.Session) error {
+	screenSizesList := []message.ScreenSize{}
+	for _, size := range h.desktop.ScreenConfigurations() {
+		for _, fps := range size.Rates {
+			screenSizesList = append(screenSizesList, message.ScreenSize{
+				Width:  size.Width,
+				Height: size.Height,
+				Rate:   int(fps),
+			})
+		}
+	}
+
+	return session.Send(
+		message.SystemAdmin{
+			Event:           event.SYSTEM_ADMIN,
+			ScreenSizesList: screenSizesList,
+			BroadcastStatus: message.BroadcastStatus{
+				IsActive: h.capture.BroadcastEnabled(),
+				URL:      h.capture.BroadcastUrl(),
+			},
+		})
+}