mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 02:21:29 +08:00
103 lines
3.3 KiB
Go
103 lines
3.3 KiB
Go
package ios
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/httprunner/httprunner/v4/hrp/internal/builtin"
|
|
"github.com/httprunner/httprunner/v4/hrp/internal/sdk"
|
|
)
|
|
|
|
// mountCmd represents the mount command
|
|
var mountCmd = &cobra.Command{
|
|
Use: "mount",
|
|
Short: "A brief description of your command",
|
|
RunE: func(cmd *cobra.Command, args []string) (err error) {
|
|
startTime := time.Now()
|
|
defer func() {
|
|
sdk.SendGA4Event("hrp_ios_mount", map[string]interface{}{
|
|
"args": strings.Join(args, "-"),
|
|
"success": err == nil,
|
|
"engagement_time_msec": time.Since(startTime).Milliseconds(),
|
|
})
|
|
}()
|
|
|
|
device, err := getDevice(udid)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
value, err := device.GetValue("", "ProductVersion")
|
|
if err != nil {
|
|
return fmt.Errorf("get device ProductVersion failed: %v", err)
|
|
}
|
|
log.Info().Str("version", value.(string)).Msg("get device version")
|
|
|
|
imageSignatures, errImage := device.Images()
|
|
|
|
if listDeveloperDiskImage {
|
|
for i, imgSign := range imageSignatures {
|
|
fmt.Printf("[%d] %s\n", i+1, base64.StdEncoding.EncodeToString(imgSign))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
if errImage == nil && len(imageSignatures) > 0 {
|
|
log.Info().Msg("ios developer image is already mounted")
|
|
return nil
|
|
}
|
|
|
|
log.Info().Str("dir", developerDiskImageDir).Msg("start to mount ios developer image")
|
|
|
|
if !builtin.IsFolderPathExists(developerDiskImageDir) {
|
|
return fmt.Errorf("developer disk image directory not exist: %s", developerDiskImageDir)
|
|
}
|
|
|
|
ver := strings.Split(value.(string), ".")
|
|
if len(ver) < 2 {
|
|
return fmt.Errorf("got invalid device ProductVersion: %v", value)
|
|
}
|
|
version := ver[0] + "." + ver[1]
|
|
|
|
var dmgPath, signaturePath string
|
|
if builtin.IsFilePathExists(filepath.Join(developerDiskImageDir, "DeveloperDiskImage.dmg")) {
|
|
dmgPath = filepath.Join(developerDiskImageDir, "DeveloperDiskImage.dmg")
|
|
signaturePath = filepath.Join(developerDiskImageDir, "DeveloperDiskImage.dmg.signature")
|
|
} else if builtin.IsFilePathExists(filepath.Join(developerDiskImageDir, version, "DeveloperDiskImage.dmg")) {
|
|
dmgPath = filepath.Join(developerDiskImageDir, version, "DeveloperDiskImage.dmg")
|
|
signaturePath = filepath.Join(developerDiskImageDir, version, "DeveloperDiskImage.dmg.signature")
|
|
} else {
|
|
log.Error().Str("dir", developerDiskImageDir).Msgf(
|
|
"developer disk image %s not found in directory", version)
|
|
return fmt.Errorf("developer disk image %s not found", version)
|
|
}
|
|
|
|
if err = device.MountDeveloperDiskImage(dmgPath, signaturePath); err != nil {
|
|
return fmt.Errorf("mount developer disk image %s failed: %s", version, err)
|
|
}
|
|
|
|
log.Info().Msg("mount developer disk image successfully")
|
|
return nil
|
|
},
|
|
}
|
|
|
|
const defaultDeveloperDiskImageDir = "/Applications/Xcode.app/Contents/Developer/Platforms/iPhoneOS.platform/DeviceSupport/"
|
|
|
|
var (
|
|
developerDiskImageDir string
|
|
listDeveloperDiskImage bool
|
|
)
|
|
|
|
func init() {
|
|
mountCmd.Flags().BoolVar(&listDeveloperDiskImage, "list", false, "list developer disk images")
|
|
mountCmd.Flags().StringVarP(&developerDiskImageDir, "dir", "d", defaultDeveloperDiskImageDir, "specify DeveloperDiskImage directory")
|
|
mountCmd.Flags().StringVarP(&udid, "udid", "u", "", "specify device by udid")
|
|
iosRootCmd.AddCommand(mountCmd)
|
|
}
|