1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
// Copyright 2000-2021 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package com.intellij.codeInspection.incorrectFormatting
import com.intellij.codeInsight.intention.preview.IntentionPreviewInfo.EMPTY
import com.intellij.codeInspection.LocalQuickFix
import com.intellij.codeInspection.ProblemDescriptor
import com.intellij.lang.LangBundle
import com.intellij.openapi.editor.RangeMarker
import com.intellij.openapi.project.Project
import com.intellij.profile.codeInspection.InspectionProjectProfileManager
import com.intellij.psi.PsiDocumentManager
import com.intellij.psi.PsiFile
import com.intellij.psi.codeStyle.CodeStyleManager
import org.jetbrains.annotations.Nls
import java.util.concurrent.atomic.AtomicBoolean
class ReplaceQuickFix(val replacements: List<Pair<RangeMarker, String>>) : LocalQuickFix {
override fun getFamilyName() = LangBundle.message("inspection.incorrect.formatting.fix.replace")
override fun getFileModifierForPreview(target: PsiFile) = ReplaceQuickFix(replacements)
private val applied = AtomicBoolean(false)
override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
if (!applied.compareAndSet(false, true)) return
descriptor
.psiElement
.containingFile
?.viewProvider
?.document
?.let { doc ->
replacements
.sortedByDescending { (range, _) -> range.startOffset }
.forEach { (range, replacement) ->
if (range.isValid) {
doc.replaceString(range.startOffset, range.endOffset, replacement)
}
}
PsiDocumentManager.getInstance(project).commitDocument(doc)
}
}
}
object ReformatQuickFix : LocalQuickFix {
override fun getFamilyName() = LangBundle.message("inspection.incorrect.formatting.fix.reformat")
override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
val innerFile = descriptor.psiElement.containingFile
val file = innerFile.viewProvider.run { getPsi(baseLanguage) }
CodeStyleManager.getInstance(project).reformatText(file, 0, file.textLength)
}
}
abstract class ReconfigureQuickFix(@Nls val family: String, val reconfigure: IncorrectFormattingInspection.() -> Unit) : LocalQuickFix {
override fun getFamilyName() = family
override fun startInWriteAction() = false
override fun generatePreview(project: Project, previewDescriptor: ProblemDescriptor) = EMPTY
override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
val file = descriptor.psiElement.containingFile
InspectionProjectProfileManager
.getInstance(project)
.currentProfile
.modifyToolSettings(INSPECTION_KEY, file) { inspection ->
inspection.reconfigure()
}
}
}
object ShowDetailedReportIntention : ReconfigureQuickFix(
LangBundle.message("inspection.incorrect.formatting.fix.show.details"),
{ reportPerFile = false }
)
object HideDetailedReportIntention : ReconfigureQuickFix(
LangBundle.message("inspection.incorrect.formatting.fix.hide.details"),
{ reportPerFile = true }
)
|