/**
    * This file Copyright (c) 2013 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.module.delta;
  
  import info.magnolia.cms.util.RequestDispatchUtil;
  import info.magnolia.module.InstallContext;
  import info.magnolia.repository.RepositoryConstants;
  
  import javax.jcr.Node;
  import javax.jcr.NodeIterator;
  import javax.jcr.RepositoryException;
  import javax.jcr.Session;
  import javax.jcr.query.Query;
  import javax.jcr.query.QueryManager;
  import javax.jcr.query.QueryResult;
  
  import org.apache.commons.lang.StringUtils;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  /**
   * Task for changing Virtual URI mappings - replaces old toURI and oldURI values with new ones.
   */
  public class ChangeVirtualUriMappingTask extends AbstractRepositoryTask {
  
      private static final Logger log = LoggerFactory.getLogger(ChangeVirtualUriMappingTask.class);
  
      protected static final String VIRTUAL_URI_MAPPING = "virtualURIMapping";
      protected static final String FROM_URI = "fromURI";
      protected static final String TO_URI = "toURI";
  
      private String currentValue;
      private String newValue;
      private String query;
  
      public ChangeVirtualUriMappingTask(String name, String description, String currentValue, String newValue) {
          this(name, description, currentValue, newValue, new String[] { RequestDispatchUtil.FORWARD_PREFIX, RequestDispatchUtil.REDIRECT_PREFIX, RequestDispatchUtil.PERMANENT_PREFIX });
      }
  
      public ChangeVirtualUriMappingTask(String name, String description, String currentValue, String newValue, String[] includedPrefixes) {
  
          super(name, description);
          this.currentValue = currentValue;
          this.newValue = newValue;
  
          query = "select * from [nt:base] as t WHERE ISDESCENDANTNODE([/modules])  AND (contains(t.*,'" + this.currentValue + "')";
          for (String prefix : includedPrefixes) {
              query += " OR contains(t.*,'" + prefix + this.currentValue + "')";
          }
          query += ")";
      }
  
      @Override
      protected void doExecute(InstallContext installContext) throws RepositoryException, TaskExecutionException {
  
          Session config = installContext.getJCRSession(RepositoryConstants.CONFIG);
          QueryManager qm = config.getWorkspace().getQueryManager();
          Query q = qm.createQuery(query, Query.JCR_SQL2);
          QueryResult queryResult = q.execute();
          NodeIterator nodeIterator = queryResult.getNodes();
  
          while (nodeIterator.hasNext()) {
              Node node = nodeIterator.nextNode();
              if (!VIRTUAL_URI_MAPPING.equals(node.getParent().getName())) {
                  continue;
              }
  
              if (node.hasProperty(FROM_URI)) {
                 String fromUri = node.getProperty(FROM_URI).getString();
 
                 if (fromUri.equals(currentValue)) {
                     node.setProperty(FROM_URI, newValue);
                     log.info("'{}/{}' changed from '{}' to '{}'", node.getPath(), FROM_URI, fromUri, newValue);
                 }
             }
 
             if (node.hasProperty(TO_URI)) {
                 String oldToUri = node.getProperty(TO_URI).getString();
                 String prefix = StringUtils.substringBefore(oldToUri, ":");
 
                 if (oldToUri.equals(prefix + ":" + currentValue)) {
                     String newToUri = prefix + ":" + newValue;
                     node.setProperty(TO_URI, newToUri);
                     log.info("'{}/{}' changed from '{}' to '{}'", node.getPath(), TO_URI, oldToUri, newToUri);
                 }
             }
         }
     }
 }