From f2cb5f36967887979e8d577036ae70762ed4ae44 Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Tue, 17 Sep 2013 12:40:46 +0400
Subject: [PATCH] new inference: initial lambda

---
 .../graphInference/InferenceSession.java      |  5 ++
 ...mbdaExpressionCompatibilityConstraint.java | 47 +++++++++++++++++--
 2 files changed, 48 insertions(+), 4 deletions(-)

diff --git a/java/java-psi-impl/src/com/intellij/psi/impl/source/resolve/graphInference/InferenceSession.java b/java/java-psi-impl/src/com/intellij/psi/impl/source/resolve/graphInference/InferenceSession.java
index 68f4aafd20f6..2a3cb272a117 100644
--- a/java/java-psi-impl/src/com/intellij/psi/impl/source/resolve/graphInference/InferenceSession.java
+++ b/java/java-psi-impl/src/com/intellij/psi/impl/source/resolve/graphInference/InferenceSession.java
@@ -240,6 +240,11 @@ public class InferenceSession {
       for (InferenceVariable inferenceVariable : variables) {
 
         if (inferenceVariable.getInstantiation() != null) continue;
+        final List<PsiType> eqBounds = inferenceVariable.getBounds(InferenceBound.EQ);
+        if (!eqBounds.isEmpty()) {
+          inferenceVariable.setInstantiation(eqBounds.get(0));
+          continue;
+        }
 
         final List<PsiType> lowerBounds = inferenceVariable.getBounds(InferenceBound.LOWER);
         PsiType lub = null;
diff --git a/java/java-psi-impl/src/com/intellij/psi/impl/source/resolve/graphInference/constraints/LambdaExpressionCompatibilityConstraint.java b/java/java-psi-impl/src/com/intellij/psi/impl/source/resolve/graphInference/constraints/LambdaExpressionCompatibilityConstraint.java
index 67e07f1fec14..9d9dad46ffc8 100644
--- a/java/java-psi-impl/src/com/intellij/psi/impl/source/resolve/graphInference/constraints/LambdaExpressionCompatibilityConstraint.java
+++ b/java/java-psi-impl/src/com/intellij/psi/impl/source/resolve/graphInference/constraints/LambdaExpressionCompatibilityConstraint.java
@@ -1,10 +1,9 @@
 package com.intellij.psi.impl.source.resolve.graphInference.constraints;
 
-import com.intellij.psi.LambdaHighlightingUtil;
-import com.intellij.psi.PsiLambdaExpression;
-import com.intellij.psi.PsiType;
+import com.intellij.psi.*;
 import com.intellij.psi.impl.source.resolve.graphInference.InferenceSession;
 import com.intellij.psi.impl.source.resolve.graphInference.InferenceVariable;
+import com.intellij.psi.util.PsiUtil;
 
 import java.util.List;
 
@@ -30,7 +29,47 @@ public class LambdaExpressionCompatibilityConstraint implements ConstraintFormul
     if (LambdaHighlightingUtil.checkInterfaceFunctional(myT) != null) {
       return false;
     }
-    
+
+    if (myExpression.hasFormalParameterTypes()) {
+    }
+    final PsiMethod interfaceMethod = LambdaUtil.getFunctionalInterfaceMethod(myT);
+    if (interfaceMethod == null) {
+      return false;
+    }
+    final PsiSubstitutor substitutor = LambdaUtil.getSubstitutor(interfaceMethod, PsiUtil.resolveGenericsClassInType(myT));
+    final PsiParameter[] parameters = interfaceMethod.getParameterList().getParameters();
+    for (PsiParameter parameter : parameters) {
+      if (!session.isProperType(parameter.getType())) {
+        delayedConstraints.add(this);
+        return true;
+      }
+    }
+
+    final PsiParameter[] lambdaParameters = myExpression.getParameterList().getParameters();
+    if (lambdaParameters.length != parameters.length) {
+      return false;
+    }
+    if (myExpression.hasFormalParameterTypes()) {
+      for (int i = 0; i < lambdaParameters.length; i++) {
+        constraints.add(new TypeEqualityConstraint(lambdaParameters[i].getType(), substitutor.substitute(parameters[i].getType())));
+      }
+    }
+
+    final PsiType returnType = interfaceMethod.getReturnType();
+    if (returnType != null) {
+      if (returnType.equals(PsiType.VOID)) {
+        if (!myExpression.isVoidCompatible() && !(myExpression.getBody() instanceof PsiExpression)) {
+          return false;
+        }
+      } else {
+        if (myExpression.isVoidCompatible()) {  //not value-compatible
+          return false;
+        }
+        for (PsiExpression returnExpressions : LambdaUtil.getReturnExpressions(myExpression)) {
+          constraints.add(new ExpressionCompatibilityConstraint(returnExpressions, returnType));
+        }
+      }
+    }
     return true;
   }
 }