org.eclipse.egit.ui.internal.sharing.SharingWizard.java Source code

Introduction

Here is the source code for org.eclipse.egit.ui.internal.sharing.SharingWizard.java

Source

/*******************************************************************************
 * Copyright (C) 2007, Robin Rosenberg <robin.rosenberg@dewire.com>
 * Copyright (C) 2007, Shawn O. Pearce <spearce@spearce.org>
 * 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.sharing;

import java.io.File;
import java.lang.reflect.InvocationTargetException;
import java.util.HashSet;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Path;
import org.eclipse.egit.core.op.ConnectProviderOperation;
import org.eclipse.egit.ui.Activator;
import org.eclipse.egit.ui.internal.UIText;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.team.ui.IConfigurationWizard;
import org.eclipse.team.ui.IConfigurationWizardExtension;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.ide.undo.MoveProjectOperation;

/**
 * The dialog used for activating Team>Share, i.e. to create a new Git
 * repository or associate projects with one.
 */
public class SharingWizard extends Wizard implements IConfigurationWizard, IConfigurationWizardExtension {
    IProject[] projects;

    private ExistingOrNewPage existingPage;

    /**
     * Construct the Git Sharing Wizard for connecting Git project to Eclipse
     */
    public SharingWizard() {
        setWindowTitle(UIText.SharingWizard_windowTitle);
        setNeedsProgressMonitor(true);
    }

    public void init(IWorkbench workbench, IProject[] p) {
        this.projects = new IProject[p.length];
        System.arraycopy(p, 0, this.projects, 0, p.length);
    }

    public void init(final IWorkbench workbench, final IProject p) {
        projects = new IProject[] { p };
    }

    public void addPages() {
        existingPage = new ExistingOrNewPage(this);
        addPage(existingPage);
    }

    public boolean performFinish() {
        if (!existingPage.getInternalMode()) {
            try {
                final Map<IProject, File> projectsToMove = existingPage.getProjects(true);
                final Repository selectedRepository = existingPage.getSelectedRepsoitory();
                getContainer().run(false, false, new IRunnableWithProgress() {
                    public void run(IProgressMonitor monitor)
                            throws InvocationTargetException, InterruptedException {
                        for (Map.Entry<IProject, File> entry : projectsToMove.entrySet()) {

                            IPath targetLocation = new Path(entry.getValue().getPath());
                            IPath currentLocation = entry.getKey().getLocation();
                            if (!targetLocation.equals(currentLocation)) {
                                MoveProjectOperation op = new MoveProjectOperation(entry.getKey(),
                                        entry.getValue().toURI(), UIText.SharingWizard_MoveProjectActionLabel);
                                try {
                                    IStatus result = op.execute(monitor, null);
                                    if (!result.isOK())
                                        throw new RuntimeException();
                                } catch (ExecutionException e) {
                                    if (e.getCause() != null)
                                        throw new InvocationTargetException(e.getCause());
                                    throw new InvocationTargetException(e);
                                }
                            }
                            try {
                                new ConnectProviderOperation(entry.getKey(), selectedRepository.getDirectory())
                                        .execute(monitor);
                            } catch (CoreException e) {
                                throw new InvocationTargetException(e);
                            }
                        }
                    }
                });
            } catch (InvocationTargetException e) {
                Activator.handleError(UIText.SharingWizard_failed, e.getCause(), true);
                return false;
            } catch (InterruptedException e) {
                // ignore for the moment
            }
            return true;
        } else {
            final ConnectProviderOperation op = new ConnectProviderOperation(existingPage.getProjects(true));
            try {
                getContainer().run(true, false, new IRunnableWithProgress() {
                    public void run(final IProgressMonitor monitor) throws InvocationTargetException {
                        try {
                            op.execute(monitor);
                            PlatformUI.getWorkbench().getDisplay().syncExec(new Runnable() {
                                public void run() {
                                    Set<File> filesToAdd = new HashSet<File>();
                                    // collect all files first
                                    for (Entry<IProject, File> entry : existingPage.getProjects(true).entrySet())
                                        filesToAdd.add(entry.getValue());
                                    // add the files to the repository
                                    // view
                                    for (File file : filesToAdd)
                                        Activator.getDefault().getRepositoryUtil().addConfiguredRepository(file);
                                }
                            });
                        } catch (CoreException ce) {
                            throw new InvocationTargetException(ce);
                        }
                    }
                });
                return true;
            } catch (Throwable e) {
                if (e instanceof InvocationTargetException) {
                    e = e.getCause();
                }
                if (e instanceof CoreException) {
                    IStatus status = ((CoreException) e).getStatus();
                    e = status.getException();
                }
                Activator.handleError(UIText.SharingWizard_failed, e, true);
                return false;
            }
        }
    }

    @Override
    public boolean canFinish() {
        return existingPage.isPageComplete();
    }
}