refactor: move hrp/ to root folder

cmd/adb/devices.go

@@ -0,0 +1,64 @@
+package adb
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/spf13/cobra"
+
+	"github.com/httprunner/httprunner/v5/internal/sdk"
+	"github.com/httprunner/httprunner/v5/pkg/uixt"
+)
+
+func format(data map[string]string) string {
+	result, _ := json.MarshalIndent(data, "", "\t")
+	return string(result)
+}
+
+var listAndroidDevicesCmd = &cobra.Command{
+	Use:   "devices",
+	Short: "List all Android devices",
+	RunE: func(cmd *cobra.Command, args []string) (err error) {
+		startTime := time.Now()
+		defer func() {
+			sdk.SendGA4Event("hrp_adb_devices", map[string]interface{}{
+				"args":                 strings.Join(args, "-"),
+				"success":              err == nil,
+				"engagement_time_msec": time.Since(startTime).Milliseconds(),
+			})
+		}()
+
+		deviceList, err := uixt.GetAndroidDevices(serial)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(0)
+		}
+
+		for _, d := range deviceList {
+			if isDetail {
+				fmt.Println(format(d.DeviceInfo()))
+			} else {
+				if usb, err := d.Usb(); err != nil {
+					fmt.Println(d.Serial())
+				} else {
+					fmt.Println(d.Serial(), usb)
+				}
+			}
+		}
+		return nil
+	},
+}
+
+var (
+	serial   string
+	isDetail bool
+)
+
+func init() {
+	listAndroidDevicesCmd.Flags().StringVarP(&serial, "serial", "s", "", "filter by device's serial")
+	listAndroidDevicesCmd.Flags().BoolVarP(&isDetail, "detail", "d", false, "print device's detail")
+	androidRootCmd.AddCommand(listAndroidDevicesCmd)
+}

cmd/adb/init.go

@@ -0,0 +1,31 @@
+package adb
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+
+	"github.com/httprunner/httprunner/v5/pkg/gadb"
+	"github.com/httprunner/httprunner/v5/pkg/uixt"
+)
+
+var androidRootCmd = &cobra.Command{
+	Use:              "adb",
+	Short:            "simple utils for android device management",
+	PersistentPreRun: func(cmd *cobra.Command, args []string) {},
+}
+
+func getDevice(serial string) (*gadb.Device, error) {
+	devices, err := uixt.GetAndroidDevices(serial)
+	if err != nil {
+		return nil, err
+	}
+	if len(devices) > 1 {
+		return nil, fmt.Errorf("found multiple attached devices, please specify android serial")
+	}
+	return devices[0], nil
+}
+
+func Init(rootCmd *cobra.Command) {
+	rootCmd.AddCommand(androidRootCmd)
+}

cmd/adb/install.go

@@ -0,0 +1,69 @@
+package adb
+
+import (
+	"fmt"
+	"strings"
+	"time"
+
+	"github.com/spf13/cobra"
+
+	"github.com/httprunner/httprunner/v5/internal/sdk"
+	"github.com/httprunner/httprunner/v5/pkg/uixt"
+)
+
+var (
+	replace   bool
+	downgrade bool
+	grant     bool
+)
+
+var installCmd = &cobra.Command{
+	Use:   "install [flags] PACKAGE",
+	Short: "push package to the device and install them automatically",
+	Args:  cobra.MinimumNArgs(1),
+	RunE: func(cmd *cobra.Command, args []string) (err error) {
+		startTime := time.Now()
+		defer func() {
+			sdk.SendGA4Event("hrp_adb_devices", map[string]interface{}{
+				"args":                 strings.Join(args, "-"),
+				"success":              err == nil,
+				"engagement_time_msec": time.Since(startTime).Milliseconds(),
+			})
+		}()
+		_, err = getDevice(serial)
+		if err != nil {
+			return err
+		}
+
+		device, err := uixt.NewAndroidDevice(uixt.WithSerialNumber(serial))
+		if err != nil {
+			fmt.Println(err)
+			return err
+		}
+		driverExt, err := device.NewDriver()
+		if err != nil {
+			fmt.Println(err)
+			return err
+		}
+
+		err = driverExt.Install(args[0],
+			uixt.WithReinstall(replace),
+			uixt.WithDowngrade(downgrade),
+			uixt.WithGrantPermission(grant),
+		)
+		if err != nil {
+			fmt.Println(err)
+			return err
+		}
+		fmt.Println("success")
+		return nil
+	},
+}
+
+func init() {
+	installCmd.Flags().StringVarP(&serial, "serial", "s", "", "filter by device's serial")
+	installCmd.Flags().BoolVarP(&replace, "replace", "r", false, "replace existing application")
+	installCmd.Flags().BoolVarP(&downgrade, "downgrade", "d", false, "allow version code downgrade (debuggable packages only)")
+	installCmd.Flags().BoolVarP(&grant, "grant", "g", false, "grant all runtime permissions")
+	androidRootCmd.AddCommand(installCmd)
+}

cmd/adb/screencap.go

@@ -0,0 +1,50 @@
+package adb
+
+import (
+	"fmt"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/spf13/cobra"
+
+	"github.com/httprunner/httprunner/v5/internal/builtin"
+	"github.com/httprunner/httprunner/v5/internal/sdk"
+)
+
+var screencapAndroidDevicesCmd = &cobra.Command{
+	Use:   "screencap",
+	Short: "Start android screen capture",
+	RunE: func(cmd *cobra.Command, args []string) (err error) {
+		startTime := time.Now()
+		defer func() {
+			sdk.SendGA4Event("hrp_adb_screencap", map[string]interface{}{
+				"args":                 strings.Join(args, "-"),
+				"success":              err == nil,
+				"engagement_time_msec": time.Since(startTime).Milliseconds(),
+			})
+		}()
+
+		device, err := getDevice(serial)
+		if err != nil {
+			return err
+		}
+
+		res, err := device.ScreenCap()
+		if err != nil {
+			return err
+		}
+
+		filepath := fmt.Sprintf("%s.png", builtin.GenNameWithTimestamp("screencap_%d"))
+		if err = os.WriteFile(filepath, res, 0o644); err != nil {
+			return err
+		}
+		fmt.Println("screencap saved to", filepath)
+		return nil
+	},
+}
+
+func init() {
+	screencapAndroidDevicesCmd.Flags().StringVarP(&serial, "serial", "s", "", "filter by device's serial")
+	androidRootCmd.AddCommand(screencapAndroidDevicesCmd)
+}