如何在gradle插件中添加新的语言源目录？

Question

我想为任意语言提供以下工作gradle脚本foo:

sourceSets {
    main {
        java {
            srcDir "${project.buildDir}/generated-sources/gen-java"
        }
        foo {
            srcDir "${project.buildDir}/generated-sources/gen-foo"
        }
    }
    test {
        java {
            srcDir "${project.buildDir}/generated-sources/gen-test-java"
        }
        foo {
            srcDir "${project.buildDir}/generated-sources/gen-test-foo"
        }
    }
}

我也希望有标准的foo源目录src/main/foo, src/test/foo.

如何编写gradle插件来实现此类功能？有可能这样做吗？

我有一个"解决方案"的解决方法,我的需求发布在下面,但仍然想要了解添加新语言源目录的正确方法.

Answer 1

您的用例的源代码会很长，但这里有一些指导，希望可以帮助您继续下去。

看一下Scala 插件。它完全满足了您的需要（因为看起来您遵循了源代码的 java 约定），而且还做了更多的事情。如果您尝试编写任何插件，我建议您查看整个 gradle 源代码。

您想参观的具体地点是：

1. ScalaBasePlugin#configureSourceSetDefaults- 这是顶层配置发生的地方
2. DefaultScalaSourceSet- 实际的源集类

在您的情况下，您可能只想将所有scala字符串重命名为foo并删除所有不需要的配置（例如实际的编译）。

该插件说明了如何添加默认源码目录及方法：

public ScalaSourceSet scala(Closure configureClosure) {
    configure(configureClosure, getScala());
    return this;
}

负责添加生成的源。基本上它只需要默认的源目录工厂并使用它。您可以使用注入来添加所有默认的 gradle 工厂（请参阅 Scala 插件，简单地使用javax.inject.Inject作品）。

您还可以检查 groovy 插件。请注意，例如，它DefaultGroovySourceSet看起来像 scala 插件中的那个。

Answer 2

为了实现这样的功能，应该扩展现有的源集（或者至少main源test集）。看起来main源test集是在JavaPlugin. 通过对象可以进行扩展Convention。

public class FooPlugin implements Plugin<ProjectInternal> {
    @Override
    public void apply(ProjectInternal project) {
        project.getPluginManager().apply(JavaPlugin.class);
        FooExtension ext = project.getExtensions().create(
            "foo",
            FooExtension.class,
            project,
            project.getFileResolver()
        );
        SourceSetContainer cont = (SourceSetContainer) project.getProperties().get("sourceSets");
        cont.all((SourceSet ss) -> {
            String name = ss.getName();
            File sources = project.file("src/" + name + "/foo");
            FooSourceSet fss = ext.getSourceSetsContainer().maybeCreate(name);
            SourceDirectorySet sds = fss.getFoo();
            sds.srcDir(sources);
            Convention sourceSetConvention = (Convention) InvokerHelper.getProperty(ss, "convention");
            sourceSetConvention.getPlugins().put("foo", fss);
        });
        project.task("compileFoo");
    }
}

public class FooExtension {

    private final NamedDomainObjectContainer<FooSourceSet> sourceSetsContainer;

    public FooExtension(Project project, FileResolver fileResolver) {
        sourceSetsContainer = project.container(
            FooSourceSet.class,
            new FooSourceSetFactory(fileResolver)
        );
    }

    public NamedDomainObjectContainer<FooSourceSet> getSourceSetsContainer() {
        return sourceSetsContainer;
    }

    public void srcDir(String file) {
        sourceSetsContainer.getByName("main").getFoo().srcDir(file);
    }
}

public class FooSourceSetFactory implements NamedDomainObjectFactory<FooSourceSet> {

    private final FileResolver fileResolver;

    public FooSourceSetFactory(FileResolver fileResolver) {
        this.fileResolver = fileResolver;
    }

    @Override
    public FooSourceSet create(String name) {
        return new DefaultFooSourceSet(name, fileResolver);
    }
}

public interface FooSourceSet {
    public String getName();
    public SourceDirectorySet getFoo();
    public FooSourceSet foo(Closure clsr);
}

public class DefaultFooSourceSet implements FooSourceSet {

    final String name;
    final SourceDirectorySet foo;

    public DefaultFooSourceSet(String displayName, FileResolver fileResolver) {
        this.name = displayName;
        DefaultDirectoryFileTreeFactory ddftf = new DefaultDirectoryFileTreeFactory();
        foo = new DefaultSourceDirectorySet(name, fileResolver, ddftf);
    }

    @Override
    public String getName() {
        return name;
    }

    @Override
    public SourceDirectorySet getFoo() {
        return foo;
    }

    @Override
    public FooSourceSet foo(Closure clsr) {
        ConfigureUtil.configure(clsr, foo);
        return this;
    }
}

public class CompileFooTask extends DefaultTask {

    @TaskAction
    public void compileFoo() {
        SourceSetContainer cont = (SourceSetContainer) getProject().getProperties().get("sourceSets");
        cont.all((SourceSet ss) -> {
            FooSourceSet fss = getProject()
                .getExtensions()
                .getByType(FooExtension.class)
                .getSourceSetsContainer()
                .maybeCreate(ss.getName());
            System.out.println("directories under " + ss.getName()
                + ": " + fss.getFoo().getSrcDirs());
        });
    }
}

任务compileFoo证明插件确实有效。给出问题中的构建脚本片段，它会打印如下行：

directories under main: [<root>/src/main/foo, <root>/build/generated-sources/gen-foo]
directories under test: [<root>/src/test/foo, <root>/build/generated-sources/gen-test-foo]