/*******************************************************************************
* Copyright (C) 2011, Mathias Kinzler <mathias.kinzler@sap.com>
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*******************************************************************************/
package org.eclipse.egit.ui.internal.history.command;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.List;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.egit.core.op.DeleteBranchOperation;
import org.eclipse.egit.ui.Activator;
import org.eclipse.egit.ui.internal.UIText;
import org.eclipse.egit.ui.internal.dialogs.BranchSelectionDialog;
import org.eclipse.egit.ui.internal.dialogs.UnmergedBranchDialog;
import org.eclipse.egit.ui.internal.history.GitHistoryPage;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.dialogs.ProgressMonitorDialog;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.window.Window;
import org.eclipse.jgit.lib.Ref;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.Shell;
/**
* Delete a branch pointing to a commit.
*/
public class DeleteBranchOnCommitHandler extends AbstractHistoryCommandHandler {
public Object execute(ExecutionEvent event) throws ExecutionException {
final Repository repository = getRepository(event);
if (repository == null)
return null;
IStructuredSelection selection = getSelection(event);
int totalBranchCount;
List<Ref> branchesOfCommit;
try {
totalBranchCount = getBranchesOfCommit(selection, repository, false)
.size();
branchesOfCommit = getBranchesOfCommit(selection, repository, true);
} catch (IOException e) {
throw new ExecutionException("Could not obtain current Branch", e); //$NON-NLS-1$
}
// this should have been checked by isEnabled()
if (branchesOfCommit.isEmpty())
return null;
final List<Ref> unmergedBranches = new ArrayList<Ref>();
final Shell shell = getPart(event).getSite().getShell();
final List<Ref> branchesToDelete;
// we will show the dialog if there are either multiple branches that might be
// deleted or if one of the branches is the current head (which we can't delete);
// in the latter case, we may show the dialog even if there is only one branch to
// delete instead of quietly deleting an unexpected one, for example a remote
// tracking branch
if (totalBranchCount > 1) {
BranchSelectionDialog<Ref> dlg = new BranchSelectionDialog<Ref>(
shell,
branchesOfCommit,
UIText.DeleteBranchOnCommitHandler_SelectBranchDialogTitle,
UIText.DeleteBranchOnCommitHandler_SelectBranchDialogMessage,
SWT.MULTI);
if (dlg.open() != Window.OK)
return null;
branchesToDelete = dlg.getSelectedNodes();
} else
branchesToDelete = branchesOfCommit;
try {
new ProgressMonitorDialog(shell).run(false, false,
new IRunnableWithProgress() {
public void run(IProgressMonitor monitor)
throws InvocationTargetException,
InterruptedException {
try {
monitor.beginTask(
UIText.DeleteBranchCommand_DeletingBranchesProgress,
branchesToDelete.size());
for (Ref refNode : branchesToDelete) {
int result = deleteBranch(repository,
refNode, false);
if (result == DeleteBranchOperation.REJECTED_CURRENT) {
throw new CoreException(
Activator
.createErrorStatus(
UIText.DeleteBranchCommand_CannotDeleteCheckedOutBranch,
null));
} else if (result == DeleteBranchOperation.REJECTED_UNMERGED) {
unmergedBranches.add(refNode);
} else
monitor.worked(1);
}
if (!unmergedBranches.isEmpty()) {
MessageDialog messageDialog = new UnmergedBranchDialog<Ref>(
shell, unmergedBranches);
if (messageDialog.open() == Window.OK) {
for (Ref node : unmergedBranches) {
deleteBranch(repository, node, true);
monitor.worked(1);
}
}
}
} catch (CoreException ex) {
throw new InvocationTargetException(ex);
} finally {
monitor.done();
}
}
});
} catch (InvocationTargetException e1) {
Activator.handleError(
UIText.RepositoriesView_BranchDeletionFailureMessage,
e1.getCause(), true);
} catch (InterruptedException e1) {
// ignore
}
return null;
}
private int deleteBranch(Repository repo, final Ref ref, boolean force)
throws CoreException {
DeleteBranchOperation dbop = new DeleteBranchOperation(repo, ref, force);
dbop.execute(null);
return dbop.getStatus();
}
@Override
public boolean isEnabled() {
GitHistoryPage page = getPage();
Repository repository = getRepository(page);
if (repository == null)
return false;
List<Ref> branchesOfCommit;
try {
branchesOfCommit = getBranchesOfCommit(getSelection(page),
repository, true);
} catch (IOException e) {
Activator.logError("Could not calculate Enablement", e); //$NON-NLS-1$
return false;
}
return !branchesOfCommit.isEmpty();
}
}