org.jboss.tools.common.jdt.debug.internal.SourceLookupUtil.java Source code

Introduction

Here is the source code for org.jboss.tools.common.jdt.debug.internal.SourceLookupUtil.java

Source

/*************************************************************************************
 * Copyright (c) 2015 Red Hat, Inc. and others.
 * 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:
 *     JBoss by Red Hat - Initial implementation.
 ************************************************************************************/
package org.jboss.tools.common.jdt.debug.internal;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.debug.core.DebugPlugin;
import org.eclipse.debug.core.ILaunchConfiguration;
import org.eclipse.debug.core.ILaunchConfigurationWorkingCopy;
import org.eclipse.debug.core.ILaunchManager;
import org.eclipse.debug.core.model.IPersistableSourceLocator;
import org.eclipse.debug.core.sourcelookup.AbstractSourceLookupDirector;
import org.eclipse.debug.core.sourcelookup.ISourceContainer;
import org.eclipse.debug.core.sourcelookup.ISourceLookupDirector;
import org.eclipse.jdt.core.IJavaElement;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.launching.sourcelookup.containers.JavaProjectSourceContainer;
import org.jboss.tools.common.jdt.debug.RemoteDebugActivator;

public class SourceLookupUtil {
    public void addSelectedProjects(ILaunchConfigurationWorkingCopy wc, IJavaElement[] selection,
            IJavaProject javaProject) throws CoreException {
        if ((selection == null || selection.length == 0)) {
            // Nothing to do
            wc.setAttribute(RemoteDebugActivator.ATTR_SELECTED_PROJECTS, (List<String>) null);
        } else {
            ILaunchManager manager = DebugPlugin.getDefault().getLaunchManager();
            ISourceLookupDirector director = addSourceContainers(manager, wc, selection, javaProject);
            if (director != null) {
                wc.setAttribute(ILaunchConfiguration.ATTR_SOURCE_LOCATOR_MEMENTO, director.getMemento());
                wc.setAttribute(ILaunchConfiguration.ATTR_SOURCE_LOCATOR_ID, director.getId());
            }
        }
    }

    private ISourceLookupDirector addSourceContainers(ILaunchManager manager, ILaunchConfigurationWorkingCopy wc,
            IJavaElement[] selection, IJavaProject javaProject) throws CoreException {
        String memento = null;
        String locatorId = null;
        try {
            memento = wc.getAttribute(ILaunchConfiguration.ATTR_SOURCE_LOCATOR_MEMENTO, (String) null);
            locatorId = wc.getAttribute(ILaunchConfiguration.ATTR_SOURCE_LOCATOR_ID, (String) null);
            if (locatorId == null) {
                locatorId = wc.getType().getSourceLocatorId();
            }
        } catch (CoreException e) {
            RemoteDebugActivator.pluginLog().logError(e);
        }
        IPersistableSourceLocator locator = manager.newSourceLocator(locatorId);
        if (memento != null) {
            locator.initializeFromMemento(memento);
        } else {
            locator.initializeDefaults(wc);
        }
        if (locator instanceof AbstractSourceLookupDirector) {
            ISourceLookupDirector director = (ISourceLookupDirector) locator;
            ArrayList<ISourceContainer> sourceContainers = new ArrayList<ISourceContainer>(
                    Arrays.asList(director.getSourceContainers()));
            Set<String> projectNames = new LinkedHashSet<String>();
            if (javaProject != null) {
                projectNames.add(javaProject.getElementName());
            }
            for (int i = 0; i < selection.length; i++) {
                IJavaProject project = selection[i].getJavaProject();
                if (project instanceof IJavaProject && !project.equals(javaProject)) {
                    projectNames.add(((IJavaProject) project).getElementName());
                    sourceContainers.add(new JavaProjectSourceContainer((IJavaProject) project));
                }
            }

            director.setSourceContainers(
                    (ISourceContainer[]) sourceContainers.toArray(new ISourceContainer[sourceContainers.size()]));
            director.setFindDuplicates(true);
            List<String> projectsList = new ArrayList<String>();
            projectsList.addAll(projectNames);
            wc.setAttribute(RemoteDebugActivator.ATTR_SELECTED_PROJECTS, projectsList);
            wc.setAttribute(ILaunchConfiguration.ATTR_SOURCE_LOCATOR_MEMENTO, director.getMemento());
            wc.setAttribute(ILaunchConfiguration.ATTR_SOURCE_LOCATOR_ID, director.getId());
            return director;
        } else {
            RemoteDebugActivator.pluginLog().logWarning("Launch configuration doesn't support source lookup");
        }
        return null;
    }
}