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();
     }
 }