/*******************************************************************************
* Copyright (c) 2000, 2017 IBM Corporation 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
*
*******************************************************************************/
package org.eclipse.dltk.internal.ui.wizards.buildpath.newsourcepage;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import org.eclipse.core.resources.IFolder;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.OperationCanceledException;
import org.eclipse.core.runtime.SubProgressMonitor;
import org.eclipse.dltk.core.DLTKCore;
import org.eclipse.dltk.core.DLTKLanguageManager;
import org.eclipse.dltk.core.IBuildpathEntry;
import org.eclipse.dltk.core.IModelElement;
import org.eclipse.dltk.core.IScriptFolder;
import org.eclipse.dltk.core.IScriptProject;
import org.eclipse.dltk.internal.corext.buildpath.BuildpathModifier;
import org.eclipse.dltk.internal.ui.wizards.NewWizardMessages;
import org.eclipse.dltk.internal.ui.wizards.buildpath.BPListElement;
import org.eclipse.dltk.internal.ui.wizards.buildpath.BuildPathBasePage;
import org.eclipse.dltk.ui.DLTKPluginImages;
import org.eclipse.dltk.ui.DLTKUIPlugin;
import org.eclipse.dltk.ui.dialogs.StatusInfo;
import org.eclipse.jface.action.Action;
import org.eclipse.jface.dialogs.ErrorDialog;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.SelectionChangedEvent;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchPart;
import org.eclipse.ui.IWorkbenchPartReference;
import org.eclipse.ui.IWorkbenchSite;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.part.ISetSelectionTarget;
public class AddFolderToBuildpathAction extends Action
implements ISelectionChangedListener {
private final IWorkbenchSite fSite;
// IScriptProject || IPackageFrament || IFolder
private final List<Object> fSelectedElements;
public AddFolderToBuildpathAction(IWorkbenchSite site) {
super(NewWizardMessages.NewSourceContainerWorkbookPage_ToolBar_AddSelSFToCP_label,
DLTKPluginImages.DESC_OBJS_PACKFRAG_ROOT);
setToolTipText(
NewWizardMessages.NewSourceContainerWorkbookPage_ToolBar_AddSelSFToCP_tooltip);
fSite = site;
fSelectedElements = new ArrayList<>();
}
@Override
public void run() {
final IScriptProject project;
Object object = fSelectedElements.get(0);
if (object instanceof IScriptProject) {
project = (IScriptProject) object;
} else if (object instanceof IScriptFolder) {
project = ((IScriptFolder) object).getScriptProject();
} else {
IFolder folder = (IFolder) object;
project = DLTKCore.create(folder.getProject());
if (project == null)
return;
}
final boolean removeProjectFromBuildpath;
if (fSelectedElements.size() == 1
&& fSelectedElements.get(0) instanceof IScriptProject) {
/*
* if only the project should be added, then the query does not need
* to be executed
*/
removeProjectFromBuildpath = true;
} else {
removeProjectFromBuildpath = false;
}
try {
final IRunnableWithProgress runnable = monitor -> {
try {
List<IModelElement> result = addToBuildpath(
fSelectedElements, project,
removeProjectFromBuildpath, monitor);
selectAndReveal(new StructuredSelection(result));
} catch (CoreException e) {
throw new InvocationTargetException(e);
}
};
PlatformUI.getWorkbench().getProgressService().run(true, false,
runnable);
} catch (final InvocationTargetException e) {
if (e.getCause() instanceof CoreException) {
showExceptionDialog((CoreException) e.getCause());
} else {
DLTKUIPlugin.log(e);
}
} catch (final InterruptedException e) {
}
}
private List<IModelElement> addToBuildpath(List<Object> elements,
IScriptProject project, boolean removeProjectFromBuildpath,
IProgressMonitor monitor)
throws OperationCanceledException, CoreException {
if (!DLTKLanguageManager.hasScriptNature(project.getProject())) {
StatusInfo rootStatus = new StatusInfo();
rootStatus.setError(
NewWizardMessages.BuildpathModifier_Error_NoNatures);
throw new CoreException(rootStatus);
}
try {
monitor.beginTask(
NewWizardMessages.BuildpathModifier_Monitor_AddToBuildpath,
elements.size() + 4);
monitor.worked(1);
List<BPListElement> existingEntries = BuildpathModifier
.getExistingEntries(project);
if (removeProjectFromBuildpath) {
BuildpathModifier.removeFromBuildpath(project, existingEntries,
new SubProgressMonitor(monitor, 1));
} else {
monitor.worked(1);
}
List<BPListElement> newEntries = new ArrayList<>();
for (int i = 0; i < elements.size(); i++) {
Object element = elements.get(i);
BPListElement entry;
if (element instanceof IResource)
entry = BuildpathModifier.addToBuildpath(
(IResource) element, existingEntries, newEntries,
project, new SubProgressMonitor(monitor, 1));
else
entry = BuildpathModifier.addToBuildpath(
(IModelElement) element, existingEntries,
newEntries, project,
new SubProgressMonitor(monitor, 1));
newEntries.add(entry);
}
Set<BPListElement> modifiedSourceEntries = new HashSet<>();
BuildPathBasePage.fixNestingConflicts(
newEntries.toArray(new BPListElement[newEntries.size()]),
existingEntries
.toArray(new BPListElement[existingEntries.size()]),
modifiedSourceEntries);
BuildpathModifier.setNewEntry(existingEntries, newEntries, project,
new SubProgressMonitor(monitor, 1));
BuildpathModifier.commitBuildPath(existingEntries, project,
new SubProgressMonitor(monitor, 1));
List<IModelElement> result = new ArrayList<>();
for (int i = 0; i < newEntries.size(); i++) {
IBuildpathEntry entry = newEntries.get(i).getBuildpathEntry();
IModelElement root;
if (entry.getPath().equals(project.getPath()))
root = project;
else
root = project.findProjectFragment(entry.getPath());
if (root != null) {
result.add(root);
}
}
return result;
} finally {
monitor.done();
}
}
@Override
public void selectionChanged(final SelectionChangedEvent event) {
final ISelection selection = event.getSelection();
if (selection instanceof IStructuredSelection) {
setEnabled(canHandle((IStructuredSelection) selection));
} else {
setEnabled(canHandle(StructuredSelection.EMPTY));
}
}
private boolean canHandle(IStructuredSelection elements) {
if (elements.size() == 0)
return false;
try {
fSelectedElements.clear();
for (Iterator<?> iter = elements.iterator(); iter.hasNext();) {
Object element = iter.next();
if (element instanceof IScriptProject) {
if (BuildpathModifier
.isSourceFolder((IScriptProject) element))
return false;
fSelectedElements.add(element);
} else if (element instanceof IProject) {
IScriptProject scriptProject = DLTKCore
.create((IProject) element);
if (!scriptProject.isValid() || BuildpathModifier
.isSourceFolder(scriptProject)) {
return false;
}
fSelectedElements.add(scriptProject);
} else if (element instanceof IScriptFolder) {
int type = DialogPackageExplorerActionGroup.getType(element,
((IScriptFolder) element).getScriptProject());
if (type != DialogPackageExplorerActionGroup.PACKAGE_FRAGMENT
&& type != DialogPackageExplorerActionGroup.INCLUDED_FOLDER)
return false;
fSelectedElements.add(element);
} else if (element instanceof IFolder) {
IProject project = ((IFolder) element).getProject();
IScriptProject scriptProject = DLTKCore.create(project);
if (scriptProject == null || !scriptProject.exists())
return false;
fSelectedElements.add(element);
} else {
return false;
}
}
return true;
} catch (CoreException e) {
}
return false;
}
private void showExceptionDialog(CoreException exception) {
showError(exception, fSite.getShell(),
NewWizardMessages.AddSourceFolderToBuildpathAction_ErrorTitle,
exception.getMessage());
}
private void showError(CoreException e, Shell shell, String title,
String message) {
IStatus status = e.getStatus();
if (status != null) {
ErrorDialog.openError(shell, message, title, status);
} else {
MessageDialog.openError(shell, title, message);
}
}
private void selectAndReveal(final ISelection selection) {
// validate the input
IWorkbenchPage page = fSite.getPage();
if (page == null)
return;
// get all the view and editor parts
List<IWorkbenchPart> parts = new ArrayList<>();
IWorkbenchPartReference refs[] = page.getViewReferences();
for (int i = 0; i < refs.length; i++) {
IWorkbenchPart part = refs[i].getPart(false);
if (part != null)
parts.add(part);
}
refs = page.getEditorReferences();
for (int i = 0; i < refs.length; i++) {
if (refs[i].getPart(false) != null)
parts.add(refs[i].getPart(false));
}
for (IWorkbenchPart part : parts) {
// get the part's ISetSelectionTarget implementation
ISetSelectionTarget target = null;
if (part instanceof ISetSelectionTarget)
target = (ISetSelectionTarget) part;
else
target = part.getAdapter(ISetSelectionTarget.class);
if (target != null) {
// select and reveal resource
final ISetSelectionTarget finalTarget = target;
page.getWorkbenchWindow().getShell().getDisplay()
.asyncExec(() -> finalTarget.selectReveal(selection));
}
}
}
}