1 files changed, 51 insertions, 3 deletions

diff --git a/Application/hu.bme.mit.inf.dslreasoner.application/src/hu/bme/mit/inf/dslreasoner/application/linking/ApplicationConfigurationLinkingService.xtend b/Application/hu.bme.mit.inf.dslreasoner.application/src/hu/bme/mit/inf/dslreasoner/application/linking/ApplicationConfigurationLinkingService.xtend
index 15567f88..35d4008b 100644
--- a/Application/hu.bme.mit.inf.dslreasoner.application/src/hu/bme/mit/inf/dslreasoner/application/linking/ApplicationConfigurationLinkingService.xtend
+++ b/Application/hu.bme.mit.inf.dslreasoner.application/src/hu/bme/mit/inf/dslreasoner/application/linking/ApplicationConfigurationLinkingService.xtend
@@ -6,6 +6,7 @@ import hu.bme.mit.inf.dslreasoner.application.applicationConfiguration.EPackageI
 import hu.bme.mit.inf.dslreasoner.application.applicationConfiguration.ViatraImport

 import java.util.Collections

 import java.util.Optional

+import org.eclipse.emf.common.util.URI

 import org.eclipse.emf.ecore.EObject

 import org.eclipse.emf.ecore.EReference

 import org.eclipse.viatra.query.patternlanguage.emf.scoping.IMetamodelProvider

@@ -14,6 +15,13 @@ import org.eclipse.xtext.conversion.ValueConverterException
 import org.eclipse.xtext.linking.impl.DefaultLinkingService

 import org.eclipse.xtext.nodemodel.ILeafNode

 import org.eclipse.xtext.nodemodel.INode

+import org.eclipse.emf.ecore.resource.Resource

+import hu.bme.mit.inf.dslreasoner.application.applicationConfiguration.PatternEntry

+import org.eclipse.emf.ecore.util.EcoreUtil

+import hu.bme.mit.inf.dslreasoner.application.applicationConfiguration.ConfigurationScript

+import org.eclipse.xtext.EcoreUtil2

+import java.util.List

+import org.eclipse.xtext.conversion.impl.QualifiedNameValueConverter

 

 class ApplicationConfigurationLinkingService extends DefaultLinkingService{

 

@@ -31,20 +39,60 @@ class ApplicationConfigurationLinkingService extends DefaultLinkingService{
                         return getEPackage(context as EPackageImport, node as ILeafNode)

                 }

         } else if(context instanceof ViatraImport) {

-                super.getLinkedObjects(context, ref, node)

+                if(ref == viatraImport_ImportedViatra && node instanceof ILeafNode) {

+                        return getViatra(context as ViatraImport, node as ILeafNode)

+                }

+        } else if(context instanceof PatternEntry) {

+                if(ref === patternEntry_Package) {

+                        return getViatraPackage(context as PatternEntry,node)

+                }

         }

         return super.getLinkedObjects(context, ref, node)

     }

+        

+        def getViatraPackage(PatternEntry entry, INode node) {

+                val document = EcoreUtil2.getContainerOfType(entry,ConfigurationScript)

+                val nodeString = valueConverterService.toValue(node.text,

+                linkingHelper.getRuleNameFrom(node.grammarElement), node).toString.replaceAll("\\s","")

+                val patternModels = document.imports.filter(ViatraImport).map[it.importedViatra].filterNull

+                val List<EObject> patternModelsWithSameNamespace = patternModels.filter[nodeString.equals(it.packageName)].filter(EObject).toList

+                return patternModelsWithSameNamespace

+        }

 

     private def getEPackage(EPackageImport packageImport, ILeafNode node) {

-        getMetamodelNsUri(node).flatMap [ uri |

+        getNSUri(node).flatMap [ uri |

             Optional.ofNullable(metamodelProvider.loadEPackage(uri, packageImport.eResource.resourceSet))

         ].map [ ePackage |

             Collections.singletonList(ePackage as EObject)

         ].orElse(emptyList)

     }

+    

+    private def getViatra(ViatraImport viatraImport, ILeafNode node) {

+        val uri = getNSUri(node)

+        if(uri.present) {

+                var URI createdURI

+                try{

+                        createdURI = URI.createURI(uri.get)

+                }catch(IllegalArgumentException e) {

+                        return super.getLinkedObjects(viatraImport, viatraImport_ImportedViatra, node)

+                }

+                var Resource res

+                try{

+                        res = viatraImport.eResource.resourceSet.getResource(createdURI,true);

+                } catch(RuntimeException  e){

+                        return super.getLinkedObjects(viatraImport, viatraImport_ImportedViatra, node)

+                }

+                if(res!==null && !res.contents.empty) {

+                        return #[res.contents.head]

+                } else {

+                        return super.getLinkedObjects(viatraImport, viatraImport_ImportedViatra, node)

+                }

+        } else {

+                return super.getLinkedObjects(viatraImport, viatraImport_ImportedViatra, node)

+        }

+    }

 

-    private def getMetamodelNsUri(ILeafNode node) {

+    private def getNSUri(ILeafNode node) {

         try {

             val convertedValue = valueConverterService.toValue(node.text,

                 linkingHelper.getRuleNameFrom(node.grammarElement), node)