com.commercehub.gradle.plugin.avro.AvroBasePlugin Maven / Gradle / Ivy

Go to download
package com.commercehub.gradle.plugin.avro;

import org.gradle.api.Action;
import org.gradle.api.Plugin;
import org.gradle.api.Project;
import org.gradle.api.internal.ConventionMapping;
import org.gradle.api.internal.IConventionAware;

import java.util.concurrent.Callable;

import static com.commercehub.gradle.plugin.avro.Constants.*;

public class AvroBasePlugin implements Plugin {
    @Override
    public void apply(final Project project) {
        configureExtension(project);
    }

    private static void configureExtension(final Project project) {
        final AvroExtension avroExtension = project.getExtensions().create(AVRO_EXTENSION_NAME, DefaultAvroExtension.class);
        ConventionMapping extensionMapping = conventionMapping(avroExtension);
        extensionMapping.map(OPTION_STRING_TYPE, new Callable() {
            @Override
            public String call() throws Exception {
                return DEFAULT_STRING_TYPE;
            }
        });
        extensionMapping.map(OPTION_FIELD_VISIBILITY, new Callable() {
            @Override
            public String call() throws Exception {
                return DEFAULT_FIELD_VISIBILITY;
            }
        });
        extensionMapping.map(OPTION_CREATE_SETTERS, new Callable() {
            @Override
            public Boolean call() throws Exception {
                return DEFAULT_CREATE_SETTERS;
            }
        });
        project.getTasks().withType(GenerateAvroJavaTask.class).all(new Action() {
            @Override
            public void execute(GenerateAvroJavaTask task) {
                ConventionMapping taskMapping = conventionMapping(task);
                taskMapping.map(OPTION_OUTPUT_CHARACTER_ENCODING, new Callable() {
                    @Override
                    public String call() throws Exception {
                        return avroExtension.getOutputCharacterEncoding();
                    }
                });
                taskMapping.map(OPTION_STRING_TYPE, new Callable() {
                    @Override
                    public String call() throws Exception {
                        return avroExtension.getStringType();
                    }
                });
                taskMapping.map(OPTION_FIELD_VISIBILITY, new Callable() {
                    @Override
                    public String call() throws Exception {
                        return avroExtension.getFieldVisibility();
                    }
                });
                taskMapping.map(OPTION_TEMPLATE_DIRECTORY, new Callable() {
                    @Override
                    public String call() throws Exception {
                        return avroExtension.getTemplateDirectory();
                    }
                });
                taskMapping.map(OPTION_CREATE_SETTERS, new Callable() {
                    @Override
                    public Boolean call() throws Exception {
                        return avroExtension.isCreateSetters();
                    }
                });
            }
        });
    }

    private static ConventionMapping conventionMapping(Object conventionAware) {
        // TODO: try other alternatives to convention mapping
        // Convention mapping is an internal API.
        // Other options here:
        // http://forums.gradle.org/gradle/topics/how_can_i_do_convention_mappings_from_java_without_depending_on_an_internal_api
        return ((IConventionAware) conventionAware).getConventionMapping();
    }
}