package contrib;
import java.io.*;
import javax.swing.filechooser.FileFilter;
/**
 * A file filter which checks the file extension for a known type.
 * 
 * The default file extension is 'xml' but can be set via the
 * alternate constructor.
 * 
 * Implements FilenameFilter
 * Extends FileFilter
 */
public class FileExtensionFilter extends FileFilter implements FilenameFilter {

	private String dotExtension = ".xml";

	/**
	 * Constructor which defaults to 'xml'
	 */
	public FileExtensionFilter() {
		super();
	}

	/**
	 * Constructor for a specific file extension
	 * 
	 * @param the 3 character file extension
	 */
	public FileExtensionFilter(String extension) {
		super();
		dotExtension = "." + extension;
	}

	/**
	 * @see FileFilter#getDescription()
	 */
	public String getDescription() {
		return "*" + dotExtension;
	}

	/**
	 * @see FileFilter#accept(File)
	 */
	public boolean accept(File file) {
		if (file.isDirectory()) {
			return true;
		} else {
			String filename = file.getName();
			return filename.endsWith(dotExtension);
		}
	}

	/**
	 * @see FilenameFilter#accept(File, String)
	 */
	public boolean accept(File newFile, String name) {
		return name.endsWith(dotExtension);
	}
}