org.sitoolkit.wt.infra.firefox.FirefoxManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sit-wt-runtime Show documentation
There is a newer version: 2.3
package org.sitoolkit.wt.infra.firefox;

import java.io.File;
import java.io.IOException;
import java.net.URL;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.SystemUtils;
import org.openqa.selenium.firefox.FirefoxBinary;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.firefox.FirefoxProfile;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.sitoolkit.wt.infra.MultiThreadUtils;
import org.sitoolkit.wt.infra.SitRepository;
import org.sitoolkit.wt.infra.process.ProcessUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.zeroturnaround.zip.ZipUtil;

public class FirefoxManager {

    private static final Logger LOG = LoggerFactory.getLogger(FirefoxManager.class);

    public FirefoxDriver startWebDriver(DesiredCapabilities capabilities) {
        return MultiThreadUtils.submitWithProgress(
                () -> new FirefoxDriver(getFirefoxBinary(), new FirefoxProfile(), capabilities));
    }

    public FirefoxBinary getFirefoxBinary() {
        File ffBinaryFile = getFirefoxBinaryFile();

        if (ffBinaryFile.exists()) {

            LOG.info("Firefoxはインストール済みです {}", ffBinaryFile.getAbsolutePath());

        } else {

            LOG.info("SIT-WT用のFirefoxがインストールされていません");
            installFirefox();

        }

        return new FirefoxBinary(ffBinaryFile);
    }

    public File getSeleniumIdeUnarchivedDir() {
        File repo = new File(SitRepository.getRepositoryPath(), "selenium/ide");

        if (!repo.exists()) {
            repo.mkdirs();
        }

        File xpi = new File(repo, "selenium_ide-2.9.1-fx.xpi");

        if (xpi.exists()) {
            LOG.info("Selenium IDEはダウンロード済みです {}", xpi.getAbsolutePath());
        } else {

            try {

                // TODO 外部化
                URL xpiUrl = new URL(
                        "https://addons.mozilla.org/firefox/downloads/latest/selenium-ide/addon-2079-latest.xpi?src=dp-btn-primary");

                LOG.info("Selenium IDEをダウンロードします {} -> {}", xpiUrl, xpi.getAbsolutePath());

                FileUtils.copyURLToFile(xpiUrl, xpi);
                ZipUtil.unpack(xpi, repo);

            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }

        return repo;
    }

    protected File getFirefoxBinaryFile() {
        if (SystemUtils.IS_OS_WINDOWS) {
            return new File(SitRepository.getRepositoryPath(), "firefox/runtime/firefox.exe");
        } else if (SystemUtils.IS_OS_MAC) {
            return new File(SitRepository.getRepositoryPath(),
                    "firefox/runtime/Firefox.app/Contents/MacOS/firefox-bin");
        } else {
            throw new UnsupportedOperationException("サポートされていないOSです");
        }
    }

    protected void installFirefox() {
        File repo = new File(SitRepository.getRepositoryPath(), "firefox");

        if (!repo.exists()) {
            repo.mkdir();
        }

        if (SystemUtils.IS_OS_WINDOWS) {
            installFirefoxWindows(repo);
        } else if (SystemUtils.IS_OS_MAC) {
            installFirefoxMacOs(repo);
        } else {
            throw new UnsupportedOperationException("サポートされていないOSです");
        }
    }

    protected void installFirefoxWindows(File repo) {
        File ffInstaller = new File(repo, "Firefox Setup 47.0.1.exe");

        if (ffInstaller.exists()) {
            LOG.info("Firefoxはダウンロード済みです {}", ffInstaller.getAbsolutePath());
        } else {
            try {
                // TODO 外部化
                URL url = new URL(
                        "https://ftp.mozilla.org/pub/firefox/releases/47.0.1/win64/ja/Firefox%20Setup%2047.0.1.exe");

                LOG.info("Firefoxをダウンロードします {} -> {}", url, ffInstaller.getAbsolutePath());

                MultiThreadUtils.submitWithProgress(() -> {
                    FileUtils.copyURLToFile(url, ffInstaller);
                    return 0;
                });

            } catch (IOException e) {
                throw new RuntimeException("Firefoxのダウンロードで例外が発生しました", e);
            }
        }

        LOG.info("Firefoxをインストールします");

        try {
            File iniFile = File.createTempFile("ff-inst", ".ini");
            iniFile.deleteOnExit();

            FileUtils.copyInputStreamToFile(ClassLoader.getSystemResourceAsStream("ff-inst.ini"),
                    iniFile);

            MultiThreadUtils.submitWithProgress(() -> {
                ProcessUtils.execute(ffInstaller.getAbsolutePath(),
                        "/INI=" + iniFile.getAbsolutePath());
                return 0;
            });
        } catch (IOException e) {
            throw new RuntimeException("Firefoxのインストールで例外が発生しました", e);
        }

    }

    protected void installFirefoxMacOs(File repo) {

        File ffInstaller = new File(repo, "Firefox 47.0.1.dmg");

        if (ffInstaller.exists()) {
            LOG.info("Firefoxはダウンロード済みです {}", ffInstaller.getAbsolutePath());
        } else {
            try {
                // TODO 外部化
                URL url = new URL(
                        "https://ftp.mozilla.org/pub/firefox/releases/47.0.1/mac/ja-JP-mac/Firefox%2047.0.1.dmg");

                LOG.info("Firefoxをダウンロードします {} -> {}", url, ffInstaller.getAbsolutePath());

                FileUtils.copyURLToFile(url, ffInstaller);

                ProcessUtils.execute("chmod", "777", ffInstaller.getAbsolutePath());
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }

        LOG.info("Firefoxのインストーラーをマウントします");
        ProcessUtils.execute("hdiutil", "attach", ffInstaller.getAbsolutePath());

        File mountedFf = new File("/Volumes/Firefox/Firefox.app");
        File ffRuntime = new File(repo, "runtime");

        LOG.info("Firefoxをインストールします {}", ffRuntime.getAbsolutePath());
        // たまにcp -Rコマンドが失敗してFirefox.appがコピーされないので3回リトライ
        for (int i = 0; i < 3; i++) {

            MultiThreadUtils.submitWithProgress(() -> {
                ProcessUtils.execute("cp", "-R", mountedFf.getAbsolutePath(),
                        ffRuntime.getAbsolutePath());
                return 0;
            });

            if (new File(ffRuntime, "Firefox.app").exists()) {
                break;
            }
        }

        LOG.info("Firefoxのインストーラーをアンマウントします");
        ProcessUtils.execute("hdiutil", "detach", "/Volumes/Firefox");

    }

}