/*
 * Copyright (C) 2005-2007 Alfresco Software Limited.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.

 * This program 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 General Public License for more details.

 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.

 * As a special exception to the terms and conditions of version 2.0 of 
 * the GPL, you may redistribute this Program in connection with Free/Libre 
 * and Open Source Software ("FLOSS") applications as described in Alfresco's 
 * FLOSS exception.  You should have recieved a copy of the text describing 
 * the FLOSS exception, and it is also available here: 
 * http://www.alfresco.com/legal/licensing
 */
package org.alfresco.web.bean.wcm;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.faces.context.FacesContext;

import org.alfresco.linkvalidation.LinkValidationReport;
import org.alfresco.model.WCMAppModel;
import org.alfresco.repo.avm.AVMNodeConverter;
import org.alfresco.service.cmr.avm.AVMService;
import org.alfresco.service.cmr.repository.NodeRef;
import org.alfresco.service.cmr.repository.NodeService;
import org.alfresco.web.bean.repository.Repository;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * Object used to retrieve and store the state of a link validaton process.
 * The object is given an initial link validation report object (the result
 * of a LinkValidationService service call) which is used by components
 * to display the result of the links check.
 * <p>
 * Further reports can then subsequently be given to this object at which
 * point a difference is calculated i.e. which links (if any) have been
 * fixed. This allows components to display the progress of link fixing.
 * </p>
 * 
 * @author gavinc
 */
public class LinkValidationState implements Serializable
{
   private static final long serialVersionUID = 3798453845183433843L;
   
   private String store;
   private String webapp;
   private boolean checkBeenReRun = false;
   private boolean maxLinksReached = false;
   private Date checkCompletedAt;
   
   private int noFilesCheckedStart = -1;
   private int noFilesCheckedLast = -1;
   
   private int noLinksCheckedStart = -1;
   private int noLinksCheckedLast = -1;
   
   private int noBrokenFilesStart = -1;
   private int noBrokenFilesLast = -1;
   
   private int noBrokenLinksStart = -1;
   private int noBrokenLinksLast = -1;
   
   private int noFixedItems = -1;
   
   private int noBrokenLinksInStaticFiles = -1;
   private int noBrokenLinksInForms = -1;
   
   private int baseSnapshotVersion = 0;
   private int latestSnapshotVersion = 0;
   
   private int maxNumberLinksInReport = -1;
   
   private List<String> brokenStaticFilesStart;
   private List<String> brokenFormsStart;
   
   private List<String> brokenStaticFilesLast;
   private List<String> brokenFormsLast;
   
   private List<String> fixedFiles;
   private List<String> fixedForms;
   
   private Map<String, List<String>> brokenLinksByFile;
   private Map<String, List<String>> brokenFilesByForm;
   
   private Throwable cause;
   
   private static Log logger = LogFactory.getLog(LinkValidationState.class);
   
   /**
    * Default constructor
    */
   public LinkValidationState(LinkValidationReport initialReport)
   {
      this.store = initialReport.getStore();
      this.webapp = initialReport.getWebapp();
      
      processReport(initialReport, false);
   }
   
   // ------------------------------------------------------------------------------
   // Getters
   
   /**
    * @return The store this validation state represents
    */
   public String getStore()
   {
      return this.store;
   }
   
   /**
    * @return The webapp within a store this validaton state represents
    */
   public String getWebapp()
   {
      return this.webapp;
   }
   
   /**
    * @return The date the check was completed
    */
   public Date getCheckCompletedAt()
   {
      return this.checkCompletedAt;
   }
   
   /**
    * @return The error that caused the last report to fail
    */
   public Throwable getError()
   {
      return this.cause;
   }
   
   /**
    * @return The number of files checked by the initial link check
    */
   public int getInitialNumberFilesChecked()
   {
      return this.noFilesCheckedStart;
   }
   
   /**
    * @return The number of links checked by the initial link check
    */
   public int getInitialNumberLinksChecked()
   {
      return this.noLinksCheckedStart;
   }
   
