Add utility method for getting class from constructor parameter

2013-05-16 18:44:58 +04:00
parent 11a1067380
commit 3a2a932e2f
4 changed files with 24 additions and 10 deletions
@@ -196,8 +196,11 @@ public class LightClassUtil {

    @Nullable
    private static PsiClass getWrappingClass(@NotNull JetDeclaration declaration) {
-        if (declaration instanceof JetParameter && declaration.getParent().getParent() instanceof JetClass) {
-            return getPsiClass((JetClassOrObject) declaration.getParent().getParent());
+        if (declaration instanceof JetParameter) {
+            JetClass constructorClass = JetPsiUtil.getClassIfParameterIsProperty((JetParameter) declaration);
+            if (constructorClass != null) {
+                return getPsiClass(constructorClass);
+            }
        }

        if (declaration instanceof JetPropertyAccessor) {