io.selendroid.server.SelendroidStandaloneServer.java Source code

Introduction

Here is the source code for io.selendroid.server.SelendroidStandaloneServer.java

Source

/*
 * Copyright 2012-2013 eBay Software Foundation and selendroid committers.
 * 
 * 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 io.selendroid.server;

import io.selendroid.SelendroidConfiguration;
import io.selendroid.exceptions.AndroidDeviceException;
import io.selendroid.exceptions.AndroidSdkException;
import io.selendroid.server.grid.SelfRegisteringRemote;
import io.selendroid.server.model.SelendroidStandaloneDriver;

import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.URI;
import java.util.concurrent.Executors;
import java.util.logging.Logger;

import org.apache.commons.lang3.StringUtils;
import org.webbitserver.WebServer;
import org.webbitserver.WebServers;
import org.webbitserver.helpers.NamingThreadFactory;

public class SelendroidStandaloneServer {
    private static final Logger log = Logger.getLogger(SelendroidStandaloneServer.class.getName());
    private WebServer webServer;
    private SelendroidConfiguration configuration;
    private SelendroidStandaloneDriver driver = null;

    /**
     * for testing only
     * 
     * @throws AndroidSdkException
     */
    protected SelendroidStandaloneServer(SelendroidConfiguration configuration, SelendroidStandaloneDriver driver)
            throws AndroidSdkException {
        this.configuration = configuration;
        this.driver = driver;
        NamingThreadFactory namingThreadFactory = new NamingThreadFactory(Executors.defaultThreadFactory(),
                "selendroid-standalone-handler");
        webServer = WebServers.createWebServer(Executors.newCachedThreadPool(namingThreadFactory),
                new InetSocketAddress(configuration.getPort()), URI.create("http://127.0.0.1"
                        + (configuration.getPort() == 80 ? "" : (":" + configuration.getPort())) + "/"));
        init();
    }

    public SelendroidStandaloneServer(SelendroidConfiguration configuration)
            throws AndroidSdkException, AndroidDeviceException {
        this.configuration = configuration;
        NamingThreadFactory namingThreadFactory = new NamingThreadFactory(Executors.defaultThreadFactory(),
                "selendroid-standalone-handler");
        webServer = WebServers.createWebServer(Executors.newCachedThreadPool(namingThreadFactory),
                new InetSocketAddress(configuration.getPort()), remoteUri(configuration.getPort()));
        driver = initializeSelendroidServer();
        init();
    }

    private static URI remoteUri(int port) {
        try {
            InetAddress address = InetAddress.getByName("0.0.0.0");

            return new URI("http://" + address.getHostAddress() + (port == 80 ? "" : (":" + port)) + "/");
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("can not create URI from HostAddress", e);
        }
    }

    protected void init() throws AndroidSdkException {
        // just make sure the connection will not be staled because
        // the long emulator starting time and therefore long time
        // it needs to create a session
        webServer.staleConnectionTimeout(configuration.getTimeoutEmulatorStart());
        webServer.add("/wd/hub/status", new StatusServlet(driver));
        webServer.add(new SelendroidServlet(driver, configuration));
    }

    protected SelendroidStandaloneDriver initializeSelendroidServer()
            throws AndroidSdkException, AndroidDeviceException {
        return new SelendroidStandaloneDriver(configuration);
    }

    public void start() {
        webServer.start();
        if (StringUtils.isBlank(configuration.getRegistrationUrl()) == false
                && StringUtils.isBlank(configuration.getServerHost()) == false) {
            try {
                new SelfRegisteringRemote(configuration, driver).performRegistration();
            } catch (Exception e) {
                log.severe("An error occured while registering selendroid into grid hub.");
                e.printStackTrace();
            }
        }
        log.info("selendroid-standalone server has been started on port: " + configuration.getPort());
    }

    public void stop() {
        log.info("About to stop selendroid-standalone server");
        driver.quitSelendroid();
        webServer.stop();
    }

    public int getPort() {
        return webServer.getPort();
    }

    protected SelendroidStandaloneDriver getDriver() {
        return driver;
    }
}