package org.cx4a.rsense.typing.vertex;

import org.jrubyparser.ast.Node;
import org.jrubyparser.ast.INameNode;

import org.cx4a.rsense.ruby.Block;
import org.cx4a.rsense.typing.Propagation;

public class CallVertex extends Vertex {
    private String name;
    private Vertex receiverVertex;
    private Vertex[] argVertices;
    private Block block;
    private boolean privateVisibility;

    public CallVertex(Node node, Vertex receiverVertex, Vertex[] argVertices, Block block) {
        this(node, null, receiverVertex, argVertices, block);
    }

    public CallVertex(Node node, String name, Vertex receiverVertex, Vertex[] argVertices, Block block) {
        super(node);
        this.name = name;
        this.receiverVertex = receiverVertex;
        this.argVertices = argVertices;
        this.block = block;
        this.privateVisibility = false;
        receiverVertex.addEdge(this);
        if (argVertices != null) {
            for (Vertex v : argVertices) {
                v.addEdge(this);
            }
        }
    }

    public void cutout() {
        if (argVertices != null) {
            for (Vertex v : argVertices) {
                v.removeEdge(this);
            }
        }
        receiverVertex.removeEdge(this);
    }

    public String getName() {
        return name == null ? ((INameNode) node).getName() : name;
    }

    public Vertex getReceiverVertex() {
        return receiverVertex;
    }

    public Vertex[] getArgVertices() {
        return argVertices;
    }

    public Block getBlock() {
        return block;
    }

    public boolean hasPrivateVisibility() {
        return privateVisibility;
    }

    public void setPrivateVisibility(boolean privateVisibility) {
        this.privateVisibility = privateVisibility;
    }

    public boolean isApplicable() {
        if (receiverVertex == null || receiverVertex.isEmpty()) {
            return false;
        }

        boolean applicable = true;
        if (argVertices != null) {
            for (Vertex v : argVertices) {
                if (v.isEmpty()) {
                    applicable = false;
                    break;
                }
            }
        }

        return applicable;
    }

    @Override
    public boolean isChanged() {
        if (super.isChanged()
            || (receiverVertex != null
                && receiverVertex.isChanged())) {
            return true;
        }

        if (argVertices != null) {
            for (Vertex v : argVertices) {
                if (v.isChanged()) {
                    return true;
                }
            }
        }

        return false;
    }

    @Override
    public void markUnchanged() {
        super.markUnchanged();
        if (receiverVertex != null) {
            receiverVertex.markUnchanged();
        }
        if (argVertices != null) {
            for (Vertex v : argVertices) {
                v.markUnchanged();
            }
        }
    }

    @Override
    public boolean accept(Propagation propagation, Vertex src) {
        return propagation.getGraph().propagateCallVertex(propagation, this, src);
    }
}