完成基础功能

services/install_service.go

@@ -2,63 +2,102 @@ package services
 
 import (
 	"fmt"
+	"os"
 	"path/filepath"
+	"strings"
 	"time"
 
 	"github.com/zhuy1228/go-mobile-uiautomator/adb"
 )
 
-// InstallServiceJar 将 u2.jar 推送到设备的 /data/local/tmp/ 目录
-// addr 为 ADB 服务器地址，serial 为设备序列号
-func InstallServiceJar(addr, serial string) {
-	conn, err := adb.DialADB(addr, 15*time.Second)
+// 默认路径常量
+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 {
-		fmt.Println("连接失败:", err)
-		return
+		return false
 	}
 	defer conn.Close()
 
-	if err := adb.TransportTo(conn, serial); err != nil {
-		fmt.Println("设备路由失败:", err)
-		return
+	// 使用 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
 	}
 
-	targetPath := "/data/local/tmp/u2.jar"
-	sync := adb.InitSync(conn)
-	n, err := sync.SyncPushFile("./assets/u2.jar", targetPath, 0644, true)
-	if err != nil {
-		fmt.Println("推送失败:", err)
-	} else {
-		fmt.Printf("推送成功, 共写入 %d 字节\n", n)
+	// 检查本地文件是否存在
+	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 为设备序列号
-func InstallServiceApk(addr, serial string) {
-	conn, err := adb.DialADB(addr, 15*time.Second)
-	if err != nil {
-		fmt.Println("连接失败:", err)
-		return
+// localPath 为本地 APK 路径，传空字符串则使用默认路径
+// force 为 true 时跳过存在性检查，强制推送
+func InstallServiceApk(addr, serial, localPath string, force bool) error {
+	if localPath == "" {
+		localPath = DefaultApkLocal
 	}
 
-	if err := adb.TransportTo(conn, serial); err != nil {
-		fmt.Println("设备路由失败:", err)
-		conn.Close()
-		return
+	abs, err := filepath.Abs(localPath)
+	if err != nil {
+		return fmt.Errorf("解析 APK 路径失败: %w", err)
 	}
 
-	targetPath := "/data/local/tmp/app-uiautomator.apk"
-	sync := adb.InitSync(conn)
-	abs, _ := filepath.Abs("./assets/app-uiautomator.apk")
-	n, err := sync.SyncPushFile(abs, targetPath, 0644, true)
-	if err != nil {
-		fmt.Println("推送失败:", err)
+	// 检查本地文件是否存在
+	if _, err := os.Stat(abs); os.IsNotExist(err) {
+		return fmt.Errorf("本地 APK 文件不存在: %s", abs)
+	}
+
+	// 非强制模式下，检查设备端文件是否已存在
+	if !force && fileExistsOnDevice(addr, serial, DefaultApkRemote) {
+		// 文件已存在，跳过推送，但仍需确保已安装
 	} else {
-		fmt.Printf("推送成功, 共写入 %d 字节\n", n)
+		_, err = adb.PushFile(addr, serial, abs, DefaultApkRemote, 0644, false)
+		if err != nil {
+			return fmt.Errorf("推送 APK 失败: %w", err)
+		}
 	}
-	conn.Close()
 
 	// 在设备上安装 APK（覆盖安装）
-	adb.InstallApkOnDevice(addr, serial, targetPath, "-r", true)
+	_, err = adb.InstallApkOnDevice(addr, serial, DefaultApkRemote, "-r", false)
+	if err != nil {
+		return fmt.Errorf("安装 APK 失败: %w", err)
+	}
+	return nil
 }