/**

  * Back 2 Browser Bytecode Translator

  * Copyright (C) 2012-2015 Jaroslav Tulach <jaroslav.tulach@apidesign.org>

  *

  * This program is free software: you can redistribute it and/or modify

  * it under the terms of the GNU General Public License as published by

  * the Free Software Foundation, version 2 of the License.

  *

  * This program is distributed in the hope that it will be useful,

  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  * GNU General Public License for more details.

  *

  * You should have received a copy of the GNU General Public License

  * along with this program. Look for COPYING file in the top folder.

  * If not, see http://opensource.org/licenses/GPL-2.0.

  */

 package org.apidesign.bck2brwsr.mojo;

 import java.io.File;

 import java.io.FileNotFoundException;

 import java.io.FileOutputStream;

 import java.io.IOException;

 import java.io.OutputStreamWriter;

 import java.io.Writer;

 import java.net.MalformedURLException;

 import java.net.URL;

 import java.net.URLClassLoader;

 import java.util.ArrayList;

 import java.util.Collection;

 import java.util.List;

 import java.util.jar.Attributes;

 import java.util.jar.JarEntry;

 import java.util.jar.JarFile;

 import java.util.jar.JarOutputStream;

 import java.util.jar.Manifest;

 import org.apache.maven.artifact.Artifact;

 import org.apache.maven.plugin.AbstractMojo;

 import org.apache.maven.plugin.MojoExecutionException;

 import org.apache.maven.plugin.MojoFailureException;

 import org.apache.maven.plugins.annotations.Component;

 import org.apache.maven.plugins.annotations.LifecyclePhase;

 import org.apache.maven.plugins.annotations.Mojo;

 import org.apache.maven.plugins.annotations.Parameter;

 import org.apache.maven.plugins.annotations.ResolutionScope;

 import org.apache.maven.project.MavenProject;

 import org.apache.maven.project.MavenProjectHelper;

 import org.apidesign.bck2brwsr.aot.Bck2BrwsrJars;

 import org.apidesign.vm4brwsr.Bck2Brwsr;

 import org.apidesign.vm4brwsr.ObfuscationLevel;

 /**

  *

  * @author Jaroslav Tulach

  * @since 0.12

  */

 @Mojo(name = "library",

     requiresDependencyResolution = ResolutionScope.COMPILE_PLUS_RUNTIME,

     defaultPhase = LifecyclePhase.PACKAGE

 )

 public class AOTLibrary extends AbstractMojo {

     @Parameter(defaultValue = "${project}")

     private MavenProject prj;

     @Component

     private MavenProjectHelper projectHelper;

     @Parameter(defaultValue = "${project.build.directory}/${project.build.finalName}.jar")

     private File mainJar;    

     @Parameter(defaultValue = "${project.build.finalName}-min.js")

     private String minified;

     @Parameter(defaultValue = "${project.build.finalName}-debug.js")

     private String debug;

     @Parameter(defaultValue = "${project.build.directory}/${project.build.finalName}-bck2brwsr.jar")

     private File aotJar;

     @Parameter

     private String[] exports;

     @Parameter

     private String[] aotDeps;

     @Parameter(defaultValue = "true")

     private boolean ignoreBootClassPath;

     /**

      * The obfuscation level for the generated minified JavaScript file.

      * @since 0.5

      */

     @Parameter(defaultValue = "FULL")

     private ObfuscationLevel obfuscation;

     @Override

     public void execute() throws MojoExecutionException, MojoFailureException {

         URLClassLoader loader;

         try {

             loader = buildClassLoader(mainJar, prj.getArtifacts());

         } catch (MalformedURLException ex) {

             throw new MojoFailureException("Can't initialize classloader");

         }

         try {

             Manifest m = new Manifest();

             if (!"false".equals(minified)) {

                 Attributes attr = new Attributes();

                 attr.putValue("Bck2BrwsrArtifactId", prj.getArtifactId());

                 attr.putValue("Bck2BrwsrGroupId", prj.getGroupId());

                 attr.putValue("Bck2BrwsrVersion", prj.getVersion());

                 attr.putValue("Bck2BrwsrMinified", "true");

                 bundleName(attr, mainJar);

                 m.getEntries().put(minified, attr);

             }

             if (!"false".equals(debug)) {

                 Attributes attr = new Attributes();

                 attr.putValue("Bck2BrwsrArtifactId", prj.getArtifactId());

                 attr.putValue("Bck2BrwsrGroupId", prj.getGroupId());

                 attr.putValue("Bck2BrwsrVersion", prj.getVersion());

                 attr.putValue("Bck2BrwsrDebug", "true");

                 bundleName(attr, mainJar);

                 m.getEntries().put(debug, attr);

             }

             if (aotDeps != null) {

                 for (Artifact a : prj.getArtifacts()) {

                     if (!matches(aotDeps, a)) {

                         continue;

                     }

                     {

                         Attributes attr = new Attributes();

                         attr.putValue("Bck2BrwsrArtifactId", a.getArtifactId());

                         attr.putValue("Bck2BrwsrGroupId", a.getGroupId());

                         attr.putValue("Bck2BrwsrVersion", a.getVersion());

                         attr.putValue("Bck2BrwsrDebug", "true");

                         bundleName(attr, a.getFile());

                         m.getEntries().put(artifactName(a, true), attr);

                     }

                     {

                         Attributes attr = new Attributes();

                         attr.putValue("Bck2BrwsrArtifactId", a.getArtifactId());

                         attr.putValue("Bck2BrwsrGroupId", a.getGroupId());

                         attr.putValue("Bck2BrwsrVersion", a.getVersion());

                         attr.putValue("Bck2BrwsrMinified", "true");

                         bundleName(attr, a.getFile());

                         m.getEntries().put(artifactName(a, false), attr);

                     }

                 }

             }

             boolean notified = false;

             JarOutputStream os = null;

             if (!"false".equals(debug)) {

                 if (os == null) {

                     os = aotJar(m);

                 }

                 os.putNextEntry(new JarEntry(debug));

                 getLog().info("Generating bck2brwsr for " + mainJar);

                 notified = true;

                 Writer w = new OutputStreamWriter(os, "UTF-8");

                 configureMain(loader).

                     obfuscation(ObfuscationLevel.NONE).

                     generate(w);

                 w.flush();

                 os.closeEntry();

             }

             if (!"false".equals(minified)) {

                 if (os == null) {

                     os = aotJar(m);

                 }

                 os.putNextEntry(new JarEntry(minified));

                 if (!notified) {

                     getLog().info("Generating bck2brwsr for " + mainJar);

                 }

                 Writer w = new OutputStreamWriter(os, "UTF-8");

                 configureMain(loader).

                     obfuscation(obfuscation).

                     generate(w);

                 w.flush();

                 os.closeEntry();

             }

             if (aotDeps != null) {

                 for (Artifact a : prj.getArtifacts()) {

                     if (!matches(aotDeps, a)) {

                         continue;

                     }

                     getLog().info("Generating bck2brwsr for " + a.getFile());

                     Bck2Brwsr c = Bck2BrwsrJars.configureFrom(null, a.getFile(), loader, ignoreBootClassPath);

                     if (exports != null) {

                         for (String e : exports) {

                             c = c.addExported(e.replace('.', '/'));

                         }

                     }

                     {

                         if (os == null) {

                             os = aotJar(m);

                         }

                         os.putNextEntry(new JarEntry(artifactName(a, true)));

                         Writer w = new OutputStreamWriter(os, "UTF-8");

                         c.

                             obfuscation(ObfuscationLevel.NONE).

                             generate(w);

                         w.flush();

                         os.closeEntry();

                     }

                     {

                         os.putNextEntry(new JarEntry(artifactName(a, false)));

                         Writer w = new OutputStreamWriter(os, "UTF-8");

                         c.

                             obfuscation(ObfuscationLevel.FULL).

                             generate(w);

                         w.flush();

                         os.closeEntry();

                     }                    

                 }

             }

             if (os != null) {

                 os.close();

             }

             projectHelper.attachArtifact(prj, "jar", "bck2brwsr", aotJar);

         } catch (IOException ex) {

             throw new MojoFailureException("Cannot generate script for " + mainJar, ex);

         }

     }

     private JarOutputStream aotJar(Manifest m) throws IOException, FileNotFoundException {

         this.aotJar.getParentFile().mkdirs();

         FileOutputStream fos = new FileOutputStream(this.aotJar);

         JarOutputStream os = new JarOutputStream(fos, m);

         return os;

     }

     private Bck2Brwsr configureMain(URLClassLoader loader) throws IOException {

         Bck2Brwsr c = Bck2BrwsrJars.configureFrom(null, mainJar, loader, ignoreBootClassPath);

         if (exports != null) {

             for (String e : exports) {

                 c = c.addExported(e.replace('.', '/'));

             }

         }

         return c;

     }

     private static String artifactName(Artifact a, boolean debug) {

         return a.getGroupId() + "-" + a.getArtifactId() + (debug ? "-debug.js" : "-min.js");

     }

     private static void bundleName(Attributes attr, File file) throws IOException {

         try (JarFile jf = new JarFile(file)) {

             Attributes main = jf.getManifest().getMainAttributes();

             String version = main.getValue("Bundle-SymbolicName");

             if (version == null) {

                 version = main.getValue("OpenIDE-Module-Name");

             }

             if (version != null) {

                 attr.putValue("Bck2BrwsrName", version);

             }

         }

     }

     private static URLClassLoader buildClassLoader(File root, Collection<Artifact> deps) throws MalformedURLException {

         List<URL> arr = new ArrayList<URL>();

         if (root != null) {

             arr.add(root.toURI().toURL());

         }

         for (Artifact a : deps) {

             if (a.getFile() != null) {

                 arr.add(a.getFile().toURI().toURL());

             }

         }

         return new URLClassLoader(arr.toArray(new URL[0]), Java2JavaScript.class.getClassLoader());

     }

     private static boolean matches(String[] aotDeps, Artifact a) {

         for (String d : aotDeps) {

             String[] parts = d.split(":");

             for (int i = 0; i < parts.length; i++) {

                 if ("*".equals(parts[i])) {

                     parts[i] = null;

                 }

             }

             if (parts[0] != null && !parts[0].equals(a.getGroupId())) {

                 continue;

             }

             if (parts[1] != null && !parts[1].equals(a.getArtifactId())) {

                 continue;

             }

             if (parts.length > 2 && parts[2] != null && !parts[2].equals(a.getClassifier())) {

                 continue;

             }

             return true;

         }

         return false;

     }

 }