feat: new method runWithProject

Author: slisson

api/src/main/kotlin/org/modelix/mps/api/ContextProject.kt

@@ -0,0 +1,19 @@
+package org.modelix.mps.api
+
+import org.jetbrains.mps.openapi.project.Project
+
+object ContextProject {
+    private val threadLocal = ThreadLocal<Project>()
+
+    fun <R> runWith(project: Project, body: () -> R): R {
+        val oldValue = threadLocal.get()
+        try {
+            threadLocal.set(project)
+            return body()
+        } finally {
+            threadLocal.set(oldValue)
+        }
+    }
+
+    fun getProject(): Project? = threadLocal.get()
+}

api/src/main/kotlin/org/modelix/mps/api/IModelixMpsApi.kt

@@ -31,4 +31,10 @@ interface IModelixMpsApi {
     fun setReference(node: SNode, link: SReferenceLink, target: SNodeReference)
 
     fun forceSave(model: SModel)
+
+    fun getIdeaProject(project: Project): com.intellij.openapi.project.Project
+    fun getMPSProject(project: com.intellij.openapi.project.Project): Project
+    fun <R> runWithProject(project: Project, body: () -> R): R =
+        ContextProject.runWith(project, body)
+    fun <R> runWithProject(project: com.intellij.openapi.project.Project, body: () -> R): R = runWithProject(getMPSProject(project), body)
 }

impl203/src/main/kotlin/org/modelix/mps/api/ModelixMpsApiImpl203.kt

@@ -23,7 +23,7 @@ import java.awt.Component
 
 open class ModelixMpsApiImpl203 : IModelixMpsApi {
     override fun getMPSProjects(): List<Project> {
-        return ProjectManager.getInstance().openedProjects
+        return (listOfNotNull(ContextProject.getProject()) + ProjectManager.getInstance().openedProjects).distinct()
     }
 
     override fun getVirtualFolder(
@@ -48,7 +48,7 @@ open class ModelixMpsApiImpl203 : IModelixMpsApi {
     }
 
     override fun getRepository(project: com.intellij.openapi.project.Project): SRepository {
-        return getRepository(checkNotNull(ProjectHelper.fromIdeaProject(project)) { "No MPS project found in $project" })
+        return getRepository(getMPSProject(project))
     }
 
     override fun getRepository(editorContext: EditorContext): SRepository {
@@ -76,4 +76,14 @@ open class ModelixMpsApiImpl203 : IModelixMpsApi {
         if (model !is EditableSModel) return
         model.save(SaveOptions.FORCE)
     }
+
+    override fun getIdeaProject(project: Project): com.intellij.openapi.project.Project {
+        return ProjectHelper.toIdeaProject(project as jetbrains.mps.project.Project)
+    }
+
+    override fun getMPSProject(project: com.intellij.openapi.project.Project): Project {
+        return checkNotNull(ProjectHelper.fromIdeaProject(project)) {
+            "No MPS project found in $project"
+        }
+    }
 }

impl223/src/main/kotlin/org/modelix/mps/api/ModelixMpsApiImpl223.kt

@@ -7,6 +7,6 @@ import org.jetbrains.mps.openapi.project.Project
 open class ModelixMpsApiImpl223 : ModelixMpsApiImpl222() {
     override fun getMPSProjects(): List<Project> {
         val manager = MPSCoreComponents.getInstance()?.platform?.findComponent(ProjectManager::class.java) ?: return emptyList()
-        return manager.openedProjects
+        return (listOfNotNull(ContextProject.getProject()) + manager.openedProjects).distinct()
     }
 }

lib/api/lib.api

@@ -3,6 +3,8 @@ public final class org/modelix/mps/api/ModelixMpsApi : org/modelix/mps/api/IMode
 	public fun fixVersions (Lorg/jetbrains/mps/openapi/project/Project;Lorg/jetbrains/mps/openapi/module/SModule;)V
 	public fun forceSave (Lorg/jetbrains/mps/openapi/model/SModel;)V
 	public fun getGlobalRepository ()Lorg/jetbrains/mps/openapi/module/SRepository;
+	public fun getIdeaProject (Lorg/jetbrains/mps/openapi/project/Project;)Lcom/intellij/openapi/project/Project;
+	public fun getMPSProject (Lcom/intellij/openapi/project/Project;)Lorg/jetbrains/mps/openapi/project/Project;
 	public fun getMPSProjects ()Ljava/util/List;
 	public fun getProjectRepository ()Lorg/jetbrains/mps/openapi/module/SRepository;
 	public fun getRepository ()Lorg/jetbrains/mps/openapi/module/SRepository;
@@ -13,6 +15,8 @@ public final class org/modelix/mps/api/ModelixMpsApi : org/modelix/mps/api/IMode
 	public fun getVirtualFolder (Lorg/jetbrains/mps/openapi/module/SModule;)Ljava/lang/String;
 	public fun getVirtualFolder (Lorg/jetbrains/mps/openapi/project/Project;Lorg/jetbrains/mps/openapi/module/SModule;)Ljava/lang/String;
 	public fun getVirtualFolders (Lorg/jetbrains/mps/openapi/module/SModule;)Ljava/util/List;
+	public fun runWithProject (Lcom/intellij/openapi/project/Project;Lkotlin/jvm/functions/Function0;)Ljava/lang/Object;
+	public fun runWithProject (Lorg/jetbrains/mps/openapi/project/Project;Lkotlin/jvm/functions/Function0;)Ljava/lang/Object;
 	public fun setReference (Lorg/jetbrains/mps/openapi/model/SNode;Lorg/jetbrains/mps/openapi/language/SReferenceLink;Lorg/jetbrains/mps/openapi/model/SNodeReference;)V
 }