azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

RedundantUnitExpressionInspection: make isRedundantUnit public

Browse Source
This commit is contained in:
Dmitry Gridin
2020-06-19 19:51:15 +07:00
parent 360a5bf348
commit 4ac7dc0744
1 changed files with 45 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/kotlin/idea/inspections/RedundantUnitExpressionInspection.kt
+45 -38
View File
@@ -1,5 +1,5 @@
/*
* Copyright 2000-2018 JetBrains s.r.o. and Kotlin Programming Language contributors.
* Copyright 2000-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
*/
@@ -27,45 +27,52 @@ import org.jetbrains.kotlin.types.KotlinType
import org.jetbrains.kotlin.types.typeUtil.isUnit
class RedundantUnitExpressionInspection : AbstractKotlinInspection(), CleanupLocalInspectionTool {
override fun buildVisitor(holder: ProblemsHolder, isOnTheFly: Boolean, session: LocalInspectionToolSession): PsiElementVisitor {
return referenceExpressionVisitor(fun(expression) {
if (expression.isRedundantUnit()) {
holder.registerProblem(
expression,
KotlinBundle.message("redundant.unit"),
ProblemHighlightType.LIKE_UNUSED_SYMBOL,
RemoveRedundantUnitFix()
)
}
})
}
}
private fun KtReferenceExpression.isRedundantUnit(): Boolean {
if (!isUnitLiteral()) return false
val parent = this.parent ?: return false
if (parent is KtReturnExpression) {
val expectedReturnType = parent.expectedReturnType() ?: return false
return expectedReturnType.nameIfStandardType != KotlinBuiltIns.FQ_NAMES.any.shortName() && !expectedReturnType.isMarkedNullable
}
if (parent is KtBlockExpression) {
// Do not report just 'Unit' in function literals (return@label Unit is OK even in literals)
if (parent.getParentOfType<KtFunctionLiteral>(strict = true) != null) return false
if (this == parent.lastBlockStatementOrThis()) {
val prev = this.previousStatement() ?: return true
if (prev.isUnitLiteral()) return true
val prevType = prev.resolveToCall(BodyResolveMode.FULL)?.resultingDescriptor?.returnType
if (prevType != null) {
return prevType.isUnit()
}
if (prev !is KtDeclaration) return false
if (prev !is KtFunction) return true
return parent.getParentOfTypesAndPredicate(true, KtIfExpression::class.java, KtWhenExpression::class.java) { true } == null
override fun buildVisitor(holder: ProblemsHolder, isOnTheFly: Boolean): PsiElementVisitor = referenceExpressionVisitor(fun(expression) {
if (isRedundantUnit(expression)) {
holder.registerProblem(
expression,
KotlinBundle.message("redundant.unit"),
ProblemHighlightType.LIKE_UNUSED_SYMBOL,
RemoveRedundantUnitFix()
)
}
})
companion object {
fun isRedundantUnit(referenceExpression: KtReferenceExpression): Boolean {
if (!referenceExpression.isUnitLiteral()) return false
val parent = referenceExpression.parent ?: return false
if (parent is KtReturnExpression) {
val expectedReturnType = parent.expectedReturnType() ?: return false
return expectedReturnType.nameIfStandardType != KotlinBuiltIns.FQ_NAMES.any.shortName() && !expectedReturnType.isMarkedNullable
}
if (parent is KtBlockExpression) {
// Do not report just 'Unit' in function literals (return@label Unit is OK even in literals)
if (parent.getParentOfType<KtFunctionLiteral>(strict = true) != null) return false
if (referenceExpression == parent.lastBlockStatementOrThis()) {
val prev = referenceExpression.previousStatement() ?: return true
if (prev.isUnitLiteral()) return true
val prevType = prev.resolveToCall(BodyResolveMode.FULL)?.resultingDescriptor?.returnType
if (prevType != null) {
return prevType.isUnit()
}
if (prev !is KtDeclaration) return false
if (prev !is KtFunction) return true
return parent.getParentOfTypesAndPredicate(
true,
KtIfExpression::class.java,
KtWhenExpression::class.java
) { true } == null
}
return true
}
return false
}
return true
}
return false
}
private fun KtExpression.isUnitLiteral(): Boolean =