GemsInstaller.java

/**
 *
 */
package de.saumya.mojo.ruby.gems;

import de.saumya.mojo.ruby.script.JRubyVersion;
import de.saumya.mojo.ruby.script.Script;
import de.saumya.mojo.ruby.script.ScriptException;
import de.saumya.mojo.ruby.script.ScriptFactory;
import org.apache.maven.artifact.Artifact;
import org.apache.maven.artifact.repository.ArtifactRepository;
import org.apache.maven.plugin.descriptor.PluginDescriptor;
import org.apache.maven.project.MavenProject;
import org.codehaus.plexus.util.FileUtils;

import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.util.Collection;
import java.util.List;


public class GemsInstaller {

    private static final String OPENSSL_VERSION = "0.8.2";
    private static final String OPENSSL = "jruby-openssl";

    private static final FileFilter FILTER = new FileFilter() {

        public boolean accept(File f) {
            return f.getName().endsWith(".gemspec");
        }
    };

    public final GemsConfig     config;

    public final ScriptFactory  factory;

    public final GemManager    manager;

    public GemsInstaller(final GemsConfig config, final ScriptFactory factory,
            final GemManager manager) {
        this.config = config;
        this.factory = factory;
        this.manager = manager;
    }

    public void installPom(final MavenProject pom) throws IOException,
            ScriptException, GemException {
        installGems(pom, null);
    }

    public void installPom(final MavenProject pom,
            final ArtifactRepository localRepository) throws IOException,
            ScriptException, GemException {
        installPom(pom, localRepository, null);
    }

    public void installPom(final MavenProject pom,
            final ArtifactRepository localRepository, String scope) throws IOException,
            ScriptException, GemException {
        installGems(pom, localRepository, scope);
    }

    public MavenProject installOpenSSLGem(final Object repositorySystemSession,
            final ArtifactRepository localRepository, List<ArtifactRepository> remotes) throws GemException,
            IOException, ScriptException {
        return installGem(OPENSSL, OPENSSL_VERSION, repositorySystemSession, localRepository, remotes);
    }

    public MavenProject installGem(final String name, final String version,
            final Object repositorySystemSession,
            final ArtifactRepository localRepository, List<ArtifactRepository> remoteRepositories) throws GemException,
            IOException, ScriptException {
        final Artifact artifact;
        if (version == null) {
            artifact = this.manager.createGemArtifactWithLatestVersion(name,
                                                                       localRepository,
                                                                       remoteRepositories);
        }
        else {
            artifact = this.manager.createGemArtifact(name, version);
        }
        final MavenProject pom = this.manager.buildPom(artifact,
                                                        repositorySystemSession,
                                                        localRepository,
                                                        remoteRepositories);
        installPom(pom);
        return pom;
    }

    public void installGems(final MavenProject pom,
            final ArtifactRepository localRepository)
                    throws IOException, ScriptException, GemException
    {
        installGems(pom, localRepository, null);

    }

    public void installGems(final MavenProject pom,
            final ArtifactRepository localRepository, String scope )
        throws IOException, ScriptException, GemException {
        installGems(pom, (Collection<Artifact>)null, localRepository, scope);
    }

    public void installGems(final MavenProject pom, PluginDescriptor plugin,
            final ArtifactRepository localRepository) throws IOException,
            ScriptException, GemException {
        installGems(pom, plugin.getArtifacts(), localRepository, (String) null);
    }

    public void installGems(final MavenProject pom, final Collection<Artifact> artifacts,
            final ArtifactRepository localRepository, String scope) throws IOException,
            ScriptException, GemException {
        installGems(pom, artifacts, localRepository, pom.getRemoteArtifactRepositories(), scope);
    }
    public void installGems(final MavenProject pom, final Collection<Artifact> artifacts,
            final ArtifactRepository localRepository, List<ArtifactRepository> remoteRepos)
                    throws IOException, ScriptException, GemException {
        installGems( pom, artifacts, localRepository, remoteRepos, null );
    }

