package org.apache.maven.plugins.pdf;
/*
* 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.IOException;
import java.util.ArrayList;
import java.util.Date;
import java.util.Iterator;
import java.util.List;
import org.apache.maven.doxia.document.DocumentAuthor;
import org.apache.maven.doxia.document.DocumentCover;
import org.apache.maven.doxia.document.DocumentMeta;
import org.apache.maven.doxia.document.DocumentModel;
import org.apache.maven.doxia.document.DocumentTOC;
import org.apache.maven.doxia.document.DocumentTOCItem;
import org.apache.maven.doxia.site.decoration.DecorationModel;
import org.apache.maven.doxia.site.decoration.Menu;
import org.apache.maven.doxia.site.decoration.MenuItem;
import org.apache.maven.model.Developer;
import org.apache.maven.project.MavenProject;
import org.codehaus.plexus.util.IOUtil;
import org.codehaus.plexus.util.ReaderFactory;
import org.codehaus.plexus.util.StringUtils;
import org.apache.commons.io.input.XmlStreamReader;
/**
* Construct a DocumentModel from a MavenProject and related information.
*
* @author ltheussl
* @version $Id: DocumentModelBuilder.java 1396150 2012-10-09 18:15:29Z krosenvold $
*/
public class DocumentModelBuilder
{
/** A MavenProject to extract the information. */
private final MavenProject project;
/** A DecorationModel to extract additional information. */
private final DecorationModel decorationModel;
/**
* Constructor. Initialize a MavenProject to extract information from.
*
* @param project a MavenProject. May be null.
*/
public DocumentModelBuilder( MavenProject project )
{
this( project, null );
}
/**
* Constructor. Initialize a MavenProject and a DecorationModel to extract information from.
*
* @param project a MavenProject. May be null.
* @param decorationModel a DecorationModel. May be null.
*/
public DocumentModelBuilder( MavenProject project, DecorationModel decorationModel )
{
this.project = project;
this.decorationModel = decorationModel;
}
/**
* Get a DocumentModel.
*
* @return a DocumentModel. Always non-null.
*/
public DocumentModel getDocumentModel()
{
return getDocumentModel( project, decorationModel, null );
}
/**
* Get a DocumentModel.
*
* @param date overrides the default date in meta- and cover information.
* @return a DocumentModel. Always non-null.
*/
public DocumentModel getDocumentModel( Date date )
{
return getDocumentModel( project, decorationModel, date );
}
// ----------------------------------------------------------------------
// Private methods
// ----------------------------------------------------------------------
/**
* Extract a DocumentModel from a MavenProject.
*
* @param project a MavenProject. May be null.
* @param decorationModel a DecorationModel. May be null.
* @param date the date of the TOC. May be null in which case the build date will be used.
*
* @return a DocumentModel. Always non-null.
*/
private static DocumentModel getDocumentModel( MavenProject project,
DecorationModel decorationModel, Date date )
{
final Date now = ( date == null ? new Date() : date );
final DocumentModel docModel = new DocumentModel();
docModel.setModelEncoding( getProjectModelEncoding( project ) );
docModel.setOutputName( project == null || project.getArtifactId() == null
? "unnamed" : project.getArtifactId() );
docModel.setMeta( getDocumentMeta( project, now ) );
docModel.setCover( getDocumentCover( project, now ) );
docModel.setToc( getDocumentTOC( decorationModel ) );
return docModel;
}
/**
* Extract a DocumentTOC from a DecorationModel.
*
* @param decorationModel a DecorationModel. May be null.
* @return a DocumentTOC, always non-null.
*/
private static DocumentTOC getDocumentTOC( DecorationModel decorationModel )
{
final DocumentTOC toc = new DocumentTOC();
if ( decorationModel != null && decorationModel.getMenus() != null )
{
for ( final Iterator it = decorationModel.getMenus().iterator(); it.hasNext(); )
{
final Menu menu = (Menu) it.next();
for ( final Iterator it2 = menu.getItems().iterator(); it2.hasNext(); )
{
final MenuItem item = (MenuItem) it2.next();
final DocumentTOCItem documentTOCItem = new DocumentTOCItem();
documentTOCItem.setName( item.getName() );
documentTOCItem.setRef( item.getHref() );
toc.addItem( documentTOCItem );
}
}
}
return toc;
}
/**
* Extract meta information from a MavenProject.
*
* @param project a MavenProject. May be null.
* @param date the date to use in meta. May be null.
*
* @return a DocumentMeta object. Always non-null.
*/
private static DocumentMeta getDocumentMeta( MavenProject project, Date date )
{
final DocumentMeta meta = new DocumentMeta();
meta.setAuthors( getAuthors( project ) );
meta.setCreationDate( date );
meta.setCreator( System.getProperty( "user.name" ) );
meta.setDate( date );
meta.setDescription( project == null ? null : project.getDescription() );
//meta.setGenerator( generator );
meta.setInitialCreator( System.getProperty( "user.name" ) );
//meta.setLanguage( locale == null ? null : locale.getLanguage() );
//meta.setPageSize( pageSize );
meta.setSubject( getProjectName( project ) );
meta.setTitle( getProjectName( project ) );
return meta;
}
/**
* Extract information for a DocumentCover from a MavenProject.
*
* @param project a MavenProject. May be null.
* @param date the cover date. May be null.
*
* @return a DocumentCover object. Always non-null.
*/
private static DocumentCover getDocumentCover( MavenProject project, Date date )
{
final DocumentCover cover = new DocumentCover();
cover.setAuthors( getAuthors( project ) );
//cover.setCompanyLogo( companyLogo );
cover.setCompanyName( getProjectOrganizationName( project ) );
cover.setCoverDate( date );
cover.setCoverSubTitle( project == null ? null : "v. " + project.getVersion() );
cover.setCoverTitle( getProjectName( project ) );
//cover.setCoverType( type );
cover.setCoverVersion( project == null ? null : project.getVersion() );
//cover.setProjectLogo( projectLogo );
cover.setProjectName( getProjectName( project ) );
return cover;
}
/**
* Wrap the list of project {@link Developer} to a list of {@link DocumentAuthor}.
*
* @param project the MavenProject to extract the authors from.
* @return a list of DocumentAuthors from the project developers.
* Returns null if project is null or contains no developers.
*/
private static List getAuthors( MavenProject project )
{
if ( project == null || project.getDevelopers() == null )
{
return null;
}
final List ret = new ArrayList( 4 );
for ( final Iterator it = project.getDevelopers().iterator(); it.hasNext(); )
{
final Developer developer = (Developer) it.next();
final DocumentAuthor author = new DocumentAuthor();
author.setName( developer.getName() );
author.setEmail( developer.getEmail() );
author.setCompanyName( developer.getOrganization() );
StringBuilder roles = null;
for ( final Iterator it2 = developer.getRoles().iterator(); it2.hasNext(); )
{
final String role = (String) it2.next();
if ( roles == null )
{
roles = new StringBuilder( 32 );
}
roles.append( role );
if ( it2.hasNext() )
{
roles.append( ',' ).append( ' ' );
}
}
if ( roles != null )
{
author.setPosition( roles.toString() );
}
ret.add( author );
}
return ret;
}
/**
* @param project the MavenProject to extract the project organization name from.
* @return the project organization name if not empty, or the current System user name otherwise.
*/
private static String getProjectOrganizationName( MavenProject project )
{
if ( project != null && project.getOrganization() != null
&& StringUtils.isNotEmpty( project.getOrganization().getName() ) )
{
return project.getOrganization().getName();
}
return System.getProperty( "user.name" );
}
/**
* Extract the name of the project.
*
* @param project the MavenProject to extract the project name from.
* @return the project name, or the project groupId and artifactId if
* the project name is empty, or null if project is null.
*/
private static String getProjectName( MavenProject project )
{
if ( project == null )
{
return null;
}
if ( StringUtils.isEmpty( project.getName() ) )
{
return project.getGroupId() + ":" + project.getArtifactId();
}
return project.getName();
}
/**
* Extract the encoding.
*
* @param project the MavenProject to extract the encoding name from.
* @return the project encoding if defined, or UTF-8 otherwise, or null if project is null.
*/
private static String getProjectModelEncoding( MavenProject project )
{
if ( project == null )
{
return null;
}
String encoding = project.getModel().getModelEncoding();
// Workaround for MNG-4289
XmlStreamReader reader = null;
try
{
reader = new XmlStreamReader( project.getFile() );
encoding = reader.getEncoding();
}
catch ( IOException e )
{
// nop
}
finally
{
IOUtil.close( reader );
}
if ( StringUtils.isEmpty( encoding ) )
{
return "UTF-8";
}
return encoding;
}
}