org.eclipse.egit.ui/src/org/eclipse/egit/ui/internal/clone/GitCloneWizard.java

   1 /*******************************************************************************
   2  * Copyright (C) 2008, Roger C. Soares <rogersoares@intelinet.com.br>
   3  * Copyright (C) 2008, Shawn O. Pearce <spearce@spearce.org>
   4  * Copyright (C) 2008, Marek Zawirski <marek.zawirski@gmail.com>
   5  * Copyright (C) 2008, Robin Rosenberg <robin.rosenberg@dewire.com>
   6  *
   7  * All rights reserved. This program and the accompanying materials
   8  * are made available under the terms of the Eclipse Public License v1.0
   9  * which accompanies this distribution, and is available at
  10  * http://www.eclipse.org/legal/epl-v10.html
  11  *******************************************************************************/
  12 package org.eclipse.egit.ui.internal.clone;
  13
  14 import java.io.File;
  15 import java.lang.reflect.InvocationTargetException;
  16 import java.util.Collection;
  17
  18 import org.eclipse.core.runtime.IProgressMonitor;
  19 import org.eclipse.core.runtime.IStatus;
  20 import org.eclipse.core.runtime.Status;
  21 import org.eclipse.core.runtime.jobs.Job;
  22 import org.eclipse.egit.core.op.CloneOperation;
  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.RepositorySelectionPage;
  27 import org.eclipse.jface.dialogs.ErrorDialog;
  28 import org.eclipse.jface.dialogs.MessageDialog;
  29 import org.eclipse.jface.viewers.IStructuredSelection;
  30 import org.eclipse.jface.wizard.Wizard;
  31 import org.eclipse.jgit.lib.Ref;
  32 import org.eclipse.jgit.transport.URIish;
  33 import org.eclipse.osgi.util.NLS;
  34 import org.eclipse.ui.IImportWizard;
  35 import org.eclipse.ui.IWorkbench;
  36
  37 /**
  38  * Import Git Repository Wizard. A front end to a git clone operation.
  39  */
  40 public class GitCloneWizard extends Wizard implements IImportWizard {
  41         private RepositorySelectionPage cloneSource;
  42
  43         private SourceBranchPage validSource;
  44
  45         private CloneDestinationPage cloneDestination;
  46
  47         private GitProjectsImportPage importProject;
  48
  49         public void init(IWorkbench arg0, IStructuredSelection arg1) {
  50                 setWindowTitle(UIText.GitCloneWizard_title);
  51                 setDefaultPageImageDescriptor(UIIcons.WIZBAN_IMPORT_REPO);
  52                 setNeedsProgressMonitor(true);
  53                 cloneSource = new RepositorySelectionPage(true);
  54                 validSource = new SourceBranchPage(cloneSource);
  55                 cloneDestination = new CloneDestinationPage(cloneSource, validSource);
  56                 importProject = new GitProjectsImportPage() {
  57                         @Override
  58                         public void setVisible(boolean visible) {
  59                                 if (visible) {
  60                                         if (cloneDestination.alreadyClonedInto == null) {
  61                                                 if (performClone(false))
  62                                                         cloneDestination.alreadyClonedInto = cloneDestination
  63                                                                         .getDestinationFile().getAbsolutePath();
  64                                         }
  65                                         setProjectsList(cloneDestination.alreadyClonedInto);
  66                                 }
  67                                 super.setVisible(visible);
  68                         }
  69                 };
  70         }
  71
  72         @Override
  73         public boolean performCancel() {
  74                 if (cloneDestination.alreadyClonedInto != null) {
  75                         if (MessageDialog
  76                                         .openQuestion(getShell(), "Aborting clone.",
  77                                                         "A complete clone was already made. Do you want to delete it?")) {
  78                                 deleteRecursively(new File(cloneDestination.alreadyClonedInto));
  79                         }
  80                 }
  81                 return true;
  82         }
  83
  84         private void deleteRecursively(File f) {
  85                 for (File i : f.listFiles()) {
  86                         if (i.isDirectory()) {
  87                                 deleteRecursively(i);
  88                         } else {
  89                                 if (!i.delete()) {
  90                                         i.deleteOnExit();
  91                                 }
  92                         }
  93                 }
  94                 if (!f.delete())
  95                         f.deleteOnExit();
  96         }
  97
  98         @Override
  99         public void addPages() {
 100                 addPage(cloneSource);
 101                 addPage(validSource);
 102                 addPage(cloneDestination);
 103                 addPage(importProject);
 104         }
 105
 106         @Override
 107         public boolean canFinish() {
 108                 return cloneDestination.isPageComplete()
 109                                 && !cloneDestination.showImportWizard.getSelection()
 110                                 || importProject.isPageComplete();
 111         }
 112
 113         @Override
 114         public boolean performFinish() {
 115                 if (!cloneDestination.showImportWizard.getSelection())
 116                         return performClone(true);
 117                 return importProject.createProjects();
 118         }
 119
 120         boolean performClone(boolean background) {
 121                 final URIish uri = cloneSource.getSelection().getURI();
 122                 final boolean allSelected = validSource.isAllSelected();
 123                 final Collection<Ref> selectedBranches = validSource
 124                                 .getSelectedBranches();
 125                 final File workdir = cloneDestination.getDestinationFile();
 126                 final String branch = cloneDestination.getInitialBranch();
 127                 final String remoteName = cloneDestination.getRemote();
 128
 129                 workdir.mkdirs();
 130                 if (!workdir.isDirectory()) {
 131                         final String errorMessage = NLS.bind(
 132                                         UIText.GitCloneWizard_errorCannotCreate, workdir.getPath());
 133                         ErrorDialog.openError(getShell(), getWindowTitle(),
 134                                         UIText.GitCloneWizard_failed, new Status(IStatus.ERROR,
 135                                                         Activator.getPluginId(), 0, errorMessage, null));
 136                         // let's give user a chance to fix this minor problem
 137                         return false;
 138                 }
 139
 140                 final CloneOperation op = new CloneOperation(uri, allSelected,
 141                                 selectedBranches, workdir, branch, remoteName);
 142                 importProject.setGitDir(op.getGitDir());
 143                 if (background) {
 144                         final Job job = new Job(NLS.bind(UIText.GitCloneWizard_jobName, uri
 145                                         .toString())) {
 146                                 @Override
 147                                 protected IStatus run(final IProgressMonitor monitor) {
 148                                         try {
 149                                                 op.run(monitor);
 150                                                 return Status.OK_STATUS;
 151                                         } catch (InterruptedException e) {
 152                                                 return Status.CANCEL_STATUS;
 153                                         } catch (InvocationTargetException e) {
 154                                                 Throwable thr = e.getCause();
 155                                                 return new Status(IStatus.ERROR, Activator
 156                                                                 .getPluginId(), 0, thr.getMessage(), thr);
 157                                         }
 158                                 }
 159                         };
 160                         job.setUser(true);
 161                         job.schedule();
 162                         return true;
 163                 } else {
 164                         try {
 165                                 // Perform clone in ModalContext thread with progress
 166                                 // reporting on the wizard.
 167                                 getContainer().run(true, true, op);
 168                                 return true;
 169                         } catch (InterruptedException e) {
 170                                 MessageDialog.openInformation(getShell(),
 171                                                 UIText.GitCloneWizard_CloneFailedHeading,
 172                                                 UIText.GitCloneWizard_CloneCanceledMessage);
 173                                 return false;
 174                         } catch (Exception e) {
 175                                 Activator.logError(UIText.GitCloneWizard_CloneFailedHeading, e);
 176                                 MessageDialog.openError(getShell(),
 177                                                 UIText.GitCloneWizard_CloneFailedHeading, e.toString());
 178                                 return false;
 179                         }
 180                 }
 181         }
 182 }