org.openrewrite.gradle.toolingapi.Assertions Maven / Gradle / Ivy
Show all versions of model Show documentation
/*
* Copyright 2024 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.openrewrite.gradle.toolingapi;
import org.jspecify.annotations.Nullable;
import org.openrewrite.InMemoryExecutionContext;
import org.openrewrite.SourceFile;
import org.openrewrite.gradle.UpdateGradleWrapper;
import org.openrewrite.gradle.marker.GradleProject;
import org.openrewrite.gradle.marker.GradleSettings;
import org.openrewrite.gradle.util.GradleWrapper;
import org.openrewrite.groovy.tree.G;
import org.openrewrite.marker.OperatingSystemProvenance;
import org.openrewrite.properties.tree.Properties;
import org.openrewrite.test.UncheckedConsumer;
import java.io.File;
import java.io.IOException;
import java.io.UncheckedIOException;
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardOpenOption;
import java.nio.file.attribute.PosixFilePermission;
import java.util.List;
import java.util.Set;
import static java.util.Objects.requireNonNull;
public class Assertions {
public static UncheckedConsumer> withToolingApi(URI distributionUrl) {
return withToolingApi(GradleWrapper.create(distributionUrl, new InMemoryExecutionContext()));
}
@SuppressWarnings("unused")
public static UncheckedConsumer> withToolingApi(URI distributionUrl, String initScriptContents) {
return withToolingApi(GradleWrapper.create(distributionUrl, new InMemoryExecutionContext()), initScriptContents);
}
public static UncheckedConsumer> withToolingApi(@Nullable GradleWrapper gradleWrapper) {
return withToolingApi(gradleWrapper, null);
}
public static UncheckedConsumer> withToolingApi(@Nullable GradleWrapper gradleWrapper, @Nullable String initScriptContents) {
return sourceFiles -> {
try {
Path tempDirectory = Files.createTempDirectory("project");
// Usage of Assertions.mavenProject() might result in gradle files inside a subdirectory
Path projectDir = tempDirectory;
try {
for (SourceFile sourceFile : sourceFiles) {
if (sourceFile instanceof G.CompilationUnit) {
G.CompilationUnit g = (G.CompilationUnit) sourceFile;
if (g.getSourcePath().toString().endsWith(".gradle")) {
Path groovyGradle = tempDirectory.resolve(g.getSourcePath());
if (!tempDirectory.equals(groovyGradle.getParent()) && tempDirectory.equals(groovyGradle.getParent().getParent())) {
projectDir = groovyGradle.getParent();
}
Files.createDirectories(groovyGradle.getParent());
Files.write(groovyGradle, g.printAllAsBytes());
}
} else if (sourceFile instanceof Properties.File) {
Properties.File f = (Properties.File) sourceFile;
if (f.getSourcePath().endsWith("gradle.properties")) {
Path gradleProperties = tempDirectory.resolve(f.getSourcePath());
if (!tempDirectory.equals(gradleProperties.getParent()) && tempDirectory.equals(gradleProperties.getParent().getParent())) {
projectDir = gradleProperties.getParent();
}
Files.createDirectories(gradleProperties.getParent());
Files.write(gradleProperties, f.printAllAsBytes());
}
}
}
if(gradleWrapper != null) {
Files.createDirectories(projectDir.resolve("gradle/wrapper/"));
Files.write(projectDir.resolve(GradleWrapper.WRAPPER_PROPERTIES_LOCATION),
("distributionBase=GRADLE_USER_HOME\n" +
"distributionPath=wrapper/dists\n" +
"distributionUrl=" + gradleWrapper.getPropertiesFormattedUrl() + "\n" +
((gradleWrapper.getDistributionChecksum() == null) ? "" : "distributionSha256Sum=" + gradleWrapper.getDistributionChecksum().getHexValue() + "\n") +
"zipStoreBase=GRADLE_USER_HOME\n" +
"zipStorePath=wrapper/dists").getBytes(StandardCharsets.UTF_8), StandardOpenOption.CREATE_NEW);
Files.write(projectDir.resolve(GradleWrapper.WRAPPER_JAR_LOCATION), gradleWrapper.wrapperJar().printAllAsBytes(), StandardOpenOption.CREATE_NEW);
Path gradleSh = projectDir.resolve(GradleWrapper.WRAPPER_SCRIPT_LOCATION);
Files.copy(requireNonNull(UpdateGradleWrapper.class.getResourceAsStream("/gradlew")), gradleSh);
OperatingSystemProvenance current = OperatingSystemProvenance.current();
if (current.isLinux() || current.isMacOsX()) {
Set permissions = Files.getPosixFilePermissions(gradleSh);
permissions.add(PosixFilePermission.OWNER_EXECUTE);
Files.setPosixFilePermissions(gradleSh, permissions);
}
Files.copy(requireNonNull(UpdateGradleWrapper.class.getResourceAsStream("/gradlew.bat")), projectDir.resolve(GradleWrapper.WRAPPER_BATCH_LOCATION));
}
for (int i = 0; i < sourceFiles.size(); i++) {
SourceFile sourceFile = sourceFiles.get(i);
if (sourceFile.getSourcePath().toString().endsWith(".gradle")) {
if (sourceFile.getSourcePath().endsWith("settings.gradle")) {
OpenRewriteModel model = OpenRewriteModelBuilder.forProjectDirectory(tempDirectory.resolve(sourceFile.getSourcePath()).getParent().toFile(), null, initScriptContents);
org.openrewrite.gradle.toolingapi.GradleSettings rawSettings = model.gradleSettings();
if (rawSettings != null) {
GradleSettings gradleSettings = org.openrewrite.gradle.toolingapi.GradleSettings. toMarker(rawSettings);
sourceFiles.set(i, sourceFile.withMarkers(sourceFile.getMarkers().add(gradleSettings)));
}
} else {
OpenRewriteModel model = OpenRewriteModelBuilder.forProjectDirectory(projectDir.toFile(), tempDirectory.resolve(sourceFile.getSourcePath()).toFile(), initScriptContents);
GradleProject gradleProject = org.openrewrite.gradle.toolingapi.GradleProject.toMarker(model.gradleProject());
sourceFiles.set(i, sourceFile.withMarkers(sourceFile.getMarkers().add(gradleProject)));
}
}
}
} finally {
deleteDirectory(tempDirectory.toFile());
}
} catch (IOException e) {
throw new UncheckedIOException(e);
}
};
}
public static UncheckedConsumer> withToolingApi(@Nullable String version, @Nullable String distribution) {
GradleWrapper gradleWrapper = null;
if (version != null) {
gradleWrapper = GradleWrapper.create(distribution, version, null, new InMemoryExecutionContext());
}
return withToolingApi(gradleWrapper);
}
@SuppressWarnings("unused")
public static UncheckedConsumer> withToolingApi(String version) {
return withToolingApi(version, "bin");
}
public static UncheckedConsumer> withToolingApi() {
return withToolingApi((GradleWrapper) null, null);
}
private static void deleteDirectory(File directoryToBeDeleted) {
File[] allContents = directoryToBeDeleted.listFiles();
if (allContents != null) {
for (File file : allContents) {
deleteDirectory(file);
}
}
//noinspection ResultOfMethodCallIgnored
directoryToBeDeleted.delete();
}
}