OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # | |
3 # Copyright 2016 Google Inc. | |
4 # | |
5 # Use of this source code is governed by a BSD-style license that can be | |
6 # found in the LICENSE file. | |
7 | |
8 | |
9 """Isolate a locally-managed Android SDK.""" | |
10 | |
11 | |
12 import argparse | |
13 import os | |
14 import shlex | |
15 import shutil | |
16 import subprocess | |
17 import sys | |
18 import utils | |
19 | |
20 | |
21 INFRA_BOTS_DIR = os.path.realpath(os.path.dirname(os.path.abspath(__file__))) | |
22 ISOLATE_FILE_NAME = 'android_sdk.isolate' | |
23 REPO_SKIA = 'https://skia.googlesource.com/skia.git' | |
24 SDK_DIR_NAME = 'android-sdk' | |
25 | |
26 | |
27 def get_isolate_binary(): | |
28 """Find or, if necessary, obtain the isolate binary.""" | |
29 # Try to find isolate locally. | |
30 platform = 'linux64' | |
31 if sys.platform == 'win32': | |
32 platform = 'win64' | |
33 elif sys.platform == 'darwin': | |
34 platform = 'mac64' | |
35 repo_isolate = os.path.join(INFRA_BOTS_DIR, | |
36 'tools', 'luci-go', platform) | |
37 path = os.pathsep.join((repo_isolate, os.environ['PATH'])) | |
38 try: | |
39 output = subprocess.check_output( | |
40 ['which', 'isolate'], | |
41 env={'PATH':path}).rstrip() | |
42 print 'Found isolate binary: %s' % output | |
43 return output | |
44 except subprocess.CalledProcessError: | |
45 pass | |
46 | |
47 # Download isolate from GS. | |
48 print 'Unable to find isolate binary; attempting to download...' | |
49 try: | |
50 subprocess.check_call( | |
51 ['download_from_google_storage', | |
52 '--bucket', 'chromium-luci', | |
53 '-d', repo_isolate]) | |
54 except OSError as e: | |
55 raise Exception('Failed to download isolate binary. ' | |
56 'Is depot_tools in PATH? Error: %s' % e) | |
57 except subprocess.CalledProcessError as e: | |
58 raise Exception('Failed to download isolate binary. ' | |
59 'Are you authenticated to Google Storage? Error: %s' % e) | |
60 | |
61 output = subprocess.check_output( | |
62 ['which', 'isolate'], | |
63 env={'PATH':path}).rstrip() | |
64 return output | |
65 | |
66 | |
67 def check_isolate_auth(isolate): | |
68 """Ensure that we're authenticated to the isolate server.""" | |
69 not_logged_in = 'Not logged in' | |
70 try: | |
71 output = subprocess.check_output([isolate, 'whoami']) | |
72 | |
73 except subprocess.CalledProcessError: | |
74 output = not_logged_in | |
75 if output == not_logged_in: | |
76 raise Exception('Not authenticated to isolate server. You probably need to ' | |
77 'run:\n$ %s login' % isolate) | |
78 | |
79 | |
80 def isolate_android_sdk(android_sdk_root): | |
81 """Isolate the Android SDK and return the isolated hash.""" | |
82 repo_isolate_file = os.path.join(INFRA_BOTS_DIR, ISOLATE_FILE_NAME) | |
83 with utils.tmp_dir(): | |
84 # Copy the SDK dir contents into a directory with a known name. | |
85 sdk_dir = os.path.join(os.getcwd(), SDK_DIR_NAME) | |
86 shutil.copytree(android_sdk_root, sdk_dir) | |
87 isolate_file = os.path.join(os.getcwd(), ISOLATE_FILE_NAME) | |
88 shutil.copyfile(repo_isolate_file, isolate_file) | |
89 | |
90 # Isolate the SDK. | |
91 isolate = get_isolate_binary() | |
92 check_isolate_auth(isolate) | |
93 android_sdk_relpath = os.path.relpath( | |
94 sdk_dir, os.path.dirname(isolate_file)) | |
95 isolate_cmd = [isolate, 'archive', '--quiet', | |
96 '--isolate-server', 'https://isolateserver.appspot.com', | |
97 '-i', isolate_file, | |
98 '-s', 'android_sdk.isolated', | |
99 '--extra-variable', 'ANDROID_SDK_DIR=%s' % android_sdk_relpath] | |
100 isolate_out = subprocess.check_output(isolate_cmd).rstrip() | |
101 return shlex.split(isolate_out)[0] | |
102 | |
103 | |
104 def update_sdk_file(skia_path, isolated_hash): | |
105 """Edit the android_sdk_hash file, upload a CL.""" | |
106 with utils.chdir(skia_path): | |
107 with utils.git_branch(): | |
108 hash_file = os.path.join('infra', 'bots', 'android_sdk_hash') | |
109 with open(hash_file, 'w') as f: | |
110 f.write(isolated_hash) | |
111 subprocess.check_call([utils.GIT, 'add', hash_file]) | |
112 subprocess.check_call([utils.GIT, 'commit', '-m', 'Update Android SDK']) | |
113 subprocess.check_call([utils.GIT, 'cl', 'upload', '--bypass-hooks']) | |
114 | |
115 | |
116 def main(): | |
117 parser = argparse.ArgumentParser() | |
118 parser.add_argument('--android_sdk_root', required=True) | |
119 args = parser.parse_args() | |
120 skia_path = os.path.abspath(os.path.join(INFRA_BOTS_DIR, | |
121 os.pardir, os.pardir)) | |
122 | |
123 with utils.print_timings(): | |
124 isolated_hash = isolate_android_sdk(args.android_sdk_root) | |
125 update_sdk_file(skia_path, isolated_hash) | |
126 | |
127 | |
128 if __name__ == '__main__': | |
129 main() | |
OLD | NEW |