From fec8397886fe536aabf6c4d3b683fcc86ea9d367 Mon Sep 17 00:00:00 2001
From: Tagir Valeev <Tagir.Valeev@jetbrains.com>
Date: Tue, 21 Apr 2020 15:00:39 +0700
Subject: [PATCH] ReplaceMethodCallFix preview-friendly

GitOrigin-RevId: 0031f1a08a2412ad6997be3ce138a0a51c86870e
---
 .../deprecation/deprecationInspectionQuickFixes.kt        | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/java/java-analysis-impl/src/com/intellij/codeInspection/deprecation/deprecationInspectionQuickFixes.kt b/java/java-analysis-impl/src/com/intellij/codeInspection/deprecation/deprecationInspectionQuickFixes.kt
index c45efebc3c8e..b2760f7103f5 100644
--- a/java/java-analysis-impl/src/com/intellij/codeInspection/deprecation/deprecationInspectionQuickFixes.kt
+++ b/java/java-analysis-impl/src/com/intellij/codeInspection/deprecation/deprecationInspectionQuickFixes.kt
@@ -15,6 +15,7 @@
  */
 package com.intellij.codeInspection.deprecation
 
+import com.intellij.codeInsight.intention.FileModifier
 import com.intellij.codeInspection.LocalQuickFixOnPsiElement
 import com.intellij.openapi.project.Project
 import com.intellij.psi.*
@@ -66,6 +67,13 @@ internal class ReplaceMethodCallFix(expr: PsiMethodCallExpression, replacementMe
     val replaced = expr.replace(newMethodCall) as PsiMethodCallExpression
     JavaCodeStyleManager.getInstance(project).shortenClassReferences(replaced.methodExpression)
   }
+
+  override fun getFileModifierForPreview(target: PsiFile): FileModifier? {
+    val method = myReplacementMethodPointer.element
+    val expr = startElement as PsiMethodCallExpression?
+    if (method == null || expr == null) return null
+    return ReplaceMethodCallFix(expr, method)
+  }
 }
 
 internal class ReplaceFieldReferenceFix(expr: PsiReferenceExpression, replacementField: PsiField) : LocalQuickFixOnPsiElement(expr) {