/*
 * Copyright 2012-2021 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.boot.build.toolchain;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import org.gradle.api.Plugin;
import org.gradle.api.Project;
import org.gradle.api.plugins.JavaPluginExtension;
import org.gradle.api.tasks.GradleBuild;
import org.gradle.api.tasks.compile.JavaCompile;
import org.gradle.api.tasks.javadoc.Javadoc;
import org.gradle.api.tasks.testing.Test;
import org.gradle.jvm.toolchain.JavaLanguageVersion;
import org.gradle.jvm.toolchain.JavaToolchainSpec;

/**
 * {@link Plugin} for customizing Gradle's toolchain support.
 *
 * @author Christoph Dreis
 */
public class ToolchainPlugin implements Plugin<Project> {

	@Override
	public void apply(Project project) {
		configureToolchain(project);
	}

	private void configureToolchain(Project project) {
		ToolchainExtension toolchain = project.getExtensions().create("toolchain", ToolchainExtension.class, project);
		JavaLanguageVersion toolchainVersion = toolchain.getJavaVersion();
		if (toolchainVersion != null) {
			project.afterEvaluate((evaluated) -> configure(evaluated, toolchain));
		}
	}

	private void configure(Project project, ToolchainExtension toolchain) {
		if (!isJavaVersionSupported(toolchain, toolchain.getJavaVersion())) {
			disableToolchainTasks(project);
		}
		else {
			JavaToolchainSpec toolchainSpec = project.getExtensions().getByType(JavaPluginExtension.class)
					.getToolchain();
			toolchainSpec.getLanguageVersion().set(toolchain.getJavaVersion());
			configureJavaCompileToolchain(project, toolchain);
			configureTestToolchain(project, toolchain);
		}
	}

	private boolean isJavaVersionSupported(ToolchainExtension toolchain, JavaLanguageVersion toolchainVersion) {
		return toolchain.getMaximumCompatibleJavaVersion().map((version) -> version.canCompileOrRun(toolchainVersion))
				.getOrElse(true);
	}

	private void disableToolchainTasks(Project project) {
		project.getTasks().withType(JavaCompile.class, (task) -> task.setEnabled(false));
		project.getTasks().withType(Javadoc.class, (task) -> task.setEnabled(false));
		project.getTasks().withType(Test.class, (task) -> task.setEnabled(false));
		project.getTasks().withType(GradleBuild.class, (task) -> task.setEnabled(false));
	}

	private void configureJavaCompileToolchain(Project project, ToolchainExtension toolchain) {
		project.getTasks().withType(JavaCompile.class, (compile) -> {
			compile.getOptions().setFork(true);
			// See https://github.com/gradle/gradle/issues/15538
			List<String> forkArgs = Arrays.asList("--add-opens", "jdk.compiler/com.sun.tools.javac.code=ALL-UNNAMED");
			compile.getOptions().getForkOptions().getJvmArgs().addAll(forkArgs);
		});
	}

	private void configureTestToolchain(Project project, ToolchainExtension toolchain) {
		project.getTasks().withType(Test.class, (test) -> {
			List<String> arguments = Collections.singletonList("--illegal-access=warn");
			test.jvmArgs(arguments);
		});
	}

}

@@ -671,33 +671,8 @@
 n;%0A%0A
-import java.util.Arrays;%0A
 impo
@@ -2096,62 +2096,8 @@
 ));%0A
-%09%09%09configureJavaCompileToolchain(project, toolchain);%0A
 %09%09%09c
@@ -2125,27 +2125,16 @@
 (project
-, toolchain
 );%0A%09%09%7D%0A%09
@@ -2796,511 +2796,37 @@
 gure
-JavaCompileToolchain(Project project, ToolchainExtension toolchain) %7B%0A%09%09project.getTasks().withType(JavaCompile.class, (compile) -%3E %7B%0A%09%09%09compile.getOptions().setFork(true);%0A%09%09%09// See https://github.com/gradle/gradle/issues/15538%0A%09%09%09List%3CString%3E forkArgs = Arrays.asList(%22--add-opens%22, %22jdk.compiler/com.sun.tools.javac.code=ALL-UNNAMED%22);%0A%09%09%09compile.getOptions().getForkOptions().getJvmArgs().addAll(forkArgs);%0A%09%09%7D);%0A%09%7D%0A%0A%09private void configureTestToolchain(Project project, ToolchainExtension toolchain
+TestToolchain(Project project
 ) %7B%0A

Remove add-opens workaround in ToolchainPlugin