From 67291898d6fa8b0ee27a235f018bdf22a1f0566d Mon Sep 17 00:00:00 2001
From: Alexander Lobas <Alexander.Lobas@jetbrains.com>
Date: Thu, 21 Nov 2013 20:42:24 +0400
Subject: [PATCH] SelectAll action

---
 .../designer/actions/DesignerActionPanel.java  |  2 +-
 .../designer/componentTree/ComponentTree.java  | 18 +++++++++++++++++-
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/plugins/ui-designer-core/src/com/intellij/designer/actions/DesignerActionPanel.java b/plugins/ui-designer-core/src/com/intellij/designer/actions/DesignerActionPanel.java
index 39cd4a6ed12d..526c2b6d39a4 100644
--- a/plugins/ui-designer-core/src/com/intellij/designer/actions/DesignerActionPanel.java
+++ b/plugins/ui-designer-core/src/com/intellij/designer/actions/DesignerActionPanel.java
@@ -129,7 +129,7 @@ public class DesignerActionPanel implements DataProvider {
     return group;
   }
 
-  protected SelectAllAction createSelectAllAction(EditableArea area) {
+  public AnAction createSelectAllAction(EditableArea area) {
     return new SelectAllAction(area);
   }
 
diff --git a/plugins/ui-designer-core/src/com/intellij/designer/componentTree/ComponentTree.java b/plugins/ui-designer-core/src/com/intellij/designer/componentTree/ComponentTree.java
index f7735c52eb0e..c128c8110497 100644
--- a/plugins/ui-designer-core/src/com/intellij/designer/componentTree/ComponentTree.java
+++ b/plugins/ui-designer-core/src/com/intellij/designer/componentTree/ComponentTree.java
@@ -39,10 +39,12 @@ import org.jetbrains.annotations.Nullable;
 
 import javax.swing.*;
 import javax.swing.border.LineBorder;
+import javax.swing.plaf.TreeUI;
 import javax.swing.tree.DefaultMutableTreeNode;
 import javax.swing.tree.DefaultTreeModel;
 import javax.swing.tree.TreePath;
 import java.awt.*;
+import java.awt.event.ActionEvent;
 import java.awt.event.MouseEvent;
 import java.util.List;
 
@@ -75,6 +77,19 @@ public final class ComponentTree extends Tree implements DataProvider {
     myInplaceEditingAction = DesignerActionPanel.createInplaceEditingAction(this);
   }
 
+  @Override
+  public void setUI(TreeUI ui) {
+    super.setUI(ui);
+    getActionMap().put("selectAll", new AbstractAction() {
+      @Override
+      public void actionPerformed(ActionEvent e) {
+        if (myDesigner != null) {
+          myDesigner.getActionPanel().createSelectAllAction(myDesigner.getSurfaceArea()).actionPerformed(null);
+        }
+      }
+    });
+  }
+
   public void newModel() {
     setModel(new DefaultTreeModel(new DefaultMutableTreeNode()));
   }
@@ -173,7 +188,8 @@ public final class ComponentTree extends Tree implements DataProvider {
 
     if (level != null) {
       TextAttributesKey attributesKey =
-        SeverityRegistrar.getSeverityRegistrar(myDesigner.getProject()).getHighlightInfoTypeBySeverity(level.getSeverity()).getAttributesKey();
+        SeverityRegistrar.getSeverityRegistrar(myDesigner.getProject()).getHighlightInfoTypeBySeverity(level.getSeverity())
+          .getAttributesKey();
       final TextAttributes textAttributes = EditorColorsManager.getInstance().getGlobalScheme().getAttributes(attributesKey);
 
       wrapper = new AttributeWrapper() {