io.github.mike10004.extensibleffdriver.AddonSupport Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of extensible-firefox-webdriver Show documentation

There is a newer version: 3.141.59x0.16

Show newest version

package io.github.mike10004.extensibleffdriver;

import io.github.mike10004.extensibleffdriver.LimitedCommandExecutor.Commands;
import org.openqa.selenium.firefox.GeckoDriverService;
import org.openqa.selenium.remote.Command;
import org.openqa.selenium.remote.Response;
import org.openqa.selenium.remote.SessionId;
import org.openqa.selenium.remote.internal.ApacheHttpClient;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.function.Supplier;

class AddonSupport {

    private final Supplier parentDriver;
    private final LimitedCommandExecutor commandExecutor;

    public AddonSupport(Supplier parentDriver, GeckoDriverService driverService) {
        this.parentDriver = Objects.requireNonNull(parentDriver);
        commandExecutor = LimitedCommandExecutor.forService(driverService, new ApacheHttpClient.Factory());
    }

    /**
     * Installs an addon.
     * @param request installation request parameters
     * @throws IOException
     */
    public void installAddon(AddonInstallRequest request) throws IOException {
        Map params = new HashMap<>();
        request.toParameters(params);
        Command command = new Command(parentDriver.get(), Commands.NAME_INSTALL_ADDON, params);
        Response response = commandExecutor.execute(command);
        if (!isSuccess(response)) {
            throw new NonSuccessResponseException(response);
        }
    }

    /**
     * Uninstalls an addon.
     * @param request uninstallation request parameters
     * @throws IOException
     */
    public void uninstallAddon(AddonUninstallRequest request) throws IOException {
        Map params = new HashMap<>();
        request.toParameters(params);
        Command command = new Command(parentDriver.get(), Commands.NAME_UNINSTALL_ADDON, params);
        Response response = commandExecutor.execute(command);
        if (!isSuccess(response)) {
            throw new NonSuccessResponseException(response);
        }
    }

    private boolean isSuccess(Response response) {
        return "success".equals(response.getState());
    }

    static class NonSuccessResponseException extends IOException {
        public NonSuccessResponseException(Response response) {
            super(String.format("status %s in session %s: %s", response.getStatus(), response.getSessionId(), response.getState()));
        }
    }

}