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 8import find_depot_tools 9import json 10import os 11import shutil 12import subprocess 13import sys 14import tarfile 15 16SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__)) 17CHROME_SRC = os.path.abspath(os.path.join(SCRIPT_DIR, os.pardir)) 18 19 20DEPOT_PATH = find_depot_tools.add_depot_tools_to_path() 21GSUTIL_PATH = os.path.join(DEPOT_PATH, 'gsutil.py') 22 23LLVM_BUILD_PATH = os.path.join(CHROME_SRC, 'third_party', 'llvm-build', 24 'Release+Asserts') 25CLANG_UPDATE_PY = os.path.join(CHROME_SRC, 'tools', 'clang', 'scripts', 26 'update.py') 27 28CLANG_BUCKET = 'gs://chromium-browser-clang' 29 30 31def 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 53if __name__ == '__main__': 54 sys.exit(main()) 55