org.nasdanika.html.jstree.impl.JsTreeNodeImpl Maven / Gradle / Ivy

Go to download
package org.nasdanika.html.jstree.impl;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.function.Predicate;

import org.json.JSONArray;
import org.json.JSONObject;
import org.nasdanika.html.jstree.JsTreeNode;

class JsTreeNodeImpl implements JsTreeNode {
	
	private Object id;
	private Object text;
	private Map attributes = new HashMap<>();
	private Map aAttributes = new HashMap<>();
	private Map liAttributes = new HashMap<>();	
	private Object data;
	private Map properties = new HashMap<>();
	private List children = new ArrayList<>();			
	private boolean selected;
	private boolean opened;
	private boolean disabled;
	private String icon;
	private boolean hasChildren;
	
	@Override
	public JsTreeNode selected(boolean selected) {
		this.selected = selected;
		return this;
	}
	
	@Override
	public JsTreeNode opened(boolean opened) {
		this.opened = opened;
		return this;
	}
	
	@Override
	public JsTreeNode icon(String icon) {
		this.icon = icon;
		return this;
	}
	
	@Override
	public JsTreeNode disabled(boolean disabled) {
		this.disabled = disabled;
		return this;
	}

	@Override
	public JsTreeNode id(Object id) {
		this.id = id;
		return this;
	}
	
	@Override
	public Object getId() {
		return id;
	}

	@Override
	public JsTreeNode text(Object text) {
		this.text = text;
		return this;
	}

	@Override
	public List children() {
		return children;
	}

	@Override
	public JsTreeNode attribute(String name, Object value) {
		if (value == null) {
			attributes.remove(name);
		} else {
			attributes.put(name, value);
		}
		return this;
	}

	@Override
	public JsTreeNode listItemAttribute(String name, Object value) {
		if (value == null) {
			liAttributes.remove(name);
		} else {
			liAttributes.put(name, value);
		}
		return this;
	}

	@Override
	public JsTreeNode anchorAttribute(String name, Object value) {
		if (value == null) {
			aAttributes.remove(name);					
		} else {
			aAttributes.put(name, value);
		}
		return this;
	}

	@Override
	public JSONObject toJSON() {
		return toJSON(null);
	}	
	
	@Override
	public JSONObject toJSON(Predicate filter) {
		JSONArray jch = new JSONArray();
		for (JsTreeNode child: children) {
			if (child != null) {
				JSONObject jsonChild = ((JsTreeNodeImpl) child).toJSON(filter);
				if (jsonChild != null) {
					jch.put(jsonChild);
				}
			}
		}
		
		if (jch.length() == 0 && filter != null && !filter.test(this) && hasChildren) {
			return null; // no children, this one is not accepted.
		}
		
		JSONObject data = new JSONObject();
		JSONObject state = new JSONObject();
		if (selected) {
			state.put("selected", true);
		}
		if (opened) {
			state.put("opened", true);
		}
		if (disabled) {
			state.put("disabled", true);
		}
		if (state.length() > 0) {
			data.put("state", state);
		}
		if (icon != null) {
			data.put("icon", icon);
		} 
		if (id != null) {
			data.put("id", id.toString());
		}
		if (text != null) {
			data.put("text", text.toString());
		}
		
		if (jch.length() == 0) {
			if (hasChildren) {
				data.put("children", true);				
			}
		} else {
			data.put("children", jch);
		}

		if (!liAttributes.isEmpty()) {
			data.put("li_attr", liAttributes);
		}
		if (!aAttributes.isEmpty()) {
			data.put("a_attr", aAttributes);
		}
		
		for (Entry ae: attributes.entrySet()) {
			data.put(ae.getKey(), ae.getValue());
		}
				
		return data;
	}
	
	public String toString() {
		return toJSON().toString();
	}	
	
	@Override
	public Object getData() {
		return data;
	}
	
	@Override
	public JsTreeNode setData(Object data) {
		this.data = data;
		return this;
	}
	
	@Override
	public Object getData(Object key) {
		return properties.get(key);
	}
	
	@Override
	public JsTreeNode setData(Object key, Object data) {
		properties.put(key, data);
		return this;
	}

	@Override
	public  R accept(Collector collector) {
		List childResults = new ArrayList();
		for (JsTreeNode child: children()) {
			childResults.add(child.accept(collector));
		}
		return collector.visit(this, childResults);
	}

	@Override
	public JsTreeNode hasChildren() {
		hasChildren = true;
		return this;
	}
	
	@Override
	public JsTreeNode createChild() {
		JsTreeNodeImpl child = new JsTreeNodeImpl();
		children.add(child);
		return child;
	}

}