| #!/usr/bin/env python3 |
| # Copyright (c) 2021, the R8 project authors. Please see the AUTHORS file |
| # for details. All rights reserved. Use of this source code is governed by a |
| # BSD-style license that can be found in the LICENSE file. |
| |
| import argparse |
| import os |
| import subprocess |
| import sys |
| |
| import jdk |
| import utils |
| |
| VERSION_EXTRACTOR = """ |
| import com.android.tools.r8.Version; |
| public class VersionExtractor { |
| public static void main(String[] args) { |
| System.out.println(Version.LABEL); |
| } |
| } |
| """ |
| |
| |
| def parse_options(): |
| parser = argparse.ArgumentParser(description='Tag R8 Versions') |
| parser.add_argument('--classpath', |
| action='append', |
| help='Dependencies to add to classpath') |
| parser.add_argument('--debug-agent', |
| action='store_true', |
| default=False, |
| help='Create a socket for debugging') |
| parser.add_argument( |
| '--excldeps-variant', |
| action='store_true', |
| default=False, |
| help='Mark this artifact as an "excldeps" variant of the compiler') |
| parser.add_argument('--debug-variant', |
| action='store_true', |
| default=False, |
| help='Compile with debug flag') |
| parser.add_argument( |
| '--lib', |
| action='append', |
| help='Additional libraries (JDK 1.8 rt.jar already included)') |
| parser.add_argument('--output', |
| required=True, |
| help='The output path for the r8lib') |
| parser.add_argument('--pg-conf', action='append', help='Keep configuration') |
| parser.add_argument('--pg-map', |
| default=None, |
| help='Input map for distribution and composition') |
| parser.add_argument('--r8jar', required=True, help='The R8 jar to compile') |
| parser.add_argument('--r8compiler', |
| default='build/libs/r8_with_deps.jar', |
| help='The R8 compiler to use') |
| return parser.parse_args() |
| |
| |
| def get_r8_version(r8jar): |
| with utils.TempDir() as temp: |
| name = os.path.join(temp, "VersionExtractor.java") |
| fd = open(name, 'w') |
| fd.write(VERSION_EXTRACTOR) |
| fd.close() |
| cmd = [jdk.GetJavacExecutable(), '-cp', r8jar, name] |
| print(' '.join(cmd)) |
| cp_separator = ';' if utils.IsWindows() else ':' |
| subprocess.check_call(cmd) |
| output = subprocess.check_output([ |
| jdk.GetJavaExecutable(), '-cp', |
| cp_separator.join([r8jar, os.path.dirname(name)]), |
| 'VersionExtractor' |
| ]).decode('UTF-8').strip() |
| if output == 'main': |
| return subprocess.check_output(['git', 'rev-parse', |
| 'HEAD']).decode('UTF-8').strip() |
| else: |
| return output |
| |
| |
| def main(): |
| args = parse_options() |
| if not os.path.exists(args.r8jar): |
| print("Could not find jar: " + args.r8jar) |
| return 1 |
| version = get_r8_version(args.r8jar) |
| variant = '+excldeps' if args.excldeps_variant else '' |
| map_id_template = version + variant |
| source_file_template = 'R8_%MAP_ID_%MAP_HASH' |
| # TODO(b/139725780): See if we can remove or lower the heap size (-Xmx8g). |
| cmd = [jdk.GetJavaExecutable(), '-Xmx8g', '-ea'] |
| if args.debug_agent: |
| cmd.extend([ |
| '-agentlib:jdwp=transport=dt_socket,server=y,suspend=y,address=*:5005' |
| ]) |
| cmd.extend(['-cp', args.r8compiler, 'com.android.tools.r8.R8']) |
| cmd.append(args.r8jar) |
| if args.debug_variant: |
| cmd.append('--debug') |
| cmd.append('--classfile') |
| cmd.extend(['--map-id-template', map_id_template]) |
| cmd.extend(['--source-file-template', source_file_template]) |
| cmd.extend(['--output', args.output]) |
| cmd.extend(['--pg-map-output', args.output + '.map']) |
| cmd.extend(['--partition-map-output', args.output + '_map.zip']) |
| cmd.extend(['--lib', jdk.GetJdkHome()]) |
| if args.pg_conf: |
| for pgconf in args.pg_conf: |
| cmd.extend(['--pg-conf', pgconf]) |
| if args.lib: |
| for lib in args.lib: |
| cmd.extend(['--lib', lib]) |
| if args.classpath: |
| for cp in args.classpath: |
| cmd.extend(['--classpath', cp]) |
| if args.pg_map: |
| cmd.extend(['--pg-map', args.pg_map]) |
| print(' '.join(cmd)) |
| subprocess.check_call(cmd) |
| |
| |
| if __name__ == '__main__': |
| sys.exit(main()) |