refactor: move uixt from hrp internal to pkg

hrp/pkg/uixt/swipe.go

@@ -0,0 +1,86 @@
+package uixt
+
+import (
+	"fmt"
+
+	"github.com/rs/zerolog/log"
+)
+
+func assertRelative(p float64) bool {
+	return p >= 0 && p <= 1
+}
+
+// SwipeRelative swipe from relative position [fromX, fromY] to relative position [toX, toY]
+func (dExt *DriverExt) SwipeRelative(fromX, fromY, toX, toY float64, identifier ...string) error {
+	width := dExt.windowSize.Width
+	height := dExt.windowSize.Height
+
+	if !assertRelative(fromX) || !assertRelative(fromY) ||
+		!assertRelative(toX) || !assertRelative(toY) {
+		return fmt.Errorf("fromX(%f), fromY(%f), toX(%f), toY(%f) must be less than 1",
+			fromX, fromY, toX, toY)
+	}
+
+	fromX = float64(width) * fromX
+	fromY = float64(height) * fromY
+	toX = float64(width) * toX
+	toY = float64(height) * toY
+
+	if len(identifier) > 0 && identifier[0] != "" {
+		option := WithCustomOption("log", map[string]interface{}{
+			"enable": true,
+			"data":   identifier[0],
+		})
+		return dExt.Driver.SwipeFloat(fromX, fromY, toX, toY, option)
+	}
+	return dExt.Driver.SwipeFloat(fromX, fromY, toX, toY)
+}
+
+func (dExt *DriverExt) SwipeTo(direction string, identifier ...string) (err error) {
+	switch direction {
+	case "up":
+		return dExt.SwipeUp(identifier...)
+	case "down":
+		return dExt.SwipeDown(identifier...)
+	case "left":
+		return dExt.SwipeLeft(identifier...)
+	case "right":
+		return dExt.SwipeRight(identifier...)
+	}
+	return fmt.Errorf("unexpected direction: %s", direction)
+}
+
+func (dExt *DriverExt) SwipeUp(identifier ...string) (err error) {
+	return dExt.SwipeRelative(0.5, 0.5, 0.5, 0.1, identifier...)
+}
+
+func (dExt *DriverExt) SwipeDown(identifier ...string) (err error) {
+	return dExt.SwipeRelative(0.5, 0.5, 0.5, 0.9, identifier...)
+}
+
+func (dExt *DriverExt) SwipeLeft(identifier ...string) (err error) {
+	return dExt.SwipeRelative(0.5, 0.5, 0.1, 0.5, identifier...)
+}
+
+func (dExt *DriverExt) SwipeRight(identifier ...string) (err error) {
+	return dExt.SwipeRelative(0.5, 0.5, 0.9, 0.5, identifier...)
+}
+
+// FindCondition indicates the condition to find a UI element
+type FindCondition func(driver *DriverExt) error
+
+// FoundAction indicates the action to do after a UI element is found
+type FoundAction func(driver *DriverExt) error
+
+func (dExt *DriverExt) SwipeUntil(direction string, condition FindCondition, action FoundAction, maxTimes int) error {
+	for i := 0; i < maxTimes; i++ {
+		if err := condition(dExt); err == nil {
+			// do action after found
+			return action(dExt)
+		}
+		if err := dExt.SwipeTo(direction); err != nil {
+			log.Error().Err(err).Msgf("swipe %s failed", direction)
+		}
+	}
+	return fmt.Errorf("swipe %s %d times, match condition failed", direction, maxTimes)
+}