| 25 | } |
| 26 | |
| 27 | func NewUIADriver(capabilities Capabilities, urlPrefix string) (driver *uiaDriver, err error) { |
| 28 | log.Info().Msg("init uiautomator2 driver") |
| 29 | if capabilities == nil { |
| 30 | capabilities = NewCapabilities() |
| 31 | } |
| 32 | driver = new(uiaDriver) |
| 33 | if driver.urlPrefix, err = url.Parse(urlPrefix); err != nil { |
| 34 | return nil, err |
| 35 | } |
| 36 | var localPort int |
| 37 | { |
| 38 | tmpURL, _ := url.Parse(driver.urlPrefix.String()) |
| 39 | hostname := tmpURL.Hostname() |
| 40 | if strings.HasPrefix(hostname, forwardToPrefix) { |
| 41 | localPort, _ = strconv.Atoi(strings.TrimPrefix(hostname, forwardToPrefix)) |
| 42 | } |
| 43 | } |
| 44 | conn, err := net.Dial("tcp", fmt.Sprintf(":%d", localPort)) |
| 45 | if err != nil { |
| 46 | return nil, fmt.Errorf("adb forward: %w", err) |
| 47 | } |
| 48 | driver.client = convertToHTTPClient(conn) |
| 49 | |
| 50 | session, err := driver.NewSession(capabilities) |
| 51 | if err != nil { |
| 52 | return nil, errors.Wrap(err, "create UIAutomator session failed") |
| 53 | } |
| 54 | driver.sessionId = session.SessionId |
| 55 | return driver, nil |
| 56 | } |
| 57 | |
| 58 | type BatteryStatus int |
| 59 | |