1 files changed, 94 insertions, 0 deletions
diff --git a/platform/platform-impl/src/com/intellij/openapi/diff/actions/DiffPanelComboBoxAction.java b/platform/platform-impl/src/com/intellij/openapi/diff/actions/DiffPanelComboBoxAction.java
new file mode 100644
index 000000000000..b3ef9595782f
--- /dev/null
+++ b/platform/platform-impl/src/com/intellij/openapi/diff/actions/DiffPanelComboBoxAction.java
@@ -0,0 +1,94 @@
+package com.intellij.openapi.diff.actions;
+
+import com.intellij.openapi.actionSystem.*;
+import com.intellij.openapi.actionSystem.ex.ComboBoxAction;
+import com.intellij.openapi.diff.DiffBundle;
+import com.intellij.openapi.diff.ex.DiffPanelEx;
+import com.intellij.openapi.diff.impl.DiffPanelImpl;
+import com.intellij.openapi.project.DumbAware;
+import com.intellij.openapi.project.DumbAwareAction;
+import com.intellij.util.containers.HashMap;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import javax.swing.*;
+import java.awt.*;
+import java.util.Map;
+
+public abstract class DiffPanelComboBoxAction<T> extends ComboBoxAction implements DumbAware {
+  @NotNull private final Map<T, AnAction> myActions = new HashMap<T, AnAction>();
+  @NotNull private final T[] myActionOrder;
+
+  protected DiffPanelComboBoxAction(@NotNull T[] actionOrder) {
+    myActionOrder = actionOrder;
+  }
+
+  @NotNull
+  protected abstract String getActionName();
+
+  @NotNull
+  protected abstract T getCurrentOption(@NotNull DiffPanelEx diffPanel);
+
+  @Nullable
+  private static DiffPanelEx getDiffPanel(@NotNull DataContext context) {
+    return DiffPanelImpl.fromDataContext(context);
+  }
+
+  protected void addAction(T key, @NotNull AnAction action) {
+    myActions.put(key, action);
+  }
+
+  @Override
+  public JComponent createCustomComponent(final Presentation presentation) {
+    JPanel panel = new JPanel(new BorderLayout());
+    final JLabel label = new JLabel(getActionName());
+    label.setBorder(BorderFactory.createEmptyBorder(0, 4, 0, 4));
+    panel.add(label, BorderLayout.WEST);
+    panel.add(super.createCustomComponent(presentation), BorderLayout.CENTER);
+    return panel;
+  }
+
+  @NotNull
+  @Override
+  protected DefaultActionGroup createPopupActionGroup(JComponent button) {
+    DefaultActionGroup actionGroup = new DefaultActionGroup();
+    for (T option : myActionOrder) {
+      actionGroup.add(myActions.get(option));
+    }
+    return actionGroup;
+  }
+
+  @Override
+  public void update(AnActionEvent e) {
+    super.update(e);
+    Presentation presentation = e.getPresentation();
+    DiffPanelEx diffPanel = getDiffPanel(e.getDataContext());
+    if (diffPanel != null && diffPanel.getComponent().isDisplayable()) {
+      AnAction action = myActions.get(getCurrentOption(diffPanel));
+      Presentation templatePresentation = action.getTemplatePresentation();
+      presentation.setIcon(templatePresentation.getIcon());
+      presentation.setText(templatePresentation.getText());
+      presentation.setEnabled(true);
+    }
+    else {
+      presentation.setIcon(null);
+      presentation.setText(DiffBundle.message("diff.panel.combo.box.action.not.available.action.name"));
+      presentation.setEnabled(false);
+    }
+  }
+
+  protected static abstract class DiffPanelAction extends DumbAwareAction {
+    public DiffPanelAction(@NotNull String text) {
+      super(text);
+    }
+
+    public void actionPerformed(AnActionEvent e) {
+      final DiffPanelEx diffPanel = getDiffPanel(e.getDataContext());
+      if (diffPanel != null) {
+        perform(diffPanel);
+      }
+    }
+
+    protected abstract void perform(@NotNull DiffPanelEx diffPanel);
+  }
+}