-
Notifications
You must be signed in to change notification settings - Fork 30
/
AspectJPostCompileWeavingPlugin.java
117 lines (94 loc) · 5.82 KB
/
AspectJPostCompileWeavingPlugin.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
package io.freefair.gradle.plugins.aspectj;
import io.freefair.gradle.plugins.aspectj.internal.DefaultWeavingSourceSet;
import org.gradle.api.NonNullApi;
import org.gradle.api.Plugin;
import org.gradle.api.Project;
import org.gradle.api.Task;
import org.gradle.api.artifacts.Configuration;
import org.gradle.api.file.FileCollection;
import org.gradle.api.internal.plugins.DslObject;
import org.gradle.api.internal.tasks.compile.HasCompileOptions;
import org.gradle.api.plugins.GroovyPlugin;
import org.gradle.api.plugins.JavaBasePlugin;
import org.gradle.api.plugins.JavaPlugin;
import org.gradle.api.plugins.JavaPluginExtension;
import org.gradle.api.plugins.scala.ScalaPlugin;
import org.gradle.api.provider.Provider;
import org.gradle.api.tasks.SourceSet;
import org.gradle.api.tasks.SourceSetContainer;
import org.gradle.api.tasks.compile.AbstractCompile;
import org.gradle.jvm.toolchain.JavaLauncher;
import org.gradle.jvm.toolchain.JavaToolchainService;
import org.jetbrains.kotlin.gradle.tasks.KotlinJvmCompile;
@NonNullApi
public class AspectJPostCompileWeavingPlugin implements Plugin<Project> {
private Project project;
private AspectJBasePlugin aspectjBasePlugin;
private SourceSetContainer sourceSets;
private Provider<JavaLauncher> defaultLauncher;
@Override
public void apply(Project project) {
if (project.getPlugins().hasPlugin(AspectJPlugin.class)) {
throw new IllegalStateException("Another aspectj plugin (which is excludes this one) has already been applied to the project.");
}
project.getPlugins().withId("com.android.application", ignored -> {
throw new IllegalStateException("The 'io.freefair.aspectj.post-compile-weaving' plugin is not compatible with android projects");
});
project.getPlugins().withId("com.android.library", ignored -> {
throw new IllegalStateException("The 'io.freefair.aspectj.post-compile-weaving' plugin is not compatible with android projects");
});
this.project = project;
aspectjBasePlugin = project.getPlugins().apply(AspectJBasePlugin.class);
project.getPlugins().apply(JavaBasePlugin.class);
JavaPluginExtension javaPluginExtension = project.getExtensions().getByType(JavaPluginExtension.class);
sourceSets = javaPluginExtension.getSourceSets();
JavaToolchainService service = project.getExtensions().getByType(JavaToolchainService.class);
defaultLauncher = service.launcherFor(javaPluginExtension.getToolchain());
sourceSets.all(this::configureSourceSetDefaults);
project.getPlugins().withType(JavaPlugin.class, plugin -> this.configurePlugin("java"));
project.getPlugins().withType(GroovyPlugin.class, plugin -> this.configurePlugin("groovy"));
project.getPlugins().withType(ScalaPlugin.class, plugin -> this.configurePlugin("scala"));
project.getPlugins().withId("org.jetbrains.kotlin.jvm", plugin -> this.configurePlugin("kotlin"));
}
private void configureSourceSetDefaults(SourceSet sourceSet) {
DefaultWeavingSourceSet weavingSourceSet = new DefaultWeavingSourceSet(sourceSet, project.getObjects());
new DslObject(sourceSet).getConvention().add("aspectj", weavingSourceSet);
Configuration aspectpath = project.getConfigurations().create(WeavingSourceSet.getAspectConfigurationName(sourceSet));
WeavingSourceSet.getAspectPath(sourceSet).from(aspectpath);
Configuration inpath = project.getConfigurations().create(WeavingSourceSet.getInpathConfigurationName(sourceSet));
WeavingSourceSet.getInPath(sourceSet).from(inpath);
project.getConfigurations().getByName(sourceSet.getImplementationConfigurationName()).extendsFrom(aspectpath);
project.getConfigurations().getByName(sourceSet.getCompileOnlyConfigurationName()).extendsFrom(inpath);
}
private void configurePlugin(String language) {
sourceSets.all(sourceSet -> {
FileCollection aspectpath = WeavingSourceSet.getAspectPath(sourceSet);
FileCollection inpath = WeavingSourceSet.getInPath(sourceSet);
Configuration runtimeClasspath = project.getConfigurations().getByName(sourceSet.getRuntimeClasspathConfigurationName());
FileCollection aspectjClasspath = aspectjBasePlugin.getAspectjRuntime().inferAspectjClasspath(runtimeClasspath);
project.getTasks().named(sourceSet.getCompileTaskName(language), compileTask -> {
AjcAction ajcAction = enhanceWithWeavingAction(compileTask, aspectpath, inpath, aspectjClasspath);
if (compileTask instanceof HasCompileOptions) {
HasCompileOptions compileTaskWithOptions = (HasCompileOptions) compileTask;
ajcAction.getOptions().getBootclasspath().from(compileTaskWithOptions.getOptions().getBootstrapClasspath());
ajcAction.getOptions().getExtdirs().from(compileTaskWithOptions.getOptions().getExtensionDirs());
}
});
});
}
private AjcAction enhanceWithWeavingAction(Task compileTask, FileCollection aspectpath, FileCollection inpath, FileCollection aspectjConfiguration) {
AjcAction action = project.getObjects().newInstance(AjcAction.class);
action.getLauncher().convention(defaultLauncher);
action.getOptions().getAspectpath().from(aspectpath);
action.getOptions().getInpath().from(inpath);
if (compileTask instanceof AbstractCompile) {
action.getAdditionalInpath().from(((AbstractCompile) compileTask).getDestinationDirectory());
}
else if (compileTask instanceof KotlinJvmCompile) {
action.getAdditionalInpath().from(((KotlinJvmCompile) compileTask).getDestinationDirectory());
}
action.getClasspath().from(aspectjConfiguration);
action.addToTask(compileTask);
return action;
}
}