package org.apache.archiva.cli;
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
import com.sampullara.cli.Args;
import com.sampullara.cli.Argument;
import org.apache.archiva.admin.model.beans.ManagedRepository;
import org.apache.archiva.common.plexusbridge.PlexusSisuBridge;
import org.apache.archiva.common.plexusbridge.PlexusSisuBridgeException;
import org.apache.archiva.converter.RepositoryConversionException;
import org.apache.archiva.converter.legacy.LegacyRepositoryConverter;
import org.apache.archiva.repository.scanner.RepositoryScanStatistics;
import org.apache.archiva.repository.scanner.RepositoryScanner;
import org.apache.archiva.repository.scanner.RepositoryScannerException;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.archiva.consumers.ConsumerException;
import org.apache.archiva.consumers.InvalidRepositoryContentConsumer;
import org.apache.archiva.consumers.KnownRepositoryContentConsumer;
import org.apache.archiva.consumers.RepositoryContentConsumer;
import org.apache.maven.artifact.manager.WagonManager;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;
/**
* ArchivaCli
*
* @version $Id: ArchivaCli.java 1171453 2011-09-16 08:54:07Z olamy $
* @todo add back reading of archiva.xml from a given location
*/
public class ArchivaCli
{
// ----------------------------------------------------------------------------
// Properties controlling Repository conversion
// ----------------------------------------------------------------------------
public static final String SOURCE_REPO_PATH = "sourceRepositoryPath";
public static final String TARGET_REPO_PATH = "targetRepositoryPath";
public static final String BLACKLISTED_PATTERNS = "blacklistPatterns";
public static final String POM_PROPERTIES = "/META-INF/maven/org.apache.archiva/archiva-cli/pom.properties";
private static String getVersion()
throws IOException
{
InputStream pomStream = ArchivaCli.class.getResourceAsStream( POM_PROPERTIES );
if ( pomStream == null )
{
throw new IOException( "Failed to load " + POM_PROPERTIES );
}
try
{
Properties properties = new Properties();
properties.load( pomStream );
return properties.getProperty( "version" );
}
finally
{
IOUtils.closeQuietly( pomStream );
}
}
private ClassPathXmlApplicationContext applicationContext;
public ArchivaCli()
{
applicationContext =
new ClassPathXmlApplicationContext( new String[]{ "classpath*:/META-INF/spring-context.xml" } );
}
public static void main( String[] args )
throws Exception
{
Commands command = new Commands();
try
{
Args.parse( command, args );
}
catch ( IllegalArgumentException e )
{
System.err.println( e.getMessage() );
Args.usage( command );
return;
}
new ArchivaCli().execute( command );
}
private void execute( Commands command )
throws Exception
{
if ( command.help )
{
Args.usage( command );
}
else if ( command.version )
{
System.out.print( "Version: " + getVersion() );
}
else if ( command.convert )
{
doConversion( command.properties );
}
else if ( command.scan )
{
if ( command.repository == null )
{
System.err.println( "The repository must be specified." );
Args.usage( command );
return;
}
doScan( command.repository, command.consumers.split( "," ) );
}
else if ( command.listConsumers )
{
dumpAvailableConsumers();
}
else
{
Args.usage( command );
}
}
private void doScan( String path, String[] consumers )
throws ConsumerException, MalformedURLException, PlexusSisuBridgeException
{
// hack around poorly configurable project builder by pointing all repositories back at this location to be self
// contained
PlexusSisuBridge plexusSisuBridge = applicationContext.getBean( PlexusSisuBridge.class );
WagonManager wagonManager = plexusSisuBridge.lookup( WagonManager.class );
wagonManager.addMirror( "internal", "*", new File( path ).toURL().toExternalForm() );
ManagedRepository repo = new ManagedRepository();
repo.setId( "cliRepo" );
repo.setName( "Archiva CLI Provided Repo" );
repo.setLocation( path );
List<KnownRepositoryContentConsumer> knownConsumerList = new ArrayList<KnownRepositoryContentConsumer>();
knownConsumerList.addAll( getConsumerList( consumers ) );
List<InvalidRepositoryContentConsumer> invalidConsumerList = Collections.emptyList();
List<String> ignoredContent = new ArrayList<String>();
ignoredContent.addAll( Arrays.asList( RepositoryScanner.IGNORABLE_CONTENT ) );
RepositoryScanner scanner = (RepositoryScanner) lookup( RepositoryScanner.class );
try
{
RepositoryScanStatistics stats = scanner.scan( repo, knownConsumerList, invalidConsumerList, ignoredContent,
RepositoryScanner.FRESH_SCAN );
System.out.println( "\n" + stats.toDump( repo ) );
}
catch ( RepositoryScannerException e )
{
e.printStackTrace( System.err );
}
}
private Object lookup( Class<?> clazz )
throws PlexusSisuBridgeException
{
PlexusSisuBridge plexusSisuBridge = applicationContext.getBean( PlexusSisuBridge.class );
return plexusSisuBridge.lookup( clazz );
}
private List<KnownRepositoryContentConsumer> getConsumerList( String[] consumers )
throws ConsumerException, PlexusSisuBridgeException
{
List<KnownRepositoryContentConsumer> consumerList = new ArrayList<KnownRepositoryContentConsumer>();
Map<String, KnownRepositoryContentConsumer> availableConsumers = getConsumers();
for ( String specifiedConsumer : consumers )
{
if ( !availableConsumers.containsKey( specifiedConsumer ) )
{
System.err.println( "Specified consumer [" + specifiedConsumer + "] not found." );
dumpAvailableConsumers();
System.exit( 1 );
}
consumerList.add( availableConsumers.get( specifiedConsumer ) );
}
return consumerList;
}
private void dumpAvailableConsumers()
throws PlexusSisuBridgeException
{
Map<String, KnownRepositoryContentConsumer> availableConsumers = getConsumers();
System.out.println( ".\\ Available Consumer List \\.______________________________" );
for ( Map.Entry<String, KnownRepositoryContentConsumer> entry : availableConsumers.entrySet() )
{
String consumerHint = (String) entry.getKey();
RepositoryContentConsumer consumer = (RepositoryContentConsumer) entry.getValue();
System.out.println(
" " + consumerHint + ": " + consumer.getDescription() + " (" + consumer.getClass().getName() + ")" );
}
}
@SuppressWarnings( "unchecked" )
private Map<String, KnownRepositoryContentConsumer> getConsumers()
throws PlexusSisuBridgeException
{
Map<String, KnownRepositoryContentConsumer> beans =
applicationContext.getBeansOfType( KnownRepositoryContentConsumer.class );
// we use a naming conventions knownRepositoryContentConsumer#hint
// with plexus we used only hint so remove before#
Map<String, KnownRepositoryContentConsumer> smallNames =
new HashMap<String, KnownRepositoryContentConsumer>( beans.size() );
for ( Map.Entry<String, KnownRepositoryContentConsumer> entry : beans.entrySet() )
{
smallNames.put( StringUtils.substringAfterLast( entry.getKey(), "#" ), entry.getValue() );
}
return smallNames;
}
private void doConversion( String properties )
throws FileNotFoundException, IOException, RepositoryConversionException, PlexusSisuBridgeException
{
LegacyRepositoryConverter legacyRepositoryConverter =
(LegacyRepositoryConverter) lookup( LegacyRepositoryConverter.class );
Properties p = new Properties();
FileInputStream fis = new FileInputStream( properties );
try
{
p.load( fis );
}
finally
{
IOUtils.closeQuietly( fis );
}
File oldRepositoryPath = new File( p.getProperty( SOURCE_REPO_PATH ) );
File newRepositoryPath = new File( p.getProperty( TARGET_REPO_PATH ) );
System.out.println( "Converting " + oldRepositoryPath + " to " + newRepositoryPath );
List<String> fileExclusionPatterns = null;
String s = p.getProperty( BLACKLISTED_PATTERNS );
if ( s != null )
{
fileExclusionPatterns = Arrays.asList( StringUtils.split( s, "," ) );
}
legacyRepositoryConverter.convertLegacyRepository( oldRepositoryPath, newRepositoryPath,
fileExclusionPatterns );
}
private static class Commands
{
@Argument( description = "Display help information", value = "help", alias = "h" )
private boolean help;
@Argument( description = "Display version information", value = "version", alias = "v" )
private boolean version;
@Argument( description = "List available consumers", value = "listconsumers", alias = "l" )
private boolean listConsumers;
@Argument( description = "The consumers to use (comma delimited)", value = "consumers", alias = "u" )
private String consumers = "count-artifacts";
@Argument( description = "Scan the specified repository", value = "scan", alias = "s" )
private boolean scan;
@Argument(
description = "Convert a legacy Maven 1.x repository to a Maven 2.x repository using a properties file to describe the conversion",
value = "convert", alias = "c" )
private boolean convert;
@Argument( description = "The properties file for the conversion", value = "properties" )
private String properties = "conversion.properties";
@Argument( description = "The repository to scan", value = "repository" )
private String repository;
}
}