Icons for lighs classes

This commit is contained in:
Andrey Breslav
2012-12-25 18:41:44 +04:00
parent e3ba5bf9b7
commit 3a34ccb03e
@@ -28,6 +28,7 @@ import com.intellij.util.containers.ContainerUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.jet.asJava.JetLightClass;
import org.jetbrains.jet.asJava.KotlinLightClass;
import org.jetbrains.jet.asJava.KotlinLightClassForPackage;
import org.jetbrains.jet.lang.psi.*;
import org.jetbrains.jet.lexer.JetTokens;
@@ -102,6 +103,10 @@ public class JetIconProvider extends IconProvider {
return JetIcons.FILE;
}
if (psiElement instanceof KotlinLightClass) {
psiElement = psiElement.getNavigationElement();
}
if (psiElement instanceof JetNamedFunction) {
if (((JetFunction) psiElement).getReceiverTypeRef() != null) {
return JetIcons.EXTENSION_FUNCTION;