From 0110cdfde9fc1a09df876aca3d0f63ab81b3620b Mon Sep 17 00:00:00 2001
From: "Svetlana.Zemlyanskaya" <Svetlana.Zemlyanskaya@jetbrains.com>
Date: Thu, 24 Mar 2016 17:46:29 +0100
Subject: [PATCH] Arrangement: add action should not be available outside of
 arrangement panel

---
 .../arrangement/action/AddArrangementRuleAction.java        | 6 ++++++
 .../arrangement/action/AddArrangementSectionRuleAction.java | 1 +
 2 files changed, 7 insertions(+)

diff --git a/platform/lang-impl/src/com/intellij/application/options/codeStyle/arrangement/action/AddArrangementRuleAction.java b/platform/lang-impl/src/com/intellij/application/options/codeStyle/arrangement/action/AddArrangementRuleAction.java
index 72bf323bd99f..103f666e4222 100644
--- a/platform/lang-impl/src/com/intellij/application/options/codeStyle/arrangement/action/AddArrangementRuleAction.java
+++ b/platform/lang-impl/src/com/intellij/application/options/codeStyle/arrangement/action/AddArrangementRuleAction.java
@@ -38,6 +38,12 @@ public class AddArrangementRuleAction extends AbstractArrangementRuleAction impl
     setEnabledInModalContext(true);
   }
 
+  @Override
+  public void update(AnActionEvent e) {
+    ArrangementMatchingRulesControl control = getRulesControl(e);
+    e.getPresentation().setEnabled(control != null);
+  }
+
   @Override
   public void actionPerformed(AnActionEvent e) {
     ArrangementMatchingRulesControl control = getRulesControl(e);
diff --git a/platform/lang-impl/src/com/intellij/application/options/codeStyle/arrangement/action/AddArrangementSectionRuleAction.java b/platform/lang-impl/src/com/intellij/application/options/codeStyle/arrangement/action/AddArrangementSectionRuleAction.java
index 17acf56b36b1..667ac7d7cd4b 100644
--- a/platform/lang-impl/src/com/intellij/application/options/codeStyle/arrangement/action/AddArrangementSectionRuleAction.java
+++ b/platform/lang-impl/src/com/intellij/application/options/codeStyle/arrangement/action/AddArrangementSectionRuleAction.java
@@ -39,6 +39,7 @@ public class AddArrangementSectionRuleAction extends AddArrangementRuleAction {
   public void update(AnActionEvent e) {
     final ArrangementMatchingRulesControl control = getRulesControl(e);
     if (control == null || !(control instanceof ArrangementSectionRulesControl)) {
+      e.getPresentation().setEnabledAndVisible(false);
       return;
     }
     e.getPresentation().setEnabledAndVisible(((ArrangementSectionRulesControl)control).getSectionRuleManager() != null);