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

/**
 * Copyright (c) 2011, 2015 - Lunifera GmbH (Gross Enzersdorf, Austria), Loetz GmbH&Co.KG (69115 Heidelberg, Germany)
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0 
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *         Florian Pirchner - Initial implementation
 *
 */
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();
	}
}