plugins/org.eclipse.wst.server.preview/src/org/eclipse/wst/server/preview/internal/PreviewStarter.java

/*******************************************************************************
 * Copyright (c) 2007, 2008 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     IBM Corporation - Initial API and implementation
 *******************************************************************************/
package org.eclipse.wst.server.preview.internal;

import java.io.File;
import java.util.logging.Filter;
import java.util.logging.LogRecord;
import java.util.logging.Logger;

import org.mortbay.http.HttpContext;
import org.mortbay.http.handler.ResourceHandler;
import org.mortbay.jetty.*;
import org.mortbay.jetty.servlet.WebApplicationContext;

public class PreviewStarter {
	private static final String[] AVERTED_LOGS = new String[] {
		"org.mortbay.util.Container", "org.mortbay.http.HttpServer",
		"org.mortbay.util.Credential", "org.mortbay.http.SocketListener",
		"org.mortbay.http.HttpServer", "org.mortbay.jetty.Server"
	};

	protected String configPath;
	protected Server server;

	public PreviewStarter(String configPath) {
		this.configPath = configPath;
	}

	public static void main(String[] args) {
		PreviewStarter app = new PreviewStarter(args[0]);
		app.run();
	}

	protected void run() {
		System.out.println("Starting preview server");
		System.out.println();
		try {
			ServerConfig config = new ServerConfig(configPath);
			System.out.println("Port " + config.getPort());
			Module[] m = config.getModules();
			int size = m.length;
			if (size > 0) {
				System.out.println("Modules:");
				for (Module mm : m)
					System.out.println("  " + mm.getName());
				System.out.println();
			}
			
			for (String log : AVERTED_LOGS) {
				Logger logger = Logger.getLogger(log);
				logger.setFilter(new Filter() {
					public boolean isLoggable(LogRecord record) {
						//Trace.trace(Trace.FINEST, "Averted Jetty log: " + record.getMessage());
						//System.out.println("averted: " + record.getLoggerName() + ": " + record.getMessage());
						return false;
					}
				});
			}
			
			// helper code to find jetty loggers
			/*Logger logger = Logger.getLogger("org.mortbay.http.HttpServer");
			logger.addHandler(new java.util.logging.Handler() {
				public void close() throws SecurityException {
					// ignore
				}

				public void flush() {
					// ignore
				}

				public void publish(LogRecord record) {
					System.out.println("Logger found: " + record.getLoggerName());
				}
			});*/
			
			server = new Server();
			server.addListener(":" + config.getPort());
			server.setTrace(false);
			server.setStatsOn(false);
			
			HttpContext context2 = new HttpContext();
			context2.setContextPath("/");
			context2.addHandler(new WTPErrorPageHandler());
			context2.setAttribute(HttpContext.__ErrorHandler, new WTPErrorPageHandler());
			server.addContext(context2);
			server.setRootWebApp("/");
			
			for (Module module : m) {
				if (module.isStaticWeb()) {
					HttpContext context = new HttpContext();
					context.setContextPath(module.getContext());
					context.setResourceBase(module.getPath());
					context.addHandler(new ResourceHandler());
					context.addHandler(new WTPErrorPageHandler());
					context.setAttribute(HttpContext.__ErrorHandler, new WTPErrorPageHandler());
					server.addContext(context);
				} else {
					WebApplicationContext context = server.addWebApplication(module.getContext(), module.getPath());
					//context.getWebApplicationHandler();
					//context.addHandler(new WTPErrorPageHandler());
					context.setAttribute(HttpContext.__ErrorHandler, new WTPErrorPageHandler());
				}
			}
			
			try {
				server.start();
			} catch (Exception e) {
				e.printStackTrace();
			}
		} catch (Throwable e) {
			e.printStackTrace();
		}
	}

	public void stop() {
		try {
			System.out.println("Stop!");
			server.stop();
			//File contextWorkDir = new File(workDir, DIR_PREFIX + pid.hashCode());
			//deleteDirectory(contextWorkDir);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * deleteDirectory is a convenience method to recursively delete a directory
	 * @param directory - the directory to delete.
	 * @return was the delete succesful
	 */
	protected static boolean deleteDirectory(File directory) {
		if (directory.exists() && directory.isDirectory()) {
			File[] files = directory.listFiles();
			for (File file : files) {
				if (file.isDirectory()) {
					deleteDirectory(file);
				} else {
					file.delete();
				}
			}
		}
		return directory.delete();
	}
}