initial commit - from neko open source repository..

internal/websocket/screen.go

@@ -0,0 +1,66 @@
+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) screenSet(id string, session types.Session, payload *message.ScreenResolution) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	if err := h.remote.ChangeResolution(payload.Width, payload.Height, payload.Rate); err != nil {
+		h.logger.Warn().Err(err).Msgf("unable to change screen size")
+		return err
+	}
+
+	if err := h.sessions.Broadcast(
+		message.ScreenResolution{
+			Event:  event.SCREEN_RESOLUTION,
+			ID:     id,
+			Width:  payload.Width,
+			Height: payload.Height,
+			Rate:   payload.Rate,
+		}, nil); err != nil {
+		h.logger.Warn().Err(err).Msgf("broadcasting event %s has failed", event.SCREEN_RESOLUTION)
+		return err
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) screenResolution(id string, session types.Session) error {
+	if size := h.remote.GetScreenSize(); size != nil {
+		if err := session.Send(message.ScreenResolution{
+			Event:  event.SCREEN_RESOLUTION,
+			Width:  size.Width,
+			Height: size.Height,
+			Rate:   int(size.Rate),
+		}); err != nil {
+			h.logger.Warn().Err(err).Msgf("sending event %s has failed", event.SCREEN_RESOLUTION)
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (h *MessageHandler) screenConfigurations(id string, session types.Session) error {
+	if !session.Admin() {
+		h.logger.Debug().Msg("user not admin")
+		return nil
+	}
+
+	if err := session.Send(message.ScreenConfigurations{
+		Event:          event.SCREEN_CONFIGURATIONS,
+		Configurations: h.remote.ScreenConfigurations(),
+	}); err != nil {
+		h.logger.Warn().Err(err).Msgf("sending event %s has failed", event.SCREEN_CONFIGURATIONS)
+		return err
+	}
+
+	return nil
+}