httprunner/hrp/pkg/uixt/tap.go

package uixt

import (
	"fmt"
	"github.com/rs/zerolog/log"
)

func (dExt *DriverExt) TapAbsXY(x, y float64, options ...ActionOption) error {
	// tap on absolute coordinate [x, y]
	return dExt.Driver.TapFloat(x, y, options...)
}

func (dExt *DriverExt) TapXY(x, y float64, options ...ActionOption) error {
	// tap on [x, y] percent of window size
	if x > 1 || y > 1 {
		return fmt.Errorf("x, y percentage should be <= 1, got x=%v, y=%v", x, y)
	}

	orientation, err := dExt.Driver.Orientation()
	if err != nil {
		log.Warn().Err(err).Msgf("tap (%v, %v) get orientation failed, use default orientation",
			x, y)
		orientation = OrientationPortrait
	}
	if orientation == OrientationPortrait {
		x = x * float64(dExt.windowSize.Width)
		y = y * float64(dExt.windowSize.Height)
	} else {
		x = x * float64(dExt.windowSize.Height)
		y = y * float64(dExt.windowSize.Width)
	}
	return dExt.TapAbsXY(x, y, options...)
}

func (dExt *DriverExt) TapByOCR(ocrText string, options ...ActionOption) error {
	actionOptions := NewActionOptions(options...)

	point, err := dExt.FindScreenText(ocrText, options...)
	if err != nil {
		if actionOptions.IgnoreNotFoundError {
			return nil
		}
		return err
	}

	return dExt.TapAbsXY(point.X, point.Y, options...)
}

func (dExt *DriverExt) TapByCV(imagePath string, options ...ActionOption) error {
	actionOptions := NewActionOptions(options...)

	point, err := dExt.FindImageRectInUIKit(imagePath, options...)
	if err != nil {
		if actionOptions.IgnoreNotFoundError {
			return nil
		}
		return err
	}

	return dExt.TapAbsXY(point.X, point.Y, options...)
}

func (dExt *DriverExt) TapByUIDetection(options ...ActionOption) error {
	actionOptions := NewActionOptions(options...)

	point, err := dExt.FindUIResult(options...)
	if err != nil {
		if actionOptions.IgnoreNotFoundError {
			return nil
		}
		return err
	}

	return dExt.TapAbsXY(point.X, point.Y, options...)
}

func (dExt *DriverExt) Tap(param string, options ...ActionOption) error {
	return dExt.TapOffset(param, 0, 0, options...)
}

func (dExt *DriverExt) TapOffset(param string, xOffset, yOffset float64, options ...ActionOption) (err error) {
	actionOptions := NewActionOptions(options...)

	point, err := dExt.FindUIRectInUIKit(param, options...)
	if err != nil {
		if actionOptions.IgnoreNotFoundError {
			return nil
		}
		return err
	}

	return dExt.TapAbsXY(point.X+xOffset, point.Y+yOffset, options...)
}

func (dExt *DriverExt) DoubleTapXY(x, y float64) error {
	// double tap on coordinate: [x, y] should be relative
	if x > 1 || y > 1 {
		return fmt.Errorf("x, y percentage should be < 1, got x=%v, y=%v", x, y)
	}
	orientation, err := dExt.Driver.Orientation()
	if err != nil {
		log.Warn().Err(err).Msgf("tap (%v, %v) get orientation failed, use default orientation",
			x, y)
		orientation = OrientationPortrait
	}
	if orientation == OrientationPortrait {
		x = x * float64(dExt.windowSize.Width)
		y = y * float64(dExt.windowSize.Height)
	} else {
		x = x * float64(dExt.windowSize.Height)
		y = y * float64(dExt.windowSize.Width)
	}
	x = x * float64(dExt.windowSize.Width)
	y = y * float64(dExt.windowSize.Height)
	return dExt.Driver.DoubleTapFloat(x, y)
}

func (dExt *DriverExt) DoubleTap(param string) (err error) {
	return dExt.DoubleTapOffset(param, 0, 0)
}

func (dExt *DriverExt) DoubleTapOffset(param string, xOffset, yOffset float64) (err error) {
	point, err := dExt.FindUIRectInUIKit(param)
	if err != nil {
		return err
	}

	return dExt.Driver.DoubleTapFloat(point.X+xOffset, point.Y+yOffset)
}