Verify ChangeDependencyArtifactId works with .kts

Author: timtebeek

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

@@ -83,20 +83,19 @@ public TreeVisitor<?, ExecutionContext> getVisitor() {
             GradleProject gradleProject;
 
             @Override
-            public @org.jspecify.annotations.Nullable J visit(@org.jspecify.annotations.Nullable Tree tree, ExecutionContext ctx) {
+            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()) {
-                        return (J) tree;
-                    }
-                    gradleProject = maybeGp.get();
-                    J result = super.visit(tree, ctx);
-                    if (result != tree && result instanceof JavaSourceFile) {
-                        JavaSourceFile updated = (JavaSourceFile) result;
-                        result = updated.withMarkers(updated.getMarkers().setByType(updateGradleModel(gradleProject)));
+                    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 result;
+                    return sf;
                 }
                 return super.visit(tree, ctx);
             }

rewrite-gradle/src/test/java/org/openrewrite/gradle/ChangeDependencyArtifactIdTest.java

@@ -26,6 +26,7 @@
 
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.openrewrite.gradle.Assertions.buildGradle;
+import static org.openrewrite.gradle.Assertions.buildGradleKts;
 import static org.openrewrite.gradle.toolingapi.Assertions.withToolingApi;
 
 class ChangeDependencyArtifactIdTest implements RewriteTest {
@@ -450,6 +451,41 @@ void worksWithDependencyDefinedInBuildScript() {
         );
     }
 
+    @Test
+    void worksWithEmptyStringConfigKts() {
+        rewriteRun(
+          spec -> spec.recipe(new ChangeDependencyArtifactId("org.springframework.boot", "spring-boot-starter", "new-starter", "")),
+          buildGradleKts(
+            """
+              plugins {
+                  id("java-library")
+              }
+
+              repositories {
+                  mavenCentral()
+              }
+
+              dependencies {
+                  implementation("org.springframework.boot:spring-boot-starter:2.5.4")
+              }
+              """,
+            """
+              plugins {
+                  id("java-library")
+              }
+
+              repositories {
+                  mavenCentral()
+              }
+
+              dependencies {
+                  implementation("org.springframework.boot:new-starter:2.5.4")
+              }
+              """
+          )
+        );
+    }
+
     @Test
     void dependenciesBlockInFreestandingScript() {
         rewriteRun(