/**

 * Back 2 Browser Bytecode Translator

 * Copyright (C) 2012 Jaroslav Tulach <jaroslav.tulach@apidesign.org>

 *

 * This program is free software: you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation, version 2 of the License.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program. Look for COPYING file in the top folder.

 * If not, see http://opensource.org/licenses/GPL-2.0.

 */

package org.apidesign.bck2brwsr.launcher;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.Closeable;

import java.io.File;

import java.io.IOException;

import java.io.InputStream;

import java.io.InterruptedIOException;

import java.io.OutputStream;

import java.io.Reader;

import java.io.UnsupportedEncodingException;

import java.io.Writer;

import java.net.URI;

import java.net.URISyntaxException;

import java.net.URL;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Enumeration;

import java.util.LinkedHashSet;

import java.util.List;

import java.util.Set;

import java.util.concurrent.BlockingQueue;

import java.util.concurrent.Callable;

import java.util.concurrent.CountDownLatch;

import java.util.concurrent.LinkedBlockingQueue;

import java.util.concurrent.TimeUnit;

import java.util.logging.Level;

import java.util.logging.Logger;

import javax.tools.ToolProvider;

import org.apidesign.bck2brwsr.launcher.InvocationContext.Resource;

import org.glassfish.grizzly.PortRange;

import org.glassfish.grizzly.http.server.HttpHandler;

import org.glassfish.grizzly.http.server.HttpServer;

import org.glassfish.grizzly.http.server.NetworkListener;

import org.glassfish.grizzly.http.server.Request;

import org.glassfish.grizzly.http.server.Response;

import org.glassfish.grizzly.http.server.ServerConfiguration;

import org.glassfish.grizzly.http.server.StaticHttpHandler;

import org.glassfish.grizzly.http.util.HttpStatus;

import org.glassfish.grizzly.threadpool.ThreadPoolConfig;

import org.glassfish.grizzly.websockets.WebSocket;

import org.glassfish.grizzly.websockets.WebSocketAddOn;

import org.glassfish.grizzly.websockets.WebSocketApplication;

import org.glassfish.grizzly.websockets.WebSocketEngine;

import org.openide.util.Exceptions;

/**

 * Lightweight server to launch Bck2Brwsr applications and tests.

 * Supports execution in native browser as well as Java's internal 

 * execution engine.

 */

abstract class BaseHTTPLauncher extends Launcher implements Closeable, Callable<HttpServer> {

    static final Logger LOG = Logger.getLogger(BaseHTTPLauncher.class.getName());

    private static final InvocationContext END = new InvocationContext(null, null, null);

    private final Set<ClassLoader> loaders = new LinkedHashSet<ClassLoader>();

    private final BlockingQueue<InvocationContext> methods = new LinkedBlockingQueue<InvocationContext>();

    private long timeOut;

    private final Res resources = new Res();

    private final String cmd;

    private Object[] brwsr;

    private HttpServer server;

    private CountDownLatch wait;

    public BaseHTTPLauncher(String cmd) {

        this.cmd = cmd;

        addClassLoader(BaseHTTPLauncher.class.getClassLoader());

        addClassLoader(ToolProvider.getSystemToolClassLoader());

        setTimeout(180000);

    }

    @Override

    InvocationContext runMethod(InvocationContext c) throws IOException {

        loaders.add(c.clazz.getClassLoader());

        methods.add(c);

        try {

            c.await(timeOut);

        } catch (InterruptedException ex) {

            throw new IOException(ex);

        }

        return c;

    }

    public void setTimeout(long ms) {

        timeOut = ms;

    }

    public void addClassLoader(ClassLoader url) {

        this.loaders.add(url);

    }

    ClassLoader[] loaders() {

        return loaders.toArray(new ClassLoader[loaders.size()]);

    }

    public void showURL(String startpage) throws IOException {

        if (!startpage.startsWith("/")) {

            startpage = "/" + startpage;

        }

        HttpServer s = initServer(".", true, "");

        int last = startpage.lastIndexOf('/');

        String prefix = startpage.substring(0, last);

        String simpleName = startpage.substring(last);

        s.getServerConfiguration().addHttpHandler(new SubTree(resources, prefix), "/");

        server = s;

        try {

            launchServerAndBrwsr(s, simpleName);

        } catch (Exception ex) {

            throw new IOException(ex);

        }

    }

