hg/bck2brwsr: launcher/src/main/java/org/apidesign/bck2brwsr/dew/Compile.java@3641fd0663d3

     1 /*

     2  * To change this template, choose Tools | Templates

     3  * and open the template in the editor.

     4  */

     5 package org.apidesign.bck2brwsr.dew;

     7 import java.io.ByteArrayInputStream;

     8 import java.io.ByteArrayOutputStream;

     9 import java.io.IOException;

    10 import java.io.InputStream;

    11 import java.io.OutputStream;

    12 import java.net.URI;

    13 import java.net.URISyntaxException;

    14 import java.nio.charset.Charset;

    15 import java.util.ArrayList;

    16 import java.util.Arrays;

    17 import java.util.Collections;

    18 import java.util.HashMap;

    19 import java.util.Iterator;

    20 import java.util.List;

    21 import java.util.Locale;

    22 import java.util.Map;

    23 import java.util.Set;

    24 import java.util.logging.Level;

    25 import java.util.logging.Logger;

    26 import java.util.regex.Matcher;

    27 import java.util.regex.Pattern;

    28 import javax.tools.Diagnostic;

    29 import javax.tools.DiagnosticListener;

    30 import javax.tools.FileObject;

    31 import javax.tools.ForwardingJavaFileManager;

    32 import javax.tools.JavaCompiler;

    33 import javax.tools.JavaCompiler.CompilationTask;

    34 import javax.tools.JavaFileManager;

    35 import javax.tools.JavaFileObject;

    36 import javax.tools.JavaFileObject.Kind;

    37 import javax.tools.SimpleJavaFileObject;

    38 import javax.tools.StandardJavaFileManager;

    39 import javax.tools.StandardLocation;

    40 import javax.tools.ToolProvider;

    42 /**

    43  *

    44  * @author Jaroslav Tulach <jtulach@netbeans.org>

    45  */

    46 final class Compile implements JavaFileManager, DiagnosticListener<JavaFileObject> {

    47     private final JFO jfo;

    48     private final String pkg;

    49     private StandardJavaFileManager delegate;

    50     private List<Diagnostic<? extends JavaFileObject>> errors = new ArrayList<>();

    52     public Compile(String pkg, JFO jfo) {

    53         this.pkg = pkg;

    54         this.jfo = jfo;

    55     }

    57     /*

    58     public static Map<String,byte[]> compile(String html, String java) throws IOException {

    59         JavaCompiler jc = javax.tools.ToolProvider.getSystemJavaCompiler();

    60         String pkg = findPkg(java);

    61         String cls = findCls(java);

    63         JFO jfo = new JFO(java, pkg.replace('.', '/') + '/' + cls + ".java");

    64         final Compile cmp = new Compile(pkg, jfo);

    65         cmp.delegate = jc.getStandardFileManager(cmp, Locale.ENGLISH, Charset.forName("UTF-8"));

    67         Set<String> toCmp = Collections.singleton(pkg + '.' + cls);

    68         Set<JFO> unit = Collections.singleton(jfo);

    69         CompilationTask task = jc.getTask(null, cmp, cmp, null, null, unit);

    70         if (task.call() != true) {

    71             throw new IOException("Compilation failed: " + cmp.errors);

    72         }

    73         return Collections.emptyMap();

    74     }

    75     */

    76     public static Map<String, byte[]> compile(final String html, final String code) throws IOException {

    77         final String pkg = findPkg(code);

    78 //        String cls = findCls(code);

    80         DiagnosticListener<JavaFileObject> devNull = new DiagnosticListener<JavaFileObject>() {

    81             public void report(Diagnostic<? extends JavaFileObject> diagnostic) {

    82                 System.err.println("diagnostic=" + diagnostic);

    83             }

    84         };

    85         StandardJavaFileManager sjfm = ToolProvider.getSystemJavaCompiler().getStandardFileManager(devNull, null, null);

    87 //        sjfm.setLocation(StandardLocation.PLATFORM_CLASS_PATH, toFiles(boot));

    88 //        sjfm.setLocation(StandardLocation.CLASS_PATH, toFiles(compile));

    90         final Map<String, ByteArrayOutputStream> class2BAOS = new HashMap<String, ByteArrayOutputStream>();

    92         JavaFileObject file = new SimpleJavaFileObject(URI.create("mem://mem"), Kind.SOURCE) {

    93             @Override

    94             public CharSequence getCharContent(boolean ignoreEncodingErrors) throws IOException {

    95                 return code;

    96             }

    97         };

    98         final JavaFileObject htmlFile = new SimpleJavaFileObject(URI.create("mem://mem2"), Kind.OTHER) {

    99             @Override

   100             public CharSequence getCharContent(boolean ignoreEncodingErrors) throws IOException {

   101                 return html;

   102             }

   104             @Override

   105             public InputStream openInputStream() throws IOException {

   106                 return new ByteArrayInputStream(html.getBytes());

   107             }

   108         };

   110         final URI scratch;

   111         try {

   112             scratch = new URI("mem://mem3");

   113         } catch (URISyntaxException ex) {

   114             throw new IOException(ex);

   115         }

   117         JavaFileManager jfm = new ForwardingJavaFileManager<JavaFileManager>(sjfm) {

   118             @Override

   119             public JavaFileObject getJavaFileForOutput(Location location, String className, Kind kind, FileObject sibling) throws IOException {

   120                 if (kind  == Kind.CLASS) {

   121                     final ByteArrayOutputStream buffer = new ByteArrayOutputStream();

   123                     class2BAOS.put(className.replace('.', '/') + ".class", buffer);

   124                     return new SimpleJavaFileObject(sibling.toUri(), kind) {

   125                         @Override

   126                         public OutputStream openOutputStream() throws IOException {

   127                             return buffer;

   128                         }

   129                     };

   130                 }

   132                 if (kind == Kind.SOURCE) {

   133                     return new SimpleJavaFileObject(scratch/*sibling.toUri()*/, kind) {

   134                         private final ByteArrayOutputStream data = new ByteArrayOutputStream();

   135                         @Override

   136                         public OutputStream openOutputStream() throws IOException {

   137                             return data;

   138                         }

   140                         @Override

   141                         public CharSequence getCharContent(boolean ignoreEncodingErrors) throws IOException {

   142                             data.close();

   143                             return new String(data.toByteArray());

   144                         }

   145                     };

   146                 }

   148                 throw new IllegalStateException();

   149             }

   150 //            @Override

   151 //            public Iterable<JavaFileObject> list(Location location, String packageName, Set<Kind> kinds, boolean recurse) throws IOException {

   152 //                if (location == StandardLocation.PLATFORM_CLASS_PATH) {

   153 //                    return super.list(location, packageName, kinds, recurse);

   154 //                }

   155 //                if (location == StandardLocation.CLASS_PATH) {

   156 //                    return super.list(location, packageName, kinds, recurse);

   157 //                }

   158 //                if (location == StandardLocation.SOURCE_PATH) {

   159 //                    System.out.println("src path for " + packageName + " kinds: " + kinds);

   160 //                    if (packageName.equals(pkg) && kinds.contains(Kind.OTHER)) {

   161 //                        return Collections.<JavaFileObject>singleton(htmlFile);

   162 //                    }

   163 //                    return Collections.emptyList();

   164 //                }

   165 //                throw new UnsupportedOperationException("Loc: " + location + " pkg: " + packageName + " kinds: " + kinds + " rec: " + recurse);

   166 //            }

   168             @Override

   169             public FileObject getFileForInput(Location location, String packageName, String relativeName) throws IOException {

   170                 if (location == StandardLocation.SOURCE_PATH) {

   171 //                    System.out.println("src path for " + packageName + " kinds: " + kinds);

   172                     if (packageName.equals(pkg)) {

   173                         return htmlFile;

   174                     }

   175                 }

   177                 return null;

   178             }

   180         };

   182         ToolProvider.getSystemJavaCompiler().getTask(null, jfm, devNull, /*XXX:*/Arrays.asList("-source", "1.7", "-target", "1.7"), null, Arrays.asList(file)).call();

   184         Map<String, byte[]> result = new HashMap<String, byte[]>();

   186         for (Map.Entry<String, ByteArrayOutputStream> e : class2BAOS.entrySet()) {

   187             result.put(e.getKey(), e.getValue().toByteArray());

   188         }

   190         return result;

   191     }

   193     @Override

   194     public ClassLoader getClassLoader(Location location) {

   195         return null;//Compile.class.getClassLoader();

   196     }

   198     @Override

   199     public Iterable<JavaFileObject> list(Location location, String packageName, Set<Kind> kinds, boolean recurse) throws IOException {

   200         if (location == StandardLocation.PLATFORM_CLASS_PATH) {

   201             return delegate.list(location, packageName, kinds, recurse);

   202         }

   203         if (location == StandardLocation.CLASS_PATH) {

   204             return delegate.list(location, packageName, kinds, recurse);

   205         }

   206         if (location == StandardLocation.SOURCE_PATH) {

   207             if (packageName.equals(pkg)) {

   208                 return Collections.<JavaFileObject>singleton(jfo);

   209             }

   210             return Collections.emptyList();

   211         }

   212         throw new UnsupportedOperationException("Loc: " + location + " pkg: " + packageName + " kinds: " + kinds + " rec: " + recurse);

   213     }

   215     @Override

   216     public String inferBinaryName(Location location, JavaFileObject file) {

   217         if (file == jfo) {

   218             return pkg + "." + file.getName();

   219         }

   220         return delegate.inferBinaryName(location, file);

   221     }

   223     @Override

   224     public boolean isSameFile(FileObject a, FileObject b) {

   225         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

   226     }

   228     @Override

   229     public boolean handleOption(String current, Iterator<String> remaining) {

   230         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

   231     }

   233     @Override

   234     public boolean hasLocation(Location location) {

   235         return true;

   236     }

   238     @Override

   239     public JavaFileObject getJavaFileForInput(Location location, String className, Kind kind) throws IOException {

   240         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

   241     }

   243     @Override

   244     public JavaFileObject getJavaFileForOutput(Location location, String className, Kind kind, FileObject sibling) throws IOException {

   245         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

   246     }

   248     @Override

   249     public FileObject getFileForInput(Location location, String packageName, String relativeName) throws IOException {

   250         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

   251     }

   253     @Override

   254     public FileObject getFileForOutput(Location location, String packageName, String relativeName, FileObject sibling) throws IOException {

   255         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

   256     }

   258     @Override

   259     public void flush() throws IOException {

   260     }

   262     @Override

   263     public void close() throws IOException {

   264         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

   265     }

   267     @Override

   268     public int isSupportedOption(String option) {

   269         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

   270     }

   272     @Override

   273     public void report(Diagnostic<? extends JavaFileObject> diagnostic) {

   274         errors.add(diagnostic);

   275     }

   276     private static String findPkg(String java) throws IOException {

   277         Pattern p = Pattern.compile("package\\p{javaWhitespace}*([\\p{Alnum}\\.]+)\\p{javaWhitespace}*;", Pattern.MULTILINE);

   278         Matcher m = p.matcher(java);

   279         if (!m.find()) {

   280             throw new IOException("Can't find package declaration in the java file");

   281         }

   282         String pkg = m.group(1);

   283         return pkg;

   284     }

   285     private static String findCls(String java) throws IOException {

   286         Pattern p = Pattern.compile("class\\p{javaWhitespace}*([\\p{Alnum}\\.]+)\\p{javaWhitespace}", Pattern.MULTILINE);

   287         Matcher m = p.matcher(java);

   288         if (!m.find()) {

   289             throw new IOException("Can't find package declaration in the java file");

   290         }

   291         String cls = m.group(1);

   292         return cls;

   293     }

   294 }

author	Jaroslav Tulach <jaroslav.tulach@apidesign.org>
	Wed, 16 Jan 2013 10:52:06 +0100
branch	dew
changeset 463	3641fd0663d3
parent 462	aa69b1387624
child 464	9823859d253a
permissions	-rw-r--r--