package org.apache.archiva.consumers.lucene;

 /*

  * Licensed to the Apache Software Foundation (ASF) under one

  * or more contributor license agreements.  See the NOTICE file

  * distributed with this work for additional information

  * regarding copyright ownership.  The ASF licenses this file

  * to you under the Apache License, Version 2.0 (the

  * "License"); you may not use this file except in compliance

  * with the License.  You may obtain a copy of the License at

  *

  *  http://www.apache.org/licenses/LICENSE-2.0

  *

  * Unless required by applicable law or agreed to in writing,

  * software distributed under the License is distributed on an

  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

  * KIND, either express or implied.  See the License for the

  * specific language governing permissions and limitations

  * under the License.

  */

 import java.io.File;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.Collections;

 import java.util.Date;

 import java.util.List;

 import org.apache.maven.archiva.configuration.ArchivaConfiguration;

 import org.apache.maven.archiva.configuration.ConfigurationNames;

 import org.apache.maven.archiva.configuration.FileTypes;

 import org.apache.maven.archiva.configuration.ManagedRepositoryConfiguration;

 import org.apache.maven.archiva.consumers.AbstractMonitoredConsumer;

 import org.apache.maven.archiva.consumers.ConsumerException;

 import org.apache.maven.archiva.consumers.KnownRepositoryContentConsumer;

 import org.apache.maven.archiva.repository.content.ManagedDefaultRepositoryContent;

 import org.apache.maven.archiva.scheduled.ArchivaTaskScheduler;

 import org.apache.maven.archiva.scheduled.tasks.ArtifactIndexingTask;

 import org.apache.maven.archiva.scheduled.tasks.TaskCreator;

 import org.codehaus.plexus.personality.plexus.lifecycle.phase.Initializable;

 import org.codehaus.plexus.personality.plexus.lifecycle.phase.InitializationException;

 import org.codehaus.plexus.registry.Registry;

 import org.codehaus.plexus.registry.RegistryListener;

 import org.codehaus.plexus.taskqueue.TaskQueueException;

 import org.slf4j.Logger;

 import org.slf4j.LoggerFactory;

 import org.sonatype.nexus.index.context.IndexingContext;

 import org.sonatype.nexus.index.context.UnsupportedExistingLuceneIndexException;

 /**

  * Consumer for indexing the repository to provide search and IDE integration features.

  */

 public class NexusIndexerConsumer

     extends AbstractMonitoredConsumer

     implements KnownRepositoryContentConsumer, RegistryListener, Initializable

 {

     private static final Logger log = LoggerFactory.getLogger( NexusIndexerConsumer.class );

     private ArchivaConfiguration configuration;

     private FileTypes filetypes;

     private ManagedDefaultRepositoryContent repositoryContent;

     private File managedRepository;

     private ArchivaTaskScheduler scheduler;

     private IndexingContext context;

     private List<String> includes = new ArrayList<String>();

     private ManagedRepositoryConfiguration repository;

     public NexusIndexerConsumer( ArchivaTaskScheduler scheduler, ArchivaConfiguration configuration, FileTypes filetypes )

     {

         this.configuration = configuration;

         this.filetypes = filetypes;

         this.scheduler = scheduler;

     }

     public String getDescription()

     {

         return "Indexes the repository to provide search and IDE integration features";

     }

     public String getId()

     {

         return "index-content";

     }

     public boolean isPermanent()

     {

         return false;

     }

     public void beginScan( ManagedRepositoryConfiguration repository, Date whenGathered )

         throws ConsumerException

     {

         managedRepository = new File( repository.getLocation() );

         repositoryContent = new ManagedDefaultRepositoryContent();

         repositoryContent.setRepository( repository );

         try

         {

             log.info( "Creating indexing context for repo : " + repository.getId() );

             context = TaskCreator.createContext( repository );

         }

         catch ( IOException e )

         {

             throw new ConsumerException( e.getMessage(), e );

         }

         catch ( UnsupportedExistingLuceneIndexException e )

         {

             throw new ConsumerException( e.getMessage(), e );

         }

     }

     public void beginScan( ManagedRepositoryConfiguration repository, Date whenGathered, boolean executeOnEntireRepo )

         throws ConsumerException

     {

         if( executeOnEntireRepo )

         {

             beginScan( repository, whenGathered );

         }

         else

         {

             this.repository = repository;

             managedRepository = new File( repository.getLocation() );       

         }

     }

     public void processFile( String path )

         throws ConsumerException

     {

         File artifactFile = new File( managedRepository, path );

         ArtifactIndexingTask task =

             TaskCreator.createIndexingTask( repositoryContent.getRepository(), artifactFile,

                                             ArtifactIndexingTask.Action.ADD, context );

         try

         {

             log.debug( "Queueing indexing task + '" + task + "' to add or update the artifact in the index." );

             scheduler.queueIndexingTask( task );

         }

         catch ( TaskQueueException e )

         {

             throw new ConsumerException( e.getMessage(), e );

         }

     }

     public void processFile( String path, boolean executeOnEntireRepo )

         throws Exception

     {

         if( executeOnEntireRepo )

         {

             processFile( path );

         }

         else

         {

             File artifactFile = new File( managedRepository, path );

             // specify in indexing task that this is not a repo scan request!

             ArtifactIndexingTask task =

                 new ArtifactIndexingTask( repository, artifactFile, ArtifactIndexingTask.Action.ADD, context, false );

             try

             {

                 log.debug( "Queueing indexing task + '" + task + "' to add or update the artifact in the index." );

                 scheduler.queueIndexingTask( task );

             }

             catch ( TaskQueueException e )

             {

                 throw new ConsumerException( e.getMessage(), e );

             }

         }

     }

     public void completeScan()

     {

         ArtifactIndexingTask task =

             TaskCreator.createIndexingTask( repositoryContent.getRepository(), null,

                                             ArtifactIndexingTask.Action.FINISH, context );

         try

         {

             log.debug( "Queueing indexing task + '" + task + "' to finish indexing." );

             scheduler.queueIndexingTask( task );

         }

         catch ( TaskQueueException e )

         {

             log.error( "Error queueing task: " + task + ": " + e.getMessage(), e );

         }

         context = null;

     }

     public void completeScan( boolean executeOnEntireRepo )

     {

         if( executeOnEntireRepo )

         {

             completeScan();

         }

         // else, do nothing as the context will be closed when indexing task is executed if not a repo scan request!

     }

     public List<String> getExcludes()

     {

         return Collections.emptyList();

     }

     public void afterConfigurationChange( Registry registry, String propertyName, Object propertyValue )

     {

         if ( ConfigurationNames.isRepositoryScanning( propertyName ) )

         {

             initIncludes();

         }

     }

     public void beforeConfigurationChange( Registry registry, String propertyName, Object propertyValue )

     {

         /* do nothing */

     }

     private void initIncludes()

     {

         includes.clear();

         includes.addAll( filetypes.getFileTypePatterns( FileTypes.INDEXABLE_CONTENT ) );

         includes.addAll( filetypes.getFileTypePatterns( FileTypes.ARTIFACTS ) );

     }

     public void initialize()

         throws InitializationException

     {

         configuration.addChangeListener( this );

         initIncludes();

     }

     public List<String> getIncludes()

     {

         return includes;

     }

 }