   /**
    * @return The number of broken files found by the initial link check
    */
   public int getInitialNumberBrokenFiles()
   {
      return this.noBrokenFilesStart;
   }
   
   /**
    * @return The number of broken links found by the initial link check
    */
   public int getInitialNumberBrokenLinks()
   {
      return this.noBrokenLinksStart;
   }
   
   /**
    * @return The number of files checked by the latest link check
    */
   public int getNumberFilesChecked()
   {
      return this.noFilesCheckedLast;
   }
   
   /**
    * @return The number of links checked by the latest link check
    */
   public int getNumberLinksChecked()
   {
      return this.noLinksCheckedLast;
   }
   
   /**
    * @return The number of broken files found by the latest link check
    */
   public int getNumberBrokenFiles()
   {
      return this.noBrokenFilesLast;
   }
   
   /**
    * @return The number of broken links found by the latest link check
    */
   public int getNumberBrokenLinks()
   {
      return this.noBrokenLinksLast;
   }
   
   /**
    * @return The number of broken links in static files
    */
   public int getNoBrokenLinksInStaticFiles()
   {
      return noBrokenLinksInStaticFiles;
   }

   /**
    * @return The number of broken links in forms
    */
   public int getNoBrokenLinksInForms()
   {
      return noBrokenLinksInForms;
   }

   /**
    * @return The number of items fixed since the initial link check
    */
   public int getNumberFixedItems()
   {
      return this.noFixedItems;
   }
   
   /**
    * @return The snapshot version of the staging area the link check was run against 
    */
   public int getBaseSnapshotVersion()
   {
      return this.baseSnapshotVersion;
   }

   /**
    * @return The snapshot version of the staging area at the end of the link check
    */
   public int getLatestSnapshotVersion()
   {
      return this.latestSnapshotVersion;
   }
   
   /**
    * @return The maximum number of links a report can have
    */
   public int getMaxNumberLinksInReport()
   {
      return this.maxNumberLinksInReport;
   }
   
   /**
    * @return true if the maximum number of links was exceeded in the last check
    */
   public boolean hasMaxNumberLinksExceeded()
   {
      return this.maxLinksReached;
   }

   /**
    * @return A list of paths to non-generated files that contain broken links
    */
   public List<String> getStaticFilesWithBrokenLinks()
   {
      return this.brokenStaticFilesLast;
   }
   
   /**
    * @return A list of forms that have generated files containing broken links
    */
   public List<String> getFormsWithBrokenLinks()
   {
      return this.brokenFormsLast;
   }
   
   /**
    * @param form The name of a form to find broken files for
    * @return The list of broken files generated by the given form
    */
   public List<String> getBrokenFilesByForm(String form)
   {
      return this.brokenFilesByForm.get(form);
   }
   
   /**
    * @param file The path to a file with broken links
    * @return The list of broken links within the given file 
    */
   public List<String> getBrokenLinksForFile(String file)
   {
      return this.brokenLinksByFile.get(file);
   }
   
   /**
    * @return The list of files that have been fixed since the 
    *         initial link check
    */
   public List<String> getFixedFiles()
   {
      return this.fixedFiles;
   }
   
   /**
    * @return The list of forms that have been fixed since the
    *         initial link check
    */
   public List<String> getFixedForms()
   {
      return this.fixedForms;
   }
   
   // ------------------------------------------------------------------------------
   // Implementation
   
   /**
    * Determines whether the link validation check has been re-run since
    * the intial check
    * 
    * @return true if the link check has been re-run, false otherwise
    */
   public boolean hasCheckBeenReRun()
   {
      return this.checkBeenReRun;
   }
   
