Support local variable arrays (#270)

This PR enables the existing mechanism to support for injecting annotation on local variables to fix errors coming from NullAway.

Author: avenger2597

annotator-core/build.gradle

@@ -57,7 +57,7 @@ spotless {
 }
 
 // Should be the latest supporting version of NullAway.
-def NULLAWAY_TEST = "0.12.3"
+def NULLAWAY_TEST = "0.12.4"
 
 tasks.test.dependsOn(':annotator-scanner:publishToMavenLocal')
 

annotator-core/src/test/java/edu/ucr/cs/riple/core/CoreTest.java

@@ -30,6 +30,7 @@
 import com.google.common.collect.ImmutableList;
 import edu.ucr.cs.riple.core.tools.TReport;
 import edu.ucr.cs.riple.injector.location.OnField;
+import edu.ucr.cs.riple.injector.location.OnLocalVariable;
 import edu.ucr.cs.riple.injector.location.OnMethod;
 import edu.ucr.cs.riple.injector.location.OnParameter;
 import java.util.Collections;
@@ -597,4 +598,28 @@ public void assignNullableToNonnullArrayComponentTypeTest() {
         .enableJSpecifyMode()
         .start();
   }
+
+  @Test
+  public void assignNullableToNonnullArrayComponentTypeLocalVariableTest() {
+    coreTestHelper
+        .onTarget()
+        .withSourceLines(
+            "Main.java",
+            "package test;",
+            "public class Main {",
+            "   void run() {",
+            "     Object[] arr = new Object[1];",
+            "     arr[0] = null;",
+            "   }",
+            "}")
+        .withExpectedReports(
+            new TReport(
+                new OnLocalVariable("Main.java", "test.Main", "run()", "arr"),
+                ImmutableList.of(ImmutableList.of(1, 0)),
+                -1))
+        .disableBailOut()
+        .checkExpectedOutput("assignNullableToNonnullArrayComponentTypeLocalVariableTest/expected")
+        .enableJSpecifyMode()
+        .start();
+  }
 }

annotator-core/src/test/resources/assignNullableToNonnullArrayComponentTypeLocalVariableTest/expected/Target/src/main/java/test/Main.java

@@ -0,0 +1,8 @@
+package test;
+import org.jspecify.annotations.Nullable;
+public class Main {
+   void run() {
+     @Nullable Object[] arr = new Object[1];
+     arr[0] = null;
+   }
+}

injector/src/main/java/edu/ucr/cs/riple/injector/location/Location.java

@@ -111,6 +111,8 @@ public static Location createLocationFromArrayInfo(String[] values) {
         return new OnMethod(path, clazz, values[2]);
       case PARAMETER:
         return new OnParameter(path, clazz, values[2], Integer.parseInt(values[4]));
+      case LOCAL_VARIABLE:
+        return new OnLocalVariable(path, clazz, values[2], values[3]);
       default:
         throw new RuntimeException(
             "Cannot reach this statement, values: " + Arrays.toString(values));