//
// Source code recreated from a .class file by IntelliJ IDEA
// (powered by Fernflower decompiler)
//
package org.openqa.selenium.firefox;
import com.google.common.base.Predicate;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
import java.io.File;
import java.io.IOException;
import java.net.URI;
import java.util.HashSet;
import java.util.Iterator;
import java.util.concurrent.TimeUnit;
import org.openqa.selenium.Capabilities;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.Platform;
import org.openqa.selenium.Proxy;
import org.openqa.selenium.WebDriverException;
import org.openqa.selenium.WebDriver.Options;
import org.openqa.selenium.WebDriver.Timeouts;
import org.openqa.selenium.firefox.ExtensionConnection;
import org.openqa.selenium.firefox.FirefoxBinary;
import org.openqa.selenium.firefox.FirefoxProfile;
import org.openqa.selenium.firefox.internal.NewProfileExtensionConnection;
import org.openqa.selenium.firefox.internal.ProfilesIni;
import org.openqa.selenium.internal.Killable;
import org.openqa.selenium.internal.Lock;
import org.openqa.selenium.internal.SocketLock;
import org.openqa.selenium.logging.LocalLogs;
import org.openqa.selenium.logging.LoggingPreferences;
import org.openqa.selenium.logging.NeedsLocalLogs;
import org.openqa.selenium.remote.BeanToJsonConverter;
import org.openqa.selenium.remote.Command;
import org.openqa.selenium.remote.CommandExecutor;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.FileDetector;
import org.openqa.selenium.remote.RemoteWebDriver;
import org.openqa.selenium.remote.Response;
import org.openqa.selenium.remote.SessionNotFoundException;
import org.openqa.selenium.remote.RemoteWebDriver.RemoteWebDriverOptions;
import org.openqa.selenium.remote.RemoteWebDriver.RemoteWebDriverOptions.RemoteTimeouts;
public class FirefoxDriver extends RemoteWebDriver implements Killable {
public static final String BINARY = "firefox_binary";
public static final String PROFILE = "firefox_profile";
public static final boolean DEFAULT_ENABLE_NATIVE_EVENTS;
/** @deprecated /
@Deprecated
public static final boolean ACCEPT_UNTRUSTED_CERTIFICATES = true;
/* @deprecated */
@Deprecated
public static final boolean ASSUME_UNTRUSTED_ISSUER = true;
protected FirefoxBinary binary;
public FirefoxDriver() {
this((FirefoxBinary)(new FirefoxBinary()), (FirefoxProfile)null);
}
public FirefoxDriver(FirefoxProfile profile) {
this(new FirefoxBinary(), profile);
}
public FirefoxDriver(Capabilities desiredCapabilities) {
this(getBinary(desiredCapabilities), extractProfile(desiredCapabilities, (Capabilities)null), desiredCapabilities);
}
public FirefoxDriver(Capabilities desiredCapabilities, Capabilities requiredCapabilities) {
this(getBinary(desiredCapabilities), extractProfile(desiredCapabilities, requiredCapabilities), desiredCapabilities, requiredCapabilities);
}
private static FirefoxProfile extractProfile(Capabilities desiredCapabilities, Capabilities requiredCapabilities) {
FirefoxProfile profile = null;
Object raw = null;
if(desiredCapabilities != null && desiredCapabilities.getCapability("firefox_profile") != null) {
raw = desiredCapabilities.getCapability("firefox_profile");
}
if(requiredCapabilities != null && requiredCapabilities.getCapability("firefox_profile") != null) {
raw = requiredCapabilities.getCapability("firefox_profile");
}
if(raw != null) {
if(raw instanceof FirefoxProfile) {
profile = (FirefoxProfile)raw;
} else if(raw instanceof String) {
try {
profile = FirefoxProfile.fromJson((String)raw);
} catch (IOException var5) {
throw new WebDriverException(var5);
}
}
}
profile = getProfile(profile);
populateProfile(profile, desiredCapabilities);
populateProfile(profile, requiredCapabilities);
return profile;
}
static void populateProfile(FirefoxProfile profile, Capabilities capabilities) {
if(capabilities != null) {
Boolean nativeEventsEnabled;
if(capabilities.getCapability("webStorageEnabled") != null) {
nativeEventsEnabled = (Boolean)capabilities.getCapability("webStorageEnabled");
profile.setPreference("dom.storage.enabled", nativeEventsEnabled.booleanValue());
}
if(capabilities.getCapability("acceptSslCerts") != null) {
nativeEventsEnabled = (Boolean)capabilities.getCapability("acceptSslCerts");
profile.setAcceptUntrustedCertificates(nativeEventsEnabled.booleanValue());
}
if(capabilities.getCapability("loggingPrefs") != null) {
LoggingPreferences nativeEventsEnabled1 = (LoggingPreferences)capabilities.getCapability("loggingPrefs");
Iterator var3 = nativeEventsEnabled1.getEnabledLogTypes().iterator();
while(var3.hasNext()) {
String logtype = (String)var3.next();
profile.setPreference("webdriver.log." + logtype, nativeEventsEnabled1.getLevel(logtype).intValue());
}
}
if(capabilities.getCapability("nativeEvents") != null) {
nativeEventsEnabled = (Boolean)capabilities.getCapability("nativeEvents");
profile.setEnableNativeEvents(nativeEventsEnabled.booleanValue());
}
}
}
private static FirefoxBinary getBinary(Capabilities capabilities) {
if(capabilities != null && capabilities.getCapability("firefox_binary") != null) {
Object raw = capabilities.getCapability("firefox_binary");
if(raw instanceof FirefoxBinary) {
return (FirefoxBinary)raw;
} else {
File file = new File((String)raw);
return new FirefoxBinary(file);
}
} else {
return new FirefoxBinary();
}
}
public FirefoxDriver(FirefoxBinary binary, FirefoxProfile profile) {
this(binary, profile, DesiredCapabilities.firefox());
}
public FirefoxDriver(FirefoxBinary binary, FirefoxProfile profile, Capabilities capabilities) {
this(binary, profile, capabilities, (Capabilities)null);
}
public FirefoxDriver(FirefoxBinary binary, FirefoxProfile profile, Capabilities desiredCapabilities, Capabilities requiredCapabilities) {
super(new FirefoxDriver.LazyCommandExecutor(binary, profile, null), dropCapabilities(desiredCapabilities, new String[]{"firefox_binary", "firefox_profile"}), dropCapabilities(requiredCapabilities, new String[]{"firefox_binary", "firefox_profile"}));
this.binary = binary;
}
public void setFileDetector(FileDetector detector) {
throw new WebDriverException("Setting the file detector only works on remote webdriver instances obtained via RemoteWebDriver");
}
public void kill() {
this.binary.quit();
}
public Options manage() {
return new RemoteWebDriverOptions(this) {
public Timeouts timeouts() {
return new RemoteTimeouts(this) {
public Timeouts implicitlyWait(long time, TimeUnit unit) {
FirefoxDriver.this.execute("setTimeout", ImmutableMap.of("type", "implicit", "ms", Long.valueOf(TimeUnit.MILLISECONDS.convert(time, unit))));
return this;
}
public Timeouts setScriptTimeout(long time, TimeUnit unit) {
FirefoxDriver.this.execute("setTimeout", ImmutableMap.of("type", "script", "ms", Long.valueOf(TimeUnit.MILLISECONDS.convert(time, unit))));
return this;
}
};
}
};
}
protected void startClient() {
FirefoxDriver.LazyCommandExecutor exe = (FirefoxDriver.LazyCommandExecutor)this.getCommandExecutor();
FirefoxProfile profileToUse = getProfile(exe.profile);
ExtensionConnection connection = this.connectTo(exe.binary, profileToUse, "localhost");
exe.setConnection(connection);
try {
connection.start();
} catch (IOException var5) {
throw new WebDriverException("An error occurred while connecting to Firefox", var5);
}
}
private static FirefoxProfile getProfile(FirefoxProfile profile) {
FirefoxProfile profileToUse = profile;
String suggestedProfile = System.getProperty("webdriver.firefox.profile");
if(profile == null && suggestedProfile != null) {
profileToUse = (new ProfilesIni()).getProfile(suggestedProfile);
if(profileToUse == null) {
throw new WebDriverException(String.format("Firefox profile \'%s\' named in system property \'%s\' not found", new Object[]{suggestedProfile, "webdriver.firefox.profile"}));
}
} else if(profile == null) {
profileToUse = new FirefoxProfile();
}
return profileToUse;
}
protected ExtensionConnection connectTo(FirefoxBinary binary, FirefoxProfile profile, String host) {
Lock lock = this.obtainLock(profile);
try {
FirefoxBinary e = binary == null?new FirefoxBinary():binary;
return new NewProfileExtensionConnection(lock, e, profile, host);
} catch (Exception var6) {
throw new WebDriverException(var6);
}
}
protected Lock obtainLock(FirefoxProfile profile) {
return new SocketLock();
}
protected void stopClient() {
((FirefoxDriver.LazyCommandExecutor)this.getCommandExecutor()).quit();
}
private static Capabilities dropCapabilities(Capabilities capabilities, String... keysToRemove) {
if(capabilities == null) {
return new DesiredCapabilities();
} else {
final HashSet toRemove = Sets.newHashSet(keysToRemove);
DesiredCapabilities caps = new DesiredCapabilities(Maps.filterKeys(capabilities.asMap(), new Predicate() {
public boolean apply(String key) {
return !toRemove.contains(key);
}
}));
Proxy proxy = Proxy.extractFrom(capabilities);
if(proxy != null) {
caps.setCapability("proxy", (new BeanToJsonConverter()).convert(proxy));
}
return caps;
}
}
public <X> X getScreenshotAs(OutputType<X> target) {
String base64 = this.execute("screenshot").getValue().toString();
return target.convertFromBase64Png(base64);
}
static {
DEFAULT_ENABLE_NATIVE_EVENTS = Platform.getCurrent().is(Platform.WINDOWS);
}
public static class LazyCommandExecutor implements CommandExecutor, NeedsLocalLogs {
private ExtensionConnection connection;
private final FirefoxBinary binary;
private final FirefoxProfile profile;
private LocalLogs logs;
private LazyCommandExecutor(FirefoxBinary binary, FirefoxProfile profile) {
this.logs = LocalLogs.getNullLogger();
this.binary = binary;
this.profile = profile;
}
public void setConnection(ExtensionConnection connection) {
this.connection = connection;
connection.setLocalLogs(this.logs);
}
public void quit() {
if(this.connection != null) {
this.connection.quit();
this.connection = null;
}
if(this.profile != null) {
this.profile.cleanTemporaryModel();
}
}
public Response execute(Command command) throws IOException {
if(this.connection == null) {
if(command.getName().equals("quit")) {
return new Response();
} else {
throw new SessionNotFoundException("The FirefoxDriver cannot be used after quit() was called.");
}
} else {
return this.connection.execute(command);
}
}
public void setLocalLogs(LocalLogs logs) {
this.logs = logs;
if(this.connection != null) {
this.connection.setLocalLogs(logs);
}
}
public URI getAddressOfRemoteServer() {
return this.connection.getAddressOfRemoteServer();
}
}
public static final class SystemProperty {
public static final String BROWSER_BINARY = "webdriver.firefox.bin";
public static final String BROWSER_LOGFILE = "webdriver.firefox.logfile";
public static final String BROWSER_LIBRARY_PATH = "webdriver.firefox.library.path";
public static final String BROWSER_PROFILE = "webdriver.firefox.profile";
public static final String DRIVER_XPI_PROPERTY = "webdriver.firefox.driver";
public static final String DRIVER_USE_MARIONETTE = "webdriver.firefox.marionette";
public SystemProperty() {
}
}
}