feat(mps-sync-plugin): allow to update binding for switching a branch

Author: abstraktor

mps-sync-plugin3/build.gradle.kts

@@ -55,6 +55,7 @@ dependencies {
     testImplementation(libs.kotlin.coroutines.test)
     testImplementation(libs.logback.classic)
     testImplementation(kotlin("test"))
+    testImplementation(libs.mockk)
     testImplementation(project(":authorization"), excludeMPSLibraries)
     testImplementation(project(":model-server"), excludeMPSLibraries)
     testImplementation(libs.ktor.client.cio, excludeMPSLibraries)

mps-sync-plugin3/src/main/kotlin/org/modelix/mps/sync3/IModelSyncService.kt

@@ -36,6 +36,7 @@ interface IModelSyncService {
     fun getServerConnections(): List<IServerConnection>
     fun getUsedServerConnections(): List<IServerConnection>
     fun getBindings(): List<IBinding>
+    fun updateBinding(oldBranchRef: BranchReference, newBranchRef: BranchReference, resetLocalState: Boolean = false)
 }
 
 data class ModelServerConnectionProperties(

mps-sync-plugin3/src/main/kotlin/org/modelix/mps/sync3/ModelSyncService.kt

@@ -85,6 +85,30 @@ class ModelSyncService(val project: Project) :
         return loadedState.bindings.keys.map { Binding(it) }
     }
 
+    @Synchronized
+    override fun updateBinding(oldBranchRef: BranchReference, newBranchRef: BranchReference, resetLocalState: Boolean) {
+        updateState {
+            it.bindings.none { it.key.branchRef == oldBranchRef } &&
+                throw IllegalArgumentException("No binding for $oldBranchRef")
+
+            it.copy(
+                bindings = it.bindings.mapKeys { (key, _) ->
+                    if (key.branchRef == oldBranchRef) {
+                        key.copy(branchRef = newBranchRef)
+                    } else {
+                        key
+                    }
+                }.mapValues { (key, value) ->
+                    if (resetLocalState && key.branchRef == newBranchRef) {
+                        value.copy(versionHash = null)
+                    } else {
+                        value
+                    }
+                },
+            )
+        }
+    }
+
     @Synchronized
     fun loadState(newState: SyncServiceState) {
         val oldState: SyncServiceState = this.loadedState