org_scala_tools_maven.ScalaToolsServerInitMojo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of maven-scala-plugin Show documentation
The maven-scala-plugin is used for compiling/testing/running/documenting scala code in maven.
There is a newer version: 2.15.2
package org_scala_tools_maven;

import java.io.File;
import java.net.URL;
import java.net.URLConnection;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.apache.maven.project.MavenProject;
import org.codehaus.plexus.util.IOUtil;
import org.codehaus.plexus.util.StringUtils;
import org.yaml.snakeyaml.Yaml;
import org_scala_tools_maven_executions.JavaMainCaller;
import org_scala_tools_maven_executions.JavaMainCallerByFork;
import org_scala_tools_maven_executions.MainHelper;

/**
 * @goal sts-init
 * @requiresDependencyResolution test
 */
public class ScalaToolsServerInitMojo extends ScalaMojoSupport {
    /**
     * If you want to use an other version of scala-tools-server than the default one.
     *
     * @parameter expression="${maven.scala.stsVersion}"
     */
    protected String stsVersion = "0.1-SNAPSHOT";

    /**
     * The directory in which to place compilation output
     *
     * @parameter expression="${project.build.outputDirectory}"
     */
    protected File outputDir;

    /**
     * A list of inclusion filters for the compiler.
     * ex :
     *      *    <includes>
     *      <include>SomeFile.scala</include>
     *    </includes>
     * 
     *
     * @parameter
     */
    private Set includes = new HashSet();

    /**
     * A list of exclusion filters for the compiler.
     * ex :
     *      *    <excludes>
     *      <exclude>SomeBadFile.scala</exclude>
     *    </excludes>
     * 
     *
     * @parameter
     */
    private Set excludes = new HashSet();

    /**
     * The directory which contains scala/java source files
     *
     * @parameter expression="${project.build.sourceDirectory}/../scala"
     */
    protected File sourceDir;

    /**
     * The directory in which to place test compilation output
     *
     * @parameter expression="${project.build.testOutputDirectory}
     */
    protected File testOutputDir;

    /**
     * The directory in which to find test scala source code
     *
     * @parameter expression="${project.build.testSourceDirectory}/../scala"
     */
    protected File testSourceDir;

    @Override
    protected void doExecute() throws Exception {
        System.out.println(sendRequestAdd());
        System.out.println(sendRequestCompile());
    }

    private String toYaml(MavenProject project) throws Exception {
        HashMap dataCompile = new HashMap();
        /*
        name : sample
        sourceDirs :
          - "/home/dwayne/work/oss/scala-tools/scala-tools-server/src/main/scala"
        includes :
          - "*.scala"
        excludes :
        targetDir : "/home/dwayne/work/oss/scala-tools/scala-tools-server/target/classes"
        classpath :
          - "/home/dwayne/.m2/repository/org/scala-lang/scala-library/2.7.5/scala-library-2.7.5.jar"
          - "/home/dwayne/.m2/repository/org/scala-lang/scala-compiler/2.7.5/scala-compiler-2.7.5.jar"
          - "/home/dwayne/.m2/repository/org/jboss/netty/netty/3.1.0.GA/netty-3.1.0.GA.jar"
          - "/home/dwayne/.m2/repository/SnakeYAML/SnakeYAML/1.3/SnakeYAML-1.3.jar"
        args :
          - "-deprecation"
        */
        dataCompile.put("name", project.getArtifactId()+"-"+project.getVersion()+"/main");
        dataCompile.put("sourcesDirs", getSourceDirectories());
        if (includes != null) {
            dataCompile.put("includes", new ArrayList(includes));
        }
        if (excludes != null) {
            dataCompile.put("excludes", new ArrayList(excludes));
        }
        dataCompile.put("targetDir", outputDir.getCanonicalPath());
        dataCompile.put("classpath", project.getCompileClasspathElements());
        if (args != null) {
            dataCompile.put("args", args);
        }
        dataCompile.put("dependency-link-path", localRepo.pathOf(project.getArtifact()));

        HashMap dataTest = new HashMap();
        dataTest.put("name", project.getArtifactId()+"-"+project.getVersion() +"/test");
        dataTest.put("sourcesDirs", project.getTestCompileSourceRoots());
        if (includes != null) {
            dataTest.put("includes", new ArrayList(includes));
        }
        if (excludes != null) {
            dataTest.put("excludes", new ArrayList(excludes));
        }
        dataTest.put("targetDir", testOutputDir.getCanonicalPath());
        dataTest.put("classpath", project.getTestClasspathElements());
        if (args != null) {
            dataTest.put("args", args);
        }

        Yaml yaml = new Yaml();
        return yaml.dump(dataCompile) + "/n---/n" + yaml.dump(dataTest);
    }

    @SuppressWarnings("unchecked")
    protected List getSourceDirectories() throws Exception {
        List sources = project.getCompileSourceRoots();
        //Quick fix in case the user has not added the "add-source" goal.
        String scalaSourceDir = sourceDir.getCanonicalPath();
        if(!sources.contains(scalaSourceDir)) {
            sources.add(scalaSourceDir);
        }
        return sources;
    }

    protected String sendRequestAdd() throws Exception {
        String yamlDef = toYaml(project);
        String back = "";
        try {
            back = sendRequest("add", yamlDef);
        } catch (java.net.ConnectException exc) {
            startNewServer();
            back = sendRequest("add", yamlDef);
        }
        return back;
    }

    protected String sendRequestCompile() throws Exception {
        return sendRequest("compile", null);
    }

    protected String sendRequestStop() throws Exception {
        return sendRequest("stop", null);
    }

    protected String sendRequest(String action, String data) throws Exception {
        URL url = new URL("http://127.0.0.1:27616/" + action);
        URLConnection cnx = url.openConnection();
        cnx.setDoOutput(StringUtils.isNotEmpty(data));
        cnx.setDoInput(true);
        if (StringUtils.isNotEmpty(data)) {
            IOUtil.copy(data, cnx.getOutputStream());
            IOUtil.close(cnx.getOutputStream());
        }
        String back = IOUtil.toString(cnx.getInputStream());
        IOUtil.close(cnx.getInputStream());
        return back;
    }

    private void startNewServer() throws Exception {
        getLog().info("start scala-tools-server...");
        Set classpath = new HashSet();
        addToClasspath(SCALA_GROUPID, "scala-compiler", scalaVersion, classpath);
        addToClasspath("org.scala-tools", "scala-tools-server", stsVersion, classpath);
        JavaMainCaller jcmd = new JavaMainCallerByFork(this, "org.scala_tools.server.HttpServer", MainHelper.toMultiPath(classpath.toArray(new String[classpath.size()])), null, null, forceUseArgFile);
        jcmd.spawn(displayCmd);
        boolean started = false;
        while (!started) {
            try {
                System.out.print(".");
                Thread.sleep(1000);
                sendRequest("ping", null);
                started = true;
                System.out.println("\n started");
            } catch (java.net.ConnectException exc) {
                started = false; //useless but more readable
            }
        }
    }
}