   /**
    * Updates the link validation state with the result from a re-run of
    * the link check
    * 
    * @param newReport The report to compare the intial report with
    */
   public void updateState(LinkValidationReport newReport)
   {
      // make sure the updated report has the same store and webapp
      // as the initial report
      if ((newReport.getStore().equals(this.store) == false) || 
          (newReport.getWebapp().equals(this.webapp) == false))
      {
         throw new IllegalStateException("Can not update a report from a different store or webapp!");
      }
      
      // process the new report
      processReport(newReport, true);
   }
   
   @Override
   public String toString()
   {
      StringBuilder buffer = new StringBuilder(super.toString());
      buffer.append(" (store=").append(this.store);
      buffer.append(" webapp=").append(this.webapp);
      buffer.append(" baseSnapshot=").append(this.baseSnapshotVersion);
      buffer.append(" latestSnapshot=").append(this.latestSnapshotVersion);
      buffer.append(" error=").append(this.cause).append(")");
      return buffer.toString();
   }
   
   // ------------------------------------------------------------------------------
   // Private Helpers

   public void processReport(LinkValidationReport report, boolean updatedReport)
   {
      this.checkBeenReRun = updatedReport;
      this.cause = report.getError();
      
      // update the check completed date
      this.checkCompletedAt = report.getCheckCompletedAt();
      
      // get the snapshot versions
      this.baseSnapshotVersion = report.getBaseSnapshotVersion();
      this.latestSnapshotVersion = report.getLatestSnapshotVersion();
      
      // get whether the max number of links was exceeded for the report
      // and the maximum number of links allowed
      this.maxLinksReached = report.hasMaxNumberLinksExceeded();
      this.maxNumberLinksInReport = report.getMaxNumberLinksInReport();
         
      if (this.cause == null)
      {
         if (updatedReport == false)
         {
            // setup initial counts
            this.noBrokenFilesStart = report.getNumberBrokenFiles();
            this.noBrokenLinksStart = report.getNumberBrokenLinks();
            this.noFilesCheckedStart = report.getNumberFilesChecked();
            this.noLinksCheckedStart = report.getNumberLinksChecked();
            this.noBrokenFilesLast = report.getNumberBrokenFiles();
            this.noBrokenLinksLast = report.getNumberBrokenLinks();
            this.noFilesCheckedLast = report.getNumberFilesChecked();
            this.noLinksCheckedLast = report.getNumberLinksChecked();
            this.noFixedItems = 0;
            
            // setup fixed lists
            this.fixedFiles = new ArrayList<String>(this.noBrokenFilesStart);
            this.fixedForms = new ArrayList<String>(this.noBrokenFilesStart);
   
            // process the broken files and determine which ones are static files
            // and which ones are generated
            processFiles(report.getFilesWithBrokenLinks(), updatedReport, report);
         }
         else
         {
            // update the relevant counts
            this.noBrokenFilesLast = report.getNumberBrokenFiles();
            this.noBrokenLinksLast = report.getNumberBrokenLinks();
            this.noFilesCheckedLast = report.getNumberFilesChecked();
            this.noLinksCheckedLast = report.getNumberLinksChecked();
            
            // process the broken files and determine which ones are static files
            // and which ones are generated
            processFiles(report.getFilesWithBrokenLinks(), updatedReport, report);
            
            // go through the list of files & forms still broken and find which ones
            // were fixed in the last re-run of the report
            for (String file : this.brokenStaticFilesStart)
            {
               if (this.brokenStaticFilesLast.contains(file) == false &&
                   this.fixedFiles.contains(file) == false)
               {
                  this.fixedFiles.add(file);
               }
            }
            
            for (String file : this.brokenFormsStart)
            {
               if (this.brokenFormsLast.contains(file) == false &&
                   this.fixedForms.contains(file) == false)
               {
                  this.fixedForms.add(file);
               }
            }
            
            // calculate the number of fixed items we have
            this.noFixedItems = this.fixedFiles.size() + this.fixedForms.size();
         }
         
         // calculate the number of broken links for static files and how
         // many are from generated files (forms)
         this.noBrokenLinksInStaticFiles = 0;
         this.noBrokenLinksInForms = 0;
         
         for (String file : this.brokenStaticFilesLast)
         {
            List<String> links = this.getBrokenLinksForFile(file);
            if (links != null)
            {
               this.noBrokenLinksInStaticFiles += links.size();
            }
         }
         
         for (String form : this.brokenFormsLast)
         {
            for (String file: this.getBrokenFilesByForm(form))
            {
               List<String> links = this.getBrokenLinksForFile(file);
               if (links != null)
               {
                  this.noBrokenLinksInForms += links.size();
               }
            }
         }
      }
   }
   
