Add Kotlin script template support for Gradle KTS (#6792)

* Add Kotlin script template support for Gradle KTS parsing

Leverage K2 FIR compiler extensions to configure implicit receivers and default
imports for build.gradle.kts and settings.gradle.kts files. This improves type
information for Gradle DSL methods, enabling better recipe matching and method
resolution. Implements FirScriptConfiguratorExtension and
FirScriptResolutionConfigurationExtension to provide Project/Settings as implicit
receivers and Gradle API packages as default imports.

* Unify Gradle recipe visitors for Groovy and Kotlin DSL

Now that Kotlin script templates provide type information for .gradle.kts
files, merge separate GroovyVisitor/KotlinVisitor pairs into single
JavaVisitor implementations where possible.

- UpdateJavaCompatibility: replace dual GroovyIsoVisitor + KotlinIsoVisitor
  with a single JavaVisitor that handles both DSLs
- UpgradeTransitiveDependencyVersion: replace inline GroovyIsoVisitor and
  KotlinIsoVisitor literal visitors with shared ChangeStringLiteral, and
  simplify the constraints precondition check

* Fix FindRepository matching for KTS with type information

Replace MethodMatcher-based pluginManagement/buildscript detection with
simple name matching, since the synthetic class names (RewriteSettings,
RewriteGradleProject) don't match the real types resolved by the K2
compiler (org.gradle.api.initialization.Settings, org.gradle.api.Project).

* Add FIR-to-IR conversion phase to KotlinParser (#6793)

After FIR analysis, run JvmFir2IrPipelinePhase to populate the
irFile field on KotlinSource, enabling future use of IR-based
type mapping for improved Kotlin parsing.

* Unify more Gradle recipe visitors for Groovy and Kotlin DSL

Convert ChangeDependencyGroupId and ChangeDependencyArtifactId from
GroovyIsoVisitor to JavaIsoVisitor so they work with both Groovy and
Kotlin DSL build files.

* Use real Gradle API types in MethodMatcher patterns for KTS support

Update MethodMatcher patterns from synthetic Groovy template types
(RewriteGradleProject, RewriteSettings, PluginSpec, Plugin) to real
Gradle API types (org.gradle.api.Project, Settings, PluginDependenciesSpec,
PluginDependencySpec) with matchOverrides=true, enabling matching for
both Groovy DSL and Kotlin DSL scripts.

* Increase test heap to 2g for rewrite-gradle

The K2 compiler used for KTS parsing loads many JDK modules and
consumes significant memory. The default test worker heap is
insufficient, causing OutOfMemoryError on CI.

* Use regular imports in Plugin.groovy

* Restore `isKotlin` conditions in RemovePluginVisitor due to missing types

* Verify `ChangeDependencyArtifactId` works with .kts

* Verify `ChangeDependencyGroupId` works with .kts

* Verify `DependencyInsight` works with .kts

* Verify `DependencyConstraintToRule` works with .kts

* Merge Groovy/Kotlin visitors into single JavaIsoVisitor in `UpdateJavaCompatibility`

Author: timtebeek

rewrite-gradle/build.gradle.kts

@@ -89,6 +89,7 @@ java {
 tasks.withType<Test>().configureEach {
     dependsOn(pluginLocalTestClasspath)
     systemProperty("org.openrewrite.gradle.local.use-embedded-classpath", pluginLocalTestClasspath.files.find { it.name == "test-manifest.txt" }!!.path)
+    maxHeapSize = "2g"
 }
 
 // This seems to be the only way to get the groovy compiler to emit java-8 compatible bytecode

rewrite-gradle/src/main/groovy/Plugin.groovy

@@ -13,11 +13,14 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-interface PluginSpec {
+import org.gradle.plugin.use.PluginDependenciesSpec
+import org.gradle.plugin.use.PluginDependencySpec
+
+interface PluginSpec extends PluginDependenciesSpec {
     Plugin id(String i)
 }
 
-interface Plugin {
+interface Plugin extends PluginDependencySpec {
     Plugin version(String v)
     Plugin apply(boolean a)
 }

rewrite-gradle/src/main/java/org/openrewrite/gradle/ChangeDependencyArtifactId.java

@@ -22,11 +22,11 @@
 import org.openrewrite.gradle.marker.GradleDependencyConfiguration;
 import org.openrewrite.gradle.marker.GradleProject;
 import org.openrewrite.gradle.trait.GradleMultiDependency;
-import org.openrewrite.groovy.GroovyIsoVisitor;
-import org.openrewrite.groovy.tree.G;
 import org.openrewrite.internal.ListUtils;
 import org.openrewrite.internal.StringUtils;
+import org.openrewrite.java.JavaIsoVisitor;
 import org.openrewrite.java.tree.J;
+import org.openrewrite.java.tree.JavaSourceFile;
 import org.openrewrite.semver.DependencyMatcher;
 
 import java.util.HashMap;
@@ -76,26 +76,28 @@ public Validated<Object> validate() {
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return Preconditions.check(new IsBuildGradle<>(), new GroovyIsoVisitor<ExecutionContext>() {
+        return Preconditions.check(new IsBuildGradle<>(), new JavaIsoVisitor<ExecutionContext>() {
             final DependencyMatcher depMatcher = requireNonNull(DependencyMatcher.build(groupId + ":" + artifactId).getValue());
 
             @SuppressWarnings("NotNullFieldNotInitialized")
             GradleProject gradleProject;
 
             @Override
-            public G.CompilationUnit visitCompilationUnit(G.CompilationUnit cu, ExecutionContext ctx) {
-                Optional<GradleProject> maybeGp = cu.getMarkers().findFirst(GradleProject.class);
-                if (!maybeGp.isPresent()) {
-                    return cu;
-                }
-
-                gradleProject = maybeGp.get();
-
-                G.CompilationUnit g = super.visitCompilationUnit(cu, ctx);
-                if (g != cu) {
-                    g = g.withMarkers(g.getMarkers().setByType(updateGradleModel(gradleProject)));
+            public @Nullable J visit(@Nullable Tree tree, ExecutionContext ctx) {
+                if (tree instanceof JavaSourceFile) {
+                    JavaSourceFile sf = (JavaSourceFile) tree;
+                    Optional<GradleProject> maybeGp = sf.getMarkers().findFirst(GradleProject.class);
+                    if (maybeGp.isPresent()) {
+                        gradleProject = maybeGp.get();
+                        J result = super.visit(tree, ctx);
+                        if (result != tree && result instanceof JavaSourceFile) {
+                            JavaSourceFile updated = (JavaSourceFile) result;
+                            return updated.withMarkers(updated.getMarkers().setByType(updateGradleModel(gradleProject)));
+                        }
+                    }
+                    return sf;
                 }
-                return g;
+                return super.visit(tree, ctx);
             }
 
             @Override

rewrite-gradle/src/main/java/org/openrewrite/gradle/ChangeDependencyGroupId.java

@@ -22,11 +22,11 @@
 import org.openrewrite.gradle.marker.GradleDependencyConfiguration;
 import org.openrewrite.gradle.marker.GradleProject;
 import org.openrewrite.gradle.trait.GradleMultiDependency;
-import org.openrewrite.groovy.GroovyIsoVisitor;
-import org.openrewrite.groovy.tree.G;
 import org.openrewrite.internal.ListUtils;
 import org.openrewrite.internal.StringUtils;
+import org.openrewrite.java.JavaIsoVisitor;
 import org.openrewrite.java.tree.J;
+import org.openrewrite.java.tree.JavaSourceFile;
 import org.openrewrite.semver.DependencyMatcher;
 
 import java.util.HashMap;
@@ -76,26 +76,28 @@ public Validated<Object> validate() {
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return Preconditions.check(new IsBuildGradle<>(), new GroovyIsoVisitor<ExecutionContext>() {
+        return Preconditions.check(new IsBuildGradle<>(), new JavaIsoVisitor<ExecutionContext>() {
             final DependencyMatcher depMatcher = requireNonNull(DependencyMatcher.build(groupId + ":" + artifactId).getValue());
 
             @SuppressWarnings("NotNullFieldNotInitialized")
             GradleProject gradleProject;
 
             @Override
-            public G.CompilationUnit visitCompilationUnit(G.CompilationUnit cu, ExecutionContext ctx) {
-                Optional<GradleProject> maybeGp = cu.getMarkers().findFirst(GradleProject.class);
-                if (!maybeGp.isPresent()) {
-                    return cu;
-                }
-
-                gradleProject = maybeGp.get();
-
-                G.CompilationUnit g = super.visitCompilationUnit(cu, ctx);
-                if (g != cu) {
-                    g = g.withMarkers(g.getMarkers().setByType(updateGradleModel(gradleProject)));
+            public @Nullable J visit(@Nullable Tree tree, ExecutionContext ctx) {
+                if (tree instanceof JavaSourceFile) {
+                    JavaSourceFile sf = (JavaSourceFile) tree;
+                    Optional<GradleProject> maybeGp = sf.getMarkers().findFirst(GradleProject.class);
+                    if (maybeGp.isPresent()) {
+                        gradleProject = maybeGp.get();
+                        J result = super.visit(tree, ctx);
+                        if (result != tree && result instanceof JavaSourceFile) {
+                            JavaSourceFile updated = (JavaSourceFile) result;
+                            return updated.withMarkers(updated.getMarkers().setByType(updateGradleModel(gradleProject)));
+                        }
+                    }
+                    return sf;
                 }
-                return g;
+                return super.visit(tree, ctx);
             }
 
             @Override

rewrite-gradle/src/main/java/org/openrewrite/gradle/DependencyConstraintToRule.java

@@ -45,7 +45,7 @@
 @EqualsAndHashCode(callSuper = false)
 public class DependencyConstraintToRule extends Recipe {
 
-    private static final MethodMatcher DEPENDENCIES_DSL_MATCHER = new MethodMatcher("RewriteGradleProject dependencies(..)");
+    private static final MethodMatcher DEPENDENCIES_DSL_MATCHER = new MethodMatcher("org.gradle.api.Project dependencies(..)", true);
     private static final String CONSTRAINT_MATCHER = "org.gradle.api.artifacts.dsl.DependencyHandler *(..)";
 
     String displayName = "Dependency constraint to resolution rule";

rewrite-gradle/src/main/java/org/openrewrite/gradle/GradleParser.java

@@ -66,6 +66,9 @@ public Stream<SourceFile> parseInputs(Iterable<Input> sources, @Nullable Path re
             }
             kotlinBuildParser = KotlinParser.builder(base.kotlinParser)
                     .classpath(buildscriptClasspath)
+                    .isKotlinScript(true)
+                    .scriptImplicitReceivers("org.gradle.api.Project")
+                    .scriptDefaultImports(DefaultImportsCustomizer.DEFAULT_IMPORTS)
                     .build();
         }
         if (groovySettingsParser == null) {
@@ -88,6 +91,9 @@ public Stream<SourceFile> parseInputs(Iterable<Input> sources, @Nullable Path re
             }
             kotlinSettingsParser = KotlinParser.builder(base.kotlinParser)
                     .classpath(settingsClasspath)
+                    .isKotlinScript(true)
+                    .scriptImplicitReceivers("org.gradle.api.initialization.Settings")
+                    .scriptDefaultImports(DefaultImportsCustomizer.DEFAULT_IMPORTS)
                     .build();
         }