feat: device新增截图方法

pkg/uixt/android_device.go

@@ -111,7 +111,6 @@ func (dev *AndroidDevice) Setup() error {
 	if err != nil {
 		return errors.Wrap(code.DeviceShellExecError, err.Error())
 	}
-
 	return nil
 }
 
@@ -339,6 +338,15 @@ func (dev *AndroidDevice) GetPackageInfo(packageName string) (types.AppInfo, err
 	return appInfo, nil
 }
 
+func (dev *AndroidDevice) ScreenShot() (*bytes.Buffer, error) {
+	raw, err := dev.Device.ScreenCap()
+	if err != nil {
+		return nil, errors.Wrapf(code.DeviceScreenShotError,
+			"adb screencap failed %v", err)
+	}
+	return bytes.NewBuffer(raw), nil
+}
+
 func (dev *AndroidDevice) GetAppInfo(packageName string) (app types.AppInfo, err error) {
 	packageInfo, err := dev.RunShellCommand(
 		"CLASSPATH=/data/local/tmp/evalite", "app_process", "/",