From 350454123fc8818208aac5afb665c50e9a95a991 Mon Sep 17 00:00:00 2001
From: Shannon Pamperl <shanman190@gmail.com>
Date: Wed, 4 Sep 2024 16:31:21 -0500
Subject: [PATCH] Add missing buildscript classpath configuration (#4459)

* Add missing buildscript classpath configuration

* Rollback updates to recipes to ease release
---
 rewrite-gradle/build.gradle.kts               |  3 +-
 .../gradle/marker/GradleBuildscript.java      | 43 +++++++++++++++++++
 .../gradle/marker/GradleProject.java          | 21 ++++++++-
 .../gradle/marker/GradleSettings.java         | 40 ++++++++++++++---
 4 files changed, 98 insertions(+), 9 deletions(-)
 create mode 100644 rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleBuildscript.java

diff --git a/rewrite-gradle/build.gradle.kts b/rewrite-gradle/build.gradle.kts
index 109bf35375c..6a13b318e64 100644
--- a/rewrite-gradle/build.gradle.kts
+++ b/rewrite-gradle/build.gradle.kts
@@ -89,6 +89,7 @@ tasks.named<Copy>("processResources") {
 //Javadoc compiler will complain about the use of the internal types.
 tasks.withType<Javadoc> {
     exclude(
-        "**/GradleProject**"
+        "**/GradleProject**",
+        "**/GradleSettings**"
     )
 }
diff --git a/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleBuildscript.java b/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleBuildscript.java
new file mode 100644
index 00000000000..7c08a94dfba
--- /dev/null
+++ b/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleBuildscript.java
@@ -0,0 +1,43 @@
+/*
+ * Copyright 2024 the original author or authors.
+ * <p>
+ * 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
+ * <p>
+ * https://www.apache.org/licenses/LICENSE-2.0
+ * <p>
+ * 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.marker;
+
+import lombok.Value;
+import lombok.With;
+import org.jspecify.annotations.Nullable;
+import org.openrewrite.maven.tree.MavenRepository;
+
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.UUID;
+
+@Value
+@With
+public class GradleBuildscript implements Serializable {
+    UUID id;
+    List<MavenRepository> mavenRepositories;
+    Map<String, GradleDependencyConfiguration> nameToConfiguration;
+
+    public @Nullable GradleDependencyConfiguration getConfiguration(String name) {
+        return nameToConfiguration.get(name);
+    }
+
+    public List<GradleDependencyConfiguration> getConfigurations() {
+        return new ArrayList<>(nameToConfiguration.values());
+    }
+}
diff --git a/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleProject.java b/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleProject.java
index 3e77f0bdd4f..3aa601e2571 100644
--- a/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleProject.java
+++ b/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleProject.java
@@ -60,24 +60,40 @@ public class GradleProject implements Marker, Serializable {
     List<MavenRepository> mavenRepositories;
 
     @With
+    @Deprecated
+    @Nullable
     List<MavenRepository> mavenPluginRepositories;
 
     Map<String, GradleDependencyConfiguration> nameToConfiguration;
 
+    GradleBuildscript buildscript;
+
     // Backwards compatibility to ease convoluted release process with rewrite-gradle-tooling-model
     public GradleProject(
             UUID id,
+            String group,
             String name,
+            String version,
             String path,
             List<GradlePluginDescriptor> plugins,
             List<MavenRepository> mavenRepositories,
             List<MavenRepository> mavenPluginRepositories,
             Map<String, GradleDependencyConfiguration> nameToConfiguration
     ) {
-        this(id, "", name, "", path, plugins, mavenRepositories, mavenPluginRepositories, nameToConfiguration);
+        this(id, group, name, version, path, plugins, mavenRepositories, mavenPluginRepositories, nameToConfiguration, null);
     }
 
+    /**
+     * Get a list of Maven plugin repositories.
+     *
+     * @return list of Maven plugin repositories
+     * @deprecated Use {@link GradleBuildscript#getMavenRepositories()} instead.
+     */
+    @Deprecated
     public List<MavenRepository> getMavenPluginRepositories() {
+        if (buildscript != null) {
+            return buildscript.getMavenRepositories();
+        }
         return mavenPluginRepositories == null ? Collections.emptyList() : mavenPluginRepositories;
     }
 
@@ -151,7 +167,8 @@ public GradleProject withNameToConfiguration(Map<String, GradleDependencyConfigu
                 plugins,
                 mavenRepositories,
                 mavenPluginRepositories,
-                configurations
+                configurations,
+                buildscript
         );
     }
 }
diff --git a/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleSettings.java b/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleSettings.java
index e231f56e365..2bd2db8cd63 100644
--- a/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleSettings.java
+++ b/rewrite-gradle/src/main/java/org/openrewrite/gradle/marker/GradleSettings.java
@@ -15,6 +15,8 @@
  */
 package org.openrewrite.gradle.marker;
 
+import com.fasterxml.jackson.annotation.JsonCreator;
+import lombok.AllArgsConstructor;
 import lombok.Value;
 import lombok.With;
 import org.jspecify.annotations.Nullable;
@@ -22,23 +24,35 @@
 import org.openrewrite.maven.tree.MavenRepository;
 
 import java.io.Serializable;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
-import java.util.UUID;
+import java.util.*;
 import java.util.stream.Collectors;
 
 @Value
 @With
+@AllArgsConstructor(onConstructor_ = { @JsonCreator})
 public class GradleSettings implements Marker, Serializable {
     UUID id;
+
+    @Deprecated
+    @Nullable
     List<MavenRepository> pluginRepositories;
+
     List<GradlePluginDescriptor> plugins;
     Map<String, FeaturePreview> featurePreviews;
+    GradleBuildscript buildscript;
+
+    // Backwards compatibility to ease convoluted release process with rewrite-gradle-tooling-model
+    public GradleSettings(
+            UUID id,
+            List<MavenRepository> pluginRepositories,
+            List<GradlePluginDescriptor> plugins,
+            Map<String, FeaturePreview> featurePreviews
+    ) {
+        this(id, pluginRepositories, plugins, featurePreviews, null);
+    }
 
     public @Nullable Boolean isFeatureEnabled(String name) {
-        // Unclear how enabled status can be determined in latest gradle APIs
-        return null;
+        return featurePreviews.get(name).getEnabled();
     }
 
     public Set<FeaturePreview> getActiveFeatures() {
@@ -46,4 +60,18 @@ public Set<FeaturePreview> getActiveFeatures() {
                 .filter(FeaturePreview::isActive)
                 .collect(Collectors.toSet());
     }
+
+    /**
+     * Get a list of Maven plugin repositories.
+     *
+     * @return list of Maven plugin repositories
+     * @deprecated Use {@link GradleBuildscript#getMavenRepositories()} instead.
+     */
+    @Deprecated
+    public List<MavenRepository> getPluginRepositories() {
+        if (buildscript != null) {
+            return buildscript.getMavenRepositories();
+        }
+        return pluginRepositories == null ? Collections.emptyList() : pluginRepositories;
+    }
 }