launcher/fx/src/main/java/org/apidesign/bck2brwsr/launcher/BaseHTTPLauncher.java
author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
Tue, 26 Apr 2016 06:03:37 +0200
changeset 1939 401a4f03875b
parent 1927 cff680298793
child 1946 bafdddd2a0cf
permissions -rw-r--r--
Strict check when using precompiled version of an OSGi library
jaroslav@1033
     1
/**
jaroslav@1033
     2
 * Back 2 Browser Bytecode Translator
jaroslav@1787
     3
 * Copyright (C) 2012-2015 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
jaroslav@1033
     4
 *
jaroslav@1033
     5
 * This program is free software: you can redistribute it and/or modify
jaroslav@1033
     6
 * it under the terms of the GNU General Public License as published by
jaroslav@1033
     7
 * the Free Software Foundation, version 2 of the License.
jaroslav@1033
     8
 *
jaroslav@1033
     9
 * This program is distributed in the hope that it will be useful,
jaroslav@1033
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
jaroslav@1033
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
jaroslav@1033
    12
 * GNU General Public License for more details.
jaroslav@1033
    13
 *
jaroslav@1033
    14
 * You should have received a copy of the GNU General Public License
jaroslav@1033
    15
 * along with this program. Look for COPYING file in the top folder.
jaroslav@1033
    16
 * If not, see http://opensource.org/licenses/GPL-2.0.
jaroslav@1033
    17
 */
jaroslav@1033
    18
package org.apidesign.bck2brwsr.launcher;
jaroslav@1033
    19
jaroslav@1188
    20
import java.io.ByteArrayInputStream;
jaroslav@1249
    21
import java.io.ByteArrayOutputStream;
jaroslav@1033
    22
import java.io.Closeable;
jaroslav@1033
    23
import java.io.File;
jaroslav@1850
    24
import java.io.Flushable;
jaroslav@1033
    25
import java.io.IOException;
jaroslav@1033
    26
import java.io.InputStream;
jaroslav@1033
    27
import java.io.InterruptedIOException;
jaroslav@1033
    28
import java.io.OutputStream;
jaroslav@1088
    29
import java.io.Reader;
jaroslav@1033
    30
import java.io.UnsupportedEncodingException;
jaroslav@1033
    31
import java.io.Writer;
jaroslav@1489
    32
import java.net.JarURLConnection;
jaroslav@1033
    33
import java.net.URI;
jaroslav@1033
    34
import java.net.URISyntaxException;
jaroslav@1033
    35
import java.net.URL;
jaroslav@1746
    36
import java.net.URLConnection;
jaroslav@1033
    37
import java.util.ArrayList;
jaroslav@1033
    38
import java.util.Arrays;
jaroslav@1033
    39
import java.util.Enumeration;
jaroslav@1916
    40
import java.util.HashMap;
jaroslav@1530
    41
import java.util.HashSet;
jaroslav@1033
    42
import java.util.LinkedHashSet;
jaroslav@1033
    43
import java.util.List;
jaroslav@1746
    44
import java.util.Map;
jaroslav@1746
    45
import java.util.Properties;
jaroslav@1033
    46
import java.util.Set;
jaroslav@1033
    47
import java.util.concurrent.BlockingQueue;
jaroslav@1076
    48
import java.util.concurrent.Callable;
jaroslav@1033
    49
import java.util.concurrent.CountDownLatch;
jaroslav@1033
    50
import java.util.concurrent.LinkedBlockingQueue;
jaroslav@1033
    51
import java.util.concurrent.TimeUnit;
jaroslav@1746
    52
import java.util.jar.Attributes;
jaroslav@1746
    53
import java.util.jar.Manifest;
jaroslav@1805
    54
import java.util.logging.ConsoleHandler;
jaroslav@1033
    55
import java.util.logging.Level;
jaroslav@1033
    56
import java.util.logging.Logger;
jaroslav@1033
    57
import org.apidesign.bck2brwsr.launcher.InvocationContext.Resource;
jaroslav@1033
    58
import org.glassfish.grizzly.PortRange;
jaroslav@1033
    59
import org.glassfish.grizzly.http.server.HttpHandler;
jaroslav@1033
    60
import org.glassfish.grizzly.http.server.HttpServer;
jaroslav@1033
    61
import org.glassfish.grizzly.http.server.NetworkListener;
jaroslav@1033
    62
import org.glassfish.grizzly.http.server.Request;
jaroslav@1033
    63
import org.glassfish.grizzly.http.server.Response;
jaroslav@1033
    64
import org.glassfish.grizzly.http.server.ServerConfiguration;
jaroslav@1271
    65
import org.glassfish.grizzly.http.server.StaticHttpHandler;
jaroslav@1751
    66
import org.glassfish.grizzly.http.util.Header;
jaroslav@1033
    67
import org.glassfish.grizzly.http.util.HttpStatus;
jaroslav@1249
    68
import org.glassfish.grizzly.websockets.WebSocket;
jaroslav@1249
    69
import org.glassfish.grizzly.websockets.WebSocketAddOn;
jaroslav@1249
    70
import org.glassfish.grizzly.websockets.WebSocketApplication;
jaroslav@1249
    71
import org.glassfish.grizzly.websockets.WebSocketEngine;
jaroslav@1033
    72
jaroslav@1033
    73
/**
jaroslav@1033
    74
 * Lightweight server to launch Bck2Brwsr applications and tests.
jaroslav@1834
    75
 * Supports execution in native browser as well as Java's internal
jaroslav@1033
    76
 * execution engine.
jaroslav@1033
    77
 */
jaroslav@1850
    78
