package org.apache.maven.doxia.book.services.io;
   
   /*
    * 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.model.io.xpp3.BookModelXpp3Reader;
  import org.apache.maven.doxia.book.model.BookModel;
  import org.apache.maven.doxia.book.BookDoxiaException;
  import org.apache.maven.doxia.book.context.BookContext;
  import org.apache.maven.doxia.module.site.SiteModule;
  import org.apache.maven.doxia.module.site.manager.SiteModuleManager;
  import org.codehaus.plexus.util.IOUtil;
  import org.codehaus.plexus.util.ReaderFactory;
  import org.codehaus.plexus.util.xml.pull.XmlPullParserException;
  import org.codehaus.plexus.logging.AbstractLogEnabled;
  
  import java.io.IOException;
  import java.io.File;
  import java.io.Reader;
  import java.util.Collection;
  import java.util.Map;
  import java.util.TreeMap;
  import java.util.List;
  
  /**
   * <p>DefaultBookIo class.</p>
   *
   * @plexus.component
   * @author <a href="mailto:trygve.laugstol@objectware.no">Trygve Laugst&oslash;l</a>
   * @version $Id: DefaultBookIo.java 1090706 2011-04-09 23:15:28Z hboutemy $
   */
  public class DefaultBookIo
      extends AbstractLogEnabled
      implements BookIo
  {
      /**
       * @plexus.requirement
       */
      private SiteModuleManager siteModuleManager;
  
      // -----------------------------------------------------------------------
      // DefaultBookIo Implementation
      // -----------------------------------------------------------------------
  
      /** {@inheritDoc} */
      public BookModel readBook( File bookDescriptor )
          throws BookDoxiaException
      {
          Reader reader = null;
          try
          {
              reader = ReaderFactory.newXmlReader( bookDescriptor );
              return new BookModelXpp3Reader().read( reader, true );
          }
          catch ( IOException e )
          {
              throw new BookDoxiaException( "Error while reading book descriptor.", e );
          }
          catch ( XmlPullParserException e )
          {
              throw new BookDoxiaException( "Error while reading book descriptor.", e );
          }
          finally
          {
              IOUtil.close( reader );
          }
      }
  
      /** {@inheritDoc} */
      public void loadFiles( BookContext context, List<File> files )
      {
          // ----------------------------------------------------------------------
          // Find all the files, map the file names to ids
          // ----------------------------------------------------------------------
  
          Collection<SiteModule> siteModules = siteModuleManager.getSiteModules();
  
          for ( SiteModule siteModule : siteModules )
          {
              String extension = siteModule.getExtension();
  
              String sourceDirectory = File.separator + siteModule.getSourceDirectory() + File.separator;
  
             String parserId = siteModule.getParserId();
 
             for ( File file : files )
             {
                 String name = file.getName();
 
                 String path = file.getAbsolutePath();
 
                 // first check if the file path contains one of the recognized source dir identifiers
                 // (there's trouble if a pathname contains 2 identifiers), then match file extensions (not unique).
 
                 if ( path.indexOf( sourceDirectory ) != -1 )
                 {
                     name = name.substring( 0, name.length() - extension.length() - 1 );
 
                     context.getFiles().put( name, new BookContext.BookFile( file, parserId ) );
                 }
                 else if ( name.endsWith( extension ) )
                 {
                     name = name.substring( 0, name.length() - extension.length() - 1 );
 
                     // don't overwrite if it's there already
                     if ( !context.getFiles().containsKey( name ) )
                     {
                         context.getFiles().put( name, new BookContext.BookFile( file, parserId ) );
                     }
                 }
             }
         }
 
         if ( getLogger().isDebugEnabled() )
         {
             getLogger().debug( "Dumping document <-> id mapping:" );
 
             Map<String, BookContext.BookFile> map = new TreeMap<String, BookContext.BookFile>( context.getFiles() );
 
             for ( Map.Entry<String, BookContext.BookFile> entry : map.entrySet() )
             {
                 BookContext.BookFile file = entry.getValue();
 
                 getLogger().debug( " " + entry.getKey() + "=" + file.getFile() + ", parser: " + file.getParserId() );
             }
         }
     }
 }