File tree 1 file changed +3
-3
lines changed
marker/jvm-marker/src/main/kotlin/spp/jetbrains/marker/jvm/service
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -68,7 +68,7 @@ class JVMArtifactScopeService : IArtifactScopeService {
68
68
override fun getFunctions (element : PsiElement , includeInnerClasses : Boolean ): List <PsiNamedElement > {
69
69
return when {
70
70
ArtifactTypeService .isKotlin(element) -> element.descendantsOfType<KtNamedFunction >().filter {
71
- if (element is KtClass && ! includeInnerClasses) {
71
+ if (element is KtClassOrObject && ! includeInnerClasses) {
72
72
element == it.containingClass()
73
73
} else true
74
74
}.toList()
@@ -89,7 +89,7 @@ class JVMArtifactScopeService : IArtifactScopeService {
89
89
90
90
override fun getClasses (element : PsiElement ): List <PsiNamedElement > {
91
91
return when {
92
- ArtifactTypeService .isKotlin(element) -> element.descendantsOfType<KtClass >().toList()
92
+ ArtifactTypeService .isKotlin(element) -> element.descendantsOfType<KtClassOrObject >().toList()
93
93
else -> element.descendantsOfType<PsiClass >().toList()
94
94
}
95
95
}
@@ -117,7 +117,7 @@ class JVMArtifactScopeService : IArtifactScopeService {
117
117
118
118
override fun getParentClass (element : PsiElement ): PsiNamedElement ? {
119
119
return when {
120
- ArtifactTypeService .isKotlin(element) -> element.findParentOfType<KtClass >()
120
+ ArtifactTypeService .isKotlin(element) -> element.findParentOfType<KtClassOrObject >()
121
121
else -> element.findParentOfType<PsiClass >()
122
122
}
123
123
}
You can’t perform that action at this time.
0 commit comments