/**

 * 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.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 String html = "";

    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;

                html = tmpHtml;

                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");

            response.getOutputBuffer().write(html);

            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);

    }

}