    void showDirectory(File dir, String startpage, boolean addClasses) throws IOException {

        if (!startpage.startsWith("/")) {

            startpage = "/" + startpage;

        }

        String prefix = "";

        int last = startpage.lastIndexOf('/');

        if (last >= 0) {

            prefix = startpage.substring(0, last);

        }

        HttpServer s = initServer(dir.getPath(), addClasses, prefix);

        try {

            launchServerAndBrwsr(s, startpage);

        } catch (Exception ex) {

            throw new IOException(ex);

        }

    }

    @Override

    public void initialize() throws IOException {

        try {

            executeInBrowser();

        } catch (InterruptedException ex) {

            final InterruptedIOException iio = new InterruptedIOException(ex.getMessage());

            iio.initCause(ex);

            throw iio;

        } catch (Exception ex) {

            if (ex instanceof IOException) {

                throw (IOException)ex;

            }

            if (ex instanceof RuntimeException) {

                throw (RuntimeException)ex;

            }

            throw new IOException(ex);

        }

    }

    private HttpServer initServer(String path, boolean addClasses, String vmPrefix) throws IOException {

        HttpServer s = HttpServer.createSimpleServer(null, new PortRange(8080, 65535));

        /*

        ThreadPoolConfig fewThreads = ThreadPoolConfig.defaultConfig().copy().

            setPoolName("Fx/Bck2 Brwsr").

            setCorePoolSize(3).

            setMaxPoolSize(5);

        ThreadPoolConfig oneKernel = ThreadPoolConfig.defaultConfig().copy().

            setPoolName("Kernel Fx/Bck2").

            setCorePoolSize(3).

            setMaxPoolSize(3);

        for (NetworkListener nl : s.getListeners()) {

            nl.getTransport().setWorkerThreadPoolConfig(fewThreads);

            nl.getTransport().setKernelThreadPoolConfig(oneKernel);

        }

        */

        final ServerConfiguration conf = s.getServerConfiguration();

        VMAndPages vm = new VMAndPages();

        conf.addHttpHandler(vm, "/");

        if (vmPrefix != null) {

            vm.registerVM(vmPrefix + "/bck2brwsr.js");

        }

        if (path != null) {

            vm.addDocRoot(path);

        }

        if (addClasses) {

            conf.addHttpHandler(new Classes(resources), "/classes/");

        }

        final WebSocketAddOn addon = new WebSocketAddOn();

        for (NetworkListener listener : s.getListeners()) {

            listener.registerAddOn(addon);

        }

        return s;

    }

