package org.jboss.windup.rules.java; import java.nio.file.Files; import java.nio.file.Path; import java.nio.file.Paths; import java.util.Collections; import javax.inject.Inject; import javax.inject.Singleton; import org.apache.commons.io.FileUtils; import org.apache.commons.lang3.RandomStringUtils; import org.jboss.arquillian.container.test.api.Deployment; import org.jboss.arquillian.junit.Arquillian; import org.jboss.forge.arquillian.AddonDependencies; import org.jboss.forge.arquillian.AddonDependency; import org.jboss.forge.arquillian.archive.AddonArchive; import org.jboss.shrinkwrap.api.ShrinkWrap; import org.jboss.windup.ast.java.data.TypeReferenceLocation; import org.jboss.windup.config.AbstractRuleProvider; import org.jboss.windup.config.GraphRewrite; import org.jboss.windup.config.loader.RuleLoaderContext; import org.jboss.windup.config.metadata.MetadataBuilder; import org.jboss.windup.config.operation.iteration.AbstractIterationOperation; import org.jboss.windup.engine.predicates.RuleProviderWithDependenciesPredicate; import org.jboss.windup.exec.WindupProcessor; import org.jboss.windup.exec.configuration.WindupConfiguration; import org.jboss.windup.graph.GraphContext; import org.jboss.windup.graph.GraphContextFactory; import org.jboss.windup.rules.apps.java.condition.JavaClass; import org.jboss.windup.rules.apps.java.condition.annotation.AnnotationListCondition; import org.jboss.windup.rules.apps.java.condition.annotation.AnnotationLiteralCondition; import org.jboss.windup.rules.apps.java.condition.annotation.AnnotationTypeCondition; import org.jboss.windup.rules.apps.java.config.ScanPackagesOption; import org.jboss.windup.rules.apps.java.config.SourceModeOption; import org.jboss.windup.rules.apps.java.scan.ast.JavaTypeReferenceModel; import org.jboss.windup.rules.apps.java.scan.provider.AnalyzeJavaFilesRuleProvider; import org.junit.Assert; import org.junit.Test; import org.junit.runner.RunWith; import org.ocpsoft.rewrite.config.Configuration; import org.ocpsoft.rewrite.config.ConfigurationBuilder; import org.ocpsoft.rewrite.context.EvaluationContext; @RunWith(Arquillian.class) public class JavaClassAnnotationFilteringTest { @Deployment @AddonDependencies({ @AddonDependency(name = "org.jboss.windup.config:windup-config"), @AddonDependency(name = "org.jboss.windup.exec:windup-exec"), @AddonDependency(name = "org.jboss.windup.rules.apps:windup-rules-java"), @AddonDependency(name = "org.jboss.windup.reporting:windup-reporting"), @AddonDependency(name = "org.jboss.windup.utils:windup-utils"), @AddonDependency(name = "org.jboss.forge.furnace.container:cdi") }) public static AddonArchive getDeployment() { return ShrinkWrap.create(AddonArchive.class).addBeansXML(); } @Inject BasicAnnotationScanProvider basicProvider; @Inject ComplexAnnotationScanProvider complexProvider; @Inject private WindupProcessor processor; @Inject private GraphContextFactory factory; @Test public void testBasicAnnotationFiltering() throws Exception { Path outputPath = getDefaultPath(); try (GraphContext context = factory.create(outputPath)) { final String inputDir = "src/test/resources/org/jboss/windup/rules/annotationtests/basic"; FileUtils.deleteDirectory(outputPath.toFile()); Files.createDirectories(outputPath); final WindupConfiguration processorConfig = new WindupConfiguration(); processorConfig.setRuleProviderFilter(new RuleProviderWithDependenciesPredicate( BasicAnnotationScanProvider.class)); processorConfig.setGraphContext(context); processorConfig.addInputPath(Paths.get(inputDir)); processorConfig.setOutputDirectory(outputPath); processorConfig.setOptionValue(ScanPackagesOption.NAME, Collections.singletonList("")); processorConfig.setOptionValue(SourceModeOption.NAME, true); processor.execute(processorConfig); Assert.assertEquals(1, basicProvider.baseRuleHitCount); Assert.assertEquals(1, basicProvider.withValueFilterHitCount); Assert.assertEquals(0, basicProvider.withIncorrectFilterCount); } } @Test public void testComplexAnnotationFiltering() throws Exception { Path outputPath = getDefaultPath(); try (GraphContext context = factory.create(outputPath)) { final String inputDir = "src/test/resources/org/jboss/windup/rules/annotationtests/complex"; FileUtils.deleteDirectory(outputPath.toFile()); Files.createDirectories(outputPath); final WindupConfiguration processorConfig = new WindupConfiguration(); processorConfig.setRuleProviderFilter(new RuleProviderWithDependenciesPredicate( ComplexAnnotationScanProvider.class)); processorConfig.setGraphContext(context); processorConfig.addInputPath(Paths.get(inputDir)); processorConfig.setOutputDirectory(outputPath); processorConfig.setOptionValue(ScanPackagesOption.NAME, Collections.singletonList("")); processorConfig.setOptionValue(SourceModeOption.NAME, true); processor.execute(processorConfig); Assert.assertEquals(1, complexProvider.baseRuleHitCount); } } private Path getDefaultPath() { return FileUtils.getTempDirectory().toPath().resolve("Windup") .resolve("windupgraph_javaclassannotationfilteringtest_" + RandomStringUtils.randomAlphanumeric(6)); } @Singleton public static class BasicAnnotationScanProvider extends AbstractRuleProvider { private int baseRuleHitCount = 0; private int withValueFilterHitCount = 0; private int withIncorrectFilterCount = 0; public BasicAnnotationScanProvider() { super(MetadataBuilder.forProvider(BasicAnnotationScanProvider.class) .addExecuteAfter(AnalyzeJavaFilesRuleProvider.class)); } // @formatter:off @Override public Configuration getConfiguration(RuleLoaderContext ruleLoaderContext) { return ConfigurationBuilder.begin() .addRule().when( JavaClass.references("org.jboss.windup.rules.annotationtests.basic.SimpleTestAnnotation") .at(TypeReferenceLocation.ANNOTATION) ).perform(new AbstractIterationOperation<JavaTypeReferenceModel>() { @Override public void perform(GraphRewrite event, EvaluationContext context, JavaTypeReferenceModel payload) { baseRuleHitCount++; } }) .addRule().when( JavaClass.references("org.jboss.windup.rules.annotationtests.basic.SimpleTestAnnotation") .at(TypeReferenceLocation.ANNOTATION).annotationMatches("value2", new AnnotationLiteralCondition("value 2")) ).perform(new AbstractIterationOperation<JavaTypeReferenceModel>() { @Override public void perform(GraphRewrite event, EvaluationContext context, JavaTypeReferenceModel payload) { withValueFilterHitCount++; } }) .addRule().when( JavaClass.references("org.jboss.windup.rules.annotationtests.basic.SimpleTestAnnotation") .at(TypeReferenceLocation.ANNOTATION).annotationMatches("value2", new AnnotationLiteralCondition("wrongvalue")) ).perform(new AbstractIterationOperation<JavaTypeReferenceModel>() { @Override public void perform(GraphRewrite event, EvaluationContext context, JavaTypeReferenceModel payload) { withIncorrectFilterCount++; } }); } // @formatter:on } @Singleton public static class ComplexAnnotationScanProvider extends AbstractRuleProvider { private int baseRuleHitCount = 0; public ComplexAnnotationScanProvider() { super(MetadataBuilder.forProvider(ComplexAnnotationScanProvider.class).addExecuteAfter(AnalyzeJavaFilesRuleProvider.class)); } // @formatter:off @Override public Configuration getConfiguration(RuleLoaderContext ruleLoaderContext) { return ConfigurationBuilder.begin() .addRule().when( JavaClass.references("org.jboss.windup.rules.annotationtests.complex.ExampleComplexAnnotation") .at(TypeReferenceLocation.ANNOTATION) .annotationMatches( "nestedAnnotationArray", new AnnotationListCondition(0).addCondition(new AnnotationTypeCondition("org.jboss.windup.rules.annotationtests.complex.ExampleNestedAnnotation")) ) ).perform(new AbstractIterationOperation<JavaTypeReferenceModel>() { @Override public void perform(GraphRewrite event, EvaluationContext context, JavaTypeReferenceModel payload) { baseRuleHitCount++; } }); } // @formatter:on } }