IDEA-78031 diamonds with new qualifiers

java/java-psi-impl/src/com/intellij/psi/PsiDiamondTypeImpl.java

@@ -169,9 +169,32 @@ public class PsiDiamondTypeImpl extends PsiDiamondType {
         final Project project = newExpression.getProject();
         final JavaPsiFacade facade = JavaPsiFacade.getInstance(project);
         final PsiResolveHelper resolveHelper = facade.getResolveHelper();
+        final PsiExpression newExpressionQualifier = newExpression.getQualifier();
         final PsiElement qualifierElement = classReference.getQualifier();
         final String qualifier = qualifierElement != null ? qualifierElement.getText() : "";
-        return resolveHelper.resolveReferencedClass(StringUtil.getQualifiedName(qualifier, text), newExpression);
+        final String qualifiedName = StringUtil.getQualifiedName(qualifier, text);
+        if (newExpressionQualifier != null) {
+          final String newExpressionQualifierText = newExpressionQualifier.getText();
+          if (newExpressionQualifier instanceof PsiReferenceExpression) {
+            final PsiVariable variable = resolveHelper.resolveReferencedVariable(newExpressionQualifierText, newExpression);
+            if (variable != null) {
+              final PsiClass aClass = PsiUtil.resolveClassInType(variable.getType());
+              if (aClass != null) {
+                return aClass.findInnerClassByName(qualifiedName, false);
+              }
+            }
+          } else if (newExpressionQualifier instanceof PsiCallExpression) {
+            final JavaResolveResult resolveResult = ((PsiCallExpression)newExpressionQualifier).resolveMethodGenerics();
+            final PsiElement method = resolveResult.getElement();
+            if (method instanceof PsiMethod) {
+              final PsiClass aClass = PsiUtil.resolveClassInType(resolveResult.getSubstitutor().substitute(((PsiMethod)method).getReturnType()));
+              if (aClass != null) {
+                return aClass.findInnerClassByName(qualifiedName, false);
+              }
+            }
+          }
+        }
+        return resolveHelper.resolveReferencedClass(qualifiedName, newExpression);
       } else {
         return null;
       }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/advHighlighting7/DiamondMisc.java

@@ -181,3 +181,26 @@ class PredefinedErrorsOverRaw<T> {
     PredefinedErrorsOverRaw mc = new <Boolean>PredefinedErrorsOverRaw<<error descr="Cannot use diamonds with explicit type parameters for constructor"></error>>("");
   }
 }
+
+class Outer {
+    class Inner<T> {}
+}
+class Outer1 {}
+
+class Another {
+    public static void main(String[] args) {
+        Outer o = new Outer();
+        Outer.Inner<String> i = o.new Inner<>();
+        Outer.Inner<String> i1 = m().new Inner<>();
+        Outer.Inner<String> i2 = m1().new Inner<>();
+
+        Outer.Inner<String> i3 = m2().new <error descr="Cannot resolve symbol 'Inner'">Inner</error><>();
+
+        System.out.println(i);
+    }
+
+  static Outer m() {return null;}
+  static <T extends Outer> T m1() {return null;}
+  static <T> T m2() {return null;}
+  
+}