ProxyServer.java

package de.otto.hmac.proxy;

import com.sun.jersey.api.container.grizzly2.GrizzlyServerFactory;
import com.sun.jersey.api.core.PackagesResourceConfig;
import com.sun.jersey.api.core.ResourceConfig;
import org.glassfish.grizzly.http.server.HttpServer;

import javax.ws.rs.core.UriBuilder;
import java.io.IOException;
import java.net.URI;
import java.util.logging.Level;
import java.util.logging.Logger;

import static de.otto.hmac.proxy.CLIParameterToConfigurationReader.toConfiguration;

public class ProxyServer {

    private static HttpServer httpServer;

    private static URI getBaseURI() {
        return UriBuilder.fromUri("http://localhost/").port(ProxyConfiguration.getSourcePort()).build();
    }

    protected static void startServer() throws IOException {
        ResourceConfig rc = new PackagesResourceConfig("de.otto.hmac.proxy");
        httpServer = GrizzlyServerFactory.createHttpServer(getBaseURI(), rc);
    }

    protected static void stopServer() {
        if (httpServer != null) {
            httpServer.stop();
        }
    }

    public static void main(String[] args) throws IOException, InterruptedException {
        toConfiguration(args);

        if (!ProxyConfiguration.isVerbose()) {
            Logger.getLogger("").setLevel(Level.OFF);
        }

        startServer();

        if (ProxyConfiguration.isHelp()) {
            System.out.println("A local proxy server that forwards all requests to the given target host and port, \n" +
                    "but with hmac-authentication headers appended.\n" +
                    "USAGE:\n" +
                    "    -u --user USER             authenticated username\n" +
                    "    -p --password PASSWORD     password of user\n" +
                    "    -h --host HOST             target host\n" +
                    "    -sp --sourcePort PORT      proxy port, defaults to 9998\n" +
                    "    -tp --targetPort PORT      target port\n" +
                    "    -v --verbose               display more stuff\n" +
                    "    -d --daemon                start proxy without waiting for input, so it\n" +
                    "                               can be started as a daemon\n"
            );
            return;
        }


        System.out.println(String.format("HMAC-Proxy listens to [%s]", getBaseURI()));
        System.out.println(String.format("HMAC-Proxy forwards to [%s:%d]", ProxyConfiguration.getTargetHost(), ProxyConfiguration.getPort()));
        System.out.println(String.format("As user [%s]", ProxyConfiguration.getUser()));
        if (ProxyConfiguration.isDaemon()) {
            while(true) {
                Thread.sleep(1000);
            }
        } else {
            System.out.println("Hit enter to stop proxy...");
            System.in.read();
        }
        stopServer();
    }
}