Java Code Examples for com.intellij.psi.util.PsiTreeUtil#getChildrenOfAnyType()

The following examples show how to use com.intellij.psi.util.PsiTreeUtil#getChildrenOfAnyType() . These examples are extracted from open source projects. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source Project: NutzCodeInsight   File: SqlsXmlUtil.java    License: Apache License 2.0 6 votes vote down vote up
public static List<PsiField> getExtendsClassFields(PsiClass psiClass) {
    List<PsiField> psiFields = new ArrayList<>();
    List<PsiReferenceList> childrenOfAnyType = PsiTreeUtil.getChildrenOfAnyType(psiClass, PsiReferenceList.class);
    for (PsiReferenceList psiReferenceList : childrenOfAnyType) {
        PsiJavaCodeReferenceElement[] referenceElements = psiReferenceList.getReferenceElements();
        for (PsiJavaCodeReferenceElement referenceElement : referenceElements) {
            String qualifiedName = referenceElement.getQualifiedName();
            GlobalSearchScope globalSearchScope = GlobalSearchScope.projectScope(psiClass.getProject());
            Collection<PsiClass> psiClasses = JavaShortClassNameIndex.getInstance().get(referenceElement.getReferenceName(), psiClass.getProject(), globalSearchScope);
            for (PsiClass aClass : psiClasses) {
                if (aClass.getQualifiedName().equals(qualifiedName)) {
                    psiFields.addAll(Arrays.asList(aClass.getFields()));
                    psiFields.addAll(getExtendsClassFields(aClass));
                }
            }
        }
    }
    return psiFields;
}
 
Example 2
Source Project: NutzCodeInsight   File: SqlsXmlUtil.java    License: Apache License 2.0 6 votes vote down vote up
/**
 * JavaSuperClassNameOccurenceIndex.getInstance().get("ISqlDaoExecuteService", project, moduleWithDependenciesScope)
 *
 * @param project
 * @return
 */
public static List<String> addExecuteService(Project project) {
    if (CollectionUtils.isEmpty(SERVICE_METHOD_List)) {
        List<String> arry = new ArrayList<>();
        Collection<PsiClass> iSqlDaoExecuteService = JavaShortClassNameIndex.getInstance().get("ISqlDaoExecuteService", project, GlobalSearchScope.everythingScope(project));
        for (PsiClass psiClass : iSqlDaoExecuteService) {
            if (SERVICE_ISQL_DAO_EXECUTE_SERVICE.equals(psiClass.getQualifiedName())) {
                List<PsiMethod> methods = PsiTreeUtil.getChildrenOfAnyType(psiClass, PsiMethod.class);
                for (PsiMethod method : methods) {
                    arry.add(method.getName() + "(");
                }
            }
        }
        SERVICE_METHOD_List = arry;
    }
    return SERVICE_METHOD_List;
}
 
Example 3
Source Project: NutzCodeInsight   File: SqlsXmlUtil.java    License: Apache License 2.0 6 votes vote down vote up
public static String findXmlFileName(PsiClass psiClass) {
    List<PsiModifierList> modifierLists = PsiTreeUtil.getChildrenOfAnyType(psiClass, PsiModifierList.class);
    if (CollectionUtils.isNotEmpty(modifierLists)) {
        List<PsiAnnotation> annotations = PsiTreeUtil.getChildrenOfAnyType(modifierLists.get(0), PsiAnnotation.class);
        for (PsiAnnotation annotation : annotations) {
            if (ANNOTATION_SQLS_XML.equals(annotation.getQualifiedName())) {
                List<JvmAnnotationAttribute> attributes = annotation.getAttributes();
                for (JvmAnnotationAttribute attribute : attributes) {
                    String attributeName = attribute.getAttributeName();
                    if ("value".equals(attributeName)) {
                        return ((PsiNameValuePairImpl) attribute).getLiteralValue();
                    }
                }
                if (CollectionUtils.isEmpty(attributes)) {
                    String name = psiClass.getContainingFile().getName();
                    name = name.substring(0, name.length() - 4);
                    return name + "xml";
                }
            }
        }
    }
    return "xml";
}
 
Example 4
Source Project: NutzCodeInsight   File: SqlsXmlUtil.java    License: Apache License 2.0 6 votes vote down vote up
public static PsiElement getXmlIdBindPsiElement(PsiMethodCallExpression psiMethodCallExpression, List<String> keys, String id) {
    for (String key : keys) {
        String text = psiMethodCallExpression.getText();
        if (text.startsWith(key)
                || text.startsWith("super." + key)
                || text.startsWith("this." + key)) {
            PsiExpressionList psiExpressionList = PsiTreeUtil.getChildOfAnyType(psiMethodCallExpression, PsiExpressionList.class);
            if (Objects.nonNull(psiExpressionList)) {
                List<PsiLiteralExpression> psiLiteralExpressions = PsiTreeUtil.getChildrenOfAnyType(psiExpressionList, PsiLiteralExpression.class);
                for (PsiLiteralExpression psiLiteralExpression : psiLiteralExpressions) {
                    if (!psiLiteralExpression.getText().contains(" ") && id.equals(((PsiLiteralExpressionImpl) psiLiteralExpression).getInnerText())) {
                        return psiLiteralExpression;
                    }
                }
            }
        }
    }
    return null;
}
 
