org.eclipse.egit.ui/src/org/eclipse/egit/ui/internal/push/PushWizard.java

   1 /*******************************************************************************
   2  * Copyright (C) 2008, Marek Zawirski <marek.zawirski@gmail.com>
   3  *
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * See LICENSE for the full license text, also available.
   7  *******************************************************************************/
   8 package org.eclipse.egit.ui.internal.push;
   9
  10 import java.io.IOException;
  11 import java.lang.reflect.InvocationTargetException;
  12 import java.net.URISyntaxException;
  13 import java.util.Collection;
  14 import java.util.List;
  15
  16 import org.eclipse.core.runtime.IProgressMonitor;
  17 import org.eclipse.core.runtime.IStatus;
  18 import org.eclipse.core.runtime.Status;
  19 import org.eclipse.core.runtime.jobs.Job;
  20 import org.eclipse.egit.core.op.PushOperation;
  21 import org.eclipse.egit.core.op.PushOperationResult;
  22 import org.eclipse.egit.core.op.PushOperationSpecification;
  23 import org.eclipse.egit.ui.Activator;
  24 import org.eclipse.egit.ui.UIIcons;
  25 import org.eclipse.egit.ui.UIText;
  26 import org.eclipse.egit.ui.internal.components.RefSpecPage;
  27 import org.eclipse.egit.ui.internal.components.RepositorySelection;
  28 import org.eclipse.egit.ui.internal.components.RepositorySelectionPage;
  29 import org.eclipse.jface.dialogs.Dialog;
  30 import org.eclipse.jface.dialogs.ErrorDialog;
  31 import org.eclipse.jface.wizard.IWizardPage;
  32 import org.eclipse.jface.wizard.Wizard;
  33 import org.eclipse.osgi.util.NLS;
  34 import org.eclipse.swt.widgets.Shell;
  35 import org.eclipse.ui.PlatformUI;
  36 import org.spearce.jgit.lib.Repository;
  37 import org.spearce.jgit.lib.RepositoryConfig;
  38 import org.spearce.jgit.transport.RefSpec;
  39 import org.spearce.jgit.transport.RemoteConfig;
  40 import org.spearce.jgit.transport.RemoteRefUpdate;
  41 import org.spearce.jgit.transport.Transport;
  42 import org.spearce.jgit.transport.URIish;
  43
  44 /**
  45  * Wizard allowing user to specify all needed data to push to another repository
  46  * - including selection of remote repository and refs specifications.
  47  * <p>
  48  * Push operation is performed upon successful completion of this wizard.
  49  */
  50 public class PushWizard extends Wizard {
  51         private static String getURIsString(final Collection<URIish> uris) {
  52                 final StringBuilder sb = new StringBuilder();
  53                 boolean first = true;
  54                 for (final URIish uri : uris) {
  55                         if (first)
  56                                 first = false;
  57                         else
  58                                 sb.append(", "); //$NON-NLS-1$
  59                         sb.append(uri);
  60                 }
  61                 return sb.toString();
  62         }
  63
  64         private Repository localDb;
  65
  66         private final RepositorySelectionPage repoPage;
  67
  68         private final RefSpecPage refSpecPage;
  69
  70         private ConfirmationPage confirmPage;
  71
  72         /**
  73          * Create push wizard for specified local repository.
  74          *
  75          * @param localDb
  76          *            repository to push from.
  77          * @throws URISyntaxException
  78          *             when configuration of this repository contains illegal URIs.
  79          */
  80         public PushWizard(final Repository localDb) throws URISyntaxException {
  81                 this.localDb = localDb;
  82                 final List<RemoteConfig> remotes = RemoteConfig
  83                                 .getAllRemoteConfigs(localDb.getConfig());
  84                 repoPage = new RepositorySelectionPage(false, remotes);
  85                 refSpecPage = new RefSpecPage(localDb, true, repoPage);
  86                 confirmPage = new ConfirmationPage(localDb, repoPage, refSpecPage);
  87                 // TODO use/create another cool icon
  88                 setDefaultPageImageDescriptor(UIIcons.WIZBAN_IMPORT_REPO);
  89                 setNeedsProgressMonitor(true);
  90         }
  91
  92         @Override
  93         public void addPages() {
  94                 addPage(repoPage);
  95                 addPage(refSpecPage);
  96                 addPage(confirmPage);
  97         }
  98
  99         @Override
 100         public boolean performFinish() {
 101                 if (repoPage.getSelection().isConfigSelected()
 102                                 && refSpecPage.isSaveRequested()) {
 103                         saveRefSpecs();
 104                 }
 105
 106                 final PushOperation operation = createPushOperation();
 107                 if (operation == null)
 108                         return false;
 109                 final PushOperationResult resultToCompare;
 110                 if (confirmPage.isShowOnlyIfChangedSelected())
 111                         resultToCompare = confirmPage.getConfirmedResult();
 112                 else
 113                         resultToCompare = null;
 114                 final Job job = new PushJob(operation, resultToCompare,
 115                                 getDestinationString());
 116
 117                 job.setUser(true);
 118                 job.schedule();
 119
 120                 return true;
 121         }
 122
 123         @Override
 124         public String getWindowTitle() {
 125                 final IWizardPage currentPage = getContainer().getCurrentPage();
 126                 if (currentPage == repoPage || currentPage == null)
 127                         return UIText.PushWizard_windowTitleDefault;
 128                 final String destination = getDestinationString();
 129                 return NLS.bind(UIText.PushWizard_windowTitleWithDestination,
 130                                 destination);
 131         }
 132
 133         private void saveRefSpecs() {
 134                 final RemoteConfig rc = repoPage.getSelection().getConfig();
 135                 rc.setPushRefSpecs(refSpecPage.getRefSpecs());
 136                 final RepositoryConfig config = localDb.getConfig();
 137                 rc.update(config);
 138                 try {
 139                         config.save();
 140                 } catch (final IOException e) {
 141                         ErrorDialog.openError(getShell(), UIText.PushWizard_cantSaveTitle,
 142                                         UIText.PushWizard_cantSaveMessage, new Status(
 143                                                         IStatus.WARNING, Activator.getPluginId(), e
 144                                                                         .getMessage(), e));
 145                         // Continue, it's not critical.
 146                 }
 147         }
 148
 149         private PushOperation createPushOperation() {
 150                 try {
 151                         final PushOperationSpecification spec;
 152                         final RemoteConfig config = repoPage.getSelection().getConfig();
 153                         if (confirmPage.isConfirmed()) {
 154                                 final PushOperationResult confirmedResult = confirmPage
 155                                                 .getConfirmedResult();
 156                                 spec = confirmedResult.deriveSpecification(confirmPage
 157                                                 .isRequireUnchangedSelected());
 158                         } else {
 159                                 final Collection<RefSpec> fetchSpecs;
 160                                 if (config != null)
 161                                         fetchSpecs = config.getFetchRefSpecs();
 162                                 else
 163                                         fetchSpecs = null;
 164
 165                                 final Collection<RemoteRefUpdate> updates = Transport
 166                                                 .findRemoteRefUpdatesFor(localDb, refSpecPage
 167                                                                 .getRefSpecs(), fetchSpecs);
 168                                 if (updates.isEmpty()) {
 169                                         ErrorDialog.openError(getShell(),
 170                                                         UIText.PushWizard_missingRefsTitle, null,
 171                                                         new Status(IStatus.ERROR, Activator.getPluginId(),
 172                                                                         UIText.PushWizard_missingRefsMessage));
 173                                         return null;
 174                                 }
 175
 176                                 spec = new PushOperationSpecification();
 177                                 for (final URIish uri : repoPage.getSelection().getAllURIs())
 178                                         spec.addURIRefUpdates(uri, ConfirmationPage
 179                                                         .copyUpdates(updates));
 180                         }
 181                         return new PushOperation(localDb, spec, false, config);
 182                 } catch (final IOException e) {
 183                         ErrorDialog.openError(getShell(),
 184                                         UIText.PushWizard_cantPrepareUpdatesTitle,
 185                                         UIText.PushWizard_cantPrepareUpdatesMessage, new Status(
 186                                                         IStatus.ERROR, Activator.getPluginId(), e
 187                                                                         .getMessage(), e));
 188                         return null;
 189                 }
 190         }
 191
 192         private String getDestinationString() {
 193                 final RepositorySelection repoSelection = repoPage.getSelection();
 194                 final String destination;
 195                 if (repoSelection.isConfigSelected())
 196                         destination = repoSelection.getConfigName();
 197                 else
 198                         destination = repoSelection.getURI().toString();
 199                 return destination;
 200         }
 201
 202         private class PushJob extends Job {
 203                 private final PushOperation operation;
 204
 205                 private final PushOperationResult resultToCompare;
 206
 207                 private final String destinationString;
 208
 209                 public PushJob(final PushOperation operation,
 210                                 final PushOperationResult resultToCompare,
 211                                 final String destinationString) {
 212                         super(NLS.bind(UIText.PushWizard_jobName, getURIsString(operation
 213                                         .getSpecification().getURIs())));
 214                         this.operation = operation;
 215                         this.resultToCompare = resultToCompare;
 216                         this.destinationString = destinationString;
 217                 }
 218
 219                 @Override
 220                 protected IStatus run(final IProgressMonitor monitor) {
 221                         try {
 222                                 operation.run(monitor);
 223                         } catch (final InvocationTargetException e) {
 224                                 return new Status(IStatus.ERROR, Activator.getPluginId(),
 225                                                 UIText.PushWizard_unexpectedError, e.getCause());
 226                         }
 227
 228                         final PushOperationResult result = operation.getOperationResult();
 229                         if (!result.isSuccessfulConnectionForAnyURI()) {
 230                                 return new Status(IStatus.ERROR, Activator.getPluginId(), NLS
 231                                                 .bind(UIText.PushWizard_cantConnectToAny, result
 232                                                                 .getErrorStringForAllURis()));
 233                         }
 234
 235                         if (resultToCompare == null || !result.equals(resultToCompare)) {
 236                                 PlatformUI.getWorkbench().getDisplay().asyncExec(
 237                                                 new Runnable() {
 238                                                         public void run() {
 239                                                                 final Shell shell = PlatformUI.getWorkbench()
 240                                                                                 .getActiveWorkbenchWindow().getShell();
 241                                                                 final Dialog dialog = new PushResultDialog(
 242                                                                                 shell, localDb, result,
 243                                                                                 destinationString);
 244                                                                 dialog.open();
 245                                                         }
 246                                                 });
 247                         }
 248                         return Status.OK_STATUS;
 249                 }
 250         }
 251 }