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

   1 package org.apache.maven.archiva.consumers.core;
   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 org.apache.commons.io.FileUtils;
  23 import org.apache.maven.archiva.configuration.ManagedRepositoryConfiguration;
  24 import org.apache.maven.archiva.consumers.AbstractMonitoredConsumer;
  25 import org.apache.maven.archiva.consumers.ConsumerException;
  26 import org.apache.maven.archiva.consumers.KnownRepositoryContentConsumer;
  27
  28 import java.io.File;
  29 import java.io.IOException;
  30 import java.util.ArrayList;
  31 import java.util.Date;
  32 import java.util.HashMap;
  33 import java.util.Iterator;
  34 import java.util.List;
  35 import java.util.Map;
  36
  37 /**
  38  * AutoRenameConsumer
  39  *
  40  * @version $Id$
  41  * @plexus.component role="org.apache.maven.archiva.consumers.KnownRepositoryContentConsumer"
  42  * role-hint="auto-rename"
  43  * instantiation-strategy="per-lookup"
  44  */
  45 public class AutoRenameConsumer
  46     extends AbstractMonitoredConsumer
  47     implements KnownRepositoryContentConsumer
  48 {
  49     /**
  50      * @plexus.configuration default-value="auto-rename"
  51      */
  52     private String id;
  53
  54     /**
  55      * @plexus.configuration default-value="Automatically rename common artifact mistakes."
  56      */
  57     private String description;
  58
  59     private static final String RENAME_FAILURE = "rename_failure";
  60
  61     private File repositoryDir;
  62
  63     private List<String> includes = new ArrayList<String>();
  64
  65     private Map<String, String> extensionRenameMap = new HashMap<String, String>();
  66
  67     public AutoRenameConsumer()
  68     {
  69         includes.add( "**/*.distribution-tgz" );
  70         includes.add( "**/*.distribution-zip" );
  71         includes.add( "**/*.plugin" );
  72
  73         extensionRenameMap.put( ".distribution-tgz", ".tar.gz" );
  74         extensionRenameMap.put( ".distribution-zip", ".zip" );
  75         extensionRenameMap.put( ".plugin", ".jar" );
  76     }
  77
  78     public String getId()
  79     {
  80         return this.id;
  81     }
  82
  83     public String getDescription()
  84     {
  85         return this.description;
  86     }
  87
  88     public boolean isPermanent()
  89     {
  90         return false;
  91     }
  92
  93     public void beginScan( ManagedRepositoryConfiguration repository, Date whenGathered )
  94         throws ConsumerException
  95     {
  96         this.repositoryDir = new File( repository.getLocation() );
  97     }
  98
  99     public void completeScan()
 100     {
 101         /* do nothing */
 102     }
 103
 104     public List<String> getExcludes()
 105     {
 106         return null;
 107     }
 108
 109     public List<String> getIncludes()
 110     {
 111         return includes;
 112     }
 113
 114     public void processFile( String path )
 115         throws ConsumerException
 116     {
 117         File file = new File( this.repositoryDir, path );
 118         if ( file.exists() )
 119         {
 120             Iterator<String> itExtensions = this.extensionRenameMap.keySet().iterator();
 121             while ( itExtensions.hasNext() )
 122             {
 123                 String extension = (String) itExtensions.next();
 124                 if ( path.endsWith( extension ) )
 125                 {
 126                     String fixedExtension = (String) this.extensionRenameMap.get( extension );
 127                     String correctedPath = path.substring( 0, path.length() - extension.length() ) + fixedExtension;
 128                     File to = new File( this.repositoryDir, correctedPath );
 129                     try
 130                     {
 131                         // Rename the file.
 132                         FileUtils.moveFile( file, to );
 133                     }
 134                     catch ( IOException e )
 135                     {
 136                         triggerConsumerWarning( RENAME_FAILURE, "Unable to rename " + path + " to " + correctedPath +
 137                             ": " + e.getMessage() );
 138                     }
 139                 }
 140             }
 141
 142             triggerConsumerInfo( "(Auto) Removing File: " + file.getAbsolutePath() );
 143             file.delete();
 144         }
 145     }
 146 }