| package org.eclipse.osbp.xtext.entitymock.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.datainterchange.DataDSLPackage; |
| import org.eclipse.osbp.xtext.datainterchange.DataInterchange; |
| 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 || toImport instanceof DataInterchange; |
| } |
| |
| protected boolean doShouldProposeAllElements(EObject object, EReference reference) { |
| EClass type = reference.getEReferenceType(); |
| return DataDSLPackage.Literals.DATA_INTERCHANGE.isSuperTypeOf(type) |
| || OSBPEntityPackage.Literals.LENTITY.isSuperTypeOf(type); |
| } |
| } |