From 9aee8a98e0c19b5ae037b34b57be91e0501e14a0 Mon Sep 17 00:00:00 2001
From: Evgeny Gerashchenko <Evgeny.Gerashchenko@jetbrains.com>
Date: Fri, 4 May 2012 21:38:39 +0400
Subject: [PATCH] Added updating tree when Kotlin file changes "single-class"
 status.

---
 .../projectView/JetClassOrObjectTreeNode.java | 25 ++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/idea/src/org/jetbrains/jet/plugin/projectView/JetClassOrObjectTreeNode.java b/idea/src/org/jetbrains/jet/plugin/projectView/JetClassOrObjectTreeNode.java
index cb4ee8b2d4f..f0d90e4c552 100644
--- a/idea/src/org/jetbrains/jet/plugin/projectView/JetClassOrObjectTreeNode.java
+++ b/idea/src/org/jetbrains/jet/plugin/projectView/JetClassOrObjectTreeNode.java
@@ -17,12 +17,15 @@
 package org.jetbrains.jet.plugin.projectView;
 
 import com.intellij.ide.projectView.PresentationData;
+import com.intellij.ide.projectView.ProjectView;
 import com.intellij.ide.projectView.ViewSettings;
 import com.intellij.ide.projectView.impl.nodes.AbstractPsiBasedNode;
 import com.intellij.ide.util.treeView.AbstractTreeNode;
 import com.intellij.openapi.project.Project;
 import com.intellij.psi.PsiElement;
 import org.jetbrains.jet.lang.psi.JetClassOrObject;
+import org.jetbrains.jet.lang.psi.JetFile;
+import org.jetbrains.jet.plugin.JetIconProvider;
 
 import java.util.Collection;
 
@@ -48,11 +51,27 @@ public class JetClassOrObjectTreeNode extends AbstractPsiBasedNode<JetClassOrObj
         return getClassOrObjectChildren(getValue(), getProject(), getSettings());
     }
 
+    private void update(AbstractTreeNode node) {
+        ProjectView.getInstance(getProject()).getCurrentProjectViewPane().getTreeBuilder().addSubtreeToUpdateByElement(node);
+    }
+
     @Override
     protected void updateImpl(PresentationData data) {
-        JetClassOrObject value = getValue();
-        if (value != null) {
-            data.setPresentableText(value.getName());
+        JetClassOrObject classOrObject = getValue();
+        if (classOrObject != null) {
+            data.setPresentableText(classOrObject.getName());
+
+            AbstractTreeNode parent = getParent();
+            if (JetIconProvider.getMainClass((JetFile) classOrObject.getContainingFile()) != null) {
+                if (parent instanceof JetFileTreeNode) {
+                    update(parent.getParent());
+                }
+            }
+            else {
+                if (!(parent instanceof JetClassOrObjectTreeNode) && !(parent instanceof JetFileTreeNode)) {
+                    update(parent);
+                }
+            }
         }
     }