org.apidesign.bck2brwsr.dew.Dew.java Source code

Introduction

Here is the source code for org.apidesign.bck2brwsr.dew.Dew.java

Source

/**
 * Development Environment for Web
 * Copyright (C) 2012-2013 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.dew;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.util.List;
import java.util.Locale;
import javax.tools.Diagnostic;
import javax.tools.JavaFileObject;
import org.apidesign.vm4brwsr.Bck2Brwsr;
import org.glassfish.grizzly.http.Method;
import org.glassfish.grizzly.http.server.HttpHandler;
import org.glassfish.grizzly.http.server.HttpServer;
import org.glassfish.grizzly.http.server.Request;
import org.glassfish.grizzly.http.server.Response;
import org.glassfish.grizzly.http.util.HttpStatus;
import org.json.JSONArray;
import org.json.JSONObject;
import org.json.JSONTokener;

/**
 *
 * @author phrebejk
 */
final class Dew extends HttpHandler implements Bck2Brwsr.Resources {
    private Compile data;

    public static void main(String... args) throws Exception {
        DewLauncher l = new DewLauncher(null);
        l.addClassLoader(DewLauncher.class.getClassLoader());
        final Dew dew = new Dew();
        HttpServer s = l.initServer(dew);
        s.getServerConfiguration().addHttpHandler(dew, "/dew/");
        l.launchServerAndBrwsr(s, "/dew/");
        System.in.read();
    }

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

        if (request.getMethod() == Method.POST) {
            InputStream is = request.getInputStream();
            JSONTokener tok = new JSONTokener(new InputStreamReader(is));
            JSONObject obj = new JSONObject(tok);
            String tmpHtml = obj.getString("html");
            String tmpJava = obj.getString("java");

            Compile res = Compile.create(tmpHtml, tmpJava);
            List<Diagnostic<? extends JavaFileObject>> err = res.getErrors();
            if (err.isEmpty()) {
                data = res;
                response.getOutputStream().write("[]".getBytes());
                response.setStatus(HttpStatus.OK_200);
            } else {

                JSONArray errors = new JSONArray();

                for (Diagnostic<? extends JavaFileObject> d : err) {
                    JSONObject e = new JSONObject();
                    e.put("col", d.getColumnNumber());
                    e.put("line", d.getLineNumber());
                    e.put("kind", d.getKind().toString());
                    e.put("msg", d.getMessage(Locale.ENGLISH));
                    errors.put(e);
                }

                errors.write(response.getWriter());
                response.setStatus(HttpStatus.PRECONDITION_FAILED_412);
            }

            return;
        }

        String r = request.getHttpHandlerPath();
        if (r == null || r.equals("/")) {
            r = "index.html";
        }
        if (r.equals("/result.html")) {
            response.setContentType("text/html");
            if (data != null) {
                response.getOutputBuffer().write(data.getHtml());
            }
            response.setStatus(HttpStatus.OK_200);
            return;
        }

        if (r.startsWith("/")) {
            r = r.substring(1);
        }

        if (r.endsWith(".html") || r.endsWith(".xhtml")) {
            response.setContentType("text/html");
        }
        OutputStream os = response.getOutputStream();
        try (InputStream is = Dew.class.getResourceAsStream(r)) {
            copyStream(is, os, request.getRequestURL().toString());
        } catch (IOException ex) {
            response.setDetailMessage(ex.getLocalizedMessage());
            response.setError();
            response.setStatus(404);
        }
    }

    static void copyStream(InputStream is, OutputStream os, String baseURL) throws IOException {
        for (;;) {
            int ch = is.read();
            if (ch == -1) {
                break;
            }
            os.write(ch);
        }
    }

    @Override
    public InputStream get(String r) throws IOException {
        byte[] arr = data == null ? null : data.get(r);
        return arr == null ? null : new ByteArrayInputStream(arr);
    }
}