com.jayway.maven.plugins.android.phase_prebuild.ClasspathModifierLifecycleParticipant.java Source code

Java tutorial

Introduction

Here is the source code for com.jayway.maven.plugins.android.phase_prebuild.ClasspathModifierLifecycleParticipant.java

Source

/*******************************************************************************
 * Copyright (c) 2008, 2011 Sonatype Inc. and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Sonatype Inc. - initial API and implementation
 *******************************************************************************/
package com.jayway.maven.plugins.android.phase_prebuild;

import com.jayway.maven.plugins.android.common.AndroidExtension;
import com.jayway.maven.plugins.android.common.ArtifactResolverHelper;
import com.jayway.maven.plugins.android.common.DependencyResolver;
import com.jayway.maven.plugins.android.common.UnpackedLibHelper;
import org.apache.maven.AbstractMavenLifecycleParticipant;
import org.apache.maven.MavenExecutionException;
import org.apache.maven.artifact.Artifact;
import org.apache.maven.artifact.resolver.ArtifactResolver;
import org.apache.maven.execution.MavenSession;
import org.apache.maven.model.Dependency;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.project.MavenProject;
import org.apache.maven.shared.dependency.graph.DependencyGraphBuilder;
import org.apache.maven.shared.dependency.graph.DependencyGraphBuilderException;
import org.codehaus.plexus.component.annotations.Component;
import org.codehaus.plexus.component.annotations.Requirement;
import org.codehaus.plexus.logging.Logger;

import java.io.File;
import java.io.IOException;
import java.util.Enumeration;
import java.util.List;
import java.util.Set;
import java.util.regex.Pattern;
import java.util.zip.ZipEntry;
import java.util.zip.ZipException;
import java.util.zip.ZipFile;

/**
 * Adds classes from AAR and APK dependencies to the project compile classpath.
 */
@Component(role = AbstractMavenLifecycleParticipant.class, hint = "default")
public final class ClasspathModifierLifecycleParticipant extends AbstractMavenLifecycleParticipant {
    @Requirement
    private ArtifactResolver artifactResolver;

    @Requirement(hint = "default")
    private DependencyGraphBuilder dependencyGraphBuilder;

    @Requirement
    private Logger log;

    @Override
    public void afterProjectsRead(MavenSession session) throws MavenExecutionException {
        log.debug("");
        log.debug("ClasspathModifierLifecycleParticipant#afterProjectsRead - start");
        log.debug("");

        log.debug("CurrentProject=" + session.getCurrentProject());
        final List<MavenProject> projects = session.getProjects();
        final DependencyResolver dependencyResolver = new DependencyResolver(log, dependencyGraphBuilder);
        final ArtifactResolverHelper artifactResolverHelper = new ArtifactResolverHelper(artifactResolver, log);

        for (MavenProject project : projects) {
            log.debug("");
            log.debug("project=" + project.getArtifact());

            if (!AndroidExtension.isAndroidPackaging(project.getPackaging())) {
                continue; // do not modify classpath if not an android project.
            }

            final UnpackedLibHelper helper = new UnpackedLibHelper(artifactResolverHelper, project, log);

            final Set<Artifact> artifacts;

            // If there is an extension ClassRealm loaded for this project then use that
            // as the ContextClassLoader so that Wagon extensions can be used to resolves dependencies.
            final ClassLoader projectClassLoader = (project.getClassRealm() != null) ? project.getClassRealm()
                    : Thread.currentThread().getContextClassLoader();

            final ClassLoader originalClassLoader = Thread.currentThread().getContextClassLoader();
            try {
                Thread.currentThread().setContextClassLoader(projectClassLoader);
                artifacts = dependencyResolver.getProjectDependenciesFor(project, session);
            } catch (DependencyGraphBuilderException e) {
                // Nothing to do. The resolution failure will be displayed by the standard resolution mechanism.
                continue;
            } finally {
                Thread.currentThread().setContextClassLoader(originalClassLoader);
            }

            log.debug("projects deps: : " + artifacts);
            for (Artifact artifact : artifacts) {
                final String type = artifact.getType();
                if (type.equals(AndroidExtension.AAR)) {
                    // An AAR lib contains a classes jar that needs to be added to the classpath.
                    // Create a placeholder classes.jar and add it to the compile classpath.
                    // It will replaced with the real classes.jar by GenerateSourcesMojo.
                    addClassesToClasspath(helper, project, artifact);
                    // Add jar files in 'libs' into classpath.
                    addLibsJarsToClassPath(helper, project, artifact);
                } else if (type.equals(AndroidExtension.APK)) {
                    // The only time that an APK will likely be a dependency is when this an an APK test project.
                    // So add a placeholder (we cannot resolve the actual dep pre build) to the compile classpath.
                    // The placeholder will be replaced with the real APK jar later.
                    addClassesToClasspath(helper, project, artifact);
                } else if (type.equals(AndroidExtension.APKLIB)) {
                    // Add jar files in 'libs' into classpath.
                    addLibsJarsToClassPath(helper, project, artifact);
                }
            }
        }
        log.debug("");
        log.debug("ClasspathModifierLifecycleParticipant#afterProjectsRead - finish");
    }

