Add missing FileModificationService call (EA-107395)

2017-10-06 19:21:01 +02:00
parent c2efdfe243
commit 6c7f558d36
1 changed files with 3 additions and 0 deletions
@@ -16,6 +16,7 @@

 package org.jetbrains.kotlin.idea.inspections

+import com.intellij.codeInsight.FileModificationService
 import com.intellij.codeInspection.*
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiElementVisitor
@@ -58,6 +59,7 @@ class RedundantSamConstructorInspection : AbstractKotlinInspection() {
                    override fun getName() = "Remove redundant SAM-constructor"
                    override fun getFamilyName() = name
                    override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
+                        if (!FileModificationService.getInstance().preparePsiElementForWrite(expression)) return
                        replaceSamConstructorCall(expression)
                    }
                }
@@ -69,6 +71,7 @@ class RedundantSamConstructorInspection : AbstractKotlinInspection() {
                    override fun getFamilyName() = name
                    override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
                        for (callExpression in expressions) {
+                            if (!FileModificationService.getInstance().preparePsiElementForWrite(callExpression)) return
                            replaceSamConstructorCall(callExpression)
                        }
                    }