com.browserup.bup.proxy.guice.JettyServerProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of browserup-proxy-rest Show documentation
BrowserUp Proxy - REST
There is a newer version: 3.0.0
package com.browserup.bup.proxy.guice;

import com.browserup.bup.proxy.MitmProxyManager;
import com.browserup.bup.rest.validation.mapper.ConstraintViolationExceptionMapper;
import com.browserup.bup.rest.filter.LoggingFilter;
import com.google.inject.Inject;
import com.google.inject.Provider;
import com.google.inject.name.Named;
import com.google.inject.servlet.GuiceFilter;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.UnknownHostException;
import java.util.EnumSet;

import io.swagger.v3.jaxrs2.integration.resources.OpenApiResource;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.servlet.DefaultServlet;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlet.ServletHolder;
import org.glassfish.hk2.api.Factory;
import org.glassfish.hk2.utilities.binding.AbstractBinder;
import org.glassfish.jersey.jackson.JacksonFeature;
import org.glassfish.jersey.server.ResourceConfig;
import org.glassfish.jersey.server.ServerProperties;
import org.glassfish.jersey.servlet.ServletContainer;

import javax.servlet.DispatcherType;

public class JettyServerProvider implements Provider {
    public static final String OPENAPI_CONFIG_YAML = "openapi-config.yaml";
    public static final String OPENAPI_PACKAGE = "com.browserup.bup.rest.resource";

    private Server server;

    @Inject
    public JettyServerProvider(@Named("port") int port, @Named("address") String address, MitmProxyManager proxyManager) throws UnknownHostException {
        OpenApiResource openApiResource = new OpenApiResource();
        openApiResource.setConfigLocation(OPENAPI_CONFIG_YAML);

        ResourceConfig resourceConfig = new ResourceConfig();
        resourceConfig.packages(OPENAPI_PACKAGE);
        resourceConfig.register(openApiResource);
        resourceConfig.register(proxyManagerToHkBinder(proxyManager));
        resourceConfig.register(JacksonFeature.class);
        resourceConfig.register(ConstraintViolationExceptionMapper.class);
        resourceConfig.registerClasses(LoggingFilter.class);

        resourceConfig.property(ServerProperties.BV_SEND_ERROR_IN_RESPONSE, true);
        resourceConfig.property(ServerProperties.WADL_FEATURE_DISABLE, true);

        ServletContextHandler context = new ServletContextHandler(ServletContextHandler.SESSIONS);
        context.setContextPath("/");

        context.addFilter(GuiceFilter.class, "/*", EnumSet.of(DispatcherType.REQUEST));;
        context.addServlet(DefaultServlet.class, "/");
        context.addServlet(new ServletHolder(new ServletContainer(resourceConfig)), "/*");

        server = new Server(new InetSocketAddress(InetAddress.getByName(address), port));
        server.setHandler(context);
    }

    private AbstractBinder proxyManagerToHkBinder(MitmProxyManager proxyManager) {
        Factory proxyManagerFactory = new Factory() {

            @Override
            public MitmProxyManager provide() {
                return proxyManager;
            }

            @Override
            public void dispose(MitmProxyManager instance) {}
        };

        return new AbstractBinder() {
            @Override
            protected void configure() {
                bindFactory(proxyManagerFactory).to(MitmProxyManager.class);
            }
        };
    }

    @Override
    public Server get() {
        return server;
    }
}