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

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