    private void executeInBrowser() throws InterruptedException, URISyntaxException, IOException {

        wait = new CountDownLatch(1);

        server = initServer(".", true, "");

        final ServerConfiguration conf = server.getServerConfiguration();

        class DynamicResourceHandler extends HttpHandler {

            private final InvocationContext ic;

            private int resourcesCount;

            DynamicResourceHandler delegate;

            public DynamicResourceHandler(InvocationContext ic) {

                this.ic = ic;

                for (Resource r : ic.resources) {

                    conf.addHttpHandler(this, r.httpPath);

                }

            }

            public void close(DynamicResourceHandler del) {

                conf.removeHttpHandler(this);

                delegate = del;

            }

            @Override

            public void service(Request request, Response response) throws Exception {

                if (delegate != null) {

                    delegate.service(request, response);

                    return;

                }

                if ("/dynamic".equals(request.getRequestURI())) {

                    boolean webSocket = false;

                    String mimeType = request.getParameter("mimeType");

                    List<String> params = new ArrayList<String>();

                    for (int i = 0; ; i++) {

                        String p = request.getParameter("param" + i);

                        if (p == null) {

                            break;

                        }

                        params.add(p);

                        if ("protocol:ws".equals(p)) {

                            webSocket = true;

                            continue;

                        }                    }

                    final String cnt = request.getParameter("content");

                    String mangle = cnt.replace("%20", " ").replace("%0A", "\n");

                    ByteArrayInputStream is = new ByteArrayInputStream(mangle.getBytes("UTF-8"));

                    URI url;

                    final Resource res = new Resource(is, mimeType, "/dynamic/res" + ++resourcesCount, params.toArray(new String[params.size()]));

                    if (webSocket) {

                        url = registerWebSocket(res);

                    } else {

                        url = registerResource(res);

                    }

                    response.getWriter().write(url.toString());

                    response.getWriter().write("\n");

                    return;

                }

                for (Resource r : ic.resources) {

                    if (r.httpPath.equals(request.getRequestURI())) {

                        LOG.log(Level.INFO, "Serving HttpResource for {0}", request.getRequestURI());

                        response.setContentType(r.httpType);

                        r.httpContent.reset();

                        String[] params = null;

                        if (r.parameters.length != 0) {

                            params = new String[r.parameters.length];

                            for (int i = 0; i < r.parameters.length; i++) {

                                params[i] = request.getParameter(r.parameters[i]);

                                if (params[i] == null) {

                                    if ("http.method".equals(r.parameters[i])) {

                                        params[i] = request.getMethod().toString();

                                    } else if ("http.requestBody".equals(r.parameters[i])) {

                                        Reader rdr = request.getReader();

                                        StringBuilder sb = new StringBuilder();

                                        for (;;) {

                                            int ch = rdr.read();

                                            if (ch == -1) {

                                                break;

                                            }

                                            sb.append((char)ch);

                                        }

                                        params[i] = sb.toString();

                                    }

                                }

                                if (params[i] == null) {

                                    params[i] = "null";

                                }

                            }

                        }

                        copyStream(r.httpContent, response.getOutputStream(), null, params);

                    }

                }

            }

            private URI registerWebSocket(Resource r) {

                WebSocketEngine.getEngine().register("", r.httpPath, new WS(r));

                return pageURL("ws", server, r.httpPath);

            }

            private URI registerResource(Resource r) {

                if (!ic.resources.contains(r)) {

                    ic.resources.add(r);

                    conf.addHttpHandler(this, r.httpPath);

                }

                return pageURL("http", server, r.httpPath);

            }

        }

        conf.addHttpHandler(new Page(resources, harnessResource()), "/execute");

        conf.addHttpHandler(new HttpHandler() {

            int cnt;

            List<InvocationContext> cases = new ArrayList<InvocationContext>();

            DynamicResourceHandler prev;

            @Override

            public void service(Request request, Response response) throws Exception {

                String id = request.getParameter("request");

                String value = request.getParameter("result");

                if (value != null && value.indexOf((char)0xC5) != -1) {

                    value = toUTF8(value);

                }

                InvocationContext mi = null;

                int caseNmbr = -1;

                if (id != null && value != null) {

                    LOG.log(Level.INFO, "Received result for case {0} = {1}", new Object[]{id, value});

                    value = decodeURL(value);

                    int indx = Integer.parseInt(id);

                    cases.get(indx).result(value, null);

                    if (++indx < cases.size()) {

                        mi = cases.get(indx);

                        LOG.log(Level.INFO, "Re-executing case {0}", indx);

                        caseNmbr = indx;

                    }

                } else {

                    if (!cases.isEmpty()) {

                        LOG.info("Re-executing test cases");

                        mi = cases.get(0);

                        caseNmbr = 0;

                    }

                }

                if (mi == null) {

                    mi = methods.take();

                    caseNmbr = cnt++;

                }

                if (mi == END) {

                    response.getWriter().write("");

                    wait.countDown();

                    cnt = 0;

                    LOG.log(Level.INFO, "End of data reached. Exiting.");

                    return;

                }

                final DynamicResourceHandler newRH = new DynamicResourceHandler(mi);

                if (prev != null) {

                    prev.close(newRH);

                }

                prev = newRH;

                conf.addHttpHandler(prev, "/dynamic");

                cases.add(mi);

                final String cn = mi.clazz.getName();

                final String mn = mi.methodName;

                LOG.log(Level.INFO, "Request for {0} case. Sending {1}.{2}", new Object[]{caseNmbr, cn, mn});

                response.getWriter().write("{"

                    + "className: '" + cn + "', "

                    + "methodName: '" + mn + "', "

                    + "request: " + caseNmbr

                );

                if (mi.html != null) {

                    response.getWriter().write(", html: '");

                    response.getWriter().write(encodeJSON(mi.html));

                    response.getWriter().write("'");

                }

                response.getWriter().write("}");

            }

        }, "/data");

        this.brwsr = launchServerAndBrwsr(server, "/execute");

    }

    private static String encodeJSON(String in) {

        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < in.length(); i++) {

            char ch = in.charAt(i);

            if (ch < 32 || ch == '\'' || ch == '"') {

                sb.append("\\u");

                String hs = "0000" + Integer.toHexString(ch);

                hs = hs.substring(hs.length() - 4);

                sb.append(hs);

            } else {

                sb.append(ch);

            }

        }

