M2TestsuiteEnvironmentBuilder.java

/**
 * Copyright (c) 2008-2011 Sonatype, Inc.
 * All rights reserved. Includes the third-party code listed at http://www.sonatype.com/products/nexus/attributions.
 *
 * This program is free software: you can redistribute it and/or modify it only under the terms of the GNU Affero General
 * Public License Version 3 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied
 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License Version 3
 * for more details.
 *
 * You should have received a copy of the GNU Affero General Public License Version 3 along with this program.  If not, see
 * http://www.gnu.org/licenses.
 *
 * Sonatype Nexus (TM) Open Source Version is available from Sonatype, Inc. Sonatype and Sonatype Nexus are trademarks of
 * Sonatype, Inc. Apache Maven is a trademark of the Apache Foundation. M2Eclipse is a trademark of the Eclipse Foundation.
 * All other trademarks are the property of their respective owners.
 */
package org.sonatype.nexus.proxy;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.codehaus.plexus.PlexusContainer;
import org.codehaus.plexus.component.repository.exception.ComponentLookupException;
import org.codehaus.plexus.util.xml.Xpp3Dom;
import org.sonatype.configuration.ConfigurationException;
import org.sonatype.jettytestsuite.ServletServer;
import org.sonatype.jettytestsuite.WebappContext;
import org.sonatype.nexus.configuration.model.CLocalStorage;
import org.sonatype.nexus.configuration.model.CRemoteStorage;
import org.sonatype.nexus.configuration.model.CRepository;
import org.sonatype.nexus.configuration.model.DefaultCRepository;
import org.sonatype.nexus.proxy.maven.ChecksumPolicy;
import org.sonatype.nexus.proxy.maven.RepositoryPolicy;
import org.sonatype.nexus.proxy.maven.maven2.M2GroupRepository;
import org.sonatype.nexus.proxy.maven.maven2.M2GroupRepositoryConfiguration;
import org.sonatype.nexus.proxy.maven.maven2.M2Repository;
import org.sonatype.nexus.proxy.maven.maven2.M2RepositoryConfiguration;
import org.sonatype.nexus.proxy.repository.GroupRepository;
import org.sonatype.nexus.proxy.repository.Repository;

/**
 * The Class JettyTestsuiteEnvironment.
 * 
 * @author cstamas
 */
