Mercurial > jdk8-mips64-public > langtools / file revision

 /*

  * Copyright 2007-2008 Sun Microsystems, Inc.  All Rights Reserved.

  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

  *

  * This code is free software; you can redistribute it and/or modify it

  * under the terms of the GNU General Public License version 2 only, as

  * published by the Free Software Foundation.

  *

  * This code 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

  * version 2 for more details (a copy is included in the LICENSE file that

  * accompanied this code).

  *

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

  * 2 along with this work; if not, write to the Free Software Foundation,

  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

  *

  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

  * CA 95054 USA or visit www.sun.com if you need additional information or

  * have any questions.

  */

 import java.util.List;

 import java.util.ArrayList;

 import java.io.File;

 //for CompilerHelper

 import java.io.OutputStream;

 import java.io.PrintWriter;

 import java.lang.reflect.Method;

 import java.lang.reflect.InvocationTargetException;

 public class WsCompileExample {

     File destDir;

     File srcDir;

     protected boolean compilerDebug = false;

     protected boolean compilerOptimize = false;

     protected String userClasspath = null;

     public static void main(String[] args) {

         new WsCompileExample().do_main(args);

     }

     public void do_main(String[] args) {

         if(!args[0].equals("-s")) {

             throw new RuntimeException("specify -s for src");

         }

         //run it once

         srcDir =  new File(args[1]);

         if(!args[2].equals("-d")) {

             throw new RuntimeException("specify -d for dest");

         }

         destDir =  new File(args[3]);

         if(!destDir.exists())

             destDir.mkdirs();

         System.out.println("----test compile 1-----");

         compileGeneratedClasses();

         //run it twice

          srcDir = new File(args[1]+"1");

          destDir =  new File(args[3]+"1");

         if(!destDir.exists())

             destDir.mkdirs();

         System.out.println("----test compile 2-----");

         compileGeneratedClasses();

     }

     protected void compileGeneratedClasses() {

         List sourceFiles = new ArrayList();

         for (File f: srcDir.listFiles()) {

             if (f.getName().endsWith(".java")) {

                 sourceFiles.add(f.getAbsolutePath());

             }

         }

         if (sourceFiles.size() > 0) {

                                 String classDir = destDir.getAbsolutePath();

             String classpathString = createClasspathString();

             System.out.println("classpathString: " + classpathString);

                                 String[] args = new String[4 + (compilerDebug == true ? 1 : 0) +

                 (compilerOptimize == true ? 1 : 0) +

                 sourceFiles.size()];

             args[0] = "-d";

             args[1] = classDir;

             args[2] = "-classpath";

             args[3] = classpathString;

 //                              args[4]="-DnonBatchMode";

             int baseIndex = 4;

             if (compilerDebug) {

                 args[baseIndex++] = "-g";

             }

             if (compilerOptimize) {

                 args[baseIndex++] = "-O";

             }

             for (int i = 0; i < sourceFiles.size(); ++i) {

                 args[baseIndex + i] = (String)sourceFiles.get(i);

             }

             // ByteArrayOutputStream javacOutput = new ByteArrayOutputStream();

             JavaCompilerHelper compilerHelper = new JavaCompilerHelper(System.out);

             boolean result = compilerHelper.compile(args);

             if (!result) {

                 System.out.println("wscompile.compilation Failed");

             }

         }

     }

     protected String createClasspathString() {

         if (userClasspath == null) {

             userClasspath = "";

         }

                           String jcp = userClasspath + File.pathSeparator + System.getProperty("java.class.path");

                   return jcp;

     }

 }

 ///////////////////////////////////////////////////////////////////

 class JavaCompilerHelper {

     public JavaCompilerHelper(OutputStream out) {

                 this.out = out;

         }

         public boolean compile(String[] args) {

                 return internalCompile(args);

         }

         protected boolean internalCompile(String[] args) {

                 System.out.println("Args: ");

                 for(String arg : args){

                         System.out.print(arg+" ");

                 }

         System.out.println();

                 ClassLoader cl = Thread.currentThread().getContextClassLoader();

                 Class comSunToolsJavacMainClass = null;

                 try {

                         /* try to use the new compiler */

                         comSunToolsJavacMainClass =

                                 cl.loadClass("com.sun.tools.javac.Main");

                         try {

                                 Method compileMethod =

                                         comSunToolsJavacMainClass.getMethod(

                                                 "compile",

                                                 compile141MethodSignature);

                                 try {

                                         Object result =

                                                 compileMethod.invoke(

                                                         null,

                                                         new Object[] { args, new PrintWriter(out)});

                                         if (!(result instanceof Integer)) {

                                                 return false;

                                         }

                                         return ((Integer) result).intValue() == 0;

                                 } catch (IllegalAccessException e3) {

                                         return false;

                                 } catch (IllegalArgumentException e3) {

                                         return false;

                                 } catch (InvocationTargetException e3) {

                                         return false;

                                 }

                         } catch (NoSuchMethodException e2) {

               System.out.println("ERROR: Compile failed with error:" + e2.toString() );

                         }

                 } catch (ClassNotFoundException e) {

                         e.printStackTrace();

                         return false;

                 } catch (SecurityException e) {

                         return false;

                 }

                 return true;

         }

         protected String getGenericErrorMessage() {return "javacompiler.error"; }

         protected void run() {  }

         protected boolean parseArguments(String[] args) {return false;}

         protected OutputStream out;

         protected static final Class[] compile141MethodSignature;

         static

         {

                 compile141MethodSignature = new Class[2];

                 compile141MethodSignature[0] = (new String[0]).getClass();

                 compile141MethodSignature[1] = PrintWriter.class;

         }

 }