org.eclipse.egit.ui.internal.components.RefContentAssistProvider.java Source code

Introduction

Here is the source code for org.eclipse.egit.ui.internal.components.RefContentAssistProvider.java

Source

/*******************************************************************************
 * Copyright (C) 2011, Mathias Kinzler <mathias.kinzler@sap.com>
 * Copyright (C) 2012, Markus Duft <markus.duft@salomon.at>
 *
 * 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.components;

import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.List;

import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.egit.core.op.ListRemoteOperation;
import org.eclipse.egit.ui.Activator;
import org.eclipse.egit.ui.UIPreferences;
import org.eclipse.egit.ui.internal.UIText;
import org.eclipse.jface.dialogs.ProgressMonitorDialog;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.eclipse.jgit.lib.Constants;
import org.eclipse.jgit.lib.Ref;
import org.eclipse.jgit.lib.RefDatabase;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.transport.URIish;
import org.eclipse.swt.widgets.Shell;

/**
 * Contains functionality required to calculate content assist data for {@link Ref}s
 */
public class RefContentAssistProvider {
    private List<Ref> destinationRefs;
    private List<Ref> sourceRefs;
    private final Shell shell;
    private final Repository repo;
    private URIish uri;

    /**
     * @param repo the repository
     * @param uri the uri to fetch branches from
     * @param shell the shell used to attach progress dialogs to.
     */
    public RefContentAssistProvider(Repository repo, URIish uri, Shell shell) {
        this.repo = repo;
        this.uri = uri;
        this.shell = shell;
    }

    /**
     * @param source whether we want proposals for the source or the destination of the operation
     * @param pushMode whether the operation is a push or a fetch
     * @return a list of all refs for the given mode.
     */
    public List<Ref> getRefsForContentAssist(boolean source, boolean pushMode) {
        if (source) {
            if (sourceRefs != null)
                return sourceRefs;
        } else if (destinationRefs != null)
            return destinationRefs;

        List<Ref> result = new ArrayList<Ref>();
        try {
            boolean local = pushMode == source;
            if (!local) {
                final ListRemoteOperation lop = new ListRemoteOperation(repo, uri, Activator.getDefault()
                        .getPreferenceStore().getInt(UIPreferences.REMOTE_CONNECTION_TIMEOUT));

                new ProgressMonitorDialog(shell).run(false, true, new IRunnableWithProgress() {

                    public void run(IProgressMonitor monitor)
                            throws InvocationTargetException, InterruptedException {
                        monitor.beginTask(UIText.RefSpecDialog_GettingRemoteRefsMonitorMessage,
                                IProgressMonitor.UNKNOWN);
                        lop.run(monitor);
                        monitor.done();
                    }
                });
                for (Ref ref : lop.getRemoteRefs())
                    if (ref.getName().startsWith(Constants.R_HEADS)
                            || (!pushMode && ref.getName().startsWith(Constants.R_TAGS)))
                        result.add(ref);

            } else if (pushMode)
                for (Ref ref : repo.getRefDatabase().getRefs(RefDatabase.ALL).values()) {
                    if (ref.getName().startsWith(Constants.R_REMOTES))
                        continue;
                    result.add(ref);
                }
            else
                for (Ref ref : repo.getRefDatabase().getRefs(Constants.R_REMOTES).values())
                    result.add(ref);
        } catch (RuntimeException e) {
            throw e;
        } catch (Exception e) {
            Activator.handleError(e.getMessage(), e, true);
            return result;
        }
        if (source)
            sourceRefs = result;
        else
            destinationRefs = result;
        return result;
    }

    /**
     * @return the associated current repository.
     */
    public Repository getRepository() {
        return repo;
    }

    /**
     * @return the associated remote config to fetch branches from.
     */
    public URIish getRemoteURI() {
        return uri;
    }
}