From ebe7d7ecf4315f2edd17d816f89114785fe562e1 Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Wed, 27 Jul 2022 21:57:51 +0800
Subject: [PATCH] feat: double/long click on name or xpath

---
 hrp/step_ios_ui.go | 54 ++++++++++++++++++++++++++++++----------------
 1 file changed, 36 insertions(+), 18 deletions(-)

diff --git a/hrp/step_ios_ui.go b/hrp/step_ios_ui.go
index e466da98..3f50a502 100644
--- a/hrp/step_ios_ui.go
+++ b/hrp/step_ios_ui.go
@@ -355,20 +355,7 @@ func (w *wdaClient) doAction(action MobileAction) error {
 		}
 		// click on name or xpath
 		if param, ok := action.Params.(string); ok {
-			var selector gwda.BySelector
-			if strings.HasPrefix(param, "/") {
-				// xpath
-				selector = gwda.BySelector{
-					XPath: param,
-				}
-			} else {
-				// name
-				selector = gwda.BySelector{
-					Name: param,
-				}
-			}
-
-			ele, err := w.Driver.FindElement(selector)
+			ele, err := w.findElement(param)
 			if err != nil {
 				return errors.Wrap(err, "failed to find element")
 			}
@@ -376,11 +363,25 @@ func (w *wdaClient) doAction(action MobileAction) error {
 		}
 		return fmt.Errorf("invalid click params: %v", action.Params)
 	case uiDoubleClick:
-		// TODO
-		return errActionNotImplemented
+		// double click on name or xpath
+		if param, ok := action.Params.(string); ok {
+			ele, err := w.findElement(param)
+			if err != nil {
+				return errors.Wrap(err, "failed to find element")
+			}
+			return ele.DoubleTap()
+		}
+		return fmt.Errorf("invalid click params: %v", action.Params)
 	case uiLongClick:
-		// TODO
-		return errActionNotImplemented
+		// long click 2s on name or xpath
+		if param, ok := action.Params.(string); ok {
+			ele, err := w.findElement(param)
+			if err != nil {
+				return errors.Wrap(err, "failed to find element")
+			}
+			return ele.TouchAndHold(2)
+		}
+		return fmt.Errorf("invalid click params: %v", action.Params)
 	case uiSwipe:
 		width := w.WindowSize.Width
 		height := w.WindowSize.Height
@@ -472,6 +473,23 @@ func (w *wdaClient) doValidation(iValidators []interface{}) (validateResults []*
 	return
 }
 
+func (w *wdaClient) findElement(param string) (ele gwda.WebElement, err error) {
+	var selector gwda.BySelector
+	if strings.HasPrefix(param, "/") {
+		// xpath
+		selector = gwda.BySelector{
+			XPath: param,
+		}
+	} else {
+		// name
+		selector = gwda.BySelector{
+			Name: param,
+		}
+	}
+
+	return w.Driver.FindElement(selector)
+}
+
 func (w *wdaClient) assertName(name string, exists bool) bool {
 	selector := gwda.BySelector{
 		Name: name,