diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetDeclarationWithBody.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetDeclarationWithBody.java index 9db554c7066..992e67b9f0f 100644 --- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetDeclarationWithBody.java +++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetDeclarationWithBody.java @@ -16,16 +16,19 @@ package org.jetbrains.jet.lang.psi; +import com.intellij.psi.PsiElement; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; import java.util.List; public interface JetDeclarationWithBody extends JetDeclaration { - @Nullable JetExpression getBodyExpression(); + @Nullable + PsiElement getEqualsToken(); + @Override @Nullable String getName(); diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetFunctionLiteral.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetFunctionLiteral.java index 48c83dfdea7..99f7b1302aa 100644 --- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetFunctionLiteral.java +++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetFunctionLiteral.java @@ -52,6 +52,12 @@ public class JetFunctionLiteral extends JetFunctionNotStubbed { return (JetBlockExpression) super.getBodyExpression(); } + @Nullable + @Override + public PsiElement getEqualsToken() { + return null; + } + @NotNull public PsiElement getLBrace() { return findChildByType(JetTokens.LBRACE); diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetNamedFunction.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetNamedFunction.java index 6ed817abfde..a3ac27908fb 100644 --- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetNamedFunction.java +++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetNamedFunction.java @@ -77,6 +77,7 @@ public class JetNamedFunction extends JetTypeParameterListOwnerStub