/*
* Copyright 2000-2015 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.intellij.dvcs.push.ui;
import com.intellij.dvcs.push.*;
import com.intellij.dvcs.push.PrePushHandler;
import com.intellij.dvcs.repo.Repository;
import com.intellij.ide.actions.ShowSettingsUtilImpl;
import com.intellij.openapi.actionSystem.AnAction;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.fileEditor.FileDocumentManager;
import com.intellij.openapi.options.ShowSettingsUtil;
import com.intellij.openapi.progress.ProgressIndicator;
import com.intellij.openapi.progress.Task;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.DialogWrapper;
import com.intellij.openapi.ui.Messages;
import com.intellij.openapi.ui.OptionAction;
import com.intellij.openapi.ui.ValidationInfo;
import com.intellij.openapi.util.registry.Registry;
import com.intellij.ui.components.labels.ActionLink;
import com.intellij.util.ui.JBUI;
import com.intellij.util.ui.UIUtil;
import com.intellij.util.ui.components.BorderLayoutPanel;
import net.miginfocom.swing.MigLayout;
import org.jetbrains.annotations.CalledInAwt;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import javax.swing.*;
import java.awt.event.ActionEvent;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicReference;
public class VcsPushDialog extends DialogWrapper {
private static final String ID = "Vcs.Push.Dialog";
private final PushLog myListPanel;
private final PushController myController;
private final Map<PushSupport, VcsPushOptionsPanel> myAdditionalPanels;
private Action myPushAction;
@Nullable private ForcePushAction myForcePushAction;
public VcsPushDialog(@NotNull Project project,
@NotNull List<? extends Repository> selectedRepositories,
@Nullable Repository currentRepo) {
super(project, true, (Registry.is("ide.perProjectModality")) ? IdeModalityType.PROJECT : IdeModalityType.IDE);
myController = new PushController(project, this, selectedRepositories, currentRepo);
myAdditionalPanels = myController.createAdditionalPanels();
myListPanel = myController.getPushPanelLog();
init();
updateOkActions();
setOKButtonText("Push");
setOKButtonMnemonic('P');
setTitle("Push Commits");
}
@Override
protected JComponent createCenterPanel() {
JPanel optionsPanel = new JPanel(new MigLayout("ins 0 0, flowx"));
for (VcsPushOptionsPanel panel : myAdditionalPanels.values()) {
optionsPanel.add(panel);
}
optionsPanel.setBorder(JBUI.Borders.emptyTop(6));
BorderLayoutPanel panel = JBUI.Panels.simplePanel(optionsPanel);
if (!myController.isForcePushEnabled()) {
panel.addToTop(createForcePushInfoLabel());
}
return JBUI.Panels.simplePanel(0, 2)
.addToCenter(myListPanel)
.addToBottom(panel);
}
@NotNull
private JComponent createForcePushInfoLabel() {
JPanel text = new JPanel();
text.setLayout(new BoxLayout(text, BoxLayout.X_AXIS));
JLabel label = new JLabel("You can enable and configure Force Push in " + ShowSettingsUtil.getSettingsMenuName() + ".");
label.setEnabled(false);
label.setFont(JBUI.Fonts.smallFont());
text.add(label);
ActionLink here = new ActionLink("Configure", new AnAction() {
@Override
public void actionPerformed(AnActionEvent e) {
Project project = myController.getProject();
VcsPushDialog.this.doCancelAction(e.getInputEvent());
ShowSettingsUtilImpl.showSettingsDialog(project, "vcs.Git", "force push");
}
});
here.setFont(JBUI.Fonts.smallFont());
text.add(here);
return JBUI.Panels.simplePanel().addToRight(text).withBorder(JBUI.Borders.emptyBottom(4));
}
@Override
protected String getDimensionServiceKey() {
return ID;
}
@Nullable
@Override
protected ValidationInfo doValidate() {
updateOkActions();
return null;
}
@Override
protected boolean postponeValidation() {
return false;
}
@Override
protected void doOKAction() {
push(false);
}
@Override
@NotNull
protected Action[] createActions() {
final List<Action> actions = new ArrayList<>();
myForcePushAction = new ForcePushAction();
myForcePushAction.setEnabled(canForcePush());
myForcePushAction.putValue(Action.NAME, "&Force Push");
if (myController.isForcePushEnabled()) {
myPushAction = new ComplexPushAction(myForcePushAction);
} else {
myPushAction = new OkAction() {};
myPushAction.putValue(Action.NAME, "&Push");
}
myPushAction.putValue(DEFAULT_ACTION, Boolean.TRUE);
actions.add(myPushAction);
actions.add(getCancelAction());
actions.add(getHelpAction());
return actions.toArray(new Action[actions.size()]);
}
private boolean canPush() {
return myController.isPushAllowed(false);
}
private boolean canForcePush() {
return myController.isForcePushEnabled() && myController.getProhibitedTarget() == null && myController.isPushAllowed(true);
}
@Nullable
@Override
public JComponent getPreferredFocusedComponent() {
return myListPanel.getPreferredFocusedComponent();
}
@NotNull
@Override
protected Action getOKAction() {
return myPushAction;
}
@Override
protected String getHelpId() {
return ID;
}
@CalledInAwt
private void push(boolean forcePush) {
FileDocumentManager.getInstance().saveAllDocuments();
AtomicReference<PrePushHandler.Result> result = new AtomicReference<>(PrePushHandler.Result.OK);
new Task.Modal(myController.getProject(), "Checking Commits...", true) {
@Override
public void run(@NotNull ProgressIndicator indicator) {
result.set(myController.executeHandlers(indicator));
}
@Override
public void onSuccess() {
super.onSuccess();
if (result.get() == PrePushHandler.Result.OK) {
doPush();
}
else if (result.get() == PrePushHandler.Result.ABORT_AND_CLOSE) {
doCancelAction();
}
else if (result.get() == PrePushHandler.Result.ABORT) {
// cancel push and leave the push dialog open
}
}
private void doPush() {
myController.push(forcePush);
close(OK_EXIT_CODE);
}
@Override
public void onThrowable(@NotNull Throwable error) {
if (error instanceof PushController.HandlerException) {
super.onThrowable(error.getCause());
String handlerName = ((PushController.HandlerException)error).getHandlerName();
suggestToSkipOrPush(handlerName + " has failed. See log for more details.\n" +
"Would you like to skip pre-push checking and continue or cancel push completely?");
} else {
super.onThrowable(error);
}
}
@Override
public void onCancel() {
super.onCancel();
suggestToSkipOrPush("Would you like to skip pre-push checking and continue or cancel push completely?");
}
private void suggestToSkipOrPush(@NotNull String message) {
if (Messages.showOkCancelDialog(myProject,
message,
"Push",
"&Push Anyway",
"&Cancel",
UIUtil.getWarningIcon()) == Messages.OK) {
doPush();
}
}
}.queue();
}
public void updateOkActions() {
myPushAction.setEnabled(canPush());
if (myForcePushAction != null) {
boolean canForcePush = canForcePush();
myForcePushAction.setEnabled(canForcePush);
String tooltip = null;
if (!canForcePush) {
PushTarget target = myController.getProhibitedTarget();
tooltip = myController.isForcePushEnabled() && target != null
? "Force push to <b>" + target.getPresentation() + "</b> is prohibited"
: "<b>Force Push</b> can be enabled in the Settings";
}
myForcePushAction.putValue(Action.SHORT_DESCRIPTION, tooltip);
}
}
public void disableOkActions() {
myPushAction.setEnabled(false);
}
@Nullable
public VcsPushOptionValue getAdditionalOptionValue(@NotNull PushSupport support) {
VcsPushOptionsPanel panel = myAdditionalPanels.get(support);
return panel == null ? null : panel.getValue();
}
private class ForcePushAction extends AbstractAction {
ForcePushAction() {
super("&Force Push");
}
@Override
public void actionPerformed(ActionEvent e) {
if (myController.ensureForcePushIsNeeded()) {
push(true);
}
}
}
private class ComplexPushAction extends AbstractAction implements OptionAction {
private final Action[] myOptions;
private ComplexPushAction(Action additionalAction) {
super("&Push");
myOptions = new Action[]{additionalAction};
}
@Override
public void actionPerformed(ActionEvent e) {
push(false);
}
@Override
public void setEnabled(boolean isEnabled) {
super.setEnabled(isEnabled);
for (Action optionAction : myOptions) {
optionAction.setEnabled(isEnabled);
}
}
@NotNull
@Override
public Action[] getOptions() {
return myOptions;
}
}
}