From 2e5ee242c7f00def4f6a2922506ef16a3c6e15e8 Mon Sep 17 00:00:00 2001
From: Mikhail Glukhikh <mikhail.glukhikh@jetbrains.com>
Date: Tue, 9 Oct 2018 11:50:21 +0300
Subject: [PATCH] Minor refactoring: implicitVisibility

---
 .../kotlin/idea/core/psiModificationUtils.kt  | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/idea/idea-core/src/org/jetbrains/kotlin/idea/core/psiModificationUtils.kt b/idea/idea-core/src/org/jetbrains/kotlin/idea/core/psiModificationUtils.kt
index 48f895054a6..f91e8020a82 100644
--- a/idea/idea-core/src/org/jetbrains/kotlin/idea/core/psiModificationUtils.kt
+++ b/idea/idea-core/src/org/jetbrains/kotlin/idea/core/psiModificationUtils.kt
@@ -253,19 +253,14 @@ fun KtModifierListOwner.setVisibility(visibilityModifier: KtModifierKeywordToken
 }
 
 fun KtDeclaration.implicitVisibility(): KtModifierKeywordToken? {
-    if (this is KtPropertyAccessor && this.isSetter) {
-        val property = this.property
-        if (property.hasModifier(KtTokens.OVERRIDE_KEYWORD)) {
-            val visibility = (property.descriptor as? PropertyDescriptor)
-                ?.overriddenDescriptors
-                ?.firstOrNull()
-                ?.setter
-                ?.visibility
-                ?.toKeywordToken()
-            if (visibility != null) return visibility
-        }
-    }
     return when {
+        this is KtPropertyAccessor && isSetter && property.hasModifier(KtTokens.OVERRIDE_KEYWORD) -> {
+            (property.descriptor as? PropertyDescriptor)?.overriddenDescriptors?.forEach {
+                val visibility = it.setter?.visibility?.toKeywordToken()
+                if (visibility != null) return visibility
+            }
+            KtTokens.DEFAULT_VISIBILITY_KEYWORD
+        }
         this is KtConstructor<*> -> {
             val klass = getContainingClassOrObject()
             if (klass is KtClass && (klass.isEnum() || klass.isSealed())) KtTokens.PRIVATE_KEYWORD