AstModuleElement.java

package org.netbeans.modules.ruby.elements;

import java.util.Set;

import org.jrubyparser.ast.Colon2Node;
import org.jrubyparser.ast.ModuleNode;
import org.jrubyparser.ast.Node;
import org.netbeans.modules.csl.api.ElementKind;
import org.netbeans.modules.csl.spi.ParserResult;

public class AstModuleElement extends AstElement implements ModuleElement {

    private String fqn;
    private String extendWith;
    private Set<String> includes;

    public AstModuleElement(ParserResult info, Node node) {
        super(info, node);
    }

    @Override
    public String getName() {
        if (name == null) {
            if (node instanceof ModuleNode) {
                Node n = ((ModuleNode)node).getCPath();

                if (n instanceof Colon2Node) {
                    Colon2Node c2n = (Colon2Node)n;
                    name = c2n.getName();
                } else {
                    name = n.toString();
                }
            }

            if (name == null) {
                name = node.toString();
            }
        }

        return name;
    }

    @Override
    public ElementKind getKind() {
        return ElementKind.MODULE;
    }

    @Override
    public String getFqn() {
        if (fqn == null) {
            return getName();
        }

        return fqn;
    }

    public void setFqn(String fqn) {
        this.fqn = fqn;
    }

    public void setIncludes(Set<String> includes) {
        this.includes = includes;
    }

    @Override
    public Set<String> getIncludes() {
        return includes;
    }

    /** 
     * Class or module whose instance methods should all be copied into
     * anyone including this module 
     */
    @Override
    public String getExtendWith() {
        return extendWith;
    }

    /** 
     * Set class or module whose instance methods should all be copied into
     * anyone including this module 
     */
    public void setExtendWith(String extendWith) {
        this.extendWith = extendWith;
    }

    @Override
    public boolean isVirtual() {
        return false;
    }

}