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

 /*

  * 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.FileNotFoundException;

 import java.io.FileReader;

 import org.apache.maven.doxia.Doxia;

 import org.apache.maven.doxia.book.BookDoxiaException;

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

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

 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.index.IndexEntry;

 import org.apache.maven.doxia.index.IndexingSink;

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

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

 import org.codehaus.plexus.logging.AbstractLogEnabled;

 /**

  * Default implementation of BookIndexer.

  *

  * @plexus.component

  *

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

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

  */

 public class DefaultBookIndexer

     extends AbstractLogEnabled

     implements BookIndexer

 {

     /**

      * @plexus.requirement

      */

     private Doxia doxia;

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

     // BookIndexer Implementation

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

     /** {@inheritDoc} */

     public void indexBook( BookModel book, BookContext bookContext )

         throws BookDoxiaException

     {

         BookIndex index = new BookIndex();

         for ( Chapter chapter : book.getChapters() )

         {

             indexChapter( bookContext, index, chapter );

         }

         bookContext.setIndex( index );

     }

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

     // Private

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

     /**

      * Index a chapter.

      *

      * @param context the BookContext.

      * @param bookEntry the IndexEntry.

      * @param chapter the Chapter to index.

      * @throws BookDoxiaException if the chapter cannot be indexed.

      */

     private void indexChapter( BookContext context, IndexEntry bookEntry, Chapter chapter )

         throws BookDoxiaException

     {

         IndexEntry chapterEntry = new IndexEntry( bookEntry, chapter.getId( ) );

         chapterEntry.setTitle( chapter.getTitle() );

         for ( Section section : chapter.getSections() )

         {

             indexSection( context, chapterEntry, section );

         }

     }

     /**

      * Index a section.

      *

      * @param bookContext the BookContext.

      * @param chapterEntry the IndexEntry.

      * @param section the Section to index.

      * @throws BookDoxiaException if the section cannot be indexed.

      */

     private void indexSection( BookContext bookContext, IndexEntry chapterEntry, Section section )

         throws BookDoxiaException

     {

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

         if ( bookFile == null )

         {

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

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

         }

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

         //

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

         IndexEntry sectionEntry = new IndexEntry( chapterEntry, section.getId() );

         IndexingSink sink = new IndexingSink( sectionEntry );

         try

         {

             doxia.parse( new FileReader( bookFile.getFile() ), 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 );

         }

         sectionEntry.setTitle( sink.getTitle() );

     }

 }