IDEA-302550 [java highlighting] switch exhaustiveness: consider permitted classes inside pattern check

GitOrigin-RevId: 14c74921fb72a7915ad0c017cc1cafbab1ee5534

java/java-analysis-impl/src/com/intellij/codeInsight/daemon/impl/analysis/SwitchBlockHighlightingModel.java

@@ -31,6 +31,7 @@ import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
 import java.util.*;
+import java.util.function.Consumer;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
@@ -391,11 +392,27 @@ public class SwitchBlockHighlightingModel {
       PsiClass patternClass = PsiUtil.resolveClassInClassTypeOnly(JavaPsiPatternUtil.getPatternType(element));
       if (patternClass != null) {
         patternClasses.put(patternClass, patternLabelElement);
+        visitAllPermittedClasses(patternClass, permittedClass -> patternClasses.put(permittedClass, patternLabelElement));
       }
     }
     return patternClasses;
   }
 
+  private static void visitAllPermittedClasses(@NotNull PsiClass psiClass, Consumer<PsiClass> consumer){
+    Set<PsiClass> visitedClasses = new HashSet<>();
+    Queue<PsiClass> notVisitedClasses = new LinkedList<>();
+    notVisitedClasses.add(psiClass);
+    while (!notVisitedClasses.isEmpty()) {
+      PsiClass notVisitedClass = notVisitedClasses.poll();
+      if (!notVisitedClass.hasModifierProperty(SEALED) || visitedClasses.contains(notVisitedClass)) continue;
+      visitedClasses.add(notVisitedClass);
+      for (PsiClass permittedClass : PatternsInSwitchBlockHighlightingModel.getPermittedClasses(psiClass)) {
+        consumer.accept(permittedClass);
+        notVisitedClasses.add(permittedClass);
+      }
+    }
+  }
+
   private static @NotNull Set<PsiClass> findMissedClasses(@NotNull PsiType selectorType, Map<PsiClass, PsiPattern> patternClasses) {
     PsiClass selectorClass = PsiUtil.resolveClassInClassTypeOnly(selectorType);
     if (selectorClass instanceof PsiTypeParameter) {

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/advHighlightingPatternsInSwitch/SwitchExhaustivenessWithSealedIntersection.java

@@ -0,0 +1,20 @@
+sealed interface First {}
+sealed interface Second {}
+sealed interface Third extends First {}
+
+final class F1 implements First {}
+final class F2 implements First, Second {}
+final class F3 implements Second {}
+final class F4 implements Third, Second {} //implicit First
+
+class Test {
+  void test(First o) {
+    switch (o) {
+      case F1 x -> System.out.println();
+      case Second x -> System.out.println();
+    }
+    switch (<error descr="'switch' statement does not cover all possible input values">o</error>) {
+      case Second x -> System.out.println();
+    }
+  }
+}

java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/LightPatternsForSwitchHighlightingTest.java

@@ -53,6 +53,10 @@ public class LightPatternsForSwitchHighlightingTest extends LightJavaCodeInsight
     IdeaTestUtil.withLevel(getModule(), LanguageLevel.JDK_19_PREVIEW, this::doTest);
   }
 
+  public void testSwitchExhaustivenessWithSealedIntersection(){
+    IdeaTestUtil.withLevel(getModule(), LanguageLevel.JDK_19_PREVIEW, this::doTest);
+  }
+
   public void testSwitchExhaustivenessWithGenericsIn19Java() {
     IdeaTestUtil.withLevel(getModule(), LanguageLevel.JDK_19_PREVIEW, this::doTest);
   }