Simplify call chain fix: extract 'apply'

This commit is contained in:
Mikhail Glukhikh
2018-11-28 15:58:43 +03:00
parent c4717d17d9
commit 138e36aa66
@@ -34,8 +34,7 @@ class SimplifyCallChainFix(private val newCallText: String) : LocalQuickFix {
override fun getFamilyName() = name override fun getFamilyName() = name
override fun applyFix(project: Project, descriptor: ProblemDescriptor) { private fun apply(secondQualifiedExpression: KtQualifiedExpression) {
(descriptor.psiElement as? KtQualifiedExpression)?.let { secondQualifiedExpression ->
val factory = KtPsiFactory(secondQualifiedExpression) val factory = KtPsiFactory(secondQualifiedExpression)
val firstExpression = secondQualifiedExpression.receiverExpression val firstExpression = secondQualifiedExpression.receiverExpression
@@ -100,5 +99,8 @@ class SimplifyCallChainFix(private val newCallText: String) : LocalQuickFix {
val result = secondQualifiedExpression.replaced(newQualifiedExpression) val result = secondQualifiedExpression.replaced(newQualifiedExpression)
ShortenReferences.DEFAULT.process(result) ShortenReferences.DEFAULT.process(result)
} }
override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
(descriptor.psiElement as? KtQualifiedExpression)?.let(this::apply)
} }
} }