From b049455aa4901d725c437aa37b052a038ea372bc Mon Sep 17 00:00:00 2001
From: Alexey Sedunov <alexey.sedunov@jetbrains.com>
Date: Mon, 8 Jun 2015 18:04:15 +0300
Subject: [PATCH] Presentation: Support constructors in JetFunctionPresenter

---
 .../org/jetbrains/kotlin/psi/JetConstructor.java  |  7 +++++++
 idea/src/META-INF/plugin.xml                      |  2 +-
 .../idea/presentation/JetFunctionPresenter.java   | 15 ++++++++++-----
 3 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/compiler/frontend/src/org/jetbrains/kotlin/psi/JetConstructor.java b/compiler/frontend/src/org/jetbrains/kotlin/psi/JetConstructor.java
index e1ae9bd2e83..49640f433e4 100644
--- a/compiler/frontend/src/org/jetbrains/kotlin/psi/JetConstructor.java
+++ b/compiler/frontend/src/org/jetbrains/kotlin/psi/JetConstructor.java
@@ -17,6 +17,8 @@
 package org.jetbrains.kotlin.psi;
 
 import com.intellij.lang.ASTNode;
+import com.intellij.navigation.ItemPresentation;
+import com.intellij.navigation.ItemPresentationProviders;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.stubs.IStubElementType;
 import com.intellij.util.IncorrectOperationException;
@@ -171,6 +173,11 @@ public abstract class JetConstructor<T extends JetConstructor<T>> extends JetDec
         throw new IncorrectOperationException("setName to constructor");
     }
 
+    @Override
+    public ItemPresentation getPresentation() {
+        return ItemPresentationProviders.getItemPresentation(this);
+    }
+
     @Nullable
     public PsiElement getConstructorKeyword() {
         return findChildByType(JetTokens.CONSTRUCTOR_KEYWORD);
diff --git a/idea/src/META-INF/plugin.xml b/idea/src/META-INF/plugin.xml
index 51f82728bbd..2139f428048 100644
--- a/idea/src/META-INF/plugin.xml
+++ b/idea/src/META-INF/plugin.xml
@@ -393,7 +393,7 @@
                                     implementationClass="org.jetbrains.kotlin.idea.ktSignature.KotlinSignatureInJavaMarkerProvider"/>
     <iconProvider implementation="org.jetbrains.kotlin.idea.JetIconProvider"/>
     <itemPresentationProvider implementationClass="org.jetbrains.kotlin.idea.presentation.JetFunctionPresenter"
-                              forClass="org.jetbrains.kotlin.psi.JetNamedFunction"/>
+                              forClass="org.jetbrains.kotlin.psi.JetFunction"/>
     <itemPresentationProvider implementationClass="org.jetbrains.kotlin.idea.presentation.JetClassPresenter"
                               forClass="org.jetbrains.kotlin.psi.JetClass"/>
     <itemPresentationProvider implementationClass="org.jetbrains.kotlin.idea.presentation.JetObjectPresenter"
diff --git a/idea/src/org/jetbrains/kotlin/idea/presentation/JetFunctionPresenter.java b/idea/src/org/jetbrains/kotlin/idea/presentation/JetFunctionPresenter.java
index 7d660a4604f..3f82ef302c7 100644
--- a/idea/src/org/jetbrains/kotlin/idea/presentation/JetFunctionPresenter.java
+++ b/idea/src/org/jetbrains/kotlin/idea/presentation/JetFunctionPresenter.java
@@ -23,15 +23,15 @@ import com.intellij.navigation.ItemPresentationProvider;
 import org.apache.commons.lang.StringUtils;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.kotlin.name.FqName;
-import org.jetbrains.kotlin.psi.JetNamedFunction;
-import org.jetbrains.kotlin.psi.JetParameter;
-import org.jetbrains.kotlin.psi.JetTypeReference;
+import org.jetbrains.kotlin.psi.*;
 
 import java.util.Collection;
 
-public class JetFunctionPresenter implements ItemPresentationProvider<JetNamedFunction> {
+public class JetFunctionPresenter implements ItemPresentationProvider<JetFunction> {
     @Override
-    public ItemPresentation getPresentation(@NotNull final JetNamedFunction function) {
+    public ItemPresentation getPresentation(@NotNull final JetFunction function) {
+        if (function instanceof JetFunctionLiteral) return null;
+
         return new JetDefaultNamedDeclarationPresentation(function) {
             @Override
             public String getPresentableText() {
@@ -60,6 +60,11 @@ public class JetFunctionPresenter implements ItemPresentationProvider<JetNamedFu
 
             @Override
             public String getLocationString() {
+                if (function instanceof JetConstructor) {
+                    FqName name = ((JetConstructor) function).getClassOrObject().getFqName();
+                    return name != null ? String.format("(in %s)", name) : "";
+                }
+
                 FqName name = function.getFqName();
                 if (name != null) {
                     JetTypeReference receiverTypeRef = function.getReceiverTypeReference();