com.google.protobuf.gradle.GenerateProtoTask.groovy Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of protobuf-gradle-plugin Show documentation
Show all versions of protobuf-gradle-plugin Show documentation
Gradle build plugin to handle Protocol Buffers automated code generation and compilation
/*
* Original work copyright (c) 2015, Alex Antonov. All rights reserved.
* Modified work copyright (c) 2015, Google Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
*
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* 3. Neither the name of the copyright holder nor the names of its contributors
* may be used to endorse or promote products derived from this software without
* specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.google.protobuf.gradle
import com.google.common.base.Preconditions
import com.google.common.collect.ImmutableList
import org.gradle.api.DefaultTask
import org.gradle.api.GradleException
import org.gradle.api.Named
import org.gradle.api.NamedDomainObjectContainer
import org.gradle.api.logging.LogLevel
import org.gradle.api.tasks.SourceSet
import org.gradle.api.tasks.TaskAction
import org.gradle.util.ConfigureUtil
/**
* The task that compiles proto files into Java files.
*/
// TODO(zhangkun83): add per-plugin output dir reconfiguraiton.
public class GenerateProtoTask extends DefaultTask {
private final List includeDirs = new ArrayList()
private final NamedDomainObjectContainer builtins
private final NamedDomainObjectContainer plugins
// These fields are set by the Protobuf plugin only when initializing the
// task. Ideally they should be final fields, but Gradle task cannot have
// constructor arguments. We use the initializing flag to prevent users from
// accidentally modifying them.
private String outputBaseDir
// Tags for selectors inside protobuf.generateProtoTasks
private SourceSet sourceSet
private Object variant
private ImmutableList flavors
private String buildType
private boolean isTestVariant
/**
* If true, will set the protoc flag
* --descriptor_set_out="${outputBaseDir}/descriptor_set.desc"
*
* Default: false
*/
public boolean generateDescriptorSet
/**
* Configuration object for descriptor generation details.
*/
public class DescriptorSetOptions {
/**
* If set, specifies an alternative location than the default for storing the descriptor
* set.
*
* Default: null
*/
public GString path
/**
* If true, source information (comments, locations) will be included in the descriptor set.
*
* Default: false
*/
public boolean includeSourceInfo
/**
* If true, imports are included in the descriptor set, such that it is self-containing.
*
* Default: false
*/
public boolean includeImports
}
public final DescriptorSetOptions descriptorSetOptions = new DescriptorSetOptions();
private static enum State {
INIT, CONFIG, FINALIZED
}
private State state = State.INIT
private void checkInitializing() {
Preconditions.checkState(state == State.INIT, 'Should not be called after initilization has finished')
}
private void checkCanConfig() {
Preconditions.checkState(state == State.CONFIG || state == State.INIT,
'Should not be called after configuration has finished')
}
void setOutputBaseDir(String outputBaseDir) {
checkInitializing()
Preconditions.checkState(this.outputBaseDir == null, 'outputBaseDir is already set')
this.outputBaseDir = outputBaseDir
outputs.dir outputBaseDir
}
void setSourceSet(SourceSet sourceSet) {
checkInitializing()
Preconditions.checkState(!Utils.isAndroidProject(project),
'sourceSet should not be set in an Android project')
this.sourceSet = sourceSet
}
void setVariant(Object variant, boolean isTestVariant) {
checkInitializing()
Preconditions.checkState(Utils.isAndroidProject(project),
'variant should not be set in a Java project')
this.variant = variant
this.isTestVariant = isTestVariant
}
void setFlavors(ImmutableList flavors) {
checkInitializing()
Preconditions.checkState(Utils.isAndroidProject(project),
'flavors should not be set in a Java project')
this.flavors = flavors
}
void setBuildType(String buildType) {
checkInitializing()
Preconditions.checkState(Utils.isAndroidProject(project),
'buildType should not be set in a Java project')
this.buildType = buildType
}
SourceSet getSourceSet() {
Preconditions.checkState(!Utils.isAndroidProject(project),
'sourceSet should not be used in an Android project')
Preconditions.checkNotNull(sourceSet, 'sourceSet is not set')
return sourceSet
}
Object getVariant() {
Preconditions.checkState(Utils.isAndroidProject(project),
'variant should not be used in a Java project')
Preconditions.checkNotNull(variant, 'variant is not set')
return variant
}
boolean getIsTestVariant() {
Preconditions.checkState(Utils.isAndroidProject(project),
'isTestVariant should not be used in a Java project')
Preconditions.checkNotNull(variant, 'variant is not set')
return isTestVariant
}
ImmutableList getFlavors() {
Preconditions.checkState(Utils.isAndroidProject(project),
'flavors should not be used in a Java project')
Preconditions.checkNotNull(flavors, 'flavors is not set')
return flavors
}
String getBuildType() {
Preconditions.checkState(Utils.isAndroidProject(project),
'buildType should not be used in a Java project')
Preconditions.checkNotNull(buildType, 'buildType is not set')
return buildType
}
void doneInitializing() {
Preconditions.checkState(state == State.INIT, "Invalid state: ${state}")
state = State.CONFIG
}
void doneConfig() {
Preconditions.checkState(state == State.CONFIG, "Invalid state: ${state}")
state = State.FINALIZED
}
String getDescriptorPath() {
if (!generateDescriptorSet) {
throw new IllegalStateException(
"requested descriptor path but descriptor generation is off")
}
return descriptorSetOptions.path != null
? descriptorSetOptions.path : "${outputBaseDir}/descriptor_set.desc"
}
public GenerateProtoTask() {
builtins = project.container(PluginOptions)
plugins = project.container(PluginOptions)
}
//===========================================================================
// Configuration methods
//===========================================================================
/**
* Configures the protoc builtins in a closure, which will be maniuplating a
* NamedDomainObjectContainer.
*/
public void builtins(Closure configureClosure) {
checkCanConfig()
ConfigureUtil.configure(configureClosure, builtins)
}
/**
* Returns the container of protoc builtins.
*/
public NamedDomainObjectContainer getBuiltins() {
checkCanConfig()
return builtins
}
/**
* Configures the protoc plugins in a closure, which will be maniuplating a
* NamedDomainObjectContainer.
*/
public void plugins(Closure configureClosure) {
checkCanConfig()
ConfigureUtil.configure(configureClosure, plugins)
}
/**
* Returns the container of protoc plugins.
*/
public NamedDomainObjectContainer getPlugins() {
checkCanConfig()
return plugins
}
/**
* Add a directory to protoc's include path.
*/
public void include(Object dir) {
checkCanConfig()
if (dir instanceof File) {
includeDirs.add(dir)
} else {
includeDirs.add(project.file(dir))
}
}
/**
* The container of command-line options for a protoc plugin or a built-in output.
*/
public static class PluginOptions implements Named {
private final ArrayList options = new ArrayList()
private final String name
private String outputSubDir
public PluginOptions(String name) {
this.name = name
}
/**
* Adds a plugin option.
*/
public PluginOptions option(String option) {
options.add(option)
return this
}
public List getOptions() {
return options
}
/**
* Returns the name of the plugin or builtin.
*/
@Override
public String getName() {
return name
}
/**
* Set the output directory for this plugin, relative to {@link GenerateProtoTask#outputBaseDir}.
*/
public setOutputSubDir(String outputSubDir) {
this.outputSubDir = outputSubDir
}
/**
* Returns the relative outputDir for this plugin. If outputDir is not specified, name is used.
*/
public String getOutputSubDir() {
if (outputSubDir != null) {
return outputSubDir;
}
return name;
}
}
//===========================================================================
// protoc invocation logic
//===========================================================================
// protoc allows you to prefix comma-delimited options to the path in
// the --*_out flags, e.g.,
// - Without options: --java_out=/path/to/output
// - With options: --java_out=option1,option2:/path/to/output
// This method generates the prefix out of the given options.
static String makeOptionsPrefix(List options) {
StringBuilder prefix = new StringBuilder()
if (!options.isEmpty()) {
options.each { option ->
if (prefix.length() > 0) {
prefix.append(',')
}
prefix.append(option)
}
prefix.append(':')
}
return prefix.toString()
}
String getOutputDir(PluginOptions plugin) {
return "${outputBaseDir}/${plugin.outputSubDir}"
}
Collection getAllOutputDirs() {
ImmutableList.Builder dirs = ImmutableList.builder()
builtins.each { builtin ->
dirs.add(new File(getOutputDir(builtin)))
}
plugins.each { plugin ->
dirs.add(new File(getOutputDir(plugin)))
}
return dirs.build()
}
@TaskAction
def compile() {
Preconditions.checkState(state == State.FINALIZED, 'doneConfig() has not been called')
ToolsLocator tools = project.protobuf.tools
Set protoFiles = inputs.sourceFiles.files
[builtins, plugins]*.each { plugin ->
File outputDir = new File(getOutputDir(plugin))
outputDir.mkdirs()
}
def dirs = includeDirs*.path.collect {"-I${it}"}
logger.debug "ProtobufCompile using directories ${dirs}"
logger.debug "ProtobufCompile using files ${protoFiles}"
def cmd = [ tools.protoc.path ]
cmd.addAll(dirs)
// Handle code generation built-ins
builtins.each { builtin ->
String outPrefix = makeOptionsPrefix(builtin.options)
cmd += "--${builtin.name}_out=${outPrefix}${getOutputDir(builtin)}"
}
// Handle code generation plugins
plugins.each { plugin ->
String name = plugin.name
ExecutableLocator locator = tools.plugins.findByName(name)
if (locator == null) {
throw new GradleException("Codegen plugin ${name} not defined")
}
String pluginOutPrefix = makeOptionsPrefix(plugin.options)
cmd += "--${name}_out=${pluginOutPrefix}${getOutputDir(plugin)}"
cmd += "--plugin=protoc-gen-${name}=${locator.path}"
}
if (generateDescriptorSet) {
def path = getDescriptorPath()
// Ensure that the folder for the descriptor exists;
// the user may have set it to point outside an existing tree
def folder = new File(path).parentFile
if (!folder.exists()) {
folder.mkdirs()
}
cmd += "--descriptor_set_out=${path}"
if (descriptorSetOptions.includeImports) {
cmd += "--include_imports"
}
if (descriptorSetOptions.includeSourceInfo) {
cmd += "--include_source_info"
}
}
cmd.addAll protoFiles
logger.log(LogLevel.INFO, cmd.toString())
def stdout = new StringBuffer()
def stderr = new StringBuffer()
Process result = cmd.execute()
result.waitForProcessOutput(stdout, stderr)
def output = "protoc: stdout: ${stdout}. stderr: ${stderr}"
if (result.exitValue() == 0) {
logger.log(LogLevel.INFO, output)
} else {
throw new GradleException(output)
}
}
}