Added gradle recipe to remove enableFeaturePreview method

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

@@ -0,0 +1,87 @@
+/*
+ * 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;
+
+import java.util.List;
+import lombok.EqualsAndHashCode;
+import lombok.Value;
+import org.jetbrains.annotations.NotNull;
+import org.openrewrite.ExecutionContext;
+import org.openrewrite.Option;
+import org.openrewrite.Preconditions;
+import org.openrewrite.Recipe;
+import org.openrewrite.TreeVisitor;
+import org.openrewrite.groovy.GroovyIsoVisitor;
+import org.openrewrite.java.tree.Expression;
+import org.openrewrite.java.tree.J;
+import org.openrewrite.java.tree.J.Literal;
+import org.openrewrite.java.tree.J.MethodInvocation;
+
+@Value
+@EqualsAndHashCode(callSuper = false)
+public class RemoveEnableFeaturePreview extends Recipe {
+
+  @Option(displayName = "The feature preview name",
+      description = "The name of the feature preview to remove.",
+      example = "ONE_LOCKFILE_PER_PROJECT"
+  )
+  public String previewFeatureName;
+
+  @Override
+  public @NotNull String getDisplayName() {
+    return "Remove an enabled Gradle preview feature";
+  }
+
+  @Override
+  public @NotNull String getDescription() {
+    return "Remove an enabled Gradle preview feature from `settings.gradle`.";
+  }
+
+  @Override
+  public TreeVisitor<?, ExecutionContext> getVisitor() {
+    return Preconditions.check(
+        new IsSettingsGradle<>(),
+        new RemoveEnableFeaturePreviewVisitor());
+  }
+
+  public class RemoveEnableFeaturePreviewVisitor extends GroovyIsoVisitor<ExecutionContext> {
+
+    @Override
+    public MethodInvocation visitMethodInvocation(
+        MethodInvocation method, @NotNull ExecutionContext executionContext) {
+
+      if (!"enableFeaturePreview".equals(method.getSimpleName())) {
+        return method;
+      }
+
+      List<Expression> arguments = method.getArguments();
+      if (arguments == null || arguments.size() != 1) {
+        return method;
+      }
+
+      Expression argument = arguments.get(0);
+      if (argument instanceof J.Literal) {
+        String candidatePreviewFeatureName = (String) ((Literal) argument).getValue();
+        if (previewFeatureName.equals(candidatePreviewFeatureName)) {
+          return null;
+        }
+      }
+
+      return method;
+    }
+  }
+
+}

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

@@ -0,0 +1,160 @@
+/*
+ * 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;
+
+import java.nio.file.Paths;
+import org.junit.jupiter.api.Test;
+import org.openrewrite.test.RewriteTest;
+
+public class RemoveEnableFeaturePreviewTest implements RewriteTest {
+
+    @Test
+    void testRemoveEnableFeaturePreviewMethodRecipe_singleQuotes() {
+        //language=gradle
+        rewriteRun(
+          spec -> spec.recipe(new RemoveEnableFeaturePreview("ONE_LOCKFILE_PER_PROJECT")),
+          Assertions.settingsGradle(
+            """
+                pluginManagement {
+                    repositories {
+                        gradlePluginPortal()
+                    }
+                }
+
+                rootProject.name = 'merge-service'
+                enableFeaturePreview('ONE_LOCKFILE_PER_PROJECT')
+              """,
+            """
+                pluginManagement {
+                    repositories {
+                        gradlePluginPortal()
+                    }
+                }
+
+                rootProject.name = 'merge-service'
+              """,
+            spec -> spec.path(Paths.get("settings.gradle"))
+          )
+        );
+    }
+
+    @Test
+    void testRemoveEnableFeaturePreviewMethodRecipe_doubleQuotes() {
+        //language=gradle
+        rewriteRun(
+          spec -> spec.recipe(new RemoveEnableFeaturePreview("ONE_LOCKFILE_PER_PROJECT")),
+          Assertions.settingsGradle(
+            """
+                pluginManagement {
+                    repositories {
+                        gradlePluginPortal()
+                    }
+                }
+
+                rootProject.name = 'merge-service'
+                enableFeaturePreview("ONE_LOCKFILE_PER_PROJECT")
+
+                include 'service'
+              """,
+            """
+                pluginManagement {
+                    repositories {
+                        gradlePluginPortal()
+                    }
+                }
+
+                rootProject.name = 'merge-service'
+
+                include 'service'
+              """,
+            spec -> spec.path(Paths.get("settings.gradle"))
+          )
+        );
+    }
+
+    @Test
+    void testRemoveEnableFeaturePreviewMethodRecipe_noChange() {
+        //language=gradle
+        rewriteRun(
+          spec -> spec.recipe(new RemoveEnableFeaturePreview("ONE_LOCKFILE_PER_PROJECT")),
+          Assertions.settingsGradle(
+            """
+                pluginManagement {
+                    repositories {
+                        gradlePluginPortal()
+                    }
+                }
+
+                enableFeaturePreview("DIFFERENT_FEATURE")
+
+                rootProject.name = 'merge-service'
+
+                include 'service'
+              """,
+            spec -> spec.path(Paths.get("settings.gradle"))
+          )
+        );
+    }
+
+    @Test
+    void testRemoveEnableFeaturePreviewMethodRecipe_noChangeNoArgument() {
+        //language=gradle
+        rewriteRun(
+          spec -> spec.recipe(new RemoveEnableFeaturePreview("ONE_LOCKFILE_PER_PROJECT")),
+          Assertions.settingsGradle(
+            """
+                pluginManagement {
+                    repositories {
+                        gradlePluginPortal()
+                    }
+                }
+
+                enableFeaturePreview()
+
+                rootProject.name = 'merge-service'
+
+                include 'service'
+              """,
+            spec -> spec.path(Paths.get("settings.gradle"))
+          )
+        );
+    }
+
+    @Test
+    void testRemoveEnableFeaturePreviewMethodRecipe_noChangeNullArgument() {
+        //language=gradle
+        rewriteRun(
+          spec -> spec.recipe(new RemoveEnableFeaturePreview("ONE_LOCKFILE_PER_PROJECT")),
+          Assertions.settingsGradle(
+            """
+                pluginManagement {
+                    repositories {
+                        gradlePluginPortal()
+                    }
+                }
+
+                enableFeaturePreview(null)
+
+                rootProject.name = 'merge-service'
+
+                include 'service'
+              """,
+            spec -> spec.path(Paths.get("settings.gradle"))
+          )
+        );
+    }
+
+}