io.specto.hoverfly.junit.core.ProxyConfigurer Maven / Gradle / Ivy

Go to download
package io.specto.hoverfly.junit.core;

import io.specto.hoverfly.junit.core.config.HoverflyConfiguration;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.HashMap;
import java.util.Map;

import static io.specto.hoverfly.junit.core.SystemProperty.HTTPS_PROXY_HOST;
import static io.specto.hoverfly.junit.core.SystemProperty.HTTPS_PROXY_PORT;
import static io.specto.hoverfly.junit.core.SystemProperty.HTTP_NON_PROXY_HOSTS;
import static io.specto.hoverfly.junit.core.SystemProperty.HTTP_PROXY_HOST;
import static io.specto.hoverfly.junit.core.SystemProperty.HTTP_PROXY_PORT;

class ProxyConfigurer {

    private static final Logger LOGGER = LoggerFactory.getLogger(ProxyConfigurer.class);

    private final HoverflyConfiguration hoverflyConfig;
    private final Map originalProxyProperties = new HashMap<>();

    ProxyConfigurer(HoverflyConfiguration hoverflyConfig) {
        this.hoverflyConfig = hoverflyConfig;
    }

    /**
     * Configures the JVM system properties to use Hoverfly as a proxy
     */
    void setProxySystemProperties() {
        if (hoverflyConfig.isWebServer()) {
            // Do nothing if Hoverfly acts as a web server!
            return;
        }

        keepOriginalProxyProperties(HTTP_NON_PROXY_HOSTS, HTTP_PROXY_HOST, HTTP_PROXY_PORT, HTTPS_PROXY_HOST, HTTPS_PROXY_PORT);
        LOGGER.info("Setting proxy host to {}", hoverflyConfig.getHost());
        System.setProperty(HTTP_PROXY_HOST, hoverflyConfig.getHost());
        System.setProperty(HTTPS_PROXY_HOST, hoverflyConfig.getHost());

        if (hoverflyConfig.isProxyLocalHost()) {
            System.setProperty(HTTP_NON_PROXY_HOSTS, "");
        } else {
            System.setProperty(HTTP_NON_PROXY_HOSTS, "localhost|127.*|[::1]");
        }

        if (hoverflyConfig.isRemoteInstance()) {
            String nonProxyHosts = System.getProperty(HTTP_NON_PROXY_HOSTS);
            if (StringUtils.isNotBlank(nonProxyHosts)) {
                nonProxyHosts = String.join("|", nonProxyHosts, hoverflyConfig.getHost());
            } else {
                nonProxyHosts = hoverflyConfig.getHost();
            }
            System.setProperty(HTTP_NON_PROXY_HOSTS, nonProxyHosts);
        }

        LOGGER.info("Setting proxy proxyPort to {}", hoverflyConfig.getProxyPort());

        System.setProperty(HTTP_PROXY_PORT, String.valueOf(hoverflyConfig.getProxyPort()));
        System.setProperty(HTTPS_PROXY_PORT, String.valueOf(hoverflyConfig.getProxyPort()));
    }

    void restoreProxySystemProperties() {
        if (hoverflyConfig.isWebServer()) {
            // Do nothing if Hoverfly acts as a web server!
            return;
        }

        for (Map.Entry originalProperty : this.originalProxyProperties.entrySet()) {
            final String property = originalProperty.getKey();
            final String originalValue = originalProperty.getValue();
            if (originalValue == null) {
                System.clearProperty(property);
            } else {
                System.setProperty(property, originalValue);
            }
        }
        this.originalProxyProperties.clear();
    }

    private void keepOriginalProxyProperties(String... properties) {
        for (String property : properties) {
            this.originalProxyProperties.put(property, System.getProperty(property));
        }
    }
}