Add a flag to configure the default nullness annotation

PiperOrigin-RevId: 386953444
google · Jul 26, 2021 · de0b017 · de0b017
1 parent c5218ed
commit de0b017
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/core/src/main/java/com/google/errorprone/bugpatterns/nullness/NullnessFixes.java b/core/src/main/java/com/google/errorprone/bugpatterns/nullness/NullnessFixes.java
@@ -16,6 +16,7 @@
 
 package com.google.errorprone.bugpatterns.nullness;
 
+import com.google.errorprone.ErrorProneFlags;
 import com.google.errorprone.VisitorState;
 import com.google.errorprone.fixes.SuggestedFix;
 import com.google.errorprone.util.FindIdentifiers;
@@ -63,8 +64,14 @@ private static String getQualifiedName(VisitorState state, SuggestedFix.Builder
      */
     // TODO(cpovirk): Suggest @NullableDecl if the code uses that.
     Symbol sym = FindIdentifiers.findIdent("Nullable", state, KindSelector.VAL_TYP);
+    ErrorProneFlags flags = state.errorProneOptions().getFlags();
     String defaultType =
-        state.isAndroidCompatible() ? "androidx.annotation.Nullable" : "javax.annotation.Nullable";
+        flags
+            .get("Nullness:DefaultNullnessAnnotation")
+            .orElse(
+                state.isAndroidCompatible()
+                    ? "androidx.annotation.Nullable"
+                    : "javax.annotation.Nullable");
     if (sym != null) {
       ClassSymbol classSym = (ClassSymbol) sym;
       if (classSym.isAnnotationType()) {