/*
 * Copyright (C) 2005-2010 Alfresco Software Limited.
 *
 * This file is part of Alfresco
 *
 * Alfresco is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Alfresco is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
 */
package org.alfresco.web.bean.wcm;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import org.alfresco.repo.content.MimetypeMap;
import org.alfresco.service.cmr.avm.locking.AVMLock;
import org.alfresco.service.cmr.repository.ContentWriter;
import org.alfresco.web.app.Application;
import org.alfresco.web.forms.Form;
import org.alfresco.web.forms.FormInstanceData;
import org.alfresco.web.forms.FormNotFoundException;
import org.alfresco.web.forms.Rendition;
import org.alfresco.web.forms.XMLUtil;
import org.alfresco.web.ui.common.Utils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * Bean implementation for the "Edit Web Content Wizard" dialog
 */
public class EditWebContentWizard extends CreateWebContentWizard
{
   private static final long serialVersionUID = 439996926303151006L;

   private static final Log logger = LogFactory.getLog(EditWebContentWizard.class);
   
   private AVMNode avmNode;
   private Form form;
   
   protected List<String> locksPresentAtInit = null;

   // ------------------------------------------------------------------------------
   // Wizard implementation
   
   @Override
   public void init(final Map<String, String> parameters)
   {
      super.init(parameters);
      this.avmNode = this.avmBrowseBean.getAvmActionNode();
      if (this.avmNode == null)
      {
         throw new IllegalArgumentException("Edit Form wizard requires action node context.");
      }

      if (logger.isDebugEnabled())
         logger.debug("path is " + this.avmNode.getPath());

      this.createdPath = AVMUtil.getCorrespondingPathInPreviewStore(this.avmNode.getPath());
      this.formInstanceData = this.getFormsService().getFormInstanceData(-1, this.createdPath);
      final WebProject webProject = new WebProject(this.createdPath);
      try
      {
         this.formName = this.formInstanceData.getForm().getName();
         this.form = webProject.getForm(this.formName);
      }
      catch (FormNotFoundException fnfe)
      {
         Utils.addErrorMessage(fnfe.getMessage(), fnfe);
      }
      this.content = this.getAvmService().getContentReader(-1, this.createdPath).getContentString();
      this.fileName = this.formInstanceData.getName();
      this.mimeType = MimetypeMap.MIMETYPE_XML;
      
      // calculate which locks are present at init time
      this.locksPresentAtInit = new ArrayList<String>(4);
      AVMLock lock = this.getAvmLockingService().getLock(AVMUtil.getStoreId(this.createdPath),
               AVMUtil.getStoreRelativePath(this.createdPath));
      if (lock != null)
      {
         this.locksPresentAtInit.add(this.createdPath);
         
         if (logger.isDebugEnabled())
            logger.debug("Lock exists for xml instance " + this.createdPath + " at initialisation");
      }
      
      for (final Rendition r : this.formInstanceData.getRenditions())
      {
         String path = r.getPath();
         lock = this.getAvmLockingService().getLock(AVMUtil.getStoreId(path),
                                               AVMUtil.getStoreRelativePath(path));
         if (lock != null)
         {
            this.locksPresentAtInit.add(path);
            
            if (logger.isDebugEnabled())
               logger.debug("Lock exists for rendition " + path + " at initialisation");
         }
      }
   }

   @Override
   public String cancel()
   {
      if (this.formInstanceData != null && this.renditions != null)
      {
         if (this.locksPresentAtInit.contains(this.createdPath) == false)
         {
            // there wasn't a lock on the form at the start of the 
            // wizard so remove the one present now
            if (logger.isDebugEnabled())
               logger.debug("removing form instance data lock from " + 
                     AVMUtil.getCorrespondingPathInMainStore(this.createdPath) +
                     " as user chose to cancel and it wasn't present at initialisation");
            
            this.getAvmLockingService().removeLock(AVMUtil.getStoreId(this.createdPath),
                                                   AVMUtil.getStoreRelativePath(this.createdPath));
         }

         for (Rendition r : this.renditions)
         {
            String path = r.getPath();
            
            if (this.locksPresentAtInit.contains(path) == false)
            {
               // there wasn't a lock on the rendition at the start of
               // the wizard so remove the one present now
               if (logger.isDebugEnabled())
                  logger.debug("removing lock from rendition " + 
                        AVMUtil.getCorrespondingPathInMainStore(path) + 
                        " as user chose to cancel and it wasn't present at initialisation");

               this.getAvmLockingService().removeLock(AVMUtil.getStoreId(path),
                                                 AVMUtil.getStoreRelativePath(path));
            }
         }
      }
      
      return super.cancel();
   }

   @Override
   public String back()
   {
      if ("content".equals(Application.getWizardManager().getCurrentStepName()))
      {
         //override in order not to delete these items
         this.formInstanceData = null;
         this.renditions = null;
      }
      return super.back();
   }

