aboutsummaryrefslogtreecommitdiff
path: root/src/main/groovy/frege/gradle/plugins/FregeBasePlugin.java
blob: f45dae4271ab4da53037eb4da8c4ceb8f793b3d0 (plain)
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
package frege.gradle.plugins;

import frege.gradle.DefaultFregeSourceSet;
import frege.gradle.FregeSourceSetDirectoryFactory;
import frege.gradle.tasks.FregeCompile;
import org.gradle.api.Action;
import org.gradle.api.Plugin;
import org.gradle.api.Project;
import org.gradle.api.file.FileTreeElement;
import org.gradle.api.internal.file.FileResolver;
import org.gradle.api.internal.plugins.DslObject;
import org.gradle.api.internal.project.ProjectInternal;
import org.gradle.api.internal.tasks.DefaultSourceSet;
import org.gradle.api.plugins.JavaBasePlugin;
import org.gradle.api.plugins.JavaPluginConvention;
import org.gradle.api.specs.Spec;
import org.gradle.api.tasks.SourceSet;

import javax.inject.Inject;
import java.util.concurrent.Callable;

public class FregeBasePlugin implements Plugin<Project> {
    private FileResolver fileResolver;

    private static String EXTENSION_NAME = "frege";
    private FregePluginExtension fregePluginExtension;
    private Project project;

    @Inject
    public FregeBasePlugin(FileResolver fileResolver) {
        this.fileResolver = fileResolver;
    }

    @Override
    public void apply(final Project project) {
        // Workaround to build proper jars on Windows, see https://github.com/Frege/frege-gradle-plugin/issues/9
        this.project = project;
        System.setProperty("file.encoding", "UTF-8");
        project.getPluginManager().apply(JavaBasePlugin.class);
        fregePluginExtension = project.getExtensions().create(EXTENSION_NAME, FregePluginExtension.class);
        JavaBasePlugin javaBasePlugin = project.getPlugins().getPlugin(JavaBasePlugin.class);
        configureSourceSetDefaults(javaBasePlugin);
    }


    private void configureSourceSetDefaults(final JavaBasePlugin javaBasePlugin) {
        project.getConvention().getPlugin(JavaPluginConvention.class).getSourceSets().all(new Action<SourceSet>() {
            public void execute(final SourceSet sourceSet) {
                FregeSourceSetDirectoryFactory factory = new FregeSourceSetDirectoryFactory((ProjectInternal) project, fileResolver);
                final DefaultFregeSourceSet fregeSourceSet = new DefaultFregeSourceSet(((DefaultSourceSet) sourceSet).getDisplayName(), factory);
                new DslObject(sourceSet).getConvention().getPlugins().put("frege", fregeSourceSet);

                final String defaultSourcePath = String.format("src/%s/frege", sourceSet.getName());
                fregeSourceSet.getFrege().srcDir(defaultSourcePath);
                sourceSet.getResources().getFilter().exclude(new Spec<FileTreeElement>() {
                    public boolean isSatisfiedBy(FileTreeElement element) {
                        return fregeSourceSet.getFrege().contains(element.getFile());
                    }
                });
                sourceSet.getAllJava().source(fregeSourceSet.getFrege());
                sourceSet.getAllSource().source(fregeSourceSet.getFrege());

                String compileTaskName = sourceSet.getCompileTaskName("frege");
                FregeCompile compile = project.getTasks().create(compileTaskName, FregeCompile.class);
                compile.setModule(project.file(defaultSourcePath).getAbsolutePath());
                javaBasePlugin.configureForSourceSet(sourceSet, compile);
                compile.getConventionMapping().map("fregepath", new Callable() {
                    public Object call() throws Exception {
                        return sourceSet.getCompileClasspath();
                    }
                });

                compile.dependsOn(sourceSet.getCompileJavaTaskName());
                compile.setDescription(String.format("Compiles the %s Frege source.", sourceSet.getName()));
                compile.setSource(fregeSourceSet.getFrege());
                project.getTasks().getByName(sourceSet.getClassesTaskName()).dependsOn(compileTaskName);
                sourceSet.compiledBy(compile);
            }
        });
    }
}