        return sb.toString();

    }

    @Override

    public void shutdown() throws IOException {

        methods.offer(END);

        for (;;) {

            int prev = methods.size();

            try {

                if (wait != null && wait.await(timeOut, TimeUnit.MILLISECONDS)) {

                    break;

                }

            } catch (InterruptedException ex) {

                throw new IOException(ex);

            }

            if (prev == methods.size()) {

                LOG.log(

                    Level.WARNING, 

                    "Timeout and no test has been executed meanwhile (at {0}). Giving up.", 

                    methods.size()

                );

                break;

            }

            LOG.log(Level.INFO, 

                "Timeout, but tests got from {0} to {1}. Trying again.", 

                new Object[]{prev, methods.size()}

            );

        }

        stopServerAndBrwsr(server, brwsr);

    }

    static void copyStream(InputStream is, OutputStream os, String baseURL, String... params) throws IOException {

        for (;;) {

            int ch = is.read();

            if (ch == -1) {

                break;

            }

            if (ch == '$' && params.length > 0) {

                int cnt = is.read() - '0';

                if (baseURL != null && cnt == 'U' - '0') {

                    os.write(baseURL.getBytes("UTF-8"));

                } else {

                    if (cnt >= 0 && cnt < params.length) {

                        os.write(params[cnt].getBytes("UTF-8"));

                    } else {

                        os.write('$');

                        os.write(cnt + '0');

                    }

                }

            } else {

                os.write(ch);

            }

        }

    }

    private Object[] launchServerAndBrwsr(HttpServer server, final String page) throws IOException, URISyntaxException, InterruptedException {

        server.start();

        URI uri = pageURL("http", server, page);

        return showBrwsr(uri);

    }

    private static String toUTF8(String value) throws UnsupportedEncodingException {

        byte[] arr = new byte[value.length()];

        for (int i = 0; i < arr.length; i++) {

            arr[i] = (byte)value.charAt(i);

        }

        return new String(arr, "UTF-8");

    }

    private static String decodeURL(String s) {

        for (;;) {

            int pos = s.indexOf('%');

            if (pos == -1) {

                return s;

            }

            int i = Integer.parseInt(s.substring(pos + 1, pos + 2), 16);

            s = s.substring(0, pos) + (char)i + s.substring(pos + 2);

        }

    }

    private void stopServerAndBrwsr(HttpServer server, Object[] brwsr) throws IOException {

        if (brwsr == null) {

            return;

        }

        Process process = (Process)brwsr[0];

        server.stop();

        InputStream stdout = process.getInputStream();

        InputStream stderr = process.getErrorStream();

        drain("StdOut", stdout);

        drain("StdErr", stderr);

        process.destroy();

        int res;

        try {

            res = process.waitFor();

        } catch (InterruptedException ex) {

            throw new IOException(ex);

        }

        LOG.log(Level.INFO, "Exit code: {0}", res);

        deleteTree((File)brwsr[1]);

    }

    private static void drain(String name, InputStream is) throws IOException {

        int av = is.available();

        if (av > 0) {

            StringBuilder sb = new StringBuilder();

            sb.append("v== ").append(name).append(" ==v\n");

            while (av-- > 0) {

                sb.append((char)is.read());

            }

            sb.append("\n^== ").append(name).append(" ==^");

            LOG.log(Level.INFO, sb.toString());

        }

    }

    private void deleteTree(File file) {

        if (file == null) {

            return;

        }

        File[] arr = file.listFiles();

        if (arr != null) {

            for (File s : arr) {

                deleteTree(s);

            }

        }

        file.delete();

    }

    @Override

    public HttpServer call() throws Exception {

        return server;

    }

    @Override

    public void close() throws IOException {

        shutdown();

    }

    protected Object[] showBrwsr(URI uri) throws IOException {

        LOG.log(Level.INFO, "Showing {0}", uri);

        if (cmd == null) {

            try {

                LOG.log(Level.INFO, "Trying Desktop.browse on {0} {2} by {1}", new Object[] {

                    System.getProperty("java.vm.name"),

                    System.getProperty("java.vm.vendor"),

                    System.getProperty("java.vm.version"),

                });

                java.awt.Desktop.getDesktop().browse(uri);

                LOG.log(Level.INFO, "Desktop.browse successfully finished");

                return null;

            } catch (UnsupportedOperationException ex) {

                LOG.log(Level.INFO, "Desktop.browse not supported: {0}", ex.getMessage());

                LOG.log(Level.FINE, null, ex);

            }

        }

        {

            String cmdName = cmd == null ? "xdg-open" : cmd;

            String[] cmdArr = { 

                cmdName, uri.toString()

            };

            LOG.log(Level.INFO, "Launching {0}", Arrays.toString(cmdArr));

            final Process process = Runtime.getRuntime().exec(cmdArr);

            return new Object[] { process, null };

        }

    }

    abstract void generateBck2BrwsrJS(StringBuilder sb, Res loader) throws IOException;

    abstract String harnessResource();

    private static URI pageURL(String protocol, HttpServer server, final String page) {

        NetworkListener listener = server.getListeners().iterator().next();

        int port = listener.getPort();

        try {

            return new URI(protocol + "://localhost:" + port + page);

        } catch (URISyntaxException ex) {

            throw new IllegalStateException(ex);

        }

    }

    class Res {

        public InputStream get(String resource) throws IOException {

            URL u = null;

            for (ClassLoader l : loaders) {

                Enumeration<URL> en = l.getResources(resource);

                while (en.hasMoreElements()) {

                    u = en.nextElement();

                    if (u.toExternalForm().matches("^.*emul.*rt\\.jar.*$")) {

                        return u.openStream();

                    }

                }

            }

            if (u != null) {

                if (u.toExternalForm().contains("rt.jar")) {

                    LOG.log(Level.WARNING, "Fallback to bootclasspath for {0}", u);

                }

                return u.openStream();

            }

            throw new IOException("Can't find " + resource);

        }

    }

    private static class Page extends HttpHandler {

        final String resource;

        private final String[] args;

        private final Res res;

        public Page(Res res, String resource, String... args) {

            this.res = res;

            this.resource = resource;

            this.args = args.length == 0 ? new String[] { "$0" } : args;

        }

        @Override

        public void service(Request request, Response response) throws Exception {

            String r = computePage(request);

            if (r.startsWith("/")) {

                r = r.substring(1);

            }

            String[] replace = {};

            if (r.endsWith(".html")) {

                response.setContentType("text/html");

                LOG.info("Content type text/html");

                replace = args;

            }

            if (r.endsWith(".xhtml")) {

                response.setContentType("application/xhtml+xml");

                LOG.info("Content type application/xhtml+xml");

                replace = args;

            }

            OutputStream os = response.getOutputStream();

            try { 

                InputStream is = res.get(r);

                copyStream(is, os, request.getRequestURL().toString(), replace);

            } catch (IOException ex) {

                response.setDetailMessage(ex.getLocalizedMessage());

                response.setError();

                response.setStatus(404);

            }

        }

        protected String computePage(Request request) {

            String r = resource;

            if (r == null) {

                r = request.getHttpHandlerPath();

            }

            return r;

        }

    }

    private static class SubTree extends Page {

        public SubTree(Res res, String resource, String... args) {

            super(res, resource, args);

        }

        @Override

        protected String computePage(Request request) {

            return resource + request.getHttpHandlerPath();

        }

    }

    private class VMAndPages extends StaticHttpHandler {

        private String vmResource;

        public VMAndPages() {

            super((String[]) null);

        }

        @Override

        public void service(Request request, Response response) throws Exception {

            if (request.getRequestURI().equals(vmResource)) {

                response.setCharacterEncoding("UTF-8");

                response.setContentType("text/javascript");

                StringBuilder sb = new StringBuilder();

                generateBck2BrwsrJS(sb, BaseHTTPLauncher.this.resources);

                response.getWriter().write(sb.toString());

            } else {

                super.service(request, response);

            }

        }

        private void registerVM(String vmResource) {

            this.vmResource = vmResource;

        }

    }

    private static class Classes extends HttpHandler {

        private final Res loader;

        public Classes(Res loader) {

            this.loader = loader;

        }

        @Override

        public void service(Request request, Response response) throws Exception {

            String res = request.getHttpHandlerPath();

            if (res.startsWith("/")) {

                res = res.substring(1);

            }

            InputStream is = null;

            try {

                is = loader.get(res);

                response.setContentType("text/javascript");

                Writer w = response.getWriter();

                w.append("[");

                for (int i = 0;; i++) {

                    int b = is.read();

                    if (b == -1) {

                        break;

                    }

                    if (i > 0) {

                        w.append(", ");

                    }

                    if (i % 20 == 0) {

                        w.write("\n");

                    }

                    if (b > 127) {

                        b = b - 256;

                    }

                    w.append(Integer.toString(b));

                }

                w.append("\n]");

            } catch (IOException ex) {

                response.setStatus(HttpStatus.NOT_FOUND_404);

                response.setError();

                response.setDetailMessage(ex.getMessage());

            } finally {

                if (is != null) {

                    is.close();

                }

            }

        }

    }

    private static class WS extends WebSocketApplication {

        private final Resource r;

        private WS(Resource r) {

            this.r = r;

        }

        @Override

        public void onMessage(WebSocket socket, String text) {

            try {

                r.httpContent.reset();

                ByteArrayOutputStream out = new ByteArrayOutputStream();

                copyStream(r.httpContent, out, null, text);

                String s = new String(out.toByteArray(), "UTF-8");

                socket.send(s);

            } catch (IOException ex) {

                Exceptions.printStackTrace(ex);

            }

        }

    }}