org.mockserver.cli.Main Maven / Gradle / Ivy
package org.mockserver.cli;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.base.Joiner;
import com.google.common.base.Strings;
import org.mockserver.configuration.IntegerStringListParser;
import org.mockserver.mockserver.MockServerBuilder;
import org.mockserver.proxy.ProxyBuilder;
import org.mockserver.stop.StopEventQueue;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.PrintStream;
import java.util.*;
/**
* @author jamesdbloom
*/
public class Main {
public static final String SERVER_PORT_KEY = "serverPort";
public static final String PROXY_PORT_KEY = "proxyPort";
public static final String PROXY_REMOTE_PORT_KEY = "proxyRemotePort";
public static final String PROXY_REMOTE_HOST_KEY = "proxyRemoteHost";
public static final String USAGE = "" +
" java -jar [-serverPort ] [-proxyPort ] [-proxyRemotePort ] [-proxyRemoteHost ]" + System.getProperty("line.separator") +
" " + System.getProperty("line.separator") +
" valid options are: " + System.getProperty("line.separator") +
" -serverPort Specifies the HTTP and HTTPS port for the " + System.getProperty("line.separator") +
" MockServer. Port unification is used to " + System.getProperty("line.separator") +
" support HTTP and HTTPS on the same port. " + System.getProperty("line.separator") +
" " + System.getProperty("line.separator") +
" -proxyPort Specifies the HTTP, HTTPS, SOCKS and HTTP " + System.getProperty("line.separator") +
" CONNECT port for proxy. Port unification " + System.getProperty("line.separator") +
" supports for all protocols on the same port. " + System.getProperty("line.separator") +
" " + System.getProperty("line.separator") +
" -proxyRemotePort Specifies the port to forward all proxy " + System.getProperty("line.separator") +
" requests to (i.e. all requests received on " + System.getProperty("line.separator") +
" portPort). This setting is used to enable " + System.getProperty("line.separator") +
" the port forwarding mode therefore this " + System.getProperty("line.separator") +
" option disables the HTTP, HTTPS, SOCKS and " + System.getProperty("line.separator") +
" HTTP CONNECT support. " + System.getProperty("line.separator") +
" " + System.getProperty("line.separator") +
" -proxyRemoteHost Specified the host to forward all proxy " + System.getProperty("line.separator") +
" requests to (i.e. all requests received on " + System.getProperty("line.separator") +
" portPort). This setting is ignored unless " + System.getProperty("line.separator") +
" proxyRemotePort has been specified. If no " + System.getProperty("line.separator") +
" value is provided for proxyRemoteHost when " + System.getProperty("line.separator") +
" proxyRemotePort has been specified, " + System.getProperty("line.separator") +
" proxyRemoteHost will default to \"localhost\". " + System.getProperty("line.separator") +
" " + System.getProperty("line.separator") +
" i.e. java -jar ./mockserver-jetty-jar-with-dependencies.jar -serverPort 1080 -proxyPort 1090 -proxyRemotePort 80 -proxyRemoteHost www.mock-server.com" + System.getProperty("line.separator") +
" " + System.getProperty("line.separator");
private static final Logger LOGGER = LoggerFactory.getLogger(Main.class);
private static final IntegerStringListParser INTEGER_STRING_LIST_PARSER = new IntegerStringListParser();
@VisibleForTesting
static ProxyBuilder httpProxyBuilder = new ProxyBuilder();
@VisibleForTesting
static MockServerBuilder mockServerBuilder = new MockServerBuilder();
@VisibleForTesting
static StopEventQueue stopEventQueue = new StopEventQueue();
@VisibleForTesting
static PrintStream outputPrintStream = System.out;
@VisibleForTesting
static Runtime runtime = Runtime.getRuntime();
private static boolean usagePrinted = false;
/**
* Run the MockServer directly providing the parseArguments for the server and httpProxyBuilder as the only input parameters (if not provided the server port defaults to 8080 and the httpProxyBuilder is not started).
*
* @param arguments the entries are in pairs:
* - "-serverPort" followed by the server port if not provided the MockServer is not started,
* - "-proxyPort" followed by the proxy port if not provided the Proxy is not started,
* - "-proxyRemotePort" followed by the proxyRemotePort port,
* - "-proxyRemoteHost" followed by the proxyRemoteHost port
*/
public static void main(String... arguments) {
usagePrinted = false;
Map parsedArguments = parseArguments(arguments);
if (LOGGER.isDebugEnabled()) {
LOGGER.debug(System.getProperty("line.separator") + System.getProperty("line.separator") + "Using command line options: " +
Joiner.on(", ").withKeyValueSeparator("=").join(parsedArguments) + System.getProperty("line.separator"));
}
if (parsedArguments.size() > 0 && validateArguments(parsedArguments)) {
if (parsedArguments.containsKey(SERVER_PORT_KEY)) {
mockServerBuilder.withStopEventQueue(stopEventQueue).withHTTPPort(INTEGER_STRING_LIST_PARSER.toArray(parsedArguments.get(SERVER_PORT_KEY))).build();
}
if (parsedArguments.containsKey(PROXY_PORT_KEY)) {
ProxyBuilder proxyBuilder = httpProxyBuilder.withStopEventQueue(stopEventQueue).withLocalPort(Integer.parseInt(parsedArguments.get(PROXY_PORT_KEY)));
if (parsedArguments.containsKey(PROXY_REMOTE_PORT_KEY)) {
String remoteHost = parsedArguments.get(PROXY_REMOTE_HOST_KEY);
if (Strings.isNullOrEmpty(remoteHost)) {
remoteHost = "localhost";
}
proxyBuilder.withDirect(remoteHost, Integer.parseInt(parsedArguments.get(PROXY_REMOTE_PORT_KEY)));
}
proxyBuilder.build();
}
} else {
showUsage();
}
}
private static boolean validateArguments(Map parsedArguments) {
List errorMessages = new ArrayList();
validatePortListArgument(parsedArguments, SERVER_PORT_KEY, errorMessages);
validatePortArgument(parsedArguments, PROXY_PORT_KEY, errorMessages);
validatePortArgument(parsedArguments, PROXY_REMOTE_PORT_KEY, errorMessages);
validateHostnameArgument(parsedArguments, PROXY_REMOTE_HOST_KEY, errorMessages);
if (!errorMessages.isEmpty()) {
int maxLengthMessage = 0;
for (String errorMessage : errorMessages) {
if (errorMessage.length() > maxLengthMessage) {
maxLengthMessage = errorMessage.length();
}
}
outputPrintStream.println(System.getProperty("line.separator") + " " + Strings.padEnd("", maxLengthMessage, '='));
for (String errorMessage : errorMessages) {
outputPrintStream.println(" " + errorMessage);
}
outputPrintStream.println(" " + Strings.padEnd("", maxLengthMessage, '=') + System.getProperty("line.separator"));
return false;
}
return true;
}
private static void validatePortArgument(Map parsedArguments, String argumentKey, List errorMessages) {
if (parsedArguments.containsKey(argumentKey) && !parsedArguments.get(argumentKey).matches("^\\d+$")) {
errorMessages.add(argumentKey + " value \"" + parsedArguments.get(argumentKey) + "\" is invalid, please specify a port i.e. \"1080\"");
}
}
private static void validatePortListArgument(Map parsedArguments, String argumentKey, List errorMessages) {
if (parsedArguments.containsKey(argumentKey) && !parsedArguments.get(argumentKey).matches("^\\d+(,\\d+)*$")) {
errorMessages.add(argumentKey + " value \"" + parsedArguments.get(argumentKey) + "\" is invalid, please specify a comma separated list of ports i.e. \"1080,1081,1082\"");
}
}
private static void validateHostnameArgument(Map parsedArguments, String argumentKey, List errorMessages) {
String validIpAddressRegex = "^(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$";
String validHostnameRegex = "^(([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9\\-]*[a-zA-Z0-9])\\.)*([A-Za-z0-9]|[A-Za-z0-9][A-Za-z0-9\\-]*[A-Za-z0-9])$";
if (parsedArguments.containsKey(argumentKey) && !(parsedArguments.get(argumentKey).matches(validIpAddressRegex) || parsedArguments.get(argumentKey).matches(validHostnameRegex))) {
errorMessages.add(argumentKey + " value \"" + parsedArguments.get(argumentKey) + "\" is invalid, please specify a host name i.e. \"localhost\" or \"127.0.0.1\"");
}
}
private static Map parseArguments(String... arguments) {
Map parsedArguments = new HashMap();
Iterator argumentsIterator = Arrays.asList(arguments).iterator();
while (argumentsIterator.hasNext()) {
String argumentName = argumentsIterator.next();
if (argumentsIterator.hasNext()) {
String argumentValue = argumentsIterator.next();
if (!parsePort(parsedArguments, SERVER_PORT_KEY, argumentName, argumentValue)
&& !parsePort(parsedArguments, PROXY_PORT_KEY, argumentName, argumentValue)
&& !parsePort(parsedArguments, PROXY_REMOTE_PORT_KEY, argumentName, argumentValue)
&& !("-" + PROXY_REMOTE_HOST_KEY).equalsIgnoreCase(argumentName)) {
showUsage();
break;
}
if (("-" + PROXY_REMOTE_HOST_KEY).equalsIgnoreCase(argumentName)) {
parsedArguments.put(PROXY_REMOTE_HOST_KEY, argumentValue);
}
} else {
showUsage();
break;
}
}
return parsedArguments;
}
private static boolean parsePort(Map parsedArguments, final String key, final String argumentName, final String argumentValue) {
if (argumentName.equals("-" + key)) {
parsedArguments.put(key, argumentValue);
return true;
}
return false;
}
private static void showUsage() {
if (!usagePrinted) {
outputPrintStream.print(USAGE);
runtime.exit(1);
usagePrinted = true;
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy