jaroslav@462: /*
jaroslav@462:  * To change this template, choose Tools | Templates
jaroslav@462:  * and open the template in the editor.
jaroslav@462:  */
jaroslav@462: package org.apidesign.bck2brwsr.dew;
jaroslav@462: 
jaroslav@462: import java.io.IOException;
jaroslav@462: import java.nio.charset.Charset;
jaroslav@462: import java.util.ArrayList;
jaroslav@462: import java.util.Collections;
jaroslav@462: import java.util.Iterator;
jaroslav@462: import java.util.List;
jaroslav@462: import java.util.Locale;
jaroslav@462: import java.util.Map;
jaroslav@462: import java.util.Set;
jaroslav@462: import java.util.regex.Matcher;
jaroslav@462: import java.util.regex.Pattern;
jaroslav@462: import javax.tools.Diagnostic;
jaroslav@462: import javax.tools.DiagnosticListener;
jaroslav@462: import javax.tools.FileObject;
jaroslav@462: import javax.tools.JavaCompiler;
jaroslav@462: import javax.tools.JavaCompiler.CompilationTask;
jaroslav@462: import javax.tools.JavaFileManager;
jaroslav@462: import javax.tools.JavaFileObject;
jaroslav@462: import javax.tools.JavaFileObject.Kind;
jaroslav@462: import javax.tools.StandardJavaFileManager;
jaroslav@462: import javax.tools.StandardLocation;
jaroslav@462: 
jaroslav@462: /**
jaroslav@462:  *
jaroslav@462:  * @author Jaroslav Tulach <jtulach@netbeans.org>
jaroslav@462:  */
jaroslav@462: final class Compile implements JavaFileManager, DiagnosticListener<JavaFileObject> {
jaroslav@462:     private final JFO jfo;
jaroslav@462:     private final String pkg;
jaroslav@462:     private StandardJavaFileManager delegate;
jaroslav@462:     private List<Diagnostic<? extends JavaFileObject>> errors = new ArrayList<>();
jaroslav@462: 
jaroslav@462:     public Compile(String pkg, JFO jfo) {
jaroslav@462:         this.pkg = pkg;
jaroslav@462:         this.jfo = jfo;
jaroslav@462:     }
jaroslav@462:     
jaroslav@462:     public static Map<String,byte[]> compile(String html, String java) throws IOException {
jaroslav@462:         JavaCompiler jc = javax.tools.ToolProvider.getSystemJavaCompiler();
jaroslav@462:         String pkg = findPkg(java);
jaroslav@462:         String cls = findCls(java);
jaroslav@462:         
jaroslav@462:         JFO jfo = new JFO(java, pkg.replace('.', '/') + '/' + cls + ".java");
jaroslav@462:         final Compile cmp = new Compile(pkg, jfo);
jaroslav@462:         cmp.delegate = jc.getStandardFileManager(cmp, Locale.ENGLISH, Charset.forName("UTF-8"));
jaroslav@462:         
jaroslav@462:         Set<String> toCmp = Collections.singleton(pkg + '.' + cls);
jaroslav@462:         Set<JFO> unit = Collections.singleton(jfo);
jaroslav@462:         CompilationTask task = jc.getTask(null, cmp, cmp, null, null, unit);
jaroslav@462:         if (task.call() != true) {
jaroslav@462:             throw new IOException("Compilation failed: " + cmp.errors);
jaroslav@462:         }
jaroslav@462:         return Collections.emptyMap();
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public ClassLoader getClassLoader(Location location) {
jaroslav@462:         return null;//Compile.class.getClassLoader();
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public Iterable<JavaFileObject> list(Location location, String packageName, Set<Kind> kinds, boolean recurse) throws IOException {
jaroslav@462:         if (location == StandardLocation.PLATFORM_CLASS_PATH) {
jaroslav@462:             return delegate.list(location, packageName, kinds, recurse);
jaroslav@462:         }
jaroslav@462:         if (location == StandardLocation.CLASS_PATH) {
jaroslav@462:             return delegate.list(location, packageName, kinds, recurse);
jaroslav@462:         }
jaroslav@462:         if (location == StandardLocation.SOURCE_PATH) {
jaroslav@462:             if (packageName.equals(pkg)) {
jaroslav@462:                 return Collections.<JavaFileObject>singleton(jfo);
jaroslav@462:             }
jaroslav@462:             return Collections.emptyList();
jaroslav@462:         }
jaroslav@462:         throw new UnsupportedOperationException("Loc: " + location + " pkg: " + packageName + " kinds: " + kinds + " rec: " + recurse);
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public String inferBinaryName(Location location, JavaFileObject file) {
jaroslav@462:         if (file == jfo) {
jaroslav@462:             return pkg + "." + file.getName();
jaroslav@462:         }
jaroslav@462:         return delegate.inferBinaryName(location, file);
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public boolean isSameFile(FileObject a, FileObject b) {
jaroslav@462:         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public boolean handleOption(String current, Iterator<String> remaining) {
jaroslav@462:         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public boolean hasLocation(Location location) {
jaroslav@462:         return true;
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public JavaFileObject getJavaFileForInput(Location location, String className, Kind kind) throws IOException {
jaroslav@462:         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public JavaFileObject getJavaFileForOutput(Location location, String className, Kind kind, FileObject sibling) throws IOException {
jaroslav@462:         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public FileObject getFileForInput(Location location, String packageName, String relativeName) throws IOException {
jaroslav@462:         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public FileObject getFileForOutput(Location location, String packageName, String relativeName, FileObject sibling) throws IOException {
jaroslav@462:         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public void flush() throws IOException {
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public void close() throws IOException {
jaroslav@462:         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public int isSupportedOption(String option) {
jaroslav@462:         throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
jaroslav@462:     }
jaroslav@462: 
jaroslav@462:     @Override
jaroslav@462:     public void report(Diagnostic<? extends JavaFileObject> diagnostic) {
jaroslav@462:         errors.add(diagnostic);
jaroslav@462:     }
jaroslav@462:     private static String findPkg(String java) throws IOException {
jaroslav@462:         Pattern p = Pattern.compile("package\\p{javaWhitespace}*([\\p{Alnum}\\.]+)\\p{javaWhitespace}*;", Pattern.MULTILINE);
jaroslav@462:         Matcher m = p.matcher(java);
jaroslav@462:         if (!m.find()) {
jaroslav@462:             throw new IOException("Can't find package declaration in the java file");
jaroslav@462:         }
jaroslav@462:         String pkg = m.group(1);
jaroslav@462:         return pkg;
jaroslav@462:     }
jaroslav@462:     private static String findCls(String java) throws IOException {
jaroslav@462:         Pattern p = Pattern.compile("class\\p{javaWhitespace}*([\\p{Alnum}\\.]+)\\p{javaWhitespace}", Pattern.MULTILINE);
jaroslav@462:         Matcher m = p.matcher(java);
jaroslav@462:         if (!m.find()) {
jaroslav@462:             throw new IOException("Can't find package declaration in the java file");
jaroslav@462:         }
jaroslav@462:         String cls = m.group(1);
jaroslav@462:         return cls;
jaroslav@462:     }
jaroslav@462: }