From 1f1b74637bfe19ef1e04736dce8cadf74f2217e1 Mon Sep 17 00:00:00 2001
From: Alexander Lobas <Alexander.Lobas@jetbrains.com>
Date: Thu, 21 Feb 2013 14:54:09 +0400
Subject: [PATCH] Collapse property recursively

---
 .../designer/propertyTable/PropertyTable.java | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/platform/platform-impl/src/com/intellij/designer/propertyTable/PropertyTable.java b/platform/platform-impl/src/com/intellij/designer/propertyTable/PropertyTable.java
index af22e699c91b..aaa12bc3c226 100644
--- a/platform/platform-impl/src/com/intellij/designer/propertyTable/PropertyTable.java
+++ b/platform/platform-impl/src/com/intellij/designer/propertyTable/PropertyTable.java
@@ -653,11 +653,8 @@ public abstract class PropertyTable extends JBTable {
     int selectedRow = getSelectedRow();
     Property property = myProperties.get(rowIndex);
 
-    LOG.assertTrue(myExpandedProperties.remove(property.getPath()));
-    int size = getFilterChildren(property).size();
-    for (int i = 0; i < size; i++) {
-      myProperties.remove(rowIndex + 1);
-    }
+    int size = collapse(property, rowIndex + 1);
+    LOG.assertTrue(size > 0);
     myModel.fireTableDataChanged();
 
     if (selectedRow != -1) {
@@ -669,6 +666,18 @@ public abstract class PropertyTable extends JBTable {
     }
   }
 
+  private int collapse(Property property, int startIndex) {
+    int totalSize = 0;
+    if (myExpandedProperties.remove(property.getPath())) {
+      int size = getFilterChildren(property).size();
+      totalSize += size;
+      for (int i = 0; i < size; i++) {
+        totalSize += collapse(myProperties.remove(startIndex), startIndex);
+      }
+    }
+    return totalSize;
+  }
+
   private void expand(int rowIndex) {
     int selectedRow = getSelectedRow();
     Property property = myProperties.get(rowIndex);