src/main/java/org/apache/maven/artifact/ant/util/AntBuildWriter.java

package org.apache.maven.artifact.ant.util;

/*
 * 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.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;

import org.apache.tools.ant.DirectoryScanner;
import org.apache.tools.ant.types.FileList;
import org.apache.tools.ant.types.FileSet;
import org.apache.tools.ant.types.Path;
import org.codehaus.plexus.util.IOUtil;
import org.codehaus.plexus.util.StringUtils;
import org.codehaus.plexus.util.xml.PrettyPrintXMLWriter;
import org.codehaus.plexus.util.xml.XMLWriter;
import org.codehaus.plexus.util.xml.XmlWriterUtil;

/**
 * Utility class for writing an Ant build file.
 */
public class AntBuildWriter
{
    public static final String DEFAULT_FILE_ENCODING = "UTF-8";

    /**
     * The default line indenter
     */
    protected static final int DEFAULT_INDENTATION_SIZE = XmlWriterUtil.DEFAULT_INDENTATION_SIZE;

    private XMLWriter writer;

    private OutputStreamWriter outputStreamWriter;

    /**
     * Open an Ant build file for writing. Opens the file and prints the opening project tag.
     *
     * @param buildFile
     * @param name
     * @param defaultTarget
     * @throws IOException
     */
    public void openAntBuild( File dependenciesBuildFile, String name, String defaultTarget )
        throws IOException
    {
        String encoding = DEFAULT_FILE_ENCODING;

        if ( ! dependenciesBuildFile.getParentFile().exists() )
        {
            dependenciesBuildFile.getParentFile().mkdirs();
        }
        outputStreamWriter = new OutputStreamWriter( new FileOutputStream( dependenciesBuildFile ), encoding );

        writer =
            new PrettyPrintXMLWriter( outputStreamWriter, StringUtils.repeat( " ", DEFAULT_INDENTATION_SIZE ),
                                      encoding, null );
        writer.startElement( "project" );
        writer.addAttribute( "name", name );
        writer.addAttribute( "default", defaultTarget );

        XmlWriterUtil.writeLineBreak( writer );
    }

    /**
     * Close the ant build writer
     *
     * @throws IOException
     */
    public void closeAntBuild()
        throws IOException
    {
        writer.endElement();

        XmlWriterUtil.writeLineBreak( writer );

        IOUtil.close( outputStreamWriter );
    }

    /**
     * Open a target tag
     *
     * @param targetName
     * @throws IOException
     */
    public void openTarget( String targetName )
        throws IOException
    {
        writer.startElement( "target" );
        writer.addAttribute( "name", targetName );
    }

    /**
     * Close a tag.
     *
     * @throws IOException
     */
    public void closeTarget()
        throws IOException
    {
        writer.endElement();
    }

    /**
     * Write an Ant fileset
     *
     * @param fileSet
     * @param id
     */
    public void writeFileSet( FileSet fileSet, String id )
    {
        writer.startElement( "fileset" );

        if ( id != null )
        {
            writer.addAttribute( "id", id );
        }

        File dir = fileSet.getDir( fileSet.getProject() );
        writer.addAttribute( "dir", dir.getAbsolutePath() );

        DirectoryScanner scanner = fileSet.getDirectoryScanner( fileSet.getProject() );
        scanner.scan();
        String[] files = scanner.getIncludedFiles();

        for ( int i = 0; i < files.length; ++i )
        {
            writer.startElement( "include" );
            writer.addAttribute( "name", files[i] );
            writer.endElement();
        }

        writer.endElement();
    }

    /**
     * Write an ant property
     *
     * @param name
     * @param value
     */
    public void writeProperty( String name, String value )
    {
        writer.startElement( "property" );

        writer.addAttribute( "name", name );
        writer.addAttribute( "value", value );

        writer.endElement();
    }

    /**
     * Write an Ant echo task
     *
     * @param message
     */
    public void writeEcho( String message )
    {
        writer.startElement( "echo" );

        writer.addAttribute( "level", "info" );
        writer.addAttribute( "message", message );

        writer.endElement();
    }

    /**
     * Write an Ant file list
     *
     * @param fileList
     * @param id
     */
    public void writeFileList( FileList fileList, String id )
    {
        writer.startElement( "filelist" );
        writer.addAttribute( "id", id );
        File dir = fileList.getDir( fileList.getProject() );
        writer.addAttribute( "dir", dir.getAbsolutePath() );

        String[] files = fileList.getFiles( fileList.getProject() );
        for ( int i = 0; i < files.length; ++i )
        {
            writer.startElement( "file" );
            writer.addAttribute( "name", files[i] );
            writer.endElement();
        }
        writer.endElement();
    }

    /**
     * Write a path.
     *
     * @param path
     * @param pathId
     */
    public void writePath( Path path, String pathId )
    {
        writer.startElement( "path" );
        writer.addAttribute( "id", pathId );
        String[] paths = path.list();
        for ( int i = 0; i < paths.length; ++i )
        {
            writer.startElement( "pathelement" );
            writer.addAttribute( "path", paths[i] );
            writer.endElement();
        }
        writer.endElement();
    }
}