name.remal.gradle_plugins.lombok.AbstractLombokTask Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lombok Show documentation
lombok: lombok
There is a newer version: 2.2.14
package name.remal.gradle_plugins.lombok;

import static java.util.Collections.emptyList;
import static name.remal.gradle_plugins.lombok.JavacPackagesToOpenUtils.getJavacPackageOpenJvmArgs;
import static name.remal.gradle_plugins.lombok.JavacPackagesToOpenUtils.shouldJavacPackageOpenJvmArgsBeAdded;
import static name.remal.gradle_plugins.toolkit.JavaLauncherUtils.getJavaLauncherProviderFor;

import com.google.errorprone.annotations.ForOverride;
import java.util.List;
import java.util.Optional;
import java.util.concurrent.atomic.AtomicReference;
import javax.inject.Inject;
import lombok.val;
import name.remal.gradle_plugins.toolkit.JavaInstallationMetadataUtils;
import org.gradle.api.DefaultTask;
import org.gradle.api.JavaVersion;
import org.gradle.api.file.ConfigurableFileCollection;
import org.gradle.api.file.ProjectLayout;
import org.gradle.api.model.ObjectFactory;
import org.gradle.api.provider.Property;
import org.gradle.api.tasks.Classpath;
import org.gradle.api.tasks.InputFiles;
import org.gradle.api.tasks.Nested;
import org.gradle.api.tasks.TaskAction;
import org.gradle.jvm.toolchain.JavaLauncher;
import org.gradle.process.ExecOperations;
import org.gradle.process.JavaExecSpec;

public abstract class AbstractLombokTask extends DefaultTask {

    private static final String MAIN_CLASS = "lombok.launch.Main";


    @ForOverride
    protected List createArgs() {
        return emptyList();
    }

    @ForOverride
    protected void beforeExecute(JavaExecSpec execSpec) {
    }

    @ForOverride
    protected void afterExecute(JavaExecSpec execSpec) {
    }


    @Classpath
    @InputFiles
    public abstract ConfigurableFileCollection getToolClasspath();

    @org.gradle.api.tasks.Optional
    @Nested
    public abstract Property getJavaLauncher();


    private final String command;

    protected AbstractLombokTask(String command) {
        this.command = command;
        getJavaLauncher().convention(getJavaLauncherProviderFor(getProject()));
    }


    @TaskAction
    public void execute() {
        AtomicReference execSpecRef = new AtomicReference<>();

        getExecOperations().javaexec(execSpec -> {
            val javaLauncher = getJavaLauncher().get();
            val javaVersion = Optional.of(javaLauncher.getMetadata())
                .map(JavaInstallationMetadataUtils::getJavaInstallationVersionOf)
                .orElseGet(JavaVersion::current);
            if (shouldJavacPackageOpenJvmArgsBeAdded(javaVersion)) {
                execSpec.jvmArgs(getJavacPackageOpenJvmArgs());
            }
            execSpec.setExecutable(javaLauncher.getExecutablePath().getAsFile());
            execSpec.setClasspath(getToolClasspath());
            execSpec.getMainClass().set(MAIN_CLASS);
            execSpec.args(command);
            execSpec.args(createArgs());
            execSpec.setWorkingDir(getTemporaryDir());
            beforeExecute(execSpec);
            execSpecRef.set(execSpec);
        });

        afterExecute(execSpecRef.get());

        setDidWork(true);
    }

    @Inject
    protected abstract ExecOperations getExecOperations();

    @Inject
    protected abstract ProjectLayout getProjectLayout();

    @Inject
    protected abstract ObjectFactory getObjectFactory();

}