    /**
     * Add jar files in libs into the project classpath.
     */
    private void addLibsJarsToClassPath(UnpackedLibHelper helper, MavenProject project, Artifact artifact)
            throws MavenExecutionException {
        try {
            final File unpackLibFolder = helper.getUnpackedLibFolder(artifact);
            final File artifactFile = helper.getArtifactToFile(artifact);
            ZipFile zipFile = new ZipFile(artifactFile);
            Enumeration enumeration = zipFile.entries();
            while (enumeration.hasMoreElements()) {
                ZipEntry entry = (ZipEntry) enumeration.nextElement();
                String entryName = entry.getName();

                // Only jar files under 'libs' directory to be processed.
                if (Pattern.matches("^libs/.+\\.jar$", entryName)) {
                    final File libsJarFile = new File(unpackLibFolder, entryName);
                    log.debug("Adding jar to classpath: " + libsJarFile);

                    // In order to satisfy the LifecycleDependencyResolver on execution up to a phase that
                    // has a Mojo requiring dependency resolution I need to create a dummy classesJar here.
                    if (!libsJarFile.getParentFile().exists()) {
                        libsJarFile.getParentFile().mkdirs();
                    }
                    libsJarFile.createNewFile();

                    // Add the jar to the classpath.
                    final Dependency dependency = createSystemScopeDependency(artifact, libsJarFile,
                            libsJarFile.getName());

                    project.getModel().addDependency(dependency);
                }
            }
        } catch (MojoExecutionException e) {
            log.debug("Error extract jars");
        } catch (ZipException e) {
            log.debug("Error");
        } catch (IOException e) {
            log.debug("Error");
        }
    }

    /**
     * Add the dependent library classes to the project classpath.
     */
    private void addClassesToClasspath(UnpackedLibHelper helper, MavenProject project, Artifact artifact)
            throws MavenExecutionException {
        // Work out where the dep will be extracted and calculate the file path to the classes jar.
        // This is location where the GenerateSourcesMojo will extract the classes.
        final File classesJar = helper.getUnpackedClassesJar(artifact);
        log.debug("Adding to classpath : " + classesJar);

        // In order to satisfy the LifecycleDependencyResolver on execution up to a phase that
        // has a Mojo requiring dependency resolution I need to create a dummy classesJar here.
        classesJar.getParentFile().mkdirs();
        try {
            classesJar.createNewFile();
            log.debug("Created dummy " + classesJar.getName() + " exist=" + classesJar.exists());
        } catch (IOException e) {
            throw new MavenExecutionException("Could not add " + classesJar.getName() + " as dependency", e);
        }

        // Add the classes to the classpath
        final Dependency dependency = createSystemScopeDependency(artifact, classesJar, null);
        project.getModel().addDependency(dependency);
    }

    private Dependency createSystemScopeDependency(Artifact artifact, File location, String suffix) {
        String artifactId = artifact.getArtifactId();
        if (suffix != null) {
            artifactId = "_" + suffix;
        }
        final Dependency dependency = new Dependency();
        dependency.setGroupId(artifact.getGroupId());
        dependency.setArtifactId(artifactId);
        dependency.setVersion(artifact.getVersion());
        dependency.setScope(Artifact.SCOPE_SYSTEM);
        dependency.setSystemPath(location.getAbsolutePath());
        return dependency;
    }
}