org.eclipse.egit.ui.internal.actions.PullFromUpstreamActionHandler.java Source code

Introduction

Here is the source code for org.eclipse.egit.ui.internal.actions.PullFromUpstreamActionHandler.java

Source

/*******************************************************************************
 * Copyright (c) 2010 SAP AG.
 * 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
 *
 * Contributors:
 *    Mathias Kinzler (SAP AG) - initial implementation
 *******************************************************************************/
package org.eclipse.egit.ui.internal.actions;

import java.io.IOException;

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.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.jobs.IJobChangeEvent;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.core.runtime.jobs.JobChangeAdapter;
import org.eclipse.egit.core.op.PullOperation;
import org.eclipse.egit.ui.Activator;
import org.eclipse.egit.ui.UIPreferences;
import org.eclipse.egit.ui.UIText;
import org.eclipse.egit.ui.internal.pull.PullResultDialog;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jgit.lib.Constants;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.osgi.util.NLS;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.PlatformUI;

/**
 * Action for pulling into the currently checked-out branch.
 */
public class PullFromUpstreamActionHandler extends RepositoryActionHandler {
    public Object execute(ExecutionEvent event) throws ExecutionException {
        final Repository repository = getRepository(true, event);
        if (repository == null)
            return null;

        if (!canMerge(repository, event))
            return null;

        // obtain the currently checked-out branch
        String branchName;
        try {
            branchName = repository.getBranch();
        } catch (IOException e) {
            throw new ExecutionException(e.getMessage(), e);
        }

        String jobname = NLS.bind(UIText.PullCurrentBranchActionHandler_PullJobname, branchName);
        int timeout = Activator.getDefault().getPreferenceStore().getInt(UIPreferences.REMOTE_CONNECTION_TIMEOUT);
        final PullOperation pull = new PullOperation(repository, timeout);
        Job job = new Job(jobname) {
            @Override
            protected IStatus run(IProgressMonitor monitor) {
                try {
                    pull.execute(monitor);
                } catch (final CoreException e) {
                    return e.getStatus();
                }
                return Status.OK_STATUS;
            }
        };
        job.setUser(true);
        job.setRule(pull.getSchedulingRule());
        job.addJobChangeListener(new JobChangeAdapter() {
            @Override
            public void done(IJobChangeEvent cevent) {
                IStatus result = cevent.getJob().getResult();
                if (result.getSeverity() == IStatus.CANCEL) {
                    Display.getDefault().asyncExec(new Runnable() {
                        public void run() {
                            // don't use getShell(event) here since
                            // the active shell has changed since the
                            // execution has been triggered.
                            Shell shell = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell();
                            MessageDialog.openInformation(shell,
                                    UIText.PullCurrentBranchActionHandler_PullCanceledTitle,
                                    UIText.PullCurrentBranchActionHandler_PullCanceledMessage);
                        }
                    });
                } else if (result.isOK()) {
                    Display.getDefault().asyncExec(new Runnable() {
                        public void run() {
                            Shell shell = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell();
                            new PullResultDialog(shell, repository, pull.getResult()).open();
                        }
                    });
                }
            }
        });
        job.schedule();
        return null;
    }

    @Override
    public boolean isEnabled() {
        // we don't do the full canMerge check here, but
        // ensure that a branch is checked out
        Repository repo = getRepository();
        if (repo == null)
            return false;
        try {
            String fullBranch = repo.getFullBranch();
            return (fullBranch.startsWith(Constants.R_REFS));
        } catch (IOException e) {
            return false;
        }
    }
}