br.com.softplan.security.zap.api.authentication.WebDriverFactory Maven / Gradle / Ivy

Go to download
package br.com.softplan.security.zap.api.authentication;

import org.openqa.selenium.Proxy;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.firefox.FirefoxProfile;
import org.openqa.selenium.htmlunit.HtmlUnitDriver;
import org.openqa.selenium.phantomjs.PhantomJSDriver;
import org.openqa.selenium.remote.DesiredCapabilities;

import br.com.softplan.security.zap.api.model.AuthenticationInfo;
import br.com.softplan.security.zap.commons.ZapInfo;

public final class WebDriverFactory {

	private static final int FIREFOX_MANUAL_PROXY_CONFIGURATION_OPTION = 1;
	
	public static WebDriver makeWebDriver(ZapInfo zapInfo, AuthenticationInfo authenticationInfo) {
		String host = zapInfo.getHost();
		int port = zapInfo.getPort();
		
		switch (authenticationInfo.getSeleniumDriver()) {
			case HTMLUNIT: return makeHtmlUnitDriver(host, port);
			case FIREFOX:   return makeFirefoxDriver(host, port);
//			case CHROME:    return makeChromeDriver(host, port);
			case PHANTOMJS: return makePhantomJSDriver(host, port);
			default:        return makeFirefoxDriver(host, port);
		}
	}
	
	public static HtmlUnitDriver makeHtmlUnitDriver(String host, int port) {
		HtmlUnitDriver driver = new HtmlUnitDriver();
		driver.setProxy(host, port);
		return driver;
	}
	
	public static FirefoxDriver makeFirefoxDriver(String host, int port) {
		FirefoxProfile profile = new FirefoxProfile();
	    profile.setPreference("network.proxy.type", FIREFOX_MANUAL_PROXY_CONFIGURATION_OPTION);
	    profile.setPreference("network.proxy.share_proxy_settings", true);
	    profile.setPreference("network.proxy.no_proxies_on", "");
	    profile.setPreference("network.proxy.http", host);
	    profile.setPreference("network.proxy.http_port", port);
	    
	    return new FirefoxDriver(profile);
	}
	
	// Chrome support was dropped due to ChromeDriver being incompatible with the RemoteDriver version that PhantomJSDriver uses
//	public static ChromeDriver makeChromeDriver(String host, int port) {
//	    DesiredCapabilities capabilities = DesiredCapabilities.chrome();
//	    capabilities.setCapability("proxy", createProxy(host, port));
//	    
//	    return new ChromeDriver(capabilities);
//	}

	private static WebDriver makePhantomJSDriver(String host, int port) {
		DesiredCapabilities capabilities = DesiredCapabilities.phantomjs();
	    capabilities.setCapability("proxy", createProxy(host, port));
	    
		return new PhantomJSDriver(capabilities);
	}
	
	private static Proxy createProxy(String host, int port) {
		Proxy proxy = new Proxy();
	    proxy.setHttpProxy(host + ":" + port);
		return proxy;
	}
	
	private WebDriverFactory() {}
	
}