plugins/org.eclipse.ldt.ui/src/org/eclipse/ldt/ui/LuaDialogUtil.java

/*******************************************************************************
 * Copyright (c) 2011, 2012 Sierra Wireless 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:
 *     Sierra Wireless - initial API and implementation
 *******************************************************************************/
package org.eclipse.ldt.ui;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.core.resources.IContainer;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.jface.window.Window;
import org.eclipse.ldt.core.LuaUtils;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.dialogs.ElementListSelectionDialog;
import org.eclipse.ui.dialogs.ElementTreeSelectionDialog;
import org.eclipse.ui.model.WorkbenchContentProvider;
import org.eclipse.ui.model.WorkbenchLabelProvider;

/**
 * Utility class, contains commons methods which could be used to implements UI for Lua
 */
public final class LuaDialogUtil {
	private LuaDialogUtil() {
	}

	public static final IProject openSelectLuaProjectDialog(Shell shell, String projectName) {
		return openSelectLuaProjectDialog(shell, projectName, null, null);
	}

	public static final IProject openSelectLuaProjectDialog(Shell shell, String projectName, String title, String message) {
		// initialize selection dialog
		ElementListSelectionDialog dialog = new ElementListSelectionDialog(shell, new WorkbenchLabelProvider());
		if (title == null)
			dialog.setTitle(Messages.LuaDialogUtil_title);
		else
			dialog.setTitle(title);
		if (message == null)
			dialog.setMessage(Messages.LuaDialogUtil_message);
		else
			dialog.setMessage(message);

		dialog.setElements(LuaUtils.getLuaProjects());

		// initialize default selected project
		if (projectName != null && !projectName.isEmpty()) {
			IProject project = ResourcesPlugin.getWorkspace().getRoot().getProject(projectName);
			if (project != null)
				dialog.setInitialSelections(new Object[] { project });
		}

		// open dialog and get result
		if (dialog.open() == Window.OK) {
			Object result = dialog.getFirstResult();
			if (result instanceof IProject) {
				return (IProject) result;

			}
		}
		return null;
	}

	public static final IFile openSelectScriptFromProjectDialog(Shell shell, IProject project) {
		return openSelectScriptFromProjectDialog(shell, project, null, null);
	}

	public static final IFile openSelectScriptFromProjectDialog(Shell shell, IProject project, String title, String message) {
		ElementTreeSelectionDialog dialog = new ElementTreeSelectionDialog(shell, new WorkbenchLabelProvider(), new WorkbenchContentProvider() {
			@Override
			public Object[] getElements(Object element) {
				Object[] elements = super.getElements(element);
				List<Object> newElementList = new ArrayList<Object>();

				// filter to display only lua files
				for (Object anElement : elements) {
					if (anElement instanceof IFile) {
						IFile file = (IFile) anElement;
						if ("lua".equals(file.getFileExtension())) { //$NON-NLS-1$
							newElementList.add(file);
						}
					} else if (anElement instanceof IContainer) {
						newElementList.add(anElement);
					}
				}

				return newElementList.toArray();
			}
		});
		dialog.setInput(project);

		if (title == null)
			dialog.setTitle(Messages.LuaDialogUtil_selectScript_title);
		else
			dialog.setTitle(title);
		if (message == null)
			dialog.setMessage(Messages.LuaDialogUtil_selectScript_message);
		else
			dialog.setMessage(message);

		if (dialog.open() == Window.OK) {
			return (IFile) dialog.getResult()[0];
		}
		return null;
	}

}