com.vilt.minium.script.impl.PhantomJsDriverFactory.java Source code

Introduction

Here is the source code for com.vilt.minium.script.impl.PhantomJsDriverFactory.java

Source

/*
 * Copyright (C) 2013 The Minium Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.vilt.minium.script.impl;

import java.io.File;
import java.io.IOException;

import org.openqa.selenium.Platform;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.phantomjs.PhantomJSDriverService;
import org.openqa.selenium.remote.Augmenter;
import org.openqa.selenium.remote.BrowserType;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.base.Objects;
import com.google.common.base.Preconditions;
import com.google.common.base.Throwables;
import com.vilt.minium.prefs.AppPreferences;

public class PhantomJsDriverFactory implements WebDriverFactory {

    private static final Logger logger = LoggerFactory.getLogger(PhantomJsDriverFactory.class);

    private PhantomJSDriverService service;
    private AppPreferences preferences;

    public PhantomJsDriverFactory(AppPreferences preferences) {
        this.preferences = preferences;
    }

    @Override
    public boolean supports(DesiredCapabilities capabilities) {
        return Objects.equal(BrowserType.IE, capabilities.getBrowserName());
    }

    @Override
    public WebDriver create(DesiredCapabilities capabilities) {
        Preconditions.checkArgument(Objects.equal(BrowserType.PHANTOMJS, capabilities.getBrowserName()));
        maybeInitPhantomJsDriverService();
        WebDriver driver = new RemoteWebDriver(service.getUrl(), capabilities);
        return new Augmenter().augment(driver);
    }

    @Override
    public void destroy() {
        if (service != null) {
            logger.debug("Stopping PhantomJS driver service: {}", service.getUrl());
            service.stop();
        }
    }

    protected void maybeInitPhantomJsDriverService() {
        try {
            if (service == null) {
                String exe = Platform.getCurrent().is(Platform.WINDOWS) ? "phantomjs.exe" : "phantomjs";
                File phantomJsBin = new File(new File(preferences.getBaseDir(), "drivers"), exe);

                if (phantomJsBin.isFile() && phantomJsBin.canExecute()) {
                    service = new PhantomJSDriverService.Builder().usingPhantomJSExecutable(phantomJsBin)
                            .usingAnyFreePort().build();
                } else {
                    service = PhantomJSDriverService.createDefaultService();
                }
                service.start();
                logger.debug("PhantomJS driver service initialized: {}", service.getUrl());
            }
        } catch (IOException e) {
            throw Throwables.propagate(e);
        }
    }
}