| #!/bin/bash -eu |
| |
| set -o pipefail |
| |
| function create_base_files { |
| mkdir -p soong-test/java/integration/impllib |
| cat > soong-test/java/integration/Android.bp <<'EOF' |
| java_library { |
| name: "impl-library", |
| srcs: [ |
| "impllib/**/*.java", |
| ], |
| libs: ["provider-library"], |
| sdk_version: "35", |
| } |
| |
| java_library { |
| name: "provider-library", |
| srcs: ["provider/**/*.java"], |
| sdk_version: "35", |
| } |
| EOF |
| |
| cat > soong-test/java/integration/impllib/ExampleImpl1.java <<'EOF' |
| package soong.java.integration.impllib; |
| |
| import soong.java.integration.provider.Provider; |
| |
| class ExampleImpl1 { |
| public static final int EXAMPLE_1_CONST_1 = 100; |
| |
| private Provider mProvider; |
| |
| public ExampleImpl1() { |
| mProvider = new Provider(); |
| } |
| |
| public String getClassName() { |
| return "ExampleInt1"; |
| } |
| |
| public int getProviderConstant() { |
| return mProvider.nextProviderConst(); |
| } |
| } |
| EOF |
| |
| mkdir -p soong-test/java/integration/provider |
| |
| cat > soong-test/java/integration/provider/Provider.java <<'EOF' |
| package soong.java.integration.provider; |
| |
| public class Provider { |
| public static final int PROVIDER_CONST = -1; |
| |
| private String className; |
| |
| public Provider() { |
| this.className = "Provider"; |
| } |
| |
| public int nextProviderConst() { |
| return PROVIDER_CONST + 1; |
| } |
| |
| private String getProviderName() { |
| return className; |
| } |
| } |
| EOF |
| } |
| |
| function create_ap_files { |
| mkdir -p soong-test/java/integration/impllib |
| cat > soong-test/java/integration/Android.bp <<'EOF' |
| java_library { |
| name: "impl-library", |
| srcs: [ |
| "impllib/**/*.java", |
| ":wrapper-annotations-source", |
| ], |
| libs: ["provider-library"], |
| plugins: ["wrapper-annotation-processor"], |
| sdk_version: "35", |
| } |
| |
| java_library { |
| name: "provider-library", |
| srcs: ["provider/**/*.java"], |
| sdk_version: "35", |
| } |
| |
| java_library_host { |
| name: "annotation-processor-lib", |
| srcs: ["annotation/**/*.java"], |
| } |
| |
| java_plugin { |
| name: "wrapper-annotation-processor", |
| processor_class: "soong.java.integration.annotation.WrapperProcessor", |
| static_libs: ["annotation-processor-lib"], |
| installable: false, |
| generates_api: true, |
| } |
| |
| filegroup { |
| name: "wrapper-annotations-source", |
| srcs: ["annotation/GenerateWrapper.java"], |
| } |
| EOF |
| |
| mkdir -p soong-test/java/integration/annotation |
| cat > soong-test/java/integration/annotation/WrapperProcessor.java <<'EOF' |
| package soong.java.integration.annotation; |
| import soong.java.integration.annotation.GenerateWrapper; |
| import java.io.IOException; |
| import java.io.Writer; |
| import java.util.Set; |
| import javax.annotation.processing.AbstractProcessor; |
| import javax.annotation.processing.Filer; |
| import javax.annotation.processing.Messager; |
| import javax.annotation.processing.ProcessingEnvironment; |
| import javax.annotation.processing.Processor; |
| import javax.annotation.processing.RoundEnvironment; |
| import javax.annotation.processing.SupportedAnnotationTypes; |
| import javax.annotation.processing.SupportedSourceVersion; |
| import javax.lang.model.SourceVersion; |
| import javax.lang.model.element.Element; |
| import javax.lang.model.element.ElementKind; |
| import javax.lang.model.element.ExecutableElement; |
| import javax.lang.model.element.TypeElement; |
| import javax.lang.model.util.Elements; |
| import javax.tools.Diagnostic; |
| import javax.tools.JavaFileObject; |
| @SupportedAnnotationTypes("soong.java.integration.annotation.GenerateWrapper") |
| @SupportedSourceVersion(SourceVersion.RELEASE_17) |
| public class WrapperProcessor extends AbstractProcessor { |
| private Filer filer; |
| private Messager messager; |
| private Elements elementUtils; |
| @Override |
| public synchronized void init(ProcessingEnvironment processingEnv) { |
| super.init(processingEnv); |
| filer = processingEnv.getFiler(); |
| messager = processingEnv.getMessager(); |
| elementUtils = processingEnv.getElementUtils(); |
| } |
| @Override |
| public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) { |
| for (Element annotatedElement : roundEnv.getElementsAnnotatedWith(GenerateWrapper.class)) { |
| ExecutableElement originalMethod = (ExecutableElement) annotatedElement; |
| GenerateWrapper annotation = originalMethod.getAnnotation(GenerateWrapper.class); |
| int methodCount = annotation.methodCount(); |
| TypeElement enclosingClass = (TypeElement) originalMethod.getEnclosingElement(); |
| String packageName = elementUtils.getPackageOf(enclosingClass).getQualifiedName().toString(); |
| String originalClassNameForNaming = enclosingClass.getSimpleName().toString(); |
| String wrapperClassName = originalClassNameForNaming + "Wrapper"; |
| String wrapperMethodName = originalMethod.getSimpleName().toString() + "Wrapper"; |
| try { |
| generateAlwaysVoidNoParamWrapper(packageName, wrapperClassName, wrapperMethodName, methodCount); |
| } catch (IOException e) { |
| throw new RuntimeException("Unable to generate class using annotation"); |
| } |
| } |
| return true; |
| } |
| private void generateAlwaysVoidNoParamWrapper(String packageName, String wrapperClassName, String originalMethodNameForNaming, int methodCount) throws IOException { |
| StringBuilder sb = new StringBuilder(); |
| sb.append("package ").append(packageName).append(";\n\n"); |
| sb.append("public final class ").append(wrapperClassName).append(" {\n\n"); |
| for (int i=0; i < methodCount; i++) { |
| sb.append(" public String ").append(originalMethodNameForNaming + String.valueOf(i)).append("() {\n"); |
| sb.append(" return ").append("\"").append(wrapperClassName).append("\"").append(";\n"); |
| sb.append(" }\n"); |
| } |
| sb.append("}\n"); |
| JavaFileObject jfo = filer.createSourceFile(packageName + "." + wrapperClassName); |
| try (Writer writer = jfo.openWriter()) { |
| writer.write(sb.toString()); |
| } |
| } |
| } |
| EOF |
| |
| cat > soong-test/java/integration/annotation/GenerateWrapper.java <<'EOF' |
| package soong.java.integration.annotation; |
| import java.lang.annotation.ElementType; |
| import java.lang.annotation.Retention; |
| import java.lang.annotation.RetentionPolicy; |
| import java.lang.annotation.Target; |
| @Retention(RetentionPolicy.SOURCE) |
| @Target(ElementType.METHOD) |
| public @interface GenerateWrapper { |
| int methodCount() default 1; |
| } |
| EOF |
| |
| cat > soong-test/java/integration/impllib/AnnotationUsage.java <<'EOF' |
| package soong.java.integration.impllib; |
| import soong.java.integration.annotation.GenerateWrapper; |
| public class AnnotationUsage { |
| @GenerateWrapper(methodCount = 2) |
| public String getClassName() { |
| return "AnnotationUsage"; |
| } |
| } |
| EOF |
| |
| cat > soong-test/java/integration/impllib/ExampleImpl4.java <<'EOF' |
| package soong.java.integration.impllib; |
| import soong.java.integration.impllib.AnnotationUsageWrapper; |
| public class ExampleImpl4 { |
| private AnnotationUsageWrapper mAnnotationUsageWrapper; |
| public ExampleImpl4() { |
| this.mAnnotationUsageWrapper = new AnnotationUsageWrapper(); |
| } |
| public String readGenApi() { |
| return mAnnotationUsageWrapper.getClassNameWrapper1(); |
| } |
| } |
| EOF |
| } |
| |
| create_example_impl3_file() { |
| mkdir -p soong-test/java/integration/impllib |
| |
| cat > soong-test/java/integration/impllib/ExampleImpl3.java <<'EOF' |
| package soong.java.integration.impllib; |
| |
| import soong.java.integration.impllib.ExampleImpl1; |
| |
| public class ExampleImpl3 { |
| |
| public ExampleImpl3() { |
| } |
| |
| public String getClassName() { |
| return "ExampleImpl3"; |
| } |
| |
| private int getConstant() { |
| return ExampleImpl1.EXAMPLE_1_CONST_1; |
| } |
| } |
| EOF |
| } |
| |
| modify_example_impl1_file() { |
| mkdir -p soong-test/java/integration/impllib |
| |
| cat > soong-test/java/integration/impllib/ExampleImpl1.java <<'EOF' |
| package soong.java.integration.impllib; |
| |
| import soong.java.integration.provider.Provider; |
| |
| class ExampleImpl1 { |
| public static final int EXAMPLE_1_CONST_1_CHANGED = 10000; |
| |
| private Provider mProvider; |
| |
| public ExampleImpl1() { |
| mProvider = new Provider(); |
| } |
| |
| public String getClassName() { |
| return "ExampleInt1"; |
| } |
| |
| public int getProviderConstant() { |
| return mProvider.nextProviderConst(); |
| } |
| } |
| EOF |
| } |
| |
| modify_provider_file() { |
| mkdir -p soong-test/java/integration/provider |
| |
| cat > soong-test/java/integration/provider/Provider.java <<'EOF' |
| package soong.java.integration.provider; |
| |
| public class Provider { |
| public static final int PROVIDER_CONST = -1; |
| |
| private String className; |
| |
| public Provider() { |
| this.className = "Provider"; |
| } |
| |
| public int previousProviderConst() { |
| return PROVIDER_CONST - 1; |
| } |
| |
| private String getProviderName() { |
| return className; |
| } |
| } |
| EOF |
| } |
| |
| function modify_annotation_api { |
| mkdir -p soong-test/java/integration/impllib |
| |
| cat > soong-test/java/integration/impllib/AnnotationUsage.java <<'EOF' |
| package soong.java.integration.impllib; |
| import soong.java.integration.annotation.GenerateWrapper; |
| public class AnnotationUsage { |
| @GenerateWrapper(methodCount = 1) |
| public String getClassName() { |
| return "AnnotationUsage"; |
| } |
| } |
| EOF |
| } |
| |
| function remove_base_dir { |
| rm -rf soong-test |
| } |
| |
| # shellcheck disable=SC2120 |
| function partial_compile_setup { |
| remove_base_dir |
| create_base_files |
| set_partial_compile_flags |
| |
| local target_to_remove="${1:-}" |
| if [ -n "$target_to_remove" ]; then # Check if it's non-empty after default |
| echo "Removing target: $target_to_remove" |
| rm -rf "$target_to_remove" |
| fi |
| |
| local all_files="${2:-}" |
| local all_files_create=false |
| if [[ "$all_files" == "true" ]]; then |
| all_files_create=true |
| fi |
| if [[ "$all_files_create" == true ]]; then |
| echo "Adding All Files:" |
| create_example_impl3_file |
| fi |
| } |
| |
| function partial_compile_setup_with_ap { |
| remove_base_dir |
| create_base_files |
| create_ap_files |
| set_partial_compile_flags |
| } |
| |
| # shellcheck disable=SC2120 |
| function full_compile_setup { |
| remove_base_dir |
| create_base_files |
| unset_partial_compile_flags |
| |
| local target_to_remove="${1:-}" |
| if [ -n "$target_to_remove" ]; then # Check if it's non-empty after default |
| echo "Removing target: $target_to_remove" |
| rm -rf "$target_to_remove" |
| fi |
| |
| local all_files="${2:-}" |
| local all_files_create=false |
| if [[ "$all_files" == "true" ]]; then |
| all_files_create=true |
| fi |
| if [[ "$all_files_create" == true ]]; then |
| echo "Adding All Files:" |
| create_example_impl3_file |
| fi |
| } |
| |
| function set_partial_compile_flags { |
| export SOONG_PARTIAL_COMPILE=true |
| export SOONG_USE_PARTIAL_COMPILE=true |
| } |
| |
| function unset_partial_compile_flags { |
| export SOONG_PARTIAL_COMPILE=false |
| } |
