feat(mps-model-adapters): migrate dependency search functions

Co-authored-by: Benedek Horváth <[email protected]>

Author: mhuster23

mps-model-adapters/src/main/kotlin/org/modelix/model/mpsadapters/MPSModelAsNode.kt

@@ -17,6 +17,7 @@ import jetbrains.mps.extapi.model.SModelDescriptorStub
 import jetbrains.mps.smodel.ModelImports
 import jetbrains.mps.smodel.adapter.structure.MetaAdapterFactory
 import org.jetbrains.mps.openapi.model.SModel
+import org.jetbrains.mps.openapi.module.SModuleId
 import org.modelix.model.api.BuiltinLanguages
 import org.modelix.model.api.IChildLink
 import org.modelix.model.api.IConcept
@@ -117,4 +118,30 @@ data class MPSModelAsNode(val model: SModel) : IDefaultNodeAdapter {
             null
         }
     }
+
+    internal fun findSingleLanguageDependency(dependencyId: SModuleId): MPSSingleLanguageDependencyAsNode? {
+        if (model is SModelDescriptorStub) {
+            model.importedLanguageIds().forEach { entry ->
+                if (entry.sourceModule?.moduleId == dependencyId) {
+                    return MPSSingleLanguageDependencyAsNode(
+                        entry.sourceModuleReference,
+                        model.getLanguageImportVersion(entry),
+                        modelImporter = model,
+                    )
+                }
+            }
+        }
+        return null
+    }
+
+    internal fun findDevKitDependency(dependencyId: SModuleId): MPSDevKitDependencyAsNode? {
+        if (model is SModelDescriptorStub) {
+            model.importedDevkits().forEach { devKit ->
+                if (devKit.moduleId == dependencyId) {
+                    return MPSDevKitDependencyAsNode(devKit, modelImporter = model)
+                }
+            }
+        }
+        return null
+    }
 }

mps-model-adapters/src/main/kotlin/org/modelix/model/mpsadapters/MPSModuleAsNode.kt

@@ -154,4 +154,48 @@ data class MPSModuleAsNode(val module: SModule) : IDefaultNodeAdapter {
             false
         }
     }
+
+    internal fun findModuleDependency(dependencyId: SModuleId): MPSModuleDependencyAsNode? {
+        if (module !is AbstractModule) {
+            return null
+        }
+
+        module.moduleDescriptor?.dependencyVersions?.forEach { entry ->
+            if (entry.key.moduleId == dependencyId) {
+                return MPSModuleDependencyAsNode(
+                    moduleReference = entry.key,
+                    moduleVersion = entry.value,
+                    explicit = isDirectDependency(module, entry.key.moduleId),
+                    reexport = isReexport(module, entry.key.moduleId),
+                    importer = module,
+                    dependencyScope = getDependencyScope(module, entry.key.moduleId),
+                )
+            }
+        }
+        return null
+    }
+
+    internal fun findSingleLanguageDependency(dependencyId: SModuleId): MPSSingleLanguageDependencyAsNode? {
+        if (module !is AbstractModule) {
+            return null
+        }
+        module.moduleDescriptor?.dependencyVersions?.forEach { entry ->
+            if (entry.key.moduleId == dependencyId) {
+                return MPSSingleLanguageDependencyAsNode(entry.key, entry.value, moduleImporter = module)
+            }
+        }
+        return null
+    }
+
+    internal fun findDevKitDependency(dependencyId: SModuleId): MPSDevKitDependencyAsNode? {
+        if (module !is AbstractModule) {
+            return null
+        }
+        module.moduleDescriptor?.usedDevkits?.forEach { devKit ->
+            if (devKit.moduleId == dependencyId) {
+                return MPSDevKitDependencyAsNode(devKit, module)
+            }
+        }
+        return null
+    }
 }