sync.go

package adb

import (
	"bytes"
	"context"
	"errors"
	"fmt"
	"io/fs"
	"os"
	"path/filepath"
	"time"

	"github.com/prife/goadb/wire"
)

func ListAllSubDirs(localDir string) (list []string, err error) {
	err = filepath.WalkDir(localDir, func(path string, d fs.DirEntry, err error) error {
		if err != nil {
			return err
		}
		if path == localDir {
			return nil
		}
		// ignore special file
		if d.IsDir() {
			relativePath, _ := filepath.Rel(localDir, path)
			list = append(list, relativePath)
		}
		return nil
	})
	return
}

// adb shell mkdir
// # Android 14
//
// OP5929L1:/ $ mkdir /a /b /c
// mkdir: '/a': Read-only file system
// mkdir: '/b': Read-only file system
// mkdir: '/c': Read-only file system
// 1|OP5929L1:/ $
//
// OP5929L1:/ $ mkdir /sdcard/a /sdcard/b /sdcard/c
// OP5929L1:/ $ mkdir /sdcard/a /sdcard/b /sdcard/c
// mkdir: '/sdcard/a': File exists
// mkdir: '/sdcard/b': File exists
// mkdir: '/sdcard/c': File exists
// 1|OP5929L1:/ $
//
// OP5929L1:/ $ mkdir /data/a /data/b /data/c
// mkdir: '/data/a': Permission denied
// mkdir: '/data/b': Permission denied
// mkdir: '/data/c': Permission denied
// 1|OP5929L1:/ $
//
// OP5929L1:/ $ mkdir /sd/a /sd/b /sd/c
// mkdir: '/sd/a': No such file or directory
// mkdir: '/sd/b': No such file or directory
// mkdir: '/sd/c': No such file or directory
//
// # Android 5.1
//
// shell@A33:/ $ mkdir /sdcard/a /sdcard/b /sdcard/c
// shell@A33:/ $ mkdir /sdcard/a /sdcard/b /sdcard/c
// mkdir failed for /sdcard/a, File exists
// 255|shell@A33:/ $
//
// shell@A33:/ $ mkdir /sd/a /sd/b /sd/c
// mkdir failed for /sd/a, No such file or directory
// 255|shell@A33:/ $
//
// shell@A33:/ $ mkdir /a /b /c
// mkdir failed for /a, Read-only file system
// 255|shell@A33:/ $
//
// shell@A33:/ $ mkdir /data/a /data/b /data/c
// mkdir failed for /data/a, Permission denied
// 255|shell@A33:/ $
func filterFileExistedError(resp []byte) (errs []error) {
	lines := bytes.Split(resp, []byte("\n"))
	for _, line := range lines {
		line := bytes.TrimSpace(line)
		if len(line) > 0 && !bytes.HasSuffix(line, []byte("File exists")) {
			errs = append(errs, errors.New(string(line)))
		}
	}
	return
}

func (c *Device) Mkdirs(list []string) error {
	return c.MkdirsWithParent(list, false)
}

// adb shell mkdir [-p] <dir1> <dir2> ...
func (c *Device) MkdirsWithParent(list []string, withParent bool) error {
	var commands []string
	var commandsLen int

	var errs []error
	if withParent {
		commands = append(commands, "-p")
	}
	for _, l := range list {
		// adb 这里的长度是32768，但是由于wire/conn.go 中判断最大长度为 MaxPayloadV1Length 4096
		// 因此这里使用 4000
		if commandsLen+len(l) > 4000 {
			resp, err := c.RunCommandTimeout(time.Second*15, "mkdir", commands...)
			if err != nil {
				return err
			}

			if len(resp) > 0 {
				// fmt.Println("resp:", string(resp))
				errs = append(errs, filterFileExistedError(resp)...)
			}
			commands = make([]string, 0)
			if withParent {
				commands = append(commands, "-p")
			}
			commandsLen = 0
		}

		commands = append(commands, l)
		commandsLen = commandsLen + len(l) + 1 // and one space
	}

	if commandsLen > 0 {
		resp, err := c.RunCommandTimeout(time.Second*15, "mkdir", commands...)
		if err != nil {
			return err
		}
		if len(resp) > 0 {
			errs = append(errs, filterFileExistedError(resp)...)
		}
	}

	if len(errs) > 0 {
		return errors.Join(errs...)
	}
	return nil
}

// Rm run `adb shell rm -rf xx xx`
// it returns is meaning less in most cases, so just ignore error is ok
func (c *Device) Rm(list []string) error {
	var commands []string
	var commandsLen int

	var errs []error

	commands = append(commands, "-rf")
	for _, l := range list {
		if commandsLen+len(l) > (32768 - 7) { // len('rm -rf ') == 6
			resp, err := c.RunCommandTimeout(time.Second*15, "rm", commands...)
			if err != nil {
				return err
			}

			if len(resp) > 0 {
				errs = append(errs, errors.New(string(resp)))
			}

			// reset commands
			commands = make([]string, 0)
			commands = append(commands, "-rf")
			commandsLen = 0
		}

		commands = append(commands, l)
		commandsLen = commandsLen + len(l) + 1 // and one space
	}

	if commandsLen > 0 {
		resp, err := c.RunCommandTimeout(time.Second*15, "rm", commands...)
		if err != nil {
			return err
		}
		if len(resp) > 0 {
			errs = append(errs, errors.New(string(resp)))
		}
	}
	return errors.Join(errs...)
}

