diff --git a/Overlord-Client/cmd/agent/capture/capture.go b/Overlord-Client/cmd/agent/capture/capture.go index 26f5a30..c736eb0 100644 --- a/Overlord-Client/cmd/agent/capture/capture.go +++ b/Overlord-Client/cmd/agent/capture/capture.go @@ -216,7 +216,14 @@ func captureAllDisplaysAndSend(ctx context.Context, env *rt.Env) error { } func supportsCapture() bool { - return safeDisplayCount() > 0 + return safeActiveDisplays() > 0 +} + +func safeActiveDisplays() int { + defer func() { + _ = recover() + }() + return activeDisplays() } func safeCaptureDisplay(display int) (*image.RGBA, error) { diff --git a/Overlord-Client/cmd/agent/capture/capture_public_default.go b/Overlord-Client/cmd/agent/capture/capture_public_default.go new file mode 100644 index 0000000..bfd7911 --- /dev/null +++ b/Overlord-Client/cmd/agent/capture/capture_public_default.go @@ -0,0 +1,13 @@ +//go:build !windows + +package capture + +import "image" + +func CaptureDisplayRGBABitBlt(display int) (*image.RGBA, error) { + return CaptureDisplayRGBA(display) +} + +func CaptureDisplayRGBAPreferBitBlt(display int) (*image.RGBA, error) { + return CaptureDisplayRGBA(display) +}