Don't increase nesting level if we aren't processing current expression

This commit is contained in:
Dmitry Jemerov
2018-01-11 18:33:11 +01:00
parent 3a7e4acf22
commit 560dc920e4
@@ -349,12 +349,14 @@ class ShortenReferences(val options: (KtElement) -> Options = { Options.DEFAULT
if (filterResult == FilterResult.PROCESS) { if (filterResult == FilterResult.PROCESS) {
addQualifiedElementToAnalyze(userType) addQualifiedElementToAnalyze(userType)
nextLevel()
} }
// elements in qualifier must be under // elements in qualifier must be under
nextLevel()
userType.qualifier?.accept(this) userType.qualifier?.accept(this)
prevLevel() if (filterResult == FilterResult.PROCESS) {
prevLevel()
}
} }
} }
@@ -400,12 +402,14 @@ class ShortenReferences(val options: (KtElement) -> Options = { Options.DEFAULT
if (filterResult == FilterResult.PROCESS) { if (filterResult == FilterResult.PROCESS) {
addQualifiedElementToAnalyze(expression) addQualifiedElementToAnalyze(expression)
nextLevel()
} }
// elements in receiver must be under // elements in receiver must be under
nextLevel()
expression.receiverExpression.accept(this) expression.receiverExpression.accept(this)
prevLevel() if (filterResult == FilterResult.PROCESS) {
prevLevel()
}
} }
} }