func (c *Device) PushFile(localPath, remotePath string, handler wire.SyncFileHandler) error {
	return c.PushFileCtx(context.Background(), localPath, remotePath, handler)
}

func (c *Device) PushFileCtx(ctx context.Context, localPath, remotePath string, handler wire.SyncFileHandler) error {
	linfo, err := os.Lstat(localPath)
	if err != nil {
		return err
	}
	if !linfo.Mode().IsRegular() {
		return fmt.Errorf("not regular file: %s", localPath)
	}

	// features, err := c.DeviceFeatures()
	// if err != nil {
	// 	return fmt.Errorf("get device features: %w", err)
	// }

	fconn, err := c.NewSyncConn()
	if err != nil {
		return err
	}
	defer fconn.Close()

	// if remotePath is dir, just append src file name
	rinfo, err := fconn.Stat(remotePath)
	if err == nil && rinfo.Mode.IsDir() {
		remotePath = remotePath + "/" + linfo.Name()
	}

	var syncHandler func(n uint64)
	if handler != nil {
		total := uint64(linfo.Size())
		sent := uint64(0)
		startTime := time.Now()
		percent := 0
		syncHandler = func(n uint64) {
			sent += n
			curPercent := float64(sent) / float64(total) * 100
			if int(curPercent) > percent {
				speedMBPerSecond := float64(sent) * float64(time.Second) / 1024.0 / 1024.0 / (float64(time.Since(startTime)))
				handler(total, sent, curPercent, speedMBPerSecond)
			}
			percent = int(curPercent)
		}
	}

	ch := make(chan error, 2)
	go func() {
		err := fconn.PushFile(localPath, remotePath, syncHandler)
		ch <- err
	}()

	select {
	case <-ctx.Done():
		return fmt.Errorf("push failed by ctx done: %w", ctx.Err())
	case err := <-ch:
		if err != nil {
			return fmt.Errorf("push failed: %w", err)
		}
		return nil
	}
}

// PushDir support push dir
// push 文件夹:
// adb push src-dir dest-dir具有两种行为，与cp命令效果一致
// 1.如果'dest-dir'路径不存在，会创建'dest-dir'，其内容与`src-dir`完全一致
// 2.如果'dest-dir'路径存在，会创建'dest-dir/src-dir'，其内容与`src-dir`完全一致
//
// 本函数行为如下
// 当 withSrcDir 为 true，永远会在手机上创建src-dir
// 当 withSrcDir 为 false，则仅会 src-dir 的子文件/目录推送到目标文件夹下
func (c *Device) PushDir(local, remote string, withSrcDir bool, handler wire.SyncHandler) (err error) {
	return c.PushDirCtx(context.Background(), local, remote, withSrcDir, handler)
}

func (c *Device) PushDirCtx(ctx context.Context, local, remote string, withSrcDir bool, handler wire.SyncHandler) (err error) {
	// Android 12 之后，push 可能遇到文件夹权限问题，解决办法
	// 1. 先在手机上创建所有文件夹，如果失败则直接返回错误
	// 2. 再推送文件
	if err := MakeDirs(c, local, remote, withSrcDir); err != nil {
		return err
	}

	// push files
	fconn, err := c.NewSyncConn()
	if err != nil {
		return err
	}
	defer fconn.Close()

	ch := make(chan error, 2)
	go func() {
		err := fconn.PushDir(withSrcDir, local, remote, handler)
		ch <- err
	}()

	select {
	case <-ctx.Done():
		return fmt.Errorf("push failed by ctx done: %w", ctx.Err())
	case err := <-ch:
		if err != nil {
			return fmt.Errorf("push failed: %w", err)
		}
		return nil
	}
}

func MakeDirs(c *Device, local string, remote string, withSrcDir bool) (err error) {
	local, err = filepath.Abs(local)
	if err != nil {
		return fmt.Errorf("pushd: get abs path of %s failed: %w", local, err)
	}
	var baseName string
	if withSrcDir {
		// use filepath.Base, not path.Base
		// https://stackoverflow.com/questions/48050724/how-to-get-correct-file-base-name-on-windows-using-golang
		baseName = filepath.Base(local)
	}

	linfo, err := os.Lstat(local)
	if err != nil {
		return err
	}

	if !linfo.IsDir() {
		return fmt.Errorf("not dir: %s", local)
	}

	// mkdir sub dirs
	subdirs, err := ListAllSubDirs(local)
	if err != nil {
		return
	}
	remoteSubDirs := make([]string, 1+len(subdirs))
	if baseName != "" {
		remoteSubDirs[0] = remote + "/" + baseName
		for i, d := range subdirs {
			remoteSubDirs[i+1] = remote + "/" + baseName + "/" + d
		}
	} else {
		remoteSubDirs[0] = remote
		for i, d := range subdirs {
			remoteSubDirs[i+1] = remote + "/" + d
		}
	}
	err = c.MkdirsWithParent(remoteSubDirs, true)
	if err != nil {
		// 当创建很多文件夹时(比如推送游戏资源包到手机中)，可能会返回一个超长的错误，截断处理
		errStr := err.Error()
		if len(errStr) > 1024 {
			errStr = errStr[:1024]
		}
		return fmt.Errorf("mkdirs failed: %s", errStr)
	}
	return nil
}