abstract class BaseHTTPLauncher extends Launcher implements Flushable, Closeable, Callable<HttpServer> {
jaroslav@1088
    79
    static final Logger LOG = Logger.getLogger(BaseHTTPLauncher.class.getName());
jaroslav@1033
    80
    private static final InvocationContext END = new InvocationContext(null, null, null);
jaroslav@1165
    81
    private final Set<ClassLoader> loaders = new LinkedHashSet<ClassLoader>();
jaroslav@1165
    82
    private final BlockingQueue<InvocationContext> methods = new LinkedBlockingQueue<InvocationContext>();
jaroslav@1033
    83
    private long timeOut;
jaroslav@1033
    84
    private final Res resources = new Res();
jaroslav@1033
    85
    private final String cmd;
jaroslav@1033
    86
    private Object[] brwsr;
jaroslav@1033
    87
    private HttpServer server;
jaroslav@1033
    88
    private CountDownLatch wait;
jaroslav@1850
    89
    private Thread flushing;
jaroslav@1927
    90
    private String rootPage;
jaroslav@1834
    91
jaroslav@1041
    92
    public BaseHTTPLauncher(String cmd) {
jaroslav@1033
    93
        this.cmd = cmd;
jaroslav@1043
    94
        addClassLoader(BaseHTTPLauncher.class.getClassLoader());
jaroslav@1033
    95
        setTimeout(180000);
jaroslav@1033
    96
    }
jaroslav@1834
    97
jaroslav@1033
    98
    @Override
jaroslav@1033
    99
    InvocationContext runMethod(InvocationContext c) throws IOException {
jaroslav@1033
   100
        loaders.add(c.clazz.getClassLoader());
jaroslav@1033
   101
        methods.add(c);
jaroslav@1033
   102
        try {
jaroslav@1033
   103
            c.await(timeOut);
jaroslav@1033
   104
        } catch (InterruptedException ex) {
jaroslav@1033
   105
            throw new IOException(ex);
jaroslav@1033
   106
        }
jaroslav@1033
   107
        return c;
jaroslav@1033
   108
    }
jaroslav@1834
   109
jaroslav@1033
   110
    public void setTimeout(long ms) {
jaroslav@1033
   111
        timeOut = ms;
jaroslav@1033
   112
    }
jaroslav@1834
   113
jaroslav@1033
   114
    public void addClassLoader(ClassLoader url) {
jaroslav@1033
   115
        this.loaders.add(url);
jaroslav@1033
   116
    }
jaroslav@1834
   117
jaroslav@1041
   118
    ClassLoader[] loaders() {
jaroslav@1041
   119
        return loaders.toArray(new ClassLoader[loaders.size()]);
jaroslav@1041
   120
    }
jaroslav@1033
   121
jaroslav@1927
   122
    @Override
jaroslav@1927
   123
    void rootPage(String startpage) {
jaroslav@1927
   124
        this.rootPage = startpage;
jaroslav@1927
   125
    }
jaroslav@1927
   126
jaroslav@1927
   127
    @Override
jaroslav@1033
   128
    public void showURL(String startpage) throws IOException {
jaroslav@1033
   129
        if (!startpage.startsWith("/")) {
jaroslav@1033
   130
            startpage = "/" + startpage;
jaroslav@1033
   131
        }
jaroslav@1926
   132
        HttpServer s = initServer(".", true, "", false);
jaroslav@1033
   133
        int last = startpage.lastIndexOf('/');
jaroslav@1033
   134
        String prefix = startpage.substring(0, last);
jaroslav@1033
   135
        String simpleName = startpage.substring(last);
jaroslav@1033
   136
        s.getServerConfiguration().addHttpHandler(new SubTree(resources, prefix), "/");
jaroslav@1076
   137
        server = s;
jaroslav@1033
   138
        try {
jaroslav@1033
   139
            launchServerAndBrwsr(s, simpleName);
jaroslav@1165
   140
        } catch (Exception ex) {
jaroslav@1033
   141
            throw new IOException(ex);
jaroslav@1033
   142
        }
jaroslav@1033
   143
    }
jaroslav@1033
   144
jaroslav@1273
   145
    void showDirectory(File dir, String startpage, boolean addClasses) throws IOException {
jaroslav@1033
   146
        if (!startpage.startsWith("/")) {
jaroslav@1033
   147
            startpage = "/" + startpage;
jaroslav@1033
   148
        }
jaroslav@1615
   149
        String prefix = null;
jaroslav@1615
   150
        if (!new File(dir, "bck2brwsr.js").exists()) {
jaroslav@1615
   151
            int last = startpage.lastIndexOf('/');
jaroslav@1615
   152
            if (last >= 0) {
jaroslav@1615
   153
                prefix = startpage.substring(0, last);
jaroslav@1615
   154
            }
jaroslav@1271
   155
        }
jaroslav@1926
   156
        HttpServer s = initServer(dir.getPath(), addClasses, prefix, false);
jaroslav@1033
   157
        try {
jaroslav@1033
   158
            launchServerAndBrwsr(s, startpage);
jaroslav@1165
   159
        } catch (Exception ex) {
jaroslav@1033
   160
            throw new IOException(ex);
jaroslav@1033
   161
        }
jaroslav@1033
   162
    }
jaroslav@1033
   163
jaroslav@1033
   164
    @Override
jaroslav@1033
   165
    public void initialize() throws IOException {
jaroslav@1033
   166
        try {
jaroslav@1033
   167
            executeInBrowser();
jaroslav@1033
   168
        } catch (InterruptedException ex) {
jaroslav@1033
   169
            final InterruptedIOException iio = new InterruptedIOException(ex.getMessage());
jaroslav@1033
   170
            iio.initCause(ex);
jaroslav@1033
   171
            throw iio;
jaroslav@1033
   172
        } catch (Exception ex) {
jaroslav@1033
   173
            if (ex instanceof IOException) {
jaroslav@1033
   174
                throw (IOException)ex;
jaroslav@1033
   175
            }
jaroslav@1033
   176
            if (ex instanceof RuntimeException) {
jaroslav@1033
   177
                throw (RuntimeException)ex;
jaroslav@1033
   178
            }
jaroslav@1033
   179
            throw new IOException(ex);
jaroslav@1033
   180
        }
jaroslav@1033
   181
    }
jaroslav@1834
   182
jaroslav@1926
   183
    private HttpServer initServer(String path, boolean addClasses, String vmPrefix, boolean unitTests) throws IOException {
jaroslav@1271
   184
        HttpServer s = HttpServer.createSimpleServer(null, new PortRange(8080, 65535));
jaroslav@1253
   185
        /*
jaroslav@1092
   186
        ThreadPoolConfig fewThreads = ThreadPoolConfig.defaultConfig().copy().
jaroslav@1092
   187
            setPoolName("Fx/Bck2 Brwsr").
jaroslav@1253
   188
            setCorePoolSize(3).
jaroslav@1092
   189
            setMaxPoolSize(5);
jaroslav@1092
   190
        ThreadPoolConfig oneKernel = ThreadPoolConfig.defaultConfig().copy().
jaroslav@1092
   191
            setPoolName("Kernel Fx/Bck2").
jaroslav@1253
   192
            setCorePoolSize(3).
jaroslav@1092
   193
            setMaxPoolSize(3);
jaroslav@1092
   194
        for (NetworkListener nl : s.getListeners()) {
jaroslav@1092
   195
            nl.getTransport().setWorkerThreadPoolConfig(fewThreads);
jaroslav@1092
   196
            nl.getTransport().setKernelThreadPoolConfig(oneKernel);
jaroslav@1092
   197
        }
jaroslav@1834
   198
*/
jaroslav@1033
   199
        final ServerConfiguration conf = s.getServerConfiguration();
jaroslav@1926
   200
        VMAndPages vm = new VMAndPages(unitTests);
jaroslav@1271
   201
        conf.addHttpHandler(vm, "/");
jaroslav@1271
   202
        if (vmPrefix != null) {
jaroslav@1271
   203
            vm.registerVM(vmPrefix + "/bck2brwsr.js");
jaroslav@1271
   204
        }
jaroslav@1271
   205
        if (path != null) {
jaroslav@1271
   206
            vm.addDocRoot(path);
jaroslav@1271
   207
        }
jaroslav@1033
   208
        if (addClasses) {
jaroslav@1033
   209
            conf.addHttpHandler(new Classes(resources), "/classes/");
jaroslav@1033
   210
        }
jaroslav@1927
   211
        if (rootPage != null) {
jaroslav@1927
   212
            int last = rootPage.lastIndexOf('/');
jaroslav@1927
   213
            String prefix = rootPage.substring(0, last);
jaroslav@1927
   214
            String page = rootPage.substring(last);
jaroslav@1927
   215
            s.getServerConfiguration().addHttpHandler(new SubTree("/pages" + page, resources, prefix), "/pages/");
jaroslav@1927
   216
        }
jaroslav@1253
   217
        final WebSocketAddOn addon = new WebSocketAddOn();
jaroslav@1253
   218
        for (NetworkListener listener : s.getListeners()) {
jaroslav@1253
   219
            listener.registerAddOn(addon);
jaroslav@1253
   220
        }
jaroslav@1805
   221
        Logger l = Logger.getLogger("org.glassfish.grizzly.http.server.HttpHandler");
jaroslav@1805
   222
        l.setLevel(Level.FINE);
jaroslav@1805
   223
        l.setUseParentHandlers(false);
jaroslav@1805
   224
        ConsoleHandler ch = new ConsoleHandler();
jaroslav@1805
   225
        ch.setLevel(Level.ALL);
jaroslav@1805
   226
        l.addHandler(ch);
jaroslav@1033
   227
        return s;
jaroslav@1033
   228
    }
jaroslav@1834
   229
jaroslav@1751
   230
    private static int resourcesCount;
jaroslav@1033
   231
    private void executeInBrowser() throws InterruptedException, URISyntaxException, IOException {
jaroslav@1033
   232
        wait = new CountDownLatch(1);
jaroslav@1926
   233
        server = initServer(".", true, "", true);
jaroslav@1033
   234
        final ServerConfiguration conf = server.getServerConfiguration();
jaroslav@1834
   235
jaroslav@1188
   236
        class DynamicResourceHandler extends HttpHandler {
jaroslav@1033
   237
            private final InvocationContext ic;
jaroslav@1235
   238
            DynamicResourceHandler delegate;
jaroslav@1033
   239
            public DynamicResourceHandler(InvocationContext ic) {
jaroslav@1033
   240
                this.ic = ic;
jaroslav@1033
   241
                for (Resource r : ic.resources) {
jaroslav@1033
   242
                    conf.addHttpHandler(this, r.httpPath);
jaroslav@1033
   243
                }
jaroslav@1033
   244
            }
jaroslav@1033
   245
jaroslav@1235
   246
            public void close(DynamicResourceHandler del) {
jaroslav@1033
   247
                conf.removeHttpHandler(this);
jaroslav@1235
   248
                delegate = del;
jaroslav@1033
   249
            }
jaroslav@1834
   250
jaroslav@1033
   251
            @Override
jaroslav@1033
   252
            public void service(Request request, Response response) throws Exception {
jaroslav@1235
   253
                if (delegate != null) {
jaroslav@1235
   254
                    delegate.service(request, response);
jaroslav@1235
   255
                    return;
jaroslav@1235
   256
                }
jaroslav@1834
   257
jaroslav@1188
   258
                if ("/dynamic".equals(request.getRequestURI())) {
jaroslav@1249
   259
                    boolean webSocket = false;
jaroslav@1188
   260
                    String mimeType = request.getParameter("mimeType");
jaroslav@1188
   261
                    List<String> params = new ArrayList<String>();
jaroslav@1188
   262
                    for (int i = 0; ; i++) {
jaroslav@1188
   263
                        String p = request.getParameter("param" + i);
jaroslav@1188
   264
                        if (p == null) {
jaroslav@1188
   265
                            break;
jaroslav@1188
   266
                        }
jaroslav@1188
   267
                        params.add(p);
jaroslav@1249
   268
                        if ("protocol:ws".equals(p)) {
jaroslav@1249
   269
                            webSocket = true;
jaroslav@1249
   270
                            continue;
jaroslav@1249
   271
                        }                    }
jaroslav@1188
   272
                    final String cnt = request.getParameter("content");
jaroslav@1188
   273
                    String mangle = cnt.replace("%20", " ").replace("%0A", "\n");
jaroslav@1188
   274
                    ByteArrayInputStream is = new ByteArrayInputStream(mangle.getBytes("UTF-8"));
jaroslav@1249
   275
                    URI url;
jaroslav@1249
   276
                    final Resource res = new Resource(is, mimeType, "/dynamic/res" + ++resourcesCount, params.toArray(new String[params.size()]));
jaroslav@1249
   277
                    if (webSocket) {
jaroslav@1249
   278
                        url = registerWebSocket(res);
jaroslav@1249
   279
                    } else {
jaroslav@1249
   280
                        url = registerResource(res);
jaroslav@1249
   281
                    }
jaroslav@1751
   282
                    response.setHeader(Header.CacheControl, "no-cache");
jaroslav@1751
   283
                    response.setHeader(Header.Pragma, "no-cache");
jaroslav@1188
   284
                    response.getWriter().write(url.toString());
jaroslav@1188
   285
                    response.getWriter().write("\n");
jaroslav@1188
   286
                    return;
jaroslav@1188
   287
                }
jaroslav@1834
   288
jaroslav@1033
   289
                for (Resource r : ic.resources) {
jaroslav@1033
   290
                    if (r.httpPath.equals(request.getRequestURI())) {
jaroslav@1033
   291
                        LOG.log(Level.INFO, "Serving HttpResource for {0}", request.getRequestURI());
jaroslav@1033
   292
                        response.setContentType(r.httpType);
jaroslav@1751
   293
                        response.setHeader(Header.CacheControl, "no-cache");
jaroslav@1751
   294
                        response.setHeader(Header.Pragma, "no-cache");
jaroslav@1033
   295
                        r.httpContent.reset();
jaroslav@1033
   296
                        String[] params = null;
jaroslav@1033
   297
                        if (r.parameters.length != 0) {
jaroslav@1033
   298
                            params = new String[r.parameters.length];
jaroslav@1033
   299
                            for (int i = 0; i < r.parameters.length; i++) {
jaroslav@1033
   300
                                params[i] = request.getParameter(r.parameters[i]);
jaroslav@1088
   301
                                if (params[i] == null) {
jaroslav@1088
   302
                                    if ("http.method".equals(r.parameters[i])) {
jaroslav@1088
   303
                                        params[i] = request.getMethod().toString();
jaroslav@1088
   304
                                    } else if ("http.requestBody".equals(r.parameters[i])) {
jaroslav@1088
   305
                                        Reader rdr = request.getReader();
jaroslav@1088
   306
                                        StringBuilder sb = new StringBuilder();
jaroslav@1088
   307
                                        for (;;) {
jaroslav@1088
   308
                                            int ch = rdr.read();
jaroslav@1088
   309
                                            if (ch == -1) {
jaroslav@1088
   310
                                                break;
jaroslav@1088
   311
                                            }
jaroslav@1088
   312
                                            sb.append((char)ch);
jaroslav@1088
   313
                                        }
jaroslav@1088
   314
                                        params[i] = sb.toString();
jaroslav@1834
   315
                                    } else if (r.parameters[i].startsWith("http.header.")) {
jaroslav@1834
   316
                                        params[i] = request.getHeader(r.parameters[i].substring(12));
jaroslav@1088
   317
                                    }
jaroslav@1088
   318
                                }
jaroslav@1088
   319
                                if (params[i] == null) {
jaroslav@1088
   320
                                    params[i] = "null";
jaroslav@1088
   321
                                }
jaroslav@1033
   322
                            }
jaroslav@1033
   323
                        }
jaroslav@1834
   324
jaroslav@1033
   325
                        copyStream(r.httpContent, response.getOutputStream(), null, params);
jaroslav@1033
   326
                    }
jaroslav@1033
   327
                }
jaroslav@1033
   328
            }
jaroslav@1834
   329
jaroslav@1249
   330
            private URI registerWebSocket(Resource r) {
jaroslav@1249
   331
                WebSocketEngine.getEngine().register("", r.httpPath, new WS(r));
jaroslav@1249
   332
                return pageURL("ws", server, r.httpPath);
jaroslav@1249
   333
            }
jaroslav@1186
   334
jaroslav@1188
   335
            private URI registerResource(Resource r) {
jaroslav@1186
   336
                if (!ic.resources.contains(r)) {
jaroslav@1186
   337
                    ic.resources.add(r);
jaroslav@1186
   338
                    conf.addHttpHandler(this, r.httpPath);
jaroslav@1186
   339
                }
jaroslav@1249
   340
                return pageURL("http", server, r.httpPath);
jaroslav@1186
   341
            }
jaroslav@1033
   342
        }
jaroslav@1834
   343
jaroslav@1088
   344
        conf.addHttpHandler(new Page(resources, harnessResource()), "/execute");
jaroslav@1834
   345
jaroslav@1033
   346
        conf.addHttpHandler(new HttpHandler() {
jaroslav@1033
   347
            int cnt;
jaroslav@1165
   348
            List<InvocationContext> cases = new ArrayList<InvocationContext>();
jaroslav@1033
   349
            DynamicResourceHandler prev;
jaroslav@1033
   350
            @Override
jaroslav@1033
   351
            public void service(Request request, Response response) throws Exception {
jaroslav@1033
   352
                String id = request.getParameter("request");
jaroslav@1033
   353
                String value = request.getParameter("result");
jaroslav@1860
   354
                String timeText = request.getParameter("time");
jaroslav@1033
   355
                if (value != null && value.indexOf((char)0xC5) != -1) {
jaroslav@1033
   356
                    value = toUTF8(value);
jaroslav@1033
   357
                }
jaroslav@1860
   358
                int time;
jaroslav@1860
   359
                if (timeText != null) {
jaroslav@1860
   360
                    try {
jaroslav@1860
   361
                        time = (int) Double.parseDouble(timeText);
jaroslav@1860
   362
                    } catch (NumberFormatException numberFormatException) {
jaroslav@1860
   363
                        time = 0;
jaroslav@1860
   364
                    }
jaroslav@1860
   365
                } else {
jaroslav@1860
   366
                    time = 0;
jaroslav@1860
   367
                }
jaroslav@1834
   368
jaroslav@1033
   369
                InvocationContext mi = null;
jaroslav@1033
   370
                int caseNmbr = -1;
jaroslav@1834
   371
jaroslav@1033
   372
                if (id != null && value != null) {
jaroslav@1033
   373
                    LOG.log(Level.INFO, "Received result for case {0} = {1}", new Object[]{id, value});
jaroslav@1033
   374
                    value = decodeURL(value);
jaroslav@1033
   375
                    int indx = Integer.parseInt(id);
jaroslav@1860
   376
                    cases.get(indx).result(value, time, null);
jaroslav@1033
   377
                    if (++indx < cases.size()) {
jaroslav@1033
   378
                        mi = cases.get(indx);
jaroslav@1033
   379
                        LOG.log(Level.INFO, "Re-executing case {0}", indx);
jaroslav@1033
   380
                        caseNmbr = indx;
jaroslav@1033
   381
                    }
jaroslav@1033
   382
                } else {
jaroslav@1033
   383
                    if (!cases.isEmpty()) {
jaroslav@1033
   384
                        LOG.info("Re-executing test cases");
jaroslav@1033
   385
                        mi = cases.get(0);
jaroslav@1033
   386
                        caseNmbr = 0;
jaroslav@1033
   387
                    }
jaroslav@1033
   388
                }
jaroslav@1834
   389
jaroslav@1033
   390
                if (mi == null) {
jaroslav@1033
   391
                    mi = methods.take();
jaroslav@1033
   392
                    caseNmbr = cnt++;
jaroslav@1033
   393
                }
jaroslav@1908
   394
                final Writer w = response.getWriter();
jaroslav@1033
   395
                if (mi == END) {
jaroslav@1908
   396
                    w.write("");
jaroslav@1033
   397
                    wait.countDown();
jaroslav@1033
   398
                    cnt = 0;
jaroslav@1033
   399
                    LOG.log(Level.INFO, "End of data reached. Exiting.");
jaroslav@1033
   400
                    return;
jaroslav@1033
   401
                }
jaroslav@1235
   402
                final DynamicResourceHandler newRH = new DynamicResourceHandler(mi);
jaroslav@1235
   403
                if (prev != null) {
jaroslav@1235
   404
                    prev.close(newRH);
jaroslav@1235
   405
                }
jaroslav@1235
   406
                prev = newRH;
jaroslav@1188
   407
                conf.addHttpHandler(prev, "/dynamic");
jaroslav@1834
   408
jaroslav@1033
   409
                cases.add(mi);
jaroslav@1033
   410
                final String cn = mi.clazz.getName();
jaroslav@1033
   411
                final String mn = mi.methodName;
jaroslav@1033
   412
                LOG.log(Level.INFO, "Request for {0} case. Sending {1}.{2}", new Object[]{caseNmbr, cn, mn});
jaroslav@1908
   413
                w.write("{"
jaroslav@1033
   414
                    + "className: '" + cn + "', "
jaroslav@1033
   415
                    + "methodName: '" + mn + "', "
jaroslav@1033
   416
                    + "request: " + caseNmbr
jaroslav@1033
   417
                );
jaroslav@1908
   418
                if (mi.args != null) {
jaroslav@1908
   419
                    w.write(", args: [");
jaroslav@1908
   420
                    String sep = "";
jaroslav@1908
   421
                    for (String a : mi.args) {
jaroslav@1908
   422
                        w.write(sep);
jaroslav@1908
   423
                        w.write("'");
jaroslav@1908
   424
                        w.write(a);
jaroslav@1908
   425
                        w.write("'");
jaroslav@1908
   426
                        sep = ", ";
jaroslav@1908
   427
                    }
jaroslav@1908
   428
                    w.write("]");
jaroslav@1908
   429
                }
jaroslav@1033
   430
                if (mi.html != null) {
jaroslav@1908
   431
                    w.write(", html: '");
jaroslav@1908
   432
                    w.write(encodeJSON(mi.html));
jaroslav@1908
   433
                    w.write("'");
jaroslav@1033
   434
                }
jaroslav@1908
   435
                w.write("}");
jaroslav@1033
   436
            }
jaroslav@1033
   437
        }, "/data");
jaroslav@1033
   438
jaroslav@1927
   439
        String page = "/execute";
jaroslav@1927
   440
        if (rootPage != null) {
jaroslav@1927
   441
            int last = rootPage.lastIndexOf('/');
jaroslav@1927
   442
            page = "/pages" + rootPage.substring(last);
jaroslav@1927
   443
        }
jaroslav@1927
   444
jaroslav@1927
   445
        this.brwsr = launchServerAndBrwsr(server, page);
jaroslav@1033
   446
    }
jaroslav@1834
   447
jaroslav@1033
   448
    private static String encodeJSON(String in) {
jaroslav@1033
   449
        StringBuilder sb = new StringBuilder();
jaroslav@1033
   450
        for (int i = 0; i < in.length(); i++) {
jaroslav@1033
   451
            char ch = in.charAt(i);
jaroslav@1033
   452
            if (ch < 32 || ch == '\'' || ch == '"') {
jaroslav@1033
   453
                sb.append("\\u");
jaroslav@1033
   454
                String hs = "0000" + Integer.toHexString(ch);
jaroslav@1033
   455
                hs = hs.substring(hs.length() - 4);
jaroslav@1033
   456
                sb.append(hs);
jaroslav@1033
   457
            } else {
jaroslav@1033
   458
                sb.append(ch);
jaroslav@1033
   459
            }
jaroslav@1033
   460
        }
jaroslav@1033
   461
        return sb.toString();
jaroslav@1033
   462
    }
jaroslav@1834
   463
jaroslav@1033
   464
    @Override
jaroslav@1033
   465
    public void shutdown() throws IOException {
jaroslav@1850
   466
        synchronized (this) {
jaroslav@1850
   467
            if (flushing != null) {
jaroslav@1850
   468
                flushing.interrupt();
jaroslav@1850
   469
                flushing = null;
jaroslav@1850
   470
            }
jaroslav@1850
   471
        }
jaroslav@1033
   472
        methods.offer(END);
jaroslav@1033
   473
        for (;;) {
jaroslav@1033
   474
            int prev = methods.size();
jaroslav@1033
   475
            try {
jaroslav@1033
   476
                if (wait != null && wait.await(timeOut, TimeUnit.MILLISECONDS)) {
jaroslav@1033
   477
                    break;
jaroslav@1033
   478
                }
jaroslav@1033
   479
            } catch (InterruptedException ex) {
jaroslav@1033
   480
                throw new IOException(ex);
jaroslav@1033
   481
            }
jaroslav@1033
   482
            if (prev == methods.size()) {
jaroslav@1033
   483
                LOG.log(
jaroslav@1834
   484
                    Level.WARNING,
jaroslav@1834
   485
                    "Timeout and no test has been executed meanwhile (at {0}). Giving up.",
jaroslav@1033
   486
                    methods.size()
jaroslav@1033
   487
                );
jaroslav@1033
   488
                break;
jaroslav@1033
   489
            }
jaroslav@1834
   490
            LOG.log(Level.INFO,
jaroslav@1834
   491
                "Timeout, but tests got from {0} to {1}. Trying again.",
jaroslav@1033
   492
                new Object[]{prev, methods.size()}
jaroslav@1033
   493
            );
jaroslav@1033
   494
        }
jaroslav@1033
   495
        stopServerAndBrwsr(server, brwsr);
jaroslav@1033
   496
    }
jaroslav@1834
   497
jaroslav@1033
   498
    static void copyStream(InputStream is, OutputStream os, String baseURL, String... params) throws IOException {
jaroslav@1033
   499
        for (;;) {
jaroslav@1033
   500
            int ch = is.read();
jaroslav@1033
   501
            if (ch == -1) {
jaroslav@1033
   502
                break;
jaroslav@1033
   503
            }
jaroslav@1033
   504
            if (ch == '$' && params.length > 0) {
jaroslav@1033
   505
                int cnt = is.read() - '0';
jaroslav@1033
   506
                if (baseURL != null && cnt == 'U' - '0') {
jaroslav@1033
   507
                    os.write(baseURL.getBytes("UTF-8"));
jaroslav@1033
   508
                } else {
jaroslav@1033
   509
                    if (cnt >= 0 && cnt < params.length) {
jaroslav@1033
   510
                        os.write(params[cnt].getBytes("UTF-8"));
jaroslav@1033
   511
                    } else {
jaroslav@1033
   512
                        os.write('$');
jaroslav@1033
   513
                        os.write(cnt + '0');
jaroslav@1033
   514
                    }
jaroslav@1033
   515
                }
jaroslav@1033
   516
            } else {
jaroslav@1033
   517
                os.write(ch);
jaroslav@1033
   518
            }
jaroslav@1033
   519
        }
jaroslav@1033
   520
    }
jaroslav@1033
   521
jaroslav@1033
   522
    private Object[] launchServerAndBrwsr(HttpServer server, final String page) throws IOException, URISyntaxException, InterruptedException {
jaroslav@1033
   523
        server.start();
jaroslav@1249
   524
        URI uri = pageURL("http", server, page);
jaroslav@1041
   525
        return showBrwsr(uri);
jaroslav@1033
   526
    }
jaroslav@1033
   527
    private static String toUTF8(String value) throws UnsupportedEncodingException {
jaroslav@1033
   528
        byte[] arr = new byte[value.length()];
jaroslav@1033
   529
        for (int i = 0; i < arr.length; i++) {
jaroslav@1033
   530
            arr[i] = (byte)value.charAt(i);
jaroslav@1033
   531
        }
jaroslav@1033
   532
        return new String(arr, "UTF-8");
jaroslav@1033
   533
    }
jaroslav@1834
   534
jaroslav@1033
   535
    private static String decodeURL(String s) {
jaroslav@1033
   536
        for (;;) {
jaroslav@1033
   537
            int pos = s.indexOf('%');
jaroslav@1033
   538
            if (pos == -1) {
jaroslav@1033
   539
                return s;
jaroslav@1033
   540
            }
jaroslav@1033
   541
            int i = Integer.parseInt(s.substring(pos + 1, pos + 2), 16);
jaroslav@1033
   542
            s = s.substring(0, pos) + (char)i + s.substring(pos + 2);
jaroslav@1033
   543
        }
jaroslav@1033
   544
    }
jaroslav@1834
   545
jaroslav@1033
   546
    private void stopServerAndBrwsr(HttpServer server, Object[] brwsr) throws IOException {
jaroslav@1033
   547
        if (brwsr == null) {
jaroslav@1033
   548
            return;
jaroslav@1033
   549
        }
jaroslav@1033
   550
        Process process = (Process)brwsr[0];
jaroslav@1834
   551
jaroslav@1033
   552
        server.stop();
jaroslav@1033
   553
        InputStream stdout = process.getInputStream();
jaroslav@1033
   554
        InputStream stderr = process.getErrorStream();
jaroslav@1033
   555
        drain("StdOut", stdout);
jaroslav@1033
   556
        drain("StdErr", stderr);
jaroslav@1033
   557
        process.destroy();
jaroslav@1033
   558
        int res;
jaroslav@1033
   559
        try {
jaroslav@1033
   560
            res = process.waitFor();
jaroslav@1033
   561
        } catch (InterruptedException ex) {
jaroslav@1033
   562
            throw new IOException(ex);
jaroslav@1033
   563
        }
jaroslav@1033
   564
        LOG.log(Level.INFO, "Exit code: {0}", res);
jaroslav@1033
   565
jaroslav@1033
   566
        deleteTree((File)brwsr[1]);
jaroslav@1033
   567
    }
jaroslav@1834
   568
jaroslav@1033
   569
    private static void drain(String name, InputStream is) throws IOException {
jaroslav@1033
   570
        int av = is.available();
jaroslav@1033
   571
        if (av > 0) {
jaroslav@1033
   572
            StringBuilder sb = new StringBuilder();
jaroslav@1033
   573
            sb.append("v== ").append(name).append(" ==v\n");
jaroslav@1033
   574
            while (av-- > 0) {
jaroslav@1033
   575
                sb.append((char)is.read());
jaroslav@1033
   576
            }
jaroslav@1033
   577
            sb.append("\n^== ").append(name).append(" ==^");
jaroslav@1033
   578
            LOG.log(Level.INFO, sb.toString());
jaroslav@1033
   579
        }
jaroslav@1033
   580
    }
jaroslav@1033
   581
jaroslav@1033
   582
    private void deleteTree(File file) {
jaroslav@1033
   583
        if (file == null) {
jaroslav@1033
   584
            return;
jaroslav@1033
   585
        }
jaroslav@1033
   586
        File[] arr = file.listFiles();
jaroslav@1033
   587
        if (arr != null) {
jaroslav@1033
   588
            for (File s : arr) {
jaroslav@1033
   589
                deleteTree(s);
jaroslav@1033
   590
            }
jaroslav@1033
   591
        }
jaroslav@1033
   592
        file.delete();
jaroslav@1033
   593
    }
jaroslav@1033
   594
jaroslav@1033
   595
    @Override
jaroslav@1076
   596
    public HttpServer call() throws Exception {
jaroslav@1076
   597
        return server;
jaroslav@1076
   598
    }
jaroslav@1834
   599
jaroslav@1076
   600
    @Override
jaroslav@1850
   601
    public synchronized void flush() throws IOException {
jaroslav@1850
   602
        flushing = Thread.currentThread();
jaroslav@1850
   603
        while (flushing == Thread.currentThread()) {
jaroslav@1850
   604
            try {
jaroslav@1850
   605
                wait();
jaroslav@1850
   606
            } catch (InterruptedException ex) {
jaroslav@1850
   607
                LOG.log(Level.FINE, null, ex);
jaroslav@1850
   608
            }
jaroslav@1850
   609
        }
jaroslav@1850
   610
    }
jaroslav@1850
   611
jaroslav@1850
   612
    @Override
jaroslav@1033
   613
    public void close() throws IOException {
jaroslav@1033
   614
        shutdown();
jaroslav@1033
   615
    }
jaroslav@1033
   616
jaroslav@1041
   617
    protected Object[] showBrwsr(URI uri) throws IOException {
jaroslav@1041
   618
        LOG.log(Level.INFO, "Showing {0}", uri);
jaroslav@1041
   619
        if (cmd == null) {
jaroslav@1041
   620
            try {
jaroslav@1041
   621
                LOG.log(Level.INFO, "Trying Desktop.browse on {0} {2} by {1}", new Object[] {
jaroslav@1041
   622
                    System.getProperty("java.vm.name"),
jaroslav@1041
   623
                    System.getProperty("java.vm.vendor"),
jaroslav@1041
   624
                    System.getProperty("java.vm.version"),
jaroslav@1041
   625
                });
jaroslav@1041
   626
                java.awt.Desktop.getDesktop().browse(uri);
jaroslav@1041
   627
                LOG.log(Level.INFO, "Desktop.browse successfully finished");
jaroslav@1041
   628
                return null;
jaroslav@1041
   629
            } catch (UnsupportedOperationException ex) {
jaroslav@1041
   630
                LOG.log(Level.INFO, "Desktop.browse not supported: {0}", ex.getMessage());
jaroslav@1041
   631
                LOG.log(Level.FINE, null, ex);
jaroslav@1574
   632
            } catch (IOException ex) {
jaroslav@1574
   633
                LOG.log(Level.INFO, "Desktop.browse failed: {0}", ex.getMessage());
jaroslav@1574
   634
                LOG.log(Level.FINE, null, ex);
jaroslav@1041
   635
            }
jaroslav@1041
   636
        }
jaroslav@1041
   637
        {
jaroslav@1041
   638
            String cmdName = cmd == null ? "xdg-open" : cmd;
jaroslav@1834
   639
            String[] cmdArr = {
jaroslav@1041
   640
                cmdName, uri.toString()
jaroslav@1041
   641
            };
jaroslav@1041
   642
            LOG.log(Level.INFO, "Launching {0}", Arrays.toString(cmdArr));
jaroslav@1041
   643
            final Process process = Runtime.getRuntime().exec(cmdArr);
jaroslav@1041
   644
            return new Object[] { process, null };
jaroslav@1041
   645
        }
jaroslav@1041
   646
    }
jaroslav@1041
   647
jaroslav@1926
   648
    abstract void generateBck2BrwsrJS(StringBuilder sb, Res loader, String url, boolean unitTestMode) throws IOException;
jaroslav@1088
   649
    abstract String harnessResource();
jaroslav@1746
   650
    Object compileJar(URL jar, URL precompiled) throws IOException {
jaroslav@1489
   651
        return null;
jaroslav@1489
   652
    }
jaroslav@1505
   653
    String compileFromClassPath(URL f, Res loader) throws IOException {
jaroslav@1489
   654
        return null;
jaroslav@1489
   655
    }
jaroslav@1041
   656
jaroslav@1249
   657
    private static URI pageURL(String protocol, HttpServer server, final String page) {
jaroslav@1186
   658
        NetworkListener listener = server.getListeners().iterator().next();
jaroslav@1186
   659
        int port = listener.getPort();
jaroslav@1186
   660
        try {
jaroslav@1249
   661
            return new URI(protocol + "://localhost:" + port + page);
jaroslav@1186
   662
        } catch (URISyntaxException ex) {
jaroslav@1186
   663
            throw new IllegalStateException(ex);
jaroslav@1186
   664
        }
jaroslav@1186
   665
    }
jaroslav@1186
   666
jaroslav@1489
   667
    final class Res {
jaroslav@1530
   668
        private final Set<URL> ignore = new HashSet<URL>();
jaroslav@1834
   669
jaroslav@1746
   670
        Object compileJar(URL jarURL) throws IOException {
jaroslav@1746
   671
            List<String[]> libraries = new ArrayList<String[]>();
jaroslav@1916
   672
            Map<String,Object[]> osgiJars = new HashMap<String, Object[]>();
jaroslav@1746
   673
            for (ClassLoader loader : loaders) {
jaroslav@1746
   674
                Enumeration<URL> en = loader.getResources("META-INF/MANIFEST.MF");
jaroslav@1746
   675
                while (en.hasMoreElements()) {
jaroslav@1746
   676
                    URL e = en.nextElement();
jaroslav@1746
   677
                    Manifest mf = new Manifest(e.openStream());
jaroslav@1746
   678
                    for (Map.Entry<String, Attributes> entrySet : mf.getEntries().entrySet()) {
jaroslav@1746
   679
                        String key = entrySet.getKey();
jaroslav@1746
   680
                        Attributes attr = entrySet.getValue();
jaroslav@1834
   681
jaroslav@1746
   682
                        final String a = attr.getValue("Bck2BrwsrArtifactId");
jaroslav@1746
   683
                        final String g = attr.getValue("Bck2BrwsrGroupId");
jaroslav@1746
   684
                        final String v = attr.getValue("Bck2BrwsrVersion");
jaroslav@1746
   685
                        final String d = attr.getValue("Bck2BrwsrDebug");
jaroslav@1916
   686
                        final String n = attr.getValue("Bck2BrwsrName");
jaroslav@1834
   687
jaroslav@1746
   688
                        if (g != null && a != null && v != null && "true".equals(d)) {
jaroslav@1746
   689
                            libraries.add(new String[] {
jaroslav@1916
   690
                                a, g, v, key, n
jaroslav@1746
   691
                            });
jaroslav@1746
   692
                        }
jaroslav@1746
   693
                    }
jaroslav@1916
   694
                    final Attributes main = mf.getMainAttributes();
jaroslav@1916
   695
                    String symbol = main.getValue("Bundle-SymbolicName");
jaroslav@1916
   696
                    String version;
jaroslav@1916
   697
                    if (symbol == null) {
jaroslav@1916
   698
                        symbol = main.getValue("OpenIDE-Module-Name");
jaroslav@1916
   699
                        version = main.getValue("OpenIDE-Module-SpecificationVersion");
jaroslav@1916
   700
                    } else {
jaroslav@1916
   701
                        version = main.getValue("Bundle-Version");
jaroslav@1916
   702
                    }
jaroslav@1916
   703
                    if (symbol != null) {
jaroslav@1916
   704
                        osgiJars.put(symbol, new Object[] { e, version });
jaroslav@1916
   705
                    }
jaroslav@1746
   706
                }
jaroslav@1746
   707
            }
jaroslav@1746
   708
            URL precompiled = null;
jaroslav@1746
   709
            for (ClassLoader loader : loaders) {
jaroslav@1746
   710
                for (String[] lib : libraries) {
jaroslav@1746
   711
                    final String res = "META-INF/maven/" + lib[1] + "/" + lib[0] + "/pom.properties";
jaroslav@1746
   712
                    URL props = loader.getResource(res);
jaroslav@1746
   713
                    if (props != null) {
jaroslav@1746
   714
                        URLConnection c = props.openConnection();
jaroslav@1746
   715
                        Properties load = new Properties();
jaroslav@1746
   716
                        final InputStream is = c.getInputStream();
jaroslav@1746
   717
                        load.load(is);
jaroslav@1746
   718
                        is.close();
jaroslav@1746
   719
                        if (lib[2].equals(load.getProperty("version"))) {
jaroslav@1746
   720
                            if (c instanceof JarURLConnection) {
jaroslav@1746
   721
                                final URL definedInURL = ((JarURLConnection)c).getJarFileURL();
jaroslav@1746
   722
                                if (definedInURL.equals(jarURL)) {
jaroslav@1746
   723
                                    precompiled = loader.getResource(lib[3]);
jaroslav@1746
   724
                                }
jaroslav@1746
   725
                            }
jaroslav@1746
   726
                        }
jaroslav@1746
   727
                    }
jaroslav@1746
   728
                }
jaroslav@1746
   729
            }
jaroslav@1916
   730
            if (precompiled == null) {
jaroslav@1916
   731
                for (ClassLoader loader : loaders) {
jaroslav@1916
   732
                    for (String[] lib : libraries) {
jaroslav@1916
   733
                        Object[] urlVersion = osgiJars.get(lib[4]);
jaroslav@1921
   734
                        String expectVersion = lib[2];
jaroslav@1921
   735
                        if (expectVersion.endsWith("-SNAPSHOT")) {
jaroslav@1921
   736
                            expectVersion = expectVersion.substring(0, expectVersion.length() - 9);
jaroslav@1921
   737
                        }
jaroslav@1921
   738
                        if (urlVersion != null && urlVersion[1].toString().startsWith(expectVersion)) {
jaroslav@1939
   739
                            URL manifest = (URL) urlVersion[0];
jaroslav@1939
   740
                            if (manifest.openConnection() instanceof JarURLConnection) {
jaroslav@1939
   741
                                JarURLConnection jarConn = (JarURLConnection) manifest.openConnection();
jaroslav@1939
   742
                                if (jarConn.getJarFileURL().equals(jarURL)) {
jaroslav@1939
   743
                                    precompiled = loader.getResource(lib[3]);
jaroslav@1939
   744
                                }
jaroslav@1939
   745
                            }
jaroslav@1916
   746
                        }
jaroslav@1916
   747
                    }
jaroslav@1916
   748
                }
jaroslav@1916
   749
            }
jaroslav@1746
   750
            Object ret = BaseHTTPLauncher.this.compileJar(jarURL, precompiled);
jaroslav@1530
   751
            ignore.add(jarURL);
jaroslav@1530
   752
            return ret;
jaroslav@1489
   753
        }
jaroslav@1505
   754
        String compileFromClassPath(URL f) throws IOException {
jaroslav@1504
   755
            return BaseHTTPLauncher.this.compileFromClassPath(f, this);
jaroslav@1489
   756
        }
jaroslav@1513
   757
        public URL get(String resource, int skip) throws IOException {
jaroslav@1375
   758
            if (!resource.endsWith(".class")) {
jaroslav@1375
   759
                return getResource(resource, skip);
jaroslav@1375
   760
            }
jaroslav@1098
   761
            URL u = null;
jaroslav@1033
   762
            for (ClassLoader l : loaders) {
jaroslav@1033
   763
                Enumeration<URL> en = l.getResources(resource);
jaroslav@1033
   764
                while (en.hasMoreElements()) {
jaroslav@1033
   765
                    u = en.nextElement();
jaroslav@1098
   766
                    if (u.toExternalForm().matches("^.*emul.*rt\\.jar.*$")) {
jaroslav@1489
   767
                        return u;
jaroslav@1098
   768
                    }
jaroslav@1033
   769
                }
jaroslav@1098
   770
            }
jaroslav@1098
   771
            if (u != null) {
jaroslav@1098
   772
                if (u.toExternalForm().contains("rt.jar")) {
jaroslav@1347
   773
                    LOG.log(Level.WARNING, "No fallback to bootclasspath for {0}", u);
jaroslav@1347
   774
                    return null;
jaroslav@1033
   775
                }
jaroslav@1489
   776
                return u;
jaroslav@1033
   777
            }
jaroslav@1033
   778
            throw new IOException("Can't find " + resource);
jaroslav@1033
   779
        }
jaroslav@1513
   780
        private URL getResource(String resource, int skip) throws IOException {
jaroslav@1375
   781
            for (ClassLoader l : loaders) {
jaroslav@1375
   782
                Enumeration<URL> en = l.getResources(resource);
jaroslav@1375
   783
                while (en.hasMoreElements()) {
jaroslav@1375
   784
                    final URL now = en.nextElement();
jaroslav@1383
   785
                    if (now.toExternalForm().contains("sisu-inject-bean")) {
jaroslav@1383
   786
                        // certainly we don't want this resource, as that
jaroslav@1383
   787
                        // module is not compiled with target 1.6, currently
jaroslav@1383
   788
                        continue;
jaroslav@1383
   789
                    }
jaroslav@1530
   790
                    if (now.getProtocol().equals("jar")) {
jaroslav@1530
   791
                        JarURLConnection juc = (JarURLConnection) now.openConnection();
jaroslav@1777
   792
                        if (now.getFile().endsWith(".class") && ignore.contains(juc.getJarFileURL())) {
jaroslav@1530
   793
                            continue;
jaroslav@1530
   794
                        }
jaroslav@1530
   795
                    }
jaroslav@1375
   796
                    if (--skip < 0) {
jaroslav@1513
   797
                        return now;
jaroslav@1375
   798
                    }
jaroslav@1375
   799
                }
jaroslav@1375
   800
            }
jaroslav@1383
   801
            throw new IOException("Not found (anymore of) " + resource);
jaroslav@1375
   802
        }
jaroslav@1033
   803
    }
jaroslav@1033
   804
jaroslav@1033
   805
    private static class Page extends HttpHandler {
jaroslav@1033
   806
        final String resource;
jaroslav@1033
   807
        private final String[] args;
jaroslav@1033
   808
        private final Res res;
jaroslav@1927
   809
        private final String ensureBck2Brwsr;
jaroslav@1834
   810
jaroslav@1033
   811
        public Page(Res res, String resource, String... args) {
jaroslav@1927
   812
            this(null, res, resource, args);
jaroslav@1927
   813
        }
jaroslav@1927
   814
jaroslav@1927
   815
        Page(String ensureBck2Brwsr, Res res, String resource, String... args) {
jaroslav@1927
   816
            this.ensureBck2Brwsr = ensureBck2Brwsr;
jaroslav@1033
   817
            this.res = res;
jaroslav@1033
   818
            this.resource = resource;
jaroslav@1033
   819
            this.args = args.length == 0 ? new String[] { "$0" } : args;
jaroslav@1033
   820
        }
jaroslav@1033
   821
jaroslav@1033
   822
        @Override
jaroslav@1033
   823
        public void service(Request request, Response response) throws Exception {
jaroslav@1033
   824
            String r = computePage(request);
jaroslav@1033
   825
            if (r.startsWith("/")) {
jaroslav@1033
   826
                r = r.substring(1);
jaroslav@1033
   827
            }
jaroslav@1033
   828
            String[] replace = {};
jaroslav@1033
   829
            if (r.endsWith(".html")) {
jaroslav@1033
   830
                response.setContentType("text/html");
jaroslav@1033
   831
                LOG.info("Content type text/html");
jaroslav@1033
   832
                replace = args;
jaroslav@1033
   833
            }
jaroslav@1033
   834
            if (r.endsWith(".xhtml")) {
jaroslav@1033
   835
                response.setContentType("application/xhtml+xml");
jaroslav@1033
   836
                LOG.info("Content type application/xhtml+xml");
jaroslav@1033
   837
                replace = args;
jaroslav@1033
   838
            }
jaroslav@1834
   839
            try {
jaroslav@1513
   840
                InputStream is = res.get(r, 0).openStream();
jaroslav@1927
   841
                if (ensureBck2Brwsr != null && ensureBck2Brwsr.equals(request.getRequestURI())) {
jaroslav@1927
   842
                    ByteArrayOutputStream tmp = new ByteArrayOutputStream();
jaroslav@1927
   843
                    copyStream(is, tmp, request.getRequestURL().toString(), replace);
jaroslav@1927
   844
                    String pageText = tmp.toString("UTF-8");
jaroslav@1927
   845
                    if (!pageText.contains("bck2brwsr.js")) {
jaroslav@1927
   846
                        int last = pageText.toLowerCase().indexOf("</body>");
jaroslav@1927
   847
                        if (last == -1) {
jaroslav@1927
   848
                            last = pageText.length();
jaroslav@1927
   849
                        }
jaroslav@1927
   850
                        pageText = pageText.substring(0, last) +
jaroslav@1927
   851
                            "\n<script src='/bck2brwsr.js'></script>\n\n" +
jaroslav@1927
   852
                            pageText.substring(last);
jaroslav@1927
   853
                    }
jaroslav@1927
   854
                    response.getWriter().write(pageText);
jaroslav@1927
   855
                } else {
jaroslav@1927
   856
                    OutputStream os = response.getOutputStream();
jaroslav@1927
   857
                    copyStream(is, os, request.getRequestURL().toString(), replace);
jaroslav@1927
   858
                }
jaroslav@1033
   859
            } catch (IOException ex) {
jaroslav@1033
   860
                response.setDetailMessage(ex.getLocalizedMessage());
jaroslav@1033
   861
                response.setError();
jaroslav@1033
   862
                response.setStatus(404);
jaroslav@1033
   863
            }
jaroslav@1033
   864
        }
jaroslav@1033
   865
jaroslav@1033
   866
        protected String computePage(Request request) {
jaroslav@1033
   867
            String r = resource;
jaroslav@1033
   868
            if (r == null) {
jaroslav@1033
   869
                r = request.getHttpHandlerPath();
jaroslav@1033
   870
            }
jaroslav@1033
   871
            return r;
jaroslav@1033
   872
        }
jaroslav@1033
   873
    }
jaroslav@1834
   874
jaroslav@1033
   875
    private static class SubTree extends Page {
jaroslav@1033
   876
jaroslav@1033
   877
        public SubTree(Res res, String resource, String... args) {
jaroslav@1033
   878
            super(res, resource, args);
jaroslav@1033
   879
        }
jaroslav@1927
   880
        public SubTree(String ensureBck2Brwsr, Res res, String resource, String... args) {
jaroslav@1927
   881
            super(ensureBck2Brwsr, res, resource, args);
jaroslav@1927
   882
        }
jaroslav@1033
   883
jaroslav@1033
   884
        @Override
jaroslav@1033
   885
        protected String computePage(Request request) {
jaroslav@1033
   886
            return resource + request.getHttpHandlerPath();
jaroslav@1033
   887
        }
jaroslav@1834
   888
jaroslav@1834
   889
jaroslav@1033
   890
    }
jaroslav@1033
   891
jaroslav@1271
   892
    private class VMAndPages extends StaticHttpHandler {
jaroslav@1926
   893
        private final boolean unitTestMode;
jaroslav@1271
   894
        private String vmResource;
jaroslav@1834
   895
jaroslav@1926
   896
        public VMAndPages(boolean unitTestMode) {
jaroslav@1271
   897
            super((String[]) null);
jaroslav@1926
   898
            this.unitTestMode = unitTestMode;
jaroslav@1271
   899
        }
jaroslav@1834
   900
jaroslav@1033
   901
        @Override
jaroslav@1033
   902
        public void service(Request request, Response response) throws Exception {
jaroslav@1850
   903
            if ("true".equals(request.getParameter("exit"))) {
jaroslav@1850
   904
                LOG.info("Exit request received. Shutting down!");
jaroslav@1850
   905
                shutdown();
jaroslav@1850
   906
            }
jaroslav@1271
   907
            if (request.getRequestURI().equals(vmResource)) {
jaroslav@1271
   908
                response.setCharacterEncoding("UTF-8");
jaroslav@1271
   909
                response.setContentType("text/javascript");
jaroslav@1271
   910
                StringBuilder sb = new StringBuilder();
jaroslav@1926
   911
                generateBck2BrwsrJS(sb, BaseHTTPLauncher.this.resources, request.getRequestURL().toString(), unitTestMode);
jaroslav@1271
   912
                response.getWriter().write(sb.toString());
jaroslav@1271
   913
            } else {
jaroslav@1271
   914
                super.service(request, response);
jaroslav@1271
   915
            }
jaroslav@1271
   916
        }
jaroslav@1271
   917
jaroslav@1271
   918
        private void registerVM(String vmResource) {
jaroslav@1271
   919
            this.vmResource = vmResource;
jaroslav@1033
   920
        }
jaroslav@1033
   921
    }
jaroslav@1033
   922
jaroslav@1033
   923
    private static class Classes extends HttpHandler {
jaroslav@1033
   924
        private final Res loader;
jaroslav@1033
   925
jaroslav@1033
   926
        public Classes(Res loader) {
jaroslav@1033
   927
            this.loader = loader;
jaroslav@1033
   928
        }
jaroslav@1033
   929
jaroslav@1033
   930
        @Override
jaroslav@1033
   931
        public void service(Request request, Response response) throws Exception {
jaroslav@1033
   932
            String res = request.getHttpHandlerPath();
jaroslav@1033
   933
            if (res.startsWith("/")) {
jaroslav@1033
   934
                res = res.substring(1);
jaroslav@1033
   935
            }
jaroslav@1513
   936
            String skip = request.getParameter("skip");
jaroslav@1513
   937
            int skipCnt = skip == null ? 0 : Integer.parseInt(skip);
jaroslav@1513
   938
            URL url = loader.get(res, skipCnt);
jaroslav@1530
   939
            if (url != null && !res.equals("META-INF/MANIFEST.MF")) try {
jaroslav@1526
   940
                response.setCharacterEncoding("UTF-8");
jaroslav@1521
   941
                if (url.getProtocol().equals("jar")) {
jaroslav@1521
   942
                    JarURLConnection juc = (JarURLConnection) url.openConnection();
jaroslav@1746
   943
                    Object s = null;
jaroslav@1733
   944
                    try {
jaroslav@1733
   945
                        s = loader.compileJar(juc.getJarFileURL());
jaroslav@1733
   946
                    } catch (IOException iOException) {
jaroslav@1733
   947
                        throw new IOException("Can't compile " + url.toExternalForm(), iOException);
jaroslav@1733
   948
                    }
jaroslav@1746
   949
                    if (s instanceof String) {
jaroslav@1521
   950
                        Writer w = response.getWriter();
jaroslav@1746
   951
                        w.append((String)s);
jaroslav@1521
   952
                        w.close();
jaroslav@1521
   953
                        return;
jaroslav@1521
   954
                    }
jaroslav@1746
   955
                    if (s instanceof InputStream) {
jaroslav@1746
   956
                        copyStream((InputStream) s, response.getOutputStream(), null);
jaroslav@1746
   957
                        return;
jaroslav@1746
   958
                    }
jaroslav@1489
   959
                }
jaroslav@1521
   960
                if (url.getProtocol().equals("file")) {
jaroslav@1614
   961
                    final String filePart = url.getFile();
jaroslav@1614
   962
                    if (filePart.endsWith(res)) {
jaroslav@1614
   963
                        url = new URL(
jaroslav@1834
   964
                            url.getProtocol(),
jaroslav@1834
   965
                            url.getHost(),
jaroslav@1834
   966
                            url.getPort(),
jaroslav@1614
   967
                            filePart.substring(0, filePart.length() - res.length())
jaroslav@1614
   968
                        );
jaroslav@1614
   969
                    }
jaroslav@1521
   970
                    String s = loader.compileFromClassPath(url);
jaroslav@1521
   971
                    if (s != null) {
jaroslav@1521
   972
                        Writer w = response.getWriter();
jaroslav@1521
   973
                        w.append(s);
jaroslav@1521
   974
                        w.close();
jaroslav@1521
   975
                        return;
jaroslav@1521
   976
                    }
jaroslav@1489
   977
                }
jaroslav@1521
   978
            } catch (IOException ex) {
jaroslav@1521
   979
                LOG.log(Level.SEVERE, "Cannot handle " + res, ex);
jaroslav@1489
   980
            }
jaroslav@1165
   981
            InputStream is = null;
jaroslav@1165
   982
            try {
jaroslav@1530
   983
                if (url == null) {
jaroslav@1530
   984
                    throw new IOException("Resource not found");
jaroslav@1530
   985
                }
jaroslav@1530
   986
                is = url.openStream();
jaroslav@1033
   987
                response.setContentType("text/javascript");
jaroslav@1033
   988
                Writer w = response.getWriter();
jaroslav@1489
   989
                w.append("([");
jaroslav@1033
   990
                for (int i = 0;; i++) {
jaroslav@1033
   991
                    int b = is.read();
jaroslav@1033
   992
                    if (b == -1) {
jaroslav@1033
   993
                        break;
jaroslav@1033
   994
                    }
jaroslav@1033
   995
                    if (i > 0) {
jaroslav@1033
   996
                        w.append(", ");
jaroslav@1033
   997
                    }
jaroslav@1033
   998
                    if (i % 20 == 0) {
jaroslav@1033
   999
                        w.write("\n");
jaroslav@1033
  1000
                    }
jaroslav@1033
  1001
                    if (b > 127) {
jaroslav@1033
  1002
                        b = b - 256;
jaroslav@1033
  1003
                    }
jaroslav@1033
  1004
                    w.append(Integer.toString(b));
jaroslav@1033
  1005
                }
jaroslav@1489
  1006
                w.append("\n])");
jaroslav@1530
  1007
            } catch (IOException ex) {
jaroslav@1033
  1008
                response.setStatus(HttpStatus.NOT_FOUND_404);
jaroslav@1033
  1009
                response.setError();
jaroslav@1033
  1010
                response.setDetailMessage(ex.getMessage());
jaroslav@1530
  1011
            } finally {
jaroslav@1165
  1012
                if (is != null) {
jaroslav@1165
  1013
                    is.close();
jaroslav@1165
  1014
                }
jaroslav@1530
  1015
            }
jaroslav@1033
  1016
        }
jaroslav@1513
  1017
jaroslav@1033
  1018
    }
jaroslav@1249
  1019
    private static class WS extends WebSocketApplication {
jaroslav@1249
  1020
jaroslav@1249
  1021
        private final Resource r;
jaroslav@1249
  1022
jaroslav@1249
  1023
        private WS(Resource r) {
jaroslav@1249
  1024
            this.r = r;
jaroslav@1249
  1025
        }
jaroslav@1249
  1026
jaroslav@1249
  1027
        @Override
jaroslav@1249
  1028
        public void onMessage(WebSocket socket, String text) {
jaroslav@1249
  1029
            try {
jaroslav@1249
  1030
                r.httpContent.reset();
jaroslav@1249
  1031
                ByteArrayOutputStream out = new ByteArrayOutputStream();
jaroslav@1249
  1032
                copyStream(r.httpContent, out, null, text);
jaroslav@1249
  1033
                String s = new String(out.toByteArray(), "UTF-8");
jaroslav@1249
  1034
                socket.send(s);
jaroslav@1249
  1035
            } catch (IOException ex) {
jaroslav@1513
  1036
                LOG.log(Level.WARNING, null, ex);
jaroslav@1249
  1037
            }
jaroslav@1249
  1038
        }
jaroslav@1249
  1039
jaroslav@1249
  1040
    }}