/**
    * Copyright 2005-2014 The Kuali Foundation
    *
    * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php
    *
   * 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.kuali.rice.krad.uif.container;
  
  import com.google.common.collect.Maps;
  import org.apache.commons.lang.StringUtils;
  import org.kuali.rice.core.api.util.tree.Node;
  import org.kuali.rice.core.api.util.tree.Tree;
  import org.kuali.rice.krad.datadictionary.parse.BeanTag;
  import org.kuali.rice.krad.datadictionary.parse.BeanTagAttribute;
  import org.kuali.rice.krad.datadictionary.parse.BeanTags;
  import org.kuali.rice.krad.uif.UifConstants;
  import org.kuali.rice.krad.uif.component.BindingInfo;
  import org.kuali.rice.krad.uif.component.Component;
  import org.kuali.rice.krad.uif.component.DataBinding;
  import org.kuali.rice.krad.uif.element.Message;
  import org.kuali.rice.krad.uif.util.ComponentUtils;
  import org.kuali.rice.krad.uif.util.ObjectPropertyUtils;
  import org.kuali.rice.krad.uif.view.View;
  
  import java.util.ArrayList;
  import java.util.List;
  import java.util.Map;
  
  /**
   * Group component that is backed by a <code>Tree</code> data structure and typically
   * rendered as a tree in the user interface
   *
   * @author Kuali Rice Team (rice.collab@kuali.org)
   */
  @BeanTags({@BeanTag(name = "treeGroup-bean", parent = "Uif-TreeGroup"),
          @BeanTag(name = "treeSection-bean", parent = "Uif-TreeSection"),
          @BeanTag(name = "treeSubSection-bean", parent = "Uif-TreeSubSection")})
  public class TreeGroup extends Group implements DataBinding {
      private static final long serialVersionUID = 5841343037089286740L;
  
      private String propertyName;
      private BindingInfo bindingInfo;
  
      private Map<Class<?>, NodePrototype> nodePrototypeMap;
      private NodePrototype defaultNodePrototype;
  
      private Tree<Group, Message> treeGroups;
  
      private org.kuali.rice.krad.uif.widget.Tree tree;
  
      public TreeGroup() {
          super();
  
          treeGroups = new Tree<Group, Message>();
      }
  
      /**
       * The following actions are performed:
       *
       * <ul>
       * <li>Set fieldBindModelPath to the collection model path (since the fields
       * have to belong to the same model as the collection)</li>
       * <li>Set defaults for binding</li>
       * <li>Calls view helper service to initialize prototypes</li>
       * </ul>
       */
      @Override
      public void performInitialization(View view, Object model) {
          setFieldBindingObjectPath(getBindingInfo().getBindingObjectPath());
  
          super.performInitialization(view, model);
  
          if (bindingInfo != null) {
              bindingInfo.setDefaults(view, getPropertyName());
          }
  
          // TODO: set object path for prototypes equal to the tree group object path?
      }
  
      @Override
      public void performApplyModel(View view, Object model, Component parent) {
          super.performApplyModel(view, model, parent);
  
          buildTreeGroups(view, model);
      }
  
      /**
       * Builds the components that will be rendered as part of the tree group
       *
       * <p>
      * The component tree group mirrors the tree data structure on the model. For each node of
      * the data structure, a corresponding <code>Message</code>  will be created for the node
      * label, and a <code>Group</code> component for the node data. These are placed into a new
      * node for the component tree. After the tree is built it is set as a property on the tree group
      * to be read by the renderer
      * </p>
      *
      * @param view view instance the tree group belongs to
      * @param model object containing the view data from which the tree data will be retrieved
      */
     protected void buildTreeGroups(View view, Object model) {
         // get Tree data property
         Tree<Object, String> treeData = ObjectPropertyUtils.getPropertyValue(model, getBindingInfo().getBindingPath());
 
         // build component tree that corresponds with tree data
         Tree<Group, Message> treeGroups = new Tree<Group, Message>();
 
         String bindingPrefix = getBindingInfo().getBindingPrefixForNested();
         Node<Group, Message> rootNode = buildTreeNode(treeData.getRootElement(),
                 bindingPrefix + /* TODO: hack */ ".rootElement", "root");
         treeGroups.setRootElement(rootNode);
 
         setTreeGroups(treeGroups);
     }
 
     protected Node<Group, Message> buildTreeNode(Node<Object, String> nodeData, String bindingPrefix,
             String parentNode) {
         if (nodeData == null) {
             return null;
         }
 
         Node<Group, Message> node = new Node<Group, Message>();
         node.setNodeType(nodeData.getNodeType());
 
         NodePrototype prototype = getNodePrototype(nodeData);
 
         Message message = ComponentUtils.copy(prototype.getLabelPrototype(), parentNode);
         ComponentUtils.pushObjectToContext(message, UifConstants.ContextVariableNames.NODE, nodeData);
         message.setMessageText(nodeData.getNodeLabel());
         node.setNodeLabel(message);
 
         Group nodeGroup = ComponentUtils.copyComponent(prototype.getDataGroupPrototype(), bindingPrefix + ".data",
                 parentNode);
         ComponentUtils.pushObjectToContext(nodeGroup, UifConstants.ContextVariableNames.NODE, nodeData);
 
         String nodePath = bindingPrefix + ".data";
         if (StringUtils.isNotBlank(getBindingInfo().getBindingObjectPath())) {
             nodePath = getBindingInfo().getBindingObjectPath() + "." + nodePath;
         }
         ComponentUtils.pushObjectToContext(nodeGroup, UifConstants.ContextVariableNames.NODE_PATH, nodePath);
         node.setData(nodeGroup);
 
         List<Node<Group, Message>> nodeChildren = new ArrayList<Node<Group, Message>>();
 
         int childIndex = 0;
         for (Node<Object, String> childDataNode : nodeData.getChildren()) {
             String nextBindingPrefix = bindingPrefix + ".children[" + childIndex + "]";
             Node<Group, Message> childNode = buildTreeNode(childDataNode, nextBindingPrefix,
                     "_node_" + childIndex + ("root".equals(parentNode) ? "_parent_" : "_parent") + parentNode);
 
             nodeChildren.add(childNode);
 
             // Don't forget about me:
             ++childIndex;
         }
         node.setChildren(nodeChildren);
 
         return node;
     }
 
     /**
      * Gets the NodePrototype to use for the given Node
      */
     private NodePrototype getNodePrototype(Node<Object, String> nodeData) {
         NodePrototype result = null;
         if (nodeData != null && nodeData.getData() != null) {
             Class<?> dataClass = nodeData.getData().getClass();
             result = nodePrototypeMap.get(dataClass);
 
             // somewhat lame fallback - to do this right we'd find all entries that are assignable from the data class
             // and then figure out which one is the closest relative
             if (result == null) {
                 for (Map.Entry<Class<?>, NodePrototype> prototypeEntry : nodePrototypeMap.entrySet()) {
                     if (prototypeEntry.getKey().isAssignableFrom(dataClass)) {
                         result = prototypeEntry.getValue();
                         break;
                     }
                 }
             }
         }
 
         if (result == null) {
             result = defaultNodePrototype;
         }
 
         return result;
     }
 
     /**
      * @see org.kuali.rice.krad.uif.component.Component#getComponentsForLifecycle()
      */
     @Override
     public List<Component> getComponentsForLifecycle() {
         List<Component> components = super.getComponentsForLifecycle();
 
         components.add(tree);
         addNodeComponents(treeGroups.getRootElement(), components);
 
         return components;
     }
 
     /**
      * @see org.kuali.rice.krad.uif.component.Component#getComponentPrototypes()
      */
     @Override
     public List<Component> getComponentPrototypes() {
         List<Component> components = super.getComponentPrototypes();
 
         if (defaultNodePrototype != null) {
             components.add(defaultNodePrototype.getLabelPrototype());
             components.add(defaultNodePrototype.getDataGroupPrototype());
         }
 
         if (nodePrototypeMap != null) {
             for (Map.Entry<Class<?>, NodePrototype> prototypeEntry : nodePrototypeMap.entrySet()) {
                 NodePrototype prototype = prototypeEntry.getValue();
                 if (prototype != null) {
                     components.add(prototype.getLabelPrototype());
                     components.add(prototype.getDataGroupPrototype());
                 }
             }
         }
 
         return components;
     }
 
     /**
      * Retrieves the <code>Component</code> instances from the node for building the nested
      * components list
      *
      * @param node node to pull components from
      * @param components list to add components to
      */
     protected void addNodeComponents(Node<Group, Message> node, List<Component> components) {
         if (node != null) {
             components.add(node.getNodeLabel());
             components.add(node.getData());
 
             for (Node<Group, Message> nodeChild : node.getChildren()) {
                 addNodeComponents(nodeChild, components);
             }
         }
     }
 
     @BeanTagAttribute(name = "propertyName")
     public String getPropertyName() {
         return propertyName;
     }
 
     public void setPropertyName(String propertyName) {
         this.propertyName = propertyName;
     }
 
     @BeanTagAttribute(name = "bindingInfo", type = BeanTagAttribute.AttributeType.SINGLEBEAN)
     public BindingInfo getBindingInfo() {
         return bindingInfo;
     }
 
     public void setBindingInfo(BindingInfo bindingInfo) {
         this.bindingInfo = bindingInfo;
     }
 
     /**
      * @return the defaultNodePrototype
      */
     @BeanTagAttribute(name = "defaultNodePrototype", type = BeanTagAttribute.AttributeType.SINGLEBEAN)
     public NodePrototype getDefaultNodePrototype() {
         return this.defaultNodePrototype;
     }
 
     /**
      * @param defaultNodePrototype the defaultNodePrototype to set
      */
     public void setDefaultNodePrototype(NodePrototype defaultNodePrototype) {
         this.defaultNodePrototype = defaultNodePrototype;
     }
 
     /**
      * @return the nodePrototypeMap
      */
     @BeanTagAttribute(name = "NodePrototypeMap", type = BeanTagAttribute.AttributeType.MAPBEAN)
     public Map<Class<?>, NodePrototype> getNodePrototypeMap() {
         return this.nodePrototypeMap;
     }
 
     /**
      * @param nodePrototypeMap the nodePrototypeMap to set
      */
     public void setNodePrototypeMap(Map<Class<?>, NodePrototype> nodePrototypeMap) {
         this.nodePrototypeMap = nodePrototypeMap;
     }
 
     @BeanTagAttribute(name = "treeGroups", type = BeanTagAttribute.AttributeType.MAP2BEAN)
     public Tree<Group, Message> getTreeGroups() {
         return treeGroups;
     }
 
     public void setTreeGroups(Tree<Group, Message> treeGroups) {
         this.treeGroups = treeGroups;
     }
 
     @BeanTagAttribute(name = "tree", type = BeanTagAttribute.AttributeType.SINGLEBEAN)
     public org.kuali.rice.krad.uif.widget.Tree getTree() {
         return tree;
     }
 
     public void setTree(org.kuali.rice.krad.uif.widget.Tree tree) {
         this.tree = tree;
     }
 
     /**
      * @see org.kuali.rice.krad.uif.component.ComponentBase#copy()
      */
     @Override
     protected <T> void copyProperties(T component) {
         super.copyProperties(component);
         TreeGroup treeGroupCopy = (TreeGroup) component;
         treeGroupCopy.setPropertyName(this.propertyName);
 
         if (this.bindingInfo != null) {
             treeGroupCopy.setBindingInfo((BindingInfo) this.bindingInfo.copy());
         }
 
         if (this.defaultNodePrototype != null) {
             treeGroupCopy.setDefaultNodePrototype((NodePrototype) this.defaultNodePrototype.copy());
         }
 
         if (this.treeGroups != null) {
             Tree<Group, Message> treeGroupsCopy = new Tree<Group, Message>();
             treeGroupsCopy.setRootElement(copyNode(this.treeGroups.getRootElement()));
 
             treeGroupCopy.setTreeGroups(treeGroupsCopy);
         }
 
         if (this.tree != null) {
             treeGroupCopy.setTree((org.kuali.rice.krad.uif.widget.Tree) this.tree.copy());
         }
 
         if (this.nodePrototypeMap != null) {
             Map<Class<?>, NodePrototype> nodePrototypeMapCopy = Maps.newHashMapWithExpectedSize(
                     this.getNodePrototypeMap().size());
             for (Map.Entry<Class<?>, NodePrototype> nodePrototypeMapEntry : nodePrototypeMap.entrySet()) {
                 NodePrototype prototypeCopy = nodePrototypeMapEntry.getValue().copy();
                 nodePrototypeMapCopy.put(nodePrototypeMapEntry.getKey(), prototypeCopy);
             }
 
             treeGroupCopy.setNodePrototypeMap(nodePrototypeMapCopy);
         }
     }
 
     /**
      * Copies a {@link Node} instance and then recursively copies each of its child nodes
      *
      * @param node node instance to copy
      * @return new node instance copied from given node
      */
     protected Node<Group, Message> copyNode(Node<Group, Message> node) {
         Node<Group, Message> nodeCopy = new Node<Group, Message>();
 
         if (node == null) {
             return null;
         }
 
         if (node.getData() != null) {
             nodeCopy.setData((Group) node.getData().copy());
         }
 
         if (node.getNodeLabel() != null) {
             nodeCopy.setNodeLabel((Message) node.getNodeLabel().copy());
         }
 
         if (node.getChildren() != null) {
             List<Node<Group, Message>> childrenCopy = new ArrayList<Node<Group, Message>>();
             for (Node<Group, Message> childNode : node.getChildren()) {
                 childrenCopy.add(copyNode(childNode));
             }
 
             nodeCopy.setChildren(childrenCopy);
         }
 
         return nodeCopy;
     }
 }