Example 5
Source Project: idea-php-symfony2-plugin   File: YamlHelper.java    License: MIT License 6 votes vote down vote up
/**
 * test:
 *   DEBUG_WEB_1: 1
 *   DEBUG_WEB_2: 1
 */
@NotNull
public static Map<String, YAMLValue> getYamlArrayKeyMap(@NotNull YAMLMapping yamlHash) {
    Map<String, YAMLValue> keys = new HashMap<>();

    for(YAMLKeyValue yamlKeyValue: PsiTreeUtil.getChildrenOfAnyType(yamlHash, YAMLKeyValue.class)) {
        String keyText = yamlKeyValue.getKeyText();
        if (StringUtils.isNotBlank(keyText)) {
            YAMLValue value = yamlKeyValue.getValue();
            if (value != null) {
                keys.put(keyText, value);
            }
        }
    }

    return keys;
}
 
Example 6
Source Project: NutzCodeInsight   File: SqlsXmlUtil.java    License: Apache License 2.0 5 votes vote down vote up
public static boolean hasExecuteService(PsiClass psiClass) {
    List<PsiReferenceList> childrenOfAnyType = PsiTreeUtil.getChildrenOfAnyType(psiClass, PsiReferenceList.class);
    for (PsiReferenceList psiReferenceList : childrenOfAnyType) {
        List<PsiJavaCodeReferenceElement> referenceElements = PsiTreeUtil.getChildrenOfAnyType(psiReferenceList, PsiJavaCodeReferenceElement.class);
        for (PsiJavaCodeReferenceElement referenceElement : referenceElements) {
            if (SERVICE_ISQL_DAO_EXECUTE_SERVICE.equals(referenceElement.getQualifiedName())) {
                return true;
            }
        }
    }
    return false;
}
 
Example 7
Source Project: NutzCodeInsight   File: SqlsXmlUtil.java    License: Apache License 2.0 5 votes vote down vote up
public static List<PsiElement> findXmlPsiElement(Project project, Collection<VirtualFile> virtualFiles, String key) {
    List<PsiElement> result = new ArrayList<>();
    for (VirtualFile virtualFile : virtualFiles) {
        PsiFile psiFile = PsiManager.getInstance(project).findFile(virtualFile);
        if (psiFile instanceof XmlFileImpl) {
            XmlFileImpl xmlFile = (XmlFileImpl) psiFile;
            PsiElement navigationElement = xmlFile.getNavigationElement();
            if (Objects.nonNull(navigationElement)) {
                if (Objects.nonNull(navigationElement.getChildren())) {
                    PsiElement child = navigationElement.getChildren()[0];
                    if (Objects.nonNull(child)) {
                        if (Objects.nonNull(child.getChildren()) && child.getChildren().length > 0) {
                            PsiElement child1 = child.getChildren()[1];
                            if (Objects.nonNull(child1)) {
                                PsiElement[] psiElements = child1.getChildren();
                                if (Objects.nonNull(psiElements) && psiElements.length > 0) {
                                    List<XmlTag> xmlTags = PsiTreeUtil.getChildrenOfAnyType(child1, XmlTag.class);
                                    for (XmlTag xmlTag : xmlTags) {
                                        if ("sql".equals(xmlTag.getName())) {
                                            XmlAttribute xmlAttribute = xmlTag.getAttribute("id");
                                            String id = xmlAttribute.getValue();
                                            if (key.equals(id)) {
                                                result.add(xmlAttribute.getValueElement());
                                            }
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }
    }
    return result;
}
 
Example 8
@NotNull
@Override
public DataIndexer<String, Set<String>, FileContent> getIndexer() {
    return fileContent -> {
        Map<String, Set<String>> blocks = new HashMap<>();

        PsiFile psiFile = fileContent.getPsiFile();
        if(psiFile instanceof TwigFile) {
            for (TwigBlock twigBlock : TwigUtil.getBlocksInFile((TwigFile) psiFile)) {
                // we only index file scope
                // {% embed 'foo.html.twig' %}{% block foo %}{% endembed %}
                PsiElement embedStatement = PsiElementUtils.getParentOfType(twigBlock.getTarget(), TwigElementTypes.EMBED_STATEMENT);
                if(embedStatement == null) {
                    blocks.putIfAbsent("block", new HashSet<>());
                    blocks.get("block").add(twigBlock.getName());
                }
            }

            for(PsiElement psiElement : PsiTreeUtil.getChildrenOfAnyType(psiFile, TwigExtendsTag.class, TwigCompositeElement.class)) {
                if(psiElement instanceof TwigCompositeElement) {
                    // {% use 'foo.html.twig' %}
                    if(psiElement.getNode().getElementType() == TwigElementTypes.TAG) {
                        psiElement.acceptChildren(new PsiRecursiveElementWalkingVisitor() {
                            @Override
                            public void visitElement(PsiElement element) {
                                if(TwigPattern.getTwigTagUseNamePattern().accepts(element) && PsiElementUtils.getParentOfType(element, TwigElementTypes.EMBED_STATEMENT) == null) {
                                    String templateName = TwigUtil.normalizeTemplateName(PsiElementUtils.trimQuote(element.getText()));
                                    if(StringUtils.isNotBlank(templateName)) {
                                        blocks.putIfAbsent("use", new HashSet<>());
                                        blocks.get("use").add(templateName);
                                    }
                                }

                                super.visitElement(element);
                            }
                        });
                    }
                }
            }
        }

        return blocks;
    };
}