-
Notifications
You must be signed in to change notification settings - Fork 9
/
utils.go
94 lines (85 loc) · 1.9 KB
/
utils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
package stf
import (
"errors"
"fmt"
"io"
"log"
"net/http"
"os"
"strconv"
"strings"
"time"
adb "github.com/openatx/go-adb"
)
func PushFileFromHTTP(d *adb.Device, dst string, perms os.FileMode, urlStr string) error {
wc, err := d.OpenWrite(dst, perms, time.Now())
if err != nil {
return err
}
resp, err := http.Get(urlStr)
if err != nil {
return err
}
if resp.StatusCode != http.StatusOK {
return fmt.Errorf("http download <%s> status %v", urlStr, resp.Status)
}
defer resp.Body.Close()
log.Printf("downloading to %s ...", dst)
if _, err = io.Copy(wc, resp.Body); err != nil {
return err
}
if err := wc.Close(); err != nil {
return err
}
return nil
}
func AdbCheckOutput(d *adb.Device, name string, args ...string) (outStr string, err error) {
args = append(args, ";", "echo", ":$?")
outStr, err = d.RunCommand(name, args...)
if err != nil {
return
}
idx := strings.LastIndexByte(outStr, ':')
if idx == -1 {
return outStr, errors.New("adb shell error, parse exit code failed")
}
exitCode, _ := strconv.Atoi(strings.TrimSpace(outStr[idx+1:]))
if exitCode != 0 {
err = fmt.Errorf("[adb shell %s %s] exit code %d", name, strings.Join(args, " "), exitCode)
}
return outStr[0:idx], err
}
func AdbFileExists(d *adb.Device, path string) bool {
_, err := AdbCheckOutput(d, "test", "-f", path)
return err == nil
}
func GoFunc(f func() error) chan error {
ch := make(chan error)
go func() {
ch <- f()
}()
return ch
}
type multiError struct {
errs []error
}
func (m multiError) Error() string {
var errStrs = make([]string, 0, len(m.errs))
for _, err := range m.errs {
errStrs = append(errStrs, err.Error())
}
return strings.Join(errStrs, "; ")
}
func wrapMultiError(errs ...error) error {
merr := multiError{make([]error, 0)}
for _, err := range errs {
if err == nil {
continue
}
merr.errs = append(merr.errs, err)
}
if len(merr.errs) == 0 {
return nil
}
return merr
}