com.github.mike10004.seleniumhelp.BasicBmpConfigurator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of selenium-capture Show documentation

Capture HTTP traffic generated by Selenium WebDriver in Java

There is a newer version: 0.58

package com.github.mike10004.seleniumhelp;

import com.google.common.base.MoreObjects;
import com.browserup.bup.BrowserUpProxy;
import com.browserup.bup.BrowserUpProxyServer;
import com.browserup.bup.mitm.CertificateAndKeySource;
import org.littleshoot.proxy.ChainedProxyManager;

import javax.annotation.Nullable;

import static java.util.Objects.requireNonNull;

class BasicBmpConfigurator implements BmpConfigurator {

    private final UpstreamProxyDefinition upstreamProxyDefinition;

    public BasicBmpConfigurator(UpstreamProxyDefinition upstreamProxyDefinition) {
        this.upstreamProxyDefinition = requireNonNull(upstreamProxyDefinition, "upstreamProxyDefinition");
    }

    @Override
    public void configureUpstream(BrowserUpProxy bmp) {
        if (upstreamProxyDefinition == null) {
            bmp.setChainedProxy(null);
            if (bmp instanceof BrowserUpProxyServer) {
                ((BrowserUpProxyServer) bmp).setChainedProxyManager(null);
            }
        } else {
            ChainedProxyManager chainedProxyManager = upstreamProxyDefinition.createUpstreamProxy();
            ((BrowserUpProxyServer)bmp).setChainedProxyManager(chainedProxyManager);
        }
    }

    @Override
    public WebdrivingConfig createWebdrivingConfig(BrowserUpProxy bmp, @Nullable CertificateAndKeySource certificateAndKeySource) {
        WebdrivingProxyDefinition proxy = ProxyDefinitionBuilder.through(BrowserMobs.resolveSocketAddress(bmp))
                .buildWebdrivingProxyDefinition();
        return WebdrivingConfig.builder()
                .proxy(proxy)
                .certificateAndKeySource(certificateAndKeySource)
                .build();
    }

    @Override
    public String toString() {
        MoreObjects.ToStringHelper h = MoreObjects.toStringHelper(this);
        h.add("upstreamProxyDefinition", upstreamProxyDefinition);
        return h.toString();
    }
}