com.browserup.bup.proxy.guice.ConfigModule 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: 2.1.2
/*
 * Modifications Copyright (c) 2019 BrowserUp, Inc.
 */

package com.browserup.bup.proxy.guice;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.inject.Binder;
import com.google.inject.Key;
import com.google.inject.Module;
import com.google.inject.Provider;
import joptsimple.ArgumentAcceptingOptionSpec;
import joptsimple.OptionParser;
import joptsimple.OptionSet;

import java.io.IOException;
import java.util.Arrays;
import java.util.List;
import java.util.TimeZone;

public class ConfigModule implements Module {
    private String[] args;

    public ConfigModule(String[] args) {
        this.args = args;
    }

    @Override
    public void configure(Binder binder) {
        OptionParser parser = new OptionParser();

        ArgumentAcceptingOptionSpec portSpec =
                parser.accepts("port", "The port to listen on")
                        .withOptionalArg().ofType(Integer.class).defaultsTo(8080);

        ArgumentAcceptingOptionSpec addressSpec =
                parser.accepts("address", "The address to bind to")
                        .withOptionalArg()
                        .ofType(String.class)
                        .defaultsTo("0.0.0.0");

        ArgumentAcceptingOptionSpec proxyPortRange =
                parser.accepts("proxyPortRange", "The range of ports to use for proxies")
                        .withOptionalArg()
                        .ofType(Integer.class)
                        .defaultsTo(8081, 8581)
                        .withValuesSeparatedBy('-');

        ArgumentAcceptingOptionSpec ttlSpec =
                parser.accepts("ttl", "Time in seconds until an unused proxy is deleted")
                        .withOptionalArg()
                        .ofType(Integer.class)
                        .defaultsTo(0);

        ArgumentAcceptingOptionSpec useLittleProxy =
                parser.accepts("use-littleproxy", "Use the littleproxy backend instead of the legacy Jetty 5-based implementation")
                        .withOptionalArg()
                        .ofType(Boolean.class)
                        .defaultsTo(true);

        parser.acceptsAll(Arrays.asList("help", "?"), "This help text");

        OptionSet options = parser.parse(args);

        if (options.has("?")) {
            try {
                parser.printHelpOn(System.out);
                System.exit(0);
            } catch (IOException e) {
                // should never happen, but...
                e.printStackTrace();
            }
            return;
        }

        // temporary, until REST API is replaced
        System.out.println("Running BrowserUp Proxy, powered by LittleProxy.");

        List ports = options.valuesOf(proxyPortRange);
        if (ports.size() < 2) {
            throw new IllegalArgumentException();
        }
        Integer minPort;
        Integer maxPort;
        if (ports.get(1) > ports.get(0)) {
            minPort = ports.get(0);
            maxPort = ports.get(1);
        } else {
            minPort = ports.get(1);
            maxPort = ports.get(0);
        }
        Integer port = portSpec.value(options);
        if (port >= minPort && port <= maxPort) {
            int num = maxPort - minPort;
            minPort = port + 1;
            maxPort = minPort + num;
        }

        binder.bind(Key.get(Integer.class, new NamedImpl("port"))).toInstance(port);
        binder.bind(Key.get(String.class, new NamedImpl("address"))).toInstance(addressSpec.value(options));
        binder.bind(Key.get(Integer.class, new NamedImpl("minPort"))).toInstance(minPort);
        binder.bind(Key.get(Integer.class, new NamedImpl("maxPort"))).toInstance(maxPort);
        binder.bind(Key.get(Integer.class, new NamedImpl("ttl"))).toInstance(ttlSpec.value(options));

        // bind an ObjectMapper provider that uses the system time zone instead of UTC by default
        binder.bind(ObjectMapper.class).toProvider(new Provider() {
            @Override
            public ObjectMapper get() {
                ObjectMapper objectMapper = new ObjectMapper();

                objectMapper.setTimeZone(TimeZone.getDefault());

                return objectMapper;
            }
        });
    }
}