package org.litesoft.ddt;

import org.eclipse.jetty.http.MimeTypes;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.handler.ContextHandler;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlet.ServletHandler;

import javax.servlet.Servlet;
import java.sql.Timestamp;

public final class EmbeddedJetty extends Server {
    public EmbeddedJetty(int port, Class<? extends Servlet> pServletClass, String pPathSpec, String... pAdditionalMimeTypes) {
        super(port);

        ServletHandler zServletHandler = new ServletHandler();
        zServletHandler.addServletWithMapping(pServletClass, pPathSpec);

        ContextHandler zContextHandler = new ServletContextHandler();
        zContextHandler.setMimeTypes(getMimeTypes(pAdditionalMimeTypes));
        zContextHandler.setHandler(zServletHandler);

        setHandler(zContextHandler);

        setStopTimeout(1000);
        setStopAtShutdown(true);
    }

    public int run() throws Exception {
        ServerKiller zKiller = new ServerKiller();
        Shutdown.Instance.set(zKiller);
        start();
        while (!isRunning()) {
            if (isStopped()) {
                throw new Error("Stopped on start!");
            }
            if (isFailed()) {
                throw new Error("Failed on start!");
            }
            Thread.sleep(50);
        }
        report("running", "");
        join();
        int zExitCode = zKiller.mExitCode;
        report("exiting", " with: " + zExitCode);
        return zExitCode;
    }

    private void report(String pWhat, String pTail) {
        System.out.println("EmbeddedJetty Server " + pWhat + " @ '" + new Timestamp(System.currentTimeMillis()) + "'" + pTail);
    }

    protected MimeTypes getMimeTypes(String... pAdditionalMimeTypes) {
        // Handle files encoded in UTF-8
        MimeTypes zMimeTypes = new MimeTypes();
        if (pAdditionalMimeTypes != null) {
            for (int i = 1; i < pAdditionalMimeTypes.length; i += 2) {
                String zExtension = pAdditionalMimeTypes[i - 1];
                String zType = pAdditionalMimeTypes[i];
                zMimeTypes.addMimeMapping(zExtension, zType);
            }
        }
        return zMimeTypes;
    }

    private class ServerKiller implements Shutdown {
        private volatile int mExitCode;

        @Override
        public void request(int pExitCode) {
            if (pExitCode != 0) {
                mExitCode = pExitCode;
            }
            new Thread(new Runnable() {
                @Override
                public void run() {
                    try {
                        Thread.sleep(100);
                        stop();
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                }
            }).start();
        }
    }
}