1 files changed, 163 insertions, 0 deletions
diff --git a/python/src/com/jetbrains/python/hierarchy/call/PyStaticCallHierarchyUtil.java b/python/src/com/jetbrains/python/hierarchy/call/PyStaticCallHierarchyUtil.java
new file mode 100644
index 000000000000..5657dcb589e9
--- /dev/null
+++ b/python/src/com/jetbrains/python/hierarchy/call/PyStaticCallHierarchyUtil.java
@@ -0,0 +1,163 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.jetbrains.python.hierarchy.call;
+
+import com.google.common.collect.Lists;
+import com.intellij.find.findUsages.FindUsagesHandler;
+import com.intellij.find.findUsages.FindUsagesOptions;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.util.PsiTreeUtil;
+import com.intellij.usageView.UsageInfo;
+import com.intellij.util.ArrayUtil;
+import com.intellij.util.CommonProcessors;
+import com.jetbrains.python.PyNames;
+import com.jetbrains.python.findUsages.PyClassFindUsagesHandler;
+import com.jetbrains.python.findUsages.PyFunctionFindUsagesHandler;
+import com.jetbrains.python.psi.*;
+import com.jetbrains.python.psi.impl.PyBuiltinCache;
+import com.jetbrains.python.psi.resolve.PyResolveContext;
+import com.jetbrains.python.psi.search.PySuperMethodsSearch;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.Collection;
+import java.util.List;
+
+/**
+ * @author novokrest
+ */
+public class PyStaticCallHierarchyUtil {
+  public static Collection<PsiElement> getCallees(@NotNull PyElement element) {
+    final List<PsiElement> callees = Lists.newArrayList();
+
+    final PyRecursiveElementVisitor visitor = new PyRecursiveElementVisitor() {
+      @Override
+      public void visitPyParameterList(PyParameterList node) {
+      }
+
+      @Override
+      public void visitPyLambdaExpression(PyLambdaExpression node) {
+      }
+
+      @Override
+      public void visitPyFunction(PyFunction innerFunction) {
+        for (PyParameter parameter : innerFunction.getParameterList().getParameters()) {
+          PsiElement defaultValue = parameter.getDefaultValue();
+          if (defaultValue != null) {
+            defaultValue.accept(this);
+          }
+        }
+      }
+
+      @Override
+      public void visitPyCallExpression(PyCallExpression callExpression) {
+        super.visitPyCallExpression(callExpression);
+        PsiElement calleeFunction = callExpression.resolveCalleeFunction(PyResolveContext.defaultContext());
+        if (calleeFunction instanceof PyFunction) {
+          callees.add(calleeFunction);
+        }
+      }
+    };
+
+    visitor.visitElement(element);
+
+    return callees;
+  }
+
+  public static Collection<PsiElement> getCallers(@NotNull PyElement pyElement) {
+    final List<PsiElement> callers = Lists.newArrayList();
+    final Collection<UsageInfo> usages = findUsages(pyElement);
+
+    for (UsageInfo usage : usages) {
+      PsiElement element = usage.getElement();
+      if (element == null) {
+        continue;
+      }
+
+      element = element.getParent();
+      while (element instanceof PyParenthesizedExpression) {
+        element = element.getParent();
+      }
+
+      if (element instanceof PyCallExpression) {
+        PsiElement caller = PsiTreeUtil.getParentOfType(element, PyParameterList.class, PyFunction.class);
+        if (caller instanceof PyFunction) {
+          callers.add(caller);
+        }
+        else if (caller instanceof PyParameterList) {
+          PsiElement innerFunction = PsiTreeUtil.getParentOfType(caller, PyFunction.class);
+          PsiElement outerFunction = PsiTreeUtil.getParentOfType(innerFunction, PyFunction.class);
+          if (innerFunction != null && outerFunction != null) {
+            callers.add(outerFunction);
+          }
+        }
+      }
+    }
+
+    return callers;
+  }
+
+  private static Collection<UsageInfo> findUsages(@NotNull final PsiElement element) {
+    final FindUsagesHandler handler = createFindUsageHandler(element);
+    if (handler == null) {
+      return Lists.newArrayList();
+    }
+    final CommonProcessors.CollectProcessor<UsageInfo> processor = new CommonProcessors.CollectProcessor<UsageInfo>();
+    final PsiElement[] psiElements = ArrayUtil.mergeArrays(handler.getPrimaryElements(), handler.getSecondaryElements());
+    final FindUsagesOptions options = handler.getFindUsagesOptions(null);
+    for (PsiElement psiElement : psiElements) {
+      handler.processElementUsages(psiElement, processor, options);
+    }
+    return processor.getResults();
+  }
+
+  /**
+   * @see {@link com.jetbrains.python.findUsages.PyFindUsagesHandlerFactory#createFindUsagesHandler(com.intellij.psi.PsiElement, boolean) createFindUsagesHandler}
+   */
+  @Nullable
+  private static FindUsagesHandler createFindUsageHandler(@NotNull final PsiElement element) {
+    if (element instanceof PyFunction) {
+      final Collection<PsiElement> superMethods = PySuperMethodsSearch.search((PyFunction)element, true).findAll();
+      if (superMethods.size() > 0) {
+        final PsiElement next = superMethods.iterator().next();
+        if (next instanceof PyFunction && !isInObject((PyFunction)next)) {
+          List<PsiElement> allMethods = Lists.newArrayList();
+          allMethods.add(element);
+          allMethods.addAll(superMethods);
+
+          return new PyFunctionFindUsagesHandler(element, allMethods);
+        }
+      }
+      return new PyFunctionFindUsagesHandler(element);
+    }
+    if (element instanceof PyClass) {
+      return new PyClassFindUsagesHandler((PyClass)element);
+    }
+    return null;
+  }
+
+  /**
+   * @see {@link com.jetbrains.python.findUsages.PyFindUsagesHandlerFactory#isInObject(com.jetbrains.python.psi.PyFunction) isInObject}
+   */
+  private static boolean isInObject(PyFunction fun) {
+    final PyClass containingClass = fun.getContainingClass();
+    if (containingClass == null) {
+      return false;
+    }
+    return (PyNames.FAKE_OLD_BASE.equals(containingClass.getName()) ||
+            (PyNames.OBJECT.equals(containingClass.getName()) && PyBuiltinCache.getInstance(fun).isBuiltin(containingClass)));
+  }
+}