Search in sources :

Example 6 with ExpressionNode

use of org.ballerinalang.plugins.idea.psi.ExpressionNode in project ballerina by ballerina-lang.

the class BallerinaParameterInfoHandler method showParameterInfo.

@Override
public void showParameterInfo(@NotNull Object element, @NotNull CreateParameterInfoContext context) {
    // This method will be called with the return object of the findElementForParameterInfo(). If it is null,
    // this method will not be called.
    // Since we know the type, we check and cast the object.
    PsiElement currentElement = null;
    PsiElement parentElement = null;
    List<PsiElement> list = getParameters(element);
    if (element instanceof ExpressionListNode) {
        ExpressionListNode expressionListNode = (ExpressionListNode) element;
        // We need to get the ExpressionListNode parent of current ExpressionListNode.
        // Current ExpressionListNode - "WSO2"
        // Parent ExpressionListNode - setName("WSO2")
        // By doing this, we get the function name because setName("WSO2") is also a ExpressionNode.
        PsiElement parent = PsiTreeUtil.getParentOfType(expressionListNode, FunctionInvocationNode.class);
        // node is a FunctionInvocationNode.
        if (parent == null) {
            // So if the parent is null, we consider the FunctionInvocationNode as the parent node.
            parent = PsiTreeUtil.getParentOfType(expressionListNode, ExpressionNode.class);
        }
        if (parent == null) {
            parent = PsiTreeUtil.getParentOfType(expressionListNode, InvocationNode.class);
        }
        if (parent == null) {
            // So if the parent is null, we consider the ActionInvocationNode as the parent node.
            parent = PsiTreeUtil.getParentOfType(expressionListNode, ActionInvocationNode.class);
        }
        if (parent == null) {
            // So if the parent is null, we consider the ActionInvocationNode as the parent node.
            parent = PsiTreeUtil.getParentOfType(expressionListNode, ConnectorInitNode.class);
        }
        if (parent == null) {
            // So if the parent is null, we consider the ExpressionListNode as the parent node.
            parent = PsiTreeUtil.getParentOfType(expressionListNode, ExpressionListNode.class);
        }
        if (parent == null) {
            parent = expressionListNode;
        }
        currentElement = expressionListNode;
        parentElement = parent;
    } else if (element instanceof FunctionInvocationNode) {
        FunctionInvocationNode functionInvocationNode = (FunctionInvocationNode) element;
        currentElement = functionInvocationNode;
        parentElement = functionInvocationNode;
    } else if (element instanceof ActionInvocationNode) {
        ActionInvocationNode actionInvocationNode = (ActionInvocationNode) element;
        currentElement = actionInvocationNode;
        parentElement = actionInvocationNode;
    } else if (element instanceof ConnectorInitNode) {
        ConnectorInitNode connectorInitNode = (ConnectorInitNode) element;
        currentElement = connectorInitNode;
        parentElement = connectorInitNode;
    } else if (element instanceof ExpressionNode) {
        ExpressionNode expressionNode = (ExpressionNode) element;
        currentElement = expressionNode;
        parentElement = expressionNode;
    } else if (element instanceof NameReferenceNode) {
        NameReferenceNode nameReferenceNode = (NameReferenceNode) element;
        currentElement = nameReferenceNode;
        parentElement = nameReferenceNode;
    } else if (element instanceof FunctionReferenceNode) {
        FunctionReferenceNode functionReferenceNode = (FunctionReferenceNode) element;
        currentElement = functionReferenceNode;
        parentElement = functionReferenceNode;
    } else if (element instanceof ConnectorReferenceNode) {
        ConnectorReferenceNode connectorReferenceNode = (ConnectorReferenceNode) element;
        currentElement = connectorReferenceNode;
        parentElement = connectorReferenceNode;
    } else if (element instanceof IdentifierPSINode) {
        IdentifierPSINode identifier = (IdentifierPSINode) element;
        currentElement = identifier;
        parentElement = identifier;
    } else if (element instanceof InvocationNode) {
        InvocationNode invocationNode = (InvocationNode) element;
        currentElement = invocationNode;
        parentElement = invocationNode;
    }
    PsiElement namedIdentifierDefNode = getNameIdentifierDefinitionNode(parentElement);
    PsiElement nameIdentifier = getNameIdentifier(currentElement, namedIdentifierDefNode);
    if (currentElement == null || nameIdentifier == null || !(nameIdentifier instanceof IdentifierPSINode)) {
        return;
    }
    if (list == null) {
        list = new LinkedList<>();
    }
    // If there are no items to show, set a custom object. Otherwise set the list as an array.
    if (list.isEmpty() && canResolve(nameIdentifier)) {
        // Todo - change how to identify no parameter situation
        context.setItemsToShow(new Object[] { "Empty" });
    } else {
        context.setItemsToShow(list.toArray(new PsiElement[list.size()]));
    }
    context.showHint(currentElement, currentElement.getTextRange().getStartOffset(), this);
}
Also used : ActionInvocationNode(org.ballerinalang.plugins.idea.psi.ActionInvocationNode) ExpressionListNode(org.ballerinalang.plugins.idea.psi.ExpressionListNode) ConnectorInitNode(org.ballerinalang.plugins.idea.psi.ConnectorInitNode) ActionInvocationNode(org.ballerinalang.plugins.idea.psi.ActionInvocationNode) InvocationNode(org.ballerinalang.plugins.idea.psi.InvocationNode) FunctionInvocationNode(org.ballerinalang.plugins.idea.psi.FunctionInvocationNode) ExpressionNode(org.ballerinalang.plugins.idea.psi.ExpressionNode) IdentifierPSINode(org.ballerinalang.plugins.idea.psi.IdentifierPSINode) ConnectorReferenceNode(org.ballerinalang.plugins.idea.psi.ConnectorReferenceNode) FunctionReferenceNode(org.ballerinalang.plugins.idea.psi.FunctionReferenceNode) PsiElement(com.intellij.psi.PsiElement) LeafPsiElement(com.intellij.psi.impl.source.tree.LeafPsiElement) FunctionInvocationNode(org.ballerinalang.plugins.idea.psi.FunctionInvocationNode) NameReferenceNode(org.ballerinalang.plugins.idea.psi.NameReferenceNode)

