org.eclipse.osbp.xtext.oxtype/src/org/eclipse/osbp/xtext/oxtype/imports/EObjectImportOrganizer.java

package org.eclipse.osbp.xtext.oxtype.imports;

import java.util.List;

import org.eclipse.emf.ecore.EObject;
import org.eclipse.xtext.naming.IQualifiedNameProvider;
import org.eclipse.xtext.resource.XtextResource;
import org.eclipse.xtext.util.ReplaceRegion;

import com.google.inject.Inject;
import com.google.inject.Provider;

public class EObjectImportOrganizer {

	@Inject
	private Provider<EObjectUsageCollector> eobjectUsageCollectorProvider;

	@Inject
	private IUnresolvedEObjectResolver unresolvedTypeResolver;

	@Inject
	private IQualifiedNameProvider nameProvider;

	public List<ReplaceRegion> getOrganizedImportChanges(
			XtextResource resource,
			OXTypeRewritableImportSection newImportSection) {
		EObjectUsageCollector eObjectUsageCollector = eobjectUsageCollectorProvider
				.get();
		EObjectUsages eObjectUsages = eObjectUsageCollector
				.collectTypesToImport(resource);
		if (unresolvedTypeResolver != null)
			unresolvedTypeResolver.resolve(eObjectUsages, resource);
		return getOrganizedImportChanges(resource, eObjectUsages,
				newImportSection);
	}

	@SuppressWarnings("restriction")
	private List<ReplaceRegion> getOrganizedImportChanges(
			XtextResource resource, EObjectUsages eObjectUsages,
			OXTypeRewritableImportSection newImportSection) {
		for (EObject importx : eObjectUsages.getNamespaceImports()) {
			newImportSection.addFQNImport(toQualifiedName(importx));
		}
		
		return newImportSection.rewrite();
	}

	protected String toQualifiedName(EObject value) {
		return nameProvider.getFullyQualifiedName(value).toString();
	}
}