[kotlin] Temporary solution to allow using KotlinPsiUpdateModCommandAction in batches until IJPL-193738 is fixed

^KTIJ-34667 Fixed

GitOrigin-RevId: 9d337555690e3bc962f96bf614ec0357cb0f14e9
This commit is contained in:
Victoria.Petrakovich
2025-06-26 22:18:34 +00:00
committed by intellij-monorepo-bot
parent a8589105b0
commit ffa64d155b
@@ -8,6 +8,8 @@ import com.intellij.modcommand.*
import com.intellij.openapi.progress.ProcessCanceledException
import com.intellij.psi.PsiElement
import com.intellij.util.concurrency.annotations.RequiresBackgroundThread
import org.jetbrains.kotlin.analysis.api.permissions.KaAllowAnalysisOnEdt
import org.jetbrains.kotlin.analysis.api.permissions.allowAnalysisOnEdt
import org.jetbrains.kotlin.idea.codeinsight.api.applicable.ContextProvider
import org.jetbrains.kotlin.idea.codeinsight.api.applicable.getElementContext
import org.jetbrains.kotlin.psi.KtElement
@@ -89,9 +91,12 @@ sealed class KotlinPsiUpdateModCommandAction<E : PsiElement, C : Any>(
) : KotlinPsiUpdateModCommandAction<E, C>(null, elementClass),
ContextProvider<E, C> {
@OptIn(KaAllowAnalysisOnEdt::class)
final override fun getElementContext(
actionContext: ActionContext,
element: E,
): C? = getElementContext(element)
): C? = allowAnalysisOnEdt { // TODO: remove this workaround when IJPL-193738 is fixed.
getElementContext(element)
}
}
}