   @Override
   protected void saveContent()
      throws Exception
   {
      if (logger.isDebugEnabled())
         logger.debug("saving " + this.createdPath);
      
      AVMLock lock = this.getAvmLockingService().getLock(AVMUtil.getStoreId(this.createdPath),
                                                    AVMUtil.getStoreRelativePath(this.createdPath));
      if (lock != null)
      {
         if (logger.isDebugEnabled())
            logger.debug("transferring lock from " + lock.getStore() + 
                         " to " + AVMUtil.getStoreName(this.createdPath));
         
         this.getAvmLockingService().modifyLock(AVMUtil.getStoreId(this.createdPath),
                                           AVMUtil.getStoreRelativePath(this.createdPath),
                                           null,
                                           AVMUtil.getStoreName(this.createdPath),
                                           null,
                                           null);
      }

      final ContentWriter writer = this.getAvmService().getContentWriter(this.createdPath);
      this.content = XMLUtil.toString(this.getInstanceDataDocument(), false);
      writer.putContent(this.content);

      // XXXarielb might not need to do this reload
      this.formInstanceData = this.getFormsService().getFormInstanceData(-1, this.createdPath);
      for (final Rendition r : this.formInstanceData.getRenditions())
      {
         lock = this.getAvmLockingService().getLock(AVMUtil.getStoreId(r.getPath()),
                                               AVMUtil.getStoreRelativePath(r.getPath()));
         if (lock != null)
         {
            if (logger.isDebugEnabled())
               logger.debug("transferring lock from " + lock.getStore() + 
                            " to " + AVMUtil.getStoreName(r.getPath()));
            
            this.getAvmLockingService().modifyLock(AVMUtil.getStoreId(r.getPath()),
                                              AVMUtil.getStoreRelativePath(r.getPath()),
                                              null,
                                              AVMUtil.getStoreName(r.getPath()),
                                              null,
                                              null);
         }
      }
      
      final List<FormInstanceData.RegenerateResult> result = this.formInstanceData.regenerateRenditions();
      this.renditions = new LinkedList<Rendition>();
      for (FormInstanceData.RegenerateResult rr : result)
      {
         if (rr.getException() != null)
         {
            Utils.addErrorMessage("error regenerating rendition using " + rr.getRenderingEngineTemplate().getName() + 
                                  ": " + rr.getException().getMessage(),
                                  rr.getException());
            
            // if the renditions were locked before the regenerate, move the lock back to main store
            String path = rr.getPath();
            
            if (this.locksPresentAtInit.contains(path))
            {
               if (logger.isDebugEnabled())
                  logger.debug("transferring existing lock for " + path + 
                               " back to " + AVMUtil.getCorrespondingMainStoreName(AVMUtil.getStoreName(path)));
               
               this.getAvmLockingService().modifyLock(AVMUtil.getStoreId(path),
                                                 AVMUtil.getStoreRelativePath(path),
                                                 null,
                                                 AVMUtil.getCorrespondingMainStoreName(AVMUtil.getStoreName(path)),
                                                 null,
                                                 null);
            }
         }
         else
         {
            final Rendition r = rr.getRendition();
            this.renditions.add(r);
            
            if (logger.isDebugEnabled())
               logger.debug("transferring lock for " + r.getPath() + 
                            " back to " + AVMUtil.getCorrespondingMainStoreName(AVMUtil.getStoreName(r.getPath())));
            
            this.getAvmLockingService().modifyLock(AVMUtil.getStoreId(r.getPath()),
                                              AVMUtil.getStoreRelativePath(r.getPath()),
                                              null,
                                              AVMUtil.getCorrespondingMainStoreName(AVMUtil.getStoreName(r.getPath())),
                                              null,
                                              null);
         }
      }
      
      if (logger.isDebugEnabled())
         logger.debug("transferring form instance data lock back to " + 
                      AVMUtil.getCorrespondingMainStoreName(AVMUtil.getStoreName(this.createdPath)));
      
      this.getAvmLockingService().modifyLock(AVMUtil.getStoreId(this.createdPath),
                                        AVMUtil.getStoreRelativePath(this.createdPath),
                                        null,
                                        AVMUtil.getCorrespondingMainStoreName(AVMUtil.getStoreName(this.createdPath)),
                                        null,
                                        null);
   }

   /** Indicates whether or not the wizard is currently in edit mode */
   @Override
   public boolean getEditMode()
   {
      return true;
   }

   @Override
   public boolean getSubmittable()
   {
      return !AVMUtil.isWorkflowStore(AVMUtil.getStoreName(this.createdPath));
   }

   /** 
    * Overridden to avoid calling getWebProject since potentially there is no web project 
    * context in workflow scenario.
    */
   @Override 
   public Form getForm()
   {
      return this.form;
   }
}