source.dussan.org Git - archiva.git/blob

   1 package org.apache.maven.archiva.repository.scanner;
   2
   3 /*
   4  * Licensed to the Apache Software Foundation (ASF) under one
   5  * or more contributor license agreements.  See the NOTICE file
   6  * distributed with this work for additional information
   7  * regarding copyright ownership.  The ASF licenses this file
   8  * to you under the Apache License, Version 2.0 (the
   9  * "License"); you may not use this file except in compliance
  10  * with the License.  You may obtain a copy of the License at
  11  *
  12  *  http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing,
  15  * software distributed under the License is distributed on an
  16  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  17  * KIND, either express or implied.  See the License for the
  18  * specific language governing permissions and limitations
  19  * under the License.
  20  */
  21
  22 import java.io.File;
  23 import java.util.ArrayList;
  24 import java.util.List;
  25
  26 import org.apache.archiva.rss.processor.RssFeedProcessor;
  27 import org.apache.commons.collections.CollectionUtils;
  28 import org.apache.maven.archiva.configuration.FileTypes;
  29 import org.apache.maven.archiva.configuration.ManagedRepositoryConfiguration;
  30 import org.apache.maven.archiva.consumers.InvalidRepositoryContentConsumer;
  31 import org.apache.maven.archiva.consumers.KnownRepositoryContentConsumer;
  32 import org.apache.maven.archiva.consumers.RepositoryContentConsumer;
  33 import org.apache.maven.archiva.model.ArchivaArtifact;
  34 import org.apache.maven.archiva.model.ArtifactReference;
  35 import org.apache.maven.archiva.repository.ManagedRepositoryContent;
  36 import org.apache.maven.archiva.repository.RepositoryContentFactory;
  37 import org.apache.maven.archiva.repository.RepositoryException;
  38 import org.apache.maven.archiva.repository.RepositoryNotFoundException;
  39 import org.apache.maven.archiva.repository.layout.LayoutException;
  40 import org.codehaus.plexus.util.DirectoryWalker;
  41 import org.slf4j.Logger;
  42 import org.slf4j.LoggerFactory;
  43
  44 /**
  45  * DefaultRepositoryScanner
  46  *
  47  * @author <a href="mailto:joakime@apache.org">Joakim Erdfelt</a>
  48  * @version $Id$
  49  * @plexus.component role="org.apache.maven.archiva.repository.scanner.RepositoryScanner"
  50  */
  51 public class DefaultRepositoryScanner
  52     implements RepositoryScanner
  53 {
  54     private Logger log = LoggerFactory.getLogger( DefaultRepositoryScanner.class );
  55
  56     /**
  57      * @plexus.requirement
  58      */
  59     private FileTypes filetypes;
  60
  61     /**
  62      * @plexus.requirement
  63      */
  64     private RepositoryContentConsumers consumerUtil;
  65
  66     /**
  67      * @plexus.requirement
  68      */
  69     private RepositoryContentFactory repositoryFactory;
  70
  71     /**
  72      * @plexus.requirement role-hint="new-artifacts"
  73      */
  74     private RssFeedProcessor rssFeedProcessor;
  75
  76     public RepositoryScanStatistics scan( ManagedRepositoryConfiguration repository, long changesSince )
  77         throws RepositoryException
  78     {
  79         List<KnownRepositoryContentConsumer> knownContentConsumers = consumerUtil.getSelectedKnownConsumers();
  80         List<InvalidRepositoryContentConsumer> invalidContentConsumers = consumerUtil.getSelectedInvalidConsumers();
  81         List<String> ignoredPatterns = filetypes.getFileTypePatterns( FileTypes.IGNORED );
  82
  83         return scan( repository, knownContentConsumers, invalidContentConsumers, ignoredPatterns, changesSince );
  84     }
  85
  86     public RepositoryScanStatistics scan( ManagedRepositoryConfiguration repository,
  87                                           List<KnownRepositoryContentConsumer> knownContentConsumers,
  88                                           List<InvalidRepositoryContentConsumer> invalidContentConsumers,
  89                                           List<String> ignoredContentPatterns, long changesSince )
  90         throws RepositoryException
  91     {
  92         if ( repository == null )
  93         {
  94             throw new IllegalArgumentException( "Unable to operate on a null repository." );
  95         }
  96
  97         File repositoryBase = new File( repository.getLocation() );
  98
  99         if ( !repositoryBase.exists() )
 100         {
 101             throw new UnsupportedOperationException( "Unable to scan a repository, directory "
 102                 + repositoryBase.getAbsolutePath() + " does not exist." );
 103         }
 104
 105         if ( !repositoryBase.isDirectory() )
 106         {
 107             throw new UnsupportedOperationException( "Unable to scan a repository, path "
 108                 + repositoryBase.getAbsolutePath() + " is not a directory." );
 109         }
 110
 111         // Setup Includes / Excludes.
 112
 113         List<String> allExcludes = new ArrayList<String>();
 114         List<String> allIncludes = new ArrayList<String>();
 115
 116         if ( CollectionUtils.isNotEmpty( ignoredContentPatterns ) )
 117         {
 118             allExcludes.addAll( ignoredContentPatterns );
 119         }
 120
 121         // Scan All Content. (intentional)
 122         allIncludes.add( "**/*" );
 123
 124         // Setup Directory Walker
 125         DirectoryWalker dirWalker = new DirectoryWalker();
 126
 127         dirWalker.setBaseDir( repositoryBase );
 128
 129         dirWalker.setIncludes( allIncludes );
 130         dirWalker.setExcludes( allExcludes );
 131
 132         // Setup the Scan Instance
 133         RepositoryScannerInstance scannerInstance = new RepositoryScannerInstance( repository, knownContentConsumers,
 134                                                                                    invalidContentConsumers, changesSince );
 135
 136         dirWalker.addDirectoryWalkListener( scannerInstance );
 137
 138         // Execute scan.
 139         dirWalker.scan();
 140
 141         RepositoryScanStatistics stats = scannerInstance.getStatistics();
 142
 143         stats.setKnownConsumers( gatherIds( knownContentConsumers ) );
 144         stats.setInvalidConsumers( gatherIds( invalidContentConsumers ) );
 145
 146         // generate RSS feeds
 147         List<ArchivaArtifact> newArtifacts = getNewArtifacts( scannerInstance.getNewFiles(), repository.getId() );
 148         rssFeedProcessor.process( newArtifacts );
 149
 150         return stats;
 151     }
 152
 153     private List<String> gatherIds( List<? extends RepositoryContentConsumer> consumers )
 154     {
 155         List<String> ids = new ArrayList<String>();
 156         for ( RepositoryContentConsumer consumer : consumers )
 157         {
 158             ids.add( consumer.getId() );
 159         }
 160         return ids;
 161     }
 162
 163     private List<ArchivaArtifact> getNewArtifacts( List<File> files, String repoId )
 164     {
 165         List<ArchivaArtifact> newArtifacts = new ArrayList<ArchivaArtifact>();
 166
 167         // TODO: filter the file types of artifacts that will be included in the rss feeds
 168         try
 169         {
 170             ManagedRepositoryContent repository = repositoryFactory.getManagedRepositoryContent( repoId );
 171             for( File file : files )
 172             {
 173                 try
 174                 {
 175                     ArtifactReference ref = repository.toArtifactReference( file.getAbsolutePath() );
 176                     ArchivaArtifact artifact = new ArchivaArtifact( ref.getGroupId(),ref.getArtifactId(), ref.getVersion(),
 177                                                                    ref.getClassifier(), ref.getType() );
 178                     artifact.getModel().setRepositoryId( repoId );
 179                     newArtifacts.add( artifact );
 180                 }
 181                 catch ( LayoutException le )
 182                 {
 183
 184                 }
 185             }
 186         }
 187         catch ( RepositoryNotFoundException re )
 188         {
 189             log.error( re.getMessage() );
 190         }
 191         catch ( RepositoryException e )
 192         {
 193             log.error( e.getMessage() );
 194         }
 195
 196         return newArtifacts;
 197     }
 198 }