1#!/usr/bin/env python
2#
3# Copyright (C) 2016 The Android Open Source Project
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9#      http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16#
17"""Downloads simpleperf prebuilts from the build server."""
18import argparse
19import logging
20import os
21import shutil
22import stat
23import textwrap
24
25
26THIS_DIR = os.path.realpath(os.path.dirname(__file__))
27
28
29class InstallEntry(object):
30    def __init__(self, target, name, install_path, need_strip=False):
31        self.target = target
32        self.name = name
33        self.install_path = install_path
34        self.need_strip = need_strip
35
36
37install_list = [
38    # simpleperf on device
39    InstallEntry('sdk_arm64-sdk', 'simpleperf', 'android/arm64/simpleperf'),
40    InstallEntry('sdk_arm64-sdk', 'simpleperf32', 'android/arm/simpleperf'),
41    InstallEntry('sdk_x86_64-sdk', 'simpleperf', 'android/x86_64/simpleperf'),
42    InstallEntry('sdk_x86_64-sdk', 'simpleperf32', 'android/x86/simpleperf'),
43
44    # simpleperf on host
45    InstallEntry('sdk_arm64-sdk', 'simpleperf_host', 'linux/x86_64/simpleperf', True),
46    InstallEntry('sdk_arm64-sdk', 'simpleperf_host32', 'linux/x86/simpleperf', True),
47    InstallEntry('sdk_mac', 'simpleperf_host', 'darwin/x86_64/simpleperf'),
48    InstallEntry('sdk_mac', 'simpleperf_host32', 'darwin/x86/simpleperf'),
49    InstallEntry('sdk', 'simpleperf.exe', 'windows/x86_64/simpleperf.exe', True),
50    InstallEntry('sdk', 'simpleperf32.exe', 'windows/x86/simpleperf.exe', True),
51
52    # libsimpleperf_report.so on host
53    InstallEntry('sdk_arm64-sdk', 'libsimpleperf_report.so', 'linux/x86_64/libsimpleperf_report.so', True),
54    InstallEntry('sdk_arm64-sdk', 'libsimpleperf_report32.so', 'linux/x86/libsimpleperf_report.so', True),
55    InstallEntry('sdk_mac', 'libsimpleperf_report.dylib', 'darwin/x86_64/libsimpleperf_report.dylib'),
56    InstallEntry('sdk_mac', 'libsimpleperf_report32.so', 'darwin/x86/libsimpleperf_report.dylib'),
57    InstallEntry('sdk', 'libsimpleperf_report.dll', 'windows/x86_64/libsimpleperf_report.dll', True),
58    InstallEntry('sdk', 'libsimpleperf_report32.dll', 'windows/x86/libsimpleperf_report.dll', True),
59]
60
61
62def logger():
63    """Returns the main logger for this module."""
64    return logging.getLogger(__name__)
65
66
67def check_call(cmd):
68    """Proxy for subprocess.check_call with logging."""
69    import subprocess
70    logger().debug('check_call `%s`', ' '.join(cmd))
71    subprocess.check_call(cmd)
72
73
74def fetch_artifact(branch, build, target, pattern):
75    """Fetches and artifact from the build server."""
76    logger().info('Fetching %s from %s %s (artifacts matching %s)', build,
77                  target, branch, pattern)
78    fetch_artifact_path = '/google/data/ro/projects/android/fetch_artifact'
79    cmd = [fetch_artifact_path, '--branch', branch, '--target', target,
80           '--bid', build, pattern]
81    check_call(cmd)
82
83
84def start_branch(build):
85    """Creates a new branch in the project."""
86    branch_name = 'update-' + (build or 'latest')
87    logger().info('Creating branch %s', branch_name)
88    check_call(['repo', 'start', branch_name, '.'])
89
90
91def commit(branch, build, add_paths):
92    """Commits the new prebuilts."""
93    logger().info('Making commit')
94    check_call(['git', 'add'] + add_paths)
95    message = textwrap.dedent("""\
96        simpleperf: update simpleperf prebuilts to build {build}.
97
98        Taken from branch {branch}.""").format(branch=branch, build=build)
99    check_call(['git', 'commit', '-m', message])
100
101
102def remove_old_release(install_dir):
103    """Removes the old prebuilts."""
104    if os.path.exists(install_dir):
105        logger().info('Removing old install directory "%s"', install_dir)
106        check_call(['git', 'rm', '-rf', '--ignore-unmatch', install_dir])
107
108    # Need to check again because git won't remove directories if they have
109    # non-git files in them.
110    if os.path.exists(install_dir):
111        shutil.rmtree(install_dir)
112
113
114def install_new_release(branch, build, install_dir):
115    """Installs the new release."""
116    for entry in install_list:
117        install_entry(branch, build, install_dir, entry)
118
119
120def install_entry(branch, build, install_dir, entry):
121    """Installs the device specific components of the release."""
122    target = entry.target
123    name = entry.name
124    install_path = os.path.join(install_dir, entry.install_path)
125    need_strip = entry.need_strip
126
127    fetch_artifact(branch, build, target, name)
128    exe_stat = os.stat(name)
129    os.chmod(name, exe_stat.st_mode | stat.S_IEXEC)
130    if need_strip:
131        check_call(['strip', name])
132    dir = os.path.dirname(install_path)
133    if not os.path.isdir(dir):
134        os.makedirs(dir)
135    shutil.move(name, install_path)
136
137
138def get_args():
139    """Parses and returns command line arguments."""
140    parser = argparse.ArgumentParser()
141
142    parser.add_argument(
143        '-b', '--branch', default='aosp-master',
144        help='Branch to pull build from.')
145    parser.add_argument('--build', required=True, help='Build number to pull.')
146    parser.add_argument(
147        '--use-current-branch', action='store_true',
148        help='Perform the update in the current branch. Do not repo start.')
149    parser.add_argument(
150        '-v', '--verbose', action='count', default=0,
151        help='Increase output verbosity.')
152
153    return parser.parse_args()
154
155
156def main():
157    """Program entry point."""
158    os.chdir(THIS_DIR)
159
160    args = get_args()
161    verbose_map = (logging.WARNING, logging.INFO, logging.DEBUG)
162    verbosity = args.verbose
163    if verbosity > 2:
164        verbosity = 2
165    logging.basicConfig(level=verbose_map[verbosity])
166
167    install_dir = 'bin'
168
169    if not args.use_current_branch:
170        start_branch(args.build)
171    remove_old_release(install_dir)
172    install_new_release(args.branch, args.build, install_dir)
173    artifacts = [install_dir]
174    commit(args.branch, args.build, artifacts)
175
176
177if __name__ == '__main__':
178    main()