jaroslav@109: package org.apidesign.bck2brwsr.mojo;
jaroslav@109: 
jaroslav@109: import org.apache.maven.plugin.AbstractMojo;
jaroslav@109: 
jaroslav@109: import java.io.File;
jaroslav@109: import java.io.FileWriter;
jaroslav@109: import java.lang.reflect.Method;
jaroslav@109: import java.net.MalformedURLException;
jaroslav@109: import java.net.URL;
jaroslav@109: import java.net.URLClassLoader;
jaroslav@109: import java.util.ArrayList;
jaroslav@109: import java.util.Collection;
jaroslav@109: import java.util.List;
jaroslav@109: import org.apache.maven.artifact.Artifact;
jaroslav@109: import org.apache.maven.model.Dependency;
jaroslav@109: import org.apache.maven.plugin.MojoExecutionException;
jaroslav@109: import org.apache.maven.plugins.annotations.Component;
jaroslav@109: import org.apache.maven.plugins.annotations.LifecyclePhase;
jaroslav@109: import org.apache.maven.plugins.annotations.Mojo;
jaroslav@109: import org.apache.maven.plugins.annotations.Parameter;
jaroslav@109: import org.apache.maven.project.MavenProject;
jaroslav@109: 
jaroslav@109: /** Compiles classes into JavaScript. */
jaroslav@109: @Mojo(name="j2js", defaultPhase=LifecyclePhase.PROCESS_CLASSES)
jaroslav@109: public class Bck2BrswrMojo extends AbstractMojo {
jaroslav@109:     public Bck2BrswrMojo() {
jaroslav@109:     }
jaroslav@109:     /** Root of the class files */
jaroslav@109:     @Parameter(defaultValue="${project.build.directory}/classes")
jaroslav@109:     private File classes;
jaroslav@109:     /** File to generate. Defaults bootjava.js in the first non-empty 
jaroslav@109:      package under the classes directory */
jaroslav@109:     @Parameter
jaroslav@109:     private File javascript;
jaroslav@109:     
jaroslav@109:     @Parameter(defaultValue="${project}")
jaroslav@109:     private MavenProject prj;
jaroslav@109:     
jaroslav@109:     
jaroslav@109: 
jaroslav@109:     @Override
jaroslav@109:     public void execute() throws MojoExecutionException {
jaroslav@109:         if (!classes.isDirectory()) {
jaroslav@109:             throw new MojoExecutionException("Can't find " + classes);
jaroslav@109:         }
jaroslav@109: 
jaroslav@109:         if (javascript == null) {
jaroslav@109:             javascript = new File(findNonEmptyFolder(classes), "bootjava.js");
jaroslav@109:         }
jaroslav@109: 
jaroslav@109:         List<String> arr = new ArrayList<String>();
jaroslav@109:         collectAllClasses("", classes, arr);
jaroslav@109:         
jaroslav@109:         
jaroslav@109: 
jaroslav@109:         try {
jaroslav@109:             URLClassLoader url = buildClassLoader(classes, prj.getDependencyArtifacts());
jaroslav@109:             
jaroslav@109:             Class<?> c = Class.forName("org.apidesign.vm4brwsr.GenJS");
jaroslav@109:             Method m = c.getDeclaredMethod("compile", ClassLoader.class, Appendable.class, List.class);
jaroslav@109:             m.setAccessible(true);
jaroslav@109:             FileWriter w = new FileWriter(javascript);
jaroslav@109:             m.invoke(null, url, w, arr);
jaroslav@109:             w.close();
jaroslav@109:         } catch (Exception ex) {
jaroslav@109:             throw new MojoExecutionException("Can't compile", ex);
jaroslav@109:         }
jaroslav@109:     }
jaroslav@109: 
jaroslav@109:     private static File findNonEmptyFolder(File dir) throws MojoExecutionException {
jaroslav@109:         if (!dir.isDirectory()) {
jaroslav@109:             throw new MojoExecutionException("Not a directory " + dir);
jaroslav@109:         }
jaroslav@109:         File[] arr = dir.listFiles();
jaroslav@109:         if (arr.length == 1 && arr[0].isDirectory()) {
jaroslav@109:             return findNonEmptyFolder(arr[0]);
jaroslav@109:         }
jaroslav@109:         return dir;
jaroslav@109:     }
jaroslav@109: 
jaroslav@109:     private static void collectAllClasses(String prefix, File toCheck, List<String> arr) {
jaroslav@109:         File[] files = toCheck.listFiles();
jaroslav@109:         if (files != null) {
jaroslav@109:             for (File f : files) {
jaroslav@109:                 collectAllClasses(prefix + f.getName() + "/", f, arr);
jaroslav@109:             }
jaroslav@109:         } else if (toCheck.getName().endsWith(".class")) {
jaroslav@109:             arr.add(prefix.substring(0, prefix.length() - 7));
jaroslav@109:         }
jaroslav@109:     }
jaroslav@109: 
jaroslav@109:     private static URLClassLoader buildClassLoader(File root, Collection<Artifact> deps) throws MalformedURLException {
jaroslav@109:         List<URL> arr = new ArrayList<URL>();
jaroslav@109:         arr.add(root.toURI().toURL());
jaroslav@109:         for (Artifact a : deps) {
jaroslav@109:             arr.add(a.getFile().toURI().toURL());
jaroslav@109:         }
jaroslav@109:         return new URLClassLoader(arr.toArray(new URL[0]), Bck2BrswrMojo.class.getClassLoader());
jaroslav@109:     }
jaroslav@109: }