Mercurial > hg4j
diff src/org/tmatesoft/hg/util/FileWalker.java @ 229:1ec6b327a6ac
Scope for status reworked: explicit files or a general matcher
author | Artem Tikhomirov <tikhomirov.artem@gmail.com> |
---|---|
date | Tue, 31 May 2011 05:23:07 +0200 |
parents | fffe4f882248 |
children | ed6b74a58c66 |
line wrap: on
line diff
--- a/src/org/tmatesoft/hg/util/FileWalker.java Fri May 27 03:01:26 2011 +0200 +++ b/src/org/tmatesoft/hg/util/FileWalker.java Tue May 31 05:23:07 2011 +0200 @@ -31,14 +31,28 @@ private final Path.Source pathHelper; private final LinkedList<File> dirQueue; private final LinkedList<File> fileQueue; + private final Path.Matcher scope; private File nextFile; private Path nextPath; public FileWalker(File dir, Path.Source pathFactory) { + this(dir, pathFactory, null); + } + + /** + * + * @param dir + * @param pathFactory + * @param scopeMatcher - this matcher shall be capable to tell not only files of interest, but + * also whether directories shall be traversed or not (Paths it gets in {@link Path.Matcher#accept(Path)} may + * point to directories) + */ + public FileWalker(File dir, Path.Source pathFactory, Path.Matcher scopeMatcher) { startDir = dir; pathHelper = pathFactory; dirQueue = new LinkedList<File>(); fileQueue = new LinkedList<File>(); + scope = scopeMatcher; reset(); } @@ -71,7 +85,8 @@ } public boolean inScope(Path file) { - return true; // no limits, all files are of interest + /* by default, no limits, all files are of interest */ + return scope == null ? true : scope.accept(file); } // returns non-null @@ -91,8 +106,13 @@ while (!dirQueue.isEmpty()) { File dir = dirQueue.removeFirst(); for (File f : listFiles(dir)) { - if (f.isDirectory()) { - if (!".hg".equals(f.getName())) { + final boolean isDir = f.isDirectory(); + Path path = pathHelper.path(isDir ? ensureTrailingSlash(f.getPath()) : f.getPath()); + if (!inScope(path)) { + continue; + } + if (isDir) { + if (!".hg/".equals(path.toString())) { dirQueue.addLast(f); } } else { @@ -104,4 +124,17 @@ } return !fileQueue.isEmpty(); } + + private static String ensureTrailingSlash(String dirName) { + if (dirName.length() > 0) { + char last = dirName.charAt(dirName.length() - 1); + if (last == '/' || last == File.separatorChar) { + return dirName; + } + // if path already has platform-specific separator (which, BTW, it shall, according to File#getPath), + // add similar, otherwise use our default. + return dirName.indexOf(File.separatorChar) != -1 ? dirName.concat(File.separator) : dirName.concat("/"); + } + return dirName; + } }