498 lines
No EOL
16 KiB
Java
498 lines
No EOL
16 KiB
Java
/*
|
|
* Copyright 2023 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 rife.bld.extension;
|
|
|
|
import org.jetbrains.kotlin.cli.jvm.K2JVMCompiler;
|
|
import rife.bld.BaseProject;
|
|
import rife.bld.operations.AbstractOperation;
|
|
import rife.tools.FileUtils;
|
|
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.util.*;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
import java.util.regex.Pattern;
|
|
|
|
/**
|
|
* Compiles main and test Kotlin sources in the relevant build directories.
|
|
*
|
|
* @author <a href="https://erik.thauvin.net/">Erik C. Thauvin</a>
|
|
* @since 1.0
|
|
*/
|
|
public class CompileKotlinOperation extends AbstractOperation<CompileKotlinOperation> {
|
|
public static final Pattern KOTLIN_FILE_PATTERN = Pattern.compile("^.*\\.kt$");
|
|
private static final Logger LOGGER = Logger.getLogger(CompileKotlinOperation.class.getName());
|
|
private final Collection<String> compileMainClasspath_ = new ArrayList<>();
|
|
private final Collection<String> compileTestClasspath_ = new ArrayList<>();
|
|
private final Collection<File> mainSourceDirectories_ = new ArrayList<>();
|
|
private final Collection<File> mainSourceFiles_ = new ArrayList<>();
|
|
private final Collection<File> testSourceDirectories_ = new ArrayList<>();
|
|
private final Collection<File> testSourceFiles_ = new ArrayList<>();
|
|
private File buildMainDirectory_;
|
|
private File buildTestDirectory_;
|
|
private CompileKotlinOptions compileOptions_;
|
|
private KaptOptions kaptOptions_;
|
|
private File kotlinLibsDirectory_;
|
|
|
|
/**
|
|
* Returns the list JARs contained in a given directory.
|
|
*
|
|
* @param directory the directory
|
|
* @param regex the regular expression to match
|
|
* @return the list of JARs
|
|
*/
|
|
public static List<String> getJarList(File directory, String regex) {
|
|
var jars = new ArrayList<String>();
|
|
|
|
if (directory.isDirectory()) {
|
|
var files = directory.listFiles();
|
|
if (files != null) {
|
|
for (var f : files) {
|
|
if (!f.getName().contains("-sources") && f.getName().matches(regex)) {
|
|
jars.add(f.getAbsolutePath());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return jars;
|
|
}
|
|
|
|
public static Collection<File> getKotlinFileList(File directory) {
|
|
if (directory == null) {
|
|
return Collections.emptyList();
|
|
} else if (!directory.exists()) {
|
|
if (LOGGER.isLoggable(Level.WARNING)) {
|
|
LOGGER.warning("Directory not found: " + directory.getAbsolutePath());
|
|
}
|
|
return Collections.emptyList();
|
|
} else {
|
|
var dir_abs = directory.getAbsoluteFile();
|
|
return FileUtils.getFileList(dir_abs, KOTLIN_FILE_PATTERN, null).stream().map((file) ->
|
|
new File(dir_abs, file)).toList();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Provides the main build destination directory.
|
|
*
|
|
* @param directory the directory to use for the main build destination
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation buildMainDirectory(File directory) {
|
|
buildMainDirectory_ = directory;
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the main build destination directory.
|
|
*
|
|
* @return the main build destination
|
|
*/
|
|
public File buildMainDirectory() {
|
|
return buildMainDirectory_;
|
|
}
|
|
|
|
/**
|
|
* Provides the test build destination directory.
|
|
*
|
|
* @param directory the directory to use for the test build destination
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation buildTestDirectory(File directory) {
|
|
buildTestDirectory_ = directory;
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the test build destination directory.
|
|
*
|
|
* @return the test build destination
|
|
*/
|
|
public File buildTestDirectory() {
|
|
return buildTestDirectory_;
|
|
}
|
|
|
|
/**
|
|
* Provides entries for the main compilation classpath.
|
|
*
|
|
* @param classpath one or more classpath entries
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation compileMainClasspath(String... classpath) {
|
|
compileMainClasspath_.addAll(Arrays.asList(classpath));
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Provides a list of entries for the main compilation classpath.
|
|
*
|
|
* @param classpath a list of classpath entries
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation compileMainClasspath(Collection<String> classpath) {
|
|
compileMainClasspath_.addAll(classpath);
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the list of entries for the main compilation classpath.
|
|
*
|
|
* @return the main compilation classpath list
|
|
*/
|
|
public Collection<String> compileMainClasspath() {
|
|
return compileMainClasspath_;
|
|
}
|
|
|
|
/**
|
|
* Provides a list of compilation options to pass to the {@code kotlinc} compiler.
|
|
*
|
|
* @param options the compiler options
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation compileOptions(CompileKotlinOptions options) {
|
|
compileOptions_ = options;
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Provides entries for the test compilation classpath.
|
|
*
|
|
* @param classpath one or more classpath entries
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation compileTestClasspath(String... classpath) {
|
|
compileTestClasspath_.addAll(Arrays.asList(classpath));
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Provides a list of entries for the test compilation classpath.
|
|
*
|
|
* @param classpath a list of classpath entries
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation compileTestClasspath(Collection<String> classpath) {
|
|
compileTestClasspath_.addAll(classpath);
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the list of entries for the test compilation classpath.
|
|
*
|
|
* @return the test compilation classpath list
|
|
*/
|
|
public Collection<String> compileTestClasspath() {
|
|
return compileTestClasspath_;
|
|
}
|
|
|
|
/**
|
|
* Performs the compile operation.
|
|
*/
|
|
@Override
|
|
@SuppressWarnings("PMD.SystemPrintln")
|
|
public void execute()
|
|
throws IOException {
|
|
executeCreateBuildDirectories();
|
|
executeBuildMainSources();
|
|
executeBuildTestSources();
|
|
if (!silent()) {
|
|
System.out.println("Kotlin compilation finished successfully.");
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Part of the {@link #execute} operation, builds the main sources.
|
|
*/
|
|
protected void executeBuildMainSources()
|
|
throws IOException {
|
|
executeBuildSources(
|
|
compileMainClasspath(),
|
|
sources(mainSourceFiles(), mainSourceDirectories()),
|
|
buildMainDirectory(),
|
|
null);
|
|
}
|
|
|
|
/**
|
|
* Part of the {@link #execute} operation, build sources to a destination.
|
|
*
|
|
* @param classpath the classpath list used for the compilation
|
|
* @param sources the source files to compile
|
|
* @param destination the destination directory
|
|
* @param friendPaths the output directory for friendly modules
|
|
*/
|
|
protected void executeBuildSources(Collection<String> classpath, Collection<File> sources, File destination, File friendPaths)
|
|
throws IOException {
|
|
if (sources.isEmpty() || destination == null) {
|
|
return;
|
|
}
|
|
|
|
var k2 = new K2JVMCompiler();
|
|
var args = new ArrayList<String>();
|
|
|
|
// classpath
|
|
args.add("-cp");
|
|
args.add(FileUtils.joinPaths(classpath.stream().toList()));
|
|
|
|
// destination
|
|
args.add("-d");
|
|
args.add(destination.getAbsolutePath());
|
|
|
|
// friend-path
|
|
if (friendPaths != null) {
|
|
args.add("-Xfriend-paths");
|
|
args.add(friendPaths.getAbsolutePath());
|
|
}
|
|
|
|
// options
|
|
if (compileOptions_ != null) {
|
|
args.addAll(compileOptions_.args());
|
|
}
|
|
|
|
// kapt plugin & options
|
|
if (kotlinLibsDirectory_ != null && kaptOptions_ != null) {
|
|
var kaptJar = getJarList(kotlinLibsDirectory_, "^.*kotlin-annotation-processing.*\\.jar$");
|
|
if (kaptJar.size() == 1) {
|
|
args.add("-Xplugin=" + kaptJar.get(0));
|
|
kaptOptions_.args().forEach(a -> {
|
|
args.add("-P");
|
|
args.add(a);
|
|
});
|
|
} else if (LOGGER.isLoggable(Level.WARNING)) {
|
|
LOGGER.warning("Could not locate the Kotlin annotation processing JAR in:" + kotlinLibsDirectory_);
|
|
}
|
|
}
|
|
|
|
// sources
|
|
sources.forEach(f -> args.add(f.getAbsolutePath()));
|
|
|
|
if (LOGGER.isLoggable(Level.FINE) && !silent()) {
|
|
LOGGER.fine("kotlinc " + String.join(" ", args));
|
|
}
|
|
|
|
var exitCode = k2.exec(System.err, args.toArray(new String[0]));
|
|
if (exitCode.getCode() != 0) {
|
|
throw new IOException("Kotlin compilation failed.");
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Part of the {@link #execute} operation, builds the test sources.
|
|
*/
|
|
protected void executeBuildTestSources()
|
|
throws IOException {
|
|
executeBuildSources(
|
|
compileTestClasspath(),
|
|
sources(testSourceFiles(), testSourceDirectories()),
|
|
buildTestDirectory(), buildMainDirectory());
|
|
}
|
|
|
|
/**
|
|
* Part of the {@link #execute} operation, creates the build directories.
|
|
*/
|
|
protected void executeCreateBuildDirectories() throws IOException {
|
|
if (buildMainDirectory() != null && !buildMainDirectory().exists() && !buildMainDirectory().mkdirs()) {
|
|
throw new IOException("Could not created build main directory: " + buildMainDirectory().getAbsolutePath());
|
|
}
|
|
if (buildTestDirectory() != null && !buildTestDirectory().exists() && !buildTestDirectory().mkdirs()) {
|
|
throw new IOException("Could not created build test directory: " + buildTestDirectory().getAbsolutePath());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Configures a compile operation from a {@link BaseProject}.
|
|
*
|
|
* @param project the project to configure the compile operation from
|
|
*/
|
|
public CompileKotlinOperation fromProject(BaseProject project) {
|
|
return buildMainDirectory(project.buildMainDirectory())
|
|
.buildTestDirectory(project.buildTestDirectory())
|
|
.compileMainClasspath(project.compileMainClasspath())
|
|
.compileTestClasspath(project.compileTestClasspath())
|
|
.kotlinLibsDirectory(project.libBldDirectory())
|
|
.mainSourceFiles(getKotlinFileList(new File(project.srcMainDirectory(), "kotlin")))
|
|
.testSourceFiles(getKotlinFileList(new File(project.srcTestDirectory(), "kotlin")));
|
|
}
|
|
|
|
/**
|
|
* Provides the annotation processor ({@code kapt} options.
|
|
*
|
|
* @param options the {@code kapt} options
|
|
* @return this class instance
|
|
*/
|
|
public CompileKotlinOperation kaptOptions(KaptOptions options) {
|
|
kaptOptions_ = options;
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Provides the directory containing the Kotlin libraries (compiler, plugins, etc.) JARs.
|
|
*
|
|
* @param directory the directory location
|
|
* @return this class instance
|
|
*/
|
|
public CompileKotlinOperation kotlinLibsDirectory(File directory) {
|
|
kotlinLibsDirectory_ = directory;
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Provides the directory containing the Kotlin libraries (compiler, plugins, etc.) JARs.
|
|
*
|
|
* @param directory the directory location
|
|
* @return this class instance
|
|
*/
|
|
public CompileKotlinOperation kotlinLibsDirectory(String directory) {
|
|
kotlinLibsDirectory_ = new File(directory);
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Provides main source directories that should be compiled.
|
|
*
|
|
* @param directories one or more main source directories
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation mainSourceDirectories(File... directories) {
|
|
mainSourceDirectories_.addAll(List.of(directories));
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Provides a list of main source directories that should be compiled.
|
|
*
|
|
* @param directories a list of main source directories
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation mainSourceDirectories(Collection<File> directories) {
|
|
mainSourceDirectories_.addAll(directories);
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the list of main source directories that should be compiled.
|
|
*
|
|
* @return the list of main source directories to compile
|
|
*/
|
|
public Collection<File> mainSourceDirectories() {
|
|
return mainSourceDirectories_;
|
|
}
|
|
|
|
/**
|
|
* Provides main files that should be compiled.
|
|
*
|
|
* @param files one or more main files
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation mainSourceFiles(File... files) {
|
|
mainSourceFiles_.addAll(Arrays.asList(files));
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Provides a list of main files that should be compiled.
|
|
*
|
|
* @param files a list of main files
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation mainSourceFiles(Collection<File> files) {
|
|
mainSourceFiles_.addAll(files);
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the list of main files that should be compiled.
|
|
*
|
|
* @return the list of main files to compile
|
|
*/
|
|
public Collection<File> mainSourceFiles() {
|
|
return mainSourceFiles_;
|
|
}
|
|
|
|
// Combine Kotlin sources
|
|
private Collection<File> sources(Collection<File> files, Collection<File> directories) {
|
|
var sources = new ArrayList<>(files);
|
|
for (var directory : directories) {
|
|
sources.addAll(getKotlinFileList(directory));
|
|
}
|
|
|
|
return sources;
|
|
}
|
|
|
|
/**
|
|
* Provides test source directories that should be compiled.
|
|
*
|
|
* @param directories one or more test source directories
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation testSourceDirectories(File... directories) {
|
|
testSourceDirectories_.addAll(List.of(directories));
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Provides a list of test source directories that should be compiled.
|
|
*
|
|
* @param directories a list of test source directories
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation testSourceDirectories(Collection<File> directories) {
|
|
testSourceDirectories_.addAll(directories);
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the list of test source directories that should be compiled.
|
|
*
|
|
* @return the list of test source directories to compile
|
|
*/
|
|
public Collection<File> testSourceDirectories() {
|
|
return testSourceDirectories_;
|
|
}
|
|
|
|
/**
|
|
* Provides test files that should be compiled.
|
|
*
|
|
* @param files one or more test files
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation testSourceFiles(File... files) {
|
|
testSourceFiles_.addAll(Arrays.asList(files));
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Provides a list of test files that should be compiled.
|
|
*
|
|
* @param files a list of test files
|
|
* @return this operation instance
|
|
*/
|
|
public CompileKotlinOperation testSourceFiles(Collection<File> files) {
|
|
testSourceFiles_.addAll(files);
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the list of test files that should be compiled.
|
|
*
|
|
* @return the list of test files to compile
|
|
*/
|
|
public Collection<File> testSourceFiles() {
|
|
return testSourceFiles_;
|
|
}
|
|
} |