Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | # Copyright (C) 2017 The Android Open Source Project |
| 3 | # |
| 4 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | # you may not use this file except in compliance with the License. |
| 6 | # You may obtain a copy of the License at |
| 7 | # |
| 8 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | # |
| 10 | # Unless required by applicable law or agreed to in writing, software |
| 11 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | # See the License for the specific language governing permissions and |
| 14 | # limitations under the License. |
| 15 | |
| 16 | import argparse |
| 17 | import hashlib |
| 18 | import logging |
| 19 | import os |
| 20 | import shutil |
Primiano Tucci | 0825bc8 | 2017-09-28 18:50:23 +0100 | [diff] [blame] | 21 | import subprocess |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 22 | import sys |
| 23 | import urllib |
| 24 | import zipfile |
| 25 | |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 26 | from collections import namedtuple |
| 27 | |
Primiano Tucci | 636ede1 | 2017-10-30 16:03:07 +0000 | [diff] [blame] | 28 | # When adding a new git dependency here please also add a corresponding entry in |
| 29 | # .travis.yml under the "cache:" section. |
| 30 | |
Primiano Tucci | b60d4b0 | 2017-11-10 11:03:00 +0000 | [diff] [blame] | 31 | # The format for the deps below is the following: |
| 32 | # (target_folder, source_url, sha1, target_platform) |
| 33 | # |source_url| can be either a git repo or a http url. |
| 34 | # If a git repo, |sha1| is the committish that will be checked out. |
| 35 | # If a http url, |sha1| is the shasum of the original file. |
| 36 | # If the url is a .zip or .tgz file it will be automatically deflated under |
| 37 | # |target_folder|, taking care of stripping the root folder if it's a single |
| 38 | # root (to avoid ending up with buildtools/protobuf/protobuf-1.2.3/... and have |
| 39 | # instead just buildtools/protobuf). |
| 40 | # |target_platform| is either 'darwin', 'linux2' or 'all' and applies the dep |
| 41 | # only on the given platform (ask python why linux2 and not just linux). |
| 42 | |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 43 | # Dependencies required to build code on the host or when targeting desktop OS. |
| 44 | BUILD_DEPS_HOST = [ |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 45 | # GN |
| 46 | ('buildtools/mac/gn', |
Primiano Tucci | c2eb510 | 2018-05-15 10:40:01 +0100 | [diff] [blame] | 47 | 'https://storage.googleapis.com/chromium-gn/9be792dd9010ce303a9c3a497a67bcc5ac8c7666', |
| 48 | '9be792dd9010ce303a9c3a497a67bcc5ac8c7666', |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 49 | 'darwin' |
| 50 | ), |
| 51 | ('buildtools/linux64/gn', |
Primiano Tucci | c2eb510 | 2018-05-15 10:40:01 +0100 | [diff] [blame] | 52 | 'https://storage.googleapis.com/chromium-gn/2f27ff0b6118e5886df976da5effa6003d19d1ce', |
| 53 | '2f27ff0b6118e5886df976da5effa6003d19d1ce', |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 54 | 'linux2' |
| 55 | ), |
| 56 | |
Primiano Tucci | 104bd6d | 2017-10-12 00:10:24 +0200 | [diff] [blame] | 57 | # clang-format |
| 58 | ('buildtools/mac/clang-format', |
| 59 | 'https://storage.googleapis.com/chromium-clang-format/0679b295e2ce2fce7919d1e8d003e497475f24a3', |
| 60 | '0679b295e2ce2fce7919d1e8d003e497475f24a3', |
| 61 | 'darwin' |
| 62 | ), |
| 63 | ('buildtools/linux64/clang-format', |
| 64 | 'https://storage.googleapis.com/chromium-clang-format/5349d1954e17f6ccafb6e6663b0f13cdb2bb33c8', |
| 65 | '5349d1954e17f6ccafb6e6663b0f13cdb2bb33c8', |
| 66 | 'linux2' |
| 67 | ), |
| 68 | # Keep the SHA1 in sync with |clang_format_rev| in chromium //buildtools/DEPS. |
| 69 | ('buildtools/clang_format/script', |
| 70 | 'https://chromium.googlesource.com/chromium/llvm-project/cfe/tools/clang-format.git', |
| 71 | '0653eee0c81ea04715c635dd0885e8096ff6ba6d', |
| 72 | 'all' |
| 73 | ), |
| 74 | |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 75 | # Ninja |
| 76 | ('buildtools/mac/ninja', |
| 77 | 'https://storage.googleapis.com/fuchsia-build/fuchsia/ninja/mac/a1db595e824c50cf565fbf0af2437fd91b7babf4', |
| 78 | 'a1db595e824c50cf565fbf0af2437fd91b7babf4', |
| 79 | 'darwin' |
| 80 | ), |
| 81 | ('buildtools/linux64/ninja', |
| 82 | 'https://storage.googleapis.com/fuchsia-build/fuchsia/ninja/linux64/d35b36c84a09f7e38b25947cafada10e8bf835bc', |
| 83 | 'd35b36c84a09f7e38b25947cafada10e8bf835bc', |
| 84 | 'linux2' |
| 85 | ), |
| 86 | |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 87 | # Keep in sync with Android's //external/googletest/README.version. |
| 88 | ('buildtools/googletest.zip', |
| 89 | 'https://github.com/google/googletest/archive/ff07a5de0e81580547f1685e101194ed1a4fcd56.zip', |
| 90 | 'c7edec7d7e6db1fc37a20710de9c4d89e3a3893b', |
| 91 | 'all' |
| 92 | ), |
| 93 | |
| 94 | # Keep in sync with Android's //external/protobuf/README.version. |
| 95 | ('buildtools/protobuf.zip', |
| 96 | 'https://github.com/google/protobuf/releases/download/v3.0.0-beta-3/protobuf-cpp-3.0.0-beta-3.zip', |
| 97 | '3caec60aa9d8eefc8c3c3201b6b8ca19935edb89', |
| 98 | 'all' |
| 99 | ), |
| 100 | |
Primiano Tucci | b60d4b0 | 2017-11-10 11:03:00 +0000 | [diff] [blame] | 101 | # libc++, libc++abi and libunwind for Linux where we need to rebuild the C++ |
| 102 | # lib from sources. Keep the SHA1s in sync with Chrome's src/buildtools/DEPS. |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 103 | ('buildtools/libcxx', |
| 104 | 'https://chromium.googlesource.com/chromium/llvm-project/libcxx.git', |
Primiano Tucci | c2eb510 | 2018-05-15 10:40:01 +0100 | [diff] [blame] | 105 | 'ece1de8658d749e19c12cacd4458cc330eca94e3', |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 106 | 'all' |
| 107 | ), |
| 108 | ('buildtools/libcxxabi', |
| 109 | 'https://chromium.googlesource.com/chromium/llvm-project/libcxxabi.git', |
Primiano Tucci | c2eb510 | 2018-05-15 10:40:01 +0100 | [diff] [blame] | 110 | '05a73941f3fb177c4a891d4ff2a4ed5785e3b80c', |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 111 | 'all' |
| 112 | ), |
Primiano Tucci | 7278dea | 2017-10-31 11:50:32 +0000 | [diff] [blame] | 113 | ('buildtools/libunwind', |
| 114 | 'https://chromium.googlesource.com/external/llvm.org/libunwind.git', |
Primiano Tucci | c2eb510 | 2018-05-15 10:40:01 +0100 | [diff] [blame] | 115 | '0f3cbe4123f8afacd646bd4f5414aa6528ef8129', |
Primiano Tucci | 7278dea | 2017-10-31 11:50:32 +0000 | [diff] [blame] | 116 | 'all' |
| 117 | ), |
Hector Dearman | 88a1011 | 2017-10-12 11:07:10 +0100 | [diff] [blame] | 118 | |
Primiano Tucci | b60d4b0 | 2017-11-10 11:03:00 +0000 | [diff] [blame] | 119 | # Keep the revision in sync with Chrome's CLANG_REVISION in |
| 120 | # tools/clang/scripts/update.py. |
| 121 | ('buildtools/clang.tgz', |
Primiano Tucci | c2eb510 | 2018-05-15 10:40:01 +0100 | [diff] [blame] | 122 | 'https://commondatastorage.googleapis.com/chromium-browser-clang/Linux_x64/clang-331747-1.tgz', |
| 123 | '973073ca36ae9194019705ec7677852a30c4b54e', |
Primiano Tucci | b60d4b0 | 2017-11-10 11:03:00 +0000 | [diff] [blame] | 124 | 'linux2' |
| 125 | ), |
| 126 | |
Hector Dearman | 88a1011 | 2017-10-12 11:07:10 +0100 | [diff] [blame] | 127 | # Benchmarking tool. |
| 128 | ('buildtools/benchmark.zip', |
Lalit Maganti | c99d93c | 2018-03-22 15:09:30 +0000 | [diff] [blame] | 129 | 'https://github.com/google/benchmark/archive/v1.3.0.zip', |
| 130 | 'f387e0df37d54bfd5be239e8d0d3ea2e2c3e34f4', |
Hector Dearman | 88a1011 | 2017-10-12 11:07:10 +0100 | [diff] [blame] | 131 | 'all' |
| 132 | ), |
Primiano Tucci | 38faa6f | 2018-04-01 20:12:08 +0200 | [diff] [blame] | 133 | |
| 134 | # Libbacktrace, for stacktraces in Linux/Android debug builds. |
| 135 | ('buildtools/libbacktrace.zip', |
| 136 | 'https://github.com/ianlancetaylor/libbacktrace/archive/177940370e4a6b2509e92a0aaa9749184e64af43.zip', |
| 137 | 'b723fe9d671d1ab54df1297f6afbf2893a41c3ea', |
| 138 | 'all' |
| 139 | ), |
Lalit Maganti | 6fe26e6 | 2018-05-23 12:14:38 +0100 | [diff] [blame] | 140 | |
| 141 | # Sqlite for the trace processing library. |
| 142 | ('buildtools/sqlite.zip', |
| 143 | 'https://storage.googleapis.com/perfetto/sqlite-amalgamation-3230100.zip', |
| 144 | '98729f2c2d57d166e3d6d9862744c1d66388e286', |
| 145 | 'all' |
| 146 | ), |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 147 | ] |
| 148 | |
| 149 | # Dependencies required to build Android code. |
| 150 | # URLs and SHA1s taken from: |
| 151 | # - https://dl.google.com/android/repository/repository-11.xml |
| 152 | # - https://dl.google.com/android/repository/sys-img/android/sys-img.xml |
| 153 | BUILD_DEPS_ANDROID = [ |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 154 | # Android NDK |
| 155 | ('buildtools/ndk.zip', |
| 156 | 'https://dl.google.com/android/repository/android-ndk-r15c-darwin-x86_64.zip', |
| 157 | 'ea4b5d76475db84745aa8828000d009625fc1f98', |
| 158 | 'darwin' |
| 159 | ), |
| 160 | ('buildtools/ndk.zip', |
| 161 | 'https://dl.google.com/android/repository/android-ndk-r15c-linux-x86_64.zip', |
| 162 | '0bf02d4e8b85fd770fd7b9b2cdec57f9441f27a2', |
| 163 | 'linux2' |
| 164 | ), |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 165 | ] |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 166 | |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 167 | # Dependencies required to run Android tests. |
| 168 | TEST_DEPS_ANDROID = [ |
Lalit Maganti | 367fcd5 | 2018-02-05 16:06:13 +0000 | [diff] [blame] | 169 | # Android emulator images. |
| 170 | ('buildtools/aosp-arm.zip', |
| 171 | 'https://storage.googleapis.com/perfetto/aosp-02022018-arm.zip', |
| 172 | 'a480d5e7d3ca888b0a58fe15ce76b1791537429a', |
Primiano Tucci | e7ca7c6 | 2018-04-07 08:28:03 +0200 | [diff] [blame] | 173 | 'all' |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 174 | ), |
| 175 | |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 176 | # platform-tools.zip contains adb binaries. |
| 177 | ('buildtools/android_sdk/platform-tools.zip', |
| 178 | 'https://dl.google.com/android/repository/platform-tools_r26.0.0-darwin.zip', |
| 179 | 'e75b6137dc444f777eb02f44a6d9819b3aabff82', |
| 180 | 'darwin' |
| 181 | ), |
| 182 | ('buildtools/android_sdk/platform-tools.zip', |
| 183 | 'https://dl.google.com/android/repository/platform-tools_r26.0.0-linux.zip', |
| 184 | '00de8a6631405b617c10f68cd11ff2e1cd528e23', |
| 185 | 'linux2' |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 186 | ), |
Primiano Tucci | 0825bc8 | 2017-09-28 18:50:23 +0100 | [diff] [blame] | 187 | |
Lalit Maganti | 367fcd5 | 2018-02-05 16:06:13 +0000 | [diff] [blame] | 188 | # Android emulator binaries. |
| 189 | ('buildtools/emulator', |
| 190 | 'https://android.googlesource.com/platform/prebuilts/android-emulator.git', |
| 191 | '4b260028dc27bc92c39bee9129cb2ba839970956', |
| 192 | 'all' |
Primiano Tucci | 0825bc8 | 2017-09-28 18:50:23 +0100 | [diff] [blame] | 193 | ), |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 194 | ] |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 195 | |
Primiano Tucci | 4bdc4c4 | 2018-05-10 15:52:33 +0100 | [diff] [blame] | 196 | UI_DEPS = [ |
| 197 | ('buildtools/nodejs.tgz', |
Primiano Tucci | dca9609 | 2018-05-31 11:47:16 +0100 | [diff] [blame] | 198 | 'https://storage.googleapis.com/perfetto/node-v10.3.0-darwin-x64.tar.gz', |
| 199 | '6d9a122785f38c256add3b25f74adf125497861a', |
Primiano Tucci | 4bdc4c4 | 2018-05-10 15:52:33 +0100 | [diff] [blame] | 200 | 'darwin' |
| 201 | ), |
| 202 | ('buildtools/nodejs.tgz', |
Primiano Tucci | dca9609 | 2018-05-31 11:47:16 +0100 | [diff] [blame] | 203 | 'https://storage.googleapis.com/perfetto/node-v10.3.0-linux-x64.tar.xz', |
| 204 | '118f6ea19f75089b3f12ac2ddfce357bff872b5e', |
Primiano Tucci | 4bdc4c4 | 2018-05-10 15:52:33 +0100 | [diff] [blame] | 205 | 'linux2' |
| 206 | ), |
| 207 | ('buildtools/emsdk/emscripten.tgz', |
| 208 | 'https://storage.googleapis.com/perfetto/emscripten-1.37.40.tar.gz', |
| 209 | '588c28221321ebbdfc8e3a6f47ea6106f589669b', |
| 210 | 'all' |
| 211 | ), |
| 212 | ('buildtools/emsdk/llvm.tgz', |
| 213 | 'https://storage.googleapis.com/perfetto/emscripten-llvm-e1.37.40-darwin.tar.gz', |
| 214 | '7a894ef0a52821c62f6abaac552dc4ce5d424607', |
| 215 | 'darwin' |
| 216 | ), |
| 217 | ('buildtools/emsdk/llvm.tgz', |
Primiano Tucci | fe19990 | 2018-06-04 12:33:04 +0200 | [diff] [blame] | 218 | 'https://storage.googleapis.com/perfetto/emscripten-llvm-e1.37.40-static-linux.tar.gz', |
| 219 | '478501b9b7a14884e546c84efe209a90052cbb07', |
Primiano Tucci | 4bdc4c4 | 2018-05-10 15:52:33 +0100 | [diff] [blame] | 220 | 'linux2' |
| 221 | ), |
| 222 | ] |
| 223 | |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 224 | ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) |
| 225 | |
| 226 | |
| 227 | def ReadFile(path): |
| 228 | if not os.path.exists(path): |
| 229 | return None |
| 230 | with open(path) as f: |
| 231 | return f.read().strip() |
| 232 | |
| 233 | |
Primiano Tucci | 0825bc8 | 2017-09-28 18:50:23 +0100 | [diff] [blame] | 234 | def MkdirRecursive(path): |
| 235 | # Works with both relative and absolute paths |
| 236 | cwd = '/' if path.startswith('/') else ROOT_DIR |
| 237 | for part in path.split('/'): |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 238 | cwd = os.path.join(cwd, part) |
| 239 | if not os.path.exists(cwd): |
| 240 | os.makedirs(cwd) |
| 241 | else: |
| 242 | assert(os.path.isdir(cwd)) |
| 243 | |
| 244 | |
| 245 | def HashLocalFile(path): |
| 246 | if not os.path.exists(path): |
| 247 | return None |
| 248 | with open(path, 'rb') as f: |
| 249 | return hashlib.sha1(f.read()).hexdigest() |
| 250 | |
| 251 | |
| 252 | def ExtractZipfilePreservePermissions(zf, info, path): |
| 253 | zf.extract(info.filename, path=path) |
| 254 | target_path = os.path.join(path, info.filename) |
| 255 | min_acls = 0o755 if info.filename.endswith('/') else 0o644 |
| 256 | os.chmod(target_path, (info.external_attr >> 16L) | min_acls) |
| 257 | |
| 258 | |
Primiano Tucci | 0825bc8 | 2017-09-28 18:50:23 +0100 | [diff] [blame] | 259 | def IsGitRepoCheckoutOutAtRevision(path, revision): |
| 260 | return ReadFile(os.path.join(path, '.git', 'HEAD')) == revision |
| 261 | |
| 262 | |
| 263 | def CheckoutGitRepo(path, git_url, revision): |
| 264 | if IsGitRepoCheckoutOutAtRevision(path, revision): |
| 265 | return |
| 266 | if os.path.exists(path): |
| 267 | shutil.rmtree(path) |
| 268 | MkdirRecursive(path) |
| 269 | logging.info('Fetching %s @ %s into %s', git_url, revision, path) |
Lalit Maganti | 367fcd5 | 2018-02-05 16:06:13 +0000 | [diff] [blame] | 270 | subprocess.check_call(['git', 'init', path], cwd=path) |
| 271 | subprocess.check_call( |
| 272 | ['git', 'fetch', '--depth', '1', git_url, revision], cwd=path) |
Primiano Tucci | 0825bc8 | 2017-09-28 18:50:23 +0100 | [diff] [blame] | 273 | subprocess.check_call(['git', 'checkout', revision, '--quiet'], cwd=path) |
| 274 | assert(IsGitRepoCheckoutOutAtRevision(path, revision)) |
| 275 | |
| 276 | |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 277 | def Main(): |
| 278 | parser = argparse.ArgumentParser() |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 279 | parser.add_argument('--no-android', action='store_true') |
Primiano Tucci | 4bdc4c4 | 2018-05-10 15:52:33 +0100 | [diff] [blame] | 280 | parser.add_argument('--ui', action='store_true') |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 281 | args = parser.parse_args() |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 282 | deps = BUILD_DEPS_HOST |
| 283 | if not args.no_android: |
| 284 | deps += BUILD_DEPS_ANDROID + TEST_DEPS_ANDROID |
Primiano Tucci | 4bdc4c4 | 2018-05-10 15:52:33 +0100 | [diff] [blame] | 285 | if args.ui: |
| 286 | deps += UI_DEPS |
Primiano Tucci | d775045 | 2017-09-29 14:38:51 +0100 | [diff] [blame] | 287 | for rel_path, url, expected_sha1, platform in deps: |
Lalit Maganti | 367fcd5 | 2018-02-05 16:06:13 +0000 | [diff] [blame] | 288 | if (platform != 'all' and platform != sys.platform): |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 289 | continue |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 290 | local_path = os.path.join(ROOT_DIR, rel_path) |
Primiano Tucci | 0825bc8 | 2017-09-28 18:50:23 +0100 | [diff] [blame] | 291 | if url.endswith('.git'): |
| 292 | CheckoutGitRepo(local_path, url, expected_sha1) |
| 293 | continue |
Primiano Tucci | b60d4b0 | 2017-11-10 11:03:00 +0000 | [diff] [blame] | 294 | is_zip = local_path.endswith('.zip') or local_path.endswith('.tgz') |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 295 | zip_target_dir = local_path[:-4] if is_zip else None |
| 296 | zip_dir_stamp = os.path.join(zip_target_dir, '.stamp') if is_zip else None |
| 297 | |
| 298 | if ((not is_zip and HashLocalFile(local_path) == expected_sha1) or |
| 299 | (is_zip and ReadFile(zip_dir_stamp) == expected_sha1)): |
| 300 | continue |
| 301 | MkdirRecursive(os.path.dirname(rel_path)) |
| 302 | if HashLocalFile(local_path) != expected_sha1: |
| 303 | download_path = local_path + '.tmp' |
| 304 | logging.info('Downloading %s from %s', local_path, url) |
| 305 | urllib.urlretrieve(url, download_path) |
| 306 | os.chmod(download_path, 0o755) |
| 307 | if (HashLocalFile(download_path) != expected_sha1): |
| 308 | os.remove(download_path) |
| 309 | logging.fatal('SHA1 mismatch for %s', download_path) |
| 310 | return 1 |
| 311 | os.rename(download_path, local_path) |
| 312 | assert(HashLocalFile(local_path) == expected_sha1) |
| 313 | |
| 314 | if is_zip: |
| 315 | logging.info('Extracting %s into %s' % (local_path, zip_target_dir)) |
| 316 | assert(os.path.commonprefix((ROOT_DIR, zip_target_dir)) == ROOT_DIR) |
| 317 | if os.path.exists(zip_target_dir): |
| 318 | logging.info('Deleting stale dir %s' % zip_target_dir) |
| 319 | shutil.rmtree(zip_target_dir) |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 320 | |
Primiano Tucci | b60d4b0 | 2017-11-10 11:03:00 +0000 | [diff] [blame] | 321 | # Decompress the archive. |
| 322 | if local_path.endswith('.tgz'): |
| 323 | MkdirRecursive(zip_target_dir) |
Primiano Tucci | 4bdc4c4 | 2018-05-10 15:52:33 +0100 | [diff] [blame] | 324 | subprocess.check_call(['tar', '-xf', local_path], cwd=zip_target_dir) |
Primiano Tucci | b60d4b0 | 2017-11-10 11:03:00 +0000 | [diff] [blame] | 325 | elif local_path.endswith('.zip'): |
| 326 | with zipfile.ZipFile(local_path, 'r') as zf: |
| 327 | for info in zf.infolist(): |
| 328 | ExtractZipfilePreservePermissions(zf, info, zip_target_dir) |
| 329 | |
| 330 | # If the zip contains one root folder, rebase one level up moving all |
| 331 | # its sub files and folders inside |target_dir|. |
| 332 | subdir = os.listdir(zip_target_dir) |
| 333 | if len(subdir) == 1: |
| 334 | subdir = os.path.join(zip_target_dir, subdir[0]) |
| 335 | if os.path.isdir(subdir): |
| 336 | for subf in os.listdir(subdir): |
| 337 | shutil.move(os.path.join(subdir,subf), zip_target_dir) |
| 338 | os.rmdir(subdir) |
| 339 | |
| 340 | # Create stamp and remove the archive. |
| 341 | with open(zip_dir_stamp, 'w') as stamp_file: |
| 342 | stamp_file.write(expected_sha1) |
| 343 | os.remove(local_path) |
Primiano Tucci | ae2879e | 2017-09-27 11:02:09 +0900 | [diff] [blame] | 344 | |
| 345 | |
| 346 | if __name__ == '__main__': |
| 347 | logging.basicConfig(level=logging.INFO) |
| 348 | sys.exit(Main()) |