public class Directory {

       

        public static File[] local(File dir,final String regex) {

                return dir.listFiles(new FilenameFilter() {

                        private Pattern pattern = Pattern.compile(regex);

                        @Override

                        public boolean accept(File dir, String name) {

                                // TODO Auto-generated method stub

                                return pattern.matcher(new File(name).getName()).matches();

                        }

                });

        }

        @SuppressWarnings("static-access")

        public static void main(String[] args) {

                if(args.length == 0) {

                        System.out.println("===============AllFile=============");

                        System.out.println(new TreeInfo().walk(".",RegexString.SUFFIX_ALL));//,RegexString.SUFFIX_ALL表示正则表达式

                        System.out.println("===============.java===============");

                        System.out.println(new TreeInfo().walk(".",RegexString.SUFFIX_JAVA));

                        System.out.println("===============.class===============");

                        System.out.println(new TreeInfo().walk(".",RegexString.SUFFIX_CLASS));

                         

                }else {

                        for(String arg : args) {

                                System.out.println(new TreeInfo().walk(arg,".*"));

                        }

                }

        }

        public static class TreeInfo implements Iterable<File>{

                public List<File> files = new ArrayList<File>();//文件

                public List<File> dirs = new ArrayList<File>();//目录

               

                @Override

                public Iterator<File> iterator() {

                        // TODO Auto-generated method stub

                        return files.iterator();

                }

                void addAll(TreeInfo other) {//加入集合

                        files.addAll(other.files);

                        dirs.addAll(other.dirs);

                }

                public String toString() {

                        return "dirs: " + PPrint.pformat(dirs) + "\n\nfiles: " + PPrint.pformat(files);

                }

                public static TreeInfo walk(String start,String regex) {//文件目录，正则表达式（文件格式）

                        return recurseDirs(new File(start),regex);

                }

                private static TreeInfo recurseDirs(File startDir, String regex) {

                        // TODO Auto-generated method stub

                        TreeInfo result = new TreeInfo();

                        for (File item : startDir.listFiles()) {

                                if(item.isDirectory()) {//判断是否是目录

                                        result.dirs.add(item);

                                        result.addAll(recurseDirs(item, regex));//递归调用自己

                                }else {

                                        if(item.getName().matches(regex))//判断符合正则

                                                result.files.add(item);

                                }

                        }

                        return result;

                }

        }