[java-analysis] Parameter nullability: prefer nullability known from …

…type over nullability known from parameter declaration

Type nullability could be more precise if parameter is generic
Fixes IDEA-364343 False-positive NPE at unboxing inside lambda with JSpecify annotations

GitOrigin-RevId: 9a49f5687eccaa013e639cdf15950be911e100bc

java/java-analysis-impl/src/com/intellij/codeInspection/dataFlow/DfaPsiUtil.java

@@ -307,12 +307,12 @@ public static Nullability getFunctionalParameterNullability(PsiFunctionalExpress
     if (sam != null) {
       PsiParameter parameter = sam.getParameterList().getParameter(index);
       if (parameter != null) {
-        Nullability nullability = getElementNullability(null, parameter);
-        if (nullability != Nullability.UNKNOWN) {
-          return nullability;
-        }
         PsiType parameterType = type.resolveGenerics().getSubstitutor().substitute(parameter.getType());
-        return getTypeNullability(GenericsUtil.eliminateWildcards(parameterType, false, true));
+        NullabilityAnnotationInfo info = getTypeNullabilityInfo(GenericsUtil.eliminateWildcards(parameterType, false, true));
+        if (info != null) {
+          return info.getNullability();
+        }
+        return getElementNullability(null, parameter);
       }
     }
     return Nullability.UNKNOWN;

java/java-tests/testData/inspection/dataFlow/fixture/JSpecifyUnboxingInLambda.java

@@ -0,0 +1,21 @@
+import org.jspecify.annotations.NullMarked;
+import org.jspecify.annotations.Nullable;
+
+// IDEA-364343
+@NullMarked
+class AnotherActivity {
+  public interface ThrowingFunction<T1 extends @Nullable Object, T2 extends @Nullable Object> {
+    T2 apply(T1 input) throws Throwable;
+  }
+
+  abstract static class Decoder<T extends @Nullable Object> {
+    abstract <T2 extends @Nullable Object> Decoder<T2> then(
+      ThrowingFunction<? super T, ? extends T2> dataTransform);
+  }
+
+  native Decoder<Boolean> foo();
+
+  Decoder<Boolean> doWork() {
+    return foo().then(f -> !f);
+  }
+}

java/java-tests/testSrc/com/intellij/java/codeInspection/DataFlowInspection9Test.java

@@ -46,4 +46,10 @@ public void testJSpecifyUpperBound() {
     setupTypeUseAnnotations("org.jspecify.annotations", myFixture);
     doTest();
   }
+
+  public void testJSpecifyUnboxingInLambda() {
+    addJSpecifyNullMarked(myFixture);
+    setupTypeUseAnnotations("org.jspecify.annotations", myFixture);
+    doTest();
+  }
 }