| // Copyright (c) 2018, 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. |
| |
| package com.android.tools.r8; |
| |
| import com.android.tools.r8.TestBase.Backend; |
| import com.android.tools.r8.references.MethodReference; |
| import com.android.tools.r8.references.TypeReference; |
| import java.io.IOException; |
| import java.nio.file.Path; |
| import java.util.Arrays; |
| import java.util.Collection; |
| import java.util.List; |
| |
| public abstract class TestShrinkerBuilder< |
| C extends BaseCompilerCommand, |
| B extends BaseCompilerCommand.Builder<C, B>, |
| CR extends TestCompileResult<CR, RR>, |
| RR extends TestRunResult, |
| T extends TestShrinkerBuilder<C, B, CR, RR, T>> |
| extends TestCompilerBuilder<C, B, CR, RR, T> { |
| |
| protected boolean enableMinification = true; |
| protected boolean enableTreeShaking = true; |
| |
| TestShrinkerBuilder(TestState state, B builder, Backend backend) { |
| super(state, builder, backend); |
| } |
| |
| public T treeShaking(boolean enable) { |
| enableTreeShaking = enable; |
| return self(); |
| } |
| |
| public T noTreeShaking() { |
| return treeShaking(false); |
| } |
| |
| public T minification(boolean enable) { |
| enableMinification = enable; |
| return self(); |
| } |
| |
| public T noMinification() { |
| return minification(false); |
| } |
| |
| public abstract T addDataEntryResources(DataEntryResource... resources); |
| |
| public abstract T addKeepRuleFiles(List<Path> files); |
| |
| public T addKeepRuleFiles(Path... files) throws IOException { |
| return addKeepRuleFiles(Arrays.asList(files)); |
| } |
| |
| public abstract T addKeepRules(Collection<String> rules); |
| |
| public T addKeepRules(String... rules) { |
| return addKeepRules(Arrays.asList(rules)); |
| } |
| |
| public T addKeepAllClassesRule() { |
| return addKeepRules("-keep class ** { *; }"); |
| } |
| |
| public T addKeepAllInterfacesRule() { |
| return addKeepRules("-keep interface ** { *; }"); |
| } |
| |
| public T addKeepClassRules(Class<?>... classes) { |
| for (Class<?> clazz : classes) { |
| addKeepRules("-keep class " + clazz.getTypeName()); |
| } |
| return self(); |
| } |
| |
| public T addKeepClassAndMembersRules(Class<?>... classes) { |
| for (Class<?> clazz : classes) { |
| addKeepRules("-keep class " + clazz.getTypeName() + " { *; }"); |
| } |
| return self(); |
| } |
| |
| public T addKeepClassAndMembersRulesWithAllowObfuscation(Class<?>... classes) { |
| for (Class<?> clazz : classes) { |
| addKeepRules("-keep,allowobfuscation class " + clazz.getTypeName() + " { *; }"); |
| } |
| return self(); |
| } |
| |
| public T addKeepClassAndDefaultConstructor(Class<?>... classes) { |
| for (Class<?> clazz : classes) { |
| addKeepRules("-keep class " + clazz.getTypeName() + " { <init>(); }"); |
| } |
| return self(); |
| } |
| |
| public T addKeepPackageRules(Package pkg) { |
| return addKeepRules("-keep class " + pkg.getName() + ".*"); |
| } |
| |
| public T addKeepMainRule(Class<?> mainClass) { |
| return addKeepMainRule(mainClass.getTypeName()); |
| } |
| |
| public T addKeepMainRule(String mainClass) { |
| return addKeepRules( |
| "-keep class " + mainClass + " { public static void main(java.lang.String[]); }"); |
| } |
| |
| public T addKeepMethodRules(MethodReference... methods) { |
| for (MethodReference method : methods) { |
| addKeepRules( |
| "-keep class " |
| + method.getHolderClass().getTypeName() |
| + " { " |
| + getMethodLine(method) |
| + " }"); |
| } |
| return self(); |
| } |
| |
| public T addKeepAllAttributes() { |
| return addKeepRules("-keepattributes *"); |
| } |
| |
| public abstract T addApplyMapping(String proguardMap); |
| |
| private static String getMethodLine(MethodReference method) { |
| // Should we encode modifiers in method references? |
| StringBuilder builder = new StringBuilder(); |
| builder.append(method.getMethodName()).append("("); |
| boolean first = true; |
| for (TypeReference parameterType : method.getFormalTypes()) { |
| if (!first) { |
| builder.append(", "); |
| } |
| builder.append(parameterType.getTypeName()); |
| first = false; |
| } |
| return builder.append(");").toString(); |
| } |
| } |