/**
    * This file Copyright (c) 2011-2014 Magnolia International
    * Ltd.  (http://www.magnolia-cms.com). All rights reserved.
    *
    *
    * This file is dual-licensed under both the Magnolia
    * Network Agreement and the GNU General Public License.
    * You may elect to use one or the other of these licenses.
    *
   * This file is distributed in the hope that it will be
   * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the
   * implied warranty of MERCHANTABILITY or FITNESS FOR A
   * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT.
   * Redistribution, except as permitted by whichever of the GPL
   * or MNA you select, is prohibited.
   *
   * 1. For the GPL license (GPL), you can redistribute and/or
   * modify this file under the terms of the GNU General
   * Public License, Version 3, as published by the Free Software
   * Foundation.  You should have received a copy of the GNU
   * General Public License, Version 3 along with this program;
   * if not, write to the Free Software Foundation, Inc., 51
   * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
   *
   * 2. For the Magnolia Network Agreement (MNA), this file
   * and the accompanying materials are made available under the
   * terms of the MNA which accompanies this distribution, and
   * is available at http://www.magnolia-cms.com/mna.html
   *
   * Any modifications to this file must keep this entire header
   * intact.
   *
   */
  package info.magnolia.templating.editor.client.model;
  
  import info.magnolia.templating.editor.client.dom.MgnlElement;
  import info.magnolia.templating.editor.client.model.focus.FocusModel;
  import info.magnolia.templating.editor.client.model.focus.FocusModelImpl3;
  import info.magnolia.templating.editor.client.widget.controlbar.AbstractBar;
  import info.magnolia.templating.editor.client.widget.controlbar.AreaEndBar;
  import info.magnolia.templating.editor.client.widget.overlay.AbstractOverlay;
  import info.magnolia.templating.editor.client.widget.placeholder.AreaPlaceHolder;
  import info.magnolia.templating.editor.client.widget.placeholder.ComponentPlaceHolder;
  
  import java.util.HashMap;
  import java.util.LinkedList;
  import java.util.List;
  import java.util.Map;
  import com.google.gwt.dom.client.Node;
  
  import com.google.gwt.dom.client.Element;
  
  /**
   * Singleton keeping the model.
   */
  public class ModelStorage {
  
      private static ModelStorage storage = null;
  
      private FocusModel focusModel = new FocusModelImpl3(this);
  
      private Map<MgnlElement, AbstractBar> editBars = new HashMap<MgnlElement, AbstractBar>();
      private Map<MgnlElement, AbstractOverlay> overlays = new HashMap<MgnlElement, AbstractOverlay>();
      private Map<MgnlElement, List<Element>> elements = new HashMap<MgnlElement, List<Element>>();
      private Map<Element, MgnlElement> mgnlElements = new HashMap<Element, MgnlElement>();
      private Map<MgnlElement, AreaPlaceHolder> areaPlaceHolders = new HashMap<MgnlElement, AreaPlaceHolder>();
      private Map<MgnlElement, ComponentPlaceHolder> componentPlaceHolders = new HashMap<MgnlElement, ComponentPlaceHolder>();
      private Map<MgnlElement, AreaEndBar> areaEndBars = new HashMap<MgnlElement, AreaEndBar>();
  
  
      public List<MgnlElement> rootElements = new LinkedList<MgnlElement>();
  
  
      private MgnlElement selectedMgnlElement = null;
  
      public static ModelStorage getInstance() {
          if (storage == null) {
              storage = new ModelStorage();
          }
          return storage;
      }
  
      public void addOverlay(MgnlElement mgnlElement, AbstractOverlay overlayWidget) {
          overlays.put(mgnlElement, overlayWidget);
      }
  
      public AbstractOverlay getOverlay(MgnlElement mgnlElement) {
          return overlays.get(mgnlElement);
      }
  
      public void addEditBar(MgnlElement mgnlElement, AbstractBar editBar) {
          editBars.put(mgnlElement, editBar);
      }
  
      public AbstractBar getEditBar(MgnlElement mgnlElement) {
          return editBars.get(mgnlElement);
      }
  
      public void addElement(MgnlElement mgnlElement, Element element) {
 
         if (mgnlElement == null || element == null) {
             return;
         }
         mgnlElements.put(element, mgnlElement);
 
         if (elements.get(mgnlElement) != null) {
             elements.get(mgnlElement).add(element);
         }
         else {
             List<Element> elList = new LinkedList<Element>();
             elList.add(element);
             elements.put(mgnlElement, elList);
         }
     }
 
     public void addElements(MgnlElement mgnlElement, Element element) {
 
         if (mgnlElement == null || element == null) {
             return;
         }
         addElement(mgnlElement, element);
 
         for (int i = 0; i < element.getChildCount(); i++) {
             Node childNode = element.getChild(i);
             if (childNode.getNodeType() == Node.ELEMENT_NODE) {
                 Element child = childNode.cast();
                 addElements(mgnlElement, child);
             }
         }
     }
 
     public MgnlElement getMgnlElement(Element element) {
         MgnlElement mgnlElement = mgnlElements.get(element);
         while (mgnlElement == null && element.hasParentElement()) {
             element = element.getParentElement();
             mgnlElement = mgnlElements.get(element);
         }
         return mgnlElement;
     }
 
     public List<Element> getElements(MgnlElement mgnlElement) {
         return elements.get(mgnlElement);
     }
 
     public void addRoot(MgnlElement boundary) {
         this.rootElements.add(boundary);
     }
 
     public List<MgnlElement> getRootElements() {
         return rootElements;
     }
 
     public void setSelectedMgnlElement(MgnlElement selectedMgnlElement) {
         this.selectedMgnlElement = selectedMgnlElement;
     }
 
     public MgnlElement getSelectedMgnlElement() {
         return selectedMgnlElement;
     }
 
     public FocusModel getFocusModel() {
         return focusModel;
     }
 
     public void addAreaPlaceHolder(MgnlElement mgnlElement, AreaPlaceHolder placeHolder) {
         areaPlaceHolders.put(mgnlElement, placeHolder);
     }
 
     public AreaPlaceHolder getAreaPlaceHolder(MgnlElement mgnlElement) {
         return areaPlaceHolders.get(mgnlElement);
     }
 
     public void addComponentPlaceHolder(MgnlElement mgnlElement, ComponentPlaceHolder placeHolder) {
         componentPlaceHolders.put(mgnlElement, placeHolder);
     }
 
     public ComponentPlaceHolder getComponentPlaceHolder(MgnlElement mgnlElement) {
         return componentPlaceHolders.get(mgnlElement);
     }
 
     public void addAreaEndBar(MgnlElement mgnlElement, AreaEndBar areaEndBar) {
         areaEndBars.put(mgnlElement, areaEndBar);
     }
 
     public AreaEndBar getAreaEndBar(MgnlElement mgnlElement) {
         return areaEndBars.get(mgnlElement);
     }
 
     public void removeMgnlElement(MgnlElement mgnlElement) {
 
         // remove all occurrences of the element
         if (mgnlElements.containsValue(mgnlElement)) {
             while(mgnlElements.values().remove(mgnlElement));
         }
         elements.remove(mgnlElement);
 
         // if the element is a root node, add all children to root list
         if (rootElements.contains(mgnlElement)) {
             rootElements.remove(mgnlElement);
             rootElements.addAll(mgnlElement.getChildren());
         }
     }
 
     public MgnlElement findMgnlElementByContentId(String contentId) {
         for (MgnlElement element : elements.keySet()) {
             if(contentId.equals(element.getAttribute("content"))){
                 return element;
             }
         }
         return null;
     }
 
 }