source: main/trunk/ant-tasks/src/org/greenstone/anttasks/PatternSetToFile.java@ 23242

package org.greenstone.anttasks;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.util.ArrayList;

import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.DirectoryScanner;
import org.apache.tools.ant.Task;
import org.apache.tools.ant.types.FileSet;
import org.apache.tools.ant.types.PatternSet;

/**
 * PatternSetToFile is an Ant task used to take an Ant PatternSet and output to a
 * file all the files that it encompasses
 * @author Sam McIntosh
 */
public class PatternSetToFile extends Task
{
    //Sets whether or not only the top-level files and folders
    //are stored in the output file
    public boolean _toplevelonly = false;
    
    //The file the set will be written to
    public File _outfile = null;

    //The directory to check the PatternSet within
    public File _dir = null;
    
    //An array of the child PatternSets of this task
    public ArrayList _sets = new ArrayList();

    /**
     * This function performs the main task
     */
    public void execute() throws BuildException
    {
        try
        {
            System.out.println("Creating " + _outfile.getAbsolutePath());
            BufferedWriter out = new BufferedWriter(new FileWriter(_outfile));
            for(int i = 0; i < _sets.size(); i++)
            {
                PatternSet set = (PatternSet)_sets.get(i);
                FileSet fs = new FileSet();
                fs.setDir(_dir);
                PatternSet ps = fs.createPatternSet();
                ps.append(set, this.getProject());
                
                DirectoryScanner ds = fs.getDirectoryScanner(this.getProject());
                
                String[] includedFiles = ds.getIncludedFiles();
                
                if(_toplevelonly)
                {
                    ArrayList toplevel = new ArrayList();
                    
                    for(int j = 0; j < includedFiles.length; j++)
                    {
                        int separatorIndex = includedFiles[j].indexOf(File.separator);
                        
                        if ( separatorIndex == -1 && includedFiles[j].length() > 0 && !toplevel.contains(toplevel)) {
                            toplevel.add(includedFiles[j]);
                        }
                        else if ( separatorIndex > -1 && includedFiles[j].length() > 0) {
                            String path = includedFiles[j].substring(0, separatorIndex);
                            if ( !toplevel.contains(path) ) {
                                toplevel.add(path);
                            }
                        }
                    }
                    includedFiles = (String[])toplevel.toArray(new String[0]);
                }
                
                for (int j = 0; j < includedFiles.length; j++)
                {
                    out.write(includedFiles[j] + "\n");
                }
            }
            out.close();
        }
        catch(Exception ex)
        {
            ex.printStackTrace();
        }
    }

    public void setToplevelonly(boolean toplevelonly)
    {
        _toplevelonly = toplevelonly;
    }

    public void setOutfile(File outfile)
    {
        _outfile = outfile;
    }

    public void setDir(File dir)
    {
        _dir = dir;
    }
    
    public void addConfiguredPatternset(PatternSet patternset) 
    {
        if (patternset == null)
        {
            return;
        }
        
        _sets.add(patternset);
    }
}