SubDirInfoTask.java 3.03 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
/**
 * $RCSfile$
 * $Revision$
 * $Date$
 *
 * Copyright (C) 2004 Jive Software. All rights reserved.
 *
 * This software is published under the terms of the GNU Public License (GPL),
 * a copy of which is included in this distribution.
 */

package org.jivesoftware.ant;

import org.apache.tools.ant.Task;
import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Project;

import java.io.File;

/**
 * A simple ant task to return the sub directories of a given dir as a comma delimited string.
 *
 * This class does not need jdk 1.5 to compile.
 */
public class SubDirInfoTask extends Task {

    public static final String DEFAULT_DELIM = ",";

    private File dir;
    private String property;
    private String delimiter;
    private String ifexists;
    private String except;

    public SubDirInfoTask() {
    }

    public File getDir() {
        return dir;
    }

    public void setDir(File dir) {
        this.dir = dir;
    }

    public String getProperty() {
        return property;
    }

    public void setProperty(String property) {
        this.property = property;
    }

    public String getDelimiter() {
        if (delimiter == null) {
            return DEFAULT_DELIM;
        }
        return delimiter;
    }

    public void setDelimiter(String delimiter) {
        this.delimiter = delimiter;
    }

    public String getIfexists() {
        return ifexists;
    }

    public void setIfexists(String ifexists) {
        this.ifexists = ifexists;
    }

    public String getExcept() {
        return except;
    }

    public void setExcept(String except) {
        this.except = except;
    }

    public void execute() throws BuildException {
        // Get the siblings of the given directory, add sub directory names to the property
        File[] subdirs = dir.listFiles();
        StringBuffer buf = new StringBuffer();
        String value = null;
        String sep = "";
Bill Lynch's avatar
Bill Lynch committed
87 88 89 90 91 92 93 94 95 96 97 98
        if (subdirs != null) {
            for (int i=0; i<subdirs.length; i++) {
                File subdir = subdirs[i];
                boolean add = false;
                if (subdir.isDirectory()) {
                    if (getIfexists() != null) {
                        File file = new File(subdir, getIfexists());
                        if (file.exists()) {
                            add = true;
                        }
                    }
                    else {
99 100 101
                        add = true;
                    }
                }
Bill Lynch's avatar
Bill Lynch committed
102 103 104
                if (add && !subdir.getName().equals(except)) {
                    buf.append(sep).append(subdir.getName());
                    sep = getDelimiter();
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121
                }
            }
        }
        if (buf.length() > 0) {
            value = buf.toString();
        }
        if (value == null) {
            log("No tokens found.", Project.MSG_DEBUG);
        }
        else {
            log("Setting property '" + property + "' to " + value, Project.MSG_DEBUG);
            if (buf.length() >= 0) {
                getProject().setProperty(property, value);
            }
        }
    }
}