MatrixTreeNode.java

package net.squiz.matrix.matrixtree;

import net.squiz.matrix.core.*;
import javax.swing.tree.*;
import java.awt.*;
import java.util.*;
import java.awt.datatransfer.*;
import java.io.*;

public class MatrixTreeNode extends DefaultMutableTreeNode
    implements Serializable {


    /* The linkid that this node represents */
    private final String linkid;

    private int linkType;

    private int sort_order;

    /* The URL including paths to this node */
    private String url;
    private String webPath;

    /* Name of node */
    protected String name;

    public MatrixTreeNode(Asset asset, String linkid, int linkType, String url, String webPath, String name, int sort_order) {
        setUserObject(asset);
        this.linkid = linkid;
        this.linkType = linkType;
        this.url = url;
        this.webPath = webPath;
        this.name = name;
        this.sort_order = sort_order;
    }

    public String toString() {
        return getName() + " Linkid : " + linkid;
    }

    /* Each node can have different name, but belong to same asset*/
    public void setName(String name) {
        this.name = name;
    }

    public String getName() {
        return name;
    }

    public void setSortOrder(int sort_order) {
        this.sort_order = sort_order;
    }

    public int getSortOrder() {
        return this.sort_order;
    }

    public Asset getAsset() {
        return (Asset) getUserObject();
    }

    public int getLinkType() {
        return linkType;
    }

    public void setLinkType(int linkType) {
        this.linkType = linkType;
    }

    public String getLinkid() {
        return linkid;
    }

    public boolean isShadowAsset() {
        return (linkid.equals("0") && linkid.split(":").length > 1);
    }

    public boolean isLeaf() {
        // if the asset is not root and user has no access then make this asset a leaf node
        if (!getAsset().getId().equals("1") && !getAsset().isAccessible()) {
            return true;
        }

        return (getAsset().getNumKids() == 0);
    }

    public String getURL() {
        if (url == null) {
            return "";
        } else if (webPath == null) {
            return url;
        }
        return url + "/" + webPath;
    }

    public boolean hasChildWithLinkid(String linkid) {
        return (getChildWithLinkid(linkid) == null) ? false : true;
    }

    public MatrixTreeNode getChildWithLinkid(String linkid) {
        Enumeration children = children();
        while (children.hasMoreElements()) {
            MatrixTreeNode node = (MatrixTreeNode) children.nextElement();
            if (node.getLinkid().equals(linkid))
                return (MatrixTreeNode) node;
        }
        return null;
    }


    public void propagateUrl(String url) {
        this.url = url;
        Enumeration children = children();
        while (children.hasMoreElements()) {
            MatrixTreeNode node = (MatrixTreeNode) children.nextElement();
            node.propagateUrl(getURL());
        }
    }

    public void propagateWebPath(String webPath) {
        this.webPath = webPath;
        Enumeration children = children();
        while (children.hasMoreElements()) {
            MatrixTreeNode node = (MatrixTreeNode) children.nextElement();
            // from here on in, all the nodes under this particular node
            // only have changed their urls as this node's url + "/" + webPath
            // is its children's url
            node.propagateUrl(getURL());
        }
    }

    public String getAssetPath() {
        Object[] path = getPath();
        StringBuffer assetPath = new StringBuffer();
        for (int i = 0; i < path.length; i++) {
            assetPath.append(",").append(((MatrixTreeNode) path[i]).getAsset().getId());
        }
        return assetPath.toString();
    }

    public String getLinkPath() {
        Object[] path = getPath();
        StringBuffer linkPath = new StringBuffer();
        for (int i = 0; i < path.length; i++) {
            linkPath.append(",").append(((MatrixTreeNode) path[i]).getLinkid());
        }
        return linkPath.toString();
    }

    public boolean hasPreviousNode() {
        if (getChildCount() != 0) {
            MatrixTreeNode node = (MatrixTreeNode)getChildAt(0);
            if (node instanceof ExpandingPreviousNode) {
                return true;
            }
        }
        return false;
    }

    public boolean hasNextNode() {
        if (getChildCount() != 0) {
            MatrixTreeNode node = (MatrixTreeNode)getChildAt(getChildCount()-1);
            if (node instanceof ExpandingNextNode) {
                return true;
            }
        }
        return false;
    }

}