Use "unnecessary variable" inspection in J2K

Unused variables are no more treated as unnecessary
Related to KT-15958
This commit is contained in:
Mikhail Glukhikh
2017-07-18 16:37:07 +03:00
parent 39f1ef390e
commit 44790eccaf
12 changed files with 91 additions and 58 deletions
@@ -22,14 +22,13 @@ import com.intellij.codeInspection.ProblemHighlightType
import com.intellij.codeInspection.ProblemsHolder
import com.intellij.openapi.project.Project
import com.intellij.psi.PsiElementVisitor
import com.intellij.psi.search.LocalSearchScope
import com.intellij.psi.search.searches.ReferencesSearch
import org.jetbrains.kotlin.descriptors.FunctionDescriptor
import org.jetbrains.kotlin.descriptors.VariableDescriptor
import org.jetbrains.kotlin.idea.caches.resolve.analyze
import org.jetbrains.kotlin.idea.refactoring.inline.KotlinInlineValHandler
import org.jetbrains.kotlin.psi.KtNameReferenceExpression
import org.jetbrains.kotlin.psi.KtProperty
import org.jetbrains.kotlin.psi.KtReturnExpression
import org.jetbrains.kotlin.psi.KtVisitorVoid
import org.jetbrains.kotlin.psi.*
import org.jetbrains.kotlin.psi.psiUtil.getNextSiblingIgnoringWhitespaceAndComments
import org.jetbrains.kotlin.psi.psiUtil.getParentOfType
import org.jetbrains.kotlin.resolve.BindingContext.DECLARATION_TO_DESCRIPTOR
@@ -42,45 +41,61 @@ class UnnecessaryVariableInspection : AbstractKotlinInspection() {
override fun visitProperty(property: KtProperty) {
super.visitProperty(property)
if (!property.isLocal) return
val initializer = property.initializer ?: return
val nameIdentifier = property.nameIdentifier ?: return
val status = statusFor(property) ?: return
holder.registerProblem(
nameIdentifier,
when (status) {
UnnecessaryVariableInspection.Companion.Status.RETURN_ONLY ->
"Variable used only in following return and can be inlined"
UnnecessaryVariableInspection.Companion.Status.EXACT_COPY ->
"Variable is an exact copy of another variable and can be inlined"
},
ProblemHighlightType.GENERIC_ERROR_OR_WARNING,
InlineVariableFix()
)
}
}
if (!property.isVar && initializer is KtNameReferenceExpression && property.typeReference == null) {
val context = property.analyze()
val initializerDescriptor = context[REFERENCE_TARGET, initializer]
if (initializerDescriptor is VariableDescriptor) {
if (!initializerDescriptor.isVar &&
initializerDescriptor.containingDeclaration is FunctionDescriptor) {
holder.registerProblem(
nameIdentifier,
"Variable is an exact copy of another variable and can be inlined",
ProblemHighlightType.GENERIC_ERROR_OR_WARNING,
InlineVariableFix()
)
return
}
}
}
companion object {
private enum class Status {
RETURN_ONLY,
EXACT_COPY
}
val nextStatement = property.getNextSiblingIgnoringWhitespaceAndComments()
if (nextStatement is KtReturnExpression) {
val returned = nextStatement.returnedExpression
if (returned is KtNameReferenceExpression) {
val context = nextStatement.analyze()
if (context[REFERENCE_TARGET, returned] == context[DECLARATION_TO_DESCRIPTOR, property]) {
holder.registerProblem(
nameIdentifier,
"Variable used only in following return and can be inlined",
ProblemHighlightType.GENERIC_ERROR_OR_WARNING,
InlineVariableFix()
)
}
private fun statusFor(property: KtProperty): Status? {
val enclosingElement = KtPsiUtil.getEnclosingElementForLocalDeclaration(property) ?: return null
val initializer = property.initializer ?: return null
if (!property.isVar && initializer is KtNameReferenceExpression && property.typeReference == null) {
val context = property.analyze()
val initializerDescriptor = context[REFERENCE_TARGET, initializer]
if (initializerDescriptor is VariableDescriptor) {
if (!initializerDescriptor.isVar && initializerDescriptor.containingDeclaration is FunctionDescriptor) {
if (ReferencesSearch.search(property, LocalSearchScope(enclosingElement)).findFirst() != null) {
return Status.EXACT_COPY
}
}
}
}
val nextStatement = property.getNextSiblingIgnoringWhitespaceAndComments()
if (nextStatement is KtReturnExpression) {
val returned = nextStatement.returnedExpression
if (returned is KtNameReferenceExpression) {
val context = nextStatement.analyze()
if (context[REFERENCE_TARGET, returned] == context[DECLARATION_TO_DESCRIPTOR, property]) {
return Status.RETURN_ONLY
}
}
}
return null
}
fun isActiveFor(property: KtProperty) = statusFor(property) != null
}
class InlineVariableFix : LocalQuickFix {
override fun getName() = "Inline variable"
@@ -26,7 +26,9 @@ import org.jetbrains.kotlin.idea.caches.resolve.analyze
import org.jetbrains.kotlin.idea.core.replaced
import org.jetbrains.kotlin.idea.core.setVisibility
import org.jetbrains.kotlin.idea.inspections.RedundantSamConstructorInspection
import org.jetbrains.kotlin.idea.inspections.UnnecessaryVariableInspection
import org.jetbrains.kotlin.idea.inspections.UseExpressionBodyInspection
import org.jetbrains.kotlin.idea.inspections.findExistingEditor
import org.jetbrains.kotlin.idea.intentions.*
import org.jetbrains.kotlin.idea.intentions.branchedTransformations.intentions.FoldIfToReturnAsymmetricallyIntention
import org.jetbrains.kotlin.idea.intentions.branchedTransformations.intentions.FoldIfToReturnIntention
@@ -37,6 +39,7 @@ import org.jetbrains.kotlin.idea.intentions.conventionNameCalls.ReplaceGetOrSetI
import org.jetbrains.kotlin.idea.intentions.conventionNameCalls.ReplaceGetOrSetIntention
import org.jetbrains.kotlin.idea.quickfix.RemoveModifierFix
import org.jetbrains.kotlin.idea.quickfix.RemoveUselessCastFix
import org.jetbrains.kotlin.idea.refactoring.inline.KotlinInlineValHandler
import org.jetbrains.kotlin.idea.references.mainReference
import org.jetbrains.kotlin.idea.util.application.runWriteAction
import org.jetbrains.kotlin.lexer.KtTokens
@@ -80,6 +83,7 @@ object J2KPostProcessingRegistrar {
_processings.add(RemoveRedundantSamAdaptersProcessing())
_processings.add(RemoveRedundantCastToNullableProcessing())
_processings.add(UseExpressionBodyProcessing())
_processings.add(UnnecessaryVariableProcessing())
registerIntentionBasedProcessing(FoldInitializerAndIfToElvisIntention())
@@ -269,6 +273,18 @@ object J2KPostProcessingRegistrar {
}
}
private class UnnecessaryVariableProcessing : J2kPostProcessing {
override fun createAction(element: KtElement, diagnostics: Diagnostics): (() -> Unit)? {
if (element !is KtProperty) return null
if (!UnnecessaryVariableInspection.isActiveFor(element)) return null
return {
KotlinInlineValHandler().inlineElement(element.project, element.findExistingEditor(), element)
}
}
}
private class RemoveRedundantCastToNullableProcessing : J2kPostProcessing {
override fun createAction(element: KtElement, diagnostics: Diagnostics): (() -> Unit)? {
if (element !is KtBinaryExpressionWithTypeRHS) return null
@@ -0,0 +1,6 @@
// PROBLEM: none
fun test() {
val x = 1
val <caret>y = x
}
@@ -1472,6 +1472,12 @@ public class LocalInspectionTestGenerated extends AbstractLocalInspectionTest {
doTest(fileName);
}
@TestMetadata("copyOfValUnused.kt")
public void testCopyOfValUnused() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/inspectionsLocal/unnecessaryVariable/copyOfValUnused.kt");
doTest(fileName);
}
@TestMetadata("copyOfValWithExplicitType.kt")
public void testCopyOfValWithExplicitType() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/inspectionsLocal/unnecessaryVariable/copyOfValWithExplicitType.kt");