Improve handling of type parameter names that start with _

PiperOrigin-RevId: 351412076
google · Jan 12, 2021 · f7c5e6c · f7c5e6c
1 parent 99968c5
commit f7c5e6c
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 5 deletions.
diff --git a/check_api/src/main/java/com/google/errorprone/names/NamingConventions.java b/check_api/src/main/java/com/google/errorprone/names/NamingConventions.java
@@ -18,8 +18,8 @@
 
 import static com.google.common.collect.ImmutableList.toImmutableList;
 
+import com.google.common.base.Splitter;
 import com.google.common.collect.ImmutableList;
-import java.util.Arrays;
 import java.util.regex.Pattern;
 import java.util.stream.Collectors;
 
@@ -32,8 +32,9 @@ public class NamingConventions {
   private static final String CASE_TRANSITION = "(?<=[a-z0-9])(?=[A-Z])";
   private static final String TRAILING_DIGITS = "(?<![0-9_])(?=[0-9]+$)";
 
-  private static final Pattern TERM_SPLITTER =
-      Pattern.compile(String.format("%s|%s|%s", UNDERSCORE, CASE_TRANSITION, TRAILING_DIGITS));
+  private static final Splitter TERM_SPLITTER =
+      Splitter.onPattern(String.format("%s|%s|%s", UNDERSCORE, CASE_TRANSITION, TRAILING_DIGITS))
+          .omitEmptyStrings();
 
   /**
    * Split a Java name into terms based on either Camel Case or Underscores. We also split digits at
@@ -47,7 +48,8 @@ public static ImmutableList<String> splitToLowercaseTerms(String identifierName)
       // Degenerate case of names which contain only underscore
       return ImmutableList.of(identifierName);
     }
-    return Arrays.stream(TERM_SPLITTER.split(identifierName))
+    return TERM_SPLITTER
+        .splitToStream(identifierName)
         .map(String::toLowerCase)
         .collect(toImmutableList());
   }

diff --git a/core/src/main/java/com/google/errorprone/bugpatterns/TypeParameterNaming.java b/core/src/main/java/com/google/errorprone/bugpatterns/TypeParameterNaming.java
@@ -205,7 +205,8 @@ private static List<TypeVariableSymbol> typeVariablesEnclosing(Symbol sym) {
   }
 
   private static String suggestedSingleLetter(String id, Tree tree) {
-    char firstLetter = id.charAt(0);
+    char firstLetter =
+        Ascii.toUpperCase(NamingConventions.splitToLowercaseTerms(id).get(0).charAt(0));
     Symbol sym = ASTHelpers.getSymbol(tree);
     List<TypeVariableSymbol> enclosingTypeSymbols = typeVariablesEnclosing(sym);
 

diff --git a/core/src/test/java/com/google/errorprone/bugpatterns/TypeParameterNamingTest.java b/core/src/test/java/com/google/errorprone/bugpatterns/TypeParameterNamingTest.java
@@ -294,6 +294,25 @@ public void negativeCases_manyNumberedTypes() {
         .doTest();
   }
 
+  @Test
+  public void refactoring_underscore() {
+    refactoring
+        .addInputLines(
+            "in/Test.java", //
+            "class Test {",
+            "  public <_T> void method(_T t) {",
+            "  }",
+            "}")
+        .addOutputLines(
+            "in/Test.java", //
+            "class Test {",
+            "  public <T> void method(T t) {",
+            "  }",
+            "}")
+        .setFixChooser(FixChoosers.FIRST)
+        .doTest(TestMode.TEXT_MATCH);
+  }
+
   @Test
   public void classifyTypeName_singleLetter() {
     assertKindOfName("T").isEqualTo(LETTER_WITH_MAYBE_NUMERAL);