/*
* Copyright 2009-2011 Prime Technology.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.primefaces.component.tree;
import java.io.IOException;
import java.util.Iterator;
import java.util.Map;
import javax.faces.component.UIComponent;
import javax.faces.context.FacesContext;
import javax.faces.context.ResponseWriter;
import org.primefaces.component.api.UITree;
import org.primefaces.model.TreeNode;
import org.primefaces.renderkit.CoreRenderer;
public class TreeRenderer extends CoreRenderer {
@Override
public void decode(FacesContext context, UIComponent component) {
Tree tree = (Tree) component;
decodeSelection(context, tree);
decodeBehaviors(context, tree);
}
public void decodeSelection(FacesContext context, Tree tree) {
Map<String,String> params = context.getExternalContext().getRequestParameterMap();
String clientId = tree.getClientId(context);
if(tree.getSelectionMode() != null) {
String selection = params.get(clientId + "_selection");
boolean isSingle = tree.getSelectionMode().equalsIgnoreCase("single");
if(isValueBlank(selection)) {
if(isSingle)
tree.setSelection(null);
else
tree.setSelection(new TreeNode[0]);
}
else {
String[] selectedRowKeys = selection.split(",");
if(isSingle) {
tree.setRowKey(selectedRowKeys[0]);
tree.setSelection(tree.getRowNode());
}
else {
TreeNode[] selectedNodes = new TreeNode[selectedRowKeys.length];
for(int i = 0 ; i < selectedRowKeys.length; i++) {
tree.setRowKey(selectedRowKeys[i]);
selectedNodes[i] = tree.getRowNode();
}
tree.setSelection(selectedNodes);
}
tree.setRowKey(null);
}
}
}
@Override
public void encodeEnd(FacesContext context, UIComponent component) throws IOException {
Tree tree = (Tree) component;
if(tree.isNodeExpandRequest(context)) {
String clientId = tree.getClientId(context);
Map<String,String> params = context.getExternalContext().getRequestParameterMap();
String rowKey = params.get(clientId + "_expandNode");
tree.setRowKey(rowKey);
TreeNode node = tree.getRowNode();
node.setExpanded(true);
encodeTreeNodeChildren(context, tree, node, clientId, rowKey, tree.isDynamic(), tree.isCheckboxSelection());
tree.setRowKey(null);
}
else {
encodeMarkup(context, tree);
encodeScript(context, tree);
}
}
protected void encodeScript(FacesContext context, Tree tree) throws IOException {
ResponseWriter writer = context.getResponseWriter();
String clientId = tree.getClientId(context);
boolean dynamic = tree.isDynamic();
String selectionMode = tree.getSelectionMode();
writer.startElement("script", null);
writer.writeAttribute("type", "text/javascript", null);
writer.write("$(function() {");
writer.write(tree.resolveWidgetVar() + " = new PrimeFaces.widget.Tree('" + clientId + "', {");
writer.write("dynamic:" + dynamic);
if(dynamic) {
writer.write(",cache:" + tree.isCache());
}
//selection
if(selectionMode != null) {
writer.write(",selectionMode:'" + selectionMode + "'");
}
if(tree.getOnNodeClick() != null) {
writer.write(",onNodeClick:function(node) {" + tree.getOnNodeClick() + "}");
}
//dragdrop
if(tree.isDragdrop()) {
writer.write(",dragdrop:true");
}
//expand-collapse icon states for specific treenodes
encodeIconStates(context, tree);
encodeClientBehaviors(context, tree);
writer.write("});});");
writer.endElement("script");
}
protected void encodeMarkup(FacesContext context, Tree tree) throws IOException {
ResponseWriter writer = context.getResponseWriter();
String clientId = tree.getClientId(context);
TreeNode root = (TreeNode) tree.getValue();
boolean dynamic = tree.isDynamic();
boolean selectable = tree.getSelectionMode() != null;
boolean checkbox = selectable && tree.getSelectionMode().equals("checkbox");
//container class
String containerClass = Tree.CONTAINER_CLASS;
containerClass = selectable ? containerClass + " " + Tree.SELECTABLE_CLASS : containerClass;
containerClass = tree.getStyleClass() == null ? containerClass : containerClass + " " + tree.getStyleClass();
writer.startElement("div", tree);
writer.writeAttribute("id", clientId, null);
writer.writeAttribute("class", containerClass, null);
if(tree.getStyle() != null)
writer.writeAttribute("style", tree.getStyle(), null);
writer.startElement("ul", null);
writer.writeAttribute("class", Tree.ROOT_NODES_CLASS, null);
if(root != null) {
root.setExpanded(true);
encodeTreeNode(context, tree, root, clientId, null, dynamic, checkbox);
}
writer.endElement("ul");
if(selectable) {
encodeSelectionHolder(context, tree);
}
writer.endElement("div");
}
public void encodeTreeNode(FacesContext context, Tree tree, TreeNode node, String clientId, String rowKey, boolean dynamic, boolean checkbox) throws IOException {
if(rowKey != null) {
ResponseWriter writer = context.getResponseWriter();
tree.setRowKey(rowKey);
boolean isLeaf = node.isLeaf();
boolean expanded = node.isExpanded();
boolean selectable = node.isSelectable();
String iconClass = expanded ? Tree.EXPANDED_ICON_CLASS : Tree.COLLAPSED_ICON_CLASS;
String nodeId = clientId + "_node_" + rowKey;
UITreeNode uiTreeNode = tree.getUITreeNodeByType(node.getType());
//preselection
boolean selected = node.isSelected();
if(selected) {
tree.getSelectedRowKeys().add(rowKey);
}
//style class of node container
String containerClass = isLeaf ? Tree.LEAF_CLASS : Tree.PARENT_CLASS;
containerClass = uiTreeNode.getStyleClass() == null ? containerClass : containerClass + "" + uiTreeNode.getStyleClass();
containerClass = containerClass + " " + uiTreeNode.getType();
writer.startElement("li", null);
writer.writeAttribute("id", nodeId, null);
writer.writeAttribute("class", containerClass, null);
//label
writer.startElement("div", null);
writer.writeAttribute("class", Tree.NODE_CLASS, null);
//node content
String nodeContentClass = selected ? Tree.NODE_CONTENT_CLASS + " ui-state-highlight" : Tree.NODE_CONTENT_CLASS;
nodeContentClass = selectable ? nodeContentClass + " " + Tree.SELECTABLE_NODE_CLASS : nodeContentClass;
writer.startElement("span", null);
writer.writeAttribute("class", nodeContentClass, null);
//state icon
if(!isLeaf) {
writer.startElement("span", null);
writer.writeAttribute("class", iconClass, null);
writer.endElement("span");
}
//node icon
writer.startElement("span", null);
String icon = uiTreeNode.getIconToRender(expanded);
if(icon != null) {
writer.writeAttribute("class", icon, null);
}
writer.endElement("span");
//checkbox
if(checkbox && selectable) {
encodeCheckbox(context, tree, node, selected);
}
//content
writer.startElement("span", null);
writer.writeAttribute("class", Tree.NODE_LABEL_CLASS, null);
uiTreeNode.encodeAll(context);
writer.endElement("span");
writer.endElement("span");
writer.endElement("div");
//children nodes
boolean shouldRender = (dynamic && expanded) || !dynamic;
writer.startElement("ul", null);
writer.writeAttribute("class", Tree.NODES_CLASS , null);
if(!expanded)
writer.writeAttribute("style", "display:none", null);
if(shouldRender)
encodeTreeNodeChildren(context, tree, node, clientId, rowKey, dynamic, checkbox);
writer.endElement("ul");
writer.endElement("li");
}
else {
encodeTreeNodeChildren(context, tree, node, clientId, rowKey, dynamic, checkbox);
}
}
public void encodeTreeNodeChildren(FacesContext context, Tree tree, TreeNode node, String clientId, String rowKey, boolean dynamic, boolean checkbox) throws IOException {
int childIndex = 0;
for(Iterator<TreeNode> iterator = node.getChildren().iterator(); iterator.hasNext();) {
String childRowKey = rowKey == null ? String.valueOf(childIndex) : rowKey + UITree.SEPARATOR + childIndex;
encodeTreeNode(context, tree, iterator.next(), clientId, childRowKey, dynamic, checkbox);
childIndex++;
}
}
protected void encodeIconStates(FacesContext context, Tree tree) throws IOException {
ResponseWriter writer = context.getResponseWriter();
Map<String,UITreeNode> nodes = tree.getTreeNodes();
writer.write(",iconStates:{");
for(Iterator<String> it = nodes.keySet().iterator(); it.hasNext();) {
String type = it.next();
UITreeNode node = nodes.get(type);
String expandedIcon = node.getExpandedIcon();
String collapsedIcon = node.getCollapsedIcon();
if(expandedIcon != null && collapsedIcon != null) {
writer.write("'" + node.getType() + "' : {");
writer.write("expandedIcon:'" + expandedIcon + "'");
writer.write(",collapsedIcon:'" + collapsedIcon + "'");
writer.write("}");
if(it.hasNext())
writer.write(",");
}
}
writer.write("}");
}
protected void encodeSelectionHolder(FacesContext context, Tree tree) throws IOException {
ResponseWriter writer = context.getResponseWriter();
String id = tree.getClientId(context) + "_selection";
writer.startElement("input", null);
writer.writeAttribute("type", "hidden", null);
writer.writeAttribute("id", id, null);
writer.writeAttribute("name", id, null);
writer.writeAttribute("value", tree.getSelectedRowKeysAsString(), null);
writer.endElement("input");
}
protected void encodeCheckbox(FacesContext context, Tree tree, TreeNode node, boolean selected) throws IOException {
ResponseWriter writer = context.getResponseWriter();
String iconClass = selected ? Tree.CHECKBOX_ICON_CHECKED_CLASS : Tree.CHECKBOX_ICON_CLASS;
writer.startElement("div", null);
writer.writeAttribute("class", Tree.CHECKBOX_CLASS, null);
writer.startElement("div", null);
writer.writeAttribute("class", Tree.CHECKBOX_BOX_CLASS, null);
writer.startElement("span", null);
writer.writeAttribute("class", iconClass, null);
writer.endElement("span");
writer.endElement("div");
writer.endElement("div");
}
@Override
public void encodeChildren(FacesContext context, UIComponent component) throws IOException {
//Do nothing
}
@Override
public boolean getRendersChildren() {
return true;
}
}