104 lines
3.1 KiB
Go
104 lines
3.1 KiB
Go
package services
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/zhuy1228/go-mobile-uiautomator/adb"
|
||
)
|
||
|
||
// 默认路径常量
|
||
const (
|
||
// DefaultJarLocal 默认本地 JAR 路径
|
||
DefaultJarLocal = "assets/u2.jar"
|
||
// DefaultJarRemote 默认设备端 JAR 路径
|
||
DefaultJarRemote = "/data/local/tmp/u2.jar"
|
||
// DefaultApkLocal 默认本地 APK 路径
|
||
DefaultApkLocal = "assets/app-uiautomator.apk"
|
||
// DefaultApkRemote 默认设备端 APK 临时路径
|
||
DefaultApkRemote = "/data/local/tmp/app-uiautomator.apk"
|
||
)
|
||
|
||
// fileExistsOnDevice 通过 ADB shell 检查设备端文件是否已存在
|
||
// 返回 true 表示文件已存在，不需要重新推送
|
||
func fileExistsOnDevice(addr, serial, remotePath string) bool {
|
||
conn, err := adb.ConnectToDevice(addr, serial, 10*time.Second)
|
||
if err != nil {
|
||
return false
|
||
}
|
||
defer conn.Close()
|
||
|
||
// 使用 ls 检查文件是否存在
|
||
out, err := adb.ExecShell(conn, fmt.Sprintf("ls %s 2>/dev/null", remotePath))
|
||
if err != nil {
|
||
return false
|
||
}
|
||
return strings.TrimSpace(string(out)) == remotePath
|
||
}
|
||
|
||
// InstallServiceJar 将 u2.jar 推送到设备的 /data/local/tmp/ 目录
|
||
// addr 为 ADB 服务器地址，serial 为设备序列号
|
||
// localPath 为本地 JAR 路径，传空字符串则使用默认路径
|
||
// force 为 true 时跳过存在性检查，强制推送
|
||
func InstallServiceJar(addr, serial, localPath string, force bool) error {
|
||
if localPath == "" {
|
||
localPath = DefaultJarLocal
|
||
}
|
||
|
||
// 检查本地文件是否存在
|
||
if _, err := os.Stat(localPath); os.IsNotExist(err) {
|
||
return fmt.Errorf("本地 JAR 文件不存在: %s", localPath)
|
||
}
|
||
|
||
// 非强制模式下，检查设备端文件是否已存在
|
||
if !force && fileExistsOnDevice(addr, serial, DefaultJarRemote) {
|
||
return nil // 文件已存在，跳过推送
|
||
}
|
||
|
||
_, err := adb.PushFile(addr, serial, localPath, DefaultJarRemote, 0644, false)
|
||
if err != nil {
|
||
return fmt.Errorf("推送 u2.jar 失败: %w", err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// InstallServiceApk 将 UIAutomator2 APK 推送到设备并安装
|
||
// addr 为 ADB 服务器地址，serial 为设备序列号
|
||
// localPath 为本地 APK 路径，传空字符串则使用默认路径
|
||
// force 为 true 时跳过存在性检查，强制推送
|
||
func InstallServiceApk(addr, serial, localPath string, force bool) error {
|
||
if localPath == "" {
|
||
localPath = DefaultApkLocal
|
||
}
|
||
|
||
abs, err := filepath.Abs(localPath)
|
||
if err != nil {
|
||
return fmt.Errorf("解析 APK 路径失败: %w", err)
|
||
}
|
||
|
||
// 检查本地文件是否存在
|
||
if _, err := os.Stat(abs); os.IsNotExist(err) {
|
||
return fmt.Errorf("本地 APK 文件不存在: %s", abs)
|
||
}
|
||
|
||
// 非强制模式下，检查设备端文件是否已存在
|
||
if !force && fileExistsOnDevice(addr, serial, DefaultApkRemote) {
|
||
// 文件已存在，跳过推送，但仍需确保已安装
|
||
} else {
|
||
_, err = adb.PushFile(addr, serial, abs, DefaultApkRemote, 0644, false)
|
||
if err != nil {
|
||
return fmt.Errorf("推送 APK 失败: %w", err)
|
||
}
|
||
}
|
||
|
||
// 在设备上安装 APK（覆盖安装）
|
||
_, err = adb.InstallApkOnDevice(addr, serial, DefaultApkRemote, "-r", false)
|
||
if err != nil {
|
||
return fmt.Errorf("安装 APK 失败: %w", err)
|
||
}
|
||
return nil
|
||
}
|