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

   1 package org.apache.maven.repository.reporting;
   2
   3 /*
   4  * Copyright 2005-2006 The Apache Software Foundation.
   5  *
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18
  19 import org.apache.lucene.index.Term;
  20 import org.apache.lucene.search.TermQuery;
  21 import org.apache.maven.artifact.Artifact;
  22 import org.apache.maven.artifact.repository.ArtifactRepository;
  23 import org.apache.maven.model.Model;
  24 import org.apache.maven.repository.digest.Digester;
  25 import org.apache.maven.repository.digest.DigesterException;
  26 import org.apache.maven.repository.indexing.RepositoryArtifactIndex;
  27 import org.apache.maven.repository.indexing.RepositoryArtifactIndexFactory;
  28 import org.apache.maven.repository.indexing.RepositoryIndexSearchException;
  29 import org.apache.maven.repository.indexing.lucene.LuceneQuery;
  30 import org.apache.maven.repository.indexing.record.StandardArtifactIndexRecord;
  31 import org.apache.maven.repository.indexing.record.StandardIndexRecordFields;
  32
  33 import java.io.File;
  34 import java.util.Iterator;
  35 import java.util.List;
  36
  37 /**
  38  * Validates an artifact file for duplicates within the same groupId based from what's available in a repository index.
  39  *
  40  * @author Edwin Punzalan
  41  * @plexus.component role="org.apache.maven.repository.reporting.ArtifactReportProcessor" role-hint="duplicate"
  42  */
  43 public class DuplicateArtifactFileReportProcessor
  44     implements ArtifactReportProcessor
  45 {
  46     /**
  47      * @plexus.requirement
  48      */
  49     private Digester digester;
  50
  51     /**
  52      * @plexus.requirement
  53      */
  54     private RepositoryArtifactIndexFactory indexFactory;
  55
  56     /**
  57      * @plexus.configuration
  58      */
  59     private String indexDirectory;
  60
  61     public void processArtifact( Model model, Artifact artifact, ArtifactReporter reporter,
  62                                  ArtifactRepository repository )
  63         throws ReportProcessorException
  64     {
  65         if ( artifact.getFile() != null )
  66         {
  67             RepositoryArtifactIndex index = indexFactory.createStandardIndex( new File( indexDirectory ) );
  68
  69             String checksum;
  70             try
  71             {
  72                 checksum = digester.createChecksum( artifact.getFile(), Digester.MD5 );
  73             }
  74             catch ( DigesterException e )
  75             {
  76                 throw new ReportProcessorException( "Failed to generate checksum", e );
  77             }
  78
  79             try
  80             {
  81                 List results = index.search( new LuceneQuery(
  82                     new TermQuery( new Term( StandardIndexRecordFields.MD5, checksum.toLowerCase() ) ) ) );
  83
  84                 if ( results.isEmpty() )
  85                 {
  86                     reporter.addSuccess( artifact );
  87                 }
  88                 else
  89                 {
  90                     boolean hasDuplicates = false;
  91                     for ( Iterator i = results.iterator(); i.hasNext(); )
  92                     {
  93                         StandardArtifactIndexRecord result = (StandardArtifactIndexRecord) i.next();
  94
  95                         //make sure it is not the same artifact
  96                         if ( !result.getFilename().equals( repository.pathOf( artifact ) ) )
  97                         {
  98                             //report only duplicates from the same groupId
  99                             String groupId = artifact.getGroupId();
 100                             if ( groupId.equals( result.getGroupId() ) )
 101                             {
 102                                 hasDuplicates = true;
 103                                 reporter.addFailure( artifact, "Found duplicate for " + artifact.getId() );
 104                             }
 105                         }
 106                     }
 107
 108                     if ( !hasDuplicates )
 109                     {
 110                         reporter.addSuccess( artifact );
 111                     }
 112                 }
 113             }
 114             catch ( RepositoryIndexSearchException e )
 115             {
 116                 throw new ReportProcessorException( "Failed to search in index", e );
 117             }
 118         }
 119         else
 120         {
 121             reporter.addWarning( artifact, "Artifact file is null" );
 122         }
 123     }
 124 }