Home | History | Annotate | Download | only in io 
      1 /*
      2  * ProGuard -- shrinking, optimization, obfuscation, and preverification
      3  *             of Java bytecode.
      4  *
      5  * Copyright (c) 2002-2014 Eric Lafortune (eric (at) graphics.cornell.edu)
      6  *
      7  * This program is free software; you can redistribute it and/or modify it
      8  * under the terms of the GNU General Public License as published by the Free
      9  * Software Foundation; either version 2 of the License, or (at your option)
     10  * any later version.
     11  *
     12  * This program is distributed in the hope that it will be useful, but WITHOUT
     13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
     14  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
     15  * more details.
     16  *
     17  * You should have received a copy of the GNU General Public License along
     18  * with this program; if not, write to the Free Software Foundation, Inc.,
     19  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
     20  */
     21 package proguard.io;
     22 
     23 import java.io.*;
     24 
     25 
     26 /**
     27  * This class can read a given file or directory, recursively, applying a given
     28  * DataEntryReader to all files it comes across.
     29  *
     30  * @author Eric Lafortune
     31  */
     32 public class DirectoryPump implements DataEntryPump
     33 {
     34     private final File directory;
     35 
     36 
     37     public DirectoryPump(File directory)
     38     {
     39         this.directory = directory;
     40     }
     41 
     42 
     43     // Implementations for DataEntryPump.
     44 
     45     public void pumpDataEntries(DataEntryReader dataEntryReader)
     46     throws IOException
     47     {
     48         if (!directory.exists())
     49         {
     50             throw new IOException("No such file or directory");
     51         }
     52 
     53         readFiles(directory, dataEntryReader);
     54     }
     55 
     56 
     57     /**
     58      * Reads the given subdirectory recursively, applying the given DataEntryReader
     59      * to all files that are encountered.
     60      */
     61     private void readFiles(File file, DataEntryReader dataEntryReader)
     62     throws IOException
     63     {
     64         // Pass the file data entry to the reader.
     65         dataEntryReader.read(new FileDataEntry(directory, file));
     66 
     67         if (file.isDirectory())
     68         {
     69             // Recurse into the subdirectory.
     70             File[] files = file.listFiles();
     71 
     72             for (int index = 0; index < files.length; index++)
     73             {
     74                 readFiles(files[index], dataEntryReader);
     75             }
     76         }
     77     }
     78 }
     79