public class M2TestsuiteEnvironmentBuilder
    extends AbstractJettyEnvironmentBuilder
{

    public M2TestsuiteEnvironmentBuilder( ServletServer servletServer )
    {
        super( servletServer );
    }

    @Override
    public void buildEnvironment( AbstractProxyTestEnvironment env )
        throws ConfigurationException,
            IOException,
            ComponentLookupException
    {
        PlexusContainer container = env.getPlexusContainer();

        List<String> reposes = new ArrayList<String>();
        for ( WebappContext remoteRepo : getServletServer().getWebappContexts() )
        {
            M2Repository repo = (M2Repository) container.lookup( Repository.class, "maven2" );

            CRepository repoConf = new DefaultCRepository();

            repoConf.setProviderRole( Repository.class.getName() );
            repoConf.setProviderHint( "maven2" );
            repoConf.setId( remoteRepo.getName() );
            repoConf.setName( remoteRepo.getName() );

            repoConf.setLocalStorage( new CLocalStorage() );
            repoConf.getLocalStorage().setProvider( "file" );
            repoConf.getLocalStorage().setUrl(
                env
                    .getApplicationConfiguration().getWorkingDirectory( "proxy/store/" + remoteRepo.getName() ).toURI()
                    .toURL().toString() );

            Xpp3Dom ex = new Xpp3Dom( "externalConfiguration" );
            repoConf.setExternalConfiguration( ex );
            M2RepositoryConfiguration exConf = new M2RepositoryConfiguration( ex );
            if ( remoteRepo.getName().endsWith( "-snapshot" ) )
            {
                exConf.setRepositoryPolicy( RepositoryPolicy.SNAPSHOT );
            }
            else
            {
                exConf.setRepositoryPolicy( RepositoryPolicy.RELEASE );
            }
            exConf.setChecksumPolicy( ChecksumPolicy.STRICT_IF_EXISTS );

            repoConf.setRemoteStorage( new CRemoteStorage() );
            repoConf.getRemoteStorage().setProvider( "apacheHttpClient3x" );
            repoConf.getRemoteStorage().setUrl( getServletServer().getUrl( remoteRepo.getName() ) );

            repo.configure( repoConf );

            // repo.setCacheManager( env.getCacheManager() );
            reposes.add( repo.getId() );
            
            env.getApplicationConfiguration().getConfigurationModel().addRepository( repoConf );

            env.getRepositoryRegistry().addRepository( repo );
        }

        // ading one hosted only
        M2Repository repo = (M2Repository) container.lookup( Repository.class, "maven2" );

        CRepository repoConf = new DefaultCRepository();

        repoConf.setProviderRole( Repository.class.getName() );
        repoConf.setProviderHint( "maven2" );
        repoConf.setId( "inhouse" );

        repoConf.setLocalStorage( new CLocalStorage() );
        repoConf.getLocalStorage().setProvider( "file" );
        repoConf.getLocalStorage().setUrl(
            env.getApplicationConfiguration().getWorkingDirectory( "proxy/store/inhouse" ).toURI().toURL().toString() );

        Xpp3Dom exRepo = new Xpp3Dom( "externalConfiguration" );
        repoConf.setExternalConfiguration( exRepo );
        M2RepositoryConfiguration exRepoConf = new M2RepositoryConfiguration( exRepo );
        exRepoConf.setRepositoryPolicy( RepositoryPolicy.RELEASE );
        exRepoConf.setChecksumPolicy( ChecksumPolicy.STRICT_IF_EXISTS );

        repo.configure( repoConf );

        reposes.add( repo.getId() );

        env.getApplicationConfiguration().getConfigurationModel().addRepository( repoConf );

        env.getRepositoryRegistry().addRepository( repo );

        // add a hosted snapshot repo
        M2Repository repoSnapshot = (M2Repository) container.lookup( Repository.class, "maven2" );

        CRepository repoSnapshotConf = new DefaultCRepository();

        repoSnapshotConf.setProviderRole( Repository.class.getName() );
        repoSnapshotConf.setProviderHint( "maven2" );
        repoSnapshotConf.setId( "inhouse-snapshot" );

        repoSnapshotConf.setLocalStorage( new CLocalStorage() );
        repoSnapshotConf.getLocalStorage().setProvider( "file" );
        repoSnapshotConf.getLocalStorage().setUrl(
            env
                .getApplicationConfiguration().getWorkingDirectory( "proxy/store/inhouse-snapshot" ).toURI().toURL()
                .toString() );

        Xpp3Dom exSnapRepo = new Xpp3Dom( "externalConfiguration" );
        repoSnapshotConf.setExternalConfiguration( exSnapRepo );
        M2RepositoryConfiguration exSnapRepoConf = new M2RepositoryConfiguration( exSnapRepo );
        exSnapRepoConf.setRepositoryPolicy( RepositoryPolicy.SNAPSHOT );
        exSnapRepoConf.setChecksumPolicy( ChecksumPolicy.STRICT_IF_EXISTS );

        repoSnapshot.configure( repoSnapshotConf );

        reposes.add( repoSnapshot.getId() );

        env.getApplicationConfiguration().getConfigurationModel().addRepository( repoSnapshotConf );

        env.getRepositoryRegistry().addRepository( repoSnapshot );

        // add a group
        M2GroupRepository group = (M2GroupRepository) container.lookup( GroupRepository.class, "maven2" );

        CRepository repoGroupConf = new DefaultCRepository();

        repoGroupConf.setProviderRole( GroupRepository.class.getName() );
        repoGroupConf.setProviderHint( "maven2" );
        repoGroupConf.setId( "test" );

        repoGroupConf.setLocalStorage( new CLocalStorage() );
        repoGroupConf.getLocalStorage().setProvider( "file" );
        repoGroupConf.getLocalStorage().setUrl(
            env.getApplicationConfiguration().getWorkingDirectory( "proxy/store/test" ).toURI().toURL().toString() );

        Xpp3Dom exGroupRepo = new Xpp3Dom( "externalConfiguration" );
        repoGroupConf.setExternalConfiguration( exGroupRepo );
        M2GroupRepositoryConfiguration exGroupRepoConf = new M2GroupRepositoryConfiguration( exGroupRepo );
        exGroupRepoConf.setMemberRepositoryIds( reposes );
        exGroupRepoConf.setMergeMetadata( true );

        group.configure( repoGroupConf );

        env.getApplicationConfiguration().getConfigurationModel().addRepository( repoGroupConf );

        env.getRepositoryRegistry().addRepository( group );
    }
}