capture error (#663)

* capture error * aohanhongzhi * Update go.mod --------- Co-authored-by: Evans <[email protected]>
go-vgo · Oct 7, 2024 · 8db59aa · 8db59aa
1 parent 217d6cf
commit 8db59aa
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/robotgo.go b/robotgo.go
@@ -50,6 +50,7 @@ package robotgo
 import "C"
 
 import (
+	"errors"
 	"image"
 	"runtime"
 	"time"
@@ -366,11 +367,14 @@ func CaptureGo(args ...int) Bitmap {
 }
 
 // CaptureImg capture the screen and return image.Image
-func CaptureImg(args ...int) image.Image {
+func CaptureImg(args ...int) (image.Image, error) {
 	bit := CaptureScreen(args...)
+	if bit == nil {
+		return nil, errors.New("capture error")
+	}
 	defer FreeBitmap(bit)
 
-	return ToImage(bit)
+	return ToImage(bit), nil
 }
 
 // FreeBitmap free and dealloc the C bitmap