neko/internal/api/room/screen.go

package room

import (
	"net/http"

	"github.com/go-chi/render"

	"demodesk/neko/internal/api/utils"
)

type ScreenConfiguration struct {
	Width  int `json:"width"`
	Height int `json:"height"`
	Rate   int `json:"rate"`
}

func (a *ScreenConfiguration) Bind(r *http.Request) error {
	// Bind will run after the unmarshalling is complete, its a
	// good time to focus some post-processing after a decoding.
	return nil
}

func (a *ScreenConfiguration) Render(w http.ResponseWriter, r *http.Request) error {
	// Pre-processing before a response is marshalled and sent
	// across the wire
	return nil
}

func (h *RoomHandler) ScreenConfiguration(w http.ResponseWriter, r *http.Request) {
	size := h.desktop.GetScreenSize()

	if size == nil {
		_ = render.Render(w, r, utils.ErrMessage(500, "Unable to get screen configuration."))
		return
	}

	render.JSON(w, r, ScreenConfiguration{
		Width:  size.Width,
		Height: size.Height,
		Rate:   int(size.Rate),
	})
}

func (h *RoomHandler) ScreenConfigurationChange(w http.ResponseWriter, r *http.Request) {
	data := &ScreenConfiguration{}
	if err := render.Bind(r, data); err != nil {
		_ = render.Render(w, r, utils.ErrBadRequest(err))
		return
	}

	if err := h.capture.ChangeResolution(data.Width, data.Height, data.Rate); err != nil {
		_ = render.Render(w, r, utils.ErrUnprocessableEntity(err))
		return
	}

	// TODO: Broadcast change to all sessions.

	render.JSON(w, r, data)
}

func (h *RoomHandler) ScreenConfigurationsList(w http.ResponseWriter, r *http.Request) {
	list := []render.Renderer{}
	
	ScreenConfigurations := h.desktop.ScreenConfigurations()
	for _, size := range ScreenConfigurations {
		for _, fps := range size.Rates {
			list = append(list, &ScreenConfiguration{
				Width:  size.Width,
				Height: size.Height,
				Rate:   int(fps),
			})
		}
	}

	_ = render.RenderList(w, r, list)
}