UIContextImpl.java

/**
 * Copyright (c) Red Hat, Inc., contributors and others 2013 - 2014. All rights reserved
 *
 * Licensed under the Eclipse Public License version 1.0, available at
 * http://www.eclipse.org/legal/epl-v10.html
 */
package org.jboss.tools.forge.ui.internal.ext.context;

import java.io.File;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;

import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IWorkspace;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.IPath;
import org.eclipse.jdt.core.IJavaElement;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jface.text.ITextSelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.jboss.forge.addon.convert.Converter;
import org.jboss.forge.addon.convert.ConverterFactory;
import org.jboss.forge.addon.resource.Resource;
import org.jboss.forge.addon.ui.UIProvider;
import org.jboss.forge.addon.ui.context.AbstractUIContext;
import org.jboss.forge.addon.ui.context.UIContextListener;
import org.jboss.forge.furnace.proxy.Proxies;
import org.jboss.forge.furnace.services.Imported;
import org.jboss.tools.forge.core.furnace.FurnaceService;
import org.jboss.tools.forge.ui.internal.ForgeUIPlugin;

@SuppressWarnings({ "unchecked", "rawtypes" })
public class UIContextImpl extends AbstractUIContext {
	private final UISelectionImpl<?> currentSelection;

	private final UIProvider provider;

	public UIContextImpl(UIProvider provider, IStructuredSelection selection, ITextSelection textSelection) {
		this.provider = provider;
		List<Object> selectedElements = selection == null ? Collections.EMPTY_LIST : selection.toList();
		List<Object> result = new LinkedList<>();
		ConverterFactory converterFactory = FurnaceService.INSTANCE.getConverterFactory();
		Converter<File, Resource> converter = converterFactory.getConverter(File.class, Resource.class);

		if (selectedElements.isEmpty()) {
			// Get the Workspace directory path
			IWorkspace workspace = ResourcesPlugin.getWorkspace();
			File workspaceDirectory = workspace.getRoot().getLocation().toFile();
			Object convertedObj = converter.convert(workspaceDirectory);
			result.add(Proxies.unwrap(convertedObj));
		} else {
			for (Object object : selectedElements) {
				if (object instanceof Resource) {
					result.add(object);
				} else if (object instanceof File) {
					File file = (File) object;
					result.add(Proxies.unwrap(converter.convert(file)));
				} else if (object instanceof IResource) {
					IPath location = ((IResource) object).getLocation();
					if (location != null) {
						File file = location.toFile();
						result.add(Proxies.unwrap(converter.convert(file)));
					}
				} else if (object instanceof IJavaElement) {
					try {
						IJavaElement javaElem = (IJavaElement) object;
						IResource correspondingResource = javaElem.getCorrespondingResource();
						if (correspondingResource != null) {
							IPath location = correspondingResource.getLocation();
							if (location != null) {
								File file = location.toFile();
								result.add(Proxies.unwrap(converter.convert(file)));
							}
						}
					} catch (JavaModelException e) {
						ForgeUIPlugin.log(e);
					}
				}
			}
		}
		this.currentSelection = new UISelectionImpl(result, selection, textSelection);
		initialize();
	}

	@Override
	public UISelectionImpl<?> getInitialSelection() {
		return currentSelection;
	}

	public void initialize() {
		Imported<UIContextListener> services = FurnaceService.INSTANCE.lookupImported(UIContextListener.class);
		if (services != null)
			for (UIContextListener listener : services) {
				try {
					listener.contextInitialized(this);
				} catch (Exception e) {
					ForgeUIPlugin.log(e);
				}
			}
	}

	@Override
	public void close() {
		super.close();
		Imported<UIContextListener> services = FurnaceService.INSTANCE.lookupImported(UIContextListener.class);
		if (services != null)
			for (org.jboss.forge.addon.ui.context.UIContextListener listener : services) {
				try {
					listener.contextDestroyed(this);
				} catch (Exception e) {
					ForgeUIPlugin.log(e);
				}
			}
	}

	@Override
	public UIProvider getProvider() {
		return provider;
	}

}