/*

  * Copyright 2009, Josh Devins, Jitr.org.

  *

  * 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 org.jitr.container;

 import java.io.File;

 import java.lang.annotation.Annotation;

 import javax.servlet.ServletContext;

 import org.jitr.core.Container;

 import org.jitr.core.ContainerOperationException;

 import org.jitr.core.ConfigurationModel;

 import org.mortbay.jetty.Connector;

 import org.mortbay.jetty.Server;

 import org.mortbay.jetty.nio.SelectChannelConnector;

 import org.mortbay.jetty.webapp.WebAppContext;

 /**

  * Jetty specific container wrapper.

  * 

  * @author Josh Devins (info@joshdevins.net)

  */

 public class JettyContainer implements Container<Server> {

     private final Server server;

     public JettyContainer() {

         server = new Server();

     }

     public String getName() {

         return "Jetty";

     }

     public Server getWrappedContainer() {

         return server;

     }

     public ServletContext initialize(final ConfigurationModel configuration,

             final Annotation[] testClassAnnotations) {

         final Connector connector = new SelectChannelConnector();

         connector.setPort(configuration.getPort());

         server.setConnectors(new Connector[] { connector });

         final WebAppContext webapp = new WebAppContext();

         webapp.setContextPath(configuration.getContextPath());

         webapp.setWar(configuration.getWarPath());

         webapp.setClassLoader(Thread.currentThread().getContextClassLoader());

         server.setHandler(webapp);

         final File tempDirectory = new File(configuration.getContainerWorkPath());

         webapp.setTempDirectory(tempDirectory);

         return webapp.getServletContext().getContext(configuration.getContextPath());

     }

     public void start() throws ContainerOperationException {

         try {

             server.start();

         } catch (final Exception e) {

             throw new ContainerOperationException("Error starting Jetty.", e, this);

         }

         // TODO: Is this a valid/useful assertion?

         // make sure Jetty is running

         if (!server.isRunning()) {

             throw new ContainerOperationException("Container could not be started.", this);

         }

     }

     public void stop() throws ContainerOperationException {

         // TODO: Could log something in this case I suspect.

         if (server.isStopped() || server.isStopping()) {

             return;

         }

         try {

             server.stop();

         } catch (final Exception e) {

             throw new ContainerOperationException("Error stopping Jetty.", e, this);

         }

     }

 }