/**
    * This file Copyright (c) 2017 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.importexport.exporter;
  
  import info.magnolia.jcr.util.NodeTypes;
  
  import javax.jcr.Node;
  import javax.jcr.NodeIterator;
  import javax.jcr.PropertyType;
  import javax.jcr.RepositoryException;
  import javax.jcr.Session;
  import javax.jcr.Value;
  
  import org.apache.commons.lang3.StringUtils;
  import org.apache.jackrabbit.JcrConstants;
  import org.apache.jackrabbit.commons.xml.DocumentViewExporter;
  import org.apache.jackrabbit.util.ISO9075;
  import org.apache.jackrabbit.value.ValueHelper;
  import org.xml.sax.ContentHandler;
  import org.xml.sax.SAXException;
  
  /**
   * {@link org.apache.jackrabbit.commons.xml.Exporter} for YAML format.
   */
  public class YamlExporter extends DocumentViewExporter {
  
      public YamlExporter(Session session, ContentHandler handler, boolean recurse, boolean binary) {
          super(session, handler, recurse, binary);
      }
  
      @Override
      protected void exportNode(String uri, String local, Node node) throws RepositoryException, SAXException {
          if (node.getDepth() == 0) { //don't include root node
              final NodeIterator iterator = node.getNodes();
              while (iterator.hasNext()) {
                  final Node child = iterator.nextNode();
                  super.exportNode(StringUtils.EMPTY, serializeKey(uri, child.getName()), child); //omit jcr: prefix for all child nodes
              }
          } else {
              super.exportNode(uri, serializeKey(uri, local), node);
          }
      }
  
      @Override
      protected void exportProperty(String uri, String local, Value value) throws RepositoryException {
          //mgnl:contentNode is the default nodeType, no need to export it, this saves a lot of lines e.g. in the configuration workspace export
          if (JcrConstants.JCR_PRIMARYTYPE.equals(getXMLName(uri, local)) && NodeTypes.ContentNode.NAME.equals(value.getString())) {
              return;
          }
          final String name = ISO9075.encode(serializeKey(uri, local));
          final String attribute = serializeValue(value, value.getType());
          final String propertyType = getPropertyTypePrefix(value);
          super.addAttribute(uri, name, propertyType + attribute);
      }
  
      @Override
      protected void exportProperty(String uri, String local, int type, Value[] values) {
          try {
              final StringBuilder attribute = new StringBuilder("");
              for (Value value : values) {
                  attribute.append(getPropertyTypePrefix(value));
                  attribute.append(serializeValue(value, type));
                  attribute.append(",");
              }
              final String name = StringUtils.isEmpty(uri) ? ISO9075.encode(serializeKey(uri, local)) : local;
              final String attributeString = "[" + (values.length == 0 ? "" : attribute.substring(0, attribute.length() - 1)) + "]";
              super.addAttribute(uri, name, attributeString);
          } catch (RepositoryException e) {
             throw new RuntimeException(e);
         }
     }
 
     private String serializeValue(Value value, int type) throws RepositoryException {
         String stringValue = ValueHelper.serialize(value, false);
         switch (type) {
         case PropertyType.STRING:
         case PropertyType.URI:
         case PropertyType.NAME:
         case PropertyType.PATH:
             stringValue = "'" + stringValue.replace("'", "''") + "'";
         }
         return stringValue;
     }
 
     private String getPropertyTypePrefix(Value value) throws RepositoryException {
         String type = "";
 
         switch (value.getType()) {
         case PropertyType.BINARY:
         case PropertyType.PATH:
         case PropertyType.REFERENCE:
         case PropertyType.WEAKREFERENCE:
         case PropertyType.URI:
             type = "!" + StringUtils.lowerCase(PropertyType.nameFromValue(value.getType())) + " ";
         }
         return type;
     }
 
     private String serializeKey(String uri, String local) throws RepositoryException {
         return StringUtils.isEmpty(uri) ? "'" + local.replace("'", "''") + "'" : local;
     }
 }