Copy/Move: Fix processing of calls used as callees

#KT-18241 Fixed
2017-06-08 17:22:55 +03:00
parent 87a41293e8
commit 6b18ff1d97
7 changed files with 63 additions and 2 deletions
@@ -562,7 +562,11 @@ class ShortenReferences(val options: (KtElement) -> Options = { Options.DEFAULT
        }

        override fun shortenElement(element: KtDotQualifiedExpression): KtElement {
-            return element.replace(element.selectorExpression!!) as KtElement
+            val parens = element.parent as? KtParenthesizedExpression
+            val requiredParens = parens != null && !KtPsiUtil.areParenthesesUseless(parens)
+            val shortenedElement = element.replace(element.selectorExpression!!) as KtElement
+            if (requiredParens) return shortenedElement.parent.replaced(shortenedElement)
+            return shortenedElement
        }
    }