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