From 0f69f459a648f2139f62ea2aeb61f042eae234a3 Mon Sep 17 00:00:00 2001
From: Sergei Tachenov <sergei.tachenov@jetbrains.com>
Date: Mon, 5 May 2025 13:19:23 +0300
Subject: [PATCH] IJPL-186724 Support provided select-in contexts in
 SelectInProjectViewImpl

This is a rare case, but it's still a valid case
when the editor provides a special kind of select-in
context, so neither SimpleSelectInContext nor
EditorSelectInContext work properly.

An example is MPS-37563.

GitOrigin-RevId: 8ec91cb35049ebba3e18eb5ec401ea4ea348bcd2
---
 .../impl/SelectInProjectViewImpl.kt           | 47 ++++++++++++++-----
 1 file changed, 34 insertions(+), 13 deletions(-)

diff --git a/platform/lang-impl/src/com/intellij/ide/projectView/impl/SelectInProjectViewImpl.kt b/platform/lang-impl/src/com/intellij/ide/projectView/impl/SelectInProjectViewImpl.kt
index 383cca9c2b69..99cdcc35d34b 100644
--- a/platform/lang-impl/src/com/intellij/ide/projectView/impl/SelectInProjectViewImpl.kt
+++ b/platform/lang-impl/src/com/intellij/ide/projectView/impl/SelectInProjectViewImpl.kt
@@ -116,7 +116,7 @@ internal class SelectInProjectViewImpl(
           if (LOG.isDebugEnabled) {
             LOG.debug("Created select-in context and delegating to it: $selectInContext")
           }
-          selectInContext.selectInCurrentTarget(requestFocus = invokedManually)
+          selectInContext.selectInCurrentTarget(project, requestFocus = invokedManually)
         }
         break
       }
@@ -155,13 +155,21 @@ internal class SelectInProjectViewImpl(
       }
     }
 
-  private fun createSelectInContext(psiFilePointer: SmartPsiElementPointer<PsiFile>, fileEditor: FileEditor): SimpleSelectInContext =
-    if (fileEditor is TextEditor) {
-      EditorSelectInContext(project, psiFilePointer, fileEditor)
-    }
-    else {
-      SimpleSelectInContext(project, psiFilePointer, fileEditor)
+  private fun createSelectInContext(psiFilePointer: SmartPsiElementPointer<PsiFile>, fileEditor: FileEditor): SelectInContext {
+    val dataContext = DataManager.getInstance().getDataContext(fileEditor.component)
+    val providedSelectInContext = dataContext.getData(SelectInContext.DATA_KEY)
+    return when {
+      providedSelectInContext != null -> {
+        providedSelectInContext
+      }
+      fileEditor is TextEditor -> {
+        EditorSelectInContext(project, psiFilePointer, fileEditor)
+      }
+      else -> {
+        SimpleSelectInContext(project, psiFilePointer, fileEditor)
+      }
     }
+  }
 
   fun ensureSelected(
     paneId: String,
@@ -335,6 +343,24 @@ internal class SelectInProjectViewImpl(
 
 }
 
+private suspend fun SelectInContext.selectInCurrentTarget(project: Project, requestFocus: Boolean) {
+  if (this is SimpleSelectInContext) {
+    selectInCurrentTarget(requestFocus) // simple, but overridable
+  }
+  else {
+    simpleSelectInCurrentTarget(project, requestFocus)
+  }
+}
+
+private fun SelectInContext.simpleSelectInCurrentTarget(project: Project, requestFocus: Boolean) {
+  val currentTarget = (project.serviceOrNull<ProjectView>() as ProjectViewImpl?)?.currentSelectInTarget
+  if (LOG.isDebugEnabled) {
+    LOG.debug("The current target is $currentTarget")
+  }
+  if (currentTarget == null) return
+  currentTarget.selectIn(this, requestFocus)
+}
+
 private open class SimpleSelectInContext(
   private val project: Project,
   psiFilePointer: SmartPsiElementPointer<PsiFile>,
@@ -346,12 +372,7 @@ private open class SimpleSelectInContext(
   }
 
   open suspend fun selectInCurrentTarget(requestFocus: Boolean) {
-    val currentTarget = (project.serviceOrNull<ProjectView>() as ProjectViewImpl?)?.currentSelectInTarget
-    if (LOG.isDebugEnabled) {
-      LOG.debug("The current target is $currentTarget")
-    }
-    if (currentTarget == null) return
-    currentTarget.selectIn(this, requestFocus)
+    simpleSelectInCurrentTarget(project, requestFocus)
   }
 
   override fun toString(): String {