NoGuavaListsNewLinkedList

src/main/java/org/openrewrite/java/migrate/guava/NoGuavaListsNewLinkedList.java

@@ -0,0 +1,81 @@
+/*
+ * Copyright 2021 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.java.migrate.guava;
+
+import org.openrewrite.ExecutionContext;
+import org.openrewrite.Recipe;
+import org.openrewrite.TreeVisitor;
+import org.openrewrite.java.JavaIsoVisitor;
+import org.openrewrite.java.JavaTemplate;
+import org.openrewrite.java.JavaVisitor;
+import org.openrewrite.java.MethodMatcher;
+import org.openrewrite.java.search.UsesMethod;
+import org.openrewrite.java.tree.J;
+
+public class NoGuavaListsNewLinkedList extends Recipe {
+    private static final MethodMatcher NEW_LINKED_LIST = new MethodMatcher("com.google.common.collect.Lists newLinkedList()");
+    private static final MethodMatcher NEW_LINKED_LIST_ITERABLE = new MethodMatcher("com.google.common.collect.Lists newLinkedList(java.lang.Iterable)");
+
+    @Override
+    public String getDisplayName() {
+        return "Use `new LinkedList<>()` instead of Guava";
+    }
+
+    @Override
+    public String getDescription() {
+        return "Prefer the Java standard library over third-party usage of Guava in simple cases like this.";
+    }
+
+    @Override
+    protected TreeVisitor<?, ExecutionContext> getApplicableTest() {
+        return new JavaIsoVisitor<ExecutionContext>() {
+            @Override
+            public J.CompilationUnit visitCompilationUnit(J.CompilationUnit cu, ExecutionContext executionContext) {
+                doAfterVisit(new UsesMethod<>(NEW_LINKED_LIST));
+                doAfterVisit(new UsesMethod<>(NEW_LINKED_LIST_ITERABLE));
+                return cu;
+            }
+        };
+    }
+
+    @Override
+    protected TreeVisitor<?, ExecutionContext> getVisitor() {
+        return new JavaVisitor<ExecutionContext>() {
+            private final JavaTemplate newLinkedList = template("new LinkedList<>()")
+                    .imports("java.util.LinkedList")
+                    .build();
+
+            private final JavaTemplate newLinkedListIterable = template("new LinkedList<>(#{any(java.lang.Iterable)})")
+                    .imports("java.util.LinkedList")
+                    .build();
+
+            @Override
+            public J visitMethodInvocation(J.MethodInvocation method, ExecutionContext executionContext) {
+                if (NEW_LINKED_LIST.matches(method)) {
+                    maybeRemoveImport("com.google.common.collect.Lists");
+                    maybeAddImport("java.util.LinkedList");
+                    return method.withTemplate(newLinkedList, method.getCoordinates().replace());
+                } else if (NEW_LINKED_LIST_ITERABLE.matches(method)) {
+                    maybeRemoveImport("com.google.common.collect.Lists");
+                    maybeAddImport("java.util.LinkedList");
+                    return method.withTemplate(newLinkedListIterable, method.getCoordinates().replace(),
+                            method.getArguments().get(0));
+                }
+                return super.visitMethodInvocation(method, executionContext);
+            }
+        };
+    }
+}

src/test/kotlin/org/openrewrite/java/migrate/guava/NoGuavaListsNewLinkedListTest.kt

@@ -0,0 +1,79 @@
+/*
+ * Copyright 2021 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.java.migrate.guava
+
+import org.junit.jupiter.api.Test
+import org.openrewrite.Recipe
+import org.openrewrite.java.JavaParser
+import org.openrewrite.java.JavaRecipeTest
+
+class NoGuavaListsNewLinkedListTest: JavaRecipeTest {
+    override val parser: JavaParser
+        get() = JavaParser.fromJavaVersion()
+            .logCompilationWarningsAndErrors(true)
+            .classpath("guava")
+            .build()
+
+    override val recipe: Recipe
+        get() = NoGuavaListsNewLinkedList()
+
+    @Test
+    fun replaceWithNewLinkedList() = assertChanged(
+        before = """
+            import com.google.common.collect.*;
+            
+            import java.util.List;
+            
+            class Test {
+                List<Integer> cardinalsWorldSeries = Lists.newLinkedList();
+            }
+        """,
+        after = """
+            import java.util.LinkedList;
+            import java.util.List;
+            
+            class Test {
+                List<Integer> cardinalsWorldSeries = new LinkedList<>();
+            }
+        """
+    )
+
+    @Test
+    fun replaceWithNewLinkedListIterable() = assertChanged(
+        before = """
+            import com.google.common.collect.*;
+            
+            import java.util.Collections;
+            import java.util.LinkedList;
+            import java.util.List;
+            
+            class Test {
+                List<Integer> l = Collections.emptyList();
+                List<Integer> cardinalsWorldSeries = Lists.newLinkedList(l);
+            }
+        """,
+        after = """
+            import java.util.Collections;
+            import java.util.LinkedList;
+            import java.util.List;
+            
+            class Test {
+                List<Integer> l = Collections.emptyList();
+                List<Integer> cardinalsWorldSeries = new LinkedList<>(l);
+            }
+        """
+    )
+}