78 lines
2.5 KiB
Go
78 lines
2.5 KiB
Go
package services
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/zhuy1228/go-mobile-uiautomator/adb"
|
||
"github.com/zhuy1228/go-mobile-uiautomator/assets"
|
||
)
|
||
|
||
// 设备端路径常量
|
||
const (
|
||
// DefaultJarRemote 默认设备端 JAR 路径
|
||
DefaultJarRemote = "/data/local/tmp/u2.jar"
|
||
// 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/ 目录
|
||
// JAR 文件通过 go:embed 编译到二进制中，使用者无需关心文件路径
|
||
//
|
||
// addr 为 ADB 服务器地址，serial 为设备序列号
|
||
// force 为 true 时跳过存在性检查，强制推送
|
||
func InstallServiceJar(addr, serial string, force bool) error {
|
||
// 非强制模式下，检查设备端文件是否已存在
|
||
if !force && fileExistsOnDevice(addr, serial, DefaultJarRemote) {
|
||
return nil // 文件已存在，跳过推送
|
||
}
|
||
|
||
_, err := adb.PushData(addr, serial, assets.JarData, DefaultJarRemote, 0644, false)
|
||
if err != nil {
|
||
return fmt.Errorf("推送 u2.jar 失败: %w", err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// InstallServiceApk 将内嵌的 UIAutomator2 APK 推送到设备并安装
|
||
// APK 文件通过 go:embed 编译到二进制中，使用者无需关心文件路径
|
||
//
|
||
// addr 为 ADB 服务器地址，serial 为设备序列号
|
||
// force 为 true 时跳过存在性检查，强制推送
|
||
func InstallServiceApk(addr, serial string, force bool) error {
|
||
// 非强制模式下，检查设备端文件是否已存在
|
||
if !force && fileExistsOnDevice(addr, serial, DefaultApkRemote) {
|
||
// 文件已存在，跳过推送，但仍需确保已安装
|
||
} else {
|
||
_, err := adb.PushData(addr, serial, assets.ApkData, 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
|
||
}
|