iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

generate screenshot using Xlib.

Browse Source
This commit is contained in:
Miroslav Šedivý
2021-01-21 20:44:09 +01:00
parent 7ff6ada205
commit afd3dd2f56
8 changed files with 110 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
internal/api/room/handler.go
View File

@ -61,7 +61,8 @@ func (h *RoomHandler) Route(r chi.Router) {
})
r.Route("/screen", func(r chi.Router) {
r.Get("/", h.screenConfiguration)
r.With(auth.CanWatchOnly).Get("/", h.screenConfiguration)
r.With(auth.CanWatchOnly).Get("/image", h.screenImageGet)
r.With(auth.AdminsOnly).Post("/", h.screenConfigurationChange)
r.With(auth.AdminsOnly).Get("/configurations", h.screenConfigurationsList)

23
internal/api/room/screen.go
View File

@ -1,6 +1,9 @@
package room
import (
"bytes"
"image/jpeg"
"strconv"
"net/http"
"demodesk/neko/internal/types"
@ -72,3 +75,23 @@ func (h *RoomHandler) screenConfigurationsList(w http.ResponseWriter, r *http.Re
utils.HttpSuccess(w, list)
}
func (h *RoomHandler) screenImageGet(w http.ResponseWriter, r *http.Request) {
var options *jpeg.Options
if quality, err := strconv.Atoi(r.URL.Query().Get("quality")); err == nil {
options = &jpeg.Options{ quality }
} else {
options = &jpeg.Options{ 90 }
}
img := h.desktop.GetScreenshotImage()
out := new(bytes.Buffer)
err := jpeg.Encode(out, img, options)
if err != nil {
utils.HttpInternalServerError(w, err)
return
}
w.Header().Set("Content-Type", "image/jpeg")
w.Write(out.Bytes())
}