org.eclipse.xtext.resource.impl.LiveShadowedResourceDescriptions Maven / Gradle / Ivy

Go to download
/*******************************************************************************
 * Copyright (c) 2011 itemis AG (http://www.itemis.eu) 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.xtext.resource.impl;

import java.util.Map;

import org.eclipse.emf.common.notify.Notifier;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.xtext.EcoreUtil2;
import org.eclipse.xtext.naming.QualifiedName;
import org.eclipse.xtext.resource.IEObjectDescription;
import org.eclipse.xtext.resource.IResourceDescription;
import org.eclipse.xtext.resource.IResourceDescriptions;
import org.eclipse.xtext.resource.ISelectable;

import com.google.common.base.Function;
import com.google.common.base.Predicate;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Iterables;
import com.google.inject.Inject;

/**
 * @author Jan Koehnlein - Initial contribution and API
 * @author Sebastian Zarnekow - Applied implementation pattern from DirtyStateAwareResourceDescriptions
 * @author Knut Wannheden - Fixed performance issue caused by the inheritance from ResourceSetBasedResourceDescriptions
 *         
 *         The implementation now manages local resources using an injected {@link ResourceSetBasedResourceDescriptions}
 *         field. The class only inherits from {@link ResourceSetBasedResourceDescriptions} to remain backwards
 *         compatible with the initial 2.1 release.
 * 
 * @since 2.1
 */
public class LiveShadowedResourceDescriptions extends ResourceSetBasedResourceDescriptions {

	@Inject
	private ResourceSetBasedResourceDescriptions localDescriptions;

	@Inject
	private IResourceDescriptions globalDescriptions;

	@Override
	public void setContext(Notifier ctx) {
		localDescriptions.setContext(ctx);
		if(globalDescriptions instanceof IResourceDescriptions.IContextAware)
			((IResourceDescriptions.IContextAware) globalDescriptions).setContext(ctx);
	}

	@Override
	public IResourceDescription getResourceDescription(URI uri) {
		IResourceDescription result = localDescriptions.getResourceDescription(uri);
		if (result == null && !isExistingOrRenamedResourceURI(uri))
			result = globalDescriptions.getResourceDescription(uri);
		return result;
	}

	@Override
	public Iterable getAllResourceDescriptions() {
		Iterable notInLiveResourceSet = Iterables.filter(globalDescriptions.getAllResourceDescriptions(), new Predicate() {
			public boolean apply(IResourceDescription input) {
				return !isExistingOrRenamedResourceURI(input.getURI());
			}
		});
		Iterable result = Iterables.concat(localDescriptions.getAllResourceDescriptions(), notInLiveResourceSet);
		return result;
	}

	@Override
	public boolean isEmpty() {
		return globalDescriptions.isEmpty() && localDescriptions.isEmpty();
	}

	@Override
	public ResourceSet getResourceSet() {
		return localDescriptions.getResourceSet();
	}

	@Override
	protected boolean hasDescription(URI uri) {
		boolean result = localDescriptions.hasDescription(uri);
		if (!result) {
			if (isExistingOrRenamedResourceURI(uri)) {
				result = false;
			} else {
				result = globalDescriptions.getResourceDescription(uri) != null;
			}
		}
		return result;
	}

	protected boolean isExistingOrRenamedResourceURI(URI uri) {
		ResourceSet resourceSet = localDescriptions.getResourceSet();
		if (resourceSet instanceof ResourceSetImpl) {
			Map map = ((ResourceSetImpl) resourceSet).getURIResourceMap();
			boolean result = map.containsKey(uri.trimFragment());
			return result;
		}
		throw new IllegalStateException("ResourceSet is not a ResourceSetImpl");
	}

	@Override
	public Iterable getExportedObjects() {
		return Iterables.concat(Iterables.transform(getAllResourceDescriptions(), new Function>() {
			public Iterable apply(ISelectable from) {
				if (from != null)
					return from.getExportedObjects();
				return ImmutableSet.of();
			}
		}));
	}

	@Override
	public Iterable getExportedObjects(EClass type, QualifiedName name, boolean ignoreCase) {
		Iterable liveDescriptions = localDescriptions.getExportedObjects(type, name, ignoreCase);
		Iterable persistentDescriptions = globalDescriptions.getExportedObjects(type, name, ignoreCase);
		return joinIterables(liveDescriptions, persistentDescriptions);
	}

	protected Iterable joinIterables(Iterable liveDescriptions,
			Iterable persistentDescriptions) {
		Iterable filteredPersistent = Iterables.filter(persistentDescriptions, new Predicate() {
			public boolean apply(IEObjectDescription input) {
				URI resourceURI = input.getEObjectURI().trimFragment();
				if (isExistingOrRenamedResourceURI(resourceURI))
					return false;
				return true;
			}
		});
		return Iterables.concat(liveDescriptions, filteredPersistent);
	}

	@Override
	public Iterable getExportedObjectsByType(EClass type) {
		Iterable liveDescriptions = localDescriptions.getExportedObjectsByType(type);
		Iterable persistentDescriptions = globalDescriptions.getExportedObjectsByType(type);
		return joinIterables(liveDescriptions, persistentDescriptions);
	}

	@Override
	public Iterable getExportedObjectsByObject(EObject object) {
		URI resourceURI = EcoreUtil2.getPlatformResourceOrNormalizedURI(object).trimFragment();
		if (localDescriptions.hasDescription(resourceURI))
			return localDescriptions.getExportedObjectsByObject(object);
		return globalDescriptions.getExportedObjectsByObject(object);
	}

}