From ebdd676e6149640a718f9608fb997683dffe216e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=BD=99=E6=B3=93=E9=93=AE?= <yuhongzheng@bytedance.com>
Date: Tue, 25 Feb 2025 17:51:27 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20device=E6=96=B0=E5=A2=9E=E6=88=AA?=
 =?UTF-8?q?=E5=9B=BE=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/uixt/browser_device.go | 5 +++++
 pkg/uixt/device.go         | 3 ++-
 pkg/uixt/harmony_device.go | 5 +++++
 3 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/pkg/uixt/browser_device.go b/pkg/uixt/browser_device.go
index 5dd6e440..1595d6fa 100644
--- a/pkg/uixt/browser_device.go
+++ b/pkg/uixt/browser_device.go
@@ -1,6 +1,7 @@
 package uixt
 
 import (
+	"bytes"
 	"github.com/httprunner/httprunner/v5/pkg/uixt/option"
 	"github.com/httprunner/httprunner/v5/pkg/uixt/types"
 	"github.com/pkg/errors"
@@ -73,3 +74,7 @@ func (dev *BrowserDevice) NewDriver() (driver IDriver, err error) {
 	}
 	return driver, nil
 }
+
+func (dev *BrowserDevice) ScreenShot() (*bytes.Buffer, error) {
+	return nil, errors.New("not support")
+}
diff --git a/pkg/uixt/device.go b/pkg/uixt/device.go
index 652e5453..6e2dc10d 100644
--- a/pkg/uixt/device.go
+++ b/pkg/uixt/device.go
@@ -1,6 +1,7 @@
 package uixt
 
 import (
+	"bytes"
 	"github.com/httprunner/httprunner/v5/pkg/uixt/option"
 	"github.com/httprunner/httprunner/v5/pkg/uixt/types"
 )
@@ -17,7 +18,7 @@ type IDevice interface {
 	Uninstall(packageName string) error
 
 	GetPackageInfo(packageName string) (types.AppInfo, error)
-
+	ScreenShot() (*bytes.Buffer, error)
 	// TODO: remove?
 	LogEnabled() bool
 }
diff --git a/pkg/uixt/harmony_device.go b/pkg/uixt/harmony_device.go
index 5c0869a5..e2bc5f04 100644
--- a/pkg/uixt/harmony_device.go
+++ b/pkg/uixt/harmony_device.go
@@ -1,6 +1,7 @@
 package uixt
 
 import (
+	"bytes"
 	"code.byted.org/iesqa/ghdc"
 	"github.com/pkg/errors"
 	"github.com/rs/zerolog/log"
@@ -106,3 +107,7 @@ func (dev *HarmonyDevice) NewDriver() (IDriver, error) {
 	}
 	return driver, nil
 }
+
+func (dev *HarmonyDevice) ScreenShot() (*bytes.Buffer, error) {
+	return nil, errors.New("not implemented")
+}