/*
    * Copyright (c) 2002-2014, Mairie de Paris
    * All rights reserved.
    *
    * Redistribution and use in source and binary forms, with or without
    * modification, are permitted provided that the following conditions
    * are met:
    *
    *  1. Redistributions of source code must retain the above copyright notice
   *     and the following disclaimer.
   *
   *  2. Redistributions in binary form must reproduce the above copyright notice
   *     and the following disclaimer in the documentation and/or other materials
   *     provided with the distribution.
   *
   *  3. Neither the name of 'Mairie de Paris' nor 'Lutece' nor the names of its
   *     contributors may be used to endorse or promote products derived from
   *     this software without specific prior written permission.
   *
   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
   * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
   * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
   * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
   * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
   * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
   * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
   * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
   * POSSIBILITY OF SUCH DAMAGE.
   *
   * License 1.0
   */
  package fr.paris.lutece.util.annotation;
  
  import fr.paris.lutece.portal.service.util.AppLogService;
  import fr.paris.lutece.portal.service.util.AppPathService;
  
  import org.scannotation.AnnotationDB;
  
  import java.io.File;
  import java.io.FilenameFilter;
  import java.io.IOException;
  
  import java.lang.annotation.Annotation;
  
  import java.net.MalformedURLException;
  import java.net.URL;
  
  import java.util.Date;
  import java.util.HashSet;
  import java.util.Map;
  import java.util.Set;
  
  
  /**
   * Uses {@link AnnotationDB} from scannotation.
   * @see #init()
   */
  public class ScannotationDB implements IAnnotationDB
  {
      private static final String CONSTANT_WEB_INF_LIB = "/WEB-INF/lib/";
      private static final String CONSTANT_WEB_INF_CLASS = "/WEB-INF/classes/";
  
      /**
       * Default filename filter matches <strong>ONLY</strong> lutece jars.
       */
      private static final String CONSTANT_DEFAULT_FILENAME_FILTER = "(plugin-.*\\.jar)|(module-.*\\.jar)|(lutece-core-.*\\.jar)|(library-.*\\.jar)";
      private final AnnotationDB _db;
      private String _strFileFilter;
  
      /**
       * Constructor.
       */
      public ScannotationDB(  )
      {
          _db = new AnnotationDB(  );
      }
  
      /**
       * Gets the filename filter
       * @return the filename filter
       */
      public String getFileFilter(  )
      {
          return _strFileFilter;
      }
  
      /**
       * Sets the filename filter
       * @param strFileFilter the filename filter
       */
      public void setFileFilter( String strFileFilter )
      {
          _strFileFilter = strFileFilter;
      }
  
      /**
       * Scans the WEB-INF/classes and WEB-INF/lib using {@link #getFileFilter()}
      * to filter jars.
      */
     @Override
     public void init(  )
     {
         AppLogService.info( "ScannotationDB Scanning classpath..." );
 
         if ( getFileFilter(  ) == null )
         {
             setFileFilter( CONSTANT_DEFAULT_FILENAME_FILTER );
             AppLogService.info( "Using default filename filter" );
         }
         else
         {
             AppLogService.info( "Using " + getFileFilter(  ) + " as filename filter" );
         }
 
         Date start = new Date(  );
         File libDirectory = new File( AppPathService.getWebAppPath(  ) + CONSTANT_WEB_INF_LIB );
 
         String[] allJars = libDirectory.list( new FilenameFilter(  )
                 {
                     /**
                      * {@inheritDoc}
                      */
                     @Override
                     public boolean accept( File dir, String name )
                     {
                         return name.matches( _strFileFilter );
                     }
                 } );
 
         for ( String strJar : allJars )
         {
             try
             {
                 if ( AppLogService.isDebugEnabled(  ) )
                 {
                     AppLogService.debug( "Scanning " + strJar );
                 }
 
                 _db.scanArchives( new URL( "file:///" + AppPathService.getWebAppPath(  ) + CONSTANT_WEB_INF_LIB +
                         strJar ) );
             }
             catch ( MalformedURLException e )
             {
                 AppLogService.error( e.getMessage(  ), e );
             }
             catch ( IOException e )
             {
                 AppLogService.error( e.getMessage(  ), e );
             }
         }
 
         AppLogService.info( "ScannotationDB WEB-INF/lib scanned" );
 
         try
         {
             _db.scanArchives( new URL( "file:///" + AppPathService.getWebAppPath(  ) + CONSTANT_WEB_INF_CLASS ) );
         }
         catch ( MalformedURLException e )
         {
             AppLogService.error( e.getMessage(  ), e );
         }
         catch ( IOException e )
         {
             AppLogService.error( e.getMessage(  ), e );
         }
 
         AppLogService.info( "ScannotationDB Classpath scanned in " + ( new Date(  ).getTime(  ) - start.getTime(  ) ) +
             "ms" );
     }
 
     /**
      * {@inheritDoc}
      */
     @Override
     public Set<String> getClassesName( Class<?extends Annotation> annotationType )
     {
         return getClassesName( annotationType.getName(  ) );
     }
 
     /**
      * {@inheritDoc}
      */
     @Override
     public Set<String> getClassesName( String strAnnotationType )
     {
         Map<String, Set<String>> index = _db.getAnnotationIndex(  );
 
         Set<String> setClasses = index.get( strAnnotationType );
 
         if ( setClasses == null )
         {
             setClasses = new HashSet<String>(  );
         }
 
         return setClasses;
     }
 }