OLD | NEW |
| (Empty) |
1 # Copyright 2015 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 import contextlib | |
6 import json | |
7 import os | |
8 import platform | |
9 import sys | |
10 import tempfile | |
11 import threading | |
12 | |
13 CATAPULT_ROOT_PATH = os.path.abspath(os.path.join( | |
14 os.path.dirname(__file__), '..', '..', '..', 'third_party', 'catapult')) | |
15 DEPENDENCY_MANAGER_PATH = os.path.join( | |
16 CATAPULT_ROOT_PATH, 'dependency_manager') | |
17 PYMOCK_PATH = os.path.join( | |
18 CATAPULT_ROOT_PATH, 'third_party', 'mock') | |
19 | |
20 | |
21 @contextlib.contextmanager | |
22 def SysPath(path): | |
23 sys.path.append(path) | |
24 yield | |
25 if sys.path[-1] != path: | |
26 sys.path.remove(path) | |
27 else: | |
28 sys.path.pop() | |
29 | |
30 with SysPath(DEPENDENCY_MANAGER_PATH): | |
31 import dependency_manager # pylint: disable=import-error | |
32 | |
33 _ANDROID_BUILD_TOOLS = {'aapt', 'dexdump', 'split-select'} | |
34 | |
35 _DEVIL_DEFAULT_CONFIG = os.path.abspath(os.path.join( | |
36 os.path.dirname(__file__), 'devil_dependencies.json')) | |
37 | |
38 _LEGACY_ENVIRONMENT_VARIABLES = { | |
39 'ADB_PATH': { | |
40 'dependency_name': 'adb', | |
41 'platform': 'linux2_x86_64', | |
42 }, | |
43 'ANDROID_SDK_ROOT': { | |
44 'dependency_name': 'android_sdk', | |
45 'platform': 'linux2_x86_64', | |
46 }, | |
47 } | |
48 | |
49 | |
50 def _GetEnvironmentVariableConfig(): | |
51 path_config = ( | |
52 (os.environ.get(k), v) | |
53 for k, v in _LEGACY_ENVIRONMENT_VARIABLES.iteritems()) | |
54 return { | |
55 'config_type': 'BaseConfig', | |
56 'dependencies': { | |
57 c['dependency_name']: { | |
58 'file_info': { | |
59 c['platform']: { | |
60 'local_paths': [p], | |
61 }, | |
62 }, | |
63 } for p, c in path_config if p | |
64 }, | |
65 } | |
66 | |
67 | |
68 class _Environment(object): | |
69 | |
70 def __init__(self): | |
71 self._dm_init_lock = threading.Lock() | |
72 self._dm = None | |
73 | |
74 def Initialize(self, configs=None, config_files=None): | |
75 """Initialize devil's environment from configuration files. | |
76 | |
77 This uses all configurations provided via |configs| and |config_files| | |
78 to determine the locations of devil's dependencies. Configurations should | |
79 all take the form described by catapult_base.dependency_manager.BaseConfig. | |
80 If no configurations are provided, a default one will be used if available. | |
81 | |
82 Args: | |
83 configs: An optional list of dict configurations. | |
84 config_files: An optional list of files to load | |
85 """ | |
86 | |
87 # Make sure we only initialize self._dm once. | |
88 with self._dm_init_lock: | |
89 if self._dm is None: | |
90 if configs is None: | |
91 configs = [] | |
92 | |
93 env_config = _GetEnvironmentVariableConfig() | |
94 if env_config: | |
95 configs.insert(0, env_config) | |
96 self._InitializeRecursive( | |
97 configs=configs, | |
98 config_files=config_files) | |
99 assert self._dm is not None, 'Failed to create dependency manager.' | |
100 | |
101 def _InitializeRecursive(self, configs=None, config_files=None): | |
102 # This recurses through configs to create temporary files for each and | |
103 # take advantage of context managers to appropriately close those files. | |
104 # TODO(jbudorick): Remove this recursion if/when dependency_manager | |
105 # supports loading configurations directly from a dict. | |
106 if configs: | |
107 with tempfile.NamedTemporaryFile(delete=False) as next_config_file: | |
108 try: | |
109 next_config_file.write(json.dumps(configs[0])) | |
110 next_config_file.close() | |
111 self._InitializeRecursive( | |
112 configs=configs[1:], | |
113 config_files=[next_config_file.name] + (config_files or [])) | |
114 finally: | |
115 if os.path.exists(next_config_file.name): | |
116 os.remove(next_config_file.name) | |
117 else: | |
118 config_files = config_files or [] | |
119 if 'DEVIL_ENV_CONFIG' in os.environ: | |
120 config_files.append(os.environ.get('DEVIL_ENV_CONFIG')) | |
121 config_files.append(_DEVIL_DEFAULT_CONFIG) | |
122 | |
123 self._dm = dependency_manager.DependencyManager( | |
124 [dependency_manager.BaseConfig(c) for c in config_files]) | |
125 | |
126 def FetchPath(self, dependency, arch=None, device=None): | |
127 if self._dm is None: | |
128 self.Initialize() | |
129 if dependency in _ANDROID_BUILD_TOOLS: | |
130 self.FetchPath('android_build_tools_libc++', arch=arch, device=device) | |
131 return self._dm.FetchPath(dependency, GetPlatform(arch, device)) | |
132 | |
133 def LocalPath(self, dependency, arch=None, device=None): | |
134 if self._dm is None: | |
135 self.Initialize() | |
136 return self._dm.LocalPath(dependency, GetPlatform(arch, device)) | |
137 | |
138 | |
139 def GetPlatform(arch=None, device=None): | |
140 if device: | |
141 return 'android_%s' % (arch or device.product_cpu_abi) | |
142 return '%s_%s' % (sys.platform, platform.machine()) | |
143 | |
144 | |
145 config = _Environment() | |
146 | |
OLD | NEW |