feat(mps-model-adapters): implement MPSNode::replaceNode

Author: mhuster23

mps-model-adapters-plugin/src/test/kotlin/org/modelix/model/mpsadapters/ReplaceNodeTest.kt

@@ -0,0 +1,51 @@
+/*
+ * Copyright (c) 2024.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.modelix.model.mpsadapters
+
+import org.modelix.model.api.BuiltinLanguages
+import org.modelix.model.api.ConceptReference
+import org.modelix.model.api.INode
+
+class ReplaceNodeTest : MpsAdaptersTestBase("SimpleProject") {
+
+    fun testReplaceNode() {
+        readAction {
+            assertEquals(1, mpsProject.projectModules.size)
+        }
+
+        val repositoryNode: INode = MPSRepositoryAsNode(mpsProject.repository)
+
+        writeActionOnEdt {
+            val module = repositoryNode.getChildren(BuiltinLanguages.MPSRepositoryConcepts.Repository.modules)
+                .single { it.getPropertyValue(BuiltinLanguages.jetbrains_mps_lang_core.INamedConcept.name) == "Solution1" }
+            val model = module.getChildren(BuiltinLanguages.MPSRepositoryConcepts.Module.models).single()
+            val rootNode = model.getChildren(BuiltinLanguages.MPSRepositoryConcepts.Model.rootNodes).single()
+
+            val oldProperties = rootNode.getAllProperties().toSet()
+            val oldReferences = rootNode.getAllReferenceTargetRefs().toSet()
+            val oldChildren = rootNode.allChildren.toList()
+
+            val newConcept = ConceptReference("mps:f3061a53-9226-4cc5-a443-f952ceaf5816/1083245097125")
+            val newNode = rootNode.replaceNode(newConcept)
+
+            assertEquals(oldProperties, newNode.getAllProperties().toSet())
+            assertEquals(oldReferences, newNode.getAllReferenceTargetRefs().toSet())
+            assertEquals(oldChildren, newNode.allChildren.toList())
+            assertEquals(newConcept, newNode.getConceptReference())
+        }
+    }
+}

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

@@ -16,9 +16,11 @@ package org.modelix.model.mpsadapters
 import jetbrains.mps.lang.smodel.generator.smodelAdapter.SNodeOperations
 import jetbrains.mps.smodel.MPSModuleRepository
 import jetbrains.mps.smodel.adapter.MetaAdapterByDeclaration
+import jetbrains.mps.smodel.adapter.ids.SConceptId
 import jetbrains.mps.smodel.adapter.ids.SContainmentLinkId
 import jetbrains.mps.smodel.adapter.ids.SPropertyId
 import jetbrains.mps.smodel.adapter.ids.SReferenceLinkId
+import jetbrains.mps.smodel.adapter.structure.concept.SConceptAdapterById
 import jetbrains.mps.smodel.adapter.structure.link.SContainmentLinkAdapterById
 import jetbrains.mps.smodel.adapter.structure.property.SPropertyAdapterById
 import jetbrains.mps.smodel.adapter.structure.ref.SReferenceLinkAdapterById
@@ -71,6 +73,23 @@ data class MPSNode(val node: SNode) : IDefaultNodeAdapter {
             return node.children.map { MPSNode(it) }
         }
 
+    override fun replaceNode(concept: ConceptReference): INode {
+        val id = node.nodeId
+        val model = checkNotNull(node.model) { "Node is not part of a model" }
+        val newNode = model.createNode(SConceptAdapterById(SConceptId.deserialize(concept.uid), ""), id)
+        node.properties.forEach { newNode.setProperty(it, node.getProperty(it)) }
+        node.references.forEach { newNode.setReference(it.link, it.targetNodeReference) }
+        node.children.forEach { child ->
+            val link = checkNotNull(child.containmentLink) { "Containment link of child node not found" }
+            newNode.addChild(link, child)
+        }
+
+        val parent = checkNotNull(node.parent) { "Cannot replace node without a parent" }
+        parent.insertChildBefore(getMPSContainmentLink(getContainmentLink()), node, newNode)
+        node.delete()
+        return MPSNode(newNode)
+    }
+
     override fun removeChild(child: INode) {
         require(child is MPSNode) { "child must be an MPSNode" }
         node.removeChild(child.node)