com.simpligility.maven.plugins.android.phase01generatesources.ResourceClassGenerator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of android-maven-plugin Show documentation
Maven Plugin for Android Development
There is a newer version: 4.6.0
package com.simpligility.maven.plugins.android.phase01generatesources;

import com.android.builder.core.VariantConfiguration;
import com.android.builder.internal.SymbolLoader;
import com.android.builder.internal.SymbolWriter;
import com.android.utils.ILogger;
import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.Multimap;
import org.apache.maven.artifact.Artifact;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.logging.Log;

import java.io.File;
import java.io.IOException;
import java.util.Collection;
import java.util.Set;

/**
 * Generates R classes containing appropriate resource values for dependent libraries.
 *
 * @author William Ferguson 
 */
final class ResourceClassGenerator
{
    private final GenerateSourcesMojo mojo;
    private final File targetDirectory;
    private final File genDirectory;
    private final Log log;
    private final ILogger androidUtilsLog;

    ResourceClassGenerator( GenerateSourcesMojo mojo, File targetDirectory,
                                   File genDirectory )
    {
        this.mojo = mojo;
        this.targetDirectory = targetDirectory;
        this.genDirectory = genDirectory;
        this.log = mojo.getLog();
        this.androidUtilsLog = new MavenILogger( log );
    }

    public void generateLibraryRs( Set libraries, String libraryType ) throws MojoExecutionException
    {
        // list of all the symbol loaders per package names.
        final Multimap libMap = ArrayListMultimap.create();

        for ( final Artifact lib : libraries )
        {
            final File unpackedLibDirectory = mojo.getUnpackedLibFolder( lib );
            final File rFile = new File( unpackedLibDirectory, "R.txt" );

            if ( rFile.isFile() )
            {
                final File libManifestFile = new File( unpackedLibDirectory, "AndroidManifest.xml" );
                final String packageName = VariantConfiguration.getManifestPackage( libManifestFile );
                log.debug( "Reading R for " + packageName  + " at " + rFile );

                // store these symbols by associating them with the package name.
                final SymbolLoader libSymbols = loadSymbols( rFile );
                libMap.put( packageName, libSymbols );
            }
        }

        if ( libMap.isEmpty() )
        {
            return;
        }

        // load the full resources values from the R.txt calculated for the project.
        final File projectR = new File( targetDirectory, "R.txt" );
        final SymbolLoader fullSymbolValues = loadSymbols( projectR );

        // now loop on all the package name, merge all the symbols to write, and write them
        for ( final String packageName : libMap.keySet() )
        {
            log.info( "Generating R file for " + libraryType + " : " + packageName );
            final Collection symbols = libMap.get( packageName );

            final SymbolWriter writer = new SymbolWriter(
                    genDirectory.getAbsolutePath(), packageName, fullSymbolValues );
            for ( SymbolLoader symbolLoader : symbols )
            {
                writer.addSymbolsToWrite( symbolLoader );
            }
            writeSymbols( writer, packageName );
        }
    }

    private SymbolLoader loadSymbols( File file ) throws MojoExecutionException
    {
        final SymbolLoader libSymbols = new SymbolLoader( file, androidUtilsLog );
        try
        {
            libSymbols.load();
        }
        catch ( IOException e )
        {
            throw new MojoExecutionException( "Could not load " + file, e );
        }
        return libSymbols;
    }

    private void writeSymbols( SymbolWriter writer, String packageName ) throws MojoExecutionException
    {
        try
        {
            writer.write();
        }
        catch ( IOException e )
        {
            throw new MojoExecutionException( "Could not write R for " + packageName, e );
        }
    }

}