dew/src/main/java/org/apidesign/bck2brwsr/launcher/Bck2BrwsrLauncher.java
branchdew
changeset 546 79e5a4aae48d
parent 545 29b8e1b87fad
child 547 36ec0a7616d0
     1.1 --- a/dew/src/main/java/org/apidesign/bck2brwsr/launcher/Bck2BrwsrLauncher.java	Wed Jan 23 13:56:54 2013 +0100
     1.2 +++ /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.3 @@ -1,505 +0,0 @@
     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 -}