   protected void processFiles(List<String> files, boolean updatedReport, LinkValidationReport report)
   {
      AVMService avmService = Repository.getServiceRegistry(
               FacesContext.getCurrentInstance()).getAVMService();
      NodeService nodeService = Repository.getServiceRegistry(
               FacesContext.getCurrentInstance()).getNodeService();
      
      if (logger.isDebugEnabled())
      {
         if (updatedReport)
            logger.debug("Processing files from updated report: " + report);
         else
            logger.debug("Processing files from initial report: " + report);
      }
         
      // reset the 'last' lists and the maps
      this.brokenStaticFilesLast = new ArrayList<String>(this.noBrokenFilesLast);
      this.brokenFormsLast = new ArrayList<String>(this.noBrokenFilesLast);
      this.brokenFilesByForm = new HashMap<String, List<String>>(this.noBrokenFilesLast);
      this.brokenLinksByFile = new HashMap<String, List<String>>(this.noBrokenFilesLast);
      
      // iterate around the files and determine which ones are generated and static
      for (String file : files)
      {
         if (avmService.hasAspect(-1, file, WCMAppModel.ASPECT_FORM_INSTANCE_DATA))
         {
            if (avmService.hasAspect(-1, file, WCMAppModel.ASPECT_RENDITION))
            {
               if (logger.isDebugEnabled())
                  logger.debug("Processing generated file: " + file);
               
               // store the broken links in the file
               this.brokenLinksByFile.put(file, report.getBrokenLinksForFile(file));
               
               // find the XML that generated this file
               NodeRef nodeRef = AVMNodeConverter.ToNodeRef(-1, file);
               String xmlPath = (String)nodeService.getProperty(nodeRef, 
                        WCMAppModel.PROP_PRIMARY_FORM_INSTANCE_DATA);
               xmlPath = this.store + ":" + xmlPath;
               
               if (logger.isDebugEnabled())
                  logger.debug("Found source XML for '" + file + "': " + xmlPath);
               
               // store the XML as a broken form (if not already)
               if (this.brokenFormsLast.contains(xmlPath) == false)
               {
                  this.brokenFormsLast.add(xmlPath);
               }
               
               // store the list of files generated by the XML in the map
               List<String> genFiles = this.brokenFilesByForm.get(xmlPath);
               if (genFiles == null)
               {
                  genFiles = new ArrayList<String>(1);
               }
               genFiles.add(file);
               this.brokenFilesByForm.put(xmlPath, genFiles);
            }
            else
            {
               if (logger.isDebugEnabled())
                  logger.debug("Ignoring generated XML file: " + file);
            }
         }
         else
         {
            if (logger.isDebugEnabled())
               logger.debug("Processing static file: " + file);
            
            // the file does not have the form instance data aspect so it must
            // have been added manually
            this.brokenStaticFilesLast.add(file);
            this.brokenLinksByFile.put(file, report.getBrokenLinksForFile(file));
         }
      }
      
      // if this is the first run of the report setup the initial lists
      if (updatedReport == false)
      {
         this.brokenStaticFilesStart = new ArrayList<String>(this.brokenStaticFilesLast.size());
         this.brokenStaticFilesStart.addAll(this.brokenStaticFilesLast);

         this.brokenFormsStart = new ArrayList<String>(this.brokenFormsLast.size());
         this.brokenFormsStart.addAll(this.brokenFormsLast);
      }
   }
}