From 555e85aa6c7f13a9fdaaf2cba053345c11f3f625 Mon Sep 17 00:00:00 2001
From: Tagir Valeev <tagir.valeev@jetbrains.com>
Date: Mon, 25 Jul 2022 14:46:52 +0200
Subject: [PATCH] [intention-preview] PsiEditorUtilBase#findEditorByPsiElement;
 PsiEditorUtil#findEditor: fix for preview

GitOrigin-RevId: ed19a469439eb75c277ebc80b4482f386a9188d7
---
 .../lang-api/src/com/intellij/psi/util/PsiEditorUtilBase.kt   | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/platform/lang-api/src/com/intellij/psi/util/PsiEditorUtilBase.kt b/platform/lang-api/src/com/intellij/psi/util/PsiEditorUtilBase.kt
index 35d9f04731f4..ad26319a8d99 100644
--- a/platform/lang-api/src/com/intellij/psi/util/PsiEditorUtilBase.kt
+++ b/platform/lang-api/src/com/intellij/psi/util/PsiEditorUtilBase.kt
@@ -1,6 +1,7 @@
 // Copyright 2000-2019 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
 package com.intellij.psi.util
 
+import com.intellij.codeInsight.intention.preview.IntentionPreviewUtils
 import com.intellij.ide.DataManager
 import com.intellij.openapi.actionSystem.CommonDataKeys
 import com.intellij.openapi.diagnostic.logger
@@ -36,6 +37,9 @@ object PsiEditorUtilBase : PsiEditorUtil {
    */
   override fun findEditorByPsiElement(element: PsiElement): Editor? {
     val psiFile = element.containingFile
+    if (IntentionPreviewUtils.isPreviewElement(psiFile)) {
+      return IntentionPreviewUtils.getPreviewEditor()
+    }
     val virtualFile = PsiUtilCore.getVirtualFile(element) ?: return null
     val project = psiFile.project
     if (virtualFile.isInLocalFileSystem || virtualFile.fileSystem is NonPhysicalFileSystem) { // Try to find editor for the real file.