package org.apache.maven.plugins.site;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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.
   */
  
  import java.io.File;
  import java.io.FileNotFoundException;
  import java.io.FileOutputStream;
  import java.io.IOException;
  import java.io.InputStream;
  
  import java.util.HashMap;
  import java.util.List;
  import java.util.Locale;
  import java.util.Map;
  
  import org.apache.maven.doxia.siterenderer.DocumentRenderer;
  import org.apache.maven.doxia.siterenderer.SiteRenderingContext;
  import org.apache.maven.plugin.MojoExecutionException;
  import org.apache.maven.plugin.MojoFailureException;
  import org.apache.maven.plugins.annotations.Mojo;
  import org.apache.maven.plugins.annotations.Parameter;
  import org.apache.maven.plugins.site.webapp.DoxiaBean;
  import org.apache.maven.plugins.site.webapp.DoxiaFilter;
  import org.apache.maven.reporting.exec.MavenReportExecution;
  
  import org.codehaus.plexus.util.IOUtil;
  
  import org.mortbay.jetty.Connector;
  import org.mortbay.jetty.Handler;
  import org.mortbay.jetty.Server;
  import org.mortbay.jetty.handler.DefaultHandler;
  import org.mortbay.jetty.nio.SelectChannelConnector;
  import org.mortbay.jetty.webapp.WebAppContext;
  
  /**
   * Starts the site up, rendering documents as requested for faster editing.
   * It uses Jetty as the web server.
   *
   * @author <a href="mailto:brett@apache.org">Brett Porter</a>
   * @version $Id: SiteRunMojo.html 861484 2013-05-09 23:10:16Z hboutemy $
   */
  @Mojo( name = "run", aggregator = true, requiresReports = true )
  public class SiteRunMojo
      extends AbstractSiteRenderingMojo
  {
      /**
       * Where to create the dummy web application.
       */
      @Parameter( defaultValue = "${project.build.directory}/site-webapp" )
      private File tempWebappDirectory;
  
      /**
       * The port to execute the HTTP server on.
       *
       */
      @Parameter(property = "port", defaultValue = "8080")
      private int port;
  
      private static final int MAX_IDLE_TIME = 30000;
  
      /**
       * @see org.apache.maven.plugin.AbstractMojo#execute()
       */
      public void execute()
          throws MojoExecutionException, MojoFailureException
      {
          Server server = new Server();
          server.setStopAtShutdown( true );
  
          Connector defaultConnector = getDefaultConnector();
          server.setConnectors( new Connector[] { defaultConnector } );
  
          WebAppContext webapp = createWebApplication();
          webapp.setServer( server );
  
          DefaultHandler defaultHandler = new DefaultHandler();
          defaultHandler.setServer( server );
  
          Handler[] handlers = new Handler[2];
          handlers[0] = webapp;
          handlers[1] = defaultHandler;
          server.setHandlers( handlers );
 
         getLog().info( "Starting Jetty on http://localhost:" + port + "/" );
         try
         {
             server.start();
         }
         catch ( Exception e )
         {
             throw new MojoExecutionException( "Error executing Jetty: " + e.getMessage(), e );
         }
 
         // Watch it
         try
         {
             server.getThreadPool().join();
         }
         catch ( InterruptedException e )
         {
             getLog().warn( "Jetty was interrupted", e );
         }
     }
 
     private WebAppContext createWebApplication()
         throws MojoExecutionException
     {
         File webXml = new File( tempWebappDirectory, "WEB-INF/web.xml" );
         webXml.getParentFile().mkdirs();
 
         InputStream inStream = null;
         FileOutputStream outStream = null;
         try
         {
             inStream = getClass().getResourceAsStream( "/webapp/web.xml" );
             outStream = new FileOutputStream( webXml );
             IOUtil.copy( inStream, outStream );
         }
         catch ( FileNotFoundException e )
         {
             throw new MojoExecutionException( "Unable to construct temporary webapp for running site", e );
         }
         catch ( IOException e )
         {
             throw new MojoExecutionException( "Unable to construct temporary webapp for running site", e );
         }
         finally
         {
             IOUtil.close( outStream );
             IOUtil.close( inStream );
         }
 
         WebAppContext webapp = new WebAppContext();
         webapp.setContextPath( "/" );
         webapp.setResourceBase( tempWebappDirectory.getAbsolutePath() );
         webapp.setAttribute( DoxiaFilter.SITE_RENDERER_KEY, siteRenderer );
         webapp.getInitParams().put( "org.mortbay.jetty.servlet.Default.useFileMappedBuffer", "false" );
 
         // For external reports
         project.getReporting().setOutputDirectory( tempWebappDirectory.getAbsolutePath() );
         for ( MavenReportExecution mavenReportExecution : getReports() )
         {
             mavenReportExecution.getMavenReport().setReportOutputDirectory( tempWebappDirectory );
         }
 
         List<MavenReportExecution> reports = getReports(); // TODO: is it sane to call getReports() method a second time?
 
         List<Locale> localesList = siteTool.getAvailableLocales( locales );
         webapp.setAttribute( DoxiaFilter.LOCALES_LIST_KEY, localesList );
 
         // Default is first in the list
         Locale defaultLocale = localesList.get( 0 );
         Locale.setDefault( defaultLocale );
 
         try
         {
             Map<String, DoxiaBean> i18nDoxiaContexts = new HashMap<String, DoxiaBean>();
 
             for ( Locale locale : localesList )
             {
                 SiteRenderingContext i18nContext = createSiteRenderingContext( locale );
                 i18nContext.setInputEncoding( getInputEncoding() );
                 i18nContext.setOutputEncoding( getOutputEncoding() );
 
                 Map<String, DocumentRenderer> i18nDocuments = locateDocuments( i18nContext, reports, locale );
                 DoxiaBean doxiaBean;
                 if ( defaultLocale.equals( locale ) )
                 {
                     doxiaBean = new DoxiaBean( i18nContext, i18nDocuments, generatedSiteDirectory );
                 }
                 else
                 {
                     doxiaBean =
                         new DoxiaBean( i18nContext, i18nDocuments, new File( generatedSiteDirectory,
                                                                              locale.getLanguage() ) );
                 }
 
                 i18nDoxiaContexts.put( locale.getLanguage(), doxiaBean );
                 if ( defaultLocale.equals( locale ) )
                 {
                     i18nDoxiaContexts.put( "default", doxiaBean );
                 }
 
                 if ( defaultLocale.equals( locale ) )
                 {
                     siteRenderer.copyResources( i18nContext, new File( siteDirectory, "resources" ),
                                                 tempWebappDirectory );
                 }
                 else
                 {
                     siteRenderer.copyResources( i18nContext, new File( siteDirectory, "resources" ),
                                                 new File( tempWebappDirectory, locale.getLanguage() ) );
                 }
             }
 
             webapp.setAttribute( DoxiaFilter.I18N_DOXIA_CONTEXTS_KEY, i18nDoxiaContexts );
         }
         catch ( Exception e )
         {
             throw new MojoExecutionException( "Unable to set up webapp", e );
         }
         return webapp;
     }
 
     private Connector getDefaultConnector()
     {
         Connector connector = new SelectChannelConnector();
         connector.setPort( port );
         connector.setMaxIdleTime( MAX_IDLE_TIME );
         return connector;
     }
 
     public void setTempWebappDirectory( File tempWebappDirectory )
     {
         this.tempWebappDirectory = tempWebappDirectory;
     }
 
     public void setPort( int port )
     {
         this.port = port;
     }
 }