Home

Contribute

Source code


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

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();
	}
}