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 org.apache.maven.doxia.sink.Sink;

 import org.apache.maven.doxia.sink.SinkFactory;

 import org.apache.maven.doxia.sink.render.RenderingContext;

 import org.apache.maven.doxia.siterenderer.DocumentRenderer;

 import org.apache.maven.doxia.siterenderer.Renderer;

 import org.apache.maven.doxia.siterenderer.RendererException;

 import org.apache.maven.doxia.siterenderer.SiteRenderingContext;

 import org.apache.maven.doxia.siterenderer.sink.SiteRendererSink;

 import org.apache.maven.doxia.tools.MojoLogWrapper;

 import org.apache.maven.plugin.logging.Log;

 import org.apache.maven.reporting.MavenReport;

 import org.apache.maven.reporting.MavenMultiPageReport;

 import org.apache.maven.reporting.MavenReportException;

 import java.io.FileNotFoundException;

 import java.io.IOException;

 import java.io.OutputStream;

 import java.io.Writer;

 import java.io.File;

 import java.io.FileWriter;

 import java.util.ArrayList;

 import java.util.Locale;

 import java.util.List;

 import java.util.Iterator;

 /**

  * Renders a Maven report.

  *

  * @author <a href="mailto:brett@apache.org">Brett Porter</a>

  */

 public class ReportDocumentRenderer

     implements DocumentRenderer

 {

     private MavenReport report;

     private RenderingContext renderingContext;

     private Log log;

     public ReportDocumentRenderer( MavenReport report, RenderingContext renderingContext, Log log )

     {

         this.report = report;

         this.renderingContext = renderingContext;

         this.log = log;

     }

     private static class MySink

         extends SiteRendererSink

     {

         private File outputDir;

         private String outputName;

         public MySink( File outputDir, String outputName, RenderingContext ctx )

         {

             super( ctx );

             this.outputName = outputName;

             this.outputDir = outputDir;

         }

         public String getOutputName()

         {

             return outputName;

         }

         public File getOutputDir()

         {

             return outputDir;

         }

     }

     private static class MySinkFactory

         implements SinkFactory

     {

         private RenderingContext context;

         private List sinks = new ArrayList();

         public MySinkFactory( RenderingContext ctx )

         {

             this.context = ctx;

         }

         public Sink createSink( File outputDir, String outputName )

         {

             SiteRendererSink sink = new MySink( outputDir, outputName, context );

             sinks.add( sink );

             return sink;

         }

         public Sink createSink( File arg0, String arg1, String arg2 )

             throws IOException

         {

             // Not used

             return null;

         }

         public Sink createSink( OutputStream arg0 )

             throws IOException

         {

             // Not used

             return null;

         }

         public Sink createSink( OutputStream arg0, String arg1 )

             throws IOException

         {

             // Not used

             return null;

         }

         public List sinks()

         {

             return sinks;

         }

     }

     public void renderDocument( Writer writer, Renderer renderer, SiteRenderingContext siteRenderingContext )

         throws RendererException, FileNotFoundException

     {

         Locale locale = siteRenderingContext.getLocale();

         String localReportName = report.getName( locale );

         log.info( "Generating \"" + localReportName + "\" report." );

         MySinkFactory sf = new MySinkFactory( renderingContext );

         SiteRendererSink sink = new SiteRendererSink( renderingContext );

         sink.enableLogging( new MojoLogWrapper( log ) );

         try

         {

             if ( report instanceof MavenMultiPageReport )

             {

                 ( (MavenMultiPageReport) report ).generate( sink, sf, locale );

             }

             else

             {

                 try

                 {

                     report.generate( sink, locale );

                 }

                 catch ( NoSuchMethodError e )

                 {

                     throw new RendererException( "No method on " + report.getClass(), e );

                 }

             }

         }

         catch ( MavenReportException e )

         {

             throw new RendererException( "Error rendering Maven report: " + e.getMessage(), e );

         }

         finally

         {

             sink.close();

         }

         if ( !report.isExternalReport() )

         {

             try

             {

                 List sinks = sf.sinks();

                 log.debug( "Multipage report: " + sinks.size() + " subreports" );

                 for ( Iterator it = sinks.iterator(); it.hasNext(); )

                 {

                     MySink mySink = (MySink) it.next();

                     mySink.enableLogging( new MojoLogWrapper( log ) );

                     log.debug( "  Rendering " + mySink.getOutputName() );

                     Writer out = new FileWriter( new File( mySink.getOutputDir(), mySink.getOutputName() ) );

                     try

                     {

                         renderer.generateDocument( out, mySink, siteRenderingContext );

                     }

                     finally

                     {

                         mySink.close();

                     }

                 }

             }

             catch ( IOException e )

             {

                 throw new RendererException( "Cannot create writer", e );

             }

             renderer.generateDocument( writer, sink, siteRenderingContext );

         }

     }

     public String getOutputName()

     {

         return renderingContext.getOutputName();

     }

     public RenderingContext getRenderingContext()

     {

         return renderingContext;

     }

     public boolean isOverwrite()

     {

         // TODO: would be nice to query the report to see if it is modified

         return true;

     }

     /**

      * @return true if the current report is external, false otherwise

      */

     public boolean isExternalReport()

     {

         return report.isExternalReport();

     }

 }