build/download_translation_unit_tool.py

   1 #!/usr/bin/env python
   2 # Copyright 2017 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Script to download Clang translation_unit tool from google storage."""
   7
   8 import find_depot_tools
   9 import json
  10 import os
  11 import shutil
  12 import subprocess
  13 import sys
  14 import tarfile
  15
  16 SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
  17 CHROME_SRC = os.path.abspath(os.path.join(SCRIPT_DIR, os.pardir))
  18
  19
  20 DEPOT_PATH = find_depot_tools.add_depot_tools_to_path()
  21 GSUTIL_PATH = os.path.join(DEPOT_PATH, 'gsutil.py')
  22
  23 LLVM_BUILD_PATH = os.path.join(CHROME_SRC, 'third_party', 'llvm-build',
  24                                'Release+Asserts')
  25 CLANG_UPDATE_PY = os.path.join(CHROME_SRC, 'tools', 'clang', 'scripts',
  26                                'update.py')
  27
  28 CLANG_BUCKET = 'gs://chromium-browser-clang'
  29
  30
  31 def main():
  32   clang_revision = subprocess.check_output([sys.executable, CLANG_UPDATE_PY,
  33                                             '--print-revision']).rstrip()
  34   targz_name = 'translation_unit-%s.tgz' % clang_revision
  35
  36   if sys.platform == 'win32' or sys.platform == 'cygwin':
  37     cds_full_url = CLANG_BUCKET + '/Win/' + targz_name
  38   elif sys.platform == 'darwin':
  39     cds_full_url = CLANG_BUCKET + '/Mac/' + targz_name
  40   else:
  41     assert sys.platform.startswith('linux')
  42     cds_full_url = CLANG_BUCKET + '/Linux_x64/' + targz_name
  43
  44   os.chdir(LLVM_BUILD_PATH)
  45
  46   subprocess.check_call([sys.executable, GSUTIL_PATH,
  47                          'cp', cds_full_url, targz_name])
  48   tarfile.open(name=targz_name, mode='r:gz').extractall(path=LLVM_BUILD_PATH)
  49
  50   os.remove(targz_name)
  51   return 0
  52
  53 if __name__ == '__main__':
  54   sys.exit(main())