JFileChooser showing files and directory

[Gulshan Singh]

The following code is used to browse for files in my program:

   public class BrowseButtonListener implements ActionListener {
        public void actionPerformed(ActionEvent e){
            fc = new JFileChooser();
            //fc.setFileSelectionMode(JFileChooser.FILES_AND_DIRECTORIES);
            //fc.setFileFilter(new ExtensionFileFilter());
            int returnVal = fc.showOpenDialog(frame);
            if (returnVal == JFileChooser.APPROVE_OPTION) {
                if (e.getSource() == browse2)
                    answerfile.setText(fc.getSelectedFile().getAbsolutePath());
                if (e.getSource() == browse1)
                    questionfile.setText(fc.getSelectedFile().getAbsolutePath());
            }
        }
    }

Then I wanted to limit whatever comes up to directories and text files (files with a .txt extension), so I uncommented the the lines in the code that are commented above, and add this class:

class ExtensionFileFilter extends FileFilter {
    
    public boolean accept(File file) {
        String filename = file.getName();
        return filename.endsWith(".txt");
    }

    public String getDescription() {
        return "*.txt";
    }
}

This filtered all the files except for files with a .txt extension, just like I wanted, but it got rid of the directories too.. How do I change this so that it only shows file with a .txt extension and directories?

[Paul Clapham]

return filename.endsWith(".txt") || file.isDirectory();

[Gulshan Singh]

Paul Clapham wrote:

return filename.endsWith(".txt") || file.isDirectory();

Thanks, works now :)