package org.apache.maven.doxia.book.services.renderer;

 /*

  * 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.book.BookDoxiaException;

 import org.apache.maven.doxia.book.services.renderer.latex.LatexBookSink;

 import org.apache.maven.doxia.book.context.BookContext;

 import org.apache.maven.doxia.book.model.BookModel;

 import org.apache.maven.doxia.book.model.Chapter;

 import org.apache.maven.doxia.book.model.Section;

 import org.apache.maven.doxia.parser.manager.ParserNotFoundException;

 import org.apache.maven.doxia.parser.ParseException;

 import org.apache.maven.doxia.Doxia;

 import org.codehaus.plexus.util.ReaderFactory;

 import org.codehaus.plexus.util.StringUtils;

 import org.codehaus.plexus.util.IOUtil;

 import org.codehaus.plexus.util.WriterFactory;

 import java.io.File;

 import java.io.FileWriter;

 import java.io.IOException;

 import java.io.PrintWriter;

 import java.io.FileNotFoundException;

 import java.io.Reader;

 import java.io.Writer;

 import java.util.Map;

 import java.util.HashMap;

 /**

  * <p>LatexBookRenderer class.</p>

  *

  * @plexus.component role-hint="latex"

  * @author <a href="mailto:trygvis@inamo.no">Trygve Laugst&oslash;l</a>

  * @version $Id: LatexBookRenderer.java 1090706 2011-04-09 23:15:28Z hboutemy $

  */

 public class LatexBookRenderer

     implements BookRenderer

 {

     /**

      * @plexus.requirement

      */

     private Doxia doxia;

     // ----------------------------------------------------------------------

     // BookRenderer Implementatino

     // ----------------------------------------------------------------------

     /** {@inheritDoc} */

     public void renderBook( BookContext context )

         throws BookDoxiaException

     {

         BookModel book = context.getBook();

         if ( !context.getOutputDirectory().exists() )

         {

             if ( !context.getOutputDirectory().mkdirs() )

             {

                 throw new BookDoxiaException(

                     "Could not make directory: " + context.getOutputDirectory().getAbsolutePath() + "." );

             }

         }

         File bookFile = new File( context.getOutputDirectory(), book.getId() + ".tex" );

         FileWriter fileWriter = null;

         try

         {

             fileWriter = new FileWriter( bookFile );

             PrintWriter writer = new PrintWriter( fileWriter );

             writeBook( book, context, writer );

         }

         catch ( IOException e )

         {

             throw new BookDoxiaException( "Error while opening file.", e );

         }

         finally

         {

             IOUtil.close( fileWriter );

         }

     }

     // ----------------------------------------------------------------------

     // Private

     // ----------------------------------------------------------------------

     /** SectionInfo: id and title. */

     static class SectionInfo

     {

         /** id. */

         String id;

         /** title. */

         String title;

     }

     /**

      * Write a book.

      *

      * @param book the BookModel to write.

      * @param context the BookContext.

      * @param writer the writer to use.

      * @throws IOException if any.

      * @throws BookDoxiaException if the section cannot be written.

      */

     private void writeBook( BookModel book, BookContext context, PrintWriter writer )

         throws IOException, BookDoxiaException

     {

         // ----------------------------------------------------------------------

         // Process all the section documents and collect their names

         // ----------------------------------------------------------------------

         Map<String, SectionInfo> sectionInfos = new HashMap<String, SectionInfo>();

         for ( Chapter chapter : book.getChapters() )

         {

             for ( Section section : chapter.getSections() )

             {

                 SectionInfo info = writeSection( section, context );

                 sectionInfos.put( info.id, info );

             }

         }

         // ----------------------------------------------------------------------

         // Write the main .tex file

         // ----------------------------------------------------------------------

         writer.println( "\\documentclass{book}" );

         writer.println( "\\title{" + book.getTitle() + "}" );

         if ( StringUtils.isNotEmpty( book.getAuthor() ) )

         {

             writer.println( "\\author{" + book.getAuthor() + "}" );

         }

         if ( StringUtils.isNotEmpty( book.getDate() ) )

         {

             writer.println( "\\date{" + book.getDate() + "}" );

         }

         LatexBookSink sink = new LatexBookSink( writer );

         sink.defaultBookPreamble();

         writer.println( "\\begin{document}" );

         writer.println( "\\maketitle" );

         writer.println( "\\tableofcontents" );

 //        writer.println( "\\listoffigures" );

         for ( Chapter chapter : book.getChapters() )

         {

             writer.println( "\\chapter{" + chapter.getTitle() + "}" );

             for ( Section section : chapter.getSections() )

             {

                 SectionInfo info = sectionInfos.get( section.getId() );

                 writer.println( "\\input{" + info.id + "}" );

             }

         }

         writer.println( "\\end{document}" );

     }

     /**

      * Write a section.

      *

      * @param section the Section to write.

      * @param context the BookContext.

      * @return SectionInfo

      * @throws IOException if any.

      * @throws BookDoxiaException if the section cannot be written.

      */

     private SectionInfo writeSection( Section section, BookContext context )

         throws IOException, BookDoxiaException

     {

         File file = new File( context.getOutputDirectory(), ( section.getId() + ".tex" ) );

         Writer writer = WriterFactory.newWriter( file, context.getOutputEncoding() );

         LatexBookSink sink = new LatexBookSink( writer );

         BookContext.BookFile bookFile = (BookContext.BookFile) context.getFiles().get( section.getId() );

         if ( bookFile == null )

         {

             throw new BookDoxiaException( "No document that matches section with id="

                         + section.getId() + "." );

         }

         Reader reader = null;

         try

         {

             reader = ReaderFactory.newReader( bookFile.getFile(), context.getInputEncoding() );

             doxia.parse( reader, bookFile.getParserId(), sink );

         }

         catch ( ParserNotFoundException e )

         {

             throw new BookDoxiaException( "Parser not found: "

                         + bookFile.getParserId() + ".", e );

         }

         catch ( ParseException e )

         {

             throw new BookDoxiaException( "Error while parsing document: "

                         + bookFile.getFile().getAbsolutePath() + ".", e );

         }

         catch ( FileNotFoundException e )

         {

             throw new BookDoxiaException( "Could not find document: "

                         + bookFile.getFile().getAbsolutePath() + ".", e );

         }

         finally

         {

             sink.flush();

             sink.close();

             IOUtil.close( reader );

             IOUtil.close( writer );

         }

         SectionInfo info = new SectionInfo();

         info.id = section.getId();

         info.title = sink.getTitle();

         return info;

     }

 }