org.eclipse.osbp.xtext.cubedsl/src/org/eclipse/osbp/xtext/cubedsl/imports/ShouldImportProvider.java

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

import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EReference;
import org.eclipse.osbp.dsl.semantic.entity.LEntity;
import org.eclipse.osbp.dsl.semantic.entity.OSBPEntityPackage;
import org.eclipse.osbp.xtext.oxtype.imports.DefaultShouldImportProvider;

public class ShouldImportProvider extends DefaultShouldImportProvider {

	protected boolean doShouldImport(EObject toImport, EReference eRef,
			EObject context) {
		return toImport instanceof LEntity;
	}
	
	protected boolean doShouldProposeAllElements(EObject object, EReference reference) {
		EClass type = reference.getEReferenceType();
		return OSBPEntityPackage.Literals.LENTITY.isSuperTypeOf(type);
	}
}