From 3a34ccb03ee03851885502e9e0ccb340832cb950 Mon Sep 17 00:00:00 2001
From: Andrey Breslav <andrey.breslav@jetbrains.com>
Date: Tue, 25 Dec 2012 18:41:44 +0400
Subject: [PATCH] Icons for lighs classes

---
 idea/src/org/jetbrains/jet/plugin/JetIconProvider.java | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/idea/src/org/jetbrains/jet/plugin/JetIconProvider.java b/idea/src/org/jetbrains/jet/plugin/JetIconProvider.java
index 46e7a8b6fc3..772a4a4bce8 100644
--- a/idea/src/org/jetbrains/jet/plugin/JetIconProvider.java
+++ b/idea/src/org/jetbrains/jet/plugin/JetIconProvider.java
@@ -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;