Use convertOpt() for getting inner classes of script

2017-09-26 16:39:22 +02:00
parent a4f5a1335a
commit 61f6db87c4
1 changed files with 2 additions and 1 deletions
@@ -157,7 +157,8 @@ class KotlinScriptUClass(

    override fun getInitializers(): Array<UClassInitializer> = super.getInitializers()

-    override fun getInnerClasses(): Array<UClass> = super.getInnerClasses()
+    override fun getInnerClasses(): Array<UClass> =
+            psi.innerClasses.mapNotNull { getLanguagePlugin().convertOpt<UClass>(it, this) }.toTypedArray()

    override fun getMethods(): Array<UMethod> = psi.methods.map(this::createUMethod).toTypedArray()