/******************************************************************************* * 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 { @Override 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( UIText.AbstractHistoryCommitHandler_cantGetBranches, e); } // this should have been checked by isEnabled() if (branchesOfCommit.isEmpty()) return null; final List<Ref> unmergedBranches = new ArrayList<>(); 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<>( 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(true, false, new IRunnableWithProgress() { @Override public void run(final 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); } } 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 } if (!unmergedBranches.isEmpty()) { MessageDialog messageDialog = new UnmergedBranchDialog<>(shell, unmergedBranches); if (messageDialog.open() == Window.OK) { try { new ProgressMonitorDialog(shell).run(true, false, new IRunnableWithProgress() { @Override public void run(final IProgressMonitor monitor) throws InvocationTargetException, InterruptedException { try { monitor.beginTask( UIText.DeleteBranchCommand_DeletingBranchesProgress, unmergedBranches.size()); 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( UIText.AbstractHistoryCommitHandler_cantGetBranches, e); return false; } return !branchesOfCommit.isEmpty(); } }