Ituple.automationBed_Beta.utility.webDriverManager.FirefoxDriverManager.java Source code

Introduction

Here is the source code for Ituple.automationBed_Beta.utility.webDriverManager.FirefoxDriverManager.java

Source

/*
 * (C) Copyright 2016 Boni Garcia (http://bonigarcia.github.io/)
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser General Public License
 * (LGPL) version 2.1 which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/lgpl-2.1.html
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 */
package Ituple.automationBed_Beta.utility.webDriverManager;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.internal.LinkedTreeMap;

/**
 * Manager for Firefox (same as Marionette).
 *
 * @author Boni Garcia (boni.gg@gmail.com)
 * @since 1.5.0
 */
public class FirefoxDriverManager extends BrowserManager {

    private static FirefoxDriverManager instance;

    public FirefoxDriverManager() {
    }

    public static synchronized FirefoxDriverManager getInstance() {
        if (instance == null) {
            instance = new FirefoxDriverManager();
        }
        return instance;
    }

    @Override
    public List<URL> getDrivers() throws IOException {
        URL driverUrl = getDriverUrl();
        String driverVersion = versionToDownload;

        BufferedReader reader = new BufferedReader(new InputStreamReader(openGitHubConnection(driverUrl)));

        GsonBuilder gsonBuilder = new GsonBuilder();
        Gson gson = gsonBuilder.create();
        GitHubApi[] releaseArray = gson.fromJson(reader, GitHubApi[].class);

        if (driverVersion != null) {
            releaseArray = new GitHubApi[] { getVersion(releaseArray, driverVersion) };
        }

        List<URL> urls = new ArrayList<>();
        for (GitHubApi release : releaseArray) {
            if (release != null) {
                List<LinkedTreeMap<String, Object>> assets = release.getAssets();
                for (LinkedTreeMap<String, Object> asset : assets) {
                    urls.add(new URL(asset.get("browser_download_url").toString()));
                }
            }
        }

        reader.close();
        return urls;
    }

    private GitHubApi getVersion(GitHubApi[] releaseArray, String version) {
        GitHubApi out = null;
        for (GitHubApi release : releaseArray) {
            if ((release.getName() != null && release.getName().contains(version))
                    || (release.getTagName() != null && release.getTagName().contains(version))) {
                out = release;
                break;
            }
        }
        return out;
    }

    @Override
    protected String getExportParameter() {
        return WdmConfig.getString("wdm.geckoDriverExport");
    }

    @Override
    protected String getDriverVersion() {
        return WdmConfig.getString("wdm.geckoDriverVersion");
    }

    @Override
    protected List<String> getDriverName() {
        return Arrays.asList("wires", "geckodriver");
    }

    @Override
    protected URL getDriverUrl() throws MalformedURLException {
        return WdmConfig.getUrl("wdm.geckoDriverUrl");
    }

    @Override
    public String getCurrentVersion(URL url, String driverName) throws MalformedURLException {
        String currentVersion = url.getFile().substring(url.getFile().indexOf("-") + 1,
                url.getFile().lastIndexOf("-"));
        if (currentVersion.startsWith("v")) {
            currentVersion = currentVersion.substring(1);
        }
        return currentVersion;
    }
}