From 872f08303e17f55baa5d79fb96414e91ee9b4e04 Mon Sep 17 00:00:00 2001
From: James Strachan <james.strachan@gmail.com>
Date: Wed, 30 May 2012 16:43:33 +0100
Subject: [PATCH] allow the JS maven plugin to copy the kotlin JS library file
 - or append it

---
 .../kotlin/maven/K2JSCompilerMojo.java        | 92 ++++++++++++++++++-
 1 file changed, 89 insertions(+), 3 deletions(-)

diff --git a/libraries/tools/kotlin-maven-plugin/src/main/java/org/jetbrains/kotlin/maven/K2JSCompilerMojo.java b/libraries/tools/kotlin-maven-plugin/src/main/java/org/jetbrains/kotlin/maven/K2JSCompilerMojo.java
index c34c1475646..73747633575 100644
--- a/libraries/tools/kotlin-maven-plugin/src/main/java/org/jetbrains/kotlin/maven/K2JSCompilerMojo.java
+++ b/libraries/tools/kotlin-maven-plugin/src/main/java/org/jetbrains/kotlin/maven/K2JSCompilerMojo.java
@@ -16,11 +16,19 @@
 
 package org.jetbrains.kotlin.maven;
 
+import com.google.common.io.Files;
+import com.google.common.io.InputSupplier;
 import org.apache.maven.plugin.MojoExecutionException;
+import org.apache.maven.plugin.MojoFailureException;
 import org.jetbrains.jet.cli.common.CLICompiler;
 import org.jetbrains.jet.cli.common.CompilerArguments;
 import org.jetbrains.jet.cli.js.K2JSCompiler;
 import org.jetbrains.jet.cli.js.K2JSCompilerArguments;
+import org.jetbrains.jet.internal.com.intellij.openapi.util.io.FileUtil;
+import org.jetbrains.k2js.config.MetaInfServices;
+
+import java.io.*;
+import java.nio.charset.Charset;
 
 /**
  * Converts Kotlin to JavaScript code
@@ -30,13 +38,40 @@ import org.jetbrains.jet.cli.js.K2JSCompilerArguments;
  * @noinspection UnusedDeclaration
  */
 public class K2JSCompilerMojo extends KotlinCompileMojo {
+    public static final String KOTLIN_JS_LIB = "kotlin-lib.js";
+
     /**
      * The output JS file name
      *
      * @required
      * @parameter default-value="${project.build.directory}/js/${project.artifactId}.js"
      */
-    private String outFile;
+    private String outputFile;
+
+    /**
+     * The output Kotlin JS file
+     *
+     * @required
+     * @parameter default-value="${project.build.directory}/js/kotlin-lib.js"
+     * @parameter expression="${outputKotlinJSFile}"
+     */
+    private File outputKotlinJSFile;
+
+    /**
+     * Whether to copy the kotlin-lib.js file to the output directory
+     *
+     * @parameter default-value="true"
+     * @parameter expression="${copyLibraryJS}"
+     */
+    private Boolean copyLibraryJS;
+
+    /**
+     * Whether to copy the kotlin-lib.js file to the output directory
+     *
+     * @parameter default-value="false"
+     * @parameter expression="${appendLibraryJS}"
+     */
+    private Boolean appendLibraryJS;
 
     /**
      * Whether verbose logging is enabled or not.
@@ -46,13 +81,64 @@ public class K2JSCompilerMojo extends KotlinCompileMojo {
      */
     private Boolean verbose;
 
+    @Override
+    public void execute() throws MojoExecutionException, MojoFailureException {
+        super.execute();
+        if (copyLibraryJS) {
+            getLog().info("Copying kotlin JS library to " + outputKotlinJSFile);
+
+            // lets copy the kotlin library into the output directory
+            try {
+                File parentFile = outputKotlinJSFile.getParentFile();
+                parentFile.mkdirs();
+                final InputStream inputStream = MetaInfServices.loadClasspathResource(KOTLIN_JS_LIB);
+                if (inputStream == null) {
+                    System.out.println("WARNING: Could not find " + KOTLIN_JS_LIB + " on the classpath!");
+                } else {
+                    InputSupplier<InputStream> inputSupplier = new InputSupplier<InputStream>() {
+                        @Override
+                        public InputStream getInput() throws IOException {
+                            return inputStream;
+                        }
+                    };
+                    Files.copy(inputSupplier, outputKotlinJSFile);
+                }
+            } catch (IOException e) {
+                throw new MojoExecutionException(e.getMessage(), e);
+            }
+        }
+        if (appendLibraryJS) {
+            getLog().info("Appending Kotlin Library JS to the generated file " + outputFile);
+
+            // lets copy the kotlin library into the output directory
+            try {
+                final InputStream inputStream = MetaInfServices.loadClasspathResource(KOTLIN_JS_LIB);
+                if (inputStream == null) {
+                    System.out.println("WARNING: Could not find " + KOTLIN_JS_LIB + " on the classpath!");
+                } else {
+                    InputSupplier<InputStream> inputSupplier = new InputSupplier<InputStream>() {
+                        @Override
+                        public InputStream getInput() throws IOException {
+                            return inputStream;
+                        }
+                    };
+                    String text = "\n" + FileUtil.loadTextAndClose(inputStream);
+                    Charset charset = Charset.defaultCharset();
+                    Files.append(text, new File(outputFile), charset);
+                }
+            } catch (IOException e) {
+                throw new MojoExecutionException(e.getMessage(), e);
+            }
+        }
+    }
+
     @Override
     protected void configureCompilerArguments(CompilerArguments arguments) throws MojoExecutionException {
         super.configureCompilerArguments(arguments);
 
         if (arguments instanceof K2JSCompilerArguments) {
             K2JSCompilerArguments k2jsArgs = (K2JSCompilerArguments)arguments;
-            k2jsArgs.outputFile = outFile;
+            k2jsArgs.outputFile = outputFile;
             if (verbose != null) {
                 k2jsArgs.verbose = verbose;
             }
@@ -60,7 +146,7 @@ public class K2JSCompilerMojo extends KotlinCompileMojo {
                 k2jsArgs.sourceFiles = sources;
             }
         }
-        getLog().info("Compiling Kotlin src from " + arguments.getSrc() + " to JavaScript at: " + outFile);
+        getLog().info("Compiling Kotlin src from " + arguments.getSrc() + " to JavaScript at: " + outputFile);
     }
 
     @Override