Java PhpElementTypes.ARRAY_KEY属性代码示例

/**
 *
 * array('key' => '<cursor>')
 *
 * Helper for to get find value getArrayPath
 * @param psiElement array value as leaf
 * @param key key name in current content
 * @return parent array creation element
 */
@Nullable
private static ArrayCreationExpression getArrayPathValue(PsiElement psiElement, String key) {

    PsiElement arrayValue = psiElement.getContext();
    if(arrayValue == null) {
        return null;
    }

    if(arrayValue.getNode().getElementType() == PhpElementTypes.ARRAY_VALUE) {
        PsiElement arrayHashElement = arrayValue.getContext();
        if(arrayHashElement instanceof ArrayHashElement) {
            PhpPsiElement arrayKey = ((ArrayHashElement) arrayHashElement).getKey();
            if(arrayKey instanceof StringLiteralExpression && ((StringLiteralExpression) arrayKey).getContents().equals(key)) {
                PsiElement innerArrayKey = arrayKey.getParent();
                if(innerArrayKey != null && innerArrayKey.getNode().getElementType() == PhpElementTypes.ARRAY_KEY) {
                    PsiElement innerArrayHashElement = innerArrayKey.getParent();
                    if(innerArrayHashElement instanceof ArrayHashElement) {
                        PsiElement arrayCreation = innerArrayHashElement.getParent();
                        if(arrayCreation instanceof ArrayCreationExpression) {
                            return (ArrayCreationExpression) arrayCreation;
                        }
                    }
                }
            }
        }
    }

    return null;
}
 

@Nullable
private GotoCompletionProvider createTranslationGotoCompletionWithLabelSwitch(@NotNull PsiElement origin, @NotNull ArrayCreationExpression choices, Processor<ArrayCreationExpression> processor) {
    PsiElement choicesArrayValue = choices.getParent();
    if(choicesArrayValue.getNode().getElementType() == PhpElementTypes.ARRAY_VALUE) {
        PsiElement choicesValueHash = choicesArrayValue.getParent();
        if(choicesValueHash instanceof ArrayHashElement) {
            PhpPsiElement transKey = ((ArrayHashElement) choicesValueHash).getKey();
            String stringValue = PhpElementsUtil.getStringValue(transKey);

            if("choices".equals(stringValue)) {
                PsiElement choicesKey = transKey.getParent();
                if(choicesKey.getNode().getElementType() == PhpElementTypes.ARRAY_KEY) {
                    PsiElement formOptionsHash = choicesKey.getParent();
                    if(formOptionsHash instanceof ArrayHashElement) {
                        PsiElement arrayCreation = formOptionsHash.getParent();
                        if(arrayCreation instanceof ArrayCreationExpression) {
                            if(processor.process((ArrayCreationExpression) arrayCreation)) {
                                return createTranslationGotoCompletion(origin, arrayCreation);
                            }
                        }
                    }
                }
            }
        }
    }

    return null;
}
 

public void fillCompletionVariants(@NotNull CompletionParameters parameters, @NotNull CompletionResultSet result)
{
	PsiElement element = parameters.getPosition().getParent();

	ParameterList parameterList = PsiTreeUtil.getParentOfType(element, ParameterList.class);
	if (parameterList == null) {
		return;
	}
	if (element.getParent() instanceof PhpPsiElement
		&& ((PhpPsiElement) element.getParent()).getPrevPsiSibling() != null
		&& ((PhpPsiElement) element.getParent()).getPrevPsiSibling().getNode().getElementType() == PhpElementTypes.ARRAY_KEY) {
		return;
	}
	PsiElement context = parameterList.getContext();
	if (!(context instanceof MethodReference)) {
		return;
	}
	MethodReference methodReference = (MethodReference) context;
	PsiElement methodEl = methodReference.resolve();
	if (methodEl == null || !(methodEl instanceof Method)) {
		return;
	}
	Method method = (Method) methodEl;
	if (!(method.getName().equals("findBy") || method.getName().equals("getBy") || method.getName().equals("orderBy"))) {
		return;
	}

	Project project = parameters.getEditor().getProject();
	if (project == null) {
		return;
	}


	String fieldExpression = parameters.getOriginalPosition().getText();
	String[] path = fieldExpression.split("->", -1);

	Collection<PhpClass> queriedEntities = OrmUtils.findQueriedEntities(methodReference, path);
	for (PhpClass cls : queriedEntities) {

		if (cls.getDocComment() == null) {
			continue;
		}
		for (PhpDocPropertyTag phpDocPropertyTag : cls.getDocComment().getPropertyTags()) {

			Stream<String> types = phpDocPropertyTag.getType().getTypesSorted().stream()
				.filter(s -> !s.contains("Nextras\\Orm\\Relationships") && !s.equals("?"))
				.map(s -> s.startsWith("\\") ? s.substring(1) : s);

			String strPath = String.join("->", Arrays.copyOfRange(path, 0, path.length - 1));
			if (strPath.length() > 0) {
				strPath += "->";
			}
			String fieldName = phpDocPropertyTag.getProperty().getText().substring(1);
			strPath += fieldName;

			result.addElement(LookupElementBuilder.create(strPath)
				.withPresentableText(fieldName)
				.withTypeText(types.collect(Collectors.joining("|"))));
		}
		if (path.length == 1) {
			result.addElement(LookupElementBuilder.create("this").withTypeText(cls.getType().toString()));
			result.addElement(LookupElementBuilder.create(cls.getFQN()).withTypeText(cls.getType().toString()));
		}
	}
}