1# Copyright 2013 The Chromium Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5""" A simple device interface for build steps.
6
7"""
8
9import logging
10import os
11import re
12import sys
13
14import build_utils
15
16BUILD_ANDROID_DIR = os.path.join(os.path.dirname(__file__), '..', '..')
17sys.path.append(BUILD_ANDROID_DIR)
18
19from pylib import android_commands
20
21from pylib.android_commands import GetAttachedDevices
22
23
24class BuildDevice(object):
25  def __init__(self, configuration):
26    self.id = configuration['id']
27    self.description = configuration['description']
28    self.install_metadata = configuration['install_metadata']
29    self.adb = android_commands.AndroidCommands(self.id)
30
31  def RunShellCommand(self, *args, **kwargs):
32    return self.adb.RunShellCommand(*args, **kwargs)
33
34  def PushIfNeeded(self, *args, **kwargs):
35    return self.adb.PushIfNeeded(*args, **kwargs)
36
37  def GetSerialNumber(self):
38    return self.id
39
40  def Install(self, *args, **kwargs):
41    return self.adb.Install(*args, **kwargs)
42
43  def GetInstallMetadata(self, apk_package):
44    """Gets the metadata on the device for the apk_package apk."""
45    # Matches lines like:
46    # -rw-r--r-- system   system    7376582 2013-04-19 16:34 \
47    #   org.chromium.chrome.testshell.apk
48    # -rw-r--r-- system   system    7376582 2013-04-19 16:34 \
49    #   org.chromium.chrome.testshell-1.apk
50    apk_matcher = lambda s: re.match('.*%s(-[0-9]*)?.apk$' % apk_package, s)
51    matches = filter(apk_matcher, self.install_metadata)
52    return matches[0] if matches else None
53
54
55def GetConfigurationForDevice(id):
56  adb = android_commands.AndroidCommands(id)
57  configuration = None
58  has_root = False
59  is_online = adb.IsOnline()
60  if is_online:
61    cmd = 'ls -l /data/app; getprop ro.build.description'
62    cmd_output = adb.RunShellCommand(cmd)
63    has_root = not 'Permission denied' in cmd_output[0]
64    if not has_root:
65      # Disable warning log messages from EnableAdbRoot()
66      logging.getLogger().disabled = True
67      has_root = adb.EnableAdbRoot()
68      logging.getLogger().disabled = False
69      cmd_output = adb.RunShellCommand(cmd)
70
71    configuration = {
72        'id': id,
73        'description': cmd_output[-1],
74        'install_metadata': cmd_output[:-1],
75      }
76  return configuration, is_online, has_root
77
78
79def WriteConfigurations(configurations, path):
80  # Currently we only support installing to the first device.
81  build_utils.WriteJson(configurations[:1], path, only_if_changed=True)
82
83
84def ReadConfigurations(path):
85  return build_utils.ReadJson(path)
86
87
88def GetBuildDevice(configurations):
89  assert len(configurations) == 1
90  return BuildDevice(configurations[0])
91
92
93def GetBuildDeviceFromPath(path):
94  configurations = ReadConfigurations(path)
95  if len(configurations) > 0:
96    return GetBuildDevice(ReadConfigurations(path))
97  return None
98
99