    public void installGems(final MavenProject pom, final Collection<Artifact> artifacts,
                final ArtifactRepository localRepository, List<ArtifactRepository> remoteRepos,
                String scope ) throws IOException,
                ScriptException, GemException {
        // start without script object. 
        // script object will be created when first un-installed gem is found
        Script script = null;
        if (pom != null) {
            boolean hasAlreadyOpenSSL = false;
            for (final Artifact artifact : pom.getArtifacts()) {
                // assume pom.getBasedir() != null indicates the project pom
                if ( ( "compile".equals(artifact.getScope()) ||
                       "runtime".equals(artifact.getScope()) ||
                       pom.getBasedir() != null ) &&
                      ( scope == null || scope.equals(artifact.getScope()) ) ) {
                    if (!artifact.getFile().exists()) {
                        this.manager.resolve(artifact,
                                             localRepository,
                                             remoteRepos);

                    }
                    script = maybeAddArtifact(script, artifact);
                    hasAlreadyOpenSSL = hasAlreadyOpenSSL
                            || artifact.getArtifactId().equals(OPENSSL);
                }
            }
            if (artifacts != null) {
                for (final Artifact artifact : artifacts) {
                    if (!artifact.getFile().exists()) {
                        this.manager.resolve(artifact,
                                             localRepository,
                                             remoteRepos);

                    }
                    script = maybeAddArtifact(script, artifact);
                    hasAlreadyOpenSSL = hasAlreadyOpenSSL
                            || artifact.getArtifactId().equals(OPENSSL);
                }
            }
            if ( pom.getArtifact().getFile() != null
                 // to filter out target/classes
                 && pom.getArtifact().getFile().isFile()
                 // have only gem files
                 && pom.getArtifact().getFile().getName().endsWith(".gem") ) {
                script = maybeAddArtifact(script, pom.getArtifact());
            }
        }

        if (script != null) {
            script.addArg("--bindir", this.config.getBinDirectory());
            if(this.config.getBinDirectory() != null && !this.config.getBinDirectory().exists()){
                this.config.getBinDirectory().mkdirs();
            }
            script.execute();

            if (this.config.getGemHome() != null){
                // workaround for unpatched: https://github.com/rubygems/rubygems/commit/21cccd55b823848c5e941093a615b0fdd6cd8bc7
                for(File spec : new File(this.config.getGemHome(), "specifications").listFiles(FILTER)){
                    String content = FileUtils.fileRead(spec);
                    FileUtils.fileWrite(spec.getAbsolutePath(), content.replaceFirst(" 00:00:00.000000000Z", ""));
                }
            }

            this.factory.newScript( "require 'jruby/commands'; JRuby::Commands.generate_dir_info '" +
                    this.config.getGemHome() +
                    "' if JRuby::Commands.respond_to? :generate_dir_info" ).execute();
        }
    }

    private boolean exists(Artifact artifact) {
        // check if the specifications exists
        String basename = artifact.getArtifactId() + "-"
                + artifact.getVersion().replaceFirst("-SNAPSHOT$", "");
        String universalJavaBasename = basename + "-universal-java.gemspec";
        String javaBasename = basename + "-java.gemspec";
        String rubyBasename = basename + ".gemspec";

        for (File dir : this.config.getGemsDirectory()) {
            File specs  = new File( dir.getParentFile(), "specifications" );
            if (new File(specs, rubyBasename).exists()
                    || new File(specs, javaBasename).exists()
                    || new File(specs, universalJavaBasename).exists()) {
                return true;
            }
        }
        return false;
    }

    private Script maybeAddArtifact(Script script, final Artifact artifact)
            throws IOException, GemException {
        if (artifact.getType().contains("gem")) {
            if (!exists(artifact)) {
                if (script == null) {

                    script = this.factory.newScriptFromJRubyJar("gem")
                            .addArg("install")
                            .addArg("--ignore-dependencies")
                            .addArg(booleanArg(this.config.isUserInstall(), "user-install"))
                            .addArg(booleanArg(this.config.isVerbose(), "verbose"));

                    final JRubyVersion version = this.factory.getVersion();
                    if (version == null || version.isVersionLowerThan(9, 2, 10)) {
                        script.addArg(booleanArg(this.config.isAddRdoc(), "rdoc"))
                                .addArg(booleanArg(this.config.isAddRI(), "ri"));
                    } else {
                        script.addArg(booleanArg(this.config.isAddRdoc(), "document"));
                    }
                }
                if (artifact.getFile() != null)
                {
                    script.addArg(artifact.getFile());
                }
            }
        }
        return script;
    }

    private String booleanArg(final boolean flag, final String name) {
        return "--" + (flag ? "" : "no-") + name;
    }
}