dew/src/main/java/org/apidesign/bck2brwsr/launcher/Bck2BrwsrLauncher.java
branchdew
changeset 544 08ffdc3938e7
parent 543 1adce93fea0f
     1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/dew/src/main/java/org/apidesign/bck2brwsr/launcher/Bck2BrwsrLauncher.java	Wed Jan 23 13:18:46 2013 +0100
     1.3 @@ -0,0 +1,505 @@
     1.4 +/**
     1.5 + * Back 2 Browser Bytecode Translator
     1.6 + * Copyright (C) 2012 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
     1.7 + *
     1.8 + * This program is free software: you can redistribute it and/or modify
     1.9 + * it under the terms of the GNU General Public License as published by
    1.10 + * the Free Software Foundation, version 2 of the License.
    1.11 + *
    1.12 + * This program is distributed in the hope that it will be useful,
    1.13 + * but WITHOUT ANY WARRANTY; without even the implied warranty of
    1.14 + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    1.15 + * GNU General Public License for more details.
    1.16 + *
    1.17 + * You should have received a copy of the GNU General Public License
    1.18 + * along with this program. Look for COPYING file in the top folder.
    1.19 + * If not, see http://opensource.org/licenses/GPL-2.0.
    1.20 + */
    1.21 +package org.apidesign.bck2brwsr.launcher;
    1.22 +
    1.23 +import java.io.Closeable;
    1.24 +import java.io.File;
    1.25 +import java.io.IOException;
    1.26 +import java.io.InputStream;
    1.27 +import java.io.InterruptedIOException;
    1.28 +import java.io.OutputStream;
    1.29 +import java.io.Writer;
    1.30 +import java.net.URI;
    1.31 +import java.net.URISyntaxException;
    1.32 +import java.net.URL;
    1.33 +import java.util.ArrayList;
    1.34 +import java.util.Arrays;
    1.35 +import java.util.Enumeration;
    1.36 +import java.util.LinkedHashSet;
    1.37 +import java.util.List;
    1.38 +import java.util.Set;
    1.39 +import java.util.concurrent.BlockingQueue;
    1.40 +import java.util.concurrent.CountDownLatch;
    1.41 +import java.util.concurrent.LinkedBlockingQueue;
    1.42 +import java.util.concurrent.TimeUnit;
    1.43 +import java.util.logging.Level;
    1.44 +import java.util.logging.Logger;
    1.45 +import org.apidesign.bck2brwsr.dew.Dew;
    1.46 +import org.apidesign.vm4brwsr.Bck2Brwsr;
    1.47 +import org.glassfish.grizzly.PortRange;
    1.48 +import org.glassfish.grizzly.http.server.HttpHandler;
    1.49 +import org.glassfish.grizzly.http.server.HttpServer;
    1.50 +import org.glassfish.grizzly.http.server.NetworkListener;
    1.51 +import org.glassfish.grizzly.http.server.Request;
    1.52 +import org.glassfish.grizzly.http.server.Response;
    1.53 +import org.glassfish.grizzly.http.server.ServerConfiguration;
    1.54 +
    1.55 +/**
    1.56 + * Lightweight server to launch Bck2Brwsr applications and tests.
    1.57 + * Supports execution in native browser as well as Java's internal 
    1.58 + * execution engine.
    1.59 + */
    1.60 +final class Bck2BrwsrLauncher extends Launcher implements Closeable {
    1.61 +    private static final Logger LOG = Logger.getLogger(Bck2BrwsrLauncher.class.getName());
    1.62 +    private static final MethodInvocation END = new MethodInvocation(null, null, null);
    1.63 +    private Set<ClassLoader> loaders = new LinkedHashSet<>();
    1.64 +    private Set<Bck2Brwsr.Resources> xRes = new LinkedHashSet<>();
    1.65 +    private BlockingQueue<MethodInvocation> methods = new LinkedBlockingQueue<>();
    1.66 +    private long timeOut;
    1.67 +    private final Res resources = new Res();
    1.68 +    private final String cmd;
    1.69 +    private Object[] brwsr;
    1.70 +    private HttpServer server;
    1.71 +    private CountDownLatch wait;
    1.72 +
    1.73 +    public Bck2BrwsrLauncher(String cmd) {
    1.74 +        this.cmd = cmd;
    1.75 +    }
    1.76 +    
    1.77 +    @Override
    1.78 +     MethodInvocation addMethod(Class<?> clazz, String method, String html) throws IOException {
    1.79 +        loaders.add(clazz.getClassLoader());
    1.80 +        MethodInvocation c = new MethodInvocation(clazz.getName(), method, html);
    1.81 +        methods.add(c);
    1.82 +        try {
    1.83 +            c.await(timeOut);
    1.84 +        } catch (InterruptedException ex) {
    1.85 +            throw new IOException(ex);
    1.86 +        }
    1.87 +        return c;
    1.88 +    }
    1.89 +    
    1.90 +    public void setTimeout(long ms) {
    1.91 +        timeOut = ms;
    1.92 +    }
    1.93 +    
    1.94 +    public void addClassLoader(ClassLoader url) {
    1.95 +        this.loaders.add(url);
    1.96 +    }
    1.97 +
    1.98 +    public void showURL(String startpage) throws IOException {
    1.99 +        if (!startpage.startsWith("/")) {
   1.100 +            startpage = "/" + startpage;
   1.101 +        }
   1.102 +        HttpServer s = initServer();
   1.103 +        s.getServerConfiguration().addHttpHandler(new Page(resources, null), "/");
   1.104 +        try {
   1.105 +            launchServerAndBrwsr(s, startpage);
   1.106 +        } catch (URISyntaxException | InterruptedException ex) {
   1.107 +            throw new IOException(ex);
   1.108 +        }
   1.109 +    }
   1.110 +    
   1.111 +    public static void main(String... args) throws Exception {
   1.112 +        Bck2BrwsrLauncher l = new Bck2BrwsrLauncher(null);
   1.113 +        l.addClassLoader(Bck2BrwsrLauncher.class.getClassLoader());
   1.114 +        HttpServer s = l.initServer();
   1.115 +        final Dew dew = new Dew();
   1.116 +        s.getServerConfiguration().addHttpHandler(dew, "/dew/");
   1.117 +        l.xRes.add(dew);
   1.118 +        l.launchServerAndBrwsr(s, "/dew/");
   1.119 +        System.in.read();
   1.120 +    }
   1.121 +
   1.122 +    @Override
   1.123 +    public void initialize() throws IOException {
   1.124 +        try {
   1.125 +            executeInBrowser();
   1.126 +        } catch (InterruptedException ex) {
   1.127 +            final InterruptedIOException iio = new InterruptedIOException(ex.getMessage());
   1.128 +            iio.initCause(ex);
   1.129 +            throw iio;
   1.130 +        } catch (Exception ex) {
   1.131 +            if (ex instanceof IOException) {
   1.132 +                throw (IOException)ex;
   1.133 +            }
   1.134 +            if (ex instanceof RuntimeException) {
   1.135 +                throw (RuntimeException)ex;
   1.136 +            }
   1.137 +            throw new IOException(ex);
   1.138 +        }
   1.139 +    }
   1.140 +    
   1.141 +    private HttpServer initServer() {
   1.142 +        HttpServer s = HttpServer.createSimpleServer(".", new PortRange(8080, 65535));
   1.143 +
   1.144 +        final ServerConfiguration conf = s.getServerConfiguration();
   1.145 +        conf.addHttpHandler(new Page(resources, 
   1.146 +            "org/apidesign/bck2brwsr/launcher/console.xhtml",
   1.147 +            "org.apidesign.bck2brwsr.launcher.Console", "welcome", "false"
   1.148 +        ), "/console");
   1.149 +        conf.addHttpHandler(new VM(resources), "/bck2brwsr.js");
   1.150 +        conf.addHttpHandler(new VMInit(), "/vm.js");
   1.151 +        conf.addHttpHandler(new Classes(resources), "/classes/");
   1.152 +        return s;
   1.153 +    }
   1.154 +    
   1.155 +    private void executeInBrowser() throws InterruptedException, URISyntaxException, IOException {
   1.156 +        wait = new CountDownLatch(1);
   1.157 +        server = initServer();
   1.158 +        ServerConfiguration conf = server.getServerConfiguration();
   1.159 +        conf.addHttpHandler(new Page(resources, 
   1.160 +            "org/apidesign/bck2brwsr/launcher/harness.xhtml"
   1.161 +        ), "/execute");
   1.162 +        conf.addHttpHandler(new HttpHandler() {
   1.163 +            int cnt;
   1.164 +            List<MethodInvocation> cases = new ArrayList<>();
   1.165 +            @Override
   1.166 +            public void service(Request request, Response response) throws Exception {
   1.167 +                String id = request.getParameter("request");
   1.168 +                String value = request.getParameter("result");
   1.169 +                
   1.170 +                if (id != null && value != null) {
   1.171 +                    LOG.log(Level.INFO, "Received result for case {0} = {1}", new Object[]{id, value});
   1.172 +                    value = decodeURL(value);
   1.173 +                    cases.get(Integer.parseInt(id)).result(value, null);
   1.174 +                }
   1.175 +                
   1.176 +                MethodInvocation mi = methods.take();
   1.177 +                if (mi == END) {
   1.178 +                    response.getWriter().write("");
   1.179 +                    wait.countDown();
   1.180 +                    cnt = 0;
   1.181 +                    LOG.log(Level.INFO, "End of data reached. Exiting.");
   1.182 +                    return;
   1.183 +                }
   1.184 +                
   1.185 +                cases.add(mi);
   1.186 +                final String cn = mi.className;
   1.187 +                final String mn = mi.methodName;
   1.188 +                LOG.log(Level.INFO, "Request for {0} case. Sending {1}.{2}", new Object[]{cnt, cn, mn});
   1.189 +                response.getWriter().write("{"
   1.190 +                    + "className: '" + cn + "', "
   1.191 +                    + "methodName: '" + mn + "', "
   1.192 +                    + "request: " + cnt
   1.193 +                );
   1.194 +                if (mi.html != null) {
   1.195 +                    response.getWriter().write(", html: '");
   1.196 +                    response.getWriter().write(encodeJSON(mi.html));
   1.197 +                    response.getWriter().write("'");
   1.198 +                }
   1.199 +                response.getWriter().write("}");
   1.200 +                cnt++;
   1.201 +            }
   1.202 +        }, "/data");
   1.203 +
   1.204 +        this.brwsr = launchServerAndBrwsr(server, "/execute");
   1.205 +    }
   1.206 +    
   1.207 +    private static String encodeJSON(String in) {
   1.208 +        StringBuilder sb = new StringBuilder();
   1.209 +        for (int i = 0; i < in.length(); i++) {
   1.210 +            char ch = in.charAt(i);
   1.211 +            if (ch < 32 || ch == '\'' || ch == '"') {
   1.212 +                sb.append("\\u");
   1.213 +                String hs = "0000" + Integer.toHexString(ch);
   1.214 +                hs = hs.substring(hs.length() - 4);
   1.215 +                sb.append(hs);
   1.216 +            } else {
   1.217 +                sb.append(ch);
   1.218 +            }
   1.219 +        }
   1.220 +        return sb.toString();
   1.221 +    }
   1.222 +    
   1.223 +    @Override
   1.224 +    public void shutdown() throws IOException {
   1.225 +        methods.offer(END);
   1.226 +        for (;;) {
   1.227 +            int prev = methods.size();
   1.228 +            try {
   1.229 +                if (wait != null && wait.await(timeOut, TimeUnit.MILLISECONDS)) {
   1.230 +                    break;
   1.231 +                }
   1.232 +            } catch (InterruptedException ex) {
   1.233 +                throw new IOException(ex);
   1.234 +            }
   1.235 +            if (prev == methods.size()) {
   1.236 +                LOG.log(
   1.237 +                    Level.WARNING, 
   1.238 +                    "Timeout and no test has been executed meanwhile (at {0}). Giving up.", 
   1.239 +                    methods.size()
   1.240 +                );
   1.241 +                break;
   1.242 +            }
   1.243 +            LOG.log(Level.INFO, 
   1.244 +                "Timeout, but tests got from {0} to {1}. Trying again.", 
   1.245 +                new Object[]{prev, methods.size()}
   1.246 +            );
   1.247 +        }
   1.248 +        stopServerAndBrwsr(server, brwsr);
   1.249 +    }
   1.250 +    
   1.251 +    static void copyStream(InputStream is, OutputStream os, String baseURL, String... params) throws IOException {
   1.252 +        for (;;) {
   1.253 +            int ch = is.read();
   1.254 +            if (ch == -1) {
   1.255 +                break;
   1.256 +            }
   1.257 +            if (ch == '$' && params.length > 0) {
   1.258 +                int cnt = is.read() - '0';
   1.259 +                if (cnt == 'U' - '0') {
   1.260 +                    os.write(baseURL.getBytes());
   1.261 +                }
   1.262 +                if (cnt >= 0 && cnt < params.length) {
   1.263 +                    os.write(params[cnt].getBytes());
   1.264 +                }
   1.265 +            } else {
   1.266 +                os.write(ch);
   1.267 +            }
   1.268 +        }
   1.269 +    }
   1.270 +
   1.271 +    private Object[] launchServerAndBrwsr(HttpServer server, final String page) throws IOException, URISyntaxException, InterruptedException {
   1.272 +        server.start();
   1.273 +        NetworkListener listener = server.getListeners().iterator().next();
   1.274 +        int port = listener.getPort();
   1.275 +        
   1.276 +        URI uri = new URI("http://localhost:" + port + page);
   1.277 +        LOG.log(Level.INFO, "Showing {0}", uri);
   1.278 +        if (cmd == null) {
   1.279 +            try {
   1.280 +                LOG.log(Level.INFO, "Trying Desktop.browse on {0} {2} by {1}", new Object[] {
   1.281 +                    System.getProperty("java.vm.name"),
   1.282 +                    System.getProperty("java.vm.vendor"),
   1.283 +                    System.getProperty("java.vm.version"),
   1.284 +                });
   1.285 +                java.awt.Desktop.getDesktop().browse(uri);
   1.286 +                LOG.log(Level.INFO, "Desktop.browse successfully finished");
   1.287 +                return null;
   1.288 +            } catch (UnsupportedOperationException ex) {
   1.289 +                LOG.log(Level.INFO, "Desktop.browse not supported: {0}", ex.getMessage());
   1.290 +                LOG.log(Level.FINE, null, ex);
   1.291 +            }
   1.292 +        }
   1.293 +        {
   1.294 +            String cmdName = cmd == null ? "xdg-open" : cmd;
   1.295 +            String[] cmdArr = { 
   1.296 +                cmdName, uri.toString()
   1.297 +            };
   1.298 +            LOG.log(Level.INFO, "Launching {0}", Arrays.toString(cmdArr));
   1.299 +            final Process process = Runtime.getRuntime().exec(cmdArr);
   1.300 +            return new Object[] { process, null };
   1.301 +        }
   1.302 +    }
   1.303 +    
   1.304 +    private static String decodeURL(String s) {
   1.305 +        for (;;) {
   1.306 +            int pos = s.indexOf('%');
   1.307 +            if (pos == -1) {
   1.308 +                return s;
   1.309 +            }
   1.310 +            int i = Integer.parseInt(s.substring(pos + 1, pos + 2), 16);
   1.311 +            s = s.substring(0, pos) + (char)i + s.substring(pos + 2);
   1.312 +        }
   1.313 +    }
   1.314 +    
   1.315 +    private void stopServerAndBrwsr(HttpServer server, Object[] brwsr) throws IOException {
   1.316 +        if (brwsr == null) {
   1.317 +            return;
   1.318 +        }
   1.319 +        Process process = (Process)brwsr[0];
   1.320 +        
   1.321 +        server.stop();
   1.322 +        InputStream stdout = process.getInputStream();
   1.323 +        InputStream stderr = process.getErrorStream();
   1.324 +        drain("StdOut", stdout);
   1.325 +        drain("StdErr", stderr);
   1.326 +        process.destroy();
   1.327 +        int res;
   1.328 +        try {
   1.329 +            res = process.waitFor();
   1.330 +        } catch (InterruptedException ex) {
   1.331 +            throw new IOException(ex);
   1.332 +        }
   1.333 +        LOG.log(Level.INFO, "Exit code: {0}", res);
   1.334 +
   1.335 +        deleteTree((File)brwsr[1]);
   1.336 +    }
   1.337 +    
   1.338 +    private static void drain(String name, InputStream is) throws IOException {
   1.339 +        int av = is.available();
   1.340 +        if (av > 0) {
   1.341 +            StringBuilder sb = new StringBuilder();
   1.342 +            sb.append("v== ").append(name).append(" ==v\n");
   1.343 +            while (av-- > 0) {
   1.344 +                sb.append((char)is.read());
   1.345 +            }
   1.346 +            sb.append("\n^== ").append(name).append(" ==^");
   1.347 +            LOG.log(Level.INFO, sb.toString());
   1.348 +        }
   1.349 +    }
   1.350 +
   1.351 +    private void deleteTree(File file) {
   1.352 +        if (file == null) {
   1.353 +            return;
   1.354 +        }
   1.355 +        File[] arr = file.listFiles();
   1.356 +        if (arr != null) {
   1.357 +            for (File s : arr) {
   1.358 +                deleteTree(s);
   1.359 +            }
   1.360 +        }
   1.361 +        file.delete();
   1.362 +    }
   1.363 +
   1.364 +    @Override
   1.365 +    public void close() throws IOException {
   1.366 +        shutdown();
   1.367 +    }
   1.368 +
   1.369 +    private class Res implements Bck2Brwsr.Resources {
   1.370 +        @Override
   1.371 +        public InputStream get(String resource) throws IOException {
   1.372 +            for (ClassLoader l : loaders) {
   1.373 +                URL u = null;
   1.374 +                Enumeration<URL> en = l.getResources(resource);
   1.375 +                while (en.hasMoreElements()) {
   1.376 +                    u = en.nextElement();
   1.377 +                }
   1.378 +                if (u != null) {
   1.379 +                    return u.openStream();
   1.380 +                }
   1.381 +            }
   1.382 +            for (Bck2Brwsr.Resources r : xRes) {
   1.383 +                InputStream is = r.get(resource);
   1.384 +                if (is != null) {
   1.385 +                    return is;
   1.386 +                }
   1.387 +            }
   1.388 +            throw new IOException("Can't find " + resource);
   1.389 +        }
   1.390 +    }
   1.391 +
   1.392 +    private static class Page extends HttpHandler {
   1.393 +        private final String resource;
   1.394 +        private final String[] args;
   1.395 +        private final Res res;
   1.396 +        
   1.397 +        public Page(Res res, String resource, String... args) {
   1.398 +            this.res = res;
   1.399 +            this.resource = resource;
   1.400 +            this.args = args.length == 0 ? new String[] { "$0" } : args;
   1.401 +        }
   1.402 +
   1.403 +        @Override
   1.404 +        public void service(Request request, Response response) throws Exception {
   1.405 +            String r = resource;
   1.406 +            if (r == null) {
   1.407 +                r = request.getHttpHandlerPath();
   1.408 +                if (r.startsWith("/")) {
   1.409 +                    r = r.substring(1);
   1.410 +                }
   1.411 +            }
   1.412 +            String[] replace = {};
   1.413 +            if (r.endsWith(".html")) {
   1.414 +                response.setContentType("text/html");
   1.415 +                LOG.info("Content type text/html");
   1.416 +                replace = args;
   1.417 +            }
   1.418 +            if (r.endsWith(".xhtml")) {
   1.419 +                response.setContentType("application/xhtml+xml");
   1.420 +                LOG.info("Content type application/xhtml+xml");
   1.421 +                replace = args;
   1.422 +            }
   1.423 +            OutputStream os = response.getOutputStream();
   1.424 +            try (InputStream is = res.get(r)) {
   1.425 +                copyStream(is, os, request.getRequestURL().toString(), replace);
   1.426 +            } catch (IOException ex) {
   1.427 +                response.setDetailMessage(ex.getLocalizedMessage());
   1.428 +                response.setError();
   1.429 +                response.setStatus(404);
   1.430 +            }
   1.431 +        }
   1.432 +    }
   1.433 +
   1.434 +    private static class VM extends HttpHandler {
   1.435 +        private final Res loader;
   1.436 +
   1.437 +        public VM(Res loader) {
   1.438 +            this.loader = loader;
   1.439 +        }
   1.440 +
   1.441 +        @Override
   1.442 +        public void service(Request request, Response response) throws Exception {
   1.443 +            response.setCharacterEncoding("UTF-8");
   1.444 +            response.setContentType("text/javascript");
   1.445 +            Bck2Brwsr.generate(response.getWriter(), loader);
   1.446 +        }
   1.447 +    }
   1.448 +    private static class VMInit extends HttpHandler {
   1.449 +        public VMInit() {
   1.450 +        }
   1.451 +
   1.452 +        @Override
   1.453 +        public void service(Request request, Response response) throws Exception {
   1.454 +            response.setCharacterEncoding("UTF-8");
   1.455 +            response.setContentType("text/javascript");
   1.456 +            response.getWriter().append(
   1.457 +                "function ldCls(res) {\n"
   1.458 +                + "  var request = new XMLHttpRequest();\n"
   1.459 +                + "  request.open('GET', '/classes/' + res, false);\n"
   1.460 +                + "  request.send();\n"
   1.461 +                + "  var arr = eval('(' + request.responseText + ')');\n"
   1.462 +                + "  return arr;\n"
   1.463 +                + "}\n"
   1.464 +                + "var vm = new bck2brwsr(ldCls);\n");
   1.465 +        }
   1.466 +    }
   1.467 +
   1.468 +    private static class Classes extends HttpHandler {
   1.469 +        private final Res loader;
   1.470 +
   1.471 +        public Classes(Res loader) {
   1.472 +            this.loader = loader;
   1.473 +        }
   1.474 +
   1.475 +        @Override
   1.476 +        public void service(Request request, Response response) throws Exception {
   1.477 +            String res = request.getHttpHandlerPath();
   1.478 +            if (res.startsWith("/")) {
   1.479 +                res = res.substring(1);
   1.480 +            }
   1.481 +            try (InputStream is = loader.get(res)) {
   1.482 +                response.setContentType("text/javascript");
   1.483 +                Writer w = response.getWriter();
   1.484 +                w.append("[");
   1.485 +                for (int i = 0;; i++) {
   1.486 +                    int b = is.read();
   1.487 +                    if (b == -1) {
   1.488 +                        break;
   1.489 +                    }
   1.490 +                    if (i > 0) {
   1.491 +                        w.append(", ");
   1.492 +                    }
   1.493 +                    if (i % 20 == 0) {
   1.494 +                        w.write("\n");
   1.495 +                    }
   1.496 +                    if (b > 127) {
   1.497 +                        b = b - 256;
   1.498 +                    }
   1.499 +                    w.append(Integer.toString(b));
   1.500 +                }
   1.501 +                w.append("\n]");
   1.502 +            } catch (IOException ex) {
   1.503 +                response.setError();
   1.504 +                response.setDetailMessage(ex.getMessage());
   1.505 +            }
   1.506 +        }
   1.507 +    }
   1.508 +}