Call Hierarchy: Add support of local declarations

2013-09-26 15:37:35 +04:00
parent 41ea7da1e3
commit 5b7881f40c
28 changed files with 494 additions and 175 deletions
@@ -1029,4 +1029,16 @@ public class JetPsiUtil {

        return header != null ? header.getQualifiedName() : null;
    }
+
+    public static JetElement getLocalizingCodeBlock(@NotNull JetNamedDeclaration declaration) {
+        //noinspection unchecked
+        JetDeclaration container =
+                PsiTreeUtil.getParentOfType(declaration, JetNamedFunction.class, JetPropertyAccessor.class, JetClassInitializer.class);
+
+        if (container == null) return null;
+
+        return (container instanceof JetClassInitializer)
+               ? ((JetClassInitializer) container).getBody()
+               : ((JetDeclarationWithBody) container).getBodyExpression();
+    }
 }