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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jstree Show documentation
Classes for generating JsTree nodes and context menus.
There is a newer version: 2024.10.0
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;
	}

}