More correct + reused code

2015-09-15 20:02:28 +03:00
parent 7caccd7578
commit de3eeb3a4f
4 changed files with 11 additions and 11 deletions
@@ -402,3 +402,6 @@ public fun JetDeclaration.visibilityModifier(): PsiElement? {
                   .map { modifierList.getModifier(it as JetModifierKeywordToken) }
                   .firstOrNull { it != null }
 }
+
+public fun JetDeclaration.visibilityModifierType(): JetModifierKeywordToken?
+        = visibilityModifier()?.node?.elementType as JetModifierKeywordToken?