com.crawljax.web.CrawljaxServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of crawljax-web Show documentation
The newest version!
package com.crawljax.web;

import java.io.File;
import java.util.concurrent.Callable;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import ch.qos.logback.classic.LoggerContext;
import com.crawljax.web.di.CrawljaxWebModule;
import com.google.inject.Guice;
import com.google.inject.Injector;
import com.google.inject.servlet.GuiceFilter;
import com.google.inject.servlet.GuiceServletContextListener;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.ServerConnector;
import org.eclipse.jetty.server.handler.HandlerList;
import org.eclipse.jetty.util.resource.Resource;
import org.eclipse.jetty.webapp.WebAppContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.bridge.SLF4JBridgeHandler;

public class CrawljaxServer implements Callable {

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

	private Server server;
	private final CountDownLatch isRunningLatch = new CountDownLatch(1);
	private String url = null;

	public CrawljaxServer(CrawljaxServerConfigurationBuilder configurationBuilder) {
		LoggerContext loggerContext = (LoggerContext) LoggerFactory.getILoggerFactory();
		loggerContext.putProperty("output_folder", configurationBuilder.getOutputDir().getAbsolutePath());
		setupJulToSlf4();
		server = new Server(configurationBuilder.getPort());
		HandlerList handlerList = new HandlerList();
		handlerList.addHandler(setupOutputContext(configurationBuilder.getOutputDir()));
		handlerList.addHandler(setupWebContext(configurationBuilder.getOutputDir(), configurationBuilder.getPluginDir()));
		server.setHandler(handlerList);
	}

	@Override
	public Void call() {
		start(true);
		return null;
	}

	/**
	 * @param join
	 *            If you want to merge with the remote thread.
	 * @see Thread#join()
	 */
	public void start(boolean join) {
		LOG.info("Starting the server");
		try {
			server.start();

		} catch (Exception e) {
			throw new RuntimeException("Could not start Crawljax web server", e);
		}

		int port = ((ServerConnector) server.getConnectors()[0]).getLocalPort(); //Value will change if originally set to 0
		url = "http://localhost:" + port;
		LOG.info("Server started at {}", url);
		isRunningLatch.countDown();

		if(join) {
			LOG.debug("Joining server thread");
			try {
				server.join();
			} catch (InterruptedException e) {
				throw new RuntimeException("Crawljax server interrupted", e);
			}
		}
	}

	public void waitUntilRunning(long timeOut_ms) {
		try {
			isRunningLatch.await(timeOut_ms, TimeUnit.MILLISECONDS);
		} catch (InterruptedException e) {
			throw new RuntimeException(e);
		}
	}

	public void stop() {
		LOG.info("Stopping the server");
		try {
			server.stop();
			LOG.info("Shutdown complete");
		} catch (Exception e) {
			throw new RuntimeException("Could not stop Crawljax web server", e);
		}
	}

	public String getUrl() {
		return url;
	}

	private void setupJulToSlf4() {
		SLF4JBridgeHandler.removeHandlersForRootLogger();
		SLF4JBridgeHandler.install();
	}

	private WebAppContext setupWebContext(final File outputFolder, final File pluginsFolder) {
		WebAppContext webAppContext = new WebAppContext();
		webAppContext.setContextPath("/");
		webAppContext.setBaseResource(Resource.newClassPathResource("web"));
		webAppContext.setParentLoaderPriority(true);
		webAppContext.addEventListener(new GuiceServletContextListener() {

			@Override
			protected Injector getInjector() {
				return Guice.createInjector(new CrawljaxWebModule(outputFolder, pluginsFolder));
			}

		});

		webAppContext.addFilter(GuiceFilter.class, "/*", null);
		return webAppContext;
	}

	private WebAppContext setupOutputContext(final File outputFolder) {
		WebAppContext webAppContext = new WebAppContext();
		webAppContext.setContextPath("/output");
		webAppContext.setBaseResource(Resource.newResource(outputFolder));
		return webAppContext;
	}

}