/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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.apache.myfaces.application;
  
  import org.apache.myfaces.shared.util.ClassUtils;
  
  import javax.faces.component.UIComponent;
  import javax.faces.component.UIViewRoot;
  import javax.faces.context.FacesContext;
  
  import java.io.Serializable;
  import java.util.ArrayList;
  import java.util.List;
  import java.util.Map;
  
  /**
   * @author Manfred Geiler (latest modification by $Author$)
   * @version $Revision$ $Date$
   */
  public class TreeStructureManager
  {
      public TreeStructureManager()
      {
          //_facesContext = facesContext;
      }
  
      public Object buildTreeStructureToSave(UIViewRoot viewRoot)
      {
          return internalBuildTreeStructureToSave(viewRoot);
      }
  
      private TreeStructComponent internalBuildTreeStructureToSave(UIComponent component)
      {
          TreeStructComponent structComp = new TreeStructComponent(component.getClass().getName(),
                                                                   component.getId());
  
          //children
          if (component.getChildCount() > 0)
          {
              List<TreeStructComponent> structChildList = new ArrayList<TreeStructComponent>();
              for (int i = 0, childCount = component.getChildCount(); i < childCount; i++)
              {
                  UIComponent child = component.getChildren().get(i);
                  if (!child.isTransient())
                  {
                      TreeStructComponent structChild = internalBuildTreeStructureToSave(child);
                      structChildList.add(structChild);
                  }
              }
              
              TreeStructComponent[] childArray = structChildList.toArray(new TreeStructComponent[structChildList.size()]);
              structComp.setChildren(childArray);
          }
  
          //facets
          
          if (component.getFacetCount() > 0)
          {
              Map<String, UIComponent> facetMap = component.getFacets();
              List<Object[]> structFacetList = new ArrayList<Object[]>();
              for (Map.Entry<String, UIComponent> entry : facetMap.entrySet())
              {
                  UIComponent child = entry.getValue();
                  if (!child.isTransient())
                  {
                      String facetName = entry.getKey();
                      TreeStructComponent structChild = internalBuildTreeStructureToSave(child);
                      structFacetList.add(new Object[] {facetName, structChild});
                  }
              }
              
              Object[] facetArray = structFacetList.toArray(new Object[structFacetList.size()]);
              structComp.setFacets(facetArray);
          }
  
          return structComp;
      }
  
  
      public UIViewRoot restoreTreeStructure(Object treeStructRoot)
      {
          if (treeStructRoot instanceof TreeStructComponent)
          {
             return (UIViewRoot)internalRestoreTreeStructure((TreeStructComponent)treeStructRoot, true);
         }
         
         
         throw new IllegalArgumentException("TreeStructure of type " + treeStructRoot.getClass().getName() + 
                                            " is not supported.");
         
     }
 
     private UIComponent internalRestoreTreeStructure(TreeStructComponent treeStructComp, boolean checkViewRoot)
     {
         String compClass = treeStructComp.getComponentClass();
         String compId = treeStructComp.getComponentId();
         UIComponent component = (UIComponent)ClassUtils.newInstance(compClass);
         component.setId(compId);
 
         if (checkViewRoot && component instanceof UIViewRoot)
         {
             FacesContext.getCurrentInstance().setViewRoot((UIViewRoot) component);
         }
         //children
         TreeStructComponent[] childArray = treeStructComp.getChildren();
         if (childArray != null)
         {
             List<UIComponent> childList = component.getChildren();
             for (int i = 0, len = childArray.length; i < len; i++)
             {
                 UIComponent child = internalRestoreTreeStructure(childArray[i], false);
                 childList.add(child);
             }
         }
 
         //facets
         Object[] facetArray = treeStructComp.getFacets();
         if (facetArray != null)
         {
             Map<String, UIComponent> facetMap = component.getFacets();
             for (int i = 0, len = facetArray.length; i < len; i++)
             {
                 Object[] tuple = (Object[])facetArray[i];
                 String facetName = (String)tuple[0];
                 TreeStructComponent structChild = (TreeStructComponent)tuple[1];
                 UIComponent child = internalRestoreTreeStructure(structChild, false);
                 facetMap.put(facetName, child);
             }
         }
 
         return component;
     }
 
 
     public static class TreeStructComponent
             implements Serializable
     {
         private static final long serialVersionUID = 5069109074684737231L;
         private String _componentClass;
         private String _componentId;
         private TreeStructComponent[] _children = null;    // Array of children
         private Object[] _facets = null;            // Array of Array-tuples with Facetname and TreeStructComponent
 
         TreeStructComponent(String componentClass, String componentId)
         {
             _componentClass = componentClass;
             _componentId = componentId;
         }
 
         public String getComponentClass()
         {
             return _componentClass;
         }
 
         public String getComponentId()
         {
             return _componentId;
         }
 
         void setChildren(TreeStructComponent[] children)
         {
             _children = children;
         }
 
         TreeStructComponent[] getChildren()
         {
             return _children;
         }
 
         Object[] getFacets()
         {
             return _facets;
         }
 
         void setFacets(Object[] facets)
         {
             _facets = facets;
         }
     }
 
 }