Example 7 with ExpressionNode

use of org.ballerinalang.plugins.idea.psi.ExpressionNode in project ballerina by ballerina-lang.

the class BallerinaParameterInfoHandler method getParameters.

/**
 * Returns the parameter list for the given element.
 *
 * @param element
 * @return
 */
@NotNull
public static List<PsiElement> getParameters(@NotNull Object element) {
    List<PsiElement> list = new LinkedList<>();
    if (element instanceof ExpressionListNode) {
        ExpressionListNode expressionListNode = (ExpressionListNode) element;
        // We need to get the ExpressionListNode parent of current ExpressionListNode.
        // Current ExpressionListNode - "WSO2"
        // Parent ExpressionListNode - setName("WSO2")
        PsiElement parent = PsiTreeUtil.getParentOfType(expressionListNode, FunctionInvocationNode.class);
        if (parent == null) {
            // So if the parent is null, we consider the ActionInvocationNode as the parent node.
            parent = PsiTreeUtil.getParentOfType(expressionListNode, ActionInvocationNode.class);
        }
        if (parent == null) {
            parent = PsiTreeUtil.getParentOfType(expressionListNode, InvocationNode.class);
        }
        if (parent == null) {
            // So if the parent is null, we consider the ActionInvocationNode as the parent node.
            parent = PsiTreeUtil.getParentOfType(expressionListNode, ConnectorInitNode.class);
        }
        if (parent == null) {
            // So if the parent is null, we consider the FunctionInvocationNode as the parent node.
            parent = PsiTreeUtil.getParentOfType(expressionListNode, ExpressionNode.class);
        }
        if (parent == null) {
            // So if the parent is null, we consider the ExpressionListNode as the parent node.
            parent = PsiTreeUtil.getParentOfType(expressionListNode, ExpressionListNode.class);
        }
        if (parent == null) {
            parent = expressionListNode;
        }
        list = getItemsToShow(expressionListNode, parent);
    } else if (element instanceof FunctionInvocationNode) {
        FunctionInvocationNode functionInvocationNode = (FunctionInvocationNode) element;
        list = getItemsToShow(functionInvocationNode, functionInvocationNode);
    } else if (element instanceof ActionInvocationNode) {
        ActionInvocationNode actionInvocationNode = (ActionInvocationNode) element;
        list = getItemsToShow(actionInvocationNode, actionInvocationNode);
    } else if (element instanceof ConnectorInitNode) {
        ConnectorInitNode connectorInitNode = (ConnectorInitNode) element;
        list = getItemsToShow(connectorInitNode, connectorInitNode);
    } else if (element instanceof ExpressionNode) {
        ExpressionNode expressionNode = (ExpressionNode) element;
        list = getItemsToShow(expressionNode, expressionNode);
    } else if (element instanceof NameReferenceNode) {
        NameReferenceNode nameReferenceNode = (NameReferenceNode) element;
        list = getItemsToShow(nameReferenceNode, nameReferenceNode);
    } else if (element instanceof FunctionReferenceNode) {
        FunctionReferenceNode functionReferenceNode = (FunctionReferenceNode) element;
        list = getItemsToShow(functionReferenceNode, functionReferenceNode);
    } else if (element instanceof ConnectorReferenceNode) {
        ConnectorReferenceNode connectorReferenceNode = (ConnectorReferenceNode) element;
        list = getItemsToShow(connectorReferenceNode, connectorReferenceNode);
    } else if (element instanceof IdentifierPSINode) {
        IdentifierPSINode identifier = (IdentifierPSINode) element;
        list = getItemsToShow(identifier, identifier);
    } else if (element instanceof InvocationNode) {
        InvocationNode invocationNode = (InvocationNode) element;
        list = getItemsToShow(invocationNode, invocationNode);
    }
    return list;
}
Also used : ActionInvocationNode(org.ballerinalang.plugins.idea.psi.ActionInvocationNode) ExpressionListNode(org.ballerinalang.plugins.idea.psi.ExpressionListNode) ConnectorInitNode(org.ballerinalang.plugins.idea.psi.ConnectorInitNode) ActionInvocationNode(org.ballerinalang.plugins.idea.psi.ActionInvocationNode) InvocationNode(org.ballerinalang.plugins.idea.psi.InvocationNode) FunctionInvocationNode(org.ballerinalang.plugins.idea.psi.FunctionInvocationNode) LinkedList(java.util.LinkedList) ExpressionNode(org.ballerinalang.plugins.idea.psi.ExpressionNode) IdentifierPSINode(org.ballerinalang.plugins.idea.psi.IdentifierPSINode) ConnectorReferenceNode(org.ballerinalang.plugins.idea.psi.ConnectorReferenceNode) FunctionReferenceNode(org.ballerinalang.plugins.idea.psi.FunctionReferenceNode) PsiElement(com.intellij.psi.PsiElement) LeafPsiElement(com.intellij.psi.impl.source.tree.LeafPsiElement) FunctionInvocationNode(org.ballerinalang.plugins.idea.psi.FunctionInvocationNode) NameReferenceNode(org.ballerinalang.plugins.idea.psi.NameReferenceNode) NotNull(org.jetbrains.annotations.NotNull)

Aggregations

ExpressionNode (org.ballerinalang.plugins.idea.psi.ExpressionNode)7 PsiElement (com.intellij.psi.PsiElement)6 LeafPsiElement (com.intellij.psi.impl.source.tree.LeafPsiElement)5 IdentifierPSINode (org.ballerinalang.plugins.idea.psi.IdentifierPSINode)4 ConnectorInitNode (org.ballerinalang.plugins.idea.psi.ConnectorInitNode)3 ExpressionListNode (org.ballerinalang.plugins.idea.psi.ExpressionListNode)3 FunctionReferenceNode (org.ballerinalang.plugins.idea.psi.FunctionReferenceNode)3 NameReferenceNode (org.ballerinalang.plugins.idea.psi.NameReferenceNode)3 TypeConversionNode (org.ballerinalang.plugins.idea.psi.TypeConversionNode)3 TypeNameNode (org.ballerinalang.plugins.idea.psi.TypeNameNode)3 VariableReferenceNode (org.ballerinalang.plugins.idea.psi.VariableReferenceNode)3 PsiReference (com.intellij.psi.PsiReference)2 IElementType (com.intellij.psi.tree.IElementType)2 ANTLRPsiNode (org.antlr.jetbrains.adaptor.psi.ANTLRPsiNode)2 ActionInvocationNode (org.ballerinalang.plugins.idea.psi.ActionInvocationNode)2 AnnotationAttachmentNode (org.ballerinalang.plugins.idea.psi.AnnotationAttachmentNode)2 AnonStructTypeNameNode (org.ballerinalang.plugins.idea.psi.AnonStructTypeNameNode)2 BuiltInReferenceTypeNameNode (org.ballerinalang.plugins.idea.psi.BuiltInReferenceTypeNameNode)2 ConnectorReferenceNode (org.ballerinalang.plugins.idea.psi.ConnectorReferenceNode)2 ConstantDefinitionNode (org.ballerinalang.plugins.idea.psi